CreateMRML/MRMLVolumeHelper.h at master · fbudin69500/CreateMRML · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
#ifndef _MRMLVolumeHelper_h
#define _MRMLVolumeHelper_h

#include "MRMLColorableHelper.h"


class MRMLVolumeHelper : public MRMLColorableHelper
{
   public:
      MRMLVolumeHelper()
      {
         m_VolumeType = "scalar" ;
         m_LabelMap = false ;
      }
      void Print()
      {
         MRMLColorableHelper::Print() ;
         std::cout << "Image Type: " << m_VolumeType << std::endl ;
      }
      void LabelMap( bool label )
      {
        m_LabelMap = label ;
      }
      bool IsLabelMap()
      {
        return m_LabelMap ;
      }
      std::string GetType()
      {
         return "Volume" ;
      }
      int SetVolumeType( std::string type )
      {
         if( type.compare( "scalar" )
             && type.compare( "DTI" )
             && type.compare( "DWI" )
             && type.compare( "vector" )
           )
         {
            return 1 ;
         }
         m_VolumeType = type ;
         return 0 ;
      }
      std::string GetVolumeType()
      {
         return m_VolumeType ;
      }
      void PrintVolumeTypes()
      {
         std::cout << "scalar" << std::endl ;
         std::cout << "DTI" << std::endl ;
         std::cout << "DWI" << std::endl ;
         std::cout << "vector" << std::endl ;
      }
   private:
      std::string m_VolumeType ;
      bool m_LabelMap ;
};

#endif