diff --git a/src/main/java/edu/lu/uni/serval/FixPatternParser/violations/Violation.java b/src/main/java/edu/lu/uni/serval/FixPatternParser/violations/Violation.java index 409af52..4d9ec5a 100644 --- a/src/main/java/edu/lu/uni/serval/FixPatternParser/violations/Violation.java +++ b/src/main/java/edu/lu/uni/serval/FixPatternParser/violations/Violation.java @@ -7,7 +7,8 @@ import edu.lu.uni.serval.diffentry.DiffEntryHunk; import edu.lu.uni.serval.gumtree.regroup.HierarchicalActionSet; public class Violation implements Comparable { - + + private String fileName = ""; private Integer startLineNum; private int endLineNum; private int bugStartLineNum; @@ -26,6 +27,14 @@ public class Violation implements Comparable { this.actionSets = new ArrayList<>(); } + public String getFileName() { + return fileName; + } + + public void setFileName(String fileName) { + this.fileName = fileName; + } + public Integer getStartLineNum() { return startLineNum; } @@ -91,5 +100,16 @@ public class Violation implements Comparable { public String toString() { return this.startLineNum + " : " + this.endLineNum + " : " + this.alarmType; } + + @Override + public boolean equals(Object obj) { + if (obj instanceof Violation) { + Violation v = (Violation) obj; + if (this.fileName.equals(v.fileName) && this.alarmType.equals(v.alarmType) && this.startLineNum == v.startLineNum && this.endLineNum == v.endLineNum) { + return true; + } + } + return false; + } }