trafu_ctrl.h

Go to the documentation of this file.

#ifndef TRAFUCTRL_H
#define TRAFUCTRL_H

#include <wx/control.h>
#include <wx/panel.h>
#include <wx/dcbuffer.h>

#include "trafupoint.h"
#include "trafu_ctrl_event.h"
#include "histogramdata.h"
#include "trafu_data.h"

#include <list>
#include <vector>

//Forward decleration
class TraFuData;

class TransferFunctionCtrlCanvas : public wxControl
{
    DECLARE_DYNAMIC_CLASS(TransferFunctionCtrlCanvas)
    DECLARE_EVENT_TABLE()

public:
    TransferFunctionCtrlCanvas() { Init(); }
    
    TransferFunctionCtrlCanvas(wxWindow* parent,
        wxWindowID id,
        const wxPoint& pos = wxDefaultPosition,
        const wxSize& size = wxDefaultSize,
        long style = 0)
    {
        Init();
        Create(parent, id, pos, size, style);
    }

    ~TransferFunctionCtrlCanvas()
    {
    }

    void SetTraFuDataPointer(TraFuData *data);

    bool Create(wxWindow* parent,
        wxWindowID id,
        const wxPoint& pos = wxDefaultPosition,
        const wxSize& size = wxDefaultSize,
        long style = wxSUNKEN_BORDER);

    wxSize DoGetBestSize() const { return wxSize(650, 200); }
    
    int GetCanvasHeight() { return GetClientRect().GetHeight() - this->m_colour_range_height + 1; }

    int GetCanvasWidth() { return GetClientRect().GetWidth(); }
    
    wxSize GetCanvasSize() const { return GetClientRect().GetSize() - wxSize(0, this->m_colour_range_height); }

    void SetHistogramData(const HistogramData *histogramdata);

private:
    void SetMoving(Mapping_point_iterator_t target, bool holding);

    void Init();

    void NotifyChange();

    void SetPointPosition(Mapping_point_iterator_t point, const wxPoint &pos);

    Mapping_point_iterator_t GetHitPoint(const wxPoint &pos);

    bool PointContains(Mapping_point_iterator_t point, const wxPoint &pos);

    wxPoint GetPosition(const Mapping_point_iterator_t &point)
    {
        return wxPoint(point->m_density * GetCanvasWidth(), (1 - point->m_opacity) * GetCanvasHeight());
    }

    void SetPosition(Mapping_point_iterator_t &point, const wxPoint &new_pos)
    {
        point->m_density = Density(new_pos);
        point->m_opacity = Opacity(new_pos);
    }

    void SetColour(Mapping_point_t &point, const wxColor &new_color)
    {
        point.m_colour = tfRGB(new_color.Red(), new_color.Green(), new_color.Blue());
    }

    wxColor GetColour(Mapping_point_t &point)
    {
        return wxColour(point.m_colour.red, point.m_colour.green, point.m_colour.blue);
    }

    void SetRGB(Mapping_point_iterator_t &point, const tfRGB &colour)
    {
        point->m_colour = colour;
    }

    tfRGB GetRGB(Mapping_point_iterator_t &point)
    {
        return point->m_colour;
    }

    float Density(const wxPoint &pos)
    {
        return static_cast<float>(pos.x) / GetCanvasWidth();
    }

    float Opacity(const wxPoint &pos)
    {
        return 1 - static_cast<float>(pos.y) / GetCanvasHeight();
    }

    void AddNewPoint(const wxPoint &pos);
    
    void AddNewPoint(const wxPoint &pos, const wxColour &colour);

    void DeletePoint(Mapping_point_iterator_t target);

    void DrawPoints(wxDC *dc);

    void DrawHist(wxDC *dc);

    void DrawRange(wxDC *dc);

    int m_colour_range_height; 
    int m_button_radius; 
    TraFuData *m_points; 
    Mapping_point_iterator_t m_active_point; 
    const HistogramData *m_hist_data; 

protected:
    void OnPaint(wxPaintEvent& event);
    void OnMouseEvent(wxMouseEvent& event);
    void OnSize(wxSizeEvent &event);
};

#endif // TRAFUCTRL_H

---