# HG changeset patch
# User Gilles Duboscq <duboscq@ssw.jku.at>
# Date 1409738474 -7200
# Node ID 7b2e6171f455f83222d40c5584a1b6655a32faf9
# Parent  86749331247ad90b90f810c8fbf8e98965161d48
LineLocationToProbeCollectionMap: add forget method

diff -r 86749331247a -r 7b2e6171f455 graal/com.oracle.truffle.api/src/com/oracle/truffle/api/instrument/impl/LineLocationToProbeCollectionMap.java
--- a/graal/com.oracle.truffle.api/src/com/oracle/truffle/api/instrument/impl/LineLocationToProbeCollectionMap.java	Wed Sep 03 12:00:41 2014 +0200
+++ b/graal/com.oracle.truffle.api/src/com/oracle/truffle/api/instrument/impl/LineLocationToProbeCollectionMap.java	Wed Sep 03 12:01:14 2014 +0200
@@ -145,4 +145,19 @@
 
         return probes;
     }
+
+    public void forget(Source source) {
+        final Set<LineLocation> mappedLines = lineToProbesMap.keySet();
+        if (mappedLines.size() > 0) {
+            List<LineLocation> forgetLines = new ArrayList<>();
+            for (LineLocation line : mappedLines) {
+                if (line.getSource().equals(source)) {
+                    forgetLines.add(line);
+                }
+            }
+            for (LineLocation line : forgetLines) {
+                lineToProbesMap.remove(line);
+            }
+        }
+    }
 }