Skip to content
GitLab
Select Git revision
  • disassembled
  • master default
  • resumeopt
  • msbuild_portability
  • streaminggraphtolabelimage
  • change_output_mode
  • GRMBasedModule
  • v2.0
Find file BlameHistoryPermalink
lsgrmController.h 4.32 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
10 
namespace lsgrm
11 
{
12 
template<class TSegmenter>
13 
class ITK_EXPORT Controller : public itk::Object
14 
{
15 
public:
16
17 
  /** Standard class typedef */
18 
  typedef Controller Self;
19 
  typedef itk::Object                   Superclass;
20 
  typedef itk::SmartPointer<Self>       Pointer;
21 
  typedef itk::SmartPointer<const Self> ConstPointer;
22
23 
  /** Runtime information support. */
24 
  itkTypeMacro(Controller, itk::Object);
25
26 
  /** Method for creation through the object factory. */
27 
  itkNewMacro(Self);
28
29 
  /* Some convenient typedefs */
30 
  using SegmenterType = TSegmenter;
31 
  using ImageType = typename SegmenterType::ImageType;
32 
  using LabelImageType = typename SegmenterType::LabelImageType;
33 
  using SegmentationParameterType = typename SegmenterType::ParamType;
34 
  using GraphType = typename SegmenterType::GraphType;
35
36 
  /* Default constructor and destructor. */
37 
  Controller();
38 
  ~Controller();
39
40 
  void Modified();
41
42 
  void RunSegmentation();
43
44 
  void SetSpecificParameters(const SegmentationParameterType& params);
45 
  void SetInputImage(ImageType * inputImage);
46 
  void SetInternalMemoryAvailable(long long unsigned int v); // expecting a value in Mbytes.
47
48 
  /* Accessors */
49 
  void SetTilingModeNone(){m_TilingMode = LSGRM_TILING_NONE; Modified();};
50 
  void SetTilingModeUser(){m_TilingMode = LSGRM_TILING_USER; Modified();};
51 
  void SetTilingModeAuto(){m_TilingMode = LSGRM_TILING_AUTO; Modified();};
52
53 
  std::vector<std::string> GetTemporaryFilesList();
54
55 
  itkGetMacro(Margin, unsigned int);
56
57 
  itkGetMacro(TemporaryFilesPrefix, std::string);
58 
  itkSetMacro(TemporaryFilesPrefix, std::string);
59
60 
  itkGetMacro(Threshold, float);
61 
  itkSetMacro(Threshold, float);
62
63 
  itkGetMacro(NumberOfIterations, unsigned int);
64 
  itkSetMacro(NumberOfIterations, unsigned int);
65
66 
  itkGetMacro(NbTilesX, unsigned int);
67 
  itkSetMacro(NbTilesX, unsigned int);
68
69 
  itkGetMacro(NbTilesY, unsigned int);
70 
  itkSetMacro(NbTilesY, unsigned int);
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
itkGetMacro(NumberOfFirstIterations, unsigned int); itkSetMacro(NumberOfFirstIterations, unsigned int); itkGetMacro(TileWidth, unsigned int); itkSetMacro(TileWidth, unsigned int); itkGetMacro(TileHeight, unsigned int); itkSetMacro(TileHeight, unsigned int); itkGetMacro(OutputGraph, GraphType); 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(); std::size_t 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 GraphType m_OutputGraph; // Output graph }; } // end of namespace lsgrm #include "lsgrmController.txx" #endif