s2_fk45migr.hpp

/* s2_fk45migr.hpp */
/* $Id: s2_fk45migr.hpp 21022 2011-07-11 11:45:03Z urij $ */
#ifndef __s2_fk45migr
#define __s2_fk45migr

#include <complex>
#include <vector>

#include <s2proc/s2_abstractmigr.hpp>
//#include <s2proc/s2_fk45migr_rp.hpp>
#include <vel2dmod/vel2dmod.hpp>

class S2FK45Migr : public S2AbstractMigr
{
public:
  
  S2FK45Migr(float spx);
  
  virtual ~S2FK45Migr();
  
  virtual bool getImage(const WaveField &wf_up, 
            Vel2dModel &mod2d,      
            WaveField &image        
            ) const;
  
  void setSp(float spx) 
  {m_spx = spx;} 
  
protected:
  
  struct CField
  {
    std::vector< std::vector< std::complex<float> > > ar;
    float w0, dw;
    float k0, dk;
  };
  
  float m_spx;
    
  void shift(size_t img_id,          /*< id of image z-level */
         const WaveField &image, /*< image field */
         CField &up,             /*< complex spectrum for current z-level */
         Vel2dModel &mod2d,      /*< 2D velocity model */
         float dz                /*< shift */
         ) const;
  
  virtual std::complex<float> migr_step(std::complex<float> u, 
                    float w, 
                    float k, 
                    float v, 
                    float dz
                    ) const;
  
  void convolution(CField &up, std::vector< std::complex<float> > &img) const;
    
  virtual bool fft2d(const WaveField &wf, S2FK45Migr::CField &cf) const;
  
  virtual bool ifft1d(const std::vector< std::complex<float> > &img, 
              Trace &tr_re, Trace &tr_im) const;
  
private:
  
  void test_save_wf(CField &cwf, const std::string &name, int nmgl, bool create) const;
};

#endif /* s2_fk45migr.hpp */