RTVolumeRendering/Source/TransferFunctionWidget.h

Go to the documentation of this file.

#ifndef TRANSFERFUNCTIONWIDGET_H
#define TRANSFERFUNCTIONWIDGET_H

#include <wx/wx.h>
#include <vector>
#include <fstream>
#include "Color4.h"
#include "MathCommon.h"
#include "TransferTexture.h"

class TransferFunctionPlot;

class TransferPlotChangeListner
{
public:
        
        virtual void TransferPlotChanged(TransferFunctionPlot *plot) = 0;
};

class TransferFunctionPlot : public wxWindow
{
public:
        
        TransferFunctionPlot(wxWindow *parent);
        
        ~TransferFunctionPlot();
        
        enum PlotMode
        {
                MOVE = 0,
                ADD,
                REMOVE,
                
                NUMBER_OF_MODES
        };
        
        void SetPlotMode(PlotMode m);

        void ResetPlot();
        
        void DelayRealtimeUpdate(bool v) { delayUpdates = v; }
        
        void DrawAll(bool b);
        
        void DrawHistogram(bool b);
        
        void MakeRedActive();
        
        void MakeGreenActive();
        
        void MakeBlueActive();
        
        void MakeAlphaActive();
        
        void OnPaint(wxPaintEvent &e);
        
        void MouseMove(wxMouseEvent &m);
        
        void MouseLeftUp(wxMouseEvent &m);
        
        void MouseLeftDown(wxMouseEvent &m);
        
        void MouseWheel(wxMouseEvent &m);

        void MouseLeave(wxMouseEvent &m);
        
        void SetListner(TransferPlotChangeListner *l);
        
        void UpdateTexture(TransferTexture *tex);

        void Load(const char *file);

        void Save(const char *file);

        void RefreshListner();

        void SetHistogram(const unsigned int *h);
        
protected:
        
        void DrawPlot(wxPaintDC &dc, const std::vector<PlotPoint> &plot, const Color4f &color) const;
        
        void DrawPlotPoints(wxPaintDC &dc, const std::vector<PlotPoint> &plot, const Color4f &color) const;
        
        void PlotPointToScreen(const PlotPoint &p, int &x, int &y) const;
        
        void ScreenToPlotPoint(int x, int y, PlotPoint &p) const;
        
        int PlotDataValueToXCoord(int dataValue) const;
        
        int PlotValueToYCoord(float value) const;

        bool SelectPoint(wxCoord x, wxCoord y);
        
        int SelectPoint(wxCoord x, wxCoord y, const std::vector<PlotPoint> &plot) const;
        
        void AddPoint(wxCoord x, wxCoord y);
        
        void RemovePoint(wxCoord x, wxCoord y);

        bool LoadPlot(std::ifstream &file, std::vector<PlotPoint> &plot, int n);

        void SavePlot(std::ofstream &file, const std::vector<PlotPoint> &plot);
        
        std::vector<PlotPoint> redPlot;
        std::vector<PlotPoint> greenPlot;
        std::vector<PlotPoint> bluePlot;
        std::vector<PlotPoint> alphaPlot;
        
        bool delayUpdates;
        bool drawHistogram;
        bool drawAll;
        
        int currDataValue;
        float currValue;
        
        PlotMode mode;
        
        int selectedPoint; // index of the selected point
        std::vector<PlotPoint> *activePlot; // the plot that can be interacted with
        Color4f activePlotColor;
        bool mouseDown;
        
        wxBitmap bgGrid;
        wxBitmap histogram;
        
        TransferPlotChangeListner *listner;
        
        DECLARE_EVENT_TABLE()
};

class TransferFunctionWidget : public wxWindow
{
public:
        
        TransferFunctionWidget(wxWindow *parent);
        
        ~TransferFunctionWidget();
        
        void DelayRealtimeUpdate(bool v) { plot->DelayRealtimeUpdate(v); }
        
        void OnDrawAllClick(wxCommandEvent &e);
        
        void OnDrawHistogramClick(wxCommandEvent &e);
        
        void OnActivePlotSelection(wxCommandEvent &e);
        
        void OnPlotModeSelection(wxCommandEvent &e);

        void OnSaveClick(wxCommandEvent &e);
        
        void OnLoadClick(wxCommandEvent &e);
        
        void OnRefreshClick(wxCommandEvent &e);

        void OnResetClick(wxCommandEvent &e);

        void SetHistogram(const unsigned int *h) { plot->SetHistogram(h); }
        
        void SetListner(TransferPlotChangeListner *l) { plot->SetListner(l); }
        
protected:
        
        TransferFunctionPlot *plot;
        
        wxCheckBox *drawAll;
        wxCheckBox *drawHistogram;
        
        DECLARE_EVENT_TABLE()
};              

#endif

---