axisplot.h

// Name:    axisplot.h
// Purpose: axis plot declaration
// Author:  Moskvichev Andrey V.
// Created: 2008/11/07
// Copyright:   (c) 2008-2010 Moskvichev Andrey V.
// Licence: wxWidgets licence

#ifndef AXISPLOT_H_
#define AXISPLOT_H_

#include <wx/plot.h>
#include <wx/axis/axis.h>

#include <wx/areadraw.h>
#include <wx/legend.h>
#include <wx/marker.h>
#include <wx/crosshair.h>
#include <wx/chartpanel.h>

#include <wx/dynarray.h>


class WXDLLIMPEXP_FREECHART DataAxisLink
{
public:
    DataAxisLink(Dataset *dataset, Axis *axis)
    {
        m_dataset = dataset;
        m_axis = axis;
    }

    DataAxisLink(const DataAxisLink &o)
    {
        m_dataset = o.m_dataset;
        m_axis = o.m_axis;
    }

    ~DataAxisLink()
    {
    }

    Dataset *m_dataset;
    Axis *m_axis;
};

WX_DECLARE_USER_EXPORTED_OBJARRAY(DataAxisLink, DataAxisLinkArray, WXDLLIMPEXP_FREECHART);

class WXDLLIMPEXP_FREECHART AxisPlot : public Plot,
    public DrawObserver, public DatasetObserver, public AxisObserver,
    public ChartPanelObserver
{
public:
    AxisPlot();
    virtual ~AxisPlot();

    void AddAxis(Axis *axis);

    void AddDataset(Dataset *dataset);

    void RemoveDataset(Dataset *dataset);

    void RemoveDataset(size_t index);

    void AddObjects(Dataset *dataset, Axis *verticalAxis, Axis *horizontalAxis);

    size_t GetDatasetCount();

    Dataset *GetDataset(size_t index);

    void LinkDataHorizontalAxis(size_t nData, size_t nAxis);

    void LinkDataVerticalAxis(size_t nData, size_t nAxis);

    Axis *GetDatasetAxis(Dataset *dataset, size_t index, bool vertical);

    Axis *GetDatasetAxis(Dataset *dataset, bool vertical);

    Axis *GetDatasetVerticalAxis(Dataset *dataset)
    {
        return GetDatasetAxis(dataset, true);
    }

    Axis *GetDatasetHorizontalAxis(Dataset *dataset)
    {
        return GetDatasetAxis(dataset, false);
    }

    Dataset *GetAxisDataset(Axis *axis, size_t index)
    {
        return axis->GetDataset(index);
    }

    void SetDrawGrid(bool drawGridVertical, bool drawGridHorizontal);

    void SetDataBackground(AreaDraw *dataBackground);

    void SetLegend(Legend *legend);

    void SetCrosshair(Crosshair *crosshair);

    bool ToDataCoords(size_t nData, wxDC &dc, wxRect rc, wxCoord gx, wxCoord gy, double *x, double *y);

    //
    // DrawObserver
    //
    virtual void NeedRedraw(DrawObject *obj);

    //
    // DatasetObserver
    //
    virtual void DatasetChanged(Dataset *dataset);

    //
    // AxisObserver
    //
    virtual void AxisChanged(Axis *axis);

    virtual void BoundsChanged(Axis *axis);

    //
    // ChartPanelObserver
    //
    virtual void ChartMouseDown(wxPoint &pt, int key);

protected:
    //
    // Methods to be implemented by derivative classes
    //

    virtual bool AcceptAxis(Axis *axis) = 0;

    virtual bool AcceptDataset(Dataset *dataset) = 0;

    virtual void DrawDatasets(wxDC &dc, wxRect rc) = 0;

    wxCoord GetAxesExtent(wxDC &dc, AxisArray *axes);

private:
    //
    // Plot
    //
    virtual void DrawData(wxDC &dc, wxRect rc);

    virtual bool HasData();

    virtual void ChartPanelChanged(wxChartPanel *oldPanel, wxChartPanel *newPanel);


    void UpdateAxis(Dataset *dataset = NULL);

    //
    // Draw functions
    //

    void CalcDataArea(wxDC &dc, wxRect rc, wxRect &rcData, wxRect &rcLegend);

    void DrawAxes(wxDC &dc, wxRect &rc, wxRect rcData);

    void DrawAxesArray(wxDC &dc, wxRect rc, AxisArray *axes, bool vertical);


    void DrawGridLines(wxDC &dc, wxRect rcData);

    void DrawMarkers(wxDC &dc, wxRect rcData);

    void DrawDataArea(wxDC &dc, wxRect rcData);

    void DrawLegend(wxDC &dc, wxRect rcLegend);

    bool m_drawGridVertical;
    bool m_drawGridHorizontal;

    AxisArray m_leftAxes;
    AxisArray m_rightAxes;
    AxisArray m_topAxes;
    AxisArray m_bottomAxes;

    AxisArray m_horizontalAxes;
    AxisArray m_verticalAxes;

    DataAxisLinkArray m_links;

    DatasetArray m_datasets;
    AreaDraw *m_dataBackground; // data area background

    wxCoord m_legendPlotGap; // distance between plot and legend

    Legend *m_legend;

    Crosshair *m_crosshair;
};

#endif /*AXISPLOT_H_*/