mitkTubeGraphProperty.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 mitkTubeGraphProperty_h
#define mitkTubeGraphProperty_h
 
#include <MitkTubeGraphExports.h>
 
#include "mitkTubeGraph.h"
#include <mitkBaseProperty.h>
#include <mitkColorProperty.h>
 
#include <itkObject.h>
#include <map>
#include <vector>
 
namespace mitk
{
  class MITKTUBEGRAPH_EXPORT TubeGraphProperty : public BaseProperty
  {
  public:
    mitkClassMacro(TubeGraphProperty, BaseProperty);
    itkNewMacro(TubeGraphProperty);
 
    struct LabelGroup
    {
      struct Label
      {
        std::string labelName;
        bool isVisible;
        Color labelColor;
      };
      std::string labelGroupName;
      std::vector<Label *> labels;
    };
 
    typedef TubeGraph::TubeDescriptorType TubeDescriptorType;
    typedef std::vector<TubeGraphProperty::LabelGroup *> LabelGroupSetType;
    typedef std::pair<TubeDescriptorType, std::string> TubeToLabelGroupType;
 
    struct Annotation
    {
      std::string name;
      std::string description;
      TubeDescriptorType tube;
    };
 
    bool IsTubeVisible(const TubeDescriptorType &tube);
 
    void SetTubeActive(const TubeDescriptorType &tube, const bool &active);
 
    void SetTubesActive(std::vector<TubeDescriptorType> &tubes);
 
    bool IsTubeActive(const TubeDescriptorType &tube);
 
    std::vector<TubeDescriptorType> GetActiveTubes();
 
    Color GetColorOfTube(const TubeDescriptorType &tube);
 
    void SetTubesToLabels(std::map<TubeToLabelGroupType, std::string> tubeToLabelMap);
 
    std::map<TubeToLabelGroupType, std::string> GetTubesToLabels();
 
    void DeactivateAllTubes();
 
    void AddAnnotation(Annotation *annotation);
    Annotation *GetAnnotationByName(std::string annotation);
    std::vector<Annotation *> GetAnnotations();
    void RemoveAnnotation(Annotation *annotation);
 
    void AddLabelGroup(LabelGroup *labelGroup, unsigned int position); // Add LG by name and vec<labelName>??
    void RemoveLabelGroup(LabelGroup *labelGroup);
    LabelGroupSetType GetLabelGroups();
    unsigned int GetNumberOfLabelGroups();
    unsigned int GetIndexOfLabelGroup(LabelGroup *labelGroup);
    LabelGroup *GetLabelGroupByName(std::string labelGroup);
 
    void SetLabelVisibility(LabelGroup::Label *label, bool isVisible);
    void SetLabelColor(LabelGroup::Label *label, Color color);
    void RenameLabel(LabelGroup *labelGroup, LabelGroup::Label *label, std::string newName);
    void SetLabelForActivatedTubes(LabelGroup *labelGroup, LabelGroup::Label *label);
 
    LabelGroup::Label *GetLabelByName(LabelGroup *labelGroup, std::string labelName);
 
    std::string GetValueAsString() const override;
 
    bool ToJSON(nlohmann::json& j) const override;
    bool FromJSON(const nlohmann::json& j) override;
 
  protected:
    TubeGraphProperty();
    TubeGraphProperty(const TubeGraphProperty &other);
    ~TubeGraphProperty() override;
 
  private:
    std::vector<TubeDescriptorType> m_ActiveTubes;
    LabelGroupSetType m_LabelGroups;
    std::map<TubeToLabelGroupType, std::string> m_TubeToLabelsMap;
    std::vector<Annotation *> m_Annotations;
 
    bool TubeDescriptorsCompare(const TubeDescriptorType &tube1, const TubeDescriptorType &tube2);
 
    // purposely not implemented
    TubeGraphProperty &operator=(const TubeGraphProperty &);
 
    bool IsEqual(const BaseProperty &property) const override;
    bool Assign(const BaseProperty &property) override;
    itk::LightObject::Pointer InternalClone() const override;
  };
 
} // namespace mitk
 
#endif