Support comment diff at class level

tsantalis · Jul 9, 2024 · 5546e5c · 5546e5c
1 parent 773cf1b
commit 5546e5c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/java/gr/uom/java/xmi/diff/UMLAbstractClassDiff.java b/src/main/java/gr/uom/java/xmi/diff/UMLAbstractClassDiff.java
@@ -64,6 +64,7 @@ public abstract class UMLAbstractClassDiff {
 	protected List<Refactoring> refactorings;
 	protected UMLModelDiff modelDiff;
 	private UMLImplementedInterfaceListDiff interfaceListDiff;
+	private UMLCommentListDiff commentListDiff;
 	private static final List<String> collectionAPINames = List.of("get", "add", "contains", "put", "putAll", "addAll", "equals");
 
 	public UMLAbstractClassDiff(UMLAbstractClass originalClass, UMLAbstractClass nextClass, UMLModelDiff modelDiff) {
@@ -85,6 +86,7 @@ public UMLAbstractClassDiff(UMLAbstractClass originalClass, UMLAbstractClass nex
 		this.nextClass = nextClass;
 		this.modelDiff = modelDiff;
 		this.interfaceListDiff = new UMLImplementedInterfaceListDiff(originalClass.getImplementedInterfaces(), nextClass.getImplementedInterfaces());
+		this.commentListDiff = new UMLCommentListDiff(originalClass.getComments(), nextClass.getComments());
 	}
 
 	public List<UMLOperation> getAddedOperations() {
@@ -147,6 +149,10 @@ public UMLModelDiff getModelDiff() {
 		return modelDiff;
 	}
 
+	public UMLCommentListDiff getCommentListDiff() {
+		return commentListDiff;
+	}
+
 	public UMLImplementedInterfaceListDiff getInterfaceListDiff() {
 		return interfaceListDiff;
 	}