From abf96caaac59b2b021dc64d1ac45cd36f19ea9be Mon Sep 17 00:00:00 2001
From: remicres <remi.cresson@teledetection.fr>
Date: Fri, 2 Sep 2016 11:12:40 +0000
Subject: [PATCH] FIX: correct pixel memory footprint

---
 include/lsgrmController.txx | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/include/lsgrmController.txx b/include/lsgrmController.txx
index 74398a3..93c4eab 100644
--- a/include/lsgrmController.txx
+++ b/include/lsgrmController.txx
@@ -236,17 +236,17 @@ unsigned int Controller<TSegmenter>::GetNodeMemory()
   // Get the memory occupied by the graph, normalize it by n*n
   unsigned int memory = segmenter.GetGraphMemory() / (n*n);
 
-  itkDebugMacro(<<"Size of a node is " << memory);
+  itkWarningMacro(<<"Size of a node is " << memory);
 
   // Get the memory occupied by one pixel of the image
-  unsigned int pixelMemory = sizeof(typename ImageType::PixelType)
-      + sizeof(typename ImageType::InternalPixelType) * m_InputImage->GetNumberOfComponentsPerPixel();
+  unsigned int pixelMemory =  sizeof(m_InputImage->GetBufferPointer())
+      * m_InputImage->GetNumberOfComponentsPerPixel();
 
-  itkDebugMacro(<<"Size of an image pixel is " << pixelMemory);
+  itkWarningMacro(<<"Size of an image pixel is " << pixelMemory);
 
   memory += pixelMemory;
 
-  itkDebugMacro(<<"Size of a node+pixel is " << memory);
+  itkWarningMacro(<<"Size of a node+pixel is " << memory);
 
   return memory;
 }
-- 
GitLab