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

#include <QDomDocument>
#include <QDir>
#include <vtkImageAccumulate.h>
#include <vtkImageReslice.h>
#include <vtkImageData.h>
#include <vtkMatrix4x4.h>
#include <vtkPlane.h>
#include <vtkPlanes.h>
#include <vtkImageResample.h>
#include <vtkImageChangeInformation.h>
#include <vtkImageClip.h>
#include <vtkImageIterator.h>
#include <vtkPiecewiseFunction.h>
#include <vtkColorTransferFunction.h>
#include "cxImageTF3D.h"
#include "cxBoundingBox3D.h"
#include "cxImageLUT2D.h"
#include "cxRegistrationTransform.h"
#include "cxLandmark.h"
#include "cxFileManagerService.h"
#include "cxLogger.h"
#include "cxTypeConversions.h"
#include "cxUtilHelpers.h"
#include "cxVolumeHelpers.h"
#include "cxImageDefaultTFGenerator.h"
#include "cxNullDeleter.h"
#include "cxSettings.h"
#include "cxUnsignedDerivedImage.h"
#include "cxEnumConversion.h"
#include "cxCustomMetaImage.h"

typedef vtkSmartPointer<vtkImageChangeInformation> vtkImageChangeInformationPtr;

namespace cx
{

Image::ShadingStruct::ShadingStruct()
{
        on = settings()->value("View/shadingOn").value<bool>();
        ambient = 0.2;
        diffuse = 0.9;
        specular = 0.3;
        specularPower = 15.0;
}

double Image::ShadingStruct::loadAttribute(QDomNode dataNode, QString name, double defVal)
{
        QString text = dataNode.toElement().attribute(name);
        bool ok;
        double val = text.toDouble(&ok);
        if (ok)
                return val;
        return defVal;
}

void Image::ShadingStruct::addXml(QDomNode dataNode)
{
        QDomElement elem = dataNode.toElement();
        elem.setAttribute("on", on);
        elem.setAttribute("ambient", ambient);
        elem.setAttribute("diffuse", diffuse);
        elem.setAttribute("specular", specular);
        elem.setAttribute("specularPower", specularPower);
}

void Image::ShadingStruct::parseXml(QDomNode dataNode)
{
        if (dataNode.isNull())
                return;

        on = dataNode.toElement().attribute("on").toInt();
        //      std::cout << "attrib on: " << dataNode.toElement().attribute("on")  << " : " << on << std::endl;
        ambient = loadAttribute(dataNode, "ambient", ambient);
        diffuse = loadAttribute(dataNode, "diffuse", diffuse);
        specular = loadAttribute(dataNode, "specular", specular);
        specularPower = loadAttribute(dataNode, "specularPower", specularPower);
}

//---------------------------------------------------------
//---------------------------------------------------------
//---------------------------------------------------------

ImagePtr Image::create(const QString& uid, const QString& name)
{
        return ImagePtr(new Image(uid, vtkImageDataPtr(), name));
}

Image::~Image()
{
}

Image::Image(const QString& uid, const vtkImageDataPtr& data, const QString& name) :
        Data(uid, name), mBaseImageData(data), mMaxRGBIntensity(-1), mThresholdPreview(false)
{
        mInitialWindowWidth = -1;
        mInitialWindowLevel = -1;

        mInterpolationType = VTK_LINEAR_INTERPOLATION;
        mUseCropping = false;
        mCroppingBox_d = this->getInitialBoundingBox();
        mModality = imUNKNOWN;

        mImageLookupTable2D.reset();
        mImageTransferFunctions3D.reset();

        this->setAcquisitionTime(QDateTime::currentDateTime());
}

ImagePtr Image::copy()
{
        vtkImageDataPtr baseImageDataCopy;
        if(mBaseImageData)
        {
                baseImageDataCopy = vtkImageDataPtr::New();
                baseImageDataCopy->DeepCopy(mBaseImageData);
        }

        ImagePtr retval = ImagePtr(new Image(mUid, baseImageDataCopy, mName));

        retval->mUnsigned = mUnsigned;
        retval->mModality = mModality;
        retval->mImageType = mImageType;
        retval->mMaxRGBIntensity = mMaxRGBIntensity;
        retval->mInterpolationType = mInterpolationType;
        retval->mImageLookupTable2D = mImageLookupTable2D;
        retval->mImageTransferFunctions3D = mImageTransferFunctions3D;
        retval->mInitialWindowWidth = mInitialWindowWidth;
        retval->mInitialWindowLevel = mInitialWindowLevel;

        //From cx::Data
        retval->mRegistrationStatus = mRegistrationStatus;
        retval->m_rMd_History = m_rMd_History;

        return retval;
}

void Image::intitializeFromParentImage(ImagePtr parentImage)
{
        this->get_rMd_History()->setRegistration(parentImage->get_rMd());
        this->get_rMd_History()->setParentSpace(parentImage->getUid());
        ImageTF3DPtr transferFunctions = parentImage->getUnmodifiedTransferFunctions3D()->createCopy();
        ImageLUT2DPtr LUT2D = parentImage->getUnmodifiedLookupTable2D()->createCopy();
        this->setLookupTable2D(LUT2D);
        this->setTransferFunctions3D(transferFunctions);
        this->setModality(parentImage->getModality());
        this->setImageType(parentImage->getImageType());
        this->setShading(parentImage->getShading());
        mInitialWindowWidth = parentImage->getInitialWindowWidth();
        mInitialWindowLevel = parentImage->getInitialWindowLevel();
}

DoubleBoundingBox3D Image::getInitialBoundingBox() const
{
        return DoubleBoundingBox3D(-1, -1, -1, -1, -1, -1);
}

ImagePtr Image::getUnsigned(ImagePtr self)
{
        CX_ASSERT(this==self.get());

        if (!mUnsigned)
        {
                // self is unsigned: return self
                if (this->getBaseVtkImageData()->GetScalarTypeMin() >= 0)
                        return self;
                else // signed: create unsigned adapter
                        mUnsigned = UnsignedDerivedImage::create(self);
        }

        return mUnsigned;
}



void Image::resetTransferFunctions(bool _2D, bool _3D)
{
        if (!mBaseImageData)
        {
                reportWarning("Image has no image data");
                return;
        }

        mBaseImageData->GetScalarRange(); // this line updates some internal vtk value, and (on fedora) removes 4.5s in the second render().
        mMaxRGBIntensity = -1;

        ImageDefaultTFGenerator tfGenerator(ImagePtr(this, null_deleter()));
        if (_3D)
        {
                this->resetTransferFunction(tfGenerator.generate3DTFPreset());
                tfGenerator.resetShading();
        }
        if (_2D)
                this->resetTransferFunction(tfGenerator.generate2DTFPreset());
}

void Image::resetTransferFunction(ImageTF3DPtr imageTransferFunctions3D, ImageLUT2DPtr imageLookupTable2D)
{
        this->blockSignals(true); // avoid emitting two transferFunctionsChanged() for one call.

        this->resetTransferFunction(imageTransferFunctions3D);
        this->resetTransferFunction(imageLookupTable2D);

        this->blockSignals(false);
        emit transferFunctionsChanged();
}

void Image::resetTransferFunction(ImageLUT2DPtr imageLookupTable2D)
{
        if (mImageLookupTable2D)
        {
                disconnect(mImageLookupTable2D.get(), &ImageTFData::transferFunctionsChanged, this, &Image::transferFunctionsChanged);
        }

        mImageLookupTable2D = imageLookupTable2D;

        if (mImageLookupTable2D)
        {
                connect(mImageLookupTable2D.get(), &ImageTFData::transferFunctionsChanged, this, &Image::transferFunctionsChanged);
        }

        emit transferFunctionsChanged();
}

void Image::resetTransferFunction(ImageTF3DPtr imageTransferFunctions3D)
{
        if (mImageTransferFunctions3D)
        {
                disconnect(mImageTransferFunctions3D.get(), &ImageTFData::transferFunctionsChanged, this, &Image::transferFunctionsChanged);
        }

        mImageTransferFunctions3D = imageTransferFunctions3D;

        if (mImageTransferFunctions3D)
        {
                connect(mImageTransferFunctions3D.get(), &ImageTFData::transferFunctionsChanged, this, &Image::transferFunctionsChanged);
        }

        emit transferFunctionsChanged();
}

void Image::transformChangedSlot()
{
}

void Image::moveThisAndChildrenToThread(QThread* thread)
{
          // important! move thread affinity to main thread - ensures signals/slots is still called correctly
          this->moveToThread(thread);
          this->getUnmodifiedTransferFunctions3D()->moveToThread(thread);
          this->getUnmodifiedLookupTable2D()->moveToThread(thread);
          this->get_rMd_History()->moveToThread(thread);
}

void Image::setVtkImageData(const vtkImageDataPtr& data, bool resetTransferFunctions)
{
        mBaseImageData = data;
        mBaseGrayScaleImageData = NULL;
        mHistogramPtr = NULL;

        if (resetTransferFunctions)
                this->resetTransferFunctions();
        emit vtkImageDataChanged(mUid);
}

vtkImageDataPtr Image::get8bitGrayScaleVtkImageData()
{
        double windowWidth = this->getUnmodifiedLookupTable2D()->getWindow();
        double windowLevel = this->getUnmodifiedLookupTable2D()->getLevel();
        return convertImageDataTo8Bit(this->getGrayScaleVtkImageData(), windowWidth, windowLevel);
}

vtkImageDataPtr Image::getGrayScaleVtkImageData()
{
        if (mBaseGrayScaleImageData)
        {
                return mBaseGrayScaleImageData;
        }

        mBaseGrayScaleImageData = convertImageDataToGrayScale(this->getBaseVtkImageData());
        return mBaseGrayScaleImageData;
}

ImageTF3DPtr Image::getTransferFunctions3D()
{
        if(mThresholdPreview)
                return mTresholdPreviewTransferfunctions3D;
        return getUnmodifiedTransferFunctions3D();
}

ImageTF3DPtr Image::getUnmodifiedTransferFunctions3D()
{
        if(!this->mImageTransferFunctions3D)
                this->resetTransferFunctions(false, true);
        return mImageTransferFunctions3D;
}

void Image::setTransferFunctions3D(ImageTF3DPtr transferFuntion)
{
        this->resetTransferFunction(transferFuntion);
}

ImageLUT2DPtr Image::getLookupTable2D()
{
        if(mThresholdPreview)
                return mTresholdPreviewLookupTable2D;
        return getUnmodifiedLookupTable2D();
}

ImageLUT2DPtr Image::getUnmodifiedLookupTable2D()
{
        if(!mImageLookupTable2D)
                this->resetTransferFunctions(true, false);
        return mImageLookupTable2D;
}

void Image::setLookupTable2D(ImageLUT2DPtr imageLookupTable2D)
{
        this->resetTransferFunction(imageLookupTable2D);
}

vtkImageDataPtr Image::getBaseVtkImageData()
{
        return mBaseImageData;
}

DoubleBoundingBox3D Image::boundingBox() const
{
//      mBaseImageData->UpdateInformation();
        DoubleBoundingBox3D bounds(mBaseImageData->GetBounds());
        return bounds;
}

Eigen::Array3d Image::getSpacing() const
{
        return Eigen::Array3d(mBaseImageData->GetSpacing());
}

vtkImageAccumulatePtr Image::getHistogram()
{
        if (mHistogramPtr.GetPointer() == NULL)
        {
                mHistogramPtr = vtkImageAccumulatePtr::New();
                mHistogramPtr->SetInputData(this->getGrayScaleVtkImageData());
                mHistogramPtr->IgnoreZeroOn(); // required for Sonowand CT volumes, where data are placed between 31K and 35K.
                // Set up only a 1D histogram for now, so y and z values are set to 0
                mHistogramPtr->SetComponentExtent(0, this->getRange(), 0, 0, 0, 0);
                mHistogramPtr->SetComponentOrigin(this->getMin(), 0, 0);
                mHistogramPtr->SetComponentSpacing(1, 0, 0);
        }
        mHistogramPtr->Update();
        return mHistogramPtr;
}

template<typename scalartype> static int getRGBMax(vtkImageDataPtr image)
{
        int max = 0;
        vtkImageIterator<scalartype> iter(image, image->GetExtent());
        while (!iter.IsAtEnd())
        {
                typename vtkImageIterator<scalartype>::SpanIterator siter = iter.BeginSpan();
                while (siter != iter.EndSpan())
                {
                        int value = *siter;
                        ++siter;
                        value += *siter;
                        ++siter;
                        value += *siter;
                        ++siter;
                        if (value > max)
                        {
                                max = value;
                        }
                }
                iter.NextSpan();
        }
        return max/3;
}


int Image::getMax()
{
        // Alternatively create max from histogram
        //IntIntMap::iterator iter = this->getHistogram()->end();
        //iter--;
        //return (*iter).first;
        if (mBaseImageData->GetNumberOfScalarComponents() == 3)
        {
                if (mMaxRGBIntensity != -1)
                {
                        return mMaxRGBIntensity;
                }
                double max = 0.0;
                switch (mBaseImageData->GetScalarType())
                {
                case VTK_UNSIGNED_CHAR:
                        max = getRGBMax<unsigned char>(mBaseImageData);
                        break;
                case VTK_UNSIGNED_SHORT:
                        max = getRGBMax<unsigned short>(mBaseImageData);
                        break;
                default:
                        CX_LOG_ERROR() << "Unhandled RGB data type in image " << this->getUid();
                        break;
                }
                mMaxRGBIntensity = max;
                return (int)mMaxRGBIntensity;
        }
        else
        {
//              return (int) this->getTransferFunctions3D()->getScalarMax();
                return mBaseImageData->GetScalarRange()[1];
        }
}

int Image::getMin()
{
        // Alternatively create min from histogram
        //IntIntMap::iterator iter = this->getHistogram()->begin();
        //return (*iter).first;
        return mBaseImageData->GetScalarRange()[0];
//      return (int) this->getTransferFunctions3D()->getScalarMin();
}

int Image::getRange()
{
        return this->getMax() - this->getMin();
}

int Image::getMaxAlphaValue()
{
        return 255;
}

double Image::getVTKMinValue()
{
        int vtkScalarType = mBaseImageData->GetScalarType();

        if (vtkScalarType==VTK_CHAR)
                return VTK_CHAR_MIN;
        else if (vtkScalarType==VTK_UNSIGNED_CHAR)
                return VTK_UNSIGNED_CHAR_MIN;
        else if (vtkScalarType==VTK_SIGNED_CHAR)
                return VTK_SIGNED_CHAR_MIN;
        else if (vtkScalarType==VTK_UNSIGNED_SHORT)
                return VTK_UNSIGNED_SHORT_MIN;
        else if (vtkScalarType==VTK_SHORT)
                return VTK_SHORT_MIN;
        else if (vtkScalarType==VTK_UNSIGNED_INT)
                return VTK_UNSIGNED_INT_MIN;
        else if (vtkScalarType==VTK_INT)
                return VTK_INT_MIN;
        else if (vtkScalarType==VTK_FLOAT)
                return VTK_FLOAT_MIN;
        else
                reportError(QString("Unknown VTK ScalarType: %1").arg(vtkScalarType));
        return 0;
}

double Image::getVTKMaxValue()
{
        int vtkScalarType = mBaseImageData->GetScalarType();

        if (vtkScalarType==VTK_CHAR)
                return VTK_CHAR_MAX;
        else if (vtkScalarType==VTK_UNSIGNED_CHAR)
                return VTK_UNSIGNED_CHAR_MAX;
        else if (vtkScalarType==VTK_SIGNED_CHAR)
                return VTK_SIGNED_CHAR_MAX;
        else if (vtkScalarType==VTK_UNSIGNED_SHORT)
                return VTK_UNSIGNED_SHORT_MAX;
        else if (vtkScalarType==VTK_SHORT)
                return VTK_SHORT_MAX;
        else if (vtkScalarType==VTK_UNSIGNED_INT)
                return VTK_UNSIGNED_INT_MAX;
        else if (vtkScalarType==VTK_INT)
                return VTK_INT_MAX;
        else if (vtkScalarType==VTK_FLOAT)
                return VTK_FLOAT_MAX;
        else
                reportError(QString("Unknown VTK ScalarType: %1").arg(vtkScalarType));
        return 0;
}

bool Image::is2D()
{
        return this->getBaseVtkImageData()->GetDimensions()[2]==1;
}

void Image::setDicomSeriesNumber(QString seriesNumber)
{
        mDicomSeriesNumber = seriesNumber;
}

QString Image::getDicomSeriesNumber()
{
        return mDicomSeriesNumber;
}

void Image::addXml(QDomNode& dataNode)
{
        Data::addXml(dataNode);
        QDomNode imageNode = dataNode;
        QDomDocument doc = dataNode.ownerDocument();

        QDomElement tf3DNode = doc.createElement("transferfunctions");
        this->getUnmodifiedTransferFunctions3D()->addXml(tf3DNode);
        imageNode.appendChild(tf3DNode);

        QDomElement lut2DNode = doc.createElement("lookuptable2D");
        this->getUnmodifiedLookupTable2D()->addXml(lut2DNode);
        imageNode.appendChild(lut2DNode);

        QDomElement shadingNode = doc.createElement("shading");
        mShading.addXml(shadingNode);
        imageNode.appendChild(shadingNode);

//      QDomElement landmarksNode = doc.createElement("landmarks");
//      mLandmarks->addXml(landmarksNode);
//      imageNode.appendChild(landmarksNode);

        QDomElement cropNode = doc.createElement("crop");
        cropNode.setAttribute("use", mUseCropping);
        cropNode.appendChild(doc.createTextNode(qstring_cast(mCroppingBox_d)));
        imageNode.appendChild(cropNode);

        QDomElement clipNode = doc.createElement("clip");
        for (unsigned i = 0; i < mPersistentClipPlanes.size(); ++i)
        {
                QDomElement planeNode = doc.createElement("plane");
                Vector3D normal(mPersistentClipPlanes[i]->GetNormal());
                Vector3D origin(mPersistentClipPlanes[i]->GetOrigin());
                planeNode.setAttribute("normal", qstring_cast(normal));
                planeNode.setAttribute("origin", qstring_cast(origin));
                clipNode.appendChild(planeNode);
        }
        imageNode.appendChild(clipNode);

        QDomElement modalityNode = doc.createElement("modality");
        modalityNode.appendChild(doc.createTextNode(enum2string(mModality)));
        imageNode.appendChild(modalityNode);

        QDomElement imageTypeNode = doc.createElement("imageType");
        imageTypeNode.appendChild(doc.createTextNode(enum2string(mImageType)));
        imageNode.appendChild(imageTypeNode);

        QDomElement interpolationNode = doc.createElement("vtk_interpolation");
        interpolationNode.setAttribute("type", mInterpolationType);
        imageNode.appendChild(interpolationNode);

        QDomElement initialWindowNode = doc.createElement("initialWindow");
        initialWindowNode.setAttribute("width", mInitialWindowWidth);
        initialWindowNode.setAttribute("level", mInitialWindowLevel);
        imageNode.appendChild(initialWindowNode);
}

double Image::loadAttribute(QDomNode dataNode, QString name, double defVal)
{
        QString text = dataNode.toElement().attribute(name);
        bool ok;
        double val = text.toDouble(&ok);
        if (ok)
                return val;
        return defVal;
}

bool Image::load(QString path, FileManagerServicePtr filemanager)
{
        ImagePtr self = ImagePtr(this, null_deleter());
        filemanager->readInto(self, path);
        return this->getBaseVtkImageData()!=0;
}

void Image::parseXml(QDomNode& dataNode)
{
        Data::parseXml(dataNode);

        // image node must be parsed in the data manager to create this Image object
        // Only subnodes are parsed here

        if (dataNode.isNull())
                return;

        //transferefunctions
        QDomNode transferfunctionsNode = dataNode.namedItem("transferfunctions");
        if (!transferfunctionsNode.isNull())
                this->getUnmodifiedTransferFunctions3D()->parseXml(transferfunctionsNode);
        else
        {
                std::cout << "Warning: Image::parseXml() found no transferfunctions";
                std::cout << std::endl;
        }

        mInitialWindowWidth = this->loadAttribute(dataNode.namedItem("initialWindow"), "width", mInitialWindowWidth);
        mInitialWindowLevel = this->loadAttribute(dataNode.namedItem("initialWindow"), "level", mInitialWindowLevel);

        this->getUnmodifiedLookupTable2D()->parseXml(dataNode.namedItem("lookuptable2D"));

        // backward compatibility:
        mShading.on = dataNode.namedItem("shading").toElement().text().toInt();
        //Assign default values if the shading nodes don't exists to allow backward compability
        if (!dataNode.namedItem("shadingAmbient").isNull())
                mShading.ambient = dataNode.namedItem("shadingAmbient").toElement().text().toDouble();
        if (!dataNode.namedItem("shadingDiffuse").isNull())
                mShading.diffuse = dataNode.namedItem("shadingDiffuse").toElement().text().toDouble();
        if (!dataNode.namedItem("shadingSpecular").isNull())
                mShading.specular = dataNode.namedItem("shadingSpecular").toElement().text().toDouble();
        if (!dataNode.namedItem("shadingSpecularPower").isNull())
                mShading.specularPower = dataNode.namedItem("shadingSpecularPower").toElement().text().toDouble();

        // new way:
        mShading.parseXml(dataNode.namedItem("shading"));

//      mLandmarks->parseXml(dataNode.namedItem("landmarks"));

        QDomElement cropNode = dataNode.namedItem("crop").toElement();
        if (!cropNode.isNull())
        {
                mUseCropping = cropNode.attribute("use").toInt();
                mCroppingBox_d = DoubleBoundingBox3D::fromString(cropNode.text());
        }

        QDomElement clipNode = dataNode.namedItem("clip").toElement();
        QDomElement clipPlaneNode = clipNode.firstChildElement("plane");
        for (; !clipPlaneNode.isNull(); clipPlaneNode = clipPlaneNode.nextSiblingElement("plane"))
        {
                Vector3D normal = Vector3D::fromString(clipPlaneNode.attribute("normal"));
                Vector3D origin = Vector3D::fromString(clipPlaneNode.attribute("origin"));
                vtkPlanePtr plane = vtkPlanePtr::New();
                plane->SetNormal(normal.begin());
                plane->SetOrigin(origin.begin());
                mPersistentClipPlanes.push_back(plane);
        }

        mModality = convertToModality(dataNode.namedItem("modality").toElement().text());
        mImageType = convertToImageSubType(dataNode.namedItem("imageType").toElement().text());

        QDomElement interpoationNode = dataNode.namedItem("vtk_interpolation").toElement();
        if (!interpoationNode.isNull())
        {
                mInterpolationType = interpoationNode.attribute("type").toInt();
                emit vtkImageDataChanged(mUid);
        }
}

void Image::setInitialWindowLevel(double width, double level)
{
        mInitialWindowWidth = width;
        mInitialWindowLevel = level;
}

void Image::setShadingOn(bool on)
{
        mShading.on = on;
        emit transferFunctionsChanged();
}

bool Image::getShadingOn() const
{
        if (mThresholdPreview)
                return true;
        return mShading.on;
}

void Image::setShadingAmbient(double ambient)
{
        mShading.ambient = ambient;
        emit transferFunctionsChanged();
}

void Image::setShadingDiffuse(double diffuse)
{
        mShading.diffuse = diffuse;
        emit transferFunctionsChanged();
}

void Image::setShadingSpecular(double specular)
{
        mShading.specular = specular;
        emit transferFunctionsChanged();
}

void Image::setShadingSpecularPower(double specularPower)
{
        mShading.specularPower = specularPower;
        emit transferFunctionsChanged();
}

double Image::getShadingAmbient()
{
        return mShading.ambient;
}

double Image::getShadingDiffuse()
{
        return mShading.diffuse;
}

double Image::getShadingSpecular()
{
        return mShading.specular;
}

double Image::getShadingSpecularPower()
{
        return mShading.specularPower;
}

Image::ShadingStruct Image::getShading()
{
        return mShading;
}

void Image::setShading(Image::ShadingStruct shading)
{
        mShading = shading;
        emit transferFunctionsChanged();
}

// methods for defining and storing a cropping box. Image does not use these data, this is up to the mapper
void Image::setCropping(bool on)
{
        if (mUseCropping == on)
                return;

        mUseCropping = on;
        if (similar(mCroppingBox_d, this->getInitialBoundingBox()))
                mCroppingBox_d = this->boundingBox();
        emit cropBoxChanged();
}

bool Image::getCropping() const
{
        return mUseCropping;
}

void Image::setCroppingBox(const DoubleBoundingBox3D& bb_d)
{
        if (similar(mCroppingBox_d, bb_d))
                return;
        mCroppingBox_d = bb_d;
        emit cropBoxChanged();
}

DoubleBoundingBox3D Image::getCroppingBox() const
{
        if (similar(mCroppingBox_d, this->getInitialBoundingBox()))
                return this->boundingBox();
        return mCroppingBox_d;
}

void Image::mergevtkSettingsIntosscTransform()
{
        // the internal CustusX format does not handle extents starting at non-zero.
        // Move extent to zero and change rMd.
        Vector3D origin(mBaseImageData->GetOrigin());
        Vector3D spacing(mBaseImageData->GetSpacing());
        IntBoundingBox3D extent(mBaseImageData->GetExtent());
        Vector3D extentShift = multiply_elems(extent.corner(0, 0, 0).cast<double>(), spacing);

        vtkImageChangeInformationPtr info = vtkImageChangeInformationPtr::New();
        info->SetInputData(mBaseImageData);
        info->SetOutputExtentStart(0, 0, 0);
        info->SetOutputOrigin(0, 0, 0);
        info->Update();
        info->UpdateInformation();
        mBaseImageData = info->GetOutput();

        mBaseImageData->ComputeBounds();
//      mBaseImageData->Update();
//      mBaseImageData->UpdateInformation();

        this->get_rMd_History()->setRegistration(this->get_rMd() * createTransformTranslate(origin + extentShift));

        emit vtkImageDataChanged(mUid);
        emit clipPlanesChanged();
        emit cropBoxChanged();
}

IMAGE_MODALITY Image::getModality() const
{
        return mModality;
}

void Image::setModality(const IMAGE_MODALITY& val)
{
        mModality = val;
        emit propertiesChanged();
}

IMAGE_SUBTYPE Image::getImageType() const
{
        return mImageType;
}

void Image::setImageType(const IMAGE_SUBTYPE& val)
{
        mImageType = val;
        emit propertiesChanged();
}

vtkImageDataPtr Image::createDummyImageData(int axisSize, int maxVoxelValue)
{
        int size = axisSize - 1;//Modify axis size as extent starts with 0, not 1
        vtkImageDataPtr dummyImageData = vtkImageDataPtr::New();
        dummyImageData->SetExtent(0, size, 0, size, 0, size);
        dummyImageData->SetSpacing(1, 1, 1);
        //dummyImageData->SetScalarTypeToUnsignedShort();
//      dummyImageData->SetScalarTypeToUnsignedChar();
//      dummyImageData->SetNumberOfScalarComponents(1);
//      dummyImageData->AllocateScalars();
        dummyImageData->AllocateScalars(VTK_UNSIGNED_CHAR, 1);
        unsigned char* dataPtr = static_cast<unsigned char*> (dummyImageData->GetScalarPointer());

        //Init voxel colors
        int minVoxelValue = 0;
        int numVoxels = axisSize*axisSize*axisSize;
        for (int i = 0; i < numVoxels; ++i)
        {
                int voxelValue = minVoxelValue + i;
                if (i == numVoxels)
                        dataPtr[i] = maxVoxelValue;
                else if (voxelValue < maxVoxelValue)
                        dataPtr[i] = voxelValue;
                else
                        dataPtr[i] = maxVoxelValue;
        }
        setDeepModified(dummyImageData);
        return dummyImageData;
}

//void Image::setInterpolationTypeToNearest()
//{
//      this->setInterpolationType(VTK_NEAREST_INTERPOLATION);
//}
//void Image::setInterpolationTypeToLinear()
//{
//      this->setInterpolationType(VTK_LINEAR_INTERPOLATION);
//}

void Image::setInterpolationType(int val)
{
        if (mInterpolationType == val)
                return;
        if (mThresholdPreview)
                return;
        mInterpolationType = val;
        emit vtkImageDataChanged(mUid);
}
int Image::getInterpolationType() const
{
        if (mThresholdPreview)
                return VTK_NEAREST_INTERPOLATION;
        return mInterpolationType;
}

vtkImageDataPtr Image::resample(long maxVoxels)
{
        // also use grayscale as vtk is incapable of rendering 3component color.
        vtkImageDataPtr retval = this->getGrayScaleVtkImageData();

        double factor = computeResampleFactor(maxVoxels);

        if (fabs(1.0-factor)>0.01) // resampling
        {
                vtkImageResamplePtr resampler = vtkImageResamplePtr::New();
                resampler->SetInterpolationModeToLinear();
                resampler->SetAxisMagnificationFactor(0, factor);
                resampler->SetAxisMagnificationFactor(1, factor);
                resampler->SetAxisMagnificationFactor(2, factor);
                resampler->SetInputData(retval);
//              resampler->GetOutput()->Update();
                resampler->Update();
                resampler->GetOutput()->GetScalarRange();
                retval = resampler->GetOutput();

//              long voxelsDown = retval->GetNumberOfPoints();
//              long voxelsOrig = this->getBaseVtkImageData()->GetNumberOfPoints();
//              report("Created downsampled volume in Image: "
//                                                                       + this->getName()
//                                                                       + " below " + qstring_cast(voxelsDown/1000/1000) + "M. "
//                                                                       + "Ratio: " + QString::number(factor, 'g', 2) + ", "
//                                                                       + "Original size: " + qstring_cast(voxelsOrig/1000/1000) + "M.");
        }
        return retval;
}

double Image::computeResampleFactor(long maxVoxels)
{
        if (maxVoxels==0)
                return 1.0;

        long voxels = this->getBaseVtkImageData()->GetNumberOfPoints();
        double factor = (double)maxVoxels/(double)voxels;
        factor = pow(factor, 1.0/3.0);
        // cubic function leads to trouble for 138M-volume - must downsample to as low as 5-10 Mv in order to succeed on Mac.

        if (factor<0.99)
        {
                return factor;
        }
        return 1.0;
}

void Image::save(const QString& basePath, FileManagerServicePtr filemanager)
{
        QString filename = basePath + "/Images/" + this->getUid() + ".mhd";
        this->setFilename(QDir(basePath).relativeFilePath(filename));

        ImagePtr self = ImagePtr(this, null_deleter());
        filemanager->save(self, filename);
}

void Image::startThresholdPreview(const Eigen::Vector2d &threshold)
{
        mThresholdPreview = true;

        this->createThresholdPreviewTransferFunctions3D(threshold);
        this->createThresholdPreviewLookupTable2D(threshold);

        emit transferFunctionsChanged();
}

void Image::createThresholdPreviewTransferFunctions3D(const Eigen::Vector2d &threshold)
{
        ImageDefaultTFGenerator tfGenerator(ImagePtr(this, null_deleter()));

        ColorMap colors = this->createPreviewColorMap(threshold);
        IntIntMap opacity = this->createPreviewOpacityMap(threshold);

        mTresholdPreviewTransferfunctions3D = tfGenerator.generate3DTFPreset();
        mTresholdPreviewTransferfunctions3D->resetColor(colors);
        mTresholdPreviewTransferfunctions3D->resetAlpha(opacity);
}

void Image::createThresholdPreviewLookupTable2D(const Eigen::Vector2d &threshold)
{
        ImageDefaultTFGenerator tfGenerator(ImagePtr(this, null_deleter()));

        ColorMap colors = this->createPreviewColorMap(threshold);

        mTresholdPreviewLookupTable2D = tfGenerator.generate2DTFPreset();
        mTresholdPreviewLookupTable2D->resetColor(colors);
        mTresholdPreviewLookupTable2D->setLLR(threshold[0]);
}

ColorMap Image::createPreviewColorMap(const Eigen::Vector2d &threshold)
{
        double lower = threshold[0];
        ColorMap colors;
        colors[lower] = Qt::green;
        colors[this->getMax()] = Qt::green;
        return colors;
}

IntIntMap Image::createPreviewOpacityMap(const Eigen::Vector2d &threshold)
{
        double lower = threshold[0];
        double upper = threshold[1];
        IntIntMap opacity;
        opacity[lower - 1] = 0;
        opacity[lower] = this->getMaxAlphaValue();
        opacity[upper] = this->getMaxAlphaValue();
        opacity[upper + 1] = 0;
        return opacity;
}

void Image::stopThresholdPreview()
{
        mThresholdPreview = false;
        mTresholdPreviewTransferfunctions3D.reset();
        mTresholdPreviewLookupTable2D.reset();

        //Need to tag these transfer functions as modified to tell the VTK pipeline that we got new TFs
        this->getTransferFunctions3D()->getColorTF()->Modified();
        this->getTransferFunctions3D()->getOpacityTF()->Modified();

        emit transferFunctionsChanged();
}

} // namespace cx