cxElastixWidget.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 "cxElastixWidget.h"

#include <QPushButton>
#include <QLabel>
#include <QSpinBox>

#include "cxTypeConversions.h"

#include "cxTimedAlgorithm.h"
#include "cxLabeledComboBoxWidget.h"
#include "cxRegistrationProperties.h"
#include "cxFileSelectWidget.h"
#include "cxElastixSyntaxHighlighter.h"
#include "cxCheckBoxWidget.h"
#include "cxSettings.h"
#include "cxTimedAlgorithmProgressBar.h"
#include "cxElastixExecuter.h"
#include "cxStringProperty.h"
#include "cxFilePathProperty.h"
#include "cxDataLocations.h"
#include "cxHelperWidgets.h"

namespace cx
{

ElastixWidget::ElastixWidget(RegServicesPtr services, QWidget* parent) :
        RegistrationBaseWidget(services, parent, "org_custusx_registration_method_commandline_elastix_widget", "ElastiX Registration"),
        mRegisterButton(NULL),
        mParameterFileWidget0(NULL),
        mFilePreviewWidget(NULL),
        mTimedAlgorithmProgressBar(NULL),
        mOptionsWidget(NULL)
{
        this->setModified();
}

void ElastixWidget::prePaintEvent()
{
        if (!mElastixManager)
        {
                this->createUI();
        }
}

void ElastixWidget::createUI()
{
        mElastixManager.reset(new ElastixManager(mServices));
        connect(mElastixManager.get(), SIGNAL(elastixChanged()), this, SLOT(elastixChangedSlot()));

        mRegisterButton = new QPushButton("Register");
        connect(mRegisterButton, SIGNAL(clicked()), this, SLOT(registerSlot()));

        QVBoxLayout* topLayout = new QVBoxLayout(this);
        topLayout->setMargin(0);

        mOptionsWidget = this->createOptionsWidget();
        mOptionsWidget->setVisible(settings()->value("registration/elastixShowDetails").toBool());

        mTimedAlgorithmProgressBar = new cx::TimedAlgorithmProgressBar;
        mTimedAlgorithmProgressBar->attach(mElastixManager->getExecuter());

        QGridLayout* entryLayout = new QGridLayout;
        entryLayout->setColumnStretch(1, 1);

        mFixedImage.reset(new StringPropertyRegistrationFixedImage(mServices->registration(), mServices->patient()));
        new LabeledComboBoxWidget(this, mFixedImage, entryLayout, 0);
        mMovingImage.reset(new StringPropertyRegistrationMovingImage(mServices->registration(), mServices->patient()));
        new LabeledComboBoxWidget(this, mMovingImage, entryLayout, 1);

        new LabeledComboBoxWidget(this, mElastixManager->getParameters()->getCurrentPreset(), entryLayout, 2);

        QHBoxLayout* buttonsLayout = new QHBoxLayout;
        buttonsLayout->addWidget(mRegisterButton);

        this->createAction(this,
                  QIcon(":/icons/open_icon_library/system-run-5.png"),
                  "Details", "Show Elastix Settings Details",
                  SLOT(toggleDetailsSlot()),
                  buttonsLayout);

        topLayout->addLayout(entryLayout);
        topLayout->addLayout(buttonsLayout);
        topLayout->addWidget(mOptionsWidget, 1);
        topLayout->addStretch();
        topLayout->addWidget(mTimedAlgorithmProgressBar);

        this->elastixChangedSlot();
}

void ElastixWidget::toggleDetailsSlot()
{
  mOptionsWidget->setVisible(!mOptionsWidget->isVisible());
  settings()->setValue("registration/elastixShowDetails", mOptionsWidget->isVisible());
}

QWidget* ElastixWidget::createOptionsWidget()
{
        QWidget* retval = new QWidget(this);
        QGridLayout* layout = new QGridLayout(retval);
        layout->setMargin(0);

        int line = 0;

        layout->addWidget(this->createHorizontalLine(), line, 0, 1, 3);
        ++line;

        layout->addWidget(new QLabel("Parameter File", this), line, 0);
        mParameterFileWidget0 = new FileSelectWidget(this);
        connect(mParameterFileWidget0, SIGNAL(fileSelected(QString)), this, SLOT(userParameterFileSelected(QString)));
        layout->addWidget(mParameterFileWidget0, line, 1, 1, 2);
        ++line;

        QWidget* executableWidget = sscCreateDataWidget(this, mElastixManager->getParameters()->getActiveExecutable());
        layout->addWidget(executableWidget, line, 0, 1, 3);
        ++line;

        QHBoxLayout* buttonsLayout = new QHBoxLayout;
        layout->addLayout(buttonsLayout, line, 0, 1, 3);

        buttonsLayout->addWidget(new CheckBoxWidget(this, mElastixManager->getDisplayProcessMessages()));
        buttonsLayout->addWidget(new CheckBoxWidget(this, mElastixManager->getDisableRendering()));

        this->createAction(this,
                        QIcon(":/icons/preset_remove.png"),
                                        "Delete the current preset", "",
                        SLOT(deletePresetSlot()),
                        buttonsLayout);

        this->createAction(this,
                            QIcon(":/icons/preset_save.png"),
                                        "Add the current setting as a preset", "",
                        SLOT(savePresetSlot()),
                        buttonsLayout);

        ++line;

        mFilePreviewWidget = new FilePreviewWidget(this);
        mFilePreviewWidget->setSyntaxHighLighter<ElastixSyntaxHighlighter>();
        mFilePreviewWidget->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding);
        layout->addWidget(mFilePreviewWidget, line, 0, 1, 3);
        ++line;

        return retval;
}

ElastixWidget::~ElastixWidget()
{}


void ElastixWidget::savePresetSlot()
{
        ElastixParametersPtr par = mElastixManager->getParameters();

        QString newName = par->getPresetNameSuggesion();

    bool ok;
    QString text = QInputDialog::getText(this, "Save Preset",
                                         "Custom Preset Name", QLineEdit::Normal,
                                         newName, &ok);
    if (!ok || text.isEmpty())
      return;

        par->saveCurrentPreset(text);
}

void ElastixWidget::deletePresetSlot()
{
        mElastixManager->getParameters()->removeCurrentPreset();
}

void ElastixWidget::userParameterFileSelected(QString filename)
{
        mElastixManager->getParameters()->getActiveParameterFile0()->setValue(filename);
}

void ElastixWidget::recurseParameterFolders(QString root, QStringList* retval)
{
        QDir folder(root);
        folder.setFilter(QDir::AllDirs|QDir::NoDotAndDotDot);
        QFileInfoList info = folder.entryInfoList();
        for (int i=0; i<info.size(); ++i)
        {
                QString current = info[i].absoluteFilePath();
                if (current.endsWith("/par"))
                {
                        retval->append(current);
                }

                this->recurseParameterFolders(current, retval);
        }
}

void ElastixWidget::elastixChangedSlot()
{
        ElastixParametersPtr par = mElastixManager->getParameters();
        EmbeddedFilepath par0 = par->getActiveParameterFile0()->getEmbeddedPath();
        QStringList folders = par0.getRootPaths();
        QStringList parfolders;
        for (int i=0; i<folders.size(); ++i)
                this->recurseParameterFolders(folders[i], &parfolders);

        mParameterFileWidget0->setPaths(parfolders);
        QStringList nameFilters;
        nameFilters << "*";
        mParameterFileWidget0->setNameFilter(nameFilters);
        mParameterFileWidget0->setFilename(par0.getAbsoluteFilepath());


        mFilePreviewWidget->previewFileSlot(par0.getAbsoluteFilepath());
}

void ElastixWidget::registerSlot()
{
        mElastixManager->execute();
}



} /* namespace cx */