Skip to content
GitLab
Select Git revision
  • master default
  • Pokiros/readmemd-edited-1496845753631
Find file BlameHistoryPermalink
lsgrmController.h 4.30 KiB
1 
#ifndef __LSGRM_CONTROLLER_H
2 
#define __LSGRM_CONTROLLER_H
3 
#include "lsgrmGetInternalMemory.h"
4 
#include "lsgrmSplitter.h"
5 
#include "lsgrmGraphOperations.h"
6 
#include "itkObject.h"
7 
#include "itkMacro.h"
8
9 
namespace lsgrm
10 
{
11 
template<class TSegmenter>
12 
class ITK_EXPORT Controller : public itk::Object
13 
{
14 
public:
15
16 
  /** Standard class typedef */
17 
  typedef Controller Self;
18 
  typedef itk::Object                   Superclass;
19 
  typedef itk::SmartPointer<Self>       Pointer;
20 
  typedef itk::SmartPointer<const Self> ConstPointer;
21
22 
  /** Runtime information support. */
23 
  itkTypeMacro(Controller, itk::Object);
24
25 
  /** Method for creation through the object factory. */
26 
  itkNewMacro(Self);
27
28 
  /* Some convenient typedefs */
29 
  using SegmenterType = TSegmenter;
30 
  using ImageType = typename SegmenterType::ImageType;
31 
  using LabelImageType = typename SegmenterType::LabelImageType;
32 
  using SegmentationParameterType = typename SegmenterType::ParamType;
33
34 
  /* Default constructor and destructor. */
35 
  Controller();
36 
  ~Controller();
37
38 
  void Modified();
39
40 
  void RunSegmentation();
41
42 
  void SetSpecificParameters(const SegmentationParameterType& params);
43 
  void SetInputImage(ImageType * inputImage);
44 
  void SetInternalMemoryAvailable(long long unsigned int v); // expecting a value in Mbytes.
45
46 
  /* Accessors */
47 
  void SetTilingModeNone(){m_TilingMode = LSGRM_TILING_NONE; Modified();};
48 
  void SetTilingModeUser(){m_TilingMode = LSGRM_TILING_USER; Modified();};
49 
  void SetTilingModeAuto(){m_TilingMode = LSGRM_TILING_AUTO; Modified();};
50
51 
  typename LabelImageType::Pointer GetLabeledClusteredOutput();
52 
  std::vector<std::string> GetTemporaryFilesList();
53
54 
  itkGetMacro(Margin, unsigned int);
55
56 
  itkGetMacro(TemporaryFilesPrefix, std::string);
57 
  itkSetMacro(TemporaryFilesPrefix, std::string);
58
59 
  itkGetMacro(Threshold, float);
60 
  itkSetMacro(Threshold, float);
61
62 
  itkGetMacro(NumberOfIterations, unsigned int);
63 
  itkSetMacro(NumberOfIterations, unsigned int);
64
65 
  itkGetMacro(NbTilesX, unsigned int);
66 
  itkSetMacro(NbTilesX, unsigned int);
67
68 
  itkGetMacro(NbTilesY, unsigned int);
69 
  itkSetMacro(NbTilesY, unsigned int);
70
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125
itkGetMacro(NumberOfFirstIterations, unsigned int); itkSetMacro(NumberOfFirstIterations, unsigned int); itkGetMacro(TileWidth, unsigned int); itkSetMacro(TileWidth, unsigned int); itkGetMacro(TileHeight, unsigned int); itkSetMacro(TileHeight, unsigned int); private: /** Enum for tiling mode */ enum LSGRMTilingMode { LSGRM_TILING_NONE, // No tiling LSGRM_TILING_USER, // User tiling LSGRM_TILING_AUTO // Automatic tiling }; void GetAutomaticConfiguration(); void ComputeMaximumStabilityMargin(unsigned int width, unsigned int height, unsigned int &iter, unsigned int &margin); void CheckMemorySize(); unsigned int GetNodeMemory(); long unsigned int GetMaximumNumberOfNodesInMemory(); /* Parameters given by the user */ std::string m_TemporaryFilesPrefix; // Path used to store intermediate files during the process. ImageType * m_InputImage; // Input image /* Segmentation parameters */ SegmentationParameterType m_SpecificParameters; // Specific parameters float m_Threshold; // Threshold unsigned int m_NumberOfIterations; // Number of iterations /* Parameters given by the user or computed automatically * depending of the chosen mode */ long long unsigned int m_Memory; // RAM available for the computation. unsigned int m_NbTilesX; // number of tiles in x dimension unsigned int m_NbTilesY; // number of tiles in y dimension unsigned int m_NumberOfFirstIterations; // number of iterations in the first step unsigned int m_TileWidth; // regular tile width (i.e. not left tiles) unsigned int m_TileHeight; // regular tile height (i.e. not bottom tiles) /* read-only variables */ LSGRMTilingMode m_TilingMode; // tiling mode (none/user/auto) unsigned int m_Margin; // stability margin related to m_NumberOfFirstIterations std::vector<ProcessingTile> m_Tiles; // list of tiles typename LabelImageType::Pointer m_LabelImage; // output label image }; } // end of namespace lsgrm #include "lsgrmController.txx" #endif