QmitkToolRoiDataSelectionBox.cpp

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.

============================================================================*/
#include "QmitkToolRoiDataSelectionBox.h"
#include "mitkToolManagerProvider.h"
#include <QBoxLayout>
#include <QLabel>
#include <mitkProperties.h>

QmitkToolRoiDataSelectionBox::QmitkToolRoiDataSelectionBox(QWidget *parent, mitk::DataStorage *)
  : QWidget(parent), m_SelfCall(false), m_lastSelection(mitk::DataNode::New()), m_lastSelectedName(tr("none"))
{
  QBoxLayout *mainLayout = new QVBoxLayout(this);
  m_segmentationComboBox = new QComboBox(this);
  QLabel *label = new QLabel("region of interest:", this);
  m_boundingObjectWidget = new QmitkBoundingObjectWidget();

  mainLayout->addWidget(label);
  mainLayout->addWidget(m_segmentationComboBox);
  mainLayout->addWidget(m_boundingObjectWidget);

  // connect signals
  connect(
    m_segmentationComboBox, SIGNAL(activated(const QString &)), this, SLOT(OnRoiDataSelectionChanged(const QString &)));
  connect(m_boundingObjectWidget, SIGNAL(BoundingObjectsChanged()), this, SLOT(OnRoiDataSelectionChanged()));

  // create ToolManager
  m_ToolManager = mitk::ToolManagerProvider::GetInstance()->GetToolManager();

  // setup message delegates
  m_ToolManager->RoiDataChanged += mitk::MessageDelegate<QmitkToolRoiDataSelectionBox>(
    this, &QmitkToolRoiDataSelectionBox::OnToolManagerRoiDataModified);

  mainLayout->deleteLater();
  label->deleteLater();
}

QmitkToolRoiDataSelectionBox::~QmitkToolRoiDataSelectionBox()
{
  delete m_segmentationComboBox;
  delete m_boundingObjectWidget;

  m_ToolManager->GetDataStorage()->RemoveNodeEvent.RemoveListener(
    mitk::MessageDelegate1<QmitkToolRoiDataSelectionBox, const mitk::DataNode *>(
      this, &QmitkToolRoiDataSelectionBox::DataStorageChanged));
}

void QmitkToolRoiDataSelectionBox::SetDataStorage(mitk::DataStorage &storage)
{
  m_ToolManager->SetDataStorage(storage);
  m_boundingObjectWidget->SetDataStorage(&storage);
  UpdateComboBoxData();

  storage.RemoveNodeEvent.AddListener(mitk::MessageDelegate1<QmitkToolRoiDataSelectionBox, const mitk::DataNode *>(
    this, &QmitkToolRoiDataSelectionBox::DataStorageChanged));
}

mitk::DataStorage *QmitkToolRoiDataSelectionBox::GetDataStorage()
{
  return m_ToolManager->GetDataStorage();
}

void QmitkToolRoiDataSelectionBox::SetToolManager(mitk::ToolManager &manager)
{
  // remove old messagedelegates
  m_ToolManager->RoiDataChanged -= mitk::MessageDelegate<QmitkToolRoiDataSelectionBox>(
    this, &QmitkToolRoiDataSelectionBox::OnToolManagerRoiDataModified);
  // set new toolmanager
  m_ToolManager = &manager;
  // add new message delegates
  m_ToolManager->RoiDataChanged += mitk::MessageDelegate<QmitkToolRoiDataSelectionBox>(
    this, &QmitkToolRoiDataSelectionBox::OnToolManagerRoiDataModified);
}

mitk::ToolManager *QmitkToolRoiDataSelectionBox::GetToolManager()
{
  return m_ToolManager;
}

void QmitkToolRoiDataSelectionBox::OnToolManagerRoiDataModified()
{
  if (m_SelfCall)
    return;
  UpdateComboBoxData();
}

void QmitkToolRoiDataSelectionBox::DataStorageChanged(const mitk::DataNode *)
{
  if (m_SelfCall)
    return;

  if (this->GetDataStorage()->GetAll()->size() == 1)
  {
    m_boundingObjectWidget->RemoveAllItems();
  }
}

void QmitkToolRoiDataSelectionBox::OnRoiDataSelectionChanged()
{
  this->OnRoiDataSelectionChanged(tr("bounding objects"));
}

void QmitkToolRoiDataSelectionBox::OnRoiDataSelectionChanged(const QString &name)
{
  if (name.compare(tr("")) == 0)
    return;

  m_lastSelectedName = name;
  m_boundingObjectWidget->setEnabled(false);
  mitk::DataNode::Pointer selection = nullptr;

  if (name.compare(tr("none")) == 0)
    m_segmentationComboBox->setCurrentIndex(0);
  else if (name.compare(tr("bounding objects")) == 0)
  {
    m_boundingObjectWidget->setEnabled(true);
    selection = m_boundingObjectWidget->GetAllBoundingObjects();
  }
  else
  {
    selection = m_ToolManager->GetDataStorage()->GetNamedNode(name.toLocal8Bit().data());

    if (m_lastSelection.IsNotNull())
      m_lastSelection->SetProperty("outline binary", mitk::BoolProperty::New(false));
  }

  if (selection == m_lastSelection)
    return;

  m_lastSelection = selection;

  if (m_lastSelection.IsNotNull())
  {
    m_lastSelection->SetProperty("outline binary", mitk::BoolProperty::New(true));
    m_lastSelection->SetProperty("outline width", mitk::FloatProperty::New(2.0));
  }

  m_SelfCall = true;
  m_ToolManager->SetRoiData(selection);
  m_SelfCall = false;
}

void QmitkToolRoiDataSelectionBox::UpdateComboBoxData()
{
  m_segmentationComboBox->clear();
  m_segmentationComboBox->addItem(tr("none"));
  m_segmentationComboBox->insertSeparator(1);

  // predicates for combobox
  mitk::NodePredicateProperty::Pointer isBinary =
    mitk::NodePredicateProperty::New("binary", mitk::BoolProperty::New(true));
  mitk::NodePredicateDataType::Pointer isImage = mitk::NodePredicateDataType::New("Image");
  mitk::NodePredicateProperty::Pointer isHelperObject =
    mitk::NodePredicateProperty::New("helper object", mitk::BoolProperty::New(true));
  mitk::NodePredicateNot::Pointer isNotHelperObject = mitk::NodePredicateNot::New(isHelperObject);
  mitk::NodePredicateAnd::Pointer segmentationPredicate =
    mitk::NodePredicateAnd::New(isImage, isBinary, isNotHelperObject);

  mitk::DataStorage::SetOfObjects::ConstPointer allSegmentations =
    m_ToolManager->GetDataStorage()->GetSubset(segmentationPredicate);
  QStringList names;

  for (mitk::DataStorage::SetOfObjects::const_iterator it = allSegmentations->begin(); it != allSegmentations->end();
       ++it)
  {
    mitk::DataNode::Pointer node = *it;

    QString name = QString::fromLocal8Bit(node->GetName().c_str());
    names.append(name);
  }
  if (names.length() > 0)
  {
    m_segmentationComboBox->addItems(names);
    m_segmentationComboBox->insertSeparator(names.length() + 2);
  }

  m_segmentationComboBox->addItem(tr("bounding objects"));

  int id = m_segmentationComboBox->findText(m_lastSelectedName);
  if (id < 0)
    this->OnRoiDataSelectionChanged(tr("none"));
  else
    m_segmentationComboBox->setCurrentIndex(id);
}

void QmitkToolRoiDataSelectionBox::setEnabled(bool flag)
{
  if (!flag)
    this->OnRoiDataSelectionChanged(tr("none"));
  m_segmentationComboBox->setEnabled(flag);
}