#2865 BBTK Feature New Normal - new option in CleanExternalPlane DimImage+2 (Type 1)

[bbtk.git] / packages / vtk / src / bbvtkCleanExternalPlane.cxx

diff --git a/packages/vtk/src/bbvtkCleanExternalPlane.cxx b/packages/vtk/src/bbvtkCleanExternalPlane.cxx
index c7306d06451dfd0727445ada6899869c7cabf409..b6d8a6eea98b20195626935483d6cecffaf63846 100644 (file)
--- a/packages/vtk/src/bbvtkCleanExternalPlane.cxx
+++ b/packages/vtk/src/bbvtkCleanExternalPlane.cxx
@@ -35,30 +35,16 @@ namespace bbvtk
 
 BBTK_ADD_BLACK_BOX_TO_PACKAGE(vtk,CleanExternalPlane)
 BBTK_BLACK_BOX_IMPLEMENTATION(CleanExternalPlane,bbtk::AtomicBlackBox);
-//===== 
-// 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)
-//===== 
-void CleanExternalPlane::Process()
-{
-
-// THE MAIN PROCESSING METHOD BODY
-//   Here we simply set the input 'In' value to the output 'Out'
-//   And print out the output value
-// INPUT/OUTPUT ACCESSORS ARE OF THE FORM :
-//    void bbSet{Input|Output}NAME(const TYPE&)
-//    const TYPE& bbGet{Input|Output}NAME() const 
-//    Where :
-//    * NAME is the name of the input/output
-//      (the one provided in the attribute 'name' of the tag 'input')
-//    * TYPE is the C++ type of the input/output
-//      (the one provided in the attribute 'type' of the tag 'input')
 
+//----------------------------------------------------------------------
+void CleanExternalPlane::CleanTypeA(vtkImageData* image)
+{
        if ( bbGetInputIn()!=NULL )
        {
                int i,j;
                int ext[6];
                int dim[3];
-               bbGetInputIn()->GetWholeExtent(ext);
+               image->GetWholeExtent(ext);
                
                dim[0]=ext[1]-ext[0]+1;
                dim[1]=ext[3]-ext[2]+1;
@@ -69,36 +55,104 @@ void CleanExternalPlane::Process()
                {
                        for (j=0 ; j<dim[1] ; j++)
                        {
-                               bbGetInputIn()->SetScalarComponentFromDouble (i, j, 0           , 0, 0.0);
-                               bbGetInputIn()->SetScalarComponentFromDouble (i, j, dim[2]-1, 0, 0.0);
-                       }
-               }
+                               image->SetScalarComponentFromDouble (i, j, 0            , 0, 0.0);
+                               image->SetScalarComponentFromDouble (i, j, dim[2]-1, 0, 0.0);
+                       } // for j
+               }// for i
                
                // YZ plane
                for (i=0 ; i<dim[1] ; i++)
                {
                        for (j=0 ; j<dim[2] ; j++)
                        {
-                               bbGetInputIn()->SetScalarComponentFromDouble (0                 , i, j, 0, 0.0);
-                               bbGetInputIn()->SetScalarComponentFromDouble (dim[0]-1  , i, j, 0, 0.0);
-                       }
-               }
+                               image->SetScalarComponentFromDouble (0                  , i, j, 0, 0.0);
+                               image->SetScalarComponentFromDouble (dim[0]-1   , i, j, 0, 0.0);
+                       } // for j
+               } // for i
                
                // YZ plane
                for (i=0 ; i<dim[0] ; i++)
                {
                        for (j=0 ; j<dim[2] ; j++)
                        {
-                               bbGetInputIn()->SetScalarComponentFromDouble (i, 0                      , j, 0, 0.0);
-                               bbGetInputIn()->SetScalarComponentFromDouble (i, dim[1]-1       , j, 0, 0.0);
-                       }
-               }
+                               image->SetScalarComponentFromDouble (i, 0                       , j, 0, 0.0);
+                               image->SetScalarComponentFromDouble (i, dim[1]-1        , j, 0, 0.0);
+                       } // for j
+               }// for i
+       } // if In
+}
+
+//----------------------------------------------------------------------
+void CleanExternalPlane::CleanTypeB()
+{
+
+       int             ext[6]; 
+       int             dimA[3];        
+       int             dimB[3];        
+       bbGetInputIn()->GetWholeExtent(ext);
+
+       dimA[0] =       ext[1]-ext[0]+1;
+       dimA[1] =       ext[3]-ext[2]+1;
+       dimA[2] =       ext[5]-ext[4]+1;
+
+       dimB[0] =       dimA[0] + 2;
+       dimB[1] =       dimA[1] + 2;
+       dimB[2] =       dimA[2] + 2;
+
        
-               
-       }
-    
+       imageoutput->Initialize();
+       imageoutput->SetScalarType( bbGetInputIn()->GetScalarType() );  
+       imageoutput->SetSpacing( bbGetInputIn()->GetSpacing() );
+       imageoutput->SetDimensions(  dimB[0], dimB[1], dimB[2] );
+       imageoutput->AllocateScalars();
        
-       bbSetOutputOut( bbGetInputIn() );
+
+       int j,k;
+       long sizeBlock= dimA[0] * bbGetInputIn()->GetScalarSize();
+
+       for (j=0;j<dimA[1];j++)
+       {
+               for (k=0;k<dimA[2];k++)
+               {
+                       memcpy(   imageoutput->GetScalarPointer(1,j+1,k+1) , bbGetInputIn()->GetScalarPointer(0,j,k) , sizeBlock );
+               }
+       }
+       imageoutput->Update();
+       imageoutput->Modified();
+}
+
+
+//===== 
+// 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)
+//===== 
+void CleanExternalPlane::Process()
+{
+
+// THE MAIN PROCESSING METHOD BODY
+//   Here we simply set the input 'In' value to the output 'Out'
+//   And print out the output value
+// INPUT/OUTPUT ACCESSORS ARE OF THE FORM :
+//    void bbSet{Input|Output}NAME(const TYPE&)
+//    const TYPE& bbGet{Input|Output}NAME() const 
+//    Where :
+//    * NAME is the name of the input/output
+//      (the one provided in the attribute 'name' of the tag 'input')
+//    * TYPE is the C++ type of the input/output
+//      (the one provided in the attribute 'type' of the tag 'input')
+
+       if (bbGetInputType()==0)
+       {
+               CleanTypeA( bbGetInputIn() );
+               bbSetOutputOut( bbGetInputIn() );
+       }
+
+       if (bbGetInputType()==1)
+       {
+               CleanTypeB();
+               CleanTypeA(imageoutput);
+               bbSetOutputOut( imageoutput );
+       }
+
 
        
        
@@ -111,7 +165,8 @@ void CleanExternalPlane::bbUserSetDefaultValues()
 
 //  SET HERE THE DEFAULT INPUT/OUTPUT VALUES OF YOUR BOX 
 //    Here we initialize the input 'In' to 0
-   bbSetInputIn(0);
+   bbSetInputIn(NULL);
+   bbSetInputType(0);
   
 }
 //===== 
@@ -125,6 +180,7 @@ void CleanExternalPlane::bbUserInitializeProcessing()
 //    but this is where you should allocate the internal/output pointers 
 //    if any 
 
+       imageoutput = vtkImageData::New();
   
 }
 //===== 
@@ -137,9 +193,9 @@ void CleanExternalPlane::bbUserFinalizeProcessing()
 //    Here does nothing 
 //    but this is where you should desallocate the internal/output pointers 
 //    if any
-  
+       imageoutput->Delete();
 }
-}
-// EO namespace bbvtk
+
+} // EO namespace bbvtk