]> Creatis software - gdcm.git/blob - vtk/vtkGdcm4DSplitter.h
6724a41f41b9026e791dccf0ab6a117c3221f370
[gdcm.git] / vtk / vtkGdcm4DSplitter.h
1 /*=========================================================================
2                                                                                 
3   Program:   gdcm
4   Module:    $RCSfile: vtkGdcm4DSplitter.h,v $
5   Language:  C++
6   Date:      $Date: 2011/04/04 17:01:03 $
7   Version:   $Revision: 1.6 $
8                                                                                 
9   Copyright (c) CREATIS (Centre de Recherche et d'Applications en Traitement de
10   l'Image). All rights reserved. See Doc/License.txt or
11   http://www.creatis.insa-lyon.fr/Public/Gdcm/License.html for details.
12                                                                                 
13      This software is distributed WITHOUT ANY WARRANTY; without even
14      the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
15      PURPOSE.  See the above copyright notices for more information.
16                                                                                 
17 =========================================================================*/
18
19 #ifndef _VTKGDCM4DSPLITTER_H_
20 #define _VTKGDCM4DSPLITTER_H_
21
22 #include <vector>
23 #include <vtkImageData.h>
24 #include "gdcmDirList.h"
25 #include "gdcmFile.h"
26
27 typedef  bool (*FoncComp)                       (GDCM_NAME_SPACE::File *file1, GDCM_NAME_SPACE::File *file2);
28
29 #define CALL_MEMBER_FONC(object, ptrToFoncMember)  ((object).*(ptrToFoncMember))
30  
31 //namespace GDCM_NAME_SPACE
32 //{
33   class vtkGdcm4DSplitter {
34     public:
35
36        vtkGdcm4DSplitter();
37        ~vtkGdcm4DSplitter(){};
38        bool setDirName(std::string &dirName);
39        bool setVectDirName(std::vector<std::string> &vectDirName);
40        bool setVectFileName(std::vector<std::string> &vectFileName);
41
42        // Split
43        // =====
44        inline void setSplitOnPosition()   {SplitOnPosition=true;  SplitOnOrientation=false; SplitOnTag=false;}
45        inline void setSplitOnOrientation(){SplitOnPosition=false; SplitOnOrientation=true;  SplitOnTag=false;}
46        inline void setSplitOnTag(unsigned short int splitGroup, unsigned short int splitElem)
47                                           {SplitOnPosition=false; SplitOnOrientation=false; SplitOnTag=true;
48                                            SplitGroup=splitGroup;    SplitElem=splitElem;}
49        inline void setSplitConvertToFloat(bool conv) {SplitConvertToFloat=conv;}
50
51        // Sort
52        // ====
53        inline void setSortOnPosition()   {SortOnPosition=true; SortOnOrientation=false; SortOnTag=false; SortOnFileName=false; SortOnUserFunction=false;
54                                           SortOnPosition=true;}
55       // use setSortOnUserFunction, instead!
56       // inline void setSortOnTag(unsigned short int sortGroup, unsigned short int sortElem)
57       //                                   {SortOnPosition=false; SortOnOrientation=false; SortOnTag=true; SortOnFileName=false; SortOnUserFunction=false;
58       //                                    SortGroup=sortGroup;  SortElem=sortElem;}
59
60        inline void setSortOnUserFunction (FoncComp f)
61                                          { UserCompareFunction=f;
62                                            SortOnPosition=false; SortOnOrientation=false; SortOnTag=false; SortOnFileName=false; SortOnUserFunction=true;}
63
64        inline void setSortConvertToFloat(bool conv) {SortConvertToFloat=conv;}
65
66        inline void setSortOnFileName()   {SortOnPosition=false; SortOnOrientation=false; SortOnTag=false; SortOnFileName=true; SortOnUserFunction=false;}
67
68        inline void setRecursive(bool recursive) { Recursive=recursive;}
69
70        // SortOnOrientation : not yet made
71        //inline void setSortOnOrientation(){SortOnPosition=false; SortOnOrientation=true;  SortOnTag=false; SortOnFileName=false; SortOnUserFunction=false;}
72
73         std::vector<vtkImageData*> *GetImageDataVector();
74         vtkImageData *GetImageData();
75
76         bool Go();
77
78     protected:
79     private:
80        bool CompareOnSortTag              (GDCM_NAME_SPACE::File *file1, GDCM_NAME_SPACE::File *file2);
81        bool CompareOnSortTagConvertToFloat(GDCM_NAME_SPACE::File *file1, GDCM_NAME_SPACE::File *file2);
82
83     // Data
84     // ----
85     public:
86     protected:
87     private:
88        std::string DirName;
89        std::vector<std::string> VectDirName;
90        std::vector<std::string> VectFileName;
91
92        bool SplitOnPosition;
93        bool SplitOnOrientation;
94        bool SplitOnTag;
95
96        unsigned short int SplitGroup;
97        unsigned short int SplitElem;
98        bool SplitConvertToFloat;
99
100        bool SortOnPosition;
101        bool SortOnOrientation;
102        bool SortOnTag;
103        bool SortOnFileName;
104        bool SortOnUserFunction;
105
106        FoncComp UserCompareFunction;
107
108        unsigned short int SortGroup;
109        unsigned short int SortElem;
110        bool SortConvertToFloat;
111
112        bool Recursive;
113        int TypeDir;
114        int TypeResult;
115        bool verbose;
116        std::vector<vtkImageData*> *ImageDataVector;
117        vtkImageData *ImageData;
118   };
119
120 //} // end namespace gdcm
121
122 //-----------------------------------------------------------------------------
123 #endif