uniform_range.hpp

Go to the documentation of this file.

#ifndef UNIFORM_RANGE_HPP
#define UNIFORM_RANGE_HPP

// This struct is where we need to work with a linear range of values, but think it's more efficient
// to recalculate values than to store them.
struct UniformRange{
    int n;
    double val_min;
    double val_max;
    double dval;

    UniformRange(){}

    UniformRange(int n, double val_min, double val_max)
     : n(n),
       val_min(val_min),
       val_max(val_max),
       dval( (val_max-val_min)/(n-1) )
    {}

    KOKKOS_INLINE_FUNCTION double get_val(int i) const{
        return val_min + i*dval;
    }

    KOKKOS_INLINE_FUNCTION bool is_in_bounds(double val) const{
        return (val<=val_max && val>=val_min);
    }

    KOKKOS_INLINE_FUNCTION int get_index(double val) const{
        return (int)((val - val_min)/dval);
    }

    // Generate view containing the linear grid
    template<class Device>
    View<double*,CLayout,Device> generate_view(const std::string& name) const{
        // Make a local copy of *this to pass into lambda
        auto uniform_range = *this;
        View<double*,CLayout,Device> view(NoInit(name), n);

        Kokkos::parallel_for("generate_view", Kokkos::RangePolicy<typename Device::execution_space>( 0,n ), KOKKOS_LAMBDA( const int i){
            view(i) = uniform_range.get_val(i);
        });
        Kokkos::fence();
        return view;
    }
};

#endif