Add comments

[crea.git] / appli / creaNewProject / NewProject / bbtk_mySamplePackage_PKG / src / bbmySamplePackagemySampleBoxWithITK.h

diff --git a/appli/creaNewProject/NewProject/bbtk_mySamplePackage_PKG/src/bbmySamplePackagemySampleBoxWithITK.h b/appli/creaNewProject/NewProject/bbtk_mySamplePackage_PKG/src/bbmySamplePackagemySampleBoxWithITK.h
index 18789b6a266e3391ce89e28887883c7996458fe8..c2c36f6af2290bf4bc708c3792f829c4aef61ab3 100644 (file)
--- a/appli/creaNewProject/NewProject/bbtk_mySamplePackage_PKG/src/bbmySamplePackagemySampleBoxWithITK.h
+++ b/appli/creaNewProject/NewProject/bbtk_mySamplePackage_PKG/src/bbmySamplePackagemySampleBoxWithITK.h
@@ -1,10 +1,59 @@
 
 #ifndef __bbmySamplePackagemySampleBoxWithITK_h_INCLUDED__
 #define __bbmySamplePackagemySampleBoxWithITK_h_INCLUDED__
+
+//
+//
+// To create a new Package with boxes that use itk don't forget 3 steps:
+//
+// 1. Modify de CMakeList.txt file of the whole project to include cmake itk mechanism.
+//             1.1 Put ON the option          SET(USE_ITK       ON)
+//
+// 2. Modify de CMakeList.txt file of this package.
+//             2.1. Uncomment the line  SET(${BBTK_PACKAGE_NAME}_USE_ITK  ON)
+//             2.2. Add the line:       
+//                             ${BBTK_DIR}/../../../include/bbitk
+//                  In the SET(${BBTK_PACKAGE_NAME}_INCLUDE_DIRS   section
+//
+// 3. In the .h file of your box, don't forget:
+//             3.1. Define the DIM and the TYPE of the images
+//                   (Comment out the BBTK_ITK_IMAGE_DIM_xxx
+//                    and the BBTK_ITK_IMAGE_TYPE_xxx_t you don't need)
+//             3.2. Define the ProcessTemplate method with the 
+//                BBTK_TEMPLATE_ITK_IMAGE_SWITCH mechanism.
+//
+//     Note: We use a generic bbtk_itk_pointer to transfere the itkImage..
+//
+//
+
+
+
+// Definition of the DIM and TYPE of the itkImage ..
+#define BBTK_ITK_IMAGE_DIM_2
+#define BBTK_ITK_IMAGE_DIM_3
+#define BBTK_ITK_IMAGE_DIM_4
+#define BBTK_ITK_IMAGE_TYPE_int8_t
+#define BBTK_ITK_IMAGE_TYPE_uint8_t
+#define BBTK_ITK_IMAGE_TYPE_int16_t
+#define BBTK_ITK_IMAGE_TYPE_uint16_t
+#define BBTK_ITK_IMAGE_TYPE_int32_t
+#define BBTK_ITK_IMAGE_TYPE_uint32_t
+#define BBTK_ITK_IMAGE_TYPE_float
+#define BBTK_ITK_IMAGE_TYPE_double
+
+
+
 #include "bbmySamplePackage_EXPORT.h"
 #include "bbtkAtomicBlackBox.h"
 #include "iostream"
 
+#include "bbitkImage.h"
+
+#include <itkBinaryThresholdImageFilter.h>
+
+#include "itkResampleImageFilter.h"
+#include "itkNearestNeighborInterpolateImageFunction.h"
+
 namespace bbmySamplePackage
 {
 
@@ -14,22 +63,67 @@ class bbmySamplePackage_EXPORT mySampleBoxWithITK
 {
   BBTK_BLACK_BOX_INTERFACE(mySampleBoxWithITK,bbtk::AtomicBlackBox);
 
-  BBTK_DECLARE_INPUT(In,double);
-  BBTK_DECLARE_OUTPUT(Out,double);
+  BBTK_DECLARE_INPUT(In,bbitk::anyImagePointer);
+  BBTK_DECLARE_INPUT(Lower,double);
+  BBTK_DECLARE_INPUT(Upper,double);
+  BBTK_DECLARE_OUTPUT(Out,bbitk::anyImagePointer);
+       
   BBTK_PROCESS(Process);
-  void Process();
-
+private:
+  inline void Process();
+  template <class T> void ProcessTemplated();
+  itk::Object* mOutput;
 };
 
 BBTK_BEGIN_DESCRIBE_BLACK_BOX(mySampleBoxWithITK,bbtk::AtomicBlackBox);
-BBTK_NAME("mySampleBoxWithITK");
-BBTK_AUTHOR("_author_");
-BBTK_DESCRIPTION("_description_");
-BBTK_CATEGORY("");
-BBTK_INPUT(mySampleBoxWithITK,In,"First input",double,"");
-BBTK_OUTPUT(mySampleBoxWithITK,Out,"First output",double,"");
+       BBTK_NAME("mySampleBoxWithITK");
+       BBTK_AUTHOR("Info-Dev");
+       BBTK_DESCRIPTION("Example ITK box - CreaNewProject");
+       BBTK_CATEGORY("filter");
+       BBTK_INPUT(mySampleBoxWithITK,In,"Input image. Can be any itk::Image<T,D>*",bbitk::anyImagePointer,"");
+       BBTK_INPUT(mySampleBoxWithITK,Lower,"(200 default) Lower value for the threshold*",double,"");
+       BBTK_INPUT(mySampleBoxWithITK,Upper,"(1200 default) Upper value for the threshold*",double,"");
+       BBTK_OUTPUT(mySampleBoxWithITK,Out,"Output Image",bbitk::anyImagePointer,"");
 BBTK_END_DESCRIBE_BLACK_BOX(mySampleBoxWithITK);
 
+       
+       
+//===================================================
+void mySampleBoxWithITK::Process()
+{
+       bbtk::TypeInfo t = bbGetInputIn().type();
+       BBTK_TEMPLATE_ITK_IMAGE_SWITCH(t, this->ProcessTemplated);
+}
+//===================================================
+       
+
+//===================================================
+template <class ImageType> 
+void mySampleBoxWithITK::ProcessTemplated()
+{
+       bbtkDebugMessageInc("Core",9,"bbitk::BinaryThresholdImageFilter::Process<"<<bbtk::TypeName<ImageType>()<<">()"<<std::endl);
+
+       typedef itk::BinaryThresholdImageFilter<ImageType,ImageType> FilterType;
+       typename FilterType::Pointer filter = FilterType::New();
+       
+       // Input
+       ImageType* in = this->bbGetInputIn().get<ImageType*>();
+       filter->SetInput( in );
+       filter->SetInsideValue (255);
+       filter->SetOutsideValue (0);
+       filter->SetLowerThreshold( bbGetInputLower() );
+       filter->SetUpperThreshold( bbGetInputUpper() );
+
+       filter->Update();
+       filter->GetOutput()->Register();
+       if (mOutput) mOutput->UnRegister();
+       mOutput = filter->GetOutput();
+       this->bbSetOutputOut( filter->GetOutput() );
+       
+       bbtkDebugDecTab("Core",9);
+}
+//===================================================
+       
 }
 // EO namespace bbmySamplePackage