From 7d49abfb271b5491ca6cf56db3249e9281528544 Mon Sep 17 00:00:00 2001
From: Kui LIU <brucekuiliu@gmail.com>
Date: Sat, 12 Aug 2017 13:30:14 +0200
Subject: [PATCH] Limit the distance between the buggy code line and inserted
 code line.

---
 .../lu/uni/serval/gumtree/regroup/HunkActionFilter.java   | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/main/java/edu/lu/uni/serval/gumtree/regroup/HunkActionFilter.java b/src/main/java/edu/lu/uni/serval/gumtree/regroup/HunkActionFilter.java
index 9fcda9a..559356e 100644
--- a/src/main/java/edu/lu/uni/serval/gumtree/regroup/HunkActionFilter.java
+++ b/src/main/java/edu/lu/uni/serval/gumtree/regroup/HunkActionFilter.java
@@ -394,7 +394,10 @@ public class HunkActionFilter {
 							}
 						} else {
 							if (isRanged(actionSet, violation)) {
-								violation.getActionSets().add(actionSet);
+								if (Math.abs(violationStartLine - actionFixStartLine) <= 5 
+										&& Math.abs(violationEndLine - actionFixEndLine) <= 5) {
+									violation.getActionSets().add(actionSet);
+								}
 							}
 						}
 					}
@@ -491,6 +494,7 @@ public class HunkActionFilter {
 								fixStarts = 0;
 								continue;
 							}
+							
 							return true;
 						}
 					}
@@ -644,6 +648,8 @@ public class HunkActionFilter {
 		actionSet.setBugEndLineNum(actionBugEndLine);
 		actionSet.setFixStartLineNum(actionFixStartLine);
 		actionSet.setFixEndLineNum(actionFixEndLine);
+		actionSet.setBugEndPosition(bugEndPosition);
+		actionSet.setFixEndPosition(fixEndPosition);
 		
 		return actionBugStartLine;
 	}