ff_projection.hpp

Go to the documentation of this file.

#ifndef FF_PROJECTION_HPP
#define FF_PROJECTION_HPP

#include "space_settings.hpp"
#include "domain_decomposition.hpp"
#include "plane.hpp"

template<class Device>
struct Projection{
    // This line enables private access between the host and device implementations of this class
    // i.e. Makes the copy from host to device more straightforward
    template<class Device2> friend class Projection;

    enum Method{
        OneStep=0,
        TwoSteps
    };

    bool node_based; 

    View<int**,CLayout,Device> tr; 
    View<int***,CLayout,Device> nd; 
    View<double***,CLayout,Device> p;
    View<double**,CLayout,Device> dx;

    Projection(){}

    Projection(NLReader::NamelistReader& nlr, const MagneticField<DeviceType>& magnetic_field, Method method, bool node_based,
               const Plane<DeviceType>& plane_start0, const Plane<DeviceType>& plane_dest0, double phi_start0, double phi_dest0, const View<int**,CLayout,DeviceType>& tr_node_dest0,
               const Plane<DeviceType>& plane_start1, const Plane<DeviceType>& plane_dest1, double phi_start1, double phi_dest1, const View<int**,CLayout,DeviceType>& tr_node_dest1);

    // Create a mirror with a different device type
    template<class Device2>
    inline Projection<Device2> mirror() const{
        Projection<Device2> m;

        m.node_based = node_based;

        m.tr = my_mirror_view(tr, Device2());
        mirror_copy(m.tr, tr);
        m.nd = my_mirror_view(nd, Device2());
        mirror_copy(m.nd, nd);
        m.p = my_mirror_view(p, Device2());
        mirror_copy(m.p, p);
        m.dx = my_mirror_view(dx, Device2());
        mirror_copy(m.dx, dx);
        return m;
    }

    Projection(const DomainDecomposition<DeviceType>& pol_decomp, const Projection<HostType>& proj_in, int iphi);
};

void ff_write_projections_to_file(const Plane<HostType>& midplane, const Plane<HostType>& lplane, const Plane<HostType>& rplane, const Projection<HostType>& half_plane_ff, const Projection<HostType>& ff_lplane_to_neighbors, const Projection<HostType>& ff_to_midplane);


void check_ff_property(const Projection<HostType>& projection, const std::string label);

#endif