boundary.hpp

Go to the documentation of this file.

#ifndef BOUNDARY_HPP
#define BOUNDARY_HPP

#include "field.hpp"

struct rtype{
    int start;
    int end;
};

template<class Device>
class Boundary{

    using exspace = typename Device::execution_space; 

    public:

    View<rtype*,CLayout,Device> iseg;

    // Initialize by copying from a fortran pointer
    Boundary(int nseg, rtype* bd_fort)
      : iseg(NoInit("iseg"),nseg)
    {
        array_deep_copy(iseg,bd_fort);
    }

    KOKKOS_INLINE_FUNCTION bool is_inside(int i) const {
        bool inside = true; // - the other way would be faster -- # of inside nodes > # of boundary nodes

        for (int is=0; is<iseg.size(); is++){
            if( iseg(is).start <= i+1 && i+1 <= iseg(is).end ){
                inside = false;
                break;
            }
        }
        return inside;
    }

    // Note: this only works for Device==HostType right now
    void set_to_value(const View<double*,CLayout,Device>& view, double value) const{
        for (int is=0; is<iseg.size(); is++){
            Kokkos::parallel_for("get_tr_vol_and_area", Kokkos::RangePolicy<exspace>( iseg(is).start-1, iseg(is).end), KOKKOS_LAMBDA( const int i){
                view(i) = value;
            });
        }
        Kokkos::fence();
    }

    // Note: this only works for Device==HostType right now
    // Set to value but for Field rather than double view
    void set_to_value(const View<Field<VarType::Scalar,PhiInterpType::None>*,CLayout,Device>& view, double value) const{
        for (int is=0; is<iseg.size(); is++){
            Kokkos::parallel_for("get_tr_vol_and_area", Kokkos::RangePolicy<exspace>( iseg(is).start-1, iseg(is).end), KOKKOS_LAMBDA( const int i){
                view(i).S = value;
            });
        }
        Kokkos::fence();
    }
};

#endif