private static void GenerateTechnique(EffectCodeGenerator generator, TechniqueSymbol technique,
bool generateVertexShader)
{
ScriptTextWriter writer = generator.Writer;
writer.WriteLine("// -------- technique {0} --------", technique.Name);
writer.WriteLine("technique {0}", technique.Name);
writer.WriteLine("{");
foreach (TechniquePassSymbol pass in technique.Passes)
{
GeneratePass(generator, technique, pass, generateVertexShader);
}
writer.WriteLine("};");
}
private static void GeneratePass(EffectCodeGenerator generator, TechniqueSymbol technique, TechniquePassSymbol pass,
bool generateVertexShader)
{
ScriptTextWriter writer = generator.Writer;
writer.WriteLine("\tpass {0}", pass.Name);
writer.WriteLine("\t{");
if (generateVertexShader)
{
writer.WriteLine("\t\tVertexShader = compile vs_{0} {1}_{2}_vs();",
generator.Context.TargetShaderProfile.GetDescription(),
technique.Name, pass.Name);
}
writer.WriteLine("\t\tPixelShader = compile ps_{0} {1}_{2}_ps();",
generator.Context.TargetShaderProfile.GetDescription(),
technique.Name, pass.Name);
writer.WriteLine("\t}");
}
private static void WriteVertexShader(EffectCodeGenerator generator, StitchedFragmentSymbol fragment,
string uniquePassName, Dictionary <string, List <string> > exports)
{
ScriptTextWriter writer = generator.Writer;
writer.WriteLine("// -------- vertex shader {0} --------", fragment.UniqueName);
ShaderCodeBlockNode shader = fragment.FragmentNode.VertexShaders.GetCodeBlock(generator.Context.TargetShaderProfile);
if (shader != null)
{
string shaderCode = StitchedEffectPreProcessor.PreProcessCodeBlock(fragment.UniqueName, shader, exports);
shaderCode = ReplaceOutputCalls(shaderCode, fragment.UniqueName);
WriteShaderCode(generator, fragment, shaderCode, "VERTEXINPUT", "VERTEXOUTPUT", "vs", uniquePassName);
}
else
{
// Need to auto-generate vertex shader. Simply pass through all vertex inputs.
writer.WriteLine("void {0}_{1}_vs({0}_{1}_VERTEXINPUT input, inout {0}_VERTEXOUTPUT output)", uniquePassName, fragment.UniqueName);
writer.WriteLine("{");
writer.Write(GetVertexPassThroughCode(fragment));
writer.WriteLine("}");
}
writer.WriteLine();
writer.WriteLine();
}
private static void WriteVertexInputStructures(EffectCodeGenerator generator,
TechniqueSymbol technique, TechniquePassSymbol pass, string uniquePassName)
{
ScriptTextWriter writer = generator.Writer;
writer.WriteLine("// -------- vertex input structures --------");
SemanticGenerator semanticGenerator = new SemanticGenerator("TEXCOORD");
pass.ForEachFragment(generator, (g, f, s) => WriteVertexInputStructure(g, f, s, semanticGenerator));
writer.WriteLine("struct {0}_VERTEXINPUT", uniquePassName);
writer.WriteLine("{");
pass.ForEachFragment(generator, (g, f, s) => g.Writer.WriteLine("\t{0}_{1}_VERTEXINPUT {1};", uniquePassName, f.UniqueName));
writer.WriteLine("};");
writer.WriteLine();
writer.WriteLine("static {0}_VERTEXINPUT gVertexInput_{0};", uniquePassName);
writer.WriteLine();
}
private static void WriteVertexOutputStructs(EffectCodeGenerator generator,
TechniqueSymbol technique, TechniquePassSymbol pass, string uniquePassName)
{
ScriptTextWriter writer = generator.Writer;
writer.WriteLine("// -------- vertex output structures --------");
writer.WriteLine("struct {0}_VERTEXOUTPUT", uniquePassName);
writer.WriteLine("{");
writer.WriteLine("\tfloat4 position : POSITION;");
pass.ForEachFragment(generator, (g, f, s) =>
{
if (f.FragmentNode.Interpolators != null && f.FragmentNode.Interpolators.VariableDeclarations.Any())
{
g.Writer.WriteLine("\t{0}_{1}_PIXELINPUT {1};", uniquePassName, f.UniqueName);
}
});
writer.WriteLine("};");
writer.WriteLine();
}
private static void WritePixelOutputStructs(EffectCodeGenerator generator,
TechniqueSymbol technique, TechniquePassSymbol pass, string uniquePassName)
{
ScriptTextWriter writer = generator.Writer;
writer.WriteLine("// -------- pixel output structures --------");
SemanticGenerator semanticGenerator = new SemanticGenerator("COLOR", 1);
pass.ForEachFragment(generator, (g, f, n) => WritePixelOutputStructure(g, f, n, semanticGenerator));
writer.WriteLine("struct {0}_PIXELOUTPUT", uniquePassName);
writer.WriteLine("{");
writer.WriteLine("\tfloat4 color : COLOR0;");
pass.ForEachFragment(generator, (g, f, s) =>
{
if (f.FragmentNode.PixelOutputs != null && f.FragmentNode.PixelOutputs.VariableDeclarations.Any())
{
writer.WriteLine("\t{0}_{1}_PIXELOUTPUT {1};", s, f.UniqueName);
}
});
writer.WriteLine("};");
writer.WriteLine();
}
public static void WriteVertexShaders(EffectCodeGenerator generator,
TechniqueSymbol technique, TechniquePassSymbol pass, string uniquePassName)
{
Dictionary <string, List <string> > exports = new Dictionary <string, List <string> >();
pass.ForEachFragment(generator, (codeGenerator, symbol, s) => WriteVertexShader(codeGenerator, symbol, s, exports));
ScriptTextWriter writer = generator.Writer;
// Write one vertex shader for each technique pass.
writer.WriteLine("// -------- technique {0}, pass {1} vertex shader entrypoint --------", technique.Name, pass.Name);
writer.WriteLine("{0}_VERTEXOUTPUT {0}_vs(const {0}_VERTEXINPUT i)", uniquePassName);
writer.WriteLine("{");
writer.WriteLine("\tgVertexInput_{0} = i;", uniquePassName);
writer.WriteLine();
writer.WriteLine("\t{0}_VERTEXOUTPUT output = ({0}_VERTEXOUTPUT) 0;", uniquePassName);
writer.WriteLine();
pass.ForEachFragment(generator, (g, f, s) => g.Writer.WriteLine("\t{0}_{1}_vs(gVertexInput_{0}.{1}, output);", uniquePassName, f.UniqueName));
writer.WriteLine();
writer.WriteLine("\treturn output;");
writer.WriteLine("}");
writer.WriteLine();
}
private static void WritePixelShaders(EffectCodeGenerator generator,
TechniqueSymbol technique, TechniquePassSymbol pass, string uniquePassName)
{
Dictionary <string, List <string> > exports = new Dictionary <string, List <string> >();
pass.ForEachFragment(generator, (codeGenerator, symbol, s) => WritePixelShader(codeGenerator, symbol, s, exports));
ScriptTextWriter writer = generator.Writer;
writer.WriteLine("// -------- technique {0}, pass {1} pixel shader entrypoint --------", technique.Name, pass.Name);
writer.WriteLine("{0}_PIXELOUTPUT {0}_ps(const {0}_PIXELINPUT i)", uniquePassName);
writer.WriteLine("{");
writer.WriteLine("\tgPixelInput_{0} = i;", uniquePassName);
writer.WriteLine();
writer.WriteLine("\t{0}_PIXELOUTPUT output = ({0}_PIXELOUTPUT) 0;", uniquePassName);
writer.WriteLine();
pass.ForEachFragment(generator, (g, f, s) =>
{
ShaderCodeBlockNode shader = f.FragmentNode.PixelShaders.GetCodeBlock(generator.Context.TargetShaderProfile);
if (shader != null)
{
writer.WriteLine("\t{0}_{1}_ps(gPixelInput_{0}.{1}, output);", uniquePassName, f.UniqueName);
}
});
writer.WriteLine();
writer.WriteLine("\treturn output;");
writer.WriteLine("}");
writer.WriteLine();
}