mitkRTStructureSetReaderService.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 MITKRTSTRUCTURESETREADER_H
#define MITKRTSTRUCTURESETREADER_H

#include <mitkAbstractFileReader.h>

#include "MitkDicomRTIOExports.h"
#include <mitkContourModelSet.h>

#include <usModuleContext.h>

namespace mitk
{
  class MITKDICOMRTIO_EXPORT RTStructureSetReaderService : public mitk::AbstractFileReader
  {

    class RoiEntry
    {
    public:
      RoiEntry();
      RoiEntry(const RoiEntry& src);
      virtual ~RoiEntry();
      RoiEntry& operator=(const RoiEntry& src);

      void SetPolyData(ContourModelSet::Pointer roiPolyData);

      unsigned int Number;
      std::string  Name;
      std::string  Description;
      double       DisplayColor[3];
      mitk::ContourModelSet::Pointer ContourModelSet;
    };

  public:
    RTStructureSetReaderService();
    RTStructureSetReaderService(const RTStructureSetReaderService& other);

    ~RTStructureSetReaderService() override;

    using AbstractFileReader::Read;
    std::vector<itk::SmartPointer<BaseData> > Read() override;

  private:
    RTStructureSetReaderService* Clone() const override;

    std::vector<RoiEntry> ROISequenceVector;

    size_t GetNumberOfROIs() const;

    RoiEntry* FindRoiByNumber(unsigned int roiNum);

    us::ServiceRegistration<mitk::IFileReader> m_FileReaderServiceReg;
  };
}

#endif // MITKRTSTRUCTURESETREADER_H