Update the information of a violation.

2017-09-14 15:13:36 +02:00
parent 904b647936
commit 211d9e990b
1 changed files with 21 additions and 1 deletions
@@ -7,7 +7,8 @@ import edu.lu.uni.serval.diffentry.DiffEntryHunk;
 import edu.lu.uni.serval.gumtree.regroup.HierarchicalActionSet;

 public class Violation implements Comparable<Violation> {
-
+	
+	private String fileName = "";
 	private Integer startLineNum;
 	private int endLineNum;
 	private int bugStartLineNum;
@@ -26,6 +27,14 @@ public class Violation implements Comparable<Violation> {
 		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<Violation> {
 	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;
+	}
 	
 }