From 211d9e990b658561ace3971424ce6a6f9ad63f61 Mon Sep 17 00:00:00 2001
From: Kui LIU <brucekuiliu@gmail.com>
Date: Thu, 14 Sep 2017 15:13:36 +0200
Subject: [PATCH] Update the information of a violation.

---
 .../violations/Violation.java                 | 22 ++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

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<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;
+	}
 	
 }