cxFrameTreeWidget.cpp

Go to the documentation of this file.

/*=========================================================================
This file is part of CustusX, an Image Guided Therapy Application.
                 
Copyright (c) SINTEF Department of Medical Technology.
All rights reserved.
                 
CustusX is released under a BSD 3-Clause license.
                 
See Lisence.txt (https://github.com/SINTEFMedtek/CustusX/blob/master/License.txt) for details.
=========================================================================*/

#include "cxFrameTreeWidget.h"

#include <QVBoxLayout>
#include <QTreeWidget>
#include <QTreeWidgetItem>
#include "cxFrameForest.h"
#include "cxData.h"
#include "cxPatientModelService.h"

namespace cx
{

FrameTreeWidget::FrameTreeWidget(PatientModelServicePtr patientService, QWidget* parent) :
  BaseWidget(parent, "frame_tree_widget", "Frame Tree"),
  mPatientService(patientService)
{
  QVBoxLayout* layout = new QVBoxLayout(this);

  //layout->setMargin(0);
  this->setToolTip("A tree displaying relations between coordinate spaces");
  mTreeWidget = new QTreeWidget(this);
  layout->addWidget(mTreeWidget);
  mTreeWidget->setHeaderLabels(QStringList() << "Frame");

  // TODO this must also listen to all changed() in all data
  connect(mPatientService.get(), SIGNAL(dataAddedOrRemoved()), this, SLOT(dataLoadedSlot()));
}

void FrameTreeWidget::dataLoadedSlot()
{
  for (std::map<QString, DataPtr>::iterator iter=mConnectedData.begin(); iter!=mConnectedData.end(); ++iter)
  {
        disconnect(iter->second.get(), SIGNAL(transformChanged()), this, SLOT(setModified()));
  }

  mConnectedData = mPatientService->getDatas();

  for (std::map<QString, DataPtr>::iterator iter=mConnectedData.begin(); iter!=mConnectedData.end(); ++iter)
  {
        connect(iter->second.get(), SIGNAL(transformChanged()), this, SLOT(setModified()));
  }

  this->setModified();
}

void FrameTreeWidget::prePaintEvent()
{
        this->rebuild();
}

void FrameTreeWidget::rebuild()
{
  mTreeWidget->clear();

  FrameForest forest(mPatientService->getDatas());
  QDomElement root = forest.getDocument().documentElement();

  this->fill(mTreeWidget->invisibleRootItem(), root);

  mTreeWidget->expandToDepth(10);
  mTreeWidget->resizeColumnToContents(0);
}

void FrameTreeWidget::fill(QTreeWidgetItem* parent, QDomNode node)
{
  for (QDomNode child = node.firstChild(); !child.isNull(); child = child.nextSibling())
  {
    QString frameName = child.toElement().tagName();

    // if frame refers to a data, use its name instead.
        DataPtr data = mPatientService->getData(frameName);
    if (data)
      frameName = data->getName();

    QTreeWidgetItem* item = new QTreeWidgetItem(parent, QStringList() << frameName);
    this->fill(item, child);
  }
}

}