scene_Plot1D.cpp

Go to the documentation of this file.

#include "scene_Plot1D.hpp"
#include "opengl.hpp"

Plot1D::Plot1D()
: mEnabled(true)
, mPosition( Point3d(0,0,0) )
, mColor( Vec4d(1.0,1.0,1.0,1.0) )
, mSteps(100)
, mXLimits( Vec2d(-1,1) )
, mXAxis(0)
, mYAxis(1)
, mFunc1D(0)
, mYScale(0)
{
}

void Plot1D::setEnabled(bool e)
{
  mEnabled = e;
}
void Plot1D::setPosition(const Point3d& p)
{
  mPosition = p;
}
void Plot1D::setColor(const Vec4d& c)
{
  mColor = c;
}
void Plot1D::setSteps(int steps)
{
  mSteps = steps;
}
void Plot1D::setXLimits(const Vec2d& xlimits)
{
  mXLimits = xlimits;
}
void Plot1D::setXAxis(int xaxis)
{
  mXAxis = xaxis;
}
void Plot1D::setYAxis(int yaxis)
{
  mYAxis = yaxis;
}
void Plot1D::setFunction(const boost::function1<double,double>& f1d)
{
  mFunc1D = f1d;
}
void Plot1D::setYScale(double yscale)
{
  mYScale = yscale;
}

void Plot1D::display()
{
  if (mEnabled)
  {
    if (mFunc1D)
    {
      glPushAttrib(GL_ENABLE_BIT|GL_LINE_BIT);

      if (mColor[3] != 1.0)
        glEnable(GL_BLEND);

      gl_set_color(mColor);
      glLineWidth(3.0);

      glBegin(GL_LINE_STRIP);
      double unit = ( (mXLimits[1]-mXLimits[0]) / (mSteps-1) );
      for (int ix=0; ix < mSteps ; ++ix)
      {
        double x = mXLimits[0] + unit * ix;    
        double y = mFunc1D(x);

        Point3d p = mPosition;
        p[mXAxis] = x;
        p[mYAxis] = y * mYScale;
        glVertex3d(p[0],p[1],p[2]);
      }
      glEnd();

      glPopAttrib();
    }
  }
}