cxImageLUT2D.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.
=========================================================================*/


/*
 * sscImageLookupTable2D.cpp
 *
 *  Created on: Jan 9, 2009
 *      Author: christiana
 */

#include "cxImageLUT2D.h"

#include <QDomDocument>
#include <vtkLookupTable.h>
#include <vtkImageData.h>
#include <vtkColorTransferFunction.h>
#include <vtkPiecewiseFunction.h>

#include "cxVector3D.h"


namespace cx
{

ImageLUT2D::ImageLUT2D()
{
}

//void ImageLUT2D::setInitialTFFromImage(vtkImageDataPtr base)
//{
//      double smin = base->GetScalarRange()[0];
//      double smax = base->GetScalarRange()[1];
//      double srange = smax - smin;

//      // this sets the initial opacity tf to full
//      mOpacityMap.clear();
//      this->addAlphaPoint(smin - 1, 0);
//      this->addAlphaPoint(smin, 255);

//      // this also sets the initial lut to grayscale
//      mColorMap.clear();
//      this->addColorPoint(smin, Qt::black);
//      this->addColorPoint(smax, Qt::white);


//      this->internalsHaveChanged();
//}

ImageLUT2DPtr ImageLUT2D::createCopy()
{
        ImageLUT2DPtr retval(new ImageLUT2D());
        retval->deepCopy(this);
        return retval;
}

void ImageLUT2D::setFullRangeWinLevel(vtkImageDataPtr image)
{
        double smin = image->GetScalarRange()[0];
        double smax = image->GetScalarRange()[1];
        double srange = smax - smin;
        this->setWindow(srange);
        this->setLevel(smin + srange / 2.0);
}

vtkLookupTablePtr ImageLUT2D::getOutputLookupTable()
{
        if (!mOutputLUT)
        {
                mOutputLUT = vtkLookupTablePtr::New();
                this->refreshOutputLUT();
        }
        return mOutputLUT;
}

// * This is because the 2D renderer only handles llr+alpha.
// */
//void ImageLUT2D::buildOpacityMapFromLLRAlpha()
//{
//      // REMOVED CA 2014-02-07 - TODO
//      mOpacityMap.clear();
//              this->addAlphaPoint(this->getLLR() - 1, 0);
//      this->addAlphaPoint(this->getLLR(), this->getAlpha() * 255);
//}

void ImageLUT2D::internalsHaveChanged()
{
        this->refreshOutputLUT();
        emit transferFunctionsChanged();
}

std::pair<int,int> ImageLUT2D::getMapsRange()
{
        if (!mColorMap.empty() && !mOpacityMap.empty())
        {
                int imin = std::min(mColorMap.begin()->first, mOpacityMap.begin()->first);
                int imax = std::max(mColorMap.rbegin()->first, mOpacityMap.rbegin()->first);
                return std::make_pair(imin,imax);
        }
        else if (!mColorMap.empty())
        {
                int imin = mColorMap.begin()->first;
                int imax = mColorMap.rbegin()->first;
                return std::make_pair(imin,imax);
        }
        else if (!mOpacityMap.empty())
        {
                int imin = mOpacityMap.begin()->first;
                int imax = mOpacityMap.rbegin()->first;
                return std::make_pair(imin,imax);
        }
        else
        {
                return std::make_pair(0,0);
        }
}

void ImageLUT2D::refreshOutputLUT()
{
        if (!mOutputLUT)
                return;

        std::pair<int,int> range = this->getMapsRange();
        int imin = range.first;
        int imax = range.second;
        if (imin==imax)
                imax = imin+1;
        int icount = imax - imin + 1;

        vtkLookupTablePtr lut = mOutputLUT;
        lut->Build();
        lut->SetNumberOfTableValues(icount);
        lut->SetTableRange(imin, imax);

        vtkColorTransferFunctionPtr colorFunc = this->generateColorTF();
        vtkPiecewiseFunctionPtr opacityFunc = this->generateOpacityTF();

        for (int i = 0; i < icount; ++i)
        {
                double* rgb = colorFunc->GetColor(i + imin);
                double alpha = opacityFunc->GetValue(i + imin);
                lut->SetTableValue(i, rgb[0], rgb[1], rgb[2], alpha);
        }

        lut->Modified();
        lut->GetTable()->Modified();

        // HACK WARNING!!!!!
        // Setting vtkLookupTable::SetNumberOfTableValues > 256 causes
        // crash in vtkImageMapToColors. Seen when upgrading vtk6.1->6.2
        // This line hacks away the problem....
        // http://vtk.1045678.n5.nabble.com/Crash-in-vtkImageMapToColors-when-upgrading-6-1-gt-6-2-td5730804.html
        // Update: added fix in VTK head -> will be fixed in 6.3
        unsigned char input = 0;
        lut->MapScalarsThroughTable2(&input, &input, VTK_UNSIGNED_CHAR, 1, 1, 1);
        // HACK END
}

//---------------------------------------------------------
} // end namespace
//---------------------------------------------------------