[Lemon-commits] Peter Kovacs: Memory leak bugfix in BellmanFord ...

[Lemon HG]

details:   http://lemon.cs.elte.hu/hg/lemon/rev/4db8d5ccd26b
changeset: 870:4db8d5ccd26b
user:      Peter Kovacs <kpeter [at] inf.elte.hu>
date:      Sun Dec 13 22:19:08 2009 +0100
description:
	Memory leak bugfix in BellmanFord (#51)

diffstat:

 lemon/bellman_ford.h |  8 +++++++-
 1 files changed, 7 insertions(+), 1 deletions(-)

diffs (25 lines):

diff --git a/lemon/bellman_ford.h b/lemon/bellman_ford.h
--- a/lemon/bellman_ford.h
+++ b/lemon/bellman_ford.h
@@ -237,7 +237,9 @@
 	_local_dist = true;
 	_dist = Traits::createDistMap(*_gr);
       }
-      _mask = new MaskMap(*_gr, false);
+      if(!_mask) {
+        _mask = new MaskMap(*_gr);
+      }
     }
     
   public :
@@ -404,6 +406,10 @@
 	  _process.push_back(it);
 	  _mask->set(it, true);
 	}
+      } else {
+	for (NodeIt it(*_gr); it != INVALID; ++it) {
+	  _mask->set(it, false);
+	}
       }
     }