|
| 1 | +using CppSharp.AST; |
| 2 | +using CppSharp.AST.Extensions; |
| 3 | + |
| 4 | +namespace CppSharp.Passes |
| 5 | +{ |
| 6 | + /// <summary> |
| 7 | + /// Replaces anonymous types with their contained fields |
| 8 | + /// in class layouts and properties in order to avoid generating |
| 9 | + /// these very same anonymous types which is useless and |
| 10 | + /// also forces unappealing names. |
| 11 | + /// </summary> |
| 12 | + public class FlattenAnonymousTypesToFields : TranslationUnitPass |
| 13 | + { |
| 14 | + public FlattenAnonymousTypesToFields() |
| 15 | + { |
| 16 | + VisitOptions.VisitClassBases = false; |
| 17 | + VisitOptions.VisitClassFields = false; |
| 18 | + VisitOptions.VisitClassMethods = false; |
| 19 | + VisitOptions.VisitClassProperties = false; |
| 20 | + VisitOptions.VisitEventParameters = false; |
| 21 | + VisitOptions.VisitFunctionParameters = false; |
| 22 | + VisitOptions.VisitFunctionReturnType = false; |
| 23 | + VisitOptions.VisitNamespaceEnums = false; |
| 24 | + VisitOptions.VisitNamespaceTemplates = false; |
| 25 | + VisitOptions.VisitNamespaceTypedefs = false; |
| 26 | + VisitOptions.VisitNamespaceVariables = false; |
| 27 | + VisitOptions.VisitPropertyAccessors = false; |
| 28 | + VisitOptions.VisitTemplateArguments = false; |
| 29 | + } |
| 30 | + |
| 31 | + public override bool VisitClassDecl(Class @class) |
| 32 | + { |
| 33 | + if (!base.VisitClassDecl(@class) || @class.Ignore || @class.IsDependent) |
| 34 | + return false; |
| 35 | + |
| 36 | + for (int i = @class.Fields.Count - 1; i >= 0; i--) |
| 37 | + { |
| 38 | + Field field = @class.Fields[i]; |
| 39 | + Class fieldType; |
| 40 | + if (!string.IsNullOrEmpty(field.OriginalName) || |
| 41 | + !field.Type.Desugar().TryGetClass(out fieldType) || |
| 42 | + !string.IsNullOrEmpty(fieldType.OriginalName)) |
| 43 | + continue; |
| 44 | + |
| 45 | + ReplaceField(@class, i, fieldType); |
| 46 | + ReplaceLayoutField(@class, field, fieldType); |
| 47 | + fieldType.Fields.Clear(); |
| 48 | + fieldType.ExplicitlyIgnore(); |
| 49 | + } |
| 50 | + |
| 51 | + return true; |
| 52 | + } |
| 53 | + |
| 54 | + private static void ReplaceField(Class @class, int i, Class fieldType) |
| 55 | + { |
| 56 | + @class.Fields.RemoveAt(i); |
| 57 | + |
| 58 | + for (int j = 0; j < fieldType.Fields.Count; j++) |
| 59 | + { |
| 60 | + Field nestedField = fieldType.Fields[j]; |
| 61 | + nestedField.Namespace = @class; |
| 62 | + @class.Fields.Insert(i + j, nestedField); |
| 63 | + } |
| 64 | + } |
| 65 | + |
| 66 | + private static void ReplaceLayoutField(Class @class, Field field, Class fieldType) |
| 67 | + { |
| 68 | + LayoutField layoutField = @class.Layout.Fields.Find( |
| 69 | + f => f.FieldPtr == field.OriginalPtr); |
| 70 | + int layoutIndex = @class.Layout.Fields.IndexOf(layoutField); |
| 71 | + @class.Layout.Fields.RemoveAt(layoutIndex); |
| 72 | + |
| 73 | + for (int j = 0; j < fieldType.Layout.Fields.Count; j++) |
| 74 | + { |
| 75 | + LayoutField nestedlayoutField = fieldType.Layout.Fields[j]; |
| 76 | + nestedlayoutField.Offset += layoutField.Offset; |
| 77 | + @class.Layout.Fields.Insert(layoutIndex + j, nestedlayoutField); |
| 78 | + } |
| 79 | + } |
| 80 | + } |
| 81 | +} |
0 commit comments