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); }