mitkDICOMSegmentationIO.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 mitkDICOMSegmentationIO_h
#define mitkDICOMSegmentationIO_h
 
#include <mitkAbstractFileIO.h>
#include <mitkLabelSetImage.h>
 
#include <mitkDICOMTagsOfInterestAddHelper.h>
 
#include <dcmqi/JSONSegmentationMetaInformationHandler.h>
 
#include <memory>
 
namespace mitk
{
  class DICOMSegmentationIO : public mitk::AbstractFileIO
  {
  public:
    typedef mitk::MultiLabelSegmentation InputType;
    typedef itk::Image<unsigned short, 3> itkInputImageType;
    typedef itk::Image<short, 3> itkInternalImageType;
 
    DICOMSegmentationIO();
 
    // -------------- AbstractFileReader -------------
 
    using AbstractFileReader::Read;
 
    ConfidenceLevel GetReaderConfidenceLevel() const override;
 
    // -------------- AbstractFileWriter -------------
 
    void Write() override;
    ConfidenceLevel GetWriterConfidenceLevel() const override;
 
  protected:
    std::vector<itk::SmartPointer<BaseData>> DoRead() override;
 
  private:
    DICOMSegmentationIO *IOClone() const override;
 
    // -------------- DICOMSegmentationIO specific functions -------------
    const std::string CreateMetaDataJsonFile(int layer);
    void SetLabelProperties(Label *label, dcmqi::SegmentAttributes *segmentAttribute);
  };
} // end of namespace mitk
 
#endif