s2_kin_trans.hpp

/* s2_kin_trans.hpp */
/* $Id: s2_kin_trans.hpp 21640 2012-06-09 13:05:14Z guser1 $ */
#ifndef __s2_kin_trans_hpp
#define __s2_kin_trans_hpp

#ifdef GE_BUILD
#include <wave_field.hpp>
#include <least_squares_2d.hpp>
#else
#include <s2proc/wave_field.hpp>
#include <mth/least_squares_2d.hpp>
#endif

namespace S2KinTrans
{
  class AbstractKinShift
  {
  public:
    
    AbstractKinShift(bool dir = true) : m_dir(dir){};
    
    virtual ~AbstractKinShift(){};
    
    double operator()(double L) const {return calc(L);} 
    
    void setDir(bool dir) {m_dir = dir;}
    
    bool getDir() const {return m_dir;}
    
    virtual double getHodPoint(double L) const = 0;
    
  protected:
    
    virtual double calc(double L) const = 0;
    
    bool m_dir;
    
  };
  
  class HypKinShift : public AbstractKinShift
  {
  public:
    
    HypKinShift(double t0, double p, bool dir = true); 
    
    void setT0(double t0) {m_t0 = t0;}
    
    double getT0() {return m_t0;}
    
    void setP(double p) {m_p = p;}
    
    double getP() {return m_p;}
    
    virtual double getHodPoint(double L) const;
    
    void setLockedParams(double t0, double L);
    
  protected:
    
    double m_t0, m_p;
    
    virtual double calc(double L) const;
    
  private:
    
    HypKinShift();
    
    double m_locked_t0, m_locked_L;
  };
  
  class AVelKinShift : public AbstractKinShift
  {
  public:
    
    AVelKinShift(double t0, double v, bool dir = true); 
    
    virtual double getHodPoint(double L) const;
    
    void setT0(double t0) {m_t0 = t0;}
    
    double getT0() {return m_t0;}
    
    void setVel(double v) {m_v = v;}
    
    double getVel() {return m_v;}
    
    void setLockedParams(double t0, double L);
    
  protected:
    
    double m_t0, m_v;
    
    virtual double calc(double L) const;
        
  private:
    
    AVelKinShift();
    
    double m_locked_t0, m_locked_L;
  };
  
  class VelKinShift : public AbstractKinShift
  {
  public:
    
    VelKinShift(double t0, double v, double L0, bool dir = true); 
    
    virtual double getHodPoint(double L) const;
    
    void setT0(double t0) {m_t0 = t0;}
    
    double getT0() {return m_t0;}
    
    void setL0(double L0) {m_L0 = L0;}
    
    double getL0() {return m_L0;}
    
    void setVel(double v) {m_v = v;}
    
    double getVel() {return m_v;}
        
  protected:
    
    double m_t0, m_v, m_L0;
    
    virtual double calc(double L) const;
    
  private:
    
    VelKinShift();
  };
  
  class PolyKinShift : public AbstractKinShift
  {
  public:
    
    PolyKinShift(const LeastSquares2D::PolyCoeffs &coeffs, bool dir = true); 

    virtual double getHodPoint(double L) const;
    
  protected:
    
    LeastSquares2D::PolyCoeffs m_coeffs;
    
    virtual double calc(double L) const;
    
  private:
    
    PolyKinShift();
  };
    
  void transform(const AbstractKinShift &kin_shift, WaveField &field, bool offsets_from_coords = false);
};

#endif /* s2_kin_trans.hpp */