volume_canvas.h

Go to the documentation of this file.

#ifndef VOLUME_CANVAS_H
#define VOLUME_CANVAS_H

#include <glew.h>
#include <wx/glcanvas.h>
#include <memory>
#include "volprogram.h"
#include "histogramdata.h"

class TransferFunctionSampler {
public:
    struct Color {
        float r, g, b, a;
    };


    virtual Color MapDensity(float density) = 0;
};

enum Viewmode {
    VM_3D,
    VM_SLICE_X,
    VM_SLICE_Y,
    VM_SLICE_Z
};

class VolumeCanvas: public wxGLCanvas {
public:
    VolumeCanvas(wxWindow *parent, wxWindowID id = wxID_ANY);

    ~VolumeCanvas()
    {
        SetCurrent();
        delete m_volprog;
    }

    void OpenFile(const char *filename);

    Viewmode GetViewmode() const
    {
        return m_viewmode;
    }

    void SetViewmode(Viewmode viewmode)
    {
        m_viewmode = viewmode;
    }

    int GetSizeX() const
    {
        return m_sizex;
    }

    int GetSizeY() const
    {
        return m_sizey;
    }

    int GetSizeZ() const
    {
        return m_sizez;
    }

    void SetSlice(int value)
    {
        m_slice = value;
    }

    int GetSlice() const
    {
        return m_slice;
    }

    void Set3dOrientation(float rx, float ry, float rz)
    {
        m_rotx = rx;
        m_roty = ry;
        m_rotz = rz;
    }

    void Get3dOrientation(float* rx, float* ry, float* rz)
    {
        *rx = m_rotx;
        *ry = m_roty;
        *rz = m_rotz;
    }

    const HistogramData *GetHistogramData() const
    {
        return m_histogramdata.get();
    }

    void UpdateTransferFunction(TransferFunctionSampler *sampler);

    void RenderToFile(const wxString & file_name, int width, int height);

protected:

    void OnPaint(wxPaintEvent &);

    void OnSize(wxSizeEvent &);

    void OnEraseBackground(wxEraseEvent &);


private:
    DECLARE_EVENT_TABLE()

    
    enum GLStatus {
        GLS_INIT_PENDING,
        GLS_INIT_FAILED,
        GLS_OK
    };

    GLStatus m_glstatus; 
    GLuint m_volume_texture; 
    GLuint m_transfer_texture; 
    VolProgram *m_volprog; 

    GLint m_viewport_width; 
    GLint m_viewport_height; 

    Viewmode m_viewmode;

    int m_sizex;
    int m_sizey;
    int m_sizez;
    int m_slice;

    float m_rotx, m_roty, m_rotz;

    std::auto_ptr<HistogramData> m_histogramdata;

    void Render() const;

    void Render3dView() const;

    void RenderSliceView() const;

    bool InitGL();

    bool CheckGL();
};

#endif

---