Romulo:

[clitk.git] / segmentation / clitkAnatomicalFeatureDatabase.txx

diff --git a/segmentation/clitkAnatomicalFeatureDatabase.txx b/segmentation/clitkAnatomicalFeatureDatabase.txx
index 7ec7f971ac05143d9afdf69b441f1b510c860662..1aad4f6eec882f08f2ee2964feb095b75b4258f6 100644 (file)
--- a/segmentation/clitkAnatomicalFeatureDatabase.txx
+++ b/segmentation/clitkAnatomicalFeatureDatabase.txx
@@ -54,6 +54,8 @@ SetImage(TagType tag, std::string f, typename ImageType::Pointer image, bool wri
 {
   SetImageFilename(tag, f);
   m_MapOfImage[tag] = &(*image);
+  // I add a reference count because the cache is not a smartpointer
+  image->SetReferenceCount(image->GetReferenceCount()+1);
   if (write) {
     writeImage<ImageType>(image, f);
   }
@@ -71,7 +73,6 @@ ReleaseImage(std::string tag)
   }
   else {
     typename ImageType::Pointer image = GetImage<ImageType>(tag);
-    DD(image->GetReferenceCount());
     image->SetReferenceCount(image->GetReferenceCount()-1);
     m_MapOfImage.erase(tag);
   }