Skip to content

Commit

Permalink
Refactor and clean
Browse files Browse the repository at this point in the history
  • Loading branch information
@ampdes
ampdes committed Aug 14, 2024
1 parent dd524c7 commit d624050
Show file tree
Hide file tree
Showing 4 changed files with 213 additions and 230 deletions.
137 changes: 32 additions & 105 deletions cpp/demo/checkpointing/main.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,133 +24,60 @@ int main(int argc, char* argv[])
dolfinx::init_logging(argc, argv);
MPI_Init(&argc, &argv);

// {
// int i=0;
// while (i == 0)
// sleep(5);
// }

// Create mesh and function space
auto part = mesh::create_cell_partitioner(mesh::GhostMode::shared_facet);
auto mesh = std::make_shared<mesh::Mesh<float>>(mesh::create_rectangle<float>(
MPI_COMM_WORLD, {{{0.0, 0.0}, {1.0, 1.0}}}, {4, 4},
mesh::CellType::quadrilateral, part));


try
{
// Set up ADIOS2 IO and Engine
adios2::ADIOS adios("checkpointing.yml", mesh->comm());
adios2::ADIOS adios(mesh->comm());

adios2::IO io = adios.DeclareIO("mesh-write");
io.SetEngine("BP5");
adios2::Engine engine = io.Open("mesh.bp", adios2::Mode::Write);

std::vector<std::string> mytags = {"one"};
io.DefineAttribute<std::string>("tags", mytags.data(), mytags.size(), "", "", true);

std::vector<std::string> mytags2 = {"one", "two"};
io.DefineAttribute<std::string>("tags", mytags2.data(), mytags2.size());

io::native::write_mesh(io, engine, *mesh);

engine.Close();
}
catch (std::exception &e)
{
std::cout << "ERROR: ADIOS2 exception: " << e.what() << "\n";
MPI_Abort(MPI_COMM_WORLD, -1);
std::cout << "ERROR: ADIOS2 exception: " << e.what() << "\n";
MPI_Abort(MPI_COMM_WORLD, -1);
}

// Set up ADIOS2 IO and Engine
// adios2::ADIOS adios_read(mesh->comm());
// adios2::IO io_read = adios_read.DeclareIO("mesh-read");
// io_read.SetEngine("BP5");
// adios2::Engine engine_read = io_read.Open("mesh.bp", adios2::Mode::Read);

// auto mesh_read_variant
// = io::native::read_mesh_variant(io_read, engine_read, MPI_COMM_WORLD);

// engine_read.Close();

// ----------------------------------------------------------------------
// auto adios_query
// = ADIOS2Wrapper(MPI_COMM_WORLD, "mesh.bp", "mesh-read", "BP5", "read");

// auto adios_read
// = ADIOS2Wrapper(MPI_COMM_WORLD, "mesh.bp", "mesh-read", "BP5", "read");

// Can't use the same engine to query as well as read
// since, in that case BeginStep and EndStep will be called twice
// on a dataset written with a single Step

// auto reader = adios_read.engine();
// reader->BeginStep();
// auto mesh_read_variant = io::native::read_mesh_variant(
// adios_query, adios_read, mesh->comm());

// reader->EndStep();

// // We cannot resolve easily the variant
// // Hence the user can query the type and
// // call the correct read_mesh
// auto mesh_read = std::get<0>(mesh_read_variant);
// // auto mesh_read = std::get<mesh::Mesh<float>>(mesh_read_variant);

// auto adios_write
// = ADIOS2Wrapper(mesh->comm(), "mesh2.bp", "mesh-rewrite", "BP5",
// "write");

// io::native::write_mesh(adios_write, mesh_read);

// auto io_query = adios_query.io();
// auto engine_query = adios_query.engine();

// Following throws an error. engine_read->BeginStep() is needed to
// read the VariableType, but then EndStep() fails with message
// EndStep() called without a successful BeginStep()

// engine_query->BeginStep();
// std::string floating_point = io_query->VariableType("x");
// engine_query->EndStep();

// // std::string floating_point =
// dolfinx::io::native::query_type(adios_read); std::cout <<
// floating_point;

// // TODO: move type deduction inside checkpointing
// if ("float" == floating_point)
// {
// using T = float;
// mesh::Mesh<T> mesh_read
// = io::native::read_mesh<T>(adios_read, mesh->comm());
// adios_read.close();

// auto adios_write = ADIOS2Wrapper(mesh->comm(), "mesh2.bp",
// "mesh-rewrite",
// "BP5", "write");

// auto io_write = adios_write.io();

// io::native::write_mesh(adios_write, mesh_read);
// adios_write.close();
// }
// else if ("double" == floating_point)
// {
// using T = double;
// mesh::Mesh<T> mesh_read
// = io::native::read_mesh<T>(adios_read, mesh->comm());
// adios_read.close();

// auto adios_write = ADIOS2Wrapper(mesh->comm(), "mesh2.bp",
// "mesh-rewrite",
// "BP5", "write");

// auto io_write = adios_write.io();

// io::native::write_mesh(adios_write, mesh_read);
// adios_write.close();
// }
try
{
// Set up ADIOS2 IO and Engine
adios2::ADIOS adios_read(MPI_COMM_WORLD);
adios2::IO io_read = adios_read.DeclareIO("mesh-read");
io_read.SetEngine("BP5");
adios2::Engine engine_read = io_read.Open("mesh.bp", adios2::Mode::Read);

engine_read.BeginStep();
auto mesh_read = io::native::read_mesh<float>(io_read, engine_read, MPI_COMM_WORLD);
if (engine_read.BetweenStepPairs())
{
engine_read.EndStep();
}

engine_read.Close();

adios2::IO io_write = adios_read.DeclareIO("mesh-write");
io_write.SetEngine("BP5");
adios2::Engine engine_write = io_write.Open("mesh2.bp", adios2::Mode::Write);

io::native::write_mesh(io_write, engine_write, mesh_read);
engine_write.Close();
}
catch (std::exception &e)
{
std::cout << "ERROR: ADIOS2 exception: " << e.what() << "\n";
MPI_Abort(MPI_COMM_WORLD, -1);
}

MPI_Finalize();
return 0;
Expand Down
Loading

0 comments on commit d624050

Please sign in to comment.