quantum3body_simulation.hh

/* vim: set sw=4 sts=4 ft=cpp et foldmethod=syntax : */

/*
 * Copyright (c) 2011 Tiziano Müller <tm@dev-zero.ch>
 *                    Christian Reinhardt
 *
 *
 *
 */

#ifndef QUANTUM3BODY_SIMULATION_HH
#define QUANTUM3BODY_SIMULATION_HH

#include <cstring>
#include <vector>
#include <functional>
#include <complex>
// make sure <complex> is included before fftw3.h
// to have FFTW use the datatype defined by gcc's complex
// rather than it's own
#include <fftw3.h>

typedef std::complex<double> complex;

class Quantum3BodySimulation
{
public:
    typedef std::function<double (const double&, const double&)> PotentialFunction;
    typedef std::function<void (complex&, const double&, const double&)> EvolutionFunction;

    Quantum3BodySimulation(size_t xSize, size_t ySize);
    ~Quantum3BodySimulation();

    void setInitial(std::function<complex (const double&, const double&)> initialPhi);
    void setPotential(PotentialFunction V);
    void setTimeEvolution(EvolutionFunction spatial, EvolutionFunction kx_y_space, EvolutionFunction x_ky_space);

    void useDefaultEvolution();
    void useQuantum3BodyEvolution();

    double binSizeX() const;
    double binSizeY() const;

    void evolveStep(const double& dt);

    const std::vector<complex>& f_spatial() const { return _spatial; }
    const std::vector<complex>& f_momentum() const { return _momentum; }

    const std::vector<double>& x() const { return _x; }
    const std::vector<double>& y() const { return _y; }
    const std::vector<double>& kx() const { return _kx; }
    const std::vector<double>& ky() const { return _ky; }

private:
    size_t _sizeX, _sizeY;
    PotentialFunction _V;
    std::vector<double> _x, _y, _kx, _ky;
    std::vector<complex> _spatial, _momentum;
#if 1
    fftw_plan _fftPlanForwardX, _fftPlanBackwardX;
    fftw_plan _fftPlanForwardY, _fftPlanBackwardY;
#else
    fftw_plan _fftPlanForward, _fftPlanBackward;
#endif

    bool _useDefaultTimeEvolution;
 
};

#endif // QUANTUM3BODY_SIMULATION_HH