@@ -42,18 +42,21 @@ public AstManipulator(Messager messager, String sourceFolder) {
4242 }
4343
4444 public void eraseTypesAndInsertCasts (String cls , String packageName , String typeOfInterest ) {
45- messager .printMessage (Kind .NOTE , "Now parsing AST's" );
46-
45+ messager .printMessage (Kind .NOTE ,
46+ String . format ( "Now parsing AST's for class %s and type param %s" , cls , typeOfInterest ));
4747 File dir = Paths .get (sourceFolder ).toFile ();
4848 assert dir .exists ();
4949 assert dir .isDirectory ();
5050
5151 ClassData classData = computeClassData (cls , packageName , typeOfInterest );
52+ messager .printMessage (Kind .NOTE , "Collected class data:\n " + classData );
5253 Map <String , MethodData > methodMap = new HashMap <>();
5354
5455 sourceRoot .parse (packageName , cls ).accept (new MethodCollector (Arrays .asList (typeOfInterest )),
5556 methodMap );
5657
58+ messager .printMessage (Kind .NOTE , "Collected methods:\n " + methodMap .toString ());
59+
5760 changeAST (dir , classData , methodMap , "" );
5861
5962 }
@@ -90,6 +93,7 @@ private void changeAST(File dir, ClassData classData, Map<String, MethodData> me
9093
9194 Set <Pair <String , String >> varsToWatch = new HashSet <>();
9295 cu .accept (new VariableCollector (classData ), varsToWatch );
96+ messager .printMessage (Kind .NOTE , "Collected variables to watch:\n " + varsToWatch );
9397 cu .accept (new TypeEraserVisitor (classData ), null );
9498 for (Pair <String , String > var : varsToWatch ) {
9599 CastInsertionVisitor castInsertionVisitor = new CastInsertionVisitor (var , methodMap );
@@ -135,7 +139,10 @@ private void computeClassHierarchyRec(ClassHierarchyGraph<String> g, File dir, S
135139 }
136140
137141 public void applyChanges () {
138- this .sourceRoot .getCompilationUnits ().forEach (cu -> changePackageDeclaration (cu ));
142+ this .sourceRoot .getCompilationUnits ().forEach (cu -> {
143+ messager .printMessage (Kind .NOTE , "Saving cu: " + cu .toString ());
144+ changePackageDeclaration (cu );
145+ });
139146 this .sourceRoot .saveAll (
140147 CodeGenerationUtils .mavenModuleRoot (AstManipulator .class ).resolve (Paths .get (sourceFolder + "/output" )));
141148 }
0 commit comments