Simplifying solution in previous commit

tsantalis · Sep 4, 2024 · 2063839 · 2063839
1 parent 4d1c0cd
commit 2063839
Showing 1 changed file with 2 additions and 11 deletions.
diff --git a/src/main/java/gr/uom/java/xmi/UMLModelASTReader.java b/src/main/java/gr/uom/java/xmi/UMLModelASTReader.java
@@ -240,17 +240,8 @@ private List<UMLComment> extractInternalComments(CompilationUnit cu, String sour
 			else if(comment.isBlockComment()) {
 				locationInfo = generateLocationInfo(cu, sourceFile, comment, CodeElementType.BLOCK_COMMENT);
 			}
-			else if(comment.isDocComment()) {
-				if(comment.getStartPosition() + comment.getLength() + 1 <= javaFileContent.length()) {
-					String textFollowing = javaFileContent.substring(comment.getStartPosition() + comment.getLength() + 1, javaFileContent.length());
-					String[] lines = textFollowing.split("\\r?\\n|\\r");
-					if(lines.length > 0) {
-						String firstLine = lines[0];
-						if(!firstLine.contains("public") && !firstLine.contains("private") && !firstLine.contains("protected") && !firstLine.contains("@") && !firstLine.contains("static")) {
-							locationInfo = generateLocationInfo(cu, sourceFile, comment, CodeElementType.BLOCK_COMMENT);
-						}
-					}
-				}
+			else if(comment.isDocComment() && comment.getParent() == null) {
+				locationInfo = generateLocationInfo(cu, sourceFile, comment, CodeElementType.BLOCK_COMMENT);
 			}
 			if(locationInfo != null) {
 				int start = comment.getStartPosition();