[wxWidgets.git] / src / generic / plot.cpp

/////////////////////////////////////////////////////////////////////////////
// Name:        plot.cpp
// Purpose:     wxPlotWindow
// Author:      Robert Roebling
// Modified by:
// Created:     12/01/2000
// RCS-ID:      $Id$
// Copyright:   (c) Robert Roebling
// Licence:     wxWindows license
/////////////////////////////////////////////////////////////////////////////

#ifdef __GNUG__
#pragma implementation "plot.h"
#endif

// For compilers that support precompilation, includes "wx.h".
#include "wx/wxprec.h"

#ifdef __BORLANDC__
#pragma hdrstop
#endif

#ifndef WX_PRECOMP
#include "wx/object.h"
#include "wx/font.h"
#include "wx/colour.h"
#include "wx/settings.h"
#include "wx/sizer.h"
#include "wx/log.h"
#include "wx/intl.h"
#include "wx/dcclient.h"
#endif

#include "wx/generic/plot.h"

#include <math.h>

//-----------------------------------------------------------------------------
// wxPlotCurve
//-----------------------------------------------------------------------------

IMPLEMENT_ABSTRACT_CLASS(wxPlotCurve, wxObject)

wxPlotCurve::wxPlotCurve( int offsetY, double startY, double endY )
{
    m_offsetY = offsetY;
    m_startY = startY;
    m_endY = endY;
}

//-----------------------------------------------------------------------------
// wxPlotArea
//-----------------------------------------------------------------------------

IMPLEMENT_DYNAMIC_CLASS(wxPlotArea, wxWindow)

BEGIN_EVENT_TABLE(wxPlotArea, wxWindow)
  EVT_PAINT(        wxPlotArea::OnPaint)
  EVT_LEFT_DOWN(    wxPlotArea::OnMouse)
END_EVENT_TABLE()

wxPlotArea::wxPlotArea( wxPlotWindow *parent )
        : wxWindow( parent, -1, wxDefaultPosition, wxDefaultSize, wxSIMPLE_BORDER, "plotarea" )
{
    m_owner = parent;

    SetBackgroundColour( *wxWHITE );
}

void wxPlotArea::OnMouse( wxMouseEvent &event )
{
    int client_width;
    int client_height;
    GetClientSize( &client_width, &client_height);
    int view_x;
    int view_y;
    m_owner->GetViewStart( &view_x, &view_y );
    view_x *= 10;
    view_y *= 10;
    
    int x = event.GetX();
    int y = event.GetY();
    x += view_x;
    y += view_y;
    
    wxNode *node = m_owner->m_curves.First();
    while (node)
    {
        wxPlotCurve *curve = (wxPlotCurve*)node->Data();
            
        double double_client_height = (double)client_height;
        double range = curve->GetEndY() - curve->GetStartY();
        double end = curve->GetEndY();
        wxCoord offset_y = curve->GetOffsetY();
            
        double dy = (end - curve->GetY( x )) / range;
        wxCoord curve_y = (wxCoord)(dy * double_client_height) - offset_y - 1;
                
        if ((y-curve_y < 4) && (y-curve_y > -4))
        {
            m_owner->SetCurrent( curve );
            return;
        }
            
        node = node->Next();
    }
}

void wxPlotArea::OnPaint( wxPaintEvent &WXUNUSED(event) )
{
    int client_width;
    int client_height;
    GetClientSize( &client_width, &client_height);
    int view_x;
    int view_y;
    m_owner->GetViewStart( &view_x, &view_y );
    view_x *= 10;
    view_y *= 10;

    wxPaintDC dc( this );
    m_owner->PrepareDC( dc );

    wxRegionIterator upd( GetUpdateRegion() );
    
    while (upd)
    {
        int update_x = upd.GetX();
        int update_y = upd.GetY();
        int update_width = upd.GetWidth();
        
        update_x += view_x;
        update_y += view_y;
        
/*
        if (m_owner->m_current)
        {
            dc.SetPen( *wxLIGHT_GREY_PEN );
            int base_line = client_height - m_owner->m_current->GetOffsetY();
            dc.DrawLine( update_x-1, base_line-1, update_x+update_width+2, base_line-1 );
        }
*/
        
        wxNode *node = m_owner->m_curves.First();
        while (node)
        {
            wxPlotCurve *curve = (wxPlotCurve*)node->Data();
            
            if (curve == m_owner->GetCurrent())
                dc.SetPen( *wxBLACK_PEN );
            else
                dc.SetPen( *wxLIGHT_GREY_PEN );

            int start_x = wxMax( update_x-1, curve->GetStartX() );
            int end_x = wxMin( update_x+update_width+2, curve->GetEndX() );
            
            double double_client_height = (double)client_height;
            double range = curve->GetEndY() - curve->GetStartY();
            double end = curve->GetEndY();
            wxCoord offset_y = curve->GetOffsetY();
            
            wxCoord y=0,last_y=0;
            for (int x = start_x; x < end_x; x++)
            {
                double dy = (end - curve->GetY( x )) / range;
                y = (wxCoord)(dy * double_client_height) - offset_y - 1;
            
                if (x != start_x)
                    dc.DrawLine( x-1, last_y, x, y );
            
                last_y = y;
            }
            node = node->Next();
        }
        upd ++;
    }
}

//-----------------------------------------------------------------------------
// wxPlotWindow
//-----------------------------------------------------------------------------

#define  ID_ENLARGE_100   1000
#define  ID_ENLARGE_50    1001
#define  ID_SHRINK_33     1002
#define  ID_SHRINK_50     1003

#define  ID_MOVE_UP       1006
#define  ID_MOVE_DOWN     1007


IMPLEMENT_DYNAMIC_CLASS(wxPlotWindow, wxScrolledWindow)

BEGIN_EVENT_TABLE(wxPlotWindow, wxScrolledWindow)
  EVT_PAINT(        wxPlotWindow::OnPaint)
END_EVENT_TABLE()

wxPlotWindow::wxPlotWindow( wxWindow *parent, wxWindowID id, const wxPoint &pos, const wxSize &size, int flag )
        : wxScrolledWindow( parent, id, pos, size, flag, "plotcanvas" )
{
    m_area = new wxPlotArea( this );
    
    wxBoxSizer *mainsizer = new wxBoxSizer( wxHORIZONTAL );
    
    wxBoxSizer *buttonlist = new wxBoxSizer( wxVERTICAL );
    buttonlist->Add( new wxButton( this, ID_ENLARGE_100, _("+ 100%") ), 0, wxEXPAND|wxALL, 5 );
    buttonlist->Add( new wxButton( this, ID_ENLARGE_50, _("+ 50%") ), 0, wxEXPAND|wxALL, 5 );
    buttonlist->Add( new wxButton( this, ID_SHRINK_33, _("- 33%") ), 0, wxEXPAND|wxALL, 5 );
    buttonlist->Add( new wxButton( this, ID_SHRINK_50, _("- 50%") ), 0, wxEXPAND|wxALL, 5 );
    buttonlist->Add( 20,20, 0 );
    buttonlist->Add( new wxButton( this, ID_MOVE_UP, _("Up") ), 0, wxEXPAND|wxALL, 5 );
    buttonlist->Add( new wxButton( this, ID_MOVE_DOWN, _("Down") ), 0, wxEXPAND|wxALL, 5 );
    buttonlist->Add( 20,20, 1 );
    
    mainsizer->Add( buttonlist, 0, wxEXPAND );
    
    mainsizer->Add( m_area, 1, wxEXPAND|wxLEFT, 50 );
    
    SetAutoLayout( TRUE );
    SetSizer( mainsizer );

    SetTargetWindow( m_area );

    SetBackgroundColour( *wxWHITE );
    
    m_current = (wxPlotCurve*) NULL;
}

wxPlotWindow::~wxPlotWindow()
{
}

void wxPlotWindow::Add( wxPlotCurve *curve )
{
    m_curves.Append( curve );
    if (!m_current) m_current = curve;
}

size_t wxPlotWindow::GetCount()
{
    return m_curves.GetCount();
}

wxPlotCurve *wxPlotWindow::GetAt( size_t n )
{
    wxNode *node = m_curves.Nth( n );
    if (!node)
        return (wxPlotCurve*) NULL;
        
    return (wxPlotCurve*) node->Data();
}

void wxPlotWindow::SetCurrent( wxPlotCurve* current )
{
    m_current = current;
    m_area->Refresh( FALSE );
    
    wxPoint pos( m_area->GetPosition() );
    
    int client_width;
    int client_height;
    GetClientSize( &client_width, &client_height);
    wxRect rect(pos.x-45,0,45,client_height);
    Refresh(TRUE,&rect);
}

wxPlotCurve *wxPlotWindow::GetCurrent()
{
    return m_current;
}

void wxPlotWindow::OnPaint( wxPaintEvent &WXUNUSED(event) )
{
    wxPaintDC dc( this );
    
    if (!m_current) return;
    
    int client_width;
    int client_height;
    GetClientSize( &client_width, &client_height);

    dc.SetPen( *wxBLACK_PEN );
    
    wxPoint pos( m_area->GetPosition() );
    
    double range = m_current->GetEndY() - m_current->GetStartY();
    double offset = ((double) m_current->GetOffsetY() / (double)client_height ) * range;
    double start = m_current->GetStartY() - offset;
    double end = m_current->GetEndY() - offset;
    int int_log_range = (int)floor( log10( range ) );
    double step = 1.0;
    if (int_log_range > 0)
    {
        for (int i = 0; i < int_log_range; i++)
           step *= 10; 
    }
    if (int_log_range < 0)
    {
        for (int i = 0; i < -int_log_range; i++)
           step /= 10; 
    }
    double lower = ceil(start / step) * step;
    double upper = floor(end / step) * step;
    int steps = (int)ceil((upper-lower)/step);
    if (steps < 3)
    {
        step /= 2;
        if (lower-step > start) lower -= step;
    }
    
    double current = lower;
    while (current < upper+(step/2))
    {
        int y = (int)((m_current->GetEndY()-current) / range * (double)client_height) - 1;
        y -= m_current->GetOffsetY();
        if ((y > 10) && (y < client_height-7))
        {
            dc.DrawLine( pos.x-15, y, pos.x-7, y );
            wxString label;
            label.Printf( "%.1f", current );
            dc.DrawText( label, pos.x-45, y-7 );
        }

        current += step;
    }
    
    dc.DrawLine( pos.x-15, 5, pos.x-15, client_height-5 );
    dc.DrawLine( pos.x-19, 9, pos.x-15, 5 );
    dc.DrawLine( pos.x-10, 10, pos.x-15, 5 );
    
}
Commit	Line	Data
981b2508 RR	1	/////////////////////////////////////////////////////////////////////////////
	2	// Name: plot.cpp
	3	// Purpose: wxPlotWindow
	4	// Author: Robert Roebling
	5	// Modified by:
	6	// Created: 12/01/2000
	7	// RCS-ID: $Id$
	8	// Copyright: (c) Robert Roebling
	9	// Licence: wxWindows license
	10	/////////////////////////////////////////////////////////////////////////////
	11
	12	#ifdef __GNUG__
	13	#pragma implementation "plot.h"
	14	#endif
	15
	16	// For compilers that support precompilation, includes "wx.h".
	17	#include "wx/wxprec.h"
	18
	19	#ifdef __BORLANDC__
	20	#pragma hdrstop
	21	#endif
	22
	23	#ifndef WX_PRECOMP
	24	#include "wx/object.h"
	25	#include "wx/font.h"
	26	#include "wx/colour.h"
	27	#include "wx/settings.h"
	28	#include "wx/sizer.h"
	29	#include "wx/log.h"
	30	#include "wx/intl.h"
	31	#include "wx/dcclient.h"
	32	#endif
	33
	34	#include "wx/generic/plot.h"
	35
846e1424 RR	36	#include <math.h>
846e1424 RR	37
981b2508 RR	38	//-----------------------------------------------------------------------------
	39	// wxPlotCurve
	40	//-----------------------------------------------------------------------------
	41
	42	IMPLEMENT_ABSTRACT_CLASS(wxPlotCurve, wxObject)
	43
846e1424	44	wxPlotCurve::wxPlotCurve( int offsetY, double startY, double endY )
981b2508 RR	45	{
981b2508 RR	46	m_offsetY = offsetY;
846e1424 RR	47	m_startY = startY;
846e1424 RR	48	m_endY = endY;
981b2508 RR	49	}
	50
	51	//-----------------------------------------------------------------------------
	52	// wxPlotArea
	53	//-----------------------------------------------------------------------------
	54
	55	IMPLEMENT_DYNAMIC_CLASS(wxPlotArea, wxWindow)
	56
	57	BEGIN_EVENT_TABLE(wxPlotArea, wxWindow)
	58	EVT_PAINT( wxPlotArea::OnPaint)
	59	EVT_LEFT_DOWN( wxPlotArea::OnMouse)
	60	END_EVENT_TABLE()
	61
	62	wxPlotArea::wxPlotArea( wxPlotWindow *parent )
	63	: wxWindow( parent, -1, wxDefaultPosition, wxDefaultSize, wxSIMPLE_BORDER, "plotarea" )
	64	{
	65	m_owner = parent;
	66
	67	SetBackgroundColour( *wxWHITE );
	68	}
	69
	70	void wxPlotArea::OnMouse( wxMouseEvent &event )
	71	{
	72	int client_width;
	73	int client_height;
	74	GetClientSize( &client_width, &client_height);
	75	int view_x;
	76	int view_y;
	77	m_owner->GetViewStart( &view_x, &view_y );
	78	view_x *= 10;
	79	view_y *= 10;
	80
	81	int x = event.GetX();
	82	int y = event.GetY();
	83	x += view_x;
	84	y += view_y;
	85
	86	wxNode *node = m_owner->m_curves.First();
	87	while (node)
	88	{
	89	wxPlotCurve curve = (wxPlotCurve)node->Data();
	90
846e1424 RR	91	double double_client_height = (double)client_height;
	92	double range = curve->GetEndY() - curve->GetStartY();
	93	double end = curve->GetEndY();
	94	wxCoord offset_y = curve->GetOffsetY();
	95
	96	double dy = (end - curve->GetY( x )) / range;
	97	wxCoord curve_y = (wxCoord)(dy * double_client_height) - offset_y - 1;
	98
981b2508 RR	99	if ((y-curve_y < 4) && (y-curve_y > -4))
	100	{
	101	m_owner->SetCurrent( curve );
	102	return;
	103	}
	104
	105	node = node->Next();
	106	}
	107	}
	108
	109	void wxPlotArea::OnPaint( wxPaintEvent &WXUNUSED(event) )
	110	{
	111	int client_width;
	112	int client_height;
	113	GetClientSize( &client_width, &client_height);
	114	int view_x;
	115	int view_y;
	116	m_owner->GetViewStart( &view_x, &view_y );
	117	view_x *= 10;
	118	view_y *= 10;
	119
	120	wxPaintDC dc( this );
	121	m_owner->PrepareDC( dc );
	122
	123	wxRegionIterator upd( GetUpdateRegion() );
	124
	125	while (upd)
	126	{
	127	int update_x = upd.GetX();
	128	int update_y = upd.GetY();
	129	int update_width = upd.GetWidth();
	130
	131	update_x += view_x;
	132	update_y += view_y;
	133
846e1424	134	/*
981b2508 RR	135	if (m_owner->m_current)
	136	{
	137	dc.SetPen( *wxLIGHT_GREY_PEN );
	138	int base_line = client_height - m_owner->m_current->GetOffsetY();
	139	dc.DrawLine( update_x-1, base_line-1, update_x+update_width+2, base_line-1 );
	140	}
846e1424	141	*/
981b2508 RR	142
	143	wxNode *node = m_owner->m_curves.First();
	144	while (node)
	145	{
	146	wxPlotCurve curve = (wxPlotCurve)node->Data();
	147
	148	if (curve == m_owner->GetCurrent())
	149	dc.SetPen( *wxBLACK_PEN );
	150	else
	151	dc.SetPen( *wxLIGHT_GREY_PEN );
981b2508 RR	152
	153	int start_x = wxMax( update_x-1, curve->GetStartX() );
	154	int end_x = wxMin( update_x+update_width+2, curve->GetEndX() );
	155
846e1424 RR	156	double double_client_height = (double)client_height;
	157	double range = curve->GetEndY() - curve->GetStartY();
	158	double end = curve->GetEndY();
	159	wxCoord offset_y = curve->GetOffsetY();
	160
981b2508 RR	161	wxCoord y=0,last_y=0;
	162	for (int x = start_x; x < end_x; x++)
	163	{
846e1424 RR	164	double dy = (end - curve->GetY( x )) / range;
846e1424 RR	165	y = (wxCoord)(dy * double_client_height) - offset_y - 1;
981b2508 RR	166
	167	if (x != start_x)
	168	dc.DrawLine( x-1, last_y, x, y );
	169
	170	last_y = y;
	171	}
	172	node = node->Next();
	173	}
	174	upd ++;
	175	}
	176	}
	177
	178	//-----------------------------------------------------------------------------
	179	// wxPlotWindow
	180	//-----------------------------------------------------------------------------
	181
	182	#define ID_ENLARGE_100 1000
	183	#define ID_ENLARGE_50 1001
	184	#define ID_SHRINK_33 1002
	185	#define ID_SHRINK_50 1003
	186
	187	#define ID_MOVE_UP 1006
	188	#define ID_MOVE_DOWN 1007
	189
	190
	191	IMPLEMENT_DYNAMIC_CLASS(wxPlotWindow, wxScrolledWindow)
	192
	193	BEGIN_EVENT_TABLE(wxPlotWindow, wxScrolledWindow)
	194	EVT_PAINT( wxPlotWindow::OnPaint)
	195	END_EVENT_TABLE()
	196
	197	wxPlotWindow::wxPlotWindow( wxWindow *parent, wxWindowID id, const wxPoint &pos, const wxSize &size, int flag )
	198	: wxScrolledWindow( parent, id, pos, size, flag, "plotcanvas" )
	199	{
	200	m_area = new wxPlotArea( this );
	201
	202	wxBoxSizer *mainsizer = new wxBoxSizer( wxHORIZONTAL );
	203
	204	wxBoxSizer *buttonlist = new wxBoxSizer( wxVERTICAL );
	205	buttonlist->Add( new wxButton( this, ID_ENLARGE_100, _("+ 100%") ), 0, wxEXPAND\|wxALL, 5 );
	206	buttonlist->Add( new wxButton( this, ID_ENLARGE_50, _("+ 50%") ), 0, wxEXPAND\|wxALL, 5 );
	207	buttonlist->Add( new wxButton( this, ID_SHRINK_33, _("- 33%") ), 0, wxEXPAND\|wxALL, 5 );
	208	buttonlist->Add( new wxButton( this, ID_SHRINK_50, _("- 50%") ), 0, wxEXPAND\|wxALL, 5 );
	209	buttonlist->Add( 20,20, 0 );
	210	buttonlist->Add( new wxButton( this, ID_MOVE_UP, _("Up") ), 0, wxEXPAND\|wxALL, 5 );
	211	buttonlist->Add( new wxButton( this, ID_MOVE_DOWN, _("Down") ), 0, wxEXPAND\|wxALL, 5 );
	212	buttonlist->Add( 20,20, 1 );
	213
	214	mainsizer->Add( buttonlist, 0, wxEXPAND );
	215
	216	mainsizer->Add( m_area, 1, wxEXPAND\|wxLEFT, 50 );
	217
	218	SetAutoLayout( TRUE );
	219	SetSizer( mainsizer );
	220
	221	SetTargetWindow( m_area );
	222
	223	SetBackgroundColour( *wxWHITE );
	224
	225	m_current = (wxPlotCurve*) NULL;
	226	}
	227
	228	wxPlotWindow::~wxPlotWindow()
	229	{
230	}
231
232	void wxPlotWindow::Add( wxPlotCurve *curve )
233	{
234	m_curves.Append( curve );
235	if (!m_current) m_current = curve;
236	}
237
238	size_t wxPlotWindow::GetCount()
239	{
240	return m_curves.GetCount();
241	}
242
243	wxPlotCurve *wxPlotWindow::GetAt( size_t n )
244	{
245	wxNode *node = m_curves.Nth( n );
246	if (!node)
247	return (wxPlotCurve*) NULL;
248
249	return (wxPlotCurve*) node->Data();
250	}
251
252	void wxPlotWindow::SetCurrent( wxPlotCurve* current )
253	{
254	m_current = current;
846e1424	255	m_area->Refresh( FALSE );
981b2508 RR	256
	257	wxPoint pos( m_area->GetPosition() );
	258
	259	int client_width;
	260	int client_height;
	261	GetClientSize( &client_width, &client_height);
846e1424	262	wxRect rect(pos.x-45,0,45,client_height);
981b2508 RR	263	Refresh(TRUE,&rect);
	264	}
	265
	266	wxPlotCurve *wxPlotWindow::GetCurrent()
	267	{
	268	return m_current;
	269	}
	270
	271	void wxPlotWindow::OnPaint( wxPaintEvent &WXUNUSED(event) )
	272	{
	273	wxPaintDC dc( this );
	274
	275	if (!m_current) return;
	276
	277	int client_width;
	278	int client_height;
	279	GetClientSize( &client_width, &client_height);
	280
	281	dc.SetPen( *wxBLACK_PEN );
	282
	283	wxPoint pos( m_area->GetPosition() );
	284
846e1424 RR	285	double range = m_current->GetEndY() - m_current->GetStartY();
	286	double offset = ((double) m_current->GetOffsetY() / (double)client_height ) * range;
	287	double start = m_current->GetStartY() - offset;
	288	double end = m_current->GetEndY() - offset;
	289	int int_log_range = (int)floor( log10( range ) );
	290	double step = 1.0;
	291	if (int_log_range > 0)
	292	{
	293	for (int i = 0; i < int_log_range; i++)
	294	step *= 10;
	295	}
	296	if (int_log_range < 0)
	297	{
	298	for (int i = 0; i < -int_log_range; i++)
	299	step /= 10;
	300	}
	301	double lower = ceil(start / step) * step;
	302	double upper = floor(end / step) * step;
	303	int steps = (int)ceil((upper-lower)/step);
	304	if (steps < 3)
	305	{
	306	step /= 2;
	307	if (lower-step > start) lower -= step;
	308	}
	309
	310	double current = lower;
	311	while (current < upper+(step/2))
	312	{
	313	int y = (int)((m_current->GetEndY()-current) / range * (double)client_height) - 1;
	314	y -= m_current->GetOffsetY();
	315	if ((y > 10) && (y < client_height-7))
	316	{
	317	dc.DrawLine( pos.x-15, y, pos.x-7, y );
	318	wxString label;
	319	label.Printf( "%.1f", current );
	320	dc.DrawText( label, pos.x-45, y-7 );
	321	}
	322
	323	current += step;
	324	}
	325
981b2508 RR	326	dc.DrawLine( pos.x-15, 5, pos.x-15, client_height-5 );
	327	dc.DrawLine( pos.x-19, 9, pos.x-15, 5 );
	328	dc.DrawLine( pos.x-10, 10, pos.x-15, 5 );
	329
981b2508 RR	330	}
981b2508 RR	331