mitkColorTransferFunction.h

/*=========================================================================

  Program:   3DMed
  Date:      $Date: 2014-02-25 18:30:00 +0800 $
  Version:   $Version: 4.6.0 $
  Copyright: MIPG, Institute of Automation, Chinese Academy of Sciences

=========================================================================*/


#ifndef __mitkColorTransferFunction_h
#define __mitkColorTransferFunction_h

#include "mitkObject.h"
#include "mitkVisualizationIncludes.h"

struct mitkColorPoint
{
    int x;
    float r;
    float g;
    float b;
};

class mitkColorPointArray;

class MITK_VISUALIZATION_API mitkColorTransferFunction : public mitkObject
{
public:
    MITK_TYPE(mitkColorTransferFunction,mitkObject)

    mitkColorTransferFunction();        
    virtual void PrintSelf(ostream& os);    
    
    void AddPoint(int x, float rColor, float gColor, float bColor);

    void RemovePoint(int x);

    void RemoveAllPoints();

    int GetPointsCount();

    int GetDataValueAtPoint(int pointIndex);

    float GetRedColorAtPoint(int pointIndex);

    float GetGreenColorAtPoint(int pointIndex);

    float GetBlueColorAtPoint(int pointIndex);

    void SetMax(int xMax, float rMax, float gMax, float bMax);  


    int  GetMaxX() {return m_MaxX;}

    float GetRedValue(int x);

    float GetGreenValue(int x);

    float GetBlueValue(int x);  

    float* GetRData() {return m_RData;}

    float* GetGData() {return m_GData;}

    float* GetBData() {return m_BData;}

    bool IsModified() const { return m_Modified; }

    void SetUnmodified() { m_Modified = false; }

protected:  
    int m_MaxX; 
    mitkColorPointArray *m_Points;  
    float *m_RData;
    float *m_GData;
    float *m_BData;
    
    bool m_Modified;

    virtual ~mitkColorTransferFunction();               
    void _buildRFunction(int x1, float r1, int x2, float r2);
    void _buildGFunction(int x1, float g1, int x2, float g2);
    void _buildBFunction(int x1, float b1, int x2, float b2);
    
private:
    mitkColorTransferFunction(const mitkColorTransferFunction&);  
    void operator=(const mitkColorTransferFunction&); 
};

 
    //#define DEFINED_mitkColorTransferFunction
     


#endif

---