X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=bbtk_creaVtk_PKG%2Fsrc%2FbbcreaVtkApplyMask.cxx;h=03baf29676d02b9081c2e107db594ec6424ac379;hb=464deff51f697881cd76b3cfb934859cd7c31df7;hp=8d346915a5c27c3a62e2b757ed69c2a52194e986;hpb=dccc47ebcfa2ad7a2a6a8770f6975336897d7d06;p=creaVtk.git

diff --git a/bbtk_creaVtk_PKG/src/bbcreaVtkApplyMask.cxx b/bbtk_creaVtk_PKG/src/bbcreaVtkApplyMask.cxx
index 8d34691..03baf29 100644
--- a/bbtk_creaVtk_PKG/src/bbcreaVtkApplyMask.cxx
+++ b/bbtk_creaVtk_PKG/src/bbcreaVtkApplyMask.cxx
@@ -38,7 +38,7 @@ void ApplyMask::Process()
 		int 				extMask[6];
 		vtkImageData 		*resultImage=NULL;
 		bbGetInputImage()->GetExtent( extImage );
-		bbGetInputImage()->GetExtent( extMask );
+        bbGetInputMask()->GetExtent( extMask );
 		int dimXImage 		= extImage[1]-extImage[0]+1;
 		int dimYImage		= extImage[3]-extImage[2]+1;
 		int dimZImage 		= extImage[5]-extImage[4]+1;
@@ -133,7 +133,12 @@ if ( (k1omp==i) || (k2omp==i) || (k3omp==i) ||
 		} //if dim
 		resultImage->Modified();
 		bbSetOutputOut( resultImage );
-	} // if Image && Mask	
+    } else {
+        if (bbGetInputType()==2)
+        {
+            bbSetOutputOut( bbGetInputImage() );
+        }
+    }// if Image && Mask
 }
 //===== 
 // Before editing this file, make sure it's a file of your own (i.e.: it wasn't generated from xml description; if so : your modifications will be lost)