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;
    View<int**,CLayout,DeviceType> tr;
    View<double***,CLayout,DeviceType> p;
    View<double**,CLayout,DeviceType> dx;
    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
        tr(NoInit("tr"),2,grid.nnode),
        p(NoInit("p"),2,grid.nnode,3),
        dx(NoInit("dx"),2,grid.nnode),
        is_inside(NoInit("is_inside"),grid.nnode)

    {
        Kokkos::deep_copy(tr,grid.one_plane_ff.tr);
        Kokkos::deep_copy(p,grid.one_plane_ff.p);
        Kokkos::deep_copy(dx,grid.one_plane_ff.dx);

        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
        tr(NoInit("tr"),2,grid.nnode),
        p(NoInit("p"),2,grid.nnode,3),
        dx(NoInit("dx"),2,grid.nnode),
        is_inside(NoInit("is_inside"),grid.nnode)

    {
        Kokkos::deep_copy(tr,grid.one_plane_ff.tr);
        Kokkos::deep_copy(p,grid.one_plane_ff.p);
        Kokkos::deep_copy(dx,grid.one_plane_ff.dx);

        // 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 Grid<DeviceType>& grid, const GradParXTmp& gptx, const View<double**,CLayout,DeviceType>& input, const View<double**,CLayout,DeviceType>& output, Order derivative_order);

#endif