public override void Visit(Cond cond)
{
_state.ConsolidateVariables();
var expr = new ExpressionSimplifierVisitor().Visit(cond.Expr);
if (expr is Bool b)
{
if (b.Value)
{
Visit(cond.Block);
}
return;
}
var exprResult = new ExpressionGeneratorVisitor(_state).Visit(expr);
_state.ConsolidateVariables();
_state.GoToNextLabel(out var trueLabel);
var falseLabel = _state.NewLabel;
_llvmGenerator.Emit($"br i1 {exprResult.Register}, label %{trueLabel}, label %{falseLabel}");
_llvmGenerator.Emit($"{trueLabel}:");
VisitBlockWithoutRestore(cond.Block);
_state.ConsolidateVariables();
_state.RestorePreviousVarEnvWithMerge();
_llvmGenerator.Emit($"br label %{falseLabel}");
_state.CurrentLabel = falseLabel;
_llvmGenerator.Emit($"{falseLabel}:");
}
public override void Visit(While @while)
{
var expr = new ExpressionSimplifierVisitor().Visit(@while.Expr);
if (expr is Bool b && b.Value == false)
{
return;
}
_state.GoToNextLabel(out var startWhileLabel);
_llvmGenerator.Emit($"br label %{startWhileLabel}");
_llvmGenerator.Emit($"{startWhileLabel}:");
var exprResult = new ExpressionGeneratorVisitor(_state).Visit(expr);
_state.GoToNextLabel(out var whileLabel);
var endWhileLabel = _state.NewLabel;
_llvmGenerator.Emit($"br i1 {exprResult.Register}, label %{whileLabel}, label %{endWhileLabel}");
_llvmGenerator.Emit($"{whileLabel}:");
Visit(@while.Block);
_llvmGenerator.Emit($"br label %{startWhileLabel}");
_state.CurrentLabel = endWhileLabel;
_llvmGenerator.Emit($"{endWhileLabel}:");
_llvmGenerator.Emit($"br label %{endWhileLabel}");
}
public override void Visit(CondElse condElse)
{
var expr = new ExpressionSimplifierVisitor().Visit(condElse.Expr);
if (expr is Bool b)
{
Visit(b.Value ? condElse.TBlock : condElse.FBlock);
return;
}
var exprResult = new ExpressionGeneratorVisitor(_state).Visit(expr);
_state.GoToNextLabel(out var trueLabel);
string falseLabel = _state.NewLabel, endIfLabel = _state.NewLabel;
_llvmGenerator.Emit($"br i1 {exprResult.Register}, label %{trueLabel}, label %{falseLabel}");
_llvmGenerator.Emit($"{trueLabel}:");
Visit(condElse.TBlock);
_llvmGenerator.Emit($"br label %{endIfLabel}");
_state.CurrentLabel = falseLabel;
_llvmGenerator.Emit($"{falseLabel}:");
Visit(condElse.FBlock);
_llvmGenerator.Emit($"br label %{endIfLabel}");
_state.CurrentLabel = endIfLabel;
_llvmGenerator.Emit($"{endIfLabel}:");
}
public DynamicMethodBody Expression(Expression expression)
{
expression = new ExpressionSimplifierVisitor().Visit(expression);
new ILEmitterVisitor(this).Visit(
expression
);
return(this);
}
public override void Visit(Ass ass)
{
var expr = new ExpressionSimplifierVisitor().Visit(ass.Expr);
var exprResult = new ExpressionGeneratorVisitor(_state).Visit(expr);
_state.VarToLabelToRegister[ass.Id] =
new Dictionary <string, RegisterLabelContext> {
{ exprResult.Label, exprResult }
};
}
public override void Visit(While @while)
{
var expr = new ExpressionSimplifierVisitor().Visit(@while.Expr);
if (expr is Bool b && b.Value == false)
{
return;
}
_state.ConsolidateVariables();
_state.GoToNextLabel(out var startWhileLabel);
_llvmGenerator.Emit($"br label %{startWhileLabel}");
_state.DetachVarEnv();
_state.DetachVarEnv();
var reservedRegisters = _state.ReserveRegisterForCurrentVars(startWhileLabel);
_state.GoToNextLabel(out var whileLabel);
_llvmGenerator.Emit($"{whileLabel}:");
var startLength = _llvmGenerator.GetCurrentLength();
VisitBlock(@while.Block);
_state.ConsolidateVariables();
_state.RestorePreviousVarEnvWithMerge();
_llvmGenerator.Emit($"br label %{startWhileLabel}");
_state.CurrentLabel = startWhileLabel;
_llvmGenerator.Emit($"{startWhileLabel}:");
_state.RestorePreviousVarEnvWithMerge();
_state.RemoveReservedRegisters(reservedRegisters, out var removedRegisters);
_state.ConsolidateVariables(reservedRegisters);
var removedRegs = removedRegisters.ToHashSet();
var reservedToReplace = reservedRegisters.ToList().Where(reg => removedRegs.Contains(reg.Value.Register)).ToList();
_llvmGenerator.ReplaceRegisters(startLength, reservedToReplace.Select(res =>
(res.Value.Register, _state.VarToLabelToRegister[res.Key][_state.CurrentLabel].Register)).ToList());
var exprResult = new ExpressionGeneratorVisitor(_state).Visit(expr);
var endWhileLabel = _state.NewLabel;
_llvmGenerator.Emit($"br i1 {exprResult.Register}, label %{whileLabel}, label %{endWhileLabel}");
_state.CurrentLabel = endWhileLabel;
_llvmGenerator.Emit($"{endWhileLabel}:");
}
public override void Visit(Ret ret)
{
var toEmit = "ret ";
if (ret.Expr == null)
{
toEmit += "void";
}
else
{
var expr = new ExpressionSimplifierVisitor().Visit(ret.Expr);
var exprResult = new ExpressionGeneratorVisitor(_state).Visit(expr);
toEmit += $"{AstToLlvmString.Type(exprResult.Type)} {exprResult.Register}";
}
_llvmGenerator.Emit(toEmit);
}
public override void Visit(StructAss structAss)
{
var objectExpr = new ExpressionSimplifierVisitor().Visit(structAss.IdExpr);
var objectExprResult = new ExpressionGeneratorVisitor(_state).Visit(objectExpr);
var expr = new ExpressionSimplifierVisitor().Visit(structAss.Expr);
var exprResult = new ExpressionGeneratorVisitor(_state).Visit(expr);
var nextRegister1 = _state.NewRegister;
var field = _globalState.NameToClass[objectExprResult.Type.GetText()].Fields[structAss.Id];
var fieldTypeString = objectExprResult.Type.GetText();
_llvmGenerator.Emit($"{nextRegister1} = getelementptr %{fieldTypeString}, " +
$"%{fieldTypeString}* {objectExprResult.Register}, i32 0, i32 {field.Number + 1}");
_llvmGenerator.Emit($"store {AstToLlvmString.Type(exprResult.Type)} {exprResult.Register}, " +
$"{AstToLlvmString.Type(field.Type)}* {nextRegister1}");
}
public override void Visit(Ret ret)
{
var toEmit = "ret ";
if (ret.Expr == null)
{
toEmit += "void";
}
else
{
var expr = new ExpressionSimplifierVisitor().Visit(ret.Expr);
var exprResult = new ExpressionGeneratorVisitor(_state).Visit(expr);
toEmit += exprResult.Type is LatteParser.TVoidContext
? AstToLlvmString.Type(_globalState.NameToFunction[_state.CurrentFunction].Type)
: AstToLlvmString.Type(exprResult.Type);
toEmit += $" {exprResult.Register}";
}
_llvmGenerator.Emit(toEmit);
}
public override void Visit(Ass ass)
{
var expr = new ExpressionSimplifierVisitor().Visit(ass.Expr);
var exprResult = new ExpressionGeneratorVisitor(_state).Visit(expr);
if (_state.VarToLabelToRegister.ContainsKey(ass.Id))
{
var varType = _state.VarToLabelToRegister[ass.Id].ToList()[0].Value.Type;
if (varType.GetText() != exprResult.Type.GetText())
{
var nextRegister = _state.NewRegister;
_llvmGenerator.Emit($"{nextRegister} = bitcast {AstToLlvmString.Type(exprResult.Type)} " +
$"{exprResult.Register} to {AstToLlvmString.Type(varType)}");
exprResult.Register = nextRegister;
exprResult.Type = varType;
}
_state.VarToLabelToRegister[ass.Id] =
new Dictionary <string, RegisterLabelContext> {
{ exprResult.Label, exprResult }
};
}
else
{
var classDef = _globalState.CurrentClass;
var field = classDef.Fields[ass.Id];
var selfRegister = _state.VarToLabelToRegister["self"].Values.ToList()[0].Register;
var nextRegister = _state.NewRegister;
_llvmGenerator.Emit(
$"{nextRegister} = getelementptr %{classDef.Id}, %{classDef.Id}* {selfRegister}, i32 0, i32 {field.Number + 1}");
_llvmGenerator.Emit($"store {AstToLlvmString.Type(field.Type)} {exprResult.Register}, " +
$"{AstToLlvmString.Type(field.Type)}* {nextRegister}");
}
}
/// <summary>
/// Mixes shader parts to produces a single HLSL file shader.
/// </summary>
/// <param name="shaderMixinSource">The shader source.</param>
/// <param name="macros">The shader perprocessor macros.</param>
/// <param name="modifiedShaders">The list of modified shaders.</param>
/// <returns>The combined shader in AST form.</returns>
public ShaderMixinParsingResult Parse(ShaderMixinSource shaderMixinSource, Stride.Shaders.ShaderMacro[] macros = null)
{
// Make in-memory shader classes known to the source manager
foreach (var x in shaderMixinSource.Mixins.OfType <ShaderClassString>())
{
SourceManager.AddShaderSource(x.ClassName, x.ShaderSourceCode, x.ClassName);
}
// Creates a parsing result
HashSet <ModuleMixinInfo> mixinsToAnalyze;
ShaderMixinParsingResult parsingResult;
var context = ParseAndAnalyze(shaderMixinSource, macros, out parsingResult, out mixinsToAnalyze);
// Return directly if there was any errors
if (parsingResult.HasErrors)
{
return(parsingResult);
}
// Update the clone context in case new instances of classes are created
CloneContext mixCloneContext;
lock (hlslCloneContextLock)
{
if (hlslCloneContext == null)
{
hlslCloneContext = new CloneContext();
// Create the clone context with the instances of Hlsl classes
HlslSemanticAnalysis.FillCloneContext(hlslCloneContext);
}
HlslSemanticAnalysis.UpdateCloneContext(hlslCloneContext);
mixCloneContext = new CloneContext(hlslCloneContext);
}
// only clone once the stage classes
foreach (var mixinInfo in mixinsToAnalyze)
{
foreach (var mixin in mixinInfo.Mixin.MinimalContext.Where(x => x.StageOnlyClass))
{
mixin.DeepClone(mixCloneContext);
}
}
// ----------------------------------------------------------
// Perform Shader Mixer
// ----------------------------------------------------------
var externDict = new CompositionDictionary();
var finalModuleList = BuildCompositionsDictionary(shaderMixinSource, externDict, context, mixCloneContext, parsingResult);
//PerformanceLogger.Stop(PerformanceStage.DeepClone);
if (parsingResult.HasErrors)
{
return(parsingResult);
}
// look for stage compositions and add the links between variables and compositions when necessary
var extraExternDict = new Dictionary <Variable, List <ModuleMixin> >();
foreach (var item in externDict)
{
if (item.Key.Qualifiers.Contains(StrideStorageQualifier.Stage))
{
FullLinkStageCompositions(item.Key, item.Value, externDict, extraExternDict, parsingResult);
}
}
foreach (var item in extraExternDict)
{
externDict.Add(item.Key, item.Value);
}
var mixinDictionary = BuildMixinDictionary(finalModuleList);
if (finalModuleList != null)
{
var finalModule = finalModuleList[0];
//PerformanceLogger.Start(PerformanceStage.Mix);
parsingResult.Reflection = new EffectReflection();
var mixer = new StrideShaderMixer(finalModule, parsingResult, mixinDictionary, externDict, new CloneContext(mixCloneContext));
mixer.Mix();
//PerformanceLogger.Stop(PerformanceStage.Mix);
// Return directly if there was any errors
if (parsingResult.HasErrors)
{
return(parsingResult);
}
var finalShader = mixer.GetMixedShader();
// Simplifies the shader by removing dead code
var simplifier = new ExpressionSimplifierVisitor();
simplifier.Run(finalShader);
var sdShaderLinker = new ShaderLinker(parsingResult);
sdShaderLinker.Run(finalShader);
// Return directly if there was any errors
if (parsingResult.HasErrors)
{
return(parsingResult);
}
// Find all entry points
// TODO: make this configurable by CompileParameters
foreach (var stage in new[] { ShaderStage.Compute, ShaderStage.Vertex, ShaderStage.Hull, ShaderStage.Domain, ShaderStage.Geometry, ShaderStage.Pixel })
{
var entryPoint = finalShader.Declarations.OfType <MethodDefinition>().FirstOrDefault(f => f.Attributes.OfType <AttributeDeclaration>().Any(a => a.Name == "EntryPoint" && (string)a.Parameters[0].Value == stage.ToString()));
if (entryPoint == null)
{
continue;
}
parsingResult.EntryPoints[stage] = entryPoint.Name.Text;
// When this is a compute shader, there is no need to scan other stages
if (stage == ShaderStage.Compute)
{
break;
}
}
var typeCleaner = new StrideShaderCleaner();
typeCleaner.Run(finalShader);
//PerformanceLogger.Stop(PerformanceStage.Global);
//PerformanceLogger.PrintLastResult();
//SemanticPerformance.PrintResult();
//MixPerformance.PrintResult();
//GenerateShaderPerformance.PrintResult();
//StreamCreatorPerformance.PrintResult();
//ShaderLoader.PrintTime();
//PerformanceLogger.WriteOut(52);
parsingResult.Shader = finalShader;
}
return(parsingResult);
}
public override void Visit(ExpStmt expStmt)
{
var expr = new ExpressionSimplifierVisitor().Visit(expStmt.Expr);
new ExpressionGeneratorVisitor(_state).Visit(expr);
}
