fft.h

Go to the documentation of this file.

#pragma once
#include <functional>
#include <vector>
#include "superarray.h"
#include "raytrace_usp.h"
#include "constants.h"
 
namespace GOAT
{
    namespace raytracing
    {
#ifndef I 
#define I std::complex<double> (0.0,1.0)
#endif
        constexpr int N_INTEGRAL = 500;

        typedef struct TrafoParms
        {
            int nI=4;                     
            int nS=200;                
            int nR= INEL_MAX_NREFLEX;   
            double omegaStart;              
            double omegaEnd;               
            double omega0;                 
            double wvl=1.0;                 
            double dt=100;                  
            std::vector<std::function<std::complex<double>(double) > > nList; 
            int number_of_threads=5; 
        } ;
 
        
        class Trafo
        {
        public:
            Trafo();
            Trafo(TrafoParms);  
            void calc(std::vector < std::vector<SuperArray <std::vector<gridEntry> > > >& SA, double t); 
            void calc(std::vector<SuperArray <std::vector<gridEntry> > > & SA, double omegaStart, double omegaEnd, double t, bool do_clear=true);
            SuperArray<maths::Vector<std::complex<double> > >SAres; 
            void setRefractiveIndexFunctions(std::vector<std::function<std::complex<double>(double) > > nList); 
            void setReferenceTime(double tref);
            void setTrafoParms(TrafoParms trafoparms);  
            void clear(); 
            void empty(); 
            void initResult(SuperArray<maths::Vector<std::complex<double> >>& SA); 
            void initResult(double r0, INDEX_TYPE nx, INDEX_TYPE ny, INDEX_TYPE nz, std::vector<ObjectShape*> Obj, int numObjs);
            std::vector<std::function<std::complex<double>(double) > > nList; 
            double getD() { return D; } 
            void setNumberOfThreads(int n) {tp.number_of_threads=n;} 
            int getNumberOfThreads () {return tp.number_of_threads;} 
        private:
            double pulseWeight(double omega);
            void setCurrNList(double wvl)
            {
                for (int i = 0; i < nList.size(); i++)
                    currNList[i] = nList[i](wvl);
            }            
        //    void createLTexpo();
            std::complex<double> calcPhase(std::vector<stepEntry> steps, double k0);
            GOAT::maths::Vector<std::complex<double> > calcOne(std::vector<stepEntry> steps, double t);
            GOAT::maths::Vector<std::complex<double> > integrate(double t, std::vector<gridEntry> ge, double omegastart, double omegastop);
            double twoSigma2; 
            double sigma2; 
            double prefactor; // 1/(sigma * sqrt (2pi))
            std::vector<double> freq;
            double tref = 0.0;
            TrafoParms tp;
            std::vector<std::complex<double > > currNList; 
            double D = 0;
            int number_of_threads;
        };
    }
}