main.cpp

#include "Grid/grid_dist_id.hpp"
#include "data_type/aggregate.hpp"
#include "Decomposition/CartDecomposition.hpp"


constexpr size_t x = 0;
constexpr size_t y = 1;
constexpr size_t z = 2;

constexpr size_t A = 0;
constexpr size_t B = 0;


int main(int argc, char* argv[])
{

    openfpm_init(&argc,&argv);

    // domain
    Box<3,float> domain({0.0,0.0,0.0},{1.0,1.0,1.0});

    // grid sizes
    size_t sz[3] = {100,100,100};

    // ghost extension
    Ghost<3,float> g(0.03);



    // a convenient alias for aggregate<...>
    typedef aggregate<float,float> grid_point;

    grid_dist_id<3, float, grid_point> g_dist(sz,domain,g);



    auto dom = g_dist.getDomainIterator();

    while (dom.isNext())
    {



        auto key = dom.get();



        auto key_g = g_dist.getGKey(key);



        g_dist.template get<A>(key) = key_g.get(0)*key_g.get(0) + key_g.get(1)*key_g.get(1) + key_g.get(2)*key_g.get(2);



        ++dom;
    }



    g_dist.template ghost_get<A>();
    


    auto dom2 = g_dist.getDomainIterator();
    
    while (dom2.isNext())
    {
        auto key = dom2.get();

        // Laplace stencil
        g_dist.template get<B>(key) = g_dist.template get<A>(key.move(x,1)) + g_dist.template get<A>(key.move(x,-1)) +
                                         g_dist.template get<A>(key.move(y,1)) + g_dist.template get<A>(key.move(y,-1)) +
                                         g_dist.template get<A>(key.move(z,1)) + g_dist.template get<A>(key.move(z,-1)) -
                                         6*g_dist.template get<A>(key);

        ++dom2;
    }



    g_dist.write("output");



    openfpm_finalize();


}