ReceptionModelAxisPlotter.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2004 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/Polarimetry/Pulsar/ReceptionModelAxisPlotter.h
 
#ifndef __Calibration_ReceptionModelAxisPlotter_H
#define __Calibration_ReceptionModelAxisPlotter_H
 
#include "Pulsar/ReceptionModelPlotter.h"
#include "MEAL/Axis.h"
#include "Pauli.h"
 
#include <cpgplot.h>
 
namespace Calibration {

  template<class Type>
  class ReceptionModelAxisPlotter : public ReceptionModelPlotter
  {
 
  public:
 
    ReceptionModelAxisPlotter () { axis = 0; npt = 100; }

    void set_axis (MEAL::Axis<Type>* _axis) { axis = _axis; }

    void set_npt (unsigned _npt) { npt = _npt; }

    void set_min (Type _min) { min = _min; }

    void set_max (Type _max) { max = _max; }

    void plot_model ();
 
  protected:

    MEAL::Axis<Type>* axis;

    unsigned npt;

    Type min;

    Type max;
 
  };
 
}
 
template<typename T>
float cast_to_float (T t)
{
  return (float) t;
}
 
float cast_to_float (const MJD& t)
{
  return t.in_days();
}

template<class Type>
void Calibration::ReceptionModelAxisPlotter<Type>::plot_model ()
{
  Type step = (max - min) / (npt - 1);
 
  model->set_transformation_index (ipath);
  model->set_input_index (isource);
 
  for (unsigned ipt=0; ipt<npt; ipt++) try
  {
    Type index = min + step * ipt;
    float val = cast_to_float (index);
 
    axis->set_value (index);
 
    Stokes<double> stokes = coherency( model->evaluate () );
 
    if (ipt == 0)
      cpgmove (val, stokes[ipol]);
    else
      cpgdraw (val, stokes[ipol]);
    
  }
  catch (Error& error)
  {
#ifdef _DEBUG
    std::cerr << "Calibration::ReceptionModelAxisPlotter<Type>::plot_model "
                 "ipt=" << ipt << " " << error.get_message() << std::endl;
#endif
  }
}
 
 
#endif