grid_structs.hpp

Go to the documentation of this file.

#ifndef GRID_STRUCTS_HPP
#define GRID_STRUCTS_HPP

#include "simd.hpp"

// The three points of a grid triangle
struct SimdGridVec{
    Simd<double> x[3];

    // Allows array-like access pattern
    KOKKOS_INLINE_FUNCTION const Simd<double>& operator [](int i) const {return x[i];}
    KOKKOS_INLINE_FUNCTION Simd<double>& operator [](int i) {return x[i];}    

    KOKKOS_INLINE_FUNCTION void maxloc(Simd<int> &ml) const {
        for (int i_simd = 0; i_simd<SIMD_SIZE; i_simd++){
            ml[i_simd] = (x[1][i_simd] > x[0][i_simd]) ? 1 : 0;
            ml[i_simd] = (x[2][i_simd] > x[0][i_simd] && x[2][i_simd] > x[1][i_simd]) ? 2 : ml[i_simd];
        }
    }
};

struct Vertex {
    int vertex[3];
};

struct RZPair {
    double r;
    double z;
};

struct VertexMap {
    RZPair vertex[3];
};

#endif