Method Inject

Source/Features/ExportIndices/Backend/Source/Feature.cpp:77–151

}

void ExportIndicesFeature::Inject(IL::Program &program, const MessageStreamView<> &specialization) {
    // Options
    const SetInstrumentationConfigMessage config = CollapseOrDefault<SetInstrumentationConfigMessage>(specialization);

    // Visit all instructions
    IL::VisitUserInstructions(program, [&](IL::VisitContext& context, IL::BasicBlock::Iterator it) -> IL::BasicBlock::Iterator {
        // Instruction of interest?
        IL::ID outputIndex;
        switch (it->opCode) {
            default:
                return it;
                // TODO: SPIRV StoreOutput -> vertex and primitive
                // TODO: DXIL EmitIndices
            case IL::OpCode::StoreVertexOutput:
                outputIndex = it->As<IL::StoreVertexOutputInstruction>()->vertexIndex;
                break;
            case IL::OpCode::StorePrimitiveOutput:
                outputIndex = it->As<IL::StorePrimitiveOutputInstruction>()->primitiveIndex;
                break;
        }

        // Instrumentation Segmentation
        //
        //             BEFORE                                 AFTER
        //
        //   ┌─────┬─────────────┬───────┐      ┌─────┐                   ┌─────────────┬──────┐
        //   │     │             │       │      │     │        OK         │             │      │
        //   │ Pre │ Instruction │ Post  │      │ Pre ├───────────────────┤ Instruction │ Post │
        //   │     │             │       │      │     │                   │   [RESUME]  │      │
        //   └─────┴─────────────┴───────┘      └──┬──┘                   └──────┬──────┴──────┘
        //                                         │    ┌────────────┐           │
        //                                     INV │    │            │           │
        //                                         └────┤ noThreadID ├───────────┘
        //                                              │            │
        //                                              └────────────┘

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

        uint64_t line = sguidHost->GetMapping(sguid).line;
        std::string_view lineCode = sguidHost->GetSource(sguid);

        // Allocate resume
        IL::BasicBlock* resumeBlock = context.function.GetBasicBlocks().AllocBlock();

        // Split this basic block, move all instructions post and including the instrumented instruction to resume
        // ! iterator invalidated
        auto instr = context.basicBlock.Split(resumeBlock, it);

        // Perform instrumentation check
        IL::Emitter<> pre(program, context.basicBlock);

        // Failure conditions: output index is not thread ID
        IL::ID isNotThreadIndex = pre.NotEqual(outputIndex, pre.KernelValue(Backend::IL::KernelValue::FlattenedLocalThreadID));

        // isNotThreadIndex block
        IL::Emitter<> noTid(program, *context.function.GetBasicBlocks().AllocBlock());
        noTid.AddBlockFlag(BasicBlockFlag::NoInstrumentation);