mitkDataCollection.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef mitkDataCollection_h
#define mitkDataCollection_h
 
#ifdef _MSC_VER
#  pragma warning (disable : 4996)
#endif
 
//#include <itkLightObject.h>
//#include <itkDataObject.h>
//#include "itkObjectFactory.h"
//#include "mitkCommon.h"
#include <mitkImage.h>
#include <mitkDataNode.h>
 
#include<mitkBaseData.h>
 
#include <MitkDataCollectionExports.h>
 
namespace mitk
{
  class MITKDATACOLLECTION_EXPORT DataCollection  : public BaseData
  {
  public:
 
    mitkClassMacro(DataCollection, BaseData);
 
    itkFactorylessNewMacro(Self);
 
    itkCloneMacro(Self);
 
      // Needed methods from Basedata
      void UpdateOutputInformation() override;
    void SetRequestedRegionToLargestPossibleRegion() override;
    bool RequestedRegionIsOutsideOfTheBufferedRegion() override;
    bool VerifyRequestedRegion() override;
    void SetRequestedRegion(const itk::DataObject *) override;
 
    void Init(std::string name);
 
    size_t AddData(DataObject::Pointer data, std::string name, std::string filePath = "");
 
    void SetName(std::string name);
 
    std::string GetName() const;
 
    std::string GetDataFilePath(size_t index) const;
 
    size_t NameToIndex(std::string name);
 
    std::string IndexToName(size_t index) const;
 
    bool HasElement(std::string name);
 
    bool HasElement(size_t index);
 
    size_t Size() const;
 
    void SetData(itk::DataObject::Pointer data, size_t index);
 
    void SetData(itk::DataObject::Pointer data, std::string name);
 
    itk::DataObject::Pointer GetData(size_t index);
 
    itk::DataObject::Pointer GetData(std::string name);
 
    mitk::Image::Pointer GetMitkImage(size_t index);
 
    mitk::Image::Pointer GetMitkImage(std::string name);
 
    template <class ImageType>
    ImageType GetItkImage(size_t index, ImageType* itkImage);
 
    template <class ImageType>
    ImageType GetItkImage(std::string name, ImageType* itkImage);
 
    itk::DataObject::Pointer& operator[](size_t index);
 
    itk::DataObject::Pointer& operator[](std::string &name);
 
    void SetNameForIndex(size_t index, std::string &name);
 
    void SetXMLFile(std::string absoluteXMlFile);
 
    std::string GetXMLFile();
 
    void SetParent(mitk::DataCollection* parent);
 
    mitk::DataCollection* GetParent();
 
    bool RemoveIndex(size_t index);
 
    bool RemoveElement(std::string& name);
 
    void Clear() override;
 
    mitk::DataNode::Pointer GetDataNode(size_t index);
 
    mitk::DataNode::Pointer GetDataNode(std::string name);
 
    mitk::Image::Pointer GetProbabilityMap(std::vector<std::string> probabilityNamesVector);
 
  protected:
 
    DataCollection();
    ~DataCollection() override;
 
  private:
    // DATA
    std::string m_Name;
    std::vector<itk::DataObject::Pointer> m_DataVector;
    std::vector<std::string> m_NameVector;
    std::vector<std::string> m_FilePathVector;
    std::map<std::string, size_t> m_DataNames;
 
    mitk::DataCollection * m_Parent;
 
    std::string m_XMLFile; // is only filled for the hightest layer when loading a data collection
  };
} // end namespace
#endif