mathplot.cxx

Go to the documentation of this file.

// Name:        mathplot.cpp
// Purpose:     Framework for mathematical graph plotting in wxWindows
// Author:      David Schalig
// Modified by:
// Created:     21/07/2003
// Copyright:   (c) David Schalig
// Licence:     wxWindows licence

/*EED
#ifdef __GNUG__
#pragma implementation "plot.h"
#endif
*/
// For compilers that support precompilation, includes "wx.h".
#include "wx/wxprec.h"
#ifdef __BORLANDC__
#pragma hdrstop
#endif

/*EED
#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

// EED
#ifndef WX_PRECOMP
#include <wx/wx.h>
#endif
*/


#include "mathplot.h"
//#include "wx/bmpbuttn.h"
//#include "wx/module.h"

//#include <math.h>

//-----------------------------------------------------------------------------
// mpLayer
//-----------------------------------------------------------------------------

IMPLEMENT_ABSTRACT_CLASS(mpLayer, wxObject)

mpLayer::mpLayer()
{
// EED wx 2.8.5
//      SetPen( *wxBLACK_PEN);
//      SetFont(*wxNORMAL_FONT);

        wxFont ff( *wxNORMAL_FONT);
        wxPen pp( *wxBLACK_PEN);
        SetPen( pp );
        SetFont( ff );

}

//-----------------------------------------------------------------------------
// mpLayer implementations - functions
//-----------------------------------------------------------------------------

IMPLEMENT_ABSTRACT_CLASS(mpFX, mpLayer)

mpFX::mpFX(wxString name, int flags)
{ 
        SetName(name);
        m_flags = flags; 
}

void mpFX::Plot(wxDC & dc, mpWindow & w)
{
        dc.SetPen( m_pen);

        if (m_pen.GetWidth() <= 1) 
        {
                for (wxCoord i = -(w.GetScrX()>>1); i < (w.GetScrX()>>1); ++i)
                {
                        dc.DrawPoint(i, (wxCoord) ((w.GetPosY() - GetY( (double)i / w.GetScaleX() + w.GetPosX()) ) * w.GetScaleY()));
                }
        }
        else
        {
                for (wxCoord i = -(w.GetScrX()>>1); i < (w.GetScrX()>>1); ++i)
                {
                        wxCoord c = (wxCoord) ((w.GetPosY() - GetY( (double)i / w.GetScaleX() + w.GetPosX()) ) * w.GetScaleY());
                        dc.DrawLine( i, c, i, c);
                }
        }

        if (!m_name.IsEmpty())
        {
                dc.SetFont(m_font);

                wxCoord tx, ty;
                dc.GetTextExtent(m_name, &tx, &ty);

                if ((m_flags & mpALIGNMASK) == mpALIGN_RIGHT)
                        tx = (w.GetScrX()>>1) - tx - 8;
                else if ((m_flags & mpALIGNMASK) == mpALIGN_CENTER)
                        tx = -tx/2;
                else
                        tx = -(w.GetScrX()>>1) + 8;

                dc.DrawText( m_name, tx, (wxCoord) ((w.GetPosY() - GetY( (double)tx / w.GetScaleX() + w.GetPosX())) * w.GetScaleY()) );
        }
}

IMPLEMENT_ABSTRACT_CLASS(mpFY, mpLayer)

mpFY::mpFY(wxString name, int flags)
{ 
        SetName(name);
        m_flags = flags;
}

void mpFY::Plot(wxDC & dc, mpWindow & w)
{
        dc.SetPen( m_pen);

        wxCoord i;

        if (m_pen.GetWidth() <= 1) 
        {
                for (i = -(w.GetScrY()>>1); i < (w.GetScrY()>>1); ++i)
                {
                        dc.DrawPoint((wxCoord) ((GetX( (double)i / w.GetScaleY() + w.GetPosY()) - w.GetPosX()) * w.GetScaleX()), -i);
                }
        }
        else
        {
                for (i = -(w.GetScrY()>>1); i < (w.GetScrY()>>1); ++i)
                {
                        wxCoord c =  (wxCoord) ((GetX( (double)i / w.GetScaleY() + w.GetPosY()) - w.GetPosX()) * w.GetScaleX());
                        dc.DrawLine(c, -i, c, -i);
                }
        }

        if (!m_name.IsEmpty())
        {
                dc.SetFont(m_font);

                wxCoord tx, ty;
                dc.GetTextExtent(m_name, &tx, &ty);

                if ((m_flags & mpALIGNMASK) == mpALIGN_TOP)
                        ty = (w.GetScrY()>>1) - 8;
                else if ((m_flags & mpALIGNMASK) == mpALIGN_CENTER)
                        ty = 16 - ty/2;
                else
                        ty = -(w.GetScrY()>>1) + 8;

                dc.DrawText( m_name, (wxCoord) ((GetX( (double)i / w.GetScaleY() + w.GetPosY()) - w.GetPosX()) * w.GetScaleX()), -ty);
        }
}

IMPLEMENT_ABSTRACT_CLASS(mpFXY, mpLayer)

mpFXY::mpFXY(wxString name, int flags)
{ 
        SetName(name);
        m_flags = flags; 
}

void mpFXY::Plot(wxDC & dc, mpWindow & w)
{
        dc.SetPen( m_pen);

        double x, y;
        Rewind();

        // for some reason DrawPoint does not use the current pen,
        // so we use DrawLine for fat pens

        if (m_pen.GetWidth() <= 1) 
        {
                while (GetNextXY(x, y))
                {
                        dc.DrawPoint( (wxCoord) ((x - w.GetPosX()) * w.GetScaleX()) ,
                                (wxCoord) ((w.GetPosY() - y) * w.GetScaleY()) );
                }
        }
        else
        {
                while (GetNextXY(x, y))
                {
                        wxCoord cx = (wxCoord) ((x - w.GetPosX()) * w.GetScaleX());
                        wxCoord cy = (wxCoord) ((w.GetPosY() - y) * w.GetScaleY());
                        dc.DrawLine(cx, cy, cx, cy);
                }
        }

        if (!m_name.IsEmpty())
        {
                dc.SetFont(m_font);

                wxCoord tx, ty;
                dc.GetTextExtent(m_name, &tx, &ty);

                // xxx implement else ... if (!HasBBox())
                {
                        const int sx = w.GetScrX()>>1;
                        const int sy = w.GetScrY()>>1;

                        if ((m_flags & mpALIGNMASK) == mpALIGN_NE)
                        {
                                tx = sx - tx - 8;
                                ty = -sy + 8;
                        }
                        else if ((m_flags & mpALIGNMASK) == mpALIGN_NW)
                        {
                                tx = -sx + 8;
                                ty = -sy + 8;
                        }
                        else if ((m_flags & mpALIGNMASK) == mpALIGN_SW)
                        {
                                tx = -sx + 8;
                                ty = sy - 8 - ty;
                        }
                        else
                        {
                                tx = sx - tx - 8;
                                ty = sy - 8 - ty;
                        }
                }
                //else
                //{
                //}

                dc.DrawText( m_name, tx, ty);
        }
}

//-----------------------------------------------------------------------------
// mpLayer implementations - furniture (scales, ...)
//-----------------------------------------------------------------------------

#define mpLN10 2.3025850929940456840179914546844

IMPLEMENT_CLASS(mpScaleX, mpLayer)

mpScaleX::mpScaleX(wxString name)
{ 
        SetName(name);/*
        SetFont(*wxSMALL_FONT);
        SetPen(*wxGREY_PEN);*/

        wxFont ff( *wxSMALL_FONT);
        wxPen pp( *wxGREY_PEN);
        SetPen( pp );
        SetFont( ff );
}

void mpScaleX::Plot(wxDC & dc, mpWindow & w)
{
        dc.SetPen( m_pen);
        dc.SetFont( m_font);

        const int orgy   = (int)(w.GetPosY() * w.GetScaleY());
        const int extend = w.GetScrX()/2;

        dc.DrawLine( -extend, orgy, extend, orgy);

        const double dig  = floor( log( 128.0 / w.GetScaleX() ) / mpLN10 );
        const double step = exp( mpLN10 * dig);
        const double end  = w.GetPosX() + (double)extend / w.GetScaleX();

        wxCoord tx, ty;
        wxString s;
        wxString fmt;
        int tmp = (int)dig;
        if (tmp>=1)
        {
                fmt = wxT("%.f");
        }
        else
        {
                tmp=8-tmp;
                fmt.Printf(wxT("%%.%df"), tmp >= -1 ? 2 : -tmp);
        }

        double n = floor( (w.GetPosX() - (double)extend / w.GetScaleX()) / step ) * step ;

        tmp=-65535;
        for (;n < end; n += step)
        {
                const int p = (int)((n - w.GetPosX()) * w.GetScaleX());
                dc.DrawLine( p, orgy, p, orgy+4);

                s.Printf(fmt, n);
                dc.GetTextExtent(s, &tx, &ty);
                if ((p-tx/2-tmp) > 64)
                {
                        dc.DrawText( s, p-tx/2, orgy+4);
                        tmp=p+tx/2;
                }
        }

        dc.GetTextExtent(m_name, &tx, &ty);
        dc.DrawText( m_name, extend - tx - 4, orgy + 4 + ty);
}

IMPLEMENT_CLASS(mpScaleY, mpLayer)

mpScaleY::mpScaleY(wxString name)
{ 
        SetName(name);/*
        SetFont(*wxSMALL_FONT);
        SetPen(*wxGREY_PEN);*/
        wxFont ff( *wxSMALL_FONT);
        wxPen pp( *wxGREY_PEN);
        SetPen( pp );
        SetFont( ff );
}

void mpScaleY::Plot(wxDC & dc, mpWindow & w, int orgy)
{
        dc.SetPen( m_pen);
        dc.SetFont( m_font);

        const int orgx   = -(int)(w.GetPosX() * w.GetScaleX());
        const int extend = w.GetScrY()/2;

        double sizedc = dc.GetSize().GetHeight()-orgy;

        dc.DrawLine( orgx, GetYTranslated(sizedc, -extend), orgx, GetYTranslated(sizedc, extend));

        const double dig  = floor( log( 128.0 / w.GetScaleY() ) / mpLN10 );
        const double step = exp( mpLN10 * dig);
        const double end  = w.GetPosY() + (double)extend / w.GetScaleY();

        wxCoord tx, ty;
        wxString s;
        wxString fmt;
        int tmp = (int)dig;
        if (tmp>=1)
        {
                fmt = wxT("%.f");
        }
        else
        {
                tmp=8-tmp;
                fmt.Printf(wxT("%%.%df"), tmp >= -1 ? 2 : -tmp);
        }

        double n = floor( (w.GetPosY() - (double)extend / w.GetScaleY()) / step ) * step ;

        tmp=65536;
        for (;n < end; n += step)
        {
                const int p = (int)((w.GetPosY() - n) * w.GetScaleY());
                dc.DrawLine( orgx, GetYTranslated(sizedc, p), orgx+4, GetYTranslated(sizedc,p));

                s.Printf(fmt, n);
                dc.GetTextExtent(s, &tx, &ty);
                if ((tmp-p+ty/2) > 32)
                {
                        dc.DrawText( s, orgx+4, GetYTranslated(sizedc,p-ty/2));
                        tmp=p-ty/2;
                }
        }

        dc.GetTextExtent(m_name, &tx, &ty);
        dc.DrawText( m_name, orgx-tx-4, GetYTranslated(sizedc,-extend + ty + 4));
}

//-----------------------------------------------------------------------------
// mpWindow
//-----------------------------------------------------------------------------

IMPLEMENT_DYNAMIC_CLASS(mpWindow, wxScrolledWindow)

BEGIN_EVENT_TABLE(mpWindow, wxScrolledWindow)
EVT_PAINT    ( mpWindow::OnPaint)
EVT_SIZE     ( mpWindow::OnSize)
EVT_SCROLLWIN( mpWindow::OnScroll2)

EVT_MIDDLE_UP( mpWindow::OnShowPopupMenu)
EVT_RIGHT_UP ( mpWindow::OnShowPopupMenu)
EVT_MENU( mpID_CENTER,    mpWindow::OnCenter)
EVT_MENU( mpID_FIT,       mpWindow::OnFit)
EVT_MENU( mpID_ZOOM_IN,   mpWindow::OnZoomIn)
EVT_MENU( mpID_ZOOM_OUT,  mpWindow::OnZoomOut)
//EVT_MENU( mpID_LINE_GUIDES, mpWindow::OnGuideLines)
// New method that belongs to the mpWindow
EVT_MENU( mpID_LOCKASPECT,mpWindow::OnLockAspect)
END_EVENT_TABLE()

mpWindow::mpWindow( wxWindow *parent, wxWindowID id, const wxPoint &pos, const wxSize &size, int flag )
: wxScrolledWindow( parent, id, wxDefaultPosition, wxDefaultSize, flag, wxT("wxPlotter") )
{
        m_scaleX = m_scaleY = 1.0;
        m_posX   = m_posY   = 0;
        m_scrX   = m_scrY   = 64;
        m_minX   = m_minY   = 0;
        m_maxX   = m_maxY   = 0;
        maxScrX = maxScrY   = 200;
        minScrX = minScrY   = 0;
        m_clickedX =  0;
        m_clickedY = 5000;
        m_lockaspect = FALSE;
        offsetX = offsetY = 0;
        offsetPixelX = offsetPixelY= 0;
        _bitmap_functions=NULL;

        real_guideLine_X = -1;
        real_guideLine_Y = -1;
        drawGuides = true;
        type=1;

        m_popmenu.Append( mpID_CENTER,     _("Center"),      _("Center plot view to this position"));
        m_popmenu.Append( mpID_FIT,        _("Fit"),         _("Set plot view to show all items"));
        m_popmenu.Append( mpID_ZOOM_IN,    _("Zoom in"),     _("Zoom in plot view."));
        m_popmenu.Append( mpID_ZOOM_OUT,   _("Zoom out"),    _("Zoom out plot view."));
        m_popmenu.AppendCheckItem( mpID_LOCKASPECT, _("Lock aspect"), _("Lock horizontal and vertical zoom aspect."));

        m_popmenu.AppendCheckItem( mpID_LINE_GUIDES, _("Turn off guide lines"), _("Enables/Disables the guide lines"));

        m_layers.DeleteContents(TRUE);
        SetBackgroundColour( *wxWHITE );
        EnableScrolling(FALSE, FALSE);
        SetSizeHints(128, 128);
        
        //_bitmap_functions= new wxBitmap(700,800);

        UpdateAll();
}

mpWindow::~mpWindow()
{
}

void mpWindow::Fit()
{
        if (UpdateBBox())
        {
                int cx, cy;
                GetClientSize( &cx, &cy);
                
                double d;
                d = m_maxX - m_minX;
                if (d!=0)
                {
                        m_scaleX = cx/d;
                        m_posX = m_minX + d/2;
                }
                d = m_maxY - m_minY;
                if (d!=0)
                {
                        m_scaleY = cy/d;
                        m_posY = m_minY + d/2;
                }

                if (m_lockaspect)
                {
                        double s = (m_scaleX + m_scaleY)/2;
                        m_scaleX = s;
                        m_scaleY = s;
                }

                UpdateAll();
        }
}

void mpWindow::ZoomIn()
{
        m_scaleX = m_scaleX * 2;
        m_scaleY = m_scaleY * 2;
        UpdateAll();
}

void mpWindow::ZoomOut()
{
        m_scaleX = m_scaleX / 2;
        m_scaleY = m_scaleY / 2;
        UpdateAll();
}

void mpWindow::LockAspect(bool enable)
{
        m_lockaspect = enable;

        m_popmenu.Check(mpID_LOCKASPECT, enable);

        if (m_lockaspect)
        {
                double s = (m_scaleX + m_scaleY)/2;
                m_scaleX = s;
                m_scaleY = s;
        }

        UpdateAll();
}

void mpWindow::OnShowPopupMenu(wxMouseEvent &event)
{
        m_clickedX = event.GetX();
        m_clickedY = event.GetY();
        PopupMenu( &m_popmenu, event.GetX(), event.GetY());
}

void mpWindow::OnLockAspect(wxCommandEvent &event)
{
        LockAspect( !m_popmenu.IsChecked(mpID_LOCKASPECT) );
}

void mpWindow::OnFit(wxCommandEvent &event)
{
        Fit();
}

void mpWindow::OnCenter(wxCommandEvent &event)
{
        int cx, cy;
        GetClientSize(&cx, &cy);
        SetPos( (double)(m_clickedX-cx/2) / m_scaleX + m_posX, (double)(cy/2-m_clickedY) / m_scaleY + m_posY);
}

void mpWindow::OnZoomIn(wxCommandEvent &event)
{
        int cx, cy;
        GetClientSize(&cx, &cy);
        m_posX = (double)(m_clickedX-cx/2) / m_scaleX + m_posX;
        m_posY = (double)(cy/2-m_clickedY) / m_scaleY + m_posY;
        ZoomIn();
}

void mpWindow::OnZoomOut(wxCommandEvent &event)
{
        ZoomOut();
}

void mpWindow::OnSize( wxSizeEvent &event )
{
        UpdateAll();
}

bool mpWindow::AddLayer( mpLayer* layer)
{
        bool ret = m_layers.Append( layer) != NULL;
        UpdateAll();
        return ret;
}

void mpWindow::DelLayer( mpLayer* layer)
{
        m_layers.DeleteObject( layer);
        UpdateAll();
}

void mpWindow::Refresh(bool eraseBackground, const wxRect* rect)
{
        wxScrolledWindow::Refresh(false);
}


void mpWindow::OnPaint( wxPaintEvent &event )
{
        wxPaintDC dc(this);
        dc.GetSize(&m_scrX, &m_scrY);
printf("EED mpWindow::OnPaint %d %d\n",m_scrX,m_scrY);

        //m_scrX=200;
        //m_scrY=200;

        wxMemoryDC temp_dc;
        //_bitmap_functions->SetWidth(m_scrX);
        //_bitmap_functions->SetHeight(m_scrY);
        _bitmap_functions=new wxBitmap(m_scrX,m_scrY);
        temp_dc.SelectObject(*_bitmap_functions);

        // Background
        //      wxColour colourParent =  wxColour(255,0,0);
        temp_dc.SetBrush(wxBrush( *wxWHITE_BRUSH  ) );
        temp_dc.SetPen(wxPen( *wxBLACK_PEN  ));
        temp_dc.DrawRectangle(0,0,m_scrX,m_scrY);
        

        wxNode *node = m_layers.GetFirst();
        while (node)
        {
                ((mpLayer*)node->GetData())->Plot( temp_dc, *this);
                node = node->GetNext();
        }
        
        //dc.SetDeviceOrigin(70,40);
        //dc.SetAxisOrientation(false,true);
        //temp_dc.SetAxisOrientation(true,false);
        
        //EED 14mai2009
        //dc.Blit(0,0, m_scrX, m_scrY, &temp_dc,-70,-m_scrY+40);
        
        temp_dc.SetDeviceOrigin(0,0);
        dc.Blit(0,0, m_scrX, m_scrY, &temp_dc,0,0);
        
        
        delete _bitmap_functions; 
        //_bitmap_functions
        //dc.SetAxisOrientation(false,true);


}

/*
void mpWindow::OnPaint( wxPaintEvent &event )
{
        wxPaintDC dc(this);
        
        //dc.BeginDrawing();
        
        dc.GetSize(&m_scrX, &m_scrY);
        
        //const int orgy   = m_scrY-40;
        //dc.SetDeviceOrigin(70,orgy);
        //dc.SetAxisOrientation(true,true);

//      wxMemoryDC temp_dc;
//      
//      //_bitmap_functions=new wxBitmap(m_scrX,m_scrY);
//      
//      _bitmap_functions->SetWidth(100);//m_scrX);
//      _bitmap_functions->SetHeight(200);//m_scrY);
//      
//      temp_dc.SelectObject(*_bitmap_functions);
//      
//      // Background
//      wxColour colourParent =  wxColour(255,0,0);
//      temp_dc.SetBrush(wxBrush( colourParent ,wxSOLID  ));
//      temp_dc.SetPen(wxPen( colourParent ,1,wxSOLID  ));
//      temp_dc.DrawRectangle(0,0,100,200);
//      dc.Blit(0,0, 100, 200, &temp_dc,0,0);
//      //dc.SetDeviceOrigin( m_scrX>>1, m_scrY>>1);

        wxNode *node = m_layers.GetFirst();
        while (node)
        {
                ((mpLayer*)node->GetData())->Plot( dc, *this);
                node = node->GetNext();
        }
        //temp_dc.SetAxisOrientation(true,false);
        //dc.EndDrawing();
        // dc.SetAxisOrientation(true,true);
         //dc.Blit(70,(int)(-m_scrY+40),m_scrX-100,m_scrY-50,&temp_dc,0,0);
}
*/

void mpWindow::OnScroll2(wxScrollWinEvent &event)
{
        int width, height;
        GetClientSize( &width, &height);
        int px, py;
        GetViewStart( &px, &py);

        if (event.GetOrientation() == wxHORIZONTAL)
        {
                SetPosX( (double)px / GetScaleX() + m_minX + (double)(width>>1)/GetScaleX());
        }
        else
        {
                SetPosY( m_maxY - (double)py / GetScaleY() - (double)(height>>1)/GetScaleY());
        }
        event.Skip();
}

bool mpWindow::UpdateBBox()
{
        bool first = TRUE;

        wxNode *node = m_layers.GetFirst();

        while(node)
        {
                mpLayer* f = (mpLayer*)node->GetData();

                if (f->HasBBox())
                {
                        if (first)
                        {
                                first = FALSE;
                                m_minX = f->GetMinX(); m_maxX=f->GetMaxX();
                                m_minY = f->GetMinY(); m_maxY=f->GetMaxY();
                        }
                        else
                        {
                                if (f->GetMinX()<m_minX) m_minX=f->GetMinX(); if (f->GetMaxX()>m_maxX) m_maxX=f->GetMaxX();
                                if (f->GetMinY()<m_minY) m_minY=f->GetMinY(); if (f->GetMaxY()>m_maxY) m_maxY=f->GetMaxY();
                        }
                }
                node = node->GetNext();
        }

        return first == FALSE;
}

void mpWindow::UpdateAll()
{
        if (UpdateBBox())
        {
                int cx, cy;
                GetClientSize( &cx, &cy);
printf("EED mpWindow::UpdateAll %d %d\n",cx,cy);

                //const int sx = (int)((m_maxX - m_minX) * GetScaleX()); // JPRx
                //const int sy = (int)((m_maxY - m_minY) * GetScaleY()); // JPRx
                //const int px = (int)((GetPosX() - m_minX) * GetScaleX() - (cx>>1)); // JPRx
                //const int py = (int)((GetPosY() - m_minY) * GetScaleY() - (cy>>1)); // JPRx
                //SetScrollbars( 1, 1, sx, sy, px, py);
        }

        FitInside();
        Refresh( false );
}

/*
* Guide lines menu handler method that reacts to the mpID_LINE_GUIDES cimmand event
* event The corresponding event to handle
*/
/*
void mpWindow :: OnGuideLines (wxCommandEvent   &event)
{
        wxString nextAction_text;
        if( drawGuides )
        {
                setLineGuidesCondition(false);
                nextAction_text << "Turn on guide lines";
                UpdateAll();
        }
        else
        {
                setLineGuidesCondition(true);
                nextAction_text << "Turn off guide lines";
                UpdateAll();
        }
        m_popmenu.SetLabel ( mpID_LINE_GUIDES, nextAction_text ); 
}
*/



bool mpWindow::drawGuideLines()
        {
                return drawGuides;
        }