Skip to content
Snippets Groups Projects
Commit 013fa54a authored by Guillaume Pasero's avatar Guillaume Pasero
Browse files

ENH: allow to access the splitter

parent 3fde3d39
No related branches found
No related tags found
1 merge request!33Multi writer
......@@ -67,6 +67,7 @@ public:
typedef typename ImageType::InternalPixelType PixelType;
typedef otb::PipelineMemoryPrintCalculator::MemoryPrintType MemoryPrintType;
typedef itk::ImageRegionSplitterBase AbstractSplitterType;
/** Type macro */
itkTypeMacro(StreamingManager, itk::LightObject);
......@@ -74,6 +75,8 @@ public:
/** Dimension of input image. */
itkStaticConstMacro(ImageDimension, unsigned int, ImageType::ImageDimension);
const AbstractSplitterType * GetSplitter() const;
/** Actually computes the stream divisions, according to the specified streaming mode,
* eventually using the input parameter to estimate memory consumption */
virtual void PrepareStreaming(itk::DataObject * input, const RegionType &region) = 0;
......@@ -103,7 +106,6 @@ protected:
RegionType m_Region;
/** The splitter used to compute the different strips */
typedef itk::ImageRegionSplitterBase AbstractSplitterType;
typedef typename AbstractSplitterType::Pointer AbstractSplitterPointerType;
AbstractSplitterPointerType m_Splitter;
......
......@@ -39,6 +39,13 @@ StreamingManager<TImage>::~StreamingManager()
{
}
template <class TImage>
const typename StreamingManager<TImage>::AbstractSplitterType *
StreamingManager<TImage>::GetSplitter() const
{
return m_Splitter;
}
template <class TImage>
typename StreamingManager<TImage>::MemoryPrintType
StreamingManager<TImage>::GetActualAvailableRAMInBytes(MemoryPrintType availableRAMInMB)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment