QmitkPatientTableInspector.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.

============================================================================*/

// semantic relations UI module
#include "QmitkPatientTableInspector.h"
#include "QmitkPatientTableHeaderView.h"

// mitk qt widgets module
#include "QmitkCustomVariants.h"
#include "QmitkEnums.h"

// qt
#include <QKeyEvent>
#include <QSignalMapper>

QmitkPatientTableInspector::QmitkPatientTableInspector(QWidget* parent/* =nullptr*/)
  : QmitkAbstractSemanticRelationsStorageInspector(parent)
{
  m_Controls.setupUi(this);

  QmitkPatientTableHeaderView* patientTableHeaderView = new QmitkPatientTableHeaderView(m_Controls.tableView);
  m_Controls.tableView->setHorizontalHeader(patientTableHeaderView);
  m_Controls.tableView->horizontalHeader()->setHighlightSections(false);
  m_Controls.tableView->horizontalHeader()->setSectionResizeMode(QHeaderView::Fixed);
  m_Controls.tableView->verticalHeader()->setHighlightSections(false);
  m_Controls.tableView->verticalHeader()->setSectionResizeMode(QHeaderView::Fixed);
  m_Controls.tableView->setSelectionMode(QAbstractItemView::SingleSelection);
  m_Controls.tableView->setSelectionBehavior(QAbstractItemView::SelectItems);
  m_Controls.tableView->setContextMenuPolicy(Qt::CustomContextMenu);

  m_StorageModel = new QmitkPatientTableModel(m_Controls.tableView);
  m_Controls.tableView->setModel(m_StorageModel);
  m_ItemDelegate = new QmitkTableItemThumbnailDelegate(m_Controls.tableView);
  //m_Controls.tableView->setItemDelegate(m_ItemDelegate);

  SetUpConnections();
}

QAbstractItemView* QmitkPatientTableInspector::GetView()
{
  return m_Controls.tableView;
}

const QAbstractItemView* QmitkPatientTableInspector::GetView() const
{
  return m_Controls.tableView;
}

void QmitkPatientTableInspector::SetSelectionMode(SelectionMode mode)
{
  m_Controls.tableView->setSelectionMode(mode);
}

QmitkPatientTableInspector::SelectionMode QmitkPatientTableInspector::GetSelectionMode() const
{
  return m_Controls.tableView->selectionMode();
}

void QmitkPatientTableInspector::SetCaseID(const mitk::SemanticTypes::CaseID& caseID)
{
  m_StorageModel->SetCaseID(caseID);
}

void QmitkPatientTableInspector::SetLesion(const mitk::SemanticTypes::Lesion& lesion)
{
  m_StorageModel->SetLesion(lesion);
}

QItemSelectionModel* QmitkPatientTableInspector::GetSelectionModel()
{
  return m_Controls.tableView->selectionModel();
}

void QmitkPatientTableInspector::Initialize()
{
  if (m_DataStorage.IsExpired())
  {
    return;
  }

  auto dataStorage = m_DataStorage.Lock();

  m_StorageModel->SetDataStorage(dataStorage);
  m_StorageModel->SetNodePredicate(m_NodePredicate);

  m_Connector->SetView(m_Controls.tableView);
}

void QmitkPatientTableInspector::OnModelUpdated()
{
  m_Controls.tableView->resizeRowsToContents();
  m_Controls.tableView->resizeColumnsToContents();
}

void QmitkPatientTableInspector::OnNodeButtonClicked(const QString& nodeType)
{
  m_StorageModel->SetNodeType(nodeType.toStdString());
}

void QmitkPatientTableInspector::OnDataNodeSelectionChanged(const QList<mitk::DataNode::Pointer>& dataNodeSelection)
{
  if (m_StorageModel->GetLesion().UID.empty())
  {
    return;
  }

  // if lesion is set, reset to empty lesion to hide the "lesion presence background highlighting" in the model
  m_StorageModel->SetLesion(mitk::SemanticTypes::Lesion());
  // need to explicitly set the data node selection
  SetCurrentSelection(dataNodeSelection);
}

void QmitkPatientTableInspector::OnItemDoubleClicked(const QModelIndex& itemIndex)
{
  if (itemIndex.isValid())
  {
    QVariant qvariantDataNode = m_StorageModel->data(itemIndex, QmitkDataNodeRawPointerRole);
    if (qvariantDataNode.canConvert<mitk::DataNode*>())
    {
      mitk::DataNode* dataNode = qvariantDataNode.value<mitk::DataNode*>();
      emit DataNodeDoubleClicked(dataNode);
    }
  }
}

void QmitkPatientTableInspector::SetUpConnections()
{
  connect(m_StorageModel, &QmitkPatientTableModel::ModelUpdated, this, &QmitkPatientTableInspector::OnModelUpdated);
  connect(m_Controls.tableView, &QTableView::customContextMenuRequested, this, &QmitkPatientTableInspector::OnContextMenuRequested);

  QSignalMapper* nodeButtonSignalMapper = new QSignalMapper(this);
  nodeButtonSignalMapper->setMapping(m_Controls.imageNodeButton, QString("Image"));
  nodeButtonSignalMapper->setMapping(m_Controls.segmentationNodeButton, QString("Segmentation"));
  connect(nodeButtonSignalMapper, static_cast<void (QSignalMapper::*)(const QString&)>(&QSignalMapper::mapped), this, &QmitkPatientTableInspector::OnNodeButtonClicked);
  connect(m_Controls.imageNodeButton, &QRadioButton::clicked, nodeButtonSignalMapper, static_cast<void(QSignalMapper::*)()>(&QSignalMapper::map));
  connect(m_Controls.segmentationNodeButton, &QRadioButton::clicked, nodeButtonSignalMapper, static_cast<void(QSignalMapper::*)()>(&QSignalMapper::map));
  m_Controls.imageNodeButton->setChecked(true);

  connect(this, &QmitkPatientTableInspector::CurrentSelectionChanged, this, &QmitkPatientTableInspector::OnDataNodeSelectionChanged);
  connect(m_Controls.tableView, &QTableView::doubleClicked, this, &QmitkPatientTableInspector::OnItemDoubleClicked);
}

void QmitkPatientTableInspector::keyPressEvent(QKeyEvent* e)
{
  mitk::DataNode* dataNode = nullptr;
  QModelIndex selectedIndex = m_Controls.tableView->currentIndex();
  if (selectedIndex.isValid())
  {
    QVariant qvariantDataNode = m_StorageModel->data(selectedIndex, QmitkDataNodeRawPointerRole);
    if (qvariantDataNode.canConvert<mitk::DataNode*>())
    {
      dataNode = qvariantDataNode.value<mitk::DataNode*>();
    }
  }

  if (nullptr == dataNode)
  {
    return;
  }

  int key = e->key();
  switch (key)
  {
  case Qt::Key_Delete:
    emit OnNodeRemoved(dataNode);
    break;
  default:
    break;
  }
}