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();
}
 
}