s2_corr.hpp

/* s2_corr.hpp */
/* $Id: s2_corr.hpp 21237 2011-11-15 10:22:54Z guser1 $ */
#ifndef __s2_corr_hpp
#define __s2_corr_hpp

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

#include <stdio.h>

namespace S2Corr
{
  typedef std::vector<Trace::AmplT> DVector;
  
  typedef std::vector<DVector> DMatrix;
  
  
  template <template<class> class Functor>
  bool traceBaseOp(Trace &tr, const DVector &bases);
  
  
  bool traceWeightSum(const Trace &tr1, 
              const Trace &tr2, 
              Trace::AmplT &koeff);
  
  
  bool traceAmplKoeffs(const Trace &tr1, 
               const Trace &tr2,
               DVector &koeffs);
  
    
  bool polycorPrepare(const WaveField &field, DMatrix &C);
  
  
  bool polycorPrepare(const WaveField &field, const Trace &tr, DVector &C);
  
  bool polycorPrepare(const WaveField &field, const Trace &trs, size_t wsize, Trace &trr);
    
  bool tracesIntercorr(const Trace &tr1, const Trace &tr2, DVector &corr);
  
  bool tracesIntercorr(const Trace &tr1, const Trace &tr2, size_t wsize, Trace &tr);

  bool GetDispersion(const Trace &trs, float &disp);

  template <typename FFT_T>
  bool tracesCCF(FFT_T &fft, 
         const Trace &tr1, 
         const Trace &tr2, 
         Trace &ccf,        
         bool norma);       
  template <typename FFT_T>
  bool traceACF(FFT_T &fft,      
        const Trace &tr, 
        Trace &acf,      
        bool norma);     
  template <typename FFT_T>
  bool SpectrWidth(FFT_T &fft, 
           const Trace &tr, 
           float &sw_val);
  

  bool diff(Trace &trace);
  
  bool fieldEnergy(const WaveField &field, DMatrix &E);

  bool fieldEnergy(const WaveField &field, size_t wsize, WaveField &E);
    
  bool traceEnergy(const Trace &tr, DVector &e);
  
  bool traceEnergy(const Trace &tr, size_t wsize, Trace &e);
  
  bool intercorr(size_t n,               
         const Trace::AmplT *x,  
         const Trace::AmplT *y,  
         Trace::AmplT       &r   
         );
  
  bool traceAutocorr(const Trace &tr_in, 
             size_t base, 
             Trace &tr_out
             );
  
  bool autocorr(size_t n, 
        const Trace::AmplT *x, 
        std::vector<Trace::AmplT> &y,
        size_t acf_size=0
        );
  
  bool energy(size_t n,                
          const Trace::AmplT *x,   
          Trace::AmplT &e,         
          bool norm_flag = false   
          ); 
  

  bool sim(size_t n,               
       const Trace::AmplT *x,  
       const Trace::AmplT *y,  
       Trace::AmplT       &s   
       );
  
  bool modsum(size_t n,                
          const Trace::AmplT *x,   
          Trace::AmplT &e          
          ); 

};

#ifdef GE_BUILD
#include <s2_corr_inline.hpp>
#else
#include <s2proc/s2_corr_inline.hpp>
#endif



#endif /* s2_corr.hpp */