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