wavefront.hpp

 
 
 
#ifndef wavefront_hpp
#define wavefront_hpp
 
#pragma GCC system_header
#include <Eigen/Dense>
 
#include "../../wfp/imagingArray.hpp"
#include "../../wfp/imagingUtils.hpp"
 
 
namespace mx
{
namespace AO
{
namespace sim
{
   
///Structure containing the phase and amplitude of a wavefront   
/** The phase and amplitude are stored separately in real valued arrays.
  */
template<typename _realT>
struct wavefront
{
   ///The floating point type used for all calculations
   typedef _realT realT;
   
   ///The data type of the real image, used to hold the phase and amplitude
   typedef  Eigen::Array<_realT, Eigen::Dynamic, Eigen::Dynamic>  realImageT; 
 
   ///The wavefront data type
   typedef wfp::imagingArray<std::complex<realT>, wfp::fftwAllocator<std::complex<realT> >, 0> complexAmplitudeT;
   
   ///The wavefront amplitude
   realImageT amplitude;
   
   ///The wavefront phase
   realImageT phase;
 
   ///The wavelength at which the wavefront is specified
   realT lambda;
   
   ///The iteration number of this wavefront
   realT iterNo;
   
   ///Zero the wavefront
   void setZero()
   {
      amplitude.setZero();
      phase.setZero();
   }
 
   void setAmplitude(const realImageT & amp)
   {
      amplitude = amp;
   }
   
   void setPhase(const realImageT & ph)
   {
      phase = ph;
   }
   
   void getWavefront(complexAmplitudeT & wf, int wfSz)
   {
      wfp::makeComplexPupil( wf, amplitude, phase, wfSz); 
   }
   
   void getWavefront( complexAmplitudeT & wf, 
                      realT dlambda, 
                      int wfSz
                    )
   {
      realImageT dphase = phase*(lambda/dlambda);
      wfp::makeComplexPupil( wf, amplitude, dphase, wfSz); 
   }
   
   
   
};
 
} //namespace sim
} //namespace AO
} //namespace mx
 
 
#endif