flux_surface_average.hpp

Go to the documentation of this file.

#ifndef FLUX_SURFACE_AVERAGE_HPP
#define FLUX_SURFACE_AVERAGE_HPP

#include "plane.hpp"
#include "volumes_and_areas.hpp"

template<class Device>
class FluxSurfaceAverage{
    Matrix<Device> to_flux_surface;
    Matrix<Device> from_flux_surface;
    View<double*, CLayout, Device> surf_vol;

    void wall_nodes_strikep_method(NLReader::NamelistReader& nlr, const MagneticField<DeviceType>& magnetic_field, const Plane<HostType>& plane, const PlaneFiles& plane_files, const View<double*,CLayout,HostType>& psi_surf, const View<int*,CLayout,DeviceType>& num_t_node, const View<int**,CLayout,DeviceType>& tr_node);

    public:

    FluxSurfaceAverage(){}

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

        m.to_flux_surface = to_flux_surface.template mirror<Device2>();
        m.from_flux_surface = from_flux_surface.template mirror<Device2>();

        m.surf_vol = my_mirror_view(surf_vol, Device2());
        mirror_copy(m.surf_vol, surf_vol);

        return m;
    }

    FluxSurfaceAverage(NLReader::NamelistReader& nlr, const MagneticField<DeviceType>& magnetic_field, const Plane<HostType>& plane, const PlaneFiles& plane_files, const VolumesAndAreas& volumes_and_areas, const View<int*,CLayout,DeviceType>& num_t_node, const View<int**,CLayout,DeviceType>& tr_node);

    void average(const View<double*, CLayout, Device>& array) const;
};


template<class Device>
void flux_surface_average(const View<double*, CLayout, Device>& input_view, const View<double*, CLayout, HostType>& output_view, const View<double*, CLayout, HostType>& tmp00_surf);

template<class Device>
void flux_surface_average(const View<double*, CLayout, Device>& input_view, const View<double*, CLayout, HostType>& output_view);

template<class Device>
void flux_surface_average_in_place(const View<double*, CLayout, Device>& input_view, View<double*, CLayout, HostType>& tmp, View<double*, CLayout, HostType>& tmp00_surf);

template<class Device>
void flux_surface_average_in_place(const View<double*, CLayout, Device>& input_view);

template<class Device>
void flux_surface_average_in_place(const View<double**, CLayout, Device>& input_view);

#endif