X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=tools%2FclitkNormalizeImageFilterGenericFilter.txx;h=ea8773334d2307d99953eab8c0a133690c215bc8;hb=7f257bb16342aa315291cef76f33d9fe727fc07b;hp=e6911afde54adce2a07e5a3656d18520ca235b43;hpb=81fa2ccb7f96adb3df3f01182f9ec67cd181f859;p=clitk.git

diff --git a/tools/clitkNormalizeImageFilterGenericFilter.txx b/tools/clitkNormalizeImageFilterGenericFilter.txx
index e6911af..ea87733 100644
--- a/tools/clitkNormalizeImageFilterGenericFilter.txx
+++ b/tools/clitkNormalizeImageFilterGenericFilter.txx
@@ -144,13 +144,28 @@ namespace clitk
         ito.GoToBegin();
         itm.GoToBegin();
 
+        double total = 0.0;
         while (!ito.IsAtEnd()) {
+          if(itm.Get() == 1) {
+            ito.Set(((float) it.Get() - minImg)/(maxImg-minImg));
+            total += ito.Get();
+          }
+          ++it;
+          ++ito;
+          ++itm;
+        }
+
+        // Normalisation wrt total ?
+        if (mArgsInfo.total_normalisation_flag) {
+          ito.GoToBegin();
+          itm.GoToBegin();
+          while (!ito.IsAtEnd()) {
             if(itm.Get() == 1) {
-                ito.Set(((float) it.Get() - minImg)/(maxImg-minImg));
+              ito.Set(ito.Get()/total);
             }
-            ++it;
             ++ito;
             ++itm;
+          }
         }
         //
         //