Eigene Dateien/Vis/src/VTransferFunction2D.h

Go to the documentation of this file.

#pragma once

#ifndef __v_transfer_function2d_h
#define __v_transfer_function2d_h

#include <map>
#include <vector>

#include "VVector.h"
#include "VFramebufferObject.h"


class QVolRendCanvas;

struct vTransferFunction2DPoint
{
        vTransferFunction2DPoint() : m_Color(), m_Alpha(1.0f), m_gmag(1.0f), m_density(0)
        {

        }

        VVector m_Color;        
        float m_Alpha;          
        float m_gmag;           
        float m_density;        
};


struct vNormal2DPoint
{
        vNormal2DPoint() :x(0.0f), y(0.0f), rgb(), alpha(0.0f)
        {

        }

        
        float x;                
        float y;                
        VVector rgb;    
        float alpha;    
};

struct vTransferFunctionWannabeTriangle
{
        vTransferFunctionWannabeTriangle( ) :a_offsetw(0.0f),a_offseth(0.0f),a_skew(0.0f)
        {

        }


        vTransferFunctionWannabeTriangle(       vNormal2DPoint p1, 
                                                                                vNormal2DPoint p2, 
                                                                                vNormal2DPoint p3,
                                                                                vNormal2DPoint p4) : point1(p1), point2(p2), point3(p3), point4(p4),
                                                                                                                         a_offsetw(0.0f),a_offseth(0.0f),a_skew(0.0f)
        {

        }

        
        vNormal2DPoint point1;                                  
        vNormal2DPoint point2;                                  
        vNormal2DPoint point3;                                  
        vNormal2DPoint point4;                                  
        float a_offsetw;                                                
        float a_offseth;                                                
        float a_skew;                                                   
};

class VTransferFunction2D
{
public:

        VTransferFunction2D() :  
                                                        m_VolumeHistogram2D(0), 
                                                        m_VolumeHistogramDisplayList2D(0),
                                                        m_hasChanged(true), 
                                                        m_TransferFunction2DHandle(0), 
                                                        m_Histogram2DTextureHandle(0),
                                                        m_FramebufferObject(NULL)
        {

        }

        ~VTransferFunction2D();

        void reinit();

        void draw2d();

        void setHistogram2D(unsigned char * volumeHistogram2d);
        
        vTransferFunction2DPoint getTransferFunctionPoint(float density, float gmag);

        vTransferFunctionWannabeTriangle getTransferFunction2dTriangle(float density, float gmag);

        VVector transferFunctionPointInRange(float selectedDensity, float gmag, float range);

        void addTransferFunctionPoint(vTransferFunction2DPoint newPoint);

        void addTransferFunctionPoint(vTransferFunctionWannabeTriangle t, vTransferFunction2DPoint newPoint);

        void removeTransferFunctionPoint(float density, float gmag, float tres);

        int findIndexTransferFunctionPoint(float density, float gmag, float tres);

        void changeTriangleSize( int index, float offsetw, float offseth );

        void changeTriangleSkew( int index, float skew );

        void changePointColor(float density, float gmag, VVector color, float a, float tres);

        void changePointPosition(int index, float newx, float newy);

        std::vector<vTransferFunctionWannabeTriangle> getFullTransferFunction()
        {
                if(m_hasChanged)
                {
                        this->interpolateTransferTriangles();
                }
                return m_FuncControlTriangles;
        }

        unsigned int getGLHandle()
        {
                if(m_hasChanged)
                {
                        this->interpolateTransferTriangles();
                }
                return m_FramebufferObject->getTextureHandle();
                //return m_TransferFunctionHandle;
        }

        void setMainCanvas(QVolRendCanvas * maincanvas)
        {
                m_MainCanvas = maincanvas;
        }
        
        bool load(std::string filename);

        bool save(std::string filename);

        void clear()
        {
                m_FuncControlTriangles.clear();
                m_FuncControlPoints.clear();
                m_TransferFunction2DHandle = 0;
                m_hasChanged =  true;
        }

        void renderTest()
        {
                m_FramebufferObject->renderToLowerRightQuad();
        }

private:
        
        void generateDisplayList2D();

        void interpolateTransferTriangles();

        double round(double Zahl, int Stellen)
        {
                double v[] = { 1, 10, 1e2, 1e3, 1e4, 1e5, 1e6, 1e7, 1e8 };  // mgl. verlängern
                return floor(Zahl * v[Stellen] + 0.5) / v[Stellen];
        }

        std::vector<vTransferFunction2DPoint> m_FuncControlPoints;                              
        std::vector<vTransferFunctionWannabeTriangle> m_FuncControlTriangles;   
        unsigned char *m_VolumeHistogram2D;                                                                     
        unsigned int m_VolumeHistogramDisplayList2D;                                            
        unsigned int m_TransferFunction2DHandle;                                                        
        unsigned int m_Histogram2DTextureHandle;                                                        
        bool m_hasChanged;                                                                                                      
        QVolRendCanvas * m_MainCanvas;                                                                          
        VFramebufferObject * m_FramebufferObject;                                                       
};

#endif //__v_transfer_function_h

---