line_of_sight.hpp

Go to the documentation of this file.

#ifndef LINE_OF_SIGHT_HPP
#define LINE_OF_SIGHT_HPP

#include "space_settings.hpp"
#include "plane.hpp"
#include "magnetic_field.hpp"
#include "grid_field.hpp"
#include <cmath>

class LineOfSight {
    bool valid;       
    double r0;        
    double z0;        
    double phi0;      
    double theta;     
    double step_size; 
    int samples;      

    public:
    //
    LineOfSight() : valid(false), r0(0.0), z0(0.0), phi0(0.0), theta(0.0), step_size(0.0), samples(0) {}

    // Constructor
    LineOfSight(const Plane<DeviceType>& plane, const double& r0_in, const double& z0_in,
                const double& theta_in, const double& step_size_in);

    //
    // Accessor functions
    //
    double get_r0() const { return r0; }
    double get_z0() const { return z0; }
    double get_phi0() const { return phi0; }
    double get_theta() const { return theta; }
    int get_samples() const { return samples; }
    bool is_valid() const { return valid; }


    // Get (R,Z) coordinates of the sample points
    void sample_points(const MagneticField<DeviceType>& magnetic_field,
            View<double*,CLayout,DeviceType> r_los, View<double*,CLayout,DeviceType> z_los,
                       View<double*,CLayout,DeviceType> psi_los) const;

    // Interpolates input field to single sample point on line-of-sight
    KOKKOS_INLINE_FUNCTION double interpolate_single(const Plane<DeviceType>& plane, const Kokkos::View<double*,CLayout,DeviceType>& field_in,
                                                     const int i) const;

    // Interpolates input field to single sample point on line-of-sight
    KOKKOS_INLINE_FUNCTION double interpolate_single(const Plane<DeviceType>& plane, const ScalarGridField& field_in,
                                                     const int i) const;

    // Interpolate ScalarGridField along line-of-sight
    View<double*,CLayout,DeviceType> interpolate_to_los(const Plane<DeviceType>& plane,
                                                        const ScalarGridField& field_in) const;
    
    // Interpolate a 1D View along line-of sight
    Kokkos::View<double*, CLayout, DeviceType> interpolate_to_los(const Plane<DeviceType>& plane,
                                                                  const Kokkos::View<double*, CLayout, DeviceType>& field_in) const;

};
#endif