Method OutputFunction

hlslparser/src/MSLGenerator.cpp:1028–1119

}

void MSLGenerator::OutputFunction(int indent, HLSLFunction* function)
{
    const char* functionName = function->name;
    const char* returnTypeName = GetTypeName(function->returnType, /*exactType=*/false);

    // Declare output tuple.
    if (function->numOutputArguments > 0) {
        returnTypeName = m_tree->AddStringFormat("%s_out%d", functionName, function->line); // @@ Find a better way to generate unique name.

        m_writer.BeginLine(indent, function->fileName, function->line);
        m_writer.Write("struct %s { ", returnTypeName);
        m_writer.EndLine();

        if (function->returnType.baseType != HLSLBaseType_Void) {
            m_writer.BeginLine(indent + 1, function->fileName, function->line);
            OutputDeclaration(function->returnType, "__result", /*defaultValue=*/NULL, /*isRef=*/false, /*isConst=*/false);
            m_writer.EndLine(";");
        }

        HLSLArgument* argument = function->argument;
        while (argument != NULL) {
            if (argument->modifier == HLSLArgumentModifier_Out || argument->modifier == HLSLArgumentModifier_Inout) {
                m_writer.BeginLine(indent + 1, function->fileName, function->line);
                OutputDeclaration(argument->type, argument->name, /*defaultValue=*/NULL, /*isRef=*/false, /*isConst=*/false);
                m_writer.EndLine(";");
            }
            argument = argument->nextArgument;
        }

        m_writer.WriteLine(indent, "};");

        // Create unique function name to avoid collision with overloads and different return types.
        m_writer.BeginLine(indent, function->fileName, function->line);
        m_writer.Write("%s %s_%d(", returnTypeName, functionName, function->line);
    }
    else {
        m_writer.BeginLine(indent, function->fileName, function->line);
        m_writer.Write("%s %s(", returnTypeName, functionName);
    }

    OutputArguments(function->argument);

    m_writer.EndLine(") {");
    m_currentFunction = function;

    // Local declarations for output arguments.
    HLSLArgument* argument = function->argument;
    while (argument != NULL) {
        if (argument->modifier == HLSLArgumentModifier_Out) {
            m_writer.BeginLine(indent + 1, function->fileName, function->line);
            OutputDeclaration(argument->type, argument->name, /*defaultValue=*/NULL, /*isRef=*/false, /*isConst=*/false);
            m_writer.EndLine(";");
        }
        argument = argument->nextArgument;
    }

    OutputStatements(indent + 1, function->statement); // @@ Modify return statements if function has multiple output arguments!