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 = 1;
 
 
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();
 
 
}