fft.hpp

Go to the documentation of this file.

/*
 * This file is part of PLaSK (https://plask.app) by Photonics Group at TUL
 * Copyright (c) 2022 Lodz University of Technology
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, version 3.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 */
#ifndef PLASK__SOLVER_OPTICAL_MODAL_FFT_H
#define PLASK__SOLVER_OPTICAL_MODAL_FFT_H
 
#include <plask/plask.hpp>
 
#include "plask/optical/modal/config.hpp"
 
#ifdef USE_FFTW
#   include <fftw3.h>
#endif
 
namespace plask { namespace optical { namespace modal { namespace FFT {
 
enum Symmetry {
    SYMMETRY_NONE = 0,
    SYMMETRY_EVEN_2 = 1,
    SYMMETRY_ODD_2 = 2,
    SYMMETRY_EVEN_1 = 5,
    SYMMETRY_ODD_1 = 6
};
 
struct PLASK_SOLVER_API Forward1D {
    Forward1D();
    Forward1D(Forward1D&& old);
    Forward1D& operator=(Forward1D&& old);
    Forward1D(std::size_t strid, std::size_t n, Symmetry symmetry);
    ~Forward1D();
    void execute(dcomplex* data, int lot=0);
  private:
    int n;
    int strid;
    Symmetry symmetry;
#ifdef USE_FFTW
    fftw_plan plan;
#else
    double *wsave;
#endif
};
 
struct PLASK_SOLVER_API Forward2D {
    Forward2D();
    Forward2D(Forward2D&& old);
    Forward2D& operator=(Forward2D&& old);
    Forward2D(std::size_t strid, std::size_t n1, std::size_t n2, Symmetry symmetry1, Symmetry symmetry2, std::size_t ld=0);
    ~Forward2D();
    void execute(dcomplex* data, int lot=0);
  private:
    int n1, n2;
    int strid1, strid2;
    Symmetry symmetry1, symmetry2;
#ifdef USE_FFTW
    fftw_plan plan;
#else
    double *wsave1, *wsave2;
#endif
};
 
struct PLASK_SOLVER_API Backward1D {
    Backward1D();
    Backward1D(Backward1D&& old);
    Backward1D& operator=(Backward1D&& old);
    Backward1D(std::size_t strid, std::size_t n, Symmetry symmetry);
    ~Backward1D();
    void execute(dcomplex* data, int lot=0);
  private:
    int n;
    int strid;
    Symmetry symmetry;
#ifdef USE_FFTW
    fftw_plan plan;
#else
    double *wsave;
#endif
};
 
struct PLASK_SOLVER_API Backward2D {
    Backward2D();
    Backward2D(Backward2D&& old);
    Backward2D& operator=(Backward2D&& old);
    Backward2D(std::size_t strid, std::size_t n1, std::size_t n2, Symmetry symmetry1, Symmetry symmetry2, std::size_t ld=0);
    ~Backward2D();
    void execute();
    void execute(dcomplex* data, int lot=0);
  private:
    int n1, n2;
    int strid1, strid2;
    Symmetry symmetry1, symmetry2;
#ifdef USE_FFTW
    fftw_plan plan;
#else
    double *wsave1, *wsave2;
#endif
};
 
}}}} // namespace plask::optical::modal::FFT
 
#endif // PLASK__SOLVER_OPTICAL_MODAL_FFT_H