diff --git a/src/Generator/Passes/RenamePass.cs b/src/Generator/Passes/RenamePass.cs index 58d1c667..699ae3fc 100644 --- a/src/Generator/Passes/RenamePass.cs +++ b/src/Generator/Passes/RenamePass.cs @@ -89,18 +89,35 @@ namespace Cxxi.Passes return base.VisitMethodDecl(method); } + + public override bool VisitFunctionDecl(Function function) + { + if (!Targets.HasFlag(RenameTargets.Function)) + return false; + + return base.VisitFunctionDecl(function); + } + + public override bool VisitParameterDecl(Parameter parameter) + { + if (!Targets.HasFlag(RenameTargets.Parameter)) + return false; + + return base.VisitParameterDecl(parameter); + } } [Flags] public enum RenameTargets { - Record, + Class, Field, Method, Function, + Parameter, Enum, EnumItem, - Any = Function | Method | Record | Field | Enum | EnumItem, + Any = Function | Method | Parameter | Class | Field | Enum | EnumItem, } ///