]> Creatis software - creaVtk.git/blobdiff - bbtk_creaVtk_PKG/src/bbcreaVtkApplyMask.cxx
#3399 creaVtk Feature New Normal - box ApplayMask new Input BackroundToAnalice
[creaVtk.git] / bbtk_creaVtk_PKG / src / bbcreaVtkApplyMask.cxx
index 264f55e4a42562403d1ade926ea4eda0e33d5646..52c1ed2d2ac50373316af68e4a7085a0eeaad67f 100644 (file)
@@ -34,61 +34,98 @@ void ApplyMask::Process()
 
        if ((bbGetInputImage()!=NULL) && (bbGetInputMask()!=NULL))
        {
-               int extImage[6];
+               int                             extImage[6];
+               int                             extMask[6];
+               vtkImageData            *resultImage=NULL;
                bbGetInputImage()->GetExtent( extImage );
-               int dimXImage = extImage[1]-extImage[0]+1;
-               int dimYImage = extImage[3]-extImage[2]+1;
-               int dimZImage = extImage[5]-extImage[4]+1;
-
-               int extMask[6];
                bbGetInputImage()->GetExtent( extMask );
-               int dimXMask = extMask[1]-extMask[0]+1;
-               int dimYMask = extMask[3]-extMask[2]+1;
-               int dimZMask = extMask[5]-extMask[4]+1;
+               int dimXImage           = extImage[1]-extImage[0]+1;
+               int dimYImage           = extImage[3]-extImage[2]+1;
+               int dimZImage           = extImage[5]-extImage[4]+1;
+               int dimXMask            = extMask[1]-extMask[0]+1;
+               int dimYMask            = extMask[3]-extMask[2]+1;
+               int dimZMask            = extMask[5]-extMask[4]+1;
+
+               double background       = bbGetInputBackground();
+               double bgToAnalice      = bbGetInputBackgroundToAnalice();
+
+               if (bbGetInputType()==0)
+               {
+                       if (bbGetOutputOut()!=NULL) 
+                       {
+                               bbGetOutputOut()->Delete();
+                       }       
+
+                       resultImage = vtkImageData::New();
+                       resultImage->Initialize();
+                       resultImage->SetSpacing( bbGetInputImage()->GetSpacing() );
+                       resultImage->SetDimensions(  dimXImage, dimYImage, dimZImage );
+                       resultImage->AllocateScalars( bbGetInputImage()->GetScalarType(),1 );
+               } // if Type==0                 
+
+               if (bbGetInputType()==1)
+               {
+                       resultImage=bbGetInputImage();
+               } // if Type==1
 
+               if (bbGetInputType()==2)
+               {
+                       resultImage=bbGetInputImage();
+               } // if Type==2
 
-               if (bbGetOutputOut()!=NULL) 
+               if (bbGetInputType()==3)
                {
-                       bbGetOutputOut()->Delete();
-               }       
+                       resultImage=bbGetInputImage();
+               } // if Type==3
 
-               vtkImageData *resultImage=NULL;
-               resultImage = vtkImageData::New();
-               resultImage->Initialize();
-               resultImage->SetSpacing( bbGetInputImage()->GetSpacing() );
-               resultImage->SetDimensions(  dimXImage, dimYImage, dimZImage );
-               resultImage->AllocateScalars( bbGetInputImage()->GetScalarType(),1 );
-                       
 
                if ((dimXImage==dimXMask) &&
                        (dimYImage==dimYMask) &&
                        (dimZImage==dimZMask)) 
                        {
-
-
                                long int i, size = dimXImage * dimYImage * dimZImage;
                                #pragma omp parallel for 
                                for (i=0; i<size;i++)
                                {
-
                                        DEF_POINTER_IMAGE_VTK_CREA(vI,ssI,pI,stI, bbGetInputImage() );
                                        DEF_POINTER_IMAGE_VTK_CREA(vM,ssM,pM,stM, bbGetInputMask() );
                                        DEF_POINTER_IMAGE_VTK_CREA(vO,ssO,pO,stO, resultImage );
-
-                                       GETVALUE2_VTK_CREA(vI,pI,stI,i);
                                        GETVALUE2_VTK_CREA(vM,pM,stM,i);
-                                       if (vM!=0)
-                                       {
-                                               vO = vI;
-                                       } else {
-                                               vO = bbGetInputBackground(); 
-                                       }
-                                       SETVALUE2_VTK_CREA(vO,pO,stO,i);        
+
+                                       if (bbGetInputType()==0)  //Applay mask to new Image
+                                       {               
+                                               GETVALUE2_VTK_CREA(vI,pI,stI,i);
+                                               if (vM!=bgToAnalice)
+                                               {
+                                                       vO = vI;
+                                               } else {
+                                                       vO = background; 
+                                               } // if vM
+                                               SETVALUE2_VTK_CREA(vO,pO,stO,i);        
+                                       } // if Type==0
+
+                                       if (bbGetInputType()==1)  //Modifiy the Input image with the Label value in Mask place
+                                       {               
+                                               if (vM!=bgToAnalice)
+                                               {
+                                                       vO = bbGetInputLabel();
+                                                       SETVALUE2_VTK_CREA(vO,pO,stO,i);        
+                                               }
+                                       } // if Type==1
+
+                                       if (bbGetInputType()==2)  //Modifiy the Input image with the Mask value
+                                       {               
+                                               if (vM!=bgToAnalice)
+                                               {
+                                                       vO = vM;
+                                                       SETVALUE2_VTK_CREA(vO,pO,stO,i);        
+                                               }
+                                       } // if Type==2
+
                                } // for
                } //if dim
-       bbSetOutputOut( resultImage );
+               bbSetOutputOut( resultImage );
        } // 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)
@@ -100,7 +137,10 @@ void ApplyMask::bbUserSetDefaultValues()
 //    Here we initialize the input 'In' to 0
    bbSetInputImage(NULL);
    bbSetInputMask(NULL);
-   bbSetInputBackground(0);
+   bbSetInputBackground(0);             // output image Type 0
+   bbSetInputBackgroundToAnalice(0);
+   bbSetInputLabel(255);
+   bbSetInputType(0);
    bbSetOutputOut(NULL);
   
 }