grid_dist_id_HDF5_chckpnt_restart_test.hpp

/*
 * grid_dist_id_HDF5_chckpnt_restart_test.hpp
 *
 *  Created on: Nov 9, 2016
 *      Author: Yaroslav Zaluzhnyi
 */

#ifndef SRC_GRID_GRID_DIST_ID_HDF5_CHCKPNT_RESTART_TEST_HPP_
#define SRC_GRID_GRID_DIST_ID_HDF5_CHCKPNT_RESTART_TEST_HPP_

#include "Grid/grid_dist_id.hpp"

BOOST_AUTO_TEST_SUITE( gd_hdf5_chckpnt_rstrt_test )

BOOST_AUTO_TEST_CASE( grid_dist_id_hdf5_save_test )
{

    // Input data
    size_t k = 2400;

    float ghost_part = 0.0;

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

    Vcluster & v_cl = create_vcluster();

    // Skip this test on big scale
    if (v_cl.getProcessingUnits() >= 32)
        return;

    // grid size
    size_t sz[2];
    sz[0] = k;
    sz[1] = k;

    // Ghost
    Ghost<2,float> g(ghost_part);

    // Distributed grid with id decomposition
    grid_dist_id<2, float, scalar<float>, CartDecomposition<2,float>> g_dist(sz,domain,g);

    // get the decomposition
    auto & dec = g_dist.getDecomposition();

    // check the consistency of the decomposition
    bool val = dec.check_consistency();
    BOOST_REQUIRE_EQUAL(val,true);

    size_t count = 0;

    auto it = g_dist.getDomainIterator();

    while (it.isNext())
    {
        //key
        auto key = it.get();

        auto keyg = g_dist.getGKey(key);

        g_dist.template get<0>(key) = keyg.get(0);

        ++it;
        count++;
    }

    openfpm::vector<size_t> count_total;
    v_cl.allGather(count,count_total);
    v_cl.execute();

    size_t sum = 0;

    for (size_t i = 0; i < count_total.size(); i++)
        sum += count_total.get(i);

    timer t;
    t.start();
    // Save the grid
    g_dist.save("grid_dist_id.h5");
    t.stop();
}

BOOST_AUTO_TEST_CASE( grid_dist_id_hdf5_load_test )
{

    // Input data
    size_t k = 2400;

    float ghost_part = 0.0;

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

    Vcluster & v_cl = create_vcluster();

    // Skip this test on big scale
    if (v_cl.getProcessingUnits() >= 32)
        return;

    // grid size
    size_t sz[2];
    sz[0] = k;
    sz[1] = k;

    // Ghost
    Ghost<2,float> g(ghost_part);

    // Distributed grid with id decomposition
    grid_dist_id<2, float, scalar<float>, CartDecomposition<2,float>> g_dist(sz,domain,g);

    g_dist.getDecomposition().write("Before_load_grid_decomposition");
    g_dist.write("Before_Loaded_grid");

    timer t;
    t.start();
    // Save the grid
    g_dist.load("grid_dist_id.h5");
    t.stop();

    g_dist.write("Loaded_grid");
    g_dist.getDecomposition().write("Loaded_grid_decomposition");

    auto it = g_dist.getDomainIterator();

    size_t count = 0;

    while (it.isNext())
    {
        //key
        auto key = it.get();

        //BOOST_CHECK_CLOSE(g_dist.template get<0>(key),1,0.0001);
        //std::cout << "Element: " << g_dist.template get<0>(key) << std::endl;

        auto keyg = g_dist.getGKey(key);

        BOOST_REQUIRE_EQUAL(g_dist.template get<0>(key), keyg.get(0));

        ++it;
        count++;
    }

    openfpm::vector<size_t> count_total;
    v_cl.allGather(count,count_total);
    v_cl.execute();

    size_t sum = 0;

    for (size_t i = 0; i < count_total.size(); i++)
        sum += count_total.get(i);

    BOOST_REQUIRE_EQUAL(sum, (size_t)k*k);
}


BOOST_AUTO_TEST_SUITE_END()

#endif /* SRC_GRID_GRID_DIST_ID_HDF5_CHCKPNT_RESTART_TEST_HPP_ */