contrib/src/plot/plot.cpp

   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/plot/plot.h"
  35 #include "wx/bmpbuttn.h"
  36 #include "wx/module.h"
  37
  38 #include <math.h>
  39
  40 // ----------------------------------------------------------------------------
  41 // XPMs
  42 // ----------------------------------------------------------------------------
  43
  44 #if !defined(__WXMSW__) && !defined(__WXPM__)
  45     #include "wx/plot/plot_enl.xpm"
  46     #include "wx/plot/plot_shr.xpm"
  47     #include "wx/plot/plot_zin.xpm"
  48     #include "wx/plot/plot_zot.xpm"
  49     #include "wx/plot/plot_up.xpm"
  50     #include "wx/plot/plot_dwn.xpm"
  51 #endif
  52
  53 //----------------------------------------------------------------------------
  54 // event types
  55 //----------------------------------------------------------------------------
  56
  57 const int wxEVT_PLOT_SEL_CHANGING = wxNewEventType();
  58 const int wxEVT_PLOT_SEL_CHANGED = wxNewEventType();
  59 const int wxEVT_PLOT_CLICKED = wxNewEventType();
  60 const int wxEVT_PLOT_DOUBLECLICKED = wxNewEventType();
  61 const int wxEVT_PLOT_ZOOM_IN = wxNewEventType();
  62 const int wxEVT_PLOT_ZOOM_OUT = wxNewEventType();
  63 const int wxEVT_PLOT_VALUE_SEL_CREATING = wxNewEventType();
  64 const int wxEVT_PLOT_VALUE_SEL_CREATED = wxNewEventType();
  65 const int wxEVT_PLOT_VALUE_SEL_CHANGING = wxNewEventType();
  66 const int wxEVT_PLOT_VALUE_SEL_CHANGED = wxNewEventType();
  67 const int wxEVT_PLOT_AREA_SEL_CREATING = wxNewEventType();
  68 const int wxEVT_PLOT_AREA_SEL_CREATED = wxNewEventType();
  69 const int wxEVT_PLOT_AREA_SEL_CHANGING = wxNewEventType();
  70 const int wxEVT_PLOT_AREA_SEL_CHANGED = wxNewEventType();
  71 const int wxEVT_PLOT_BEGIN_X_LABEL_EDIT = wxNewEventType();
  72 const int wxEVT_PLOT_END_X_LABEL_EDIT = wxNewEventType();
  73 const int wxEVT_PLOT_BEGIN_Y_LABEL_EDIT = wxNewEventType();
  74 const int wxEVT_PLOT_END_Y_LABEL_EDIT = wxNewEventType();
  75 const int wxEVT_PLOT_BEGIN_TITLE_EDIT = wxNewEventType();
  76 const int wxEVT_PLOT_END_TITLE_EDIT = wxNewEventType();
  77 const int wxEVT_PLOT_AREA_CREATE = wxNewEventType();
  78
  79 //----------------------------------------------------------------------------
  80 // accessor functions for the bitmaps (may return NULL, check for it!)
  81 //----------------------------------------------------------------------------
  82
  83 static wxBitmap *GetEnlargeBitmap();
  84 static wxBitmap *GetShrinkBitmap();
  85 static wxBitmap *GetZoomInBitmap();
  86 static wxBitmap *GetZoomOutBitmap();
  87 static wxBitmap *GetUpBitmap();
  88 static wxBitmap *GetDownBitmap();
  89
  90 //-----------------------------------------------------------------------------
  91 // consts
  92 //-----------------------------------------------------------------------------
  93
  94 #define wxPLOT_SCROLL_STEP  30
  95
  96 //-----------------------------------------------------------------------------
  97 // wxPlotEvent
  98 //-----------------------------------------------------------------------------
  99
 100 wxPlotEvent::wxPlotEvent( wxEventType commandType, int id )
 101     : wxNotifyEvent( commandType, id )
 102 {
 103     m_curve = (wxPlotCurve*) NULL;
 104     m_zoom = 1.0;
 105     m_position = 0;
 106 }
 107
 108 //-----------------------------------------------------------------------------
 109 // wxPlotCurve
 110 //-----------------------------------------------------------------------------
 111
 112 IMPLEMENT_ABSTRACT_CLASS(wxPlotCurve, wxObject)
 113
 114 wxPlotCurve::wxPlotCurve( int offsetY, double startY, double endY )
 115 {
 116     m_offsetY = offsetY;
 117     m_startY = startY;
 118     m_endY = endY;
 119 }
 120
 121 //-----------------------------------------------------------------------------
 122 // wxPlotOnOffCurve
 123 //-----------------------------------------------------------------------------
 124
 125 IMPLEMENT_CLASS(wxPlotOnOffCurve, wxObject)
 126
 127 #include "wx/arrimpl.cpp"
 128 WX_DEFINE_OBJARRAY(wxArrayPlotOnOff);
 129
 130 wxPlotOnOffCurve::wxPlotOnOffCurve( int offsetY )
 131 {
 132     m_offsetY = offsetY;
 133     m_minX = -1;
 134     m_maxX = -1;
 135 }
 136
 137 void wxPlotOnOffCurve::Add( wxInt32 on, wxInt32 off, void *clientData )
 138 {
 139     wxASSERT_MSG( on > 0, _T("plot index < 0") );
 140     wxASSERT( on <= off );
 141
 142     if (m_minX == -1)
 143         m_minX = on;
 144     if (off > m_maxX)
 145         m_maxX = off;
 146
 147     wxPlotOnOff *v = new wxPlotOnOff;
 148     v->m_on = on;
 149     v->m_off = off;
 150     v->m_clientData = clientData;
 151     m_marks.Add( v );
 152 }
 153
 154 size_t wxPlotOnOffCurve::GetCount()
 155 {
 156     return m_marks.GetCount();
 157 }
 158
 159 wxInt32 wxPlotOnOffCurve::GetOn( size_t index )
 160 {
 161     wxPlotOnOff *v = &m_marks.Item( index );
 162     return v->m_on;
 163 }
 164
 165 wxInt32 wxPlotOnOffCurve::GetOff( size_t index )
 166 {
 167     wxPlotOnOff *v = &m_marks.Item( index );
 168     return v->m_off;
 169 }
 170
 171 void* wxPlotOnOffCurve::GetClientData( size_t index )
 172 {
 173     wxPlotOnOff *v = &m_marks.Item( index );
 174     return v->m_clientData;
 175 }
 176
 177 wxPlotOnOff *wxPlotOnOffCurve::GetAt( size_t index )
 178 {
 179     return &m_marks.Item( index );
 180 }
 181
 182 void wxPlotOnOffCurve::DrawOnLine( wxDC &dc, wxCoord y, wxCoord start, wxCoord end, void *WXUNUSED(clientData) )
 183 {
 184     dc.DrawLine( start, y, start, y-30 );
 185     dc.DrawLine( start, y-30, end, y-30 );
 186     dc.DrawLine( end, y-30, end, y );
 187 }
 188
 189 void wxPlotOnOffCurve::DrawOffLine( wxDC &dc, wxCoord y, wxCoord start, wxCoord end )
 190 {
 191     dc.DrawLine( start, y, end, y );
 192 }
 193
 194 //-----------------------------------------------------------------------------
 195 // wxPlotArea
 196 //-----------------------------------------------------------------------------
 197
 198 IMPLEMENT_DYNAMIC_CLASS(wxPlotArea, wxWindow)
 199
 200 BEGIN_EVENT_TABLE(wxPlotArea, wxWindow)
 201   EVT_PAINT(        wxPlotArea::OnPaint)
 202   EVT_LEFT_DOWN(    wxPlotArea::OnMouse)
 203   EVT_LEFT_DCLICK(  wxPlotArea::OnMouse)
 204 END_EVENT_TABLE()
 205
 206 wxPlotArea::wxPlotArea( wxPlotWindow *parent )
 207         : wxWindow( parent, -1, wxDefaultPosition, wxDefaultSize, wxSIMPLE_BORDER, _T("plotarea") )
 208 {
 209     m_owner = parent;
 210
 211     m_zooming = FALSE;
 212
 213     SetBackgroundColour( *wxWHITE );
 214 }
 215
 216 void wxPlotArea::OnMouse( wxMouseEvent &event )
 217 {
 218     int client_width;
 219     int client_height;
 220     GetClientSize( &client_width, &client_height);
 221     int view_x;
 222     int view_y;
 223     m_owner->GetViewStart( &view_x, &view_y );
 224     view_x *= wxPLOT_SCROLL_STEP;
 225     view_y *= wxPLOT_SCROLL_STEP;
 226
 227     wxCoord x = event.GetX();
 228     wxCoord y = event.GetY();
 229     x += view_x;
 230     y += view_y;
 231
 232     wxNode *node = m_owner->m_curves.First();
 233     while (node)
 234     {
 235         wxPlotCurve *curve = (wxPlotCurve*)node->Data();
 236
 237         double double_client_height = (double)client_height;
 238         double range = curve->GetEndY() - curve->GetStartY();
 239         double end = curve->GetEndY();
 240         wxCoord offset_y = curve->GetOffsetY();
 241
 242         double dy = (end - curve->GetY( (wxInt32)(x/m_owner->GetZoom()) )) / range;
 243         wxCoord curve_y = (wxCoord)(dy * double_client_height) - offset_y - 1;
 244
 245         if ((y-curve_y < 4) && (y-curve_y > -4))
 246         {
 247             wxPlotEvent event1( event.ButtonDClick() ? wxEVT_PLOT_DOUBLECLICKED : wxEVT_PLOT_CLICKED, m_owner->GetId() );
 248             event1.SetEventObject( m_owner );
 249             event1.SetZoom( m_owner->GetZoom() );
 250             event1.SetCurve( curve );
 251             event1.SetPosition( (int)floor(x/m_owner->GetZoom()) );
 252             m_owner->GetEventHandler()->ProcessEvent( event1 );
 253
 254             if (curve != m_owner->GetCurrent())
 255             {
 256                 wxPlotEvent event2( wxEVT_PLOT_SEL_CHANGING, m_owner->GetId() );
 257                 event2.SetEventObject( m_owner );
 258                 event2.SetZoom( m_owner->GetZoom() );
 259                 event2.SetCurve( curve );
 260                 if (!m_owner->GetEventHandler()->ProcessEvent( event2 ) || event2.IsAllowed())
 261                 {
 262                     m_owner->SetCurrent( curve );
 263                 }
 264             }
 265             return;
 266         }
 267
 268         node = node->Next();
 269     }
 270 }
 271
 272 void wxPlotArea::DeleteCurve( wxPlotCurve *curve, int from, int to )
 273 {
 274     wxClientDC dc(this);
 275     m_owner->PrepareDC( dc );
 276     dc.SetPen( *wxWHITE_PEN );
 277     DrawCurve( &dc, curve, from, to );
 278 }
 279
 280 void wxPlotArea::DrawCurve( wxDC *dc, wxPlotCurve *curve, int from, int to )
 281 {
 282     int view_x;
 283     int view_y;
 284     m_owner->GetViewStart( &view_x, &view_y );
 285     view_x *= wxPLOT_SCROLL_STEP;
 286
 287     if (from == -1)
 288         from = view_x;
 289
 290     int client_width;
 291     int client_height;
 292     GetClientSize( &client_width, &client_height);
 293
 294     if (to == -1)
 295         to = view_x + client_width;
 296
 297     double zoom = m_owner->GetZoom();
 298
 299     int start_x = wxMax( from, (int)floor(curve->GetStartX()*zoom) );
 300     int end_x = wxMin( to, (int)floor(curve->GetEndX()*zoom) );
 301
 302     start_x = wxMax( view_x, start_x );
 303     end_x = wxMin( view_x + client_width, end_x );
 304
 305     end_x++;
 306
 307     double double_client_height = (double)client_height;
 308     double range = curve->GetEndY() - curve->GetStartY();
 309     double end = curve->GetEndY();
 310     wxCoord offset_y = curve->GetOffsetY();
 311
 312     wxCoord y=0,last_y=0;
 313     for (int x = start_x; x < end_x; x++)
 314     {
 315         double dy = (end - curve->GetY( (wxInt32)(x/zoom) )) / range;
 316         y = (wxCoord)(dy * double_client_height) - offset_y - 1;
 317
 318         if (x != start_x)
 319            dc->DrawLine( x-1, last_y, x, y );
 320
 321         last_y = y;
 322     }
 323 }
 324
 325 void wxPlotArea::DrawOnOffCurve( wxDC *dc, wxPlotOnOffCurve *curve, int from, int to )
 326 {
 327     int view_x;
 328     int view_y;
 329     m_owner->GetViewStart( &view_x, &view_y );
 330     view_x *= wxPLOT_SCROLL_STEP;
 331
 332     if (from == -1)
 333         from = view_x;
 334
 335     int client_width;
 336     int client_height;
 337     GetClientSize( &client_width, &client_height);
 338
 339     if (to == -1)
 340         to = view_x + client_width;
 341
 342     double zoom = m_owner->GetZoom();
 343
 344     int start_x = wxMax( from, (int)floor(curve->GetStartX()*zoom) );
 345     int end_x = wxMin( to, (int)floor(curve->GetEndX()*zoom) );
 346
 347     start_x = wxMax( view_x, start_x );
 348     end_x = wxMin( view_x + client_width, end_x );
 349
 350     end_x++;
 351
 352     wxCoord offset_y = curve->GetOffsetY();
 353     wxCoord last_off = -5;
 354
 355     if (curve->GetCount() == 0)
 356         return;
 357
 358     for (size_t index = 0; index < curve->GetCount(); index++)
 359     {
 360         wxPlotOnOff *p = curve->GetAt( index );
 361
 362         wxCoord on = (wxCoord)(p->m_on*zoom);
 363         wxCoord off = (wxCoord)(p->m_off*zoom);
 364
 365         if (end_x < on)
 366         {
 367             curve->DrawOffLine( *dc, client_height-offset_y, last_off, on );
 368             break;
 369         }
 370
 371         if (off >= start_x)
 372         {
 373             curve->DrawOffLine( *dc, client_height-offset_y, last_off, on );
 374             curve->DrawOnLine( *dc, client_height-offset_y, on, off, p->m_clientData );
 375         }
 376         last_off = off;
 377     }
 378
 379     wxPlotOnOff *p = curve->GetAt( curve->GetCount()-1 );
 380     wxCoord off = (wxCoord)(p->m_off*zoom);
 381     if (off < end_x)
 382         curve->DrawOffLine( *dc, client_height-offset_y, off, to );
 383 }
 384
 385 void wxPlotArea::OnPaint( wxPaintEvent &WXUNUSED(event) )
 386 {
 387     int view_x;
 388     int view_y;
 389     m_owner->GetViewStart( &view_x, &view_y );
 390     view_x *= wxPLOT_SCROLL_STEP;
 391     view_y *= wxPLOT_SCROLL_STEP;
 392
 393     wxPaintDC dc( this );
 394     m_owner->PrepareDC( dc );
 395
 396     wxRegionIterator upd( GetUpdateRegion() );
 397
 398     while (upd)
 399     {
 400         int update_x = upd.GetX();
 401         int update_y = upd.GetY();
 402         int update_width = upd.GetWidth();
 403
 404         update_x += view_x;
 405         update_y += view_y;
 406
 407 /*
 408         if (m_owner->m_current)
 409         {
 410             dc.SetPen( *wxLIGHT_GREY_PEN );
 411             int base_line = client_height - m_owner->m_current->GetOffsetY();
 412             dc.DrawLine( update_x-1, base_line-1, update_x+update_width+2, base_line-1 );
 413         }
 414 */
 415
 416         wxNode *node = m_owner->m_curves.First();
 417         while (node)
 418         {
 419             wxPlotCurve *curve = (wxPlotCurve*) node->Data();
 420
 421             if (curve == m_owner->GetCurrent())
 422                 dc.SetPen( *wxBLACK_PEN );
 423             else
 424                 dc.SetPen( *wxGREY_PEN );
 425
 426             DrawCurve( &dc, curve, update_x-1, update_x+update_width+2 );
 427
 428             node = node->Next();
 429         }
 430
 431         dc.SetPen( *wxRED_PEN );
 432
 433         node = m_owner->m_onOffCurves.First();
 434         while (node)
 435         {
 436             wxPlotOnOffCurve *curve = (wxPlotOnOffCurve*) node->Data();
 437
 438             DrawOnOffCurve( &dc, curve, update_x-1, update_x+update_width+2 );
 439
 440             node = node->Next();
 441         }
 442
 443         upd ++;
 444     }
 445 }
 446
 447 void wxPlotArea::ScrollWindow( int dx, int dy, const wxRect *rect )
 448 {
 449     wxWindow::ScrollWindow( dx, dy, rect );
 450 //    m_owner->m_xaxis->ScrollWindow( dx, 0 );
 451 }
 452
 453 //-----------------------------------------------------------------------------
 454 // wxPlotXAxisArea
 455 //-----------------------------------------------------------------------------
 456
 457 IMPLEMENT_DYNAMIC_CLASS(wxPlotXAxisArea, wxWindow)
 458
 459 BEGIN_EVENT_TABLE(wxPlotXAxisArea, wxWindow)
 460   EVT_PAINT(        wxPlotXAxisArea::OnPaint)
 461   EVT_LEFT_DOWN(    wxPlotXAxisArea::OnMouse)
 462 END_EVENT_TABLE()
 463
 464 wxPlotXAxisArea::wxPlotXAxisArea( wxPlotWindow *parent )
 465         : wxWindow( parent, -1, wxDefaultPosition, wxSize(-1,40), 0, _T("plotxaxisarea") )
 466 {
 467     m_owner = parent;
 468
 469     SetBackgroundColour( *wxWHITE );
 470     SetFont( *wxSMALL_FONT );
 471 }
 472
 473 void wxPlotXAxisArea::OnMouse( wxMouseEvent &event )
 474 {
 475     int client_width;
 476     int client_height;
 477     GetClientSize( &client_width, &client_height);
 478     int view_x;
 479     int view_y;
 480     m_owner->GetViewStart( &view_x, &view_y );
 481     view_x *= wxPLOT_SCROLL_STEP;
 482     view_y *= wxPLOT_SCROLL_STEP;
 483
 484     wxCoord x = event.GetX();
 485     wxCoord y = event.GetY();
 486     x += view_x;
 487     y += view_y;
 488
 489     /* do something here */
 490 }
 491
 492 void wxPlotXAxisArea::OnPaint( wxPaintEvent &WXUNUSED(event) )
 493 {
 494     int view_x;
 495     int view_y;
 496     m_owner->GetViewStart( &view_x, &view_y );
 497     view_x *= wxPLOT_SCROLL_STEP;
 498     view_y *= wxPLOT_SCROLL_STEP;
 499
 500     wxPaintDC dc( this );
 501
 502     int client_width;
 503     int client_height;
 504     GetClientSize( &client_width, &client_height);
 505
 506     double zoom = m_owner->GetZoom();
 507
 508     double ups = m_owner->GetUnitsPerValue() / zoom;
 509
 510     double start = view_x * ups;
 511     double end = (view_x + client_width) * ups;
 512     double range = end - start;
 513
 514     int int_log_range = (int)floor( log10( range ) );
 515     double step = 1.0;
 516     if (int_log_range > 0)
 517     {
 518         for (int i = 0; i < int_log_range; i++)
 519            step *= 10;
 520     }
 521     if (int_log_range < 0)
 522     {
 523         for (int i = 0; i < -int_log_range; i++)
 524            step /= 10;
 525     }
 526     double lower = ceil(start / step) * step;
 527     double upper = floor(end / step) * step;
 528
 529     // if too few values, shrink size
 530     if ((range/step) < 4)
 531     {
 532         step /= 2;
 533         if (lower-step > start) lower -= step;
 534         if (upper+step < end) upper += step;
 535     }
 536
 537     // if still too few, again
 538     if ((range/step) < 4)
 539     {
 540         step /= 2;
 541         if (lower-step > start) lower -= step;
 542         if (upper+step < end) upper += step;
 543     }
 544
 545     dc.SetBrush( *wxWHITE_BRUSH );
 546     dc.SetPen( *wxTRANSPARENT_PEN );
 547     dc.DrawRectangle( 4, 5, client_width-14, 10 );
 548     dc.DrawRectangle( 0, 20, client_width, 20 );
 549     dc.SetPen( *wxBLACK_PEN );
 550
 551     double current = lower;
 552     while (current < upper+(step/2))
 553     {
 554         int x = (int)ceil((current-start) / range * (double)client_width) - 1;
 555         if ((x > 4) && (x < client_width-25))
 556         {
 557             dc.DrawLine( x, 5, x, 15 );
 558             wxString label;
 559             if (range < 50)
 560             {
 561                 label.Printf( _T("%f"), current );
 562                 while (label.Last() == _T('0'))
 563                     label.RemoveLast();
 564                 if ((label.Last() == _T('.')) || (label.Last() == _T(',')))
 565                     label.Append( _T('0') );
 566             }
 567             else
 568                 label.Printf( _T("%d"), (int)floor(current) );
 569             dc.DrawText( label, x-4, 20 );
 570         }
 571
 572         current += step;
 573     }
 574
 575     dc.DrawLine( 0, 15, client_width-8, 15 );
 576     dc.DrawLine( client_width-4, 15, client_width-10, 10 );
 577     dc.DrawLine( client_width-4, 15, client_width-10, 20 );
 578 }
 579
 580 //-----------------------------------------------------------------------------
 581 // wxPlotYAxisArea
 582 //-----------------------------------------------------------------------------
 583
 584 IMPLEMENT_DYNAMIC_CLASS(wxPlotYAxisArea, wxWindow)
 585
 586 BEGIN_EVENT_TABLE(wxPlotYAxisArea, wxWindow)
 587   EVT_PAINT(        wxPlotYAxisArea::OnPaint)
 588   EVT_LEFT_DOWN(    wxPlotYAxisArea::OnMouse)
 589 END_EVENT_TABLE()
 590
 591 wxPlotYAxisArea::wxPlotYAxisArea( wxPlotWindow *parent )
 592         : wxWindow( parent, -1, wxDefaultPosition, wxSize(60,-1), 0, _T("plotyaxisarea") )
 593 {
 594     m_owner = parent;
 595
 596     SetBackgroundColour( *wxWHITE );
 597     SetFont( *wxSMALL_FONT );
 598 }
 599
 600 void wxPlotYAxisArea::OnMouse( wxMouseEvent &WXUNUSED(event) )
 601 {
 602     /* do something here */
 603 }
 604
 605 void wxPlotYAxisArea::OnPaint( wxPaintEvent &WXUNUSED(event) )
 606 {
 607     wxPaintDC dc( this );
 608
 609     wxPlotCurve *curve = m_owner->GetCurrent();
 610
 611     if (!curve) return;
 612
 613     int client_width;
 614     int client_height;
 615     GetClientSize( &client_width, &client_height);
 616
 617
 618     double range = curve->GetEndY() - curve->GetStartY();
 619     double offset = ((double) curve->GetOffsetY() / (double)client_height ) * range;
 620     double start = curve->GetStartY() - offset;
 621     double end = curve->GetEndY() - offset;
 622
 623     int int_log_range = (int)floor( log10( range ) );
 624     double step = 1.0;
 625     if (int_log_range > 0)
 626     {
 627         for (int i = 0; i < int_log_range; i++)
 628            step *= 10;
 629     }
 630     if (int_log_range < 0)
 631     {
 632         for (int i = 0; i < -int_log_range; i++)
 633            step /= 10;
 634     }
 635     double lower = ceil(start / step) * step;
 636     double upper = floor(end / step) * step;
 637
 638     // if too few values, shrink size
 639     if ((range/step) < 4)
 640     {
 641         step /= 2;
 642         if (lower-step > start) lower -= step;
 643         if (upper+step < end) upper += step;
 644     }
 645
 646     // if still too few, again
 647     if ((range/step) < 4)
 648     {
 649         step /= 2;
 650         if (lower-step > start) lower -= step;
 651         if (upper+step < end) upper += step;
 652     }
 653
 654     dc.SetPen( *wxBLACK_PEN );
 655
 656     double current = lower;
 657     while (current < upper+(step/2))
 658     {
 659         int y = (int)((curve->GetEndY()-current) / range * (double)client_height) - 1;
 660         y -= curve->GetOffsetY();
 661         if ((y > 10) && (y < client_height-7))
 662         {
 663             dc.DrawLine( client_width-15, y, client_width-7, y );
 664             wxString label;
 665             if (range < 50)
 666             {
 667                 label.Printf( _T("%f"), current );
 668                 while (label.Last() == _T('0'))
 669                     label.RemoveLast();
 670                 if ((label.Last() == _T('.')) || (label.Last() == _T(',')))
 671                     label.Append( _T('0') );
 672             }
 673             else
 674                 label.Printf( _T("%d"), (int)floor(current) );
 675             dc.DrawText( label, 5, y-7 );
 676         }
 677
 678         current += step;
 679     }
 680
 681     dc.DrawLine( client_width-15, 6, client_width-15, client_height );
 682     dc.DrawLine( client_width-15, 2, client_width-20, 8 );
 683     dc.DrawLine( client_width-15, 2, client_width-10, 8 );
 684 }
 685
 686 //-----------------------------------------------------------------------------
 687 // wxPlotWindow
 688 //-----------------------------------------------------------------------------
 689
 690 #define  ID_ENLARGE       1000
 691 #define  ID_SHRINK        1002
 692
 693 #define  ID_MOVE_UP       1006
 694 #define  ID_MOVE_DOWN     1007
 695
 696 #define  ID_ZOOM_IN       1010
 697 #define  ID_ZOOM_OUT      1011
 698
 699
 700 IMPLEMENT_DYNAMIC_CLASS(wxPlotWindow, wxScrolledWindow)
 701
 702 BEGIN_EVENT_TABLE(wxPlotWindow, wxScrolledWindow)
 703   EVT_BUTTON(  ID_MOVE_UP,     wxPlotWindow::OnMoveUp)
 704   EVT_BUTTON(  ID_MOVE_DOWN,   wxPlotWindow::OnMoveDown)
 705
 706   EVT_BUTTON(  ID_ENLARGE,  wxPlotWindow::OnEnlarge)
 707   EVT_BUTTON(  ID_SHRINK,   wxPlotWindow::OnShrink)
 708
 709   EVT_BUTTON(  ID_ZOOM_IN,     wxPlotWindow::OnZoomIn)
 710   EVT_BUTTON(  ID_ZOOM_OUT,    wxPlotWindow::OnZoomOut)
 711
 712   EVT_SCROLLWIN( wxPlotWindow::OnScroll2)
 713 END_EVENT_TABLE()
 714
 715 wxPlotWindow::wxPlotWindow( wxWindow *parent, wxWindowID id, const wxPoint &pos, const wxSize &size, int flag )
 716         : wxScrolledWindow( parent, id, pos, size, flag, _T("plotcanvas") )
 717 {
 718     m_xUnitsPerValue = 1.0;
 719     m_xZoom = 1.0;
 720
 721     m_enlargeAroundWindowCentre = FALSE;
 722     m_scrollOnThumbRelease = FALSE;
 723
 724     m_area = new wxPlotArea( this );
 725     wxBoxSizer *mainsizer = new wxBoxSizer( wxHORIZONTAL );
 726
 727     if ((GetWindowStyleFlag() & wxPLOT_BUTTON_ALL) != 0)
 728     {
 729         wxBoxSizer *buttonlist = new wxBoxSizer( wxVERTICAL );
 730         if ((GetWindowStyleFlag() & wxPLOT_BUTTON_ENLARGE) != 0)
 731         {
 732             buttonlist->Add( new wxBitmapButton( this, ID_ENLARGE, *GetEnlargeBitmap() ), 0, wxEXPAND|wxALL, 2 );
 733             buttonlist->Add( new wxBitmapButton( this, ID_SHRINK, *GetShrinkBitmap() ), 0, wxEXPAND|wxALL, 2 );
 734             buttonlist->Add( 20,10, 0 );
 735         }
 736         if ((GetWindowStyleFlag() & wxPLOT_BUTTON_MOVE) != 0)
 737         {
 738             buttonlist->Add( new wxBitmapButton( this, ID_MOVE_UP, *GetUpBitmap() ), 0, wxEXPAND|wxALL, 2 );
 739             buttonlist->Add( new wxBitmapButton( this, ID_MOVE_DOWN, *GetDownBitmap() ), 0, wxEXPAND|wxALL, 2 );
 740             buttonlist->Add( 20,10, 0 );
 741         }
 742         if ((GetWindowStyleFlag() & wxPLOT_BUTTON_ZOOM) != 0)
 743         {
 744             buttonlist->Add( new wxBitmapButton( this, ID_ZOOM_IN, *GetZoomInBitmap() ), 0, wxEXPAND|wxALL, 2 );
 745             buttonlist->Add( new wxBitmapButton( this, ID_ZOOM_OUT, *GetZoomOutBitmap() ), 0, wxEXPAND|wxALL, 2 );
 746         }
 747         mainsizer->Add( buttonlist, 0, wxEXPAND|wxALL, 4 );
 748     }
 749
 750     wxBoxSizer *plotsizer = new wxBoxSizer( wxHORIZONTAL );
 751
 752     if ((GetWindowStyleFlag() & wxPLOT_Y_AXIS) != 0)
 753     {
 754         m_yaxis = new wxPlotYAxisArea( this );
 755
 756         wxBoxSizer *vert1 = new wxBoxSizer( wxVERTICAL );
 757         plotsizer->Add( vert1, 0, wxEXPAND );
 758         vert1->Add( m_yaxis, 1 );
 759         if ((GetWindowStyleFlag() & wxPLOT_X_AXIS) != 0)
 760             vert1->Add( 60, 40 );
 761     }
 762     else
 763     {
 764         m_yaxis = (wxPlotYAxisArea*) NULL;
 765     }
 766
 767     if ((GetWindowStyleFlag() & wxPLOT_X_AXIS) != 0)
 768     {
 769         m_xaxis = new wxPlotXAxisArea( this );
 770
 771         wxBoxSizer *vert2 = new wxBoxSizer( wxVERTICAL );
 772         plotsizer->Add( vert2, 1, wxEXPAND );
 773         vert2->Add( m_area, 1, wxEXPAND );
 774         vert2->Add( m_xaxis, 0, wxEXPAND );
 775     }
 776     else
 777     {
 778         plotsizer->Add( m_area, 1, wxEXPAND );
 779         m_xaxis = (wxPlotXAxisArea*) NULL;
 780     }
 781
 782     mainsizer->Add( plotsizer, 1, wxEXPAND );
 783
 784     SetAutoLayout( TRUE );
 785     SetSizer( mainsizer );
 786
 787     SetTargetWindow( m_area );
 788
 789     SetBackgroundColour( *wxWHITE );
 790
 791     m_current = (wxPlotCurve*) NULL;
 792 }
 793
 794 wxPlotWindow::~wxPlotWindow()
 795 {
 796 }
 797
 798 void wxPlotWindow::Add( wxPlotCurve *curve )
 799 {
 800     m_curves.Append( curve );
 801     if (!m_current) m_current = curve;
 802
 803     ResetScrollbar();
 804 }
 805
 806 size_t wxPlotWindow::GetCount()
 807 {
 808     return m_curves.GetCount();
 809 }
 810
 811 wxPlotCurve *wxPlotWindow::GetAt( size_t n )
 812 {
 813     wxNode *node = m_curves.Nth( n );
 814     if (!node)
 815         return (wxPlotCurve*) NULL;
 816
 817     return (wxPlotCurve*) node->Data();
 818 }
 819
 820 void wxPlotWindow::SetCurrent( wxPlotCurve* current )
 821 {
 822     m_current = current;
 823     m_area->Refresh( FALSE );
 824
 825     RedrawYAxis();
 826
 827     wxPlotEvent event( wxEVT_PLOT_SEL_CHANGED, GetId() );
 828     event.SetEventObject( this );
 829     event.SetZoom( GetZoom() );
 830     event.SetCurve( m_current );
 831     GetEventHandler()->ProcessEvent( event );
 832 }
 833
 834 void wxPlotWindow::Delete( wxPlotCurve* curve )
 835 {
 836     wxNode *node = m_curves.Find( curve );
 837     if (!node) return;
 838
 839     m_curves.DeleteObject( curve );
 840
 841     m_area->DeleteCurve( curve );
 842     m_area->Refresh( FALSE );
 843
 844     if (curve == m_current) m_current = (wxPlotCurve *) NULL;
 845 }
 846
 847 wxPlotCurve *wxPlotWindow::GetCurrent()
 848 {
 849     return m_current;
 850 }
 851
 852 void wxPlotWindow::Add( wxPlotOnOffCurve *curve )
 853 {
 854     m_onOffCurves.Append( curve );
 855 }
 856
 857 void wxPlotWindow::Delete( wxPlotOnOffCurve* curve )
 858 {
 859     wxNode *node = m_onOffCurves.Find( curve );
 860     if (!node) return;
 861
 862     m_onOffCurves.DeleteObject( curve );
 863 }
 864
 865 size_t wxPlotWindow::GetOnOffCurveCount()
 866 {
 867     return m_onOffCurves.GetCount();
 868 }
 869
 870 wxPlotOnOffCurve *wxPlotWindow::GetOnOffCurveAt( size_t n )
 871 {
 872     wxNode *node = m_onOffCurves.Nth( n );
 873     if (!node)
 874         return (wxPlotOnOffCurve*) NULL;
 875
 876     return (wxPlotOnOffCurve*) node->Data();
 877 }
 878
 879 void wxPlotWindow::Move( wxPlotCurve* curve, int pixels_up )
 880 {
 881     m_area->DeleteCurve( curve );
 882
 883     curve->SetOffsetY( curve->GetOffsetY() + pixels_up );
 884
 885     m_area->Refresh( FALSE );
 886
 887     RedrawYAxis();
 888 }
 889
 890 void wxPlotWindow::OnMoveUp( wxCommandEvent& WXUNUSED(event) )
 891 {
 892     if (!m_current) return;
 893
 894     Move( m_current, 25 );
 895 }
 896
 897 void wxPlotWindow::OnMoveDown( wxCommandEvent& WXUNUSED(event) )
 898 {
 899     if (!m_current) return;
 900
 901     Move( m_current, -25 );
 902 }
 903
 904 void wxPlotWindow::Enlarge( wxPlotCurve *curve, double factor )
 905 {
 906     m_area->DeleteCurve( curve );
 907
 908     int client_width;
 909     int client_height;
 910     m_area->GetClientSize( &client_width, &client_height);
 911     double offset = (double)curve->GetOffsetY() / (double)client_height;
 912
 913     double range = curve->GetEndY() - curve->GetStartY();
 914     offset *= range;
 915
 916     double new_range = range / factor;
 917     double new_offset = offset / factor;
 918
 919     if (m_enlargeAroundWindowCentre)
 920     {
 921         double middle = curve->GetStartY() - offset + range/2;
 922
 923         curve->SetStartY( middle - new_range / 2 + new_offset );
 924         curve->SetEndY( middle + new_range / 2 + new_offset  );
 925     }
 926     else
 927     {
 928         curve->SetStartY( (curve->GetStartY() - offset)/factor + new_offset );
 929         curve->SetEndY( (curve->GetEndY() - offset)/factor + new_offset );
 930     }
 931
 932     m_area->Refresh( FALSE );
 933     RedrawYAxis();
 934 }
 935
 936 void wxPlotWindow::SetUnitsPerValue( double upv )
 937 {
 938     m_xUnitsPerValue = upv;
 939
 940     RedrawXAxis();
 941 }
 942
 943 void wxPlotWindow::SetZoom( double zoom )
 944 {
 945     double old_zoom = m_xZoom;
 946     m_xZoom = zoom;
 947
 948     int view_x = 0;
 949     int view_y = 0;
 950     GetViewStart( &view_x, &view_y );
 951
 952     wxInt32 max = 0;
 953     wxNode *node = m_curves.First();
 954     while (node)
 955     {
 956         wxPlotCurve *curve = (wxPlotCurve*) node->Data();
 957         if (curve->GetEndX() > max)
 958             max = curve->GetEndX();
 959         node = node->Next();
 960     }
 961     SetScrollbars( wxPLOT_SCROLL_STEP, wxPLOT_SCROLL_STEP,
 962                    (int)((max*m_xZoom)/wxPLOT_SCROLL_STEP)+1, 0,
 963                    (int)(view_x*zoom/old_zoom), 0,
 964                    TRUE );
 965
 966     RedrawXAxis();
 967     m_area->Refresh( TRUE );
 968 }
 969
 970 void wxPlotWindow::ResetScrollbar()
 971 {
 972     wxInt32 max = 0;
 973     wxNode *node = m_curves.First();
 974     while (node)
 975     {
 976         wxPlotCurve *curve = (wxPlotCurve*) node->Data();
 977         if (curve->GetEndX() > max)
 978             max = curve->GetEndX();
 979         node = node->Next();
 980     }
 981
 982     SetScrollbars( wxPLOT_SCROLL_STEP, wxPLOT_SCROLL_STEP,
 983                    (int)(((max*m_xZoom)/wxPLOT_SCROLL_STEP)+1), 0 );
 984 }
 985
 986 void wxPlotWindow::RedrawXAxis()
 987 {
 988     if (m_xaxis)
 989         m_xaxis->Refresh( FALSE );
 990 }
 991
 992 void wxPlotWindow::RedrawYAxis()
 993 {
 994     if (m_yaxis)
 995        m_yaxis->Refresh( TRUE );
 996 }
 997
 998 void wxPlotWindow::RedrawEverything()
 999 {
1000     if (m_xaxis)
1001         m_xaxis->Refresh( TRUE );
1002     if (m_yaxis)
1003         m_yaxis->Refresh( TRUE );
1004     m_area->Refresh( TRUE );
1005 }
1006
1007 void wxPlotWindow::OnZoomIn( wxCommandEvent& WXUNUSED(event) )
1008 {
1009     SetZoom( m_xZoom * 1.5 );
1010 }
1011
1012 void wxPlotWindow::OnZoomOut( wxCommandEvent& WXUNUSED(event) )
1013 {
1014     SetZoom( m_xZoom * 0.6666 );
1015 }
1016
1017 void wxPlotWindow::OnEnlarge( wxCommandEvent& WXUNUSED(event) )
1018 {
1019     if (!m_current) return;
1020
1021     Enlarge( m_current, 1.5 );
1022 }
1023
1024 void wxPlotWindow::OnShrink( wxCommandEvent& WXUNUSED(event) )
1025 {
1026     if (!m_current) return;
1027
1028     Enlarge( m_current, 0.6666666 );
1029 }
1030
1031 void wxPlotWindow::OnScroll2( wxScrollWinEvent& event )
1032 {
1033     if ((!m_scrollOnThumbRelease) || (event.GetEventType() != wxEVT_SCROLLWIN_THUMBTRACK))
1034     {
1035         wxScrolledWindow::OnScroll( event );
1036         RedrawXAxis();
1037     }
1038 }
1039
1040 // ----------------------------------------------------------------------------
1041 // global functions
1042 // ----------------------------------------------------------------------------
1043
1044 // FIXME MT-UNSAFE
1045 static wxBitmap *GetEnlargeBitmap()
1046 {
1047     static wxBitmap* s_bitmap = (wxBitmap *) NULL;
1048     static bool s_loaded = FALSE;
1049
1050     if ( !s_loaded )
1051     {
1052         s_loaded = TRUE; // set it to TRUE anyhow, we won't try again
1053
1054         #if defined(__WXMSW__) || defined(__WXPM__)
1055             s_bitmap = new wxBitmap(_T("plot_enl_bmp"), wxBITMAP_TYPE_RESOURCE);
1056         #else
1057             s_bitmap = new wxBitmap( plot_enl_xpm );
1058         #endif
1059     }
1060
1061     return s_bitmap;
1062 }
1063
1064 static wxBitmap *GetShrinkBitmap()
1065 {
1066     static wxBitmap* s_bitmap = (wxBitmap *) NULL;
1067     static bool s_loaded = FALSE;
1068
1069     if ( !s_loaded )
1070     {
1071         s_loaded = TRUE; // set it to TRUE anyhow, we won't try again
1072
1073         #if defined(__WXMSW__) || defined(__WXPM__)
1074             s_bitmap = new wxBitmap(_T("plot_shr_bmp"), wxBITMAP_TYPE_RESOURCE);
1075         #else
1076             s_bitmap = new wxBitmap( plot_shr_xpm );
1077         #endif
1078     }
1079
1080     return s_bitmap;
1081 }
1082
1083 static wxBitmap *GetZoomInBitmap()
1084 {
1085     static wxBitmap* s_bitmap = (wxBitmap *) NULL;
1086     static bool s_loaded = FALSE;
1087
1088     if ( !s_loaded )
1089     {
1090         s_loaded = TRUE; // set it to TRUE anyhow, we won't try again
1091
1092         #if defined(__WXMSW__) || defined(__WXPM__)
1093             s_bitmap = new wxBitmap(_T("plot_zin_bmp"), wxBITMAP_TYPE_RESOURCE);
1094         #else
1095             s_bitmap = new wxBitmap( plot_zin_xpm );
1096         #endif
1097     }
1098
1099     return s_bitmap;
1100 }
1101
1102 static wxBitmap *GetZoomOutBitmap()
1103 {
1104     static wxBitmap* s_bitmap = (wxBitmap *) NULL;
1105     static bool s_loaded = FALSE;
1106
1107     if ( !s_loaded )
1108     {
1109         s_loaded = TRUE; // set it to TRUE anyhow, we won't try again
1110
1111         #if defined(__WXMSW__) || defined(__WXPM__)
1112             s_bitmap = new wxBitmap(_T("plot_zot_bmp"), wxBITMAP_TYPE_RESOURCE);
1113         #else
1114             s_bitmap = new wxBitmap( plot_zot_xpm );
1115         #endif
1116     }
1117
1118     return s_bitmap;
1119 }
1120
1121 static wxBitmap *GetUpBitmap()
1122 {
1123     static wxBitmap* s_bitmap = (wxBitmap *) NULL;
1124     static bool s_loaded = FALSE;
1125
1126     if ( !s_loaded )
1127     {
1128         s_loaded = TRUE; // set it to TRUE anyhow, we won't try again
1129
1130         #if defined(__WXMSW__) || defined(__WXPM__)
1131             s_bitmap = new wxBitmap(_T("plot_up_bmp"), wxBITMAP_TYPE_RESOURCE);
1132         #else
1133             s_bitmap = new wxBitmap( plot_up_xpm );
1134         #endif
1135     }
1136
1137     return s_bitmap;
1138 }
1139
1140 static wxBitmap *GetDownBitmap()
1141 {
1142     static wxBitmap* s_bitmap = (wxBitmap *) NULL;
1143     static bool s_loaded = FALSE;
1144
1145     if ( !s_loaded )
1146     {
1147         s_loaded = TRUE; // set it to TRUE anyhow, we won't try again
1148
1149         #if defined(__WXMSW__) || defined(__WXPM__)
1150             s_bitmap = new wxBitmap(_T("plot_dwn_bmp"), wxBITMAP_TYPE_RESOURCE);
1151         #else
1152             s_bitmap = new wxBitmap( plot_dwn_xpm );
1153         #endif
1154     }
1155
1156     return s_bitmap;
1157 }
1158