ImageCorrection.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2018 by Paul Demorest
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/General/Pulsar/ImageCorrection.h
 
#ifndef __Pulsar_ImageCorrection_h
#define __Pulsar_ImageCorrection_h
 
#include "Pulsar/Transformation.h"
 
#include "TextInterface.h"
 
namespace Pulsar {
 
  class Archive;
  class Integration;

  class ImageCorrection : public Transformation<Integration> {
 
  public:

    ImageCorrection ();

    void correct (Archive*);

    void transform (Integration*);

    virtual TextInterface::Parser* get_interface ();

    class Interface;

    void set_strict(bool flag=true) { strict = flag; }

    bool get_strict() const { return strict; }

    void set_coherent(bool flag=true) { coherent = flag; }

    bool get_coherent() const { return coherent; }

    void set_fadc(double f) { f_cent_adc = f; }

    double get_fadc() const { return f_cent_adc; }

    void set_fcenter(double f) { f_cent_sky = f; }

    double get_fcenter() const { return f_cent_sky; }

    void set_band_dir(int d) { dir = d; }

    int get_band_dir() const { return dir; }

    void set_dt(double dt0, double dt1) { dt[0]=dt0; dt[1]=dt1; }

    void set_gain(double a0, double a1) { alpha[0]=a0; alpha[1]=a1; }
 
  protected:

    double f_cent_adc;

    double f_cent_sky;

    int dir;
 
    // Note, all of the above imply that the conversion between
    // sky freq (f_sky) and sampled freq (f_adc) are:
    //   f_adc = dir*(f_sky-f_cent_sky) + f_cent_adc
    double f_adc(double f_sky) const 
    { 
      return (double)dir*(f_sky-f_cent_sky) + f_cent_adc;
    }

    bool strict;

    bool coherent;

    double dt[2];

    double alpha[2];

    //  into channel at 2*f_center - f
    double ratio(double f, unsigned ipol) const;

    int match_channel(const Integration *subint, double f_sky) const;
 
  };
 
}
 
#endif