SymbolPointRep.cxx

Go to the documentation of this file.

#ifdef _MSC_VER
// include max() and min() missing from Microsoft Visual C++
#include "msdevstudio/MSconfig.h"
#endif //_MSC_VER

#include "SymbolPointRep.h"

#include "ErrorBarRep.h"

#include "datasrcs/DataPointTuple.h"
#include "datasrcs/DataSource.h"
#include "graphics/ViewBase.h"

#include "projectors/NTupleProjector.h"
#include "transforms/PeriodicBinaryTransform.h"

#include <cassert>
#include <cmath>

using namespace hippodraw;

using std::string;
using std::vector;

SymbolPointRep::
SymbolPointRep ( hippodraw::Symbol::Type symbol, float size )
  : PointRepBase ( "Symbol", size ),
    m_plot_symbol( symbol )
{
  m_error_rep = new ErrorBarRep();
}

SymbolPointRep::SymbolPointRep ( )
  : PointRepBase ( "Symbol", 2.0 ),
    m_plot_symbol( Symbol::SOLIDSQUARE )
{
  m_error_rep = new ErrorBarRep();
}

SymbolPointRep::SymbolPointRep ( const SymbolPointRep & point_rep )
  : PointRepBase( point_rep ), 
  m_x( point_rep.m_x ),
  m_y( point_rep.m_y ),
  m_x_flag( point_rep.m_x_flag ),
  m_y_flag( point_rep.m_y_flag ),
  m_plot_symbol( point_rep.m_plot_symbol )
{
  RepBase * clone = point_rep.m_error_rep->clone();
  m_error_rep = dynamic_cast< ErrorBarRep * > ( clone );
}

SymbolPointRep::~SymbolPointRep()
{
  delete m_error_rep;
}

RepBase * SymbolPointRep::clone()
{
  return new SymbolPointRep( *this );
}

void SymbolPointRep::setColor ( const Color & color )
{
  RepBase::setColor ( color );

  if ( m_error_rep ) {
    m_error_rep->setColor ( color );
  }
}

void SymbolPointRep::setStyle ( unsigned int symbol )
{
  m_plot_symbol = Symbol::convert ( symbol );
}

unsigned int
SymbolPointRep::
getStyle ( ) const
{
  return m_plot_symbol;
}

void
SymbolPointRep::
setErrorOn ( const std::string & axis, bool flag )
{
  if ( axis == "x" || axis == "X" ) {
    m_error_rep->setXError ( flag );
    m_x_flag = flag;
  }
  if ( axis == "y" || axis == "Y" ) {
    m_error_rep->setYError ( flag );
    m_y_flag = flag;
  }
}

bool SymbolPointRep::xError () const
{
  return m_x_flag;
}

bool SymbolPointRep::yError () const
{
  return m_y_flag;
}

void SymbolPointRep::transformPoints ( const TransformBase & tb,
                                       ViewBase & vb )
{
  try {
    const BinaryTransform & tf 
      = dynamic_cast < const BinaryTransform & > ( tb );

    tf.transform ( m_x, m_y );
  }
  catch ( ... ) {
    assert ( false );
  }
}

void 
SymbolPointRep::
rotate ( double & lat, double & lon,
         double alpha, double beta, double gamma )
{
  // convert to radians from degrees
  lat = M_PI * lat / 180.0; 
  lon = M_PI * lon / 180.0;
  alpha = M_PI * alpha / 180.0;
  beta  = M_PI * beta  / 180.0;
  gamma = M_PI * gamma / 180.0; 
    
  // Convert the latitude and longitudes into the cartesian coordinates
  // Assume a unit sphere ( Radii does not matter in rotation )
  double x = cos( lat ) * cos( lon );
  double y = sin( lat ) * cos( lon );
  double z = sin( lon );
  
  // First give a rotation about the x axis ( conventionally denoted by alpha )
  double rx =   x;
  double ry =   y * cos( alpha ) + z * sin( alpha );
  double rz = - y * sin( alpha ) + z * cos( alpha );
  
  x = rx; y = ry; z = rz;
  
  // Now give a rotation about the y axis ( conventionally denoted by beta )
  rx =   x * cos( beta ) - z * sin( beta );
  ry =   y;
  rz =   x * sin( beta ) + z * cos( beta );
  
  x = rx; y = ry; z = rz;
  
  // Now give a rotation about the z axis ( conventionally denoted by gamma )
  rx =   x * cos( gamma ) + y * sin( gamma );
  ry = - x * sin( gamma ) + y * cos( gamma );
  rz =   z;
  
  x = rx; y = ry; z = rz;
  
  // Convert back to latitude and longitudes ( in degrees )
  lon = ( 180.0 / M_PI ) * asin( z );
  lat = ( 180.0 / M_PI ) * atan2( y, x );
  
}

namespace dp = hippodraw::DataPoint2DTuple;

void
SymbolPointRep::
drawProjectedValues ( const DataSource * ntuple,
                      TransformBase * transform,
                      ViewBase * view )
{
  unsigned int size = ntuple -> rows ();
  if ( size == 0 ) return;

  const BinaryTransform * tf 
    = dynamic_cast < BinaryTransform * > ( transform );
  assert ( tf != 0 );
  beginPlot ( size, tf, view );

  for ( unsigned int i = 0; i < size; i++ ) {
    drawProjectedValue ( i, ntuple, tf, view );
  }  

  const Color & cur_color = color ();
  view -> drawPoints ( m_x, m_y, m_plot_symbol, m_size, cur_color );

  if ( m_x_flag || m_y_flag ) {
    m_error_rep -> drawProjectedValues ( ntuple, transform, view );
  }
}

void
SymbolPointRep::
beginPlot ( unsigned int size,
            const BinaryTransform * tf, 
            const ViewBase * view )
{
  m_x.clear ();
  m_y.clear ();

  m_x.reserve ( size );
  m_y.reserve ( size );

  c_x_range = view -> getRange ( Axes::X );
  c_y_range = view -> getRange ( Axes::Y );

  if ( tf -> isPeriodic () ) {
    c_periodic_tf = dynamic_cast < const PeriodicBinaryTransform * > ( tf );
    c_beta = c_periodic_tf -> xOffset ();
    c_gamma = c_periodic_tf -> yOffset ();
  }
}

bool
SymbolPointRep::
drawProjectedValue ( unsigned int i, 
                     const DataSource * ntuple, 
                     const BinaryTransform * tf, 
                     ViewBase * view )
{
  bool retval = false;

  const vector < double > & row = ntuple -> getRow ( i );
  double x = row [ dp::X ];
  double y = row [ dp::Y ];

  double xtemp = x;
  double ytemp = y;

  if ( tf -> isPeriodic() )
    {
      xtemp = c_periodic_tf  -> moduloSubX ( x, c_beta );
      ytemp = c_periodic_tf  -> moduloSubY ( y, c_gamma );
      
      rotate ( x, y, 0.0, c_beta, c_gamma );
    }

  const Range & xrange = view -> getRange ( Axes::X );
  const Range & yrange = view -> getRange ( Axes::Y );

  if ( xrange.includes ( xtemp ) && 
       yrange.includes ( ytemp ) ) 
    {
      tf -> transform( x, y );
      m_x.push_back ( x );
      m_y.push_back ( y );
      retval = true;
    }

  return retval;
}

---