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

#include <iostream>
#include <vtkColorTransferFunction.h>
#include <QStringList>
#include <QDomElement>
#include <QDomDocument>
#include <QColor>
#include "cxTypeConversions.h"
#include "cxXmlOptionItem.h"
#include "cxImageTF3D.h"
#include "cxImageLUT2D.h"
#include "cxImageTFData.h"
#include "cxEnumConversion.h"

namespace cx
{

TransferFunctions3DPresets::TransferFunctions3DPresets(XmlOptionFile presetFile, XmlOptionFile customFile) :
                Presets(presetFile, customFile)
{
}

void TransferFunctions3DPresets::save(QString name, ImagePtr image, bool _2D, bool _3D)
{
        if (_2D)
                this->save2D(name, image);
        if (_3D)
                this->save3D(name, image);
}

void TransferFunctions3DPresets::save2D(QString name, ImagePtr image)
{
        XmlOptionFile file = this->getCustomFile();
        file = file.descend("Preset", "name", name);

        QDomNode tf2DNode = file.getElement("lookuptable2D");
        while (tf2DNode.hasChildNodes())
                tf2DNode.removeChild(tf2DNode.firstChild());

        ImageLUT2DPtr LUT2D = image->getLookupTable2D();

        // For unsigned CT: Modify transfer function values temporarily prior to save
        if ((0 <= image->getMin()) && (imCT == image->getModality()))
        {
                LUT2D->unsignedCT(false);
        }

        LUT2D->addXml(file.getElement("lookuptable2D"));

        // Revert the transfer function values back again
        if ((0 <= image->getMin()) && (imCT == image->getModality()))
        {
                LUT2D->unsignedCT(true);
        }

        file.getElement().setAttribute("modality", enum2string(image->getModality()));
        file.save();

//      emit changed();
}

void TransferFunctions3DPresets::save3D(QString name, ImagePtr image)
{
        //create the node to be saved
        XmlOptionFile file = this->getCustomFile();
        file = file.descend("Preset", "name", name);

        QDomNode tf3DNode = file.getElement("transferfunctions");
        while (tf3DNode.hasChildNodes())
                tf3DNode.removeChild(tf3DNode.firstChild());

        ImageTF3DPtr transferFunctions = image->getTransferFunctions3D();

        // For unsigned CT: Modify transfer function values temporarily prior to save
        if ((0 <= image->getMin()) && (imCT == image->getModality()))
        {
                transferFunctions->unsignedCT(false);
        }

        transferFunctions->addXml(file.getElement("transferfunctions"));
        image->getShading().addXml(file.getElement("shading"));

        // Revert the transfer function values back again
        if ((0 <= image->getMin()) && (imCT == image->getModality()))
        {
                transferFunctions->unsignedCT(true);
        }

        file.getElement().setAttribute("modality", enum2string(image->getModality()));
        file.save();
}

void TransferFunctions3DPresets::load(QString name, ImagePtr image, bool _2D, bool _3D)
{
        if (_2D)
                this->load2D(name, image);
        if (_3D)
                this->load3D(name, image);
}

void TransferFunctions3DPresets::load2D(QString name, ImagePtr image)
{
        if(!image)
                return;

        //Make sure transfer functions are reset in case something is missing from the preset
        image->resetTransferFunctions(true, false);

        ImageLUT2DPtr LUT2D = image->getLookupTable2D();
        XmlOptionFile node = this->getPresetNode(name);

        LUT2D->parseXml(node.getElement().namedItem("lookuptable2D"));

        // Transfer functions for CT data are signed, so these have to be converted if they are to be used for unsigned CT
        if ((0 <= image->getMin()) && (imCT == image->getModality()) && (name != "Transfer function preset...") )
        {
                LUT2D->unsignedCT(true);
        }
}

void TransferFunctions3DPresets::load3D(QString name, ImagePtr image)
{
        //Make sure transfer functions are reset in case something is missing from the preset
        image->resetTransferFunctions(false, true);

        ImageTF3DPtr transferFunctions = image->getTransferFunctions3D();
        XmlOptionFile node = this->getPresetNode(name);

        transferFunctions->parseXml(node.getElement().namedItem("transferfunctions"));

        Image::ShadingStruct shading = image->getShading();
        shading.parseXml(node.getElement().namedItem("shading"));
        image->setShading(shading);

        // Transfer functions for CT data are signed, so these have to be converted if they are to be used for unsigned CT
        if ((0 <= image->getMin()) && (imCT == image->getModality()) && (name != "Transfer function preset...") )
        {
                transferFunctions->unsignedCT(true);
        }
}

QStringList TransferFunctions3DPresets::generatePresetList(IMAGE_MODALITY modality)
{
        QStringList presetList;

        QDomNodeList presetNodeList = mPresetFile.getElement().elementsByTagName("Preset");
        for (int i = 0; i < presetNodeList.count(); ++i)
        {
                QString presetName = presetNodeList.item(i).toElement().attribute("name");
                if (presetName == "Default")
                        continue;
                else
                {
                        QString sourceModality = presetNodeList.item(i).toElement().attribute("modality");
                        if ( (modality == string2enum<IMAGE_MODALITY>(sourceModality)) || (imUNKNOWN == modality) || imCOUNT == modality )
                                presetList << presetName;
                }
        }

        XmlOptionFile customFile = this->getCustomFile();
        presetNodeList = customFile.getElement().elementsByTagName("Preset");
        for (int i = 0; i < presetNodeList.count(); ++i)
        {
                QString presetName = presetNodeList.item(i).toElement().attribute("name");
                QString presetModality = presetNodeList.item(i).toElement().attribute("modality");
                if ( (string2enum<IMAGE_MODALITY>(presetModality) == modality) || (imUNKNOWN == modality) || imCOUNT == modality )
                        presetList << presetName;
        }

        return presetList;
}

void TransferFunctions3DPresets::deletePresetData(QString name, bool _2D, bool _3D)
{
        //todo rewrite
        std::cout << "TODO rewrite TransferFunctions3DPresets::deletePresetData(QString name, bool _2D, bool _3D)" << std::endl;
        XmlOptionFile node = this->getPresetNode(name);

        if (_2D)
                node.descend("lookuptable2D").deleteNode();
        if (_3D)
                node.descend("transferfunctions").deleteNode();
        if (_2D && _3D)
                node.deleteNode();

        emit changed();
}

}