gradparx2.hpp

Go to the documentation of this file.

#ifndef GRADPARX2_HPP
#define GRADPARX2_HPP

#include "globals.hpp"
#include "grid.hpp"
#include "boundary.hpp"

struct GradParXTmp{
    double sgn;
    Projection<DeviceType> lplane_to_neighbors;
    Projection<DeviceType> rplane_to_neighbors;
    View<bool*,CLayout,DeviceType> is_inside;

    GradParXTmp(){}

    GradParXTmp(const Grid<DeviceType>& grid, int nseg, double sml_bt_sign, rtype* bd_fort)
      : sgn(-(sml_bt_sign<0.0 ? -1.0 : 1.0)), // Reverse sign since negative gradient is desired
        lplane_to_neighbors(grid.ff_lplane_to_neighbors.template mirror<DeviceType>()),
        rplane_to_neighbors(grid.ff_rplane_to_neighbors.template mirror<DeviceType>()),
        is_inside(NoInit("is_inside"),grid.nnode)

    {
        Boundary<DeviceType> bd(nseg, bd_fort);

        // get is_inside for every node
        set_is_inside(bd, is_inside);
    }

    // For testing - set is_insde=.true.
    GradParXTmp(const Grid<DeviceType>& grid, double sml_bt_sign)
      : sgn(-(sml_bt_sign<0.0 ? -1.0 : 1.0)), // Reverse sign since negative gradient is desired
        lplane_to_neighbors(grid.ff_lplane_to_neighbors.template mirror<DeviceType>()),
        rplane_to_neighbors(grid.ff_rplane_to_neighbors.template mirror<DeviceType>()),
        is_inside(NoInit("is_inside"),grid.nnode)

    {
        // Assume all nodes are inside
        Kokkos::deep_copy(is_inside, true);
    }

    void set_is_inside(const Boundary<DeviceType>& bd, const View<bool*,CLayout,DeviceType>& is_inside_local){
        Kokkos::parallel_for("gradparx2", Kokkos::RangePolicy<ExSpace>(0,is_inside_local.extent(0)), KOKKOS_LAMBDA( const int i ){
            is_inside_local(i) = bd.is_inside(i);
        });
    }
};

void negative_gradparx2(const GradParXTmp& gptx, const View<double**,CLayout,DeviceType>& input, const View<double**,CLayout,DeviceType>& output, Order derivative_order);

#endif