segmentation/clitkExtractLymphStationsFilter.h

   1 /*=========================================================================
   2   Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv
   3
   4   Authors belong to:
   5   - University of LYON              http://www.universite-lyon.fr/
   6   - Léon Bérard cancer center       http://oncora1.lyon.fnclcc.fr
   7   - CREATIS CNRS laboratory         http://www.creatis.insa-lyon.fr
   8
   9   This software is distributed WITHOUT ANY WARRANTY; without even
  10   the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  11   PURPOSE.  See the copyright notices for more information.
  12
  13   It is distributed under dual licence
  14
  15   - BSD        See included LICENSE.txt file
  16   - CeCILL-B   http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
  17   ======================================================================-====*/
  18
  19 #ifndef CLITKEXTRACTLYMPHSTATIONSFILTER_H
  20 #define CLITKEXTRACTLYMPHSTATIONSFILTER_H
  21
  22 // clitk
  23 #include "clitkFilterBase.h"
  24 #include "clitkFilterWithAnatomicalFeatureDatabaseManagement.h"
  25
  26 // vtk
  27 #include <vtkPolyData.h>
  28
  29 namespace clitk {
  30
  31   //--------------------------------------------------------------------
  32   /*
  33     Try to extract the LymphStations part of a thorax CT.
  34     Need a set of Anatomical Features (AFDB)
  35   */
  36   //--------------------------------------------------------------------
  37
  38   template <class TImageType>
  39   class ITK_EXPORT ExtractLymphStationsFilter:
  40     public virtual clitk::FilterBase,
  41     public clitk::FilterWithAnatomicalFeatureDatabaseManagement,
  42     public itk::ImageToImageFilter<TImageType, itk::Image<uchar, 3> >
  43   {
  44
  45   public:
  46     /** Standard class typedefs. */
  47     typedef itk::ImageToImageFilter<TImageType, itk::Image<uchar, 3> > Superclass;
  48     typedef ExtractLymphStationsFilter          Self;
  49     typedef itk::SmartPointer<Self>             Pointer;
  50     typedef itk::SmartPointer<const Self>       ConstPointer;
  51
  52     /** Method for creation through the object factory. */
  53     itkNewMacro(Self);
  54
  55     /** Run-time type information (and related methods). */
  56     itkTypeMacro(ExtractLymphStationsFilter, ImageToImageFilter);
  57
  58     /** Some convenient typedefs. */
  59     typedef TImageType                       ImageType;
  60     typedef typename ImageType::ConstPointer ImageConstPointer;
  61     typedef typename ImageType::Pointer      ImagePointer;
  62     typedef typename ImageType::RegionType   ImageRegionType;
  63     typedef typename ImageType::PixelType    ImagePixelType;
  64     typedef typename ImageType::SizeType     ImageSizeType;
  65     typedef typename ImageType::IndexType    ImageIndexType;
  66     typedef typename ImageType::PointType    ImagePointType;
  67
  68     typedef uchar MaskImagePixelType;
  69     typedef itk::Image<MaskImagePixelType, 3>    MaskImageType;
  70     typedef typename MaskImageType::Pointer      MaskImagePointer;
  71     typedef typename MaskImageType::RegionType   MaskImageRegionType;
  72     typedef typename MaskImageType::SizeType     MaskImageSizeType;
  73     typedef typename MaskImageType::IndexType    MaskImageIndexType;
  74     typedef typename MaskImageType::PointType    MaskImagePointType;
  75
  76     typedef itk::Image<MaskImagePixelType, 2>    MaskSliceType;
  77     typedef typename MaskSliceType::Pointer      MaskSlicePointer;
  78     typedef typename MaskSliceType::PointType    MaskSlicePointType;
  79     typedef typename MaskSliceType::RegionType   MaskSliceRegionType;
  80     typedef typename MaskSliceType::SizeType     MaskSliceSizeType;
  81     typedef typename MaskSliceType::IndexType    MaskSliceIndexType;
  82
  83     /** ImageDimension constants */
  84     itkStaticConstMacro(ImageDimension, unsigned int, ImageType::ImageDimension);
  85     FILTERBASE_INIT;
  86
  87     itkGetConstMacro(BackgroundValue, MaskImagePixelType);
  88     itkGetConstMacro(ForegroundValue, MaskImagePixelType);
  89     itkSetMacro(BackgroundValue, MaskImagePixelType);
  90     itkSetMacro(ForegroundValue, MaskImagePixelType);
  91
  92     // Station 8
  93     // itkSetMacro(DistanceMaxToAnteriorPartOfTheSpine, double);
  94     //itkGetConstMacro(DistanceMaxToAnteriorPartOfTheSpine, double);
  95     itkSetMacro(EsophagusDiltationForAnt, MaskImagePointType);
  96     itkGetConstMacro(EsophagusDiltationForAnt, MaskImagePointType);
  97     itkSetMacro(EsophagusDiltationForRight, MaskImagePointType);
  98     itkGetConstMacro(EsophagusDiltationForRight, MaskImagePointType);
  99     itkSetMacro(InjectedThresholdForS8, double);
 100     itkGetConstMacro(InjectedThresholdForS8, double);
 101
 102     // Station 7
 103     itkGetConstMacro(S7_UseMostInferiorPartOnlyFlag, bool);
 104     itkSetMacro(S7_UseMostInferiorPartOnlyFlag, bool);
 105     itkBooleanMacro(S7_UseMostInferiorPartOnlyFlag);
 106
 107     // All stations
 108     bool GetComputeStation(std::string s);
 109     void AddComputeStation(std::string station) ;
 110     void SetFuzzyThreshold(std::string station, std::string tag, double value);
 111     double GetFuzzyThreshold(std::string station, std::string tag);
 112     itkGetConstMacro(ComputeStationsSupportsFlag, bool);
 113     itkSetMacro(ComputeStationsSupportsFlag, bool);
 114     itkBooleanMacro(ComputeStationsSupportsFlag);
 115
 116   protected:
 117     ExtractLymphStationsFilter();
 118     virtual ~ExtractLymphStationsFilter() {}
 119
 120     virtual void GenerateOutputInformation();
 121     virtual void GenerateInputRequestedRegion();
 122     virtual void GenerateData();
 123
 124     ImageConstPointer  m_Input;
 125     MaskImagePointer   m_Mediastinum;
 126     MaskImagePointer   m_Working_Support;
 127     std::map<std::string, MaskImagePointer> m_ListOfStations;
 128     std::map<std::string, MaskImagePointer> m_ListOfSupports;
 129     MaskImagePixelType m_BackgroundValue;
 130     MaskImagePixelType m_ForegroundValue;
 131     std::map<std::string, bool> m_ComputeStationMap;
 132
 133     bool CheckForStation(std::string station);
 134     void Remove_Structures(std::string station, std::string s);
 135
 136     // Functions common to several stations
 137     double FindCarina();
 138     double FindApexOfTheChest();
 139     double FindSuperiorBorderOfAorticArch();
 140     double FindInferiorBorderOfAorticArch();
 141     void FindLeftAndRightBronchi();
 142     void FindLineForS7S8Separation(MaskImagePointType & A, MaskImagePointType & B);
 143     MaskImagePointer FindAntPostVessels();
 144     MaskImagePointer FindAntPostVessels2();
 145
 146     // Global parameters
 147     typedef std::map<std::string, double> FuzzyThresholdByStructureType;
 148     std::map<std::string, FuzzyThresholdByStructureType> m_FuzzyThreshold;
 149
 150     // Station's supports
 151     void ExtractStationSupports();
 152     void Support_SupInf_S1RL();
 153     void Support_LeftRight_S1R_S1L();
 154     void Support_SupInf_S2R_S2L();
 155     void Support_LeftRight_S2R_S2L();
 156     void Support_SupInf_S4R_S4L();
 157     void Support_LeftRight_S4R_S4L();
 158     void Support_Post_S1S2S4();
 159     void Support_S3P();
 160     void Support_S3A();
 161     void Support_S5();
 162     void Support_S6();
 163
 164     MaskImagePointer LimitsWithTrachea(MaskImageType * input,
 165                                        int extremaDirection, int lineDirection,
 166                                        double offset, double maxSupPosition);
 167     MaskImagePointer LimitsWithTrachea(MaskImageType * input,
 168                                        int extremaDirection, int lineDirection,
 169                                        double offset);
 170
 171     // Station 8
 172     // double m_DistanceMaxToAnteriorPartOfTheSpine;
 173     double m_DiaphragmInferiorLimit;
 174     double m_OriginOfRightMiddleLobeBronchusZ;
 175     double m_InjectedThresholdForS8;
 176     MaskImagePointer m_Esophagus;
 177     MaskImagePointType m_EsophagusDiltationForAnt;
 178     MaskImagePointType m_EsophagusDiltationForRight;
 179
 180     void ExtractStation_8();
 181     void ExtractStation_8_SetDefaultValues();
 182     void ExtractStation_8_SI_Limits();
 183     void ExtractStation_8_Ant_Limits();
 184     void ExtractStation_8_Left_Sup_Limits();
 185     void ExtractStation_8_Left_Inf_Limits();
 186     void ExtractStation_8_Single_CCL_Limits();
 187     void ExtractStation_8_Remove_Structures();
 188
 189     // Station 3P
 190     void ExtractStation_3P();
 191     void ExtractStation_3P_SetDefaultValues();
 192     void ExtractStation_3P_LR_inf_Limits();
 193     void ExtractStation_3P_LR_sup_Limits_2();
 194     void ExtractStation_3P_Remove_Structures();
 195     void ExtractStation_3P_LR_sup_Limits();
 196
 197     // Station 3A
 198     void ExtractStation_3A();
 199     void ExtractStation_3A_SetDefaultValues();
 200     void ExtractStation_3A_AntPost_S5();
 201     void ExtractStation_3A_AntPost_S6();
 202     void ExtractStation_3A_AntPost_Superiorly();
 203     void ExtractStation_3A_Remove_Structures();
 204
 205     void ExtractStation_3A_SI_Limits();
 206     void ExtractStation_3A_Ant_Limits();
 207     void ExtractStation_3A_Post_Limits();
 208
 209     // Station 2RL
 210     void ExtractStation_2RL();
 211     void ExtractStation_2RL_SetDefaultValues();
 212     void ExtractStation_2RL_SI_Limits();
 213     void ExtractStation_2RL_Ant_Limits();
 214     void ExtractStation_2RL_Ant_Limits2();
 215     void ExtractStation_2RL_Post_Limits();
 216     void ExtractStation_2RL_LR_Limits();
 217     void ExtractStation_2RL_Remove_Structures();
 218     void ExtractStation_2RL_SeparateRL();
 219     vtkSmartPointer<vtkPolyData> Build3DMeshFrom2DContour(const std::vector<ImagePointType> & points);
 220
 221     // Station 7
 222     void ExtractStation_7();
 223     void ExtractStation_7_SetDefaultValues();
 224     void ExtractStation_7_SI_Limits();
 225     void ExtractStation_7_RL_Interior_Limits();
 226
 227
 228     void ExtractStation_7_RL_Limits_OLD();
 229     void ExtractStation_7_Posterior_Limits();
 230     void ExtractStation_7_Remove_Structures();
 231     bool m_S7_UseMostInferiorPartOnlyFlag;
 232     bool m_ComputeStationsSupportsFlag;
 233     MaskImagePointer m_Working_Trachea;
 234     MaskImagePointer m_LeftBronchus;
 235     MaskImagePointer m_RightBronchus;
 236     typedef std::vector<MaskImageType::PointType> ListOfPointsType;
 237     ListOfPointsType  m_RightMostInLeftBronchus;
 238     ListOfPointsType  m_AntMostInLeftBronchus;
 239     ListOfPointsType  m_PostMostInLeftBronchus;
 240     ListOfPointsType  m_LeftMostInRightBronchus;
 241     ListOfPointsType  m_AntMostInRightBronchus;
 242     ListOfPointsType  m_PostMostInRightBronchus;
 243
 244     void FindExtremaPointsInBronchus(MaskImagePointer input,
 245                                      int direction,
 246                                      double distance_max_from_center_point,
 247                                      ListOfPointsType & LR,
 248                                      ListOfPointsType & Ant,
 249                                      ListOfPointsType & Post);
 250     // Station 4RL
 251     void ExtractStation_4RL();
 252     void ExtractStation_4RL_SI_Limits();
 253     void ExtractStation_4RL_LR_Limits();
 254     void ExtractStation_4RL_AP_Limits();
 255     MaskImagePointer m_RightSupport;
 256     MaskImagePointer m_LeftSupport;
 257
 258   private:
 259     ExtractLymphStationsFilter(const Self&); //purposely not implemented
 260     void operator=(const Self&); //purposely not implemented
 261
 262   }; // end class
 263   //--------------------------------------------------------------------
 264
 265 } // end namespace clitk
 266 //--------------------------------------------------------------------
 267
 268 #ifndef ITK_MANUAL_INSTANTIATION
 269 #include "clitkExtractLymphStationsFilter.txx"
 270 #include "clitkExtractLymphStation_Supports.txx"
 271 #include "clitkExtractLymphStation_8.txx"
 272 #include "clitkExtractLymphStation_3P.txx"
 273 #include "clitkExtractLymphStation_2RL.txx"
 274 #include "clitkExtractLymphStation_3A.txx"
 275 #include "clitkExtractLymphStation_7.txx"
 276 #include "clitkExtractLymphStation_4RL.txx"
 277 #endif
 278
 279 #endif