Method Prepass

hlslparser/src/MSLGenerator.cpp:108–296

}

void MSLGenerator::Prepass(HLSLTree* tree, HLSLTarget target, HLSLFunction* entryFunction)
{
    // Hide unused arguments. @@ It would be good to do this in the other generators too.

    // PruneTree resets hidden flags to true, then marks visible elements
    // based on whether entry point visits them.
    PruneTree(tree, entryFunction->name); // Note: takes second entry

    // This sorts tree by type, but keeps ordering
    SortTree(tree);

    // This strips any unused inputs to the entry point function
    HideUnusedArguments(entryFunction);

    // Note sure if/where to add these calls.  Just wanted to point
    // out that nothing is calling them, but could be useful.
    FlattenExpressions(tree);

    HLSLRoot* root = tree->GetRoot();
    HLSLStatement* statement = root->statement;
    ASSERT(m_firstClassArgument == NULL);

    //HLSLType samplerType(HLSLBaseType_Sampler);

    int nextTextureRegister = 0;
    int nextSamplerRegister = 0;
    int nextBufferRegister = 0;

    while (statement != NULL) {
        if (statement->hidden) {
            statement = statement->nextStatement;
            continue;
        }

        if (statement->nodeType == HLSLNodeType_Declaration) {
            HLSLDeclaration* declaration = (HLSLDeclaration*)statement;

            if (IsTextureType(declaration->type)) {
                const char* textureName = declaration->name;

                int textureRegister = ParseRegister(declaration->registerName, nextTextureRegister);
                const char* textureRegisterName = m_tree->AddStringFormat("texture(%d)", textureRegister);

                if (declaration->type.addressSpace == HLSLAddressSpace_Undefined)
                    declaration->type.addressSpace = HLSLAddressSpace_Device;

                AddClassArgument(new ClassArgument(textureName, declaration->type, textureRegisterName, true));
            }
            else if (IsSamplerType(declaration->type)) {
                const char* samplerName = declaration->name;

                int samplerRegister = ParseRegister(declaration->registerName, nextSamplerRegister);
                const char* samplerRegisterName = m_tree->AddStringFormat("sampler(%d)", samplerRegister);

                if (declaration->type.addressSpace == HLSLAddressSpace_Undefined)
                    declaration->type.addressSpace = HLSLAddressSpace_Device;

                AddClassArgument(new ClassArgument(samplerName, declaration->type, samplerRegisterName, true));