string WriteSamplers(ShaderOutputContext ctx, TemplateGroup group)
{
List <Sampler> samplers = ctx.Samplers;
if (samplers.Count == 0)
{
return("");
}
StringBuilder result = new StringBuilder();
result.Append("///////////////// SAMPLERS /////////////////// ");
result.Append('\n');
foreach (var sampler in samplers)
{
StringTemplate samplerTemplate = group.GetInstanceOf("sampler");
samplerTemplate.Add("name", sampler.Name);
samplerTemplate.Add("reg", sampler.Bindpoint.Reg);
samplerTemplate.Add("space", sampler.Bindpoint.Space);
result.Append(samplerTemplate.Render() + '\n');
}
result.Append('\n');
result.Append('\n');
return(result.ToString());
}
string WriteVertexLayout(ShaderOutputContext ctx, TemplateGroup group)
{
VertexLayout vLayout = ctx.VertLayout;
StringBuilder result = new StringBuilder();
result.Append("///////////////// VERTEX LAYOUT /////////////////// ");
result.Append('\n');
StringBuilder members = new StringBuilder();
foreach (var member in vLayout.Members)
{
StringTemplate memberTemplate = group.GetInstanceOf("vlayoutvar");
memberTemplate.Add("type", Variable.ConvertHlslType(member.Type));
memberTemplate.Add("name", member.Name);
string index = "";
if (member.Semantic == "TEXCOORD" || member.Semantic == "COLOR")
{
index = member.SemanticIndex.ToString();
}
memberTemplate.Add("semantic", member.Semantic + index);
members.Append(memberTemplate.Render() + '\n');
}
StringTemplate structTemplate = group.GetInstanceOf("struct");
structTemplate.Add("name", "vIn");
structTemplate.Add("members", members);
result.Append(structTemplate.Render() + '\n' + '\n');
result.Append('\n');
result.Append('\n');
return(result.ToString());
}
string WriteTextures(ShaderOutputContext ctx, TemplateGroup group)
{
List <Texture> textures = ctx.Textures;
if (textures.Count == 0)
{
return("");
}
StringBuilder result = new StringBuilder();
result.Append("///////////////// TEXTURES /////////////////// ");
result.Append('\n');
foreach (var texture in textures)
{
StringTemplate textureTemplate = group.GetInstanceOf("tex2d");
textureTemplate.Add("name", texture.Name);
textureTemplate.Add("reg", texture.Bindpoint.Reg);
textureTemplate.Add("space", texture.Bindpoint.Space);
result.Append(textureTemplate.Render() + '\n');
}
result.Append('\n');
result.Append('\n');
return(result.ToString());
}
string WriteRootConstants(ShaderOutputContext ctx, TemplateGroup group)
{
List <UniformConstant> rootConstants = ctx.UniformConstants;
if (rootConstants.Count == 0)
{
return("");
}
StringBuilder result = new StringBuilder();
result.Append("///////////////// ROOT CONSANTS /////////////////// ");
result.Append('\n');
foreach (var rootConstant in rootConstants)
{
StringTemplate cbufferTemplate = group.GetInstanceOf("cbuffer");
cbufferTemplate.Add("name", "cb_" + rootConstant.Name);
string member = rootConstant.Type + " " + rootConstant.Name + ";";
cbufferTemplate.Add("members", member);
cbufferTemplate.Add("reg", rootConstant.Bindpoint.Reg);
cbufferTemplate.Add("space", rootConstant.Bindpoint.Space);
result.Append(cbufferTemplate.Render() + '\n' + '\n');
}
result.Append('\n');
result.Append('\n');
return(result.ToString());
}
public void AssignBindpoints(ShaderOutputContext ctx)
{
AssignBindpointsFromAnnotations(ctx.ConstantBuffers);
AutomaticallyAssignBindpoints(ctx.ConstantBuffers, m_allBindings[BindpointType.Buffer]);
// [a_vorontcov] Root constants have the same bindpoints as constant buffers.
AssignBindpointsFromAnnotations(ctx.UniformConstants);
AutomaticallyAssignBindpoints(ctx.UniformConstants, m_allBindings[BindpointType.Buffer]);
AssignBindpointsFromAnnotations(ctx.Textures);
AutomaticallyAssignBindpoints(ctx.Textures, m_allBindings[BindpointType.Texture]);
AssignBindpointsFromAnnotations(ctx.Samplers);
AutomaticallyAssignBindpoints(ctx.Samplers, m_allBindings[BindpointType.Sampler]);
}
public void WriteHeaders(ShaderOutputContext ctx, string filename)
{
string filenameUpper = filename.ToUpper();
TemplateGroup group = new Antlr4.StringTemplate.TemplateGroupFile(Program.TemplatesDir + "/hlslTemplate.stg");
StringBuilder result = new StringBuilder();
// [a_vorontcov] TODO: move this stuff to the hlsl tempate file.
result.Append("//////////////////////////////////////////////// \n");
result.Append("////////// AUTOGENERATED FILE, DO NOT EDIT !//// \n");
result.Append("//////////////////////////////////////////////// \n");
result.Append("#ifndef _" + filenameUpper + "_\n");
result.Append("#define _" + filenameUpper + "_\n");
result.Append('\n');
result.Append("#include \"autogen/CommonStructures.hlsl\" \n\n"); // [a_vorontcov] The folder context is the shader folder context, so relative paths are related to it, not to autogenerated files.
result.Append("///////////////// STRUCTURES /////////////////// ");
result.Append('\n');
result.Append(WriteStructures(group, ctx.Structures));
result.Append(WriteStructures(group, ctx.ConstantBuffers));
result.Append('\n');
result.Append('\n');
result.Append(WriteConstantBuffers(ctx, group));
string constants = WriteRootConstants(ctx, group);
result.Append(constants);
result.Append(WriteTextures(ctx, group));
result.Append(WriteSamplers(ctx, group));
result.Append(WriteVertexLayout(ctx, group));
result.Append("\n#endif // " + filenameUpper);
string outDirHlsl = Program.HlslOutputDir;
string filenameOut = outDirHlsl + "/" + filename + ".hlsl";
File.WriteAllText(filenameOut, result.ToString());
}
string WriteConstantBuffers(ShaderOutputContext ctx, TemplateGroup group)
{
List <ConstantBuffer> constantBuffers = ctx.ConstantBuffers;
if (constantBuffers.Count == 0)
{
return("");
}
StringBuilder result = new StringBuilder();
result.Append("///////////////// CONSTANT BUFFERS /////////////////// ");
result.Append('\n');
foreach (var constantBuffer in constantBuffers)
{
StringTemplate cbufferTemplate = group.GetInstanceOf("cbufferTempl");
cbufferTemplate.Add("name", constantBuffer.Name);
cbufferTemplate.Add("typename", constantBuffer.Typename);
cbufferTemplate.Add("reg", constantBuffer.Bindpoint.Reg);
cbufferTemplate.Add("space", constantBuffer.Bindpoint.Space);
if (constantBuffer.Size != 0)
{
cbufferTemplate.Add("isArray", "true");
}
if (constantBuffer.Size > 0)
{
cbufferTemplate.Add("size", constantBuffer.Size);
}
result.Append(cbufferTemplate.Render() + '\n' + '\n');
}
result.Append('\n');
result.Append('\n');
return(result.ToString());
}
public void Merge(ShaderOutputContext other)
{
Structures.AddRange(other.Structures);
ConstantBuffers.AddRange(other.ConstantBuffers);
Textures.AddRange(other.Textures);
Samplers.AddRange(other.Samplers);
UniformConstants.AddRange(other.UniformConstants);
if (VertLayout != null && other.VertLayout != null)
{
throw new DuplicateBindpointException("Vertex layout is defined twice");
}
if (VertLayout == null)
{
VertLayout = other.VertLayout;
}
if (ShaderBinding != null && other.ShaderBinding != null)
{
throw new DuplicateBindpointException("Shader bindings are defined twice");
}
if (ShaderBinding == null)
{
ShaderBinding = other.ShaderBinding;
}
}