Eigene Dateien/Vis/src/VTransferFunction.cpp

Go to the documentation of this file.

#include "VTransferFunction.h"
#include "glew.h"
#include <algorithm>
#include "QVolRendCanvas.h"

VTransferFunction::~VTransferFunction()
{
        if(m_VolumeHistogramDisplayList != 0)
        {
                glDeleteLists(m_VolumeHistogramDisplayList, 1);
        }
        m_FuncControlPoints.clear();
        if(m_VolumeHistogram)
        {
                m_VolumeHistogram->clear();
        }

        if(m_MainCanvas)
        {
                m_MainCanvas = NULL;
        }
        m_TransferFunction.clear();

        
}


void VTransferFunction::draw1d()
{
        if(m_VolumeHistogramDisplayList != 0)
        {
                glCallList(m_VolumeHistogramDisplayList);
        }

        glBegin(GL_QUADS);
        glColor4f(0.0f, 0.0f, 0.0f, 0.0f);
        glVertex3f(-1.0f, -1.0f, -0.03f);
        glColor4f(0.0f, 0.0f, 0.0f, 0.0f);
        glVertex3f(-1.0f, -0.8f, -0.03f);

        std::map<int, vTransferFunctionPoint>::iterator it = m_FuncControlPoints.begin();
        while(it != m_FuncControlPoints.end())
        {
                //end of quad
                glColor4f((*it).second.m_Color.getX(),(*it).second.m_Color.getY(), (*it).second.m_Color.getZ(), (*it).second.m_Alpha );
                glVertex3f(2*(float)(*it).first/4095.0f - 1.0f, -0.8f, -0.03f);
                glColor4f((*it).second.m_Color.getX(),(*it).second.m_Color.getY(), (*it).second.m_Color.getZ(), (*it).second.m_Alpha );
                glVertex3f(2*(float)(*it).first/4095.0f - 1.0f, -1.0f, -0.03f);
                //beginning of quad
                glColor4f((*it).second.m_Color.getX(),(*it).second.m_Color.getY(), (*it).second.m_Color.getZ(), (*it).second.m_Alpha );
                glVertex3f(2*(float)(*it).first/4095.0f - 1.0f, -1.0f, -0.03f);
                glColor4f((*it).second.m_Color.getX(),(*it).second.m_Color.getY(), (*it).second.m_Color.getZ(), (*it).second.m_Alpha );
                glVertex3f(2*(float)(*it).first/4095.0f - 1.0f, -0.8f, -0.03f);
                ++it;
        }
        glColor4f(1.0f, 1.0f, 1.0f, 1.0f);
        glVertex3f(1.0f, -0.8f, -0.03f);
        glColor4f(1.0f, 1.0f, 1.0f, 1.0f);
        glVertex3f(1.0f, -1.0f, -0.03f);
        glEnd();
        
        if(m_zoom_active&&m_MainCanvas->getLoaded())
        {
                float x1 =  (float)max(zoom_ptx - 0.15,-1.0) ;
                float x2 =  (float)min(zoom_ptx + 0.15, 1.0) ;
                float localmax = 0.0;

                std::vector<int> tmpHist = *m_VolumeHistogram;
                for (int i = (int)((x1 + 1.0f)*2047); i < (int)((x2+1.0f)*2047); i++)
                {
                        if(tmpHist[i] > localmax) localmax = tmpHist[i];
                }

                float upper = (bigzoom) ? 1.0f : -0.333f;

                glBegin(GL_QUADS);

                        glColor4f(0.9176f, 0.8235f, 0.7451f, 0.3f);

                        glVertex3f(x1,-0.8f, -0.01f);   
                        glVertex3f(x2,-0.8f, -0.01f);   
                        glVertex3f(x2,1.0f, -0.01f);    
                        glVertex3f(x1,1.0f, -0.01f);
                
                glEnd();
                
                glBegin(GL_LINES);
                for (int i = (int)((x1 + 1.0f)*2047); i < (int)((x2+1.0f)*2047); i++)
                {
                        glColor3f(0.9176f, 0.4f, 0.4f);
                        glVertex3f(2*(float)(i)/4095.0f - 1.0f, -0.8f, -0.02f);
                        glVertex3f(2*(float)(i)/4095.0f - 1.0f, (2.0f* abs(upper)*(float)m_VolumeHistogram->at(i)/localmax - 0.8f ), -0.02f);
                }
                glEnd();

        }
        
        glBegin(GL_LINE_STRIP);
        glColor3f(0.0f, 0.0f, 0.0f);
        glVertex3f(-1.0f, -0.8f, -0.1f);
        it = m_FuncControlPoints.begin();
        while(it != m_FuncControlPoints.end())
        {
                glColor3f(0.0f, 0.0f, 0.0f);
                glVertex3f((float)(*it).first/4095.0f * 2.0f - 1.0f, (*it).second.m_Alpha * 1.8f - 0.8f, -0.1f);
                ++it;
        }
        glColor3f(0.5f, 0.5f, 0.5f);
        glVertex3f(1.0f, 1.0f, -0.1f);
        glEnd();
        
        glPointSize(7.0f);
        glBegin(GL_POINTS);
        it = m_FuncControlPoints.begin();
        while(it != m_FuncControlPoints.end())
        {
                glColor4f(0.0f, 0.0f, 0.0f, 1.0f);
                glVertex3f((float)(*it).first/4095.0f * 2.0f - 1.0f, (*it).second.m_Alpha * 1.8f - 0.8f, -0.15f);
                ++it;
        }
        glEnd();

        glPointSize(5.0f);
        glBegin(GL_POINTS);
        it = m_FuncControlPoints.begin();
        while(it != m_FuncControlPoints.end())
        {
                glColor4f((*it).second.m_Color.getX(), (*it).second.m_Color.getY(), (*it).second.m_Color.getZ(), 1.0f);
                glVertex3f((float)(*it).first/4095.0f * 2.0f - 1.0f, (*it).second.m_Alpha * 1.8f - 0.8f, -0.2f);
                ++it;
        }
        glEnd();


}

void VTransferFunction::addTransferFunctionPoint(int density, vTransferFunctionPoint newPoint)
{
        m_FuncControlPoints[density] = newPoint;
        m_hasChanged = true;
        //interpolateTransferPoints();
        //m_MainCanvas->repaint();
        //m_MainCanvas->swapBuffers();
}

void VTransferFunction::removeTransferFunctionPoint(int density)
{
        m_FuncControlPoints.erase(density);
        m_hasChanged = true;
        //interpolateTransferPoints();
        //m_MainCanvas->repaint();
        //m_MainCanvas->swapBuffers();
}

void VTransferFunction::generateDisplayList()
{
        if(m_VolumeHistogramDisplayList != 0)
        {
                glDeleteLists(m_VolumeHistogramDisplayList, 1);
        }
        std::vector<int> tmpHist = *m_VolumeHistogram;
        float maxHistogramValue =  m_maxdensity;//float)(*max_element(tmpHist.begin(), tmpHist.end()));
        
        m_VolumeHistogramDisplayList = glGenLists(1);
        glNewList(m_VolumeHistogramDisplayList, GL_COMPILE);
        glBegin(GL_LINES);
        for(int i = 0; i < (int)m_VolumeHistogram->size(); ++i)
        {
                glColor3f(0.7f, 0.7f, 0.7f);
                glVertex3f(2*(float)(i)/4095.0f - 1.0f, -0.8f, 0.0f);   
                glColor3f(0.7f, 0.7f, 0.7f);
                glVertex3f(2*(float)(i)/4095.0f - 1.0f, 2* (float)m_VolumeHistogram->at(i)/maxHistogramValue - 0.8f, 0.0f);
        }
        glEnd();
        glEndList();

}



void VTransferFunction::interpolateTransferPoints()
{
        //m_MainCanvas->makeCurrent();

        if(!m_FramebufferObject)
        {
                m_FramebufferObject = new VFramebufferObject(4096,2,GL_RGBA);
        }

        if(m_FramebufferObject->getFboHandle() == 0)
        {
                m_FramebufferObject->init();
        }
        m_FramebufferObject->bind();

        //glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);   // Clear Screen And Depth Buffer

        glBegin(GL_QUADS);
        glColor4f(0.0f, 0.0f, 0.0f, 0.0f);
        glVertex3f(-1.0f, -1.0f, 0.0f);
        glColor4f(0.0f, 0.0f, 0.0f, 0.0f);
        glVertex3f(-1.0f, 1.0f, 0.0f);

        std::map<int, vTransferFunctionPoint>::iterator it = m_FuncControlPoints.begin();
        while(it != m_FuncControlPoints.end())
        {
                //end of quad
                glColor4f((*it).second.m_Color.getX(),(*it).second.m_Color.getY(), (*it).second.m_Color.getZ(), (*it).second.m_Alpha );
                glVertex3f(2*(float)(*it).first/4095.0f - 1.0f, 1.0f, 0.0f);
                glColor4f((*it).second.m_Color.getX(),(*it).second.m_Color.getY(), (*it).second.m_Color.getZ(), (*it).second.m_Alpha );
                glVertex3f(2*(float)(*it).first/4095.0f - 1.0f, -1.0f, 0.0f);
                //beginning of quad
                glColor4f((*it).second.m_Color.getX(),(*it).second.m_Color.getY(), (*it).second.m_Color.getZ(), (*it).second.m_Alpha );
                glVertex3f(2*(float)(*it).first/4095.0f - 1.0f, -1.0f, 0.0f);
                glColor4f((*it).second.m_Color.getX(),(*it).second.m_Color.getY(), (*it).second.m_Color.getZ(), (*it).second.m_Alpha );
                glVertex3f(2*(float)(*it).first/4095.0f - 1.0f, 1.0f, 0.0f);
                ++it;
        }
        glColor4f(1.0f, 1.0f, 1.0f, 1.0f);
        glVertex3f(1.0f, 1.0f, 0.0f);
        glColor4f(1.0f, 1.0f, 1.0f, 1.0f);
        glVertex3f(1.0f, -1.0f, 0.0f);
        glEnd();

        
        m_FramebufferObject->unbind();

        m_hasChanged = false;
}

bool VTransferFunction::load(std::string filepath)
{
        if(filepath == "")
        {
                return true;
        }
        FILE *fp = NULL;

        fopen_s(&fp,filepath.c_str(),"rb");

        if(!fp)
        {
                return false;
        }
        unsigned int numPoints;

        fread(&numPoints,sizeof(unsigned int),1,fp);

        m_FuncControlPoints.clear();
        for(unsigned int i = 0; i < numPoints; ++i)
        {
                int density;
                fread(&density, sizeof(int), 1, fp);

                vTransferFunctionPoint tmp;
                float x,y,z,a;
                fread(&x, sizeof(float), 1, fp);
                fread(&y, sizeof(float), 1, fp);
                fread(&z, sizeof(float), 1, fp);
                fread(&a, sizeof(float), 1, fp);
                tmp.m_Color = VVector(x,y,z);
                tmp.m_Alpha = a;
                m_FuncControlPoints[density] = tmp;

        }
        fclose(fp);
        m_hasChanged = true;
        return true;
}

bool VTransferFunction::save(std::string filepath)
{
        if(filepath == "")
        {
                return true;
        }
        FILE *fp = NULL;



        fopen_s(&fp,filepath.c_str(),"wb");

        if(!fp)
        {
                return false;
        }
        unsigned int numPoints = (unsigned int)m_FuncControlPoints.size();

        fwrite(reinterpret_cast<char*>(&numPoints), sizeof(unsigned int), 1,fp);
        
        std::map<int, vTransferFunctionPoint>::iterator it = m_FuncControlPoints.begin();

        while(it != m_FuncControlPoints.end())
        {
                float x,y,z,a;
                x = (*it).second.m_Color.getX();
                y = (*it).second.m_Color.getY();
                z = (*it).second.m_Color.getZ();
                a = (*it).second.m_Alpha;
                int density = (*it).first;
                fwrite(reinterpret_cast<char*>(&density), sizeof(int), 1, fp);
                fwrite(reinterpret_cast<char*>(&x), sizeof(float), 1, fp);
                fwrite(reinterpret_cast<char*>(&y), sizeof(float), 1, fp);
                fwrite(reinterpret_cast<char*>(&z), sizeof(float), 1, fp);
                fwrite(reinterpret_cast<char*>(&a), sizeof(float), 1, fp);
                ++it;
        }
        fclose(fp);
        return true;
}

---