Method Inject

Source/Features/Loop/Backend/Source/Feature.cpp:147–417

}

void LoopFeature::Inject(IL::Program &program, const MessageStreamView<> &specialization) {
    // Options
    const SetLoopInstrumentationConfigMessage config = FindOrDefault(specialization, SetLoopInstrumentationConfigMessage {
        .useIterationLimits = true,
        .iterationLimit = 32'000,
        .atomicIterationInterval = 256
    });

    // Get constant literals
    IL::ID interval      = program.GetConstants().UInt(config.atomicIterationInterval)->id;
    IL::ID maxIterations = program.GetConstants().UInt(config.iterationLimit)->id;
    
    // Get the data ids
    IL::ID terminationBufferDataID     = program.GetShaderDataMap().Get(terminationBufferID)->id;
    IL::ID terminationAllocationDataID = program.GetShaderDataMap().Get(terminationAllocationID)->id;

    // Get the program capabilities
    const IL::CapabilityTable &capabilityTable = program.GetCapabilityTable();

    // Allocate all the counters
    LoopCounterMap functionCounters;
    InjectLoopCounters(program, functionCounters);

    // If the program has structured control flow, we can take quite a few liberties in instrumentation
    if (capabilityTable.hasControlFlow) {
        // Visit all instructions
        IL::VisitUserInstructions(program, [&](IL::VisitContext &context, IL::BasicBlock::Iterator it) -> IL::BasicBlock::Iterator {
            IL::BranchControlFlow controlFlow;

            // Must have a continue based, i.e. loop styled, control flow
            if (!Backend::IL::GetControlFlow(it, controlFlow) || controlFlow._continue == IL::InvalidID) {
                return it;
            }

            // All basic blocks
            IL::BasicBlockList& basicBlocks = context.function.GetBasicBlocks();

            // Bind the SGUID
            ShaderSGUID sguid = sguidHost ? sguidHost->Bind(program, it) : InvalidShaderSGUID;

            // Get merge block
            // On termination this is the block we reach
            IL::BasicBlock* mergeBlock = context.function.GetBasicBlocks().GetBlock(controlFlow.merge);
            
            // Allocate blocks
            IL::BasicBlock *postEntry        = context.function.GetBasicBlocks().AllocBlock();
            IL::BasicBlock *terminationBlock = context.function.GetBasicBlocks().AllocBlock();
            IL::BasicBlock* atomicBlock      = context.function.GetBasicBlocks().AllocBlock();
            IL::BasicBlock* atomicMergeBlock = context.function.GetBasicBlocks().AllocBlock();

            // The selection merge target, typically this is the post-entry (i.e. the split body entry point block),
            // however, if the continue block is the body, then we need to split things further.
            IL::BasicBlock* selectionMerge = postEntry;

            // Expected entry point
            IL::BasicBlock* entryBlock = nullptr;

            // Determine entry point