int main(int argc, char* argv[])
{
environment env(argc, argv);
communicator comm;
clock_t before, after;
float zoltan_version;
if (Zoltan_Initialize(argc, argv, &zoltan_version) != ZOLTAN_OK) {
cout << "Zoltan_Initialize failed" << endl;
exit(EXIT_FAILURE);
}
if (comm.rank() == 0) {
cout << "Using Zoltan version " << zoltan_version << endl;
}
Dccrg<int, ArbitraryGeometry> grid;
#define GRID_SIZE 2
#define CELL_SIZE (1.0 / GRID_SIZE)
vector<double> x_coordinates, y_coordinates, z_coordinates;
for (int i = 0; i <= GRID_SIZE; i++) {
x_coordinates.push_back(i * CELL_SIZE);
}
y_coordinates.push_back(0);
y_coordinates.push_back(1);
z_coordinates.push_back(0);
z_coordinates.push_back(1);
grid.set_geometry(x_coordinates, y_coordinates, z_coordinates);
#define NEIGHBORHOOD_SIZE 1
grid.initialize(comm, "RANDOM", NEIGHBORHOOD_SIZE, 5);
if (comm.rank() == 0) {
cout << "Maximum refinement level of the grid: " << grid.get_maximum_refinement_level() << endl;
cout << "Number of cells: " << (x_coordinates.size() - 1) * (y_coordinates.size() - 1) * (z_coordinates.size() - 1) << endl << endl;
}
// every process outputs the game state into its own file
ostringstream basename, suffix(".vtk");
basename << "unrefine_simple_" << comm.rank() << "_";
ofstream outfile, visit_file;
// visualize the game with visit -o game_of_life_test.visit
if (comm.rank() == 0) {
visit_file.open("unrefine_simple.visit");
visit_file << "!NBLOCKS " << comm.size() << endl;
}
#define TIME_STEPS 8
for (int step = 0; step < TIME_STEPS; step++) {
if (comm.rank() == 0) {
cout << "step " << step << endl;
}
grid.balance_load();
vector<uint64_t> cells = grid.get_cells();
sort(cells.begin(), cells.end());
// write the game state into a file named according to the current time step
string current_output_name("");
ostringstream step_string;
step_string.fill('0');
step_string.width(5);
step_string << step;
current_output_name += basename.str();
current_output_name += step_string.str();
current_output_name += suffix.str();
// visualize the game with visit -o game_of_life_test.visit
if (comm.rank() == 0) {
for (int process = 0; process < comm.size(); process++) {
visit_file << "unrefine_simple_" << process << "_" << step_string.str() << suffix.str() << endl;
}
}
// write the grid into a file
grid.write_vtk_file(current_output_name.c_str());
// prepare to write the game data into the same file
outfile.open(current_output_name.c_str(), ofstream::app);
outfile << "CELL_DATA " << cells.size() << endl;
// write each cells neighbor count
outfile << "SCALARS neighbors int 1" << endl;
outfile << "LOOKUP_TABLE default" << endl;
for (vector<uint64_t>::const_iterator cell = cells.begin(); cell != cells.end(); cell++) {
const vector<uint64_t>* neighbors = grid.get_neighbors(*cell);
outfile << neighbors->size() << endl;
}
// write each cells process
outfile << "SCALARS process int 1" << endl;
outfile << "LOOKUP_TABLE default" << endl;
for (vector<uint64_t>::const_iterator cell = cells.begin(); cell != cells.end(); cell++) {
outfile << comm.rank() << endl;
}
// write each cells id
outfile << "SCALARS id int 1" << endl;
outfile << "LOOKUP_TABLE default" << endl;
for (vector<uint64_t>::const_iterator cell = cells.begin(); cell != cells.end(); cell++) {
outfile << *cell << endl;
}
outfile.close();
before = clock();
// refine / unrefine the smallest cell that is closest to the grid starting corner
if (step < 4) {
/*// unrefine all cells in the grid
for (vector<uint64_t>::const_iterator cell = cells.begin(); cell != cells.end(); cell++) {
grid.unrefine_completely(*cell);
}
cout << "unrefined 1" << endl;*/
grid.refine_completely_at(0.0001 * CELL_SIZE, 0.0001 * CELL_SIZE, 0.0001 * CELL_SIZE);
/*for (vector<uint64_t>::const_iterator cell = cells.begin(); cell != cells.end(); cell++) {
grid.unrefine_completely(*cell);
}
cout << "unrefined 2" << endl;*/
} else {
grid.unrefine_completely_at(0.0001 * CELL_SIZE, 0.0001 * CELL_SIZE, 0.0001 * CELL_SIZE);
}
vector<uint64_t> new_cells = grid.stop_refining();
after = clock();
cout << "Process " << comm.rank() <<": Refining / unrefining took " << double(after - before) / CLOCKS_PER_SEC << " seconds, " << new_cells.size() << " new cells created" << endl;
}
if (comm.rank() == 0) {
visit_file.close();
}
return EXIT_SUCCESS;
}
int main(int argc, char* argv[])
{
environment env(argc, argv);
communicator comm;
float zoltan_version;
if (Zoltan_Initialize(argc, argv, &zoltan_version) != ZOLTAN_OK) {
cout << "Zoltan_Initialize failed" << endl;
exit(EXIT_FAILURE);
}
if (comm.rank() == 0) {
cout << "Using Zoltan version " << zoltan_version << endl;
}
/*
Options
*/
uint64_t x_length, y_length, z_length;
unsigned int neighborhood_size, refine_n, iterations;
string load_balancing_method;
vector<string> hier_lb_methods;
vector<int> hier_lb_procs_per_level;
bool save;
boost::program_options::options_description options("Usage: program_name [options], where options are:");
options.add_options()
("help", "print this help message")
("x_length",
boost::program_options::value<uint64_t>(&x_length)->default_value(10),
"Create a grid with arg number of unrefined cells in the x direction")
("y_length",
boost::program_options::value<uint64_t>(&y_length)->default_value(10),
"Create a grid with arg number of unrefined cells in the y direction")
("z_length",
boost::program_options::value<uint64_t>(&z_length)->default_value(10),
"Create a grid with arg number of unrefined cells in the z direction")
("load_balancing_method",
boost::program_options::value<string>(&load_balancing_method)->default_value("HYPERGRAPH"),
"Use arg as the load balancing method (supported values: NONE, BLOCK, RANDOM, RCB, RIB, HSFC, GRAPH, HYPERGRAPH, HIER)")
("iterations",
boost::program_options::value<unsigned int>(&iterations)->default_value(10),
"First randomize processes of cells then balance the load using given options arg times")
("hier_lb_methods",
boost::program_options::value<vector<string> >(&hier_lb_methods)->composing(),
"Load balancing method used by HIER is specified here, once per number of hierarchies, default HYPERGRAPH, number of these must equal number of hier_lb_procs_per_level (for example --hier_... RCB --hier_... RIB --hier_... RANDOM for three hierarhies)")
("hier_lb_procs_per_level",
boost::program_options::value<vector<int> >(&hier_lb_procs_per_level)->composing(),
"Number of processes per hierarchy with HIER load balancing method, default 1, number of these must equal number of hier_lb_methods (for example --hier_... 12 --hier_... 4 --hier_... 2 for three hierarchies)")
("refine_n",
boost::program_options::value<unsigned int>(&refine_n)->default_value(0),
"Refine cells arg times")
("save",
boost::program_options::value<bool>(&save)->default_value(false),
"Save the grid after every iteration")
("neighborhood_size",
boost::program_options::value<unsigned int>(&neighborhood_size)->default_value(1),
"Size of a cell's neighborhood in cells of equal size (0 means only face neighbors are neighbors)");
// read options from command line
boost::program_options::variables_map option_variables;
boost::program_options::store(boost::program_options::parse_command_line(argc, argv, options), option_variables);
boost::program_options::notify(option_variables);
// print a help message if asked
if (option_variables.count("help") > 0) {
if (comm.rank() == 0) {
cout << options << endl;
}
comm.barrier();
return EXIT_SUCCESS;
}
// check for invalid options
if (load_balancing_method == "HIER") {
if (hier_lb_methods.size() == 0) {
if (comm.rank() == 0) {
cerr << "At least one load balancing method for HIER partitioning must be given" << endl;
}
return EXIT_FAILURE;
}
if (hier_lb_methods.size() != hier_lb_procs_per_level.size()) {
if (comm.rank() == 0) {
cerr << "Number of load balancing methods for HIER must equal the number of processes per partition options" << endl;
}
return EXIT_FAILURE;
}
for (unsigned int i = 0; i < hier_lb_procs_per_level.size(); i++) {
if (hier_lb_procs_per_level[i] <= 0) {
if (comm.rank() == 0) {
cerr << "Processes per partition must be a positive number" << endl;
}
return EXIT_FAILURE;
}
}
}
Dccrg<int> grid;
if (!grid.set_geometry(
x_length, y_length, z_length,
-0.5, -0.5, -0.5,
1.0 / x_length, 1.0 / y_length, 1.0 / z_length
)) {
if (comm.rank() == 0) {
cerr << "Couldn't set grid geometry" << endl;
}
return EXIT_FAILURE;
}
grid.initialize(
comm,
load_balancing_method.c_str(),
neighborhood_size
);
// set load balancing options
if (load_balancing_method == "HIER") {
for (unsigned int i = 0; i < hier_lb_methods.size(); i++) {
grid.add_partitioning_level(hier_lb_procs_per_level[i]);
grid.add_partitioning_option(i, "LB_METHOD", hier_lb_methods[i]);
}
}
vector<uint64_t> cells = grid.get_cells();
for (unsigned int i = 0; i < refine_n; i++) {
for (vector<uint64_t>::const_iterator cell = cells.begin(); cell != cells.end(); cell++) {
double x = grid.get_cell_x(*cell);
double y = grid.get_cell_y(*cell);
double z = grid.get_cell_z(*cell);
if (sqrt(x * x + y * y + z * z) < 0.1) {
grid.refine_completely(*cell);
}
}
grid.stop_refining();
cells = grid.get_cells();
}
if (comm.rank() == 0) {
cout << "Total cells after refining: " << all_reduce(comm, cells.size(), plus<uint64_t>()) << endl;
} else {
all_reduce(comm, cells.size(), plus<uint64_t>());
}
// every process outputs the game state into its own file
ostringstream basename, suffix(".vtk");
basename << "load_balancing_test_" << comm.rank() << "_";
ofstream outfile, visit_file;
// visualize the game with visit -o game_of_life_test.visit
if (comm.rank() == 0) {
visit_file.open("load_balancing_test.visit");
visit_file << "!NBLOCKS " << comm.size() << endl;
}
for (unsigned int step = 0; step < iterations; step++) {
// scatter cells to random processes
cells = grid.get_cells();
for (vector<uint64_t>::const_iterator
cell = cells.begin();
cell != cells.end();
cell++
) {
grid.pin(*cell, rand() % comm.size());
}
grid.migrate_cells();
grid.unpin_all_cells();
// balance the load using given options
grid.balance_load();
/*grid.start_remote_neighbor_data_update();
grid.wait_neighbor_data_update();*/
cells = grid.get_cells();
// the library writes the grid into a file in ascending cell order, do the same for the grid data at every time step
sort(cells.begin(), cells.end());
// write the game state into a file named according to the current time step
string current_output_name("");
ostringstream step_string;
step_string.fill('0');
step_string.width(5);
step_string << step;
current_output_name += basename.str();
current_output_name += step_string.str();
current_output_name += suffix.str();
// visualize the game with visit -o game_of_life_test.visit
if (comm.rank() == 0) {
for (int process = 0; process < comm.size(); process++) {
visit_file << "load_balancing_test_" << process << "_" << step_string.str() << suffix.str() << endl;
}
}
// write the grid into a file
grid.write_vtk_file(current_output_name.c_str());
// prepare to write the game data into the same file
outfile.open(current_output_name.c_str(), ofstream::app);
outfile << "CELL_DATA " << cells.size() << endl;
// write each cells neighbor count
outfile << "SCALARS neighbors int 1" << endl;
outfile << "LOOKUP_TABLE default" << endl;
for (vector<uint64_t>::const_iterator cell = cells.begin(); cell != cells.end(); cell++) {
const vector<uint64_t>* neighbors = grid.get_neighbors(*cell);
outfile << neighbors->size() << endl;
}
// write each cells process
outfile << "SCALARS process int 1" << endl;
outfile << "LOOKUP_TABLE default" << endl;
for (vector<uint64_t>::const_iterator cell = cells.begin(); cell != cells.end(); cell++) {
outfile << comm.rank() << endl;
}
// write each cells id
outfile << "SCALARS id int 1" << endl;
outfile << "LOOKUP_TABLE default" << endl;
for (vector<uint64_t>::const_iterator cell = cells.begin(); cell != cells.end(); cell++) {
outfile << *cell << endl;
}
outfile.close();
}
if (comm.rank() == 0) {
cout << "Passed" << endl;
}
return EXIT_SUCCESS;
}