Skip to content

refactor(mpi): drop commented-out write_vtk call #22

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
nikbott merged 2 commits into from
Jul 7, 2026
Merged

refactor(mpi): drop commented-out write_vtk call #22

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
b9e183c
style(mpi): clang-format main.cpp
nikbott Jul 7, 2026
ac8a95f
refactor(mpi): drop commented-out write_vtk call
nikbott Jul 7, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
45 changes: 26 additions & 19 deletions mpi/main.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,19 +2,21 @@
* @file main.cpp
* @brief Main Entry Point.
*/
#include <chrono>
#include <iostream>
#include <string>

#include "core.hpp"
#include "tree.hpp"
#include "physics.hpp"
#include "tree.hpp"
#include "viz.hpp"
#include <iostream>
#include <chrono>
#include <string>

using namespace amr;

int main(int argc, char** argv) {
amr::mpi::Context mpi(argc, argv);
if (mpi.rank == 0) std::cout << "=== AMR System (SoA + Strong Types) ===\n";
if (mpi.rank == 0)
std::cout << "=== AMR System (SoA + Strong Types) ===\n";

Config cfg;
cfg.max_level = 20;
Expand All @@ -23,50 +25,55 @@ int main(int argc, char** argv) {
cfg.radius = 0.25;

{
if (mpi.rank == 0) std::cout << "\n--- 2D Quadtree ---\n";
if (mpi.rank == 0)
std::cout << "\n--- 2D Quadtree ---\n";
DistributedTree<2> tree(cfg.max_level);
CircleOracle oracle(cfg);
int steps = 0;
while(tree.refine(oracle)) {
while (tree.refine(oracle)) {
tree.repartition();
size_t n = tree.global_size();
if (mpi.rank == 0) std::cout << "Refine step " << ++steps << ": " << n << " leaves\n";
if (mpi.rank == 0)
std::cout << "Refine step " << ++steps << ": " << n << " leaves\n";
}
if (mpi.rank == 0) std::cout << "Balancing...\n";
if (mpi.rank == 0)
std::cout << "Balancing...\n";
tree.repartition();
auto start = std::chrono::high_resolution_clock::now();
tree.balance();
auto end = std::chrono::high_resolution_clock::now();
tree.verify_global();
if (mpi.rank == 0) {
std::cout << "Balanced in "
<< std::chrono::duration_cast<std::chrono::milliseconds>(end-start).count()
std::cout << "Balanced in "
<< std::chrono::duration_cast<std::chrono::milliseconds>(end - start).count()
<< "ms. Final size: " << tree.global_size() << "\n";
}
viz::write_svg(tree, "mesh_2d_rank_" + std::to_string(mpi.rank) + ".svg");
}

{
if (mpi.rank == 0) std::cout << "\n--- 3D Octree ---\n";
if (mpi.rank == 0)
std::cout << "\n--- 3D Octree ---\n";
DistributedTree<3> tree(cfg.max_level);
SphereOracle oracle(cfg);
int steps = 0;
while(tree.refine(oracle)) {
while (tree.refine(oracle)) {
tree.repartition();
size_t n = tree.global_size();
if (mpi.rank == 0) std::cout << "Refine step " << ++steps << ": " << n << " leaves\n";
if (mpi.rank == 0)
std::cout << "Refine step " << ++steps << ": " << n << " leaves\n";
}
if (mpi.rank == 0) std::cout << "Balancing...\n";
if (mpi.rank == 0)
std::cout << "Balancing...\n";
tree.repartition();
auto start = std::chrono::high_resolution_clock::now();
tree.balance();
auto end = std::chrono::high_resolution_clock::now();
if (mpi.rank == 0) {
std::cout << "Balanced in "
<< std::chrono::duration_cast<std::chrono::milliseconds>(end-start).count()
std::cout << "Balanced in "
<< std::chrono::duration_cast<std::chrono::milliseconds>(end - start).count()
<< "ms. Final size: " << tree.global_size() << "\n";
}
//viz::write_vtk(tree, "mesh_3d_rank_" + std::to_string(mpi.rank) + ".vtk");
}
return 0;
}
}
Loading