Modify statement signature in case of anonymous class declaration within

the statement
tsantalis · Oct 23, 2024 · 35a9df1 · 35a9df1
1 parent b1d526e
commit 35a9df1
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/main/java/gr/uom/java/xmi/decomposition/StatementObject.java b/src/main/java/gr/uom/java/xmi/decomposition/StatementObject.java
@@ -56,9 +56,6 @@ public StatementObject(CompilationUnit cu, String filePath, Statement statement,
 		this.locationInfo = new LocationInfo(cu, filePath, statement, codeElementType);
 		Visitor visitor = new Visitor(cu, filePath, container, javaFileContent);
 		statement.accept(visitor);
-		int start = statement.getStartPosition();
-		int end = start + statement.getLength();
-		this.actualSignature = javaFileContent.substring(start, end);
 		this.variables = visitor.getVariables();
 		this.types = visitor.getTypes();
 		this.variableDeclarations = visitor.getVariableDeclarations();
@@ -84,6 +81,15 @@ public StatementObject(CompilationUnit cu, String filePath, Statement statement,
 		this.castExpressions = visitor.getCastExpressions();
 		this.ternaryOperatorExpressions = visitor.getTernaryOperatorExpressions();
 		this.lambdas = visitor.getLambdas();
+		int start = statement.getStartPosition();
+		int end = start + statement.getLength();
+		if(this.anonymousClassDeclarations.isEmpty()) {
+			this.actualSignature = javaFileContent.substring(start, end);
+		}
+		else {
+			int endOffset = this.anonymousClassDeclarations.get(0).getLocationInfo().getStartOffset() + 1;
+			this.actualSignature = javaFileContent.substring(start, endOffset);
+		}
 		setDepth(depth);
 		String statementAsString = stringify(statement);
 		if(Visitor.METHOD_INVOCATION_PATTERN.matcher(statementAsString).matches()) {