129 lines
3.9 KiB
C++
129 lines
3.9 KiB
C++
/** @file phase_transition_mpi.cpp
|
|
*
|
|
* @author Cory Alexander Balaton (coryab)
|
|
* @author Janita Ovidie Sandtrøen Willumsen (janitaws)
|
|
*
|
|
* @version 1.0
|
|
*
|
|
* @brief Sweep over different temperatures and generate data.
|
|
*
|
|
* @bug No known bugs
|
|
* */
|
|
#include "data_type.hpp"
|
|
#include "monte_carlo.hpp"
|
|
#include "utils.hpp"
|
|
|
|
#include <algorithm>
|
|
#include <fstream>
|
|
#include <iostream>
|
|
#include <iterator>
|
|
#include <mpi.h>
|
|
#include <sstream>
|
|
|
|
/** @brief The main function*/
|
|
int main(int argc, char **argv)
|
|
{
|
|
if (argc < 5) {
|
|
std::cout << "You need at least 4 arguments" << std::endl;
|
|
abort();
|
|
}
|
|
double t0, t1;
|
|
t0 = MPI_Wtime();
|
|
double start = atof(argv[1]), end = atof(argv[2]);
|
|
int points = atoi(argv[3]), N;
|
|
int lattice_sizes[] = {20, 40, 60, 80, 100};
|
|
double dt = (end - start) / points;
|
|
int cycles = atoi(argv[4]);
|
|
std::ofstream ofile;
|
|
|
|
data_t data[points];
|
|
|
|
// MPI stuff
|
|
int rank, cluster_size;
|
|
|
|
// Initialize MPI
|
|
MPI_Init(&argc, &argv);
|
|
|
|
// Get the cluster size and rank
|
|
MPI_Comm_size(MPI_COMM_WORLD, &cluster_size);
|
|
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
|
|
|
|
int remainder = points % cluster_size;
|
|
double i_start;
|
|
int i_points;
|
|
// Distribute temperature points
|
|
if (rank < remainder) {
|
|
i_points = points / cluster_size + 1;
|
|
i_start = start + dt * i_points * rank;
|
|
}
|
|
else {
|
|
i_points = points / cluster_size;
|
|
i_start = start + dt * (i_points * rank + remainder);
|
|
}
|
|
|
|
data_t i_data[i_points];
|
|
std::cout << "Rank " << rank << ": " << i_points << ',' << i_start << '\n';
|
|
|
|
for (int L : lattice_sizes) {
|
|
N = L * L;
|
|
for (size_t i = 0; i < i_points; i++) {
|
|
i_data[i] = monte_carlo_parallel(L, i_start + dt * i, cycles);
|
|
}
|
|
|
|
if (rank == 0) {
|
|
std::copy_n(i_data, i_points, data);
|
|
for (size_t i = 1; i < cluster_size; i++) {
|
|
if (rank < remainder) {
|
|
MPI_Recv((void *)i_data,
|
|
sizeof(data_t) * (points / cluster_size + 1),
|
|
MPI_CHAR, i, MPI_ANY_TAG, MPI_COMM_WORLD,
|
|
MPI_STATUS_IGNORE);
|
|
std::copy_n(i_data, points / cluster_size + 1,
|
|
data + (points / cluster_size) * i);
|
|
}
|
|
else {
|
|
MPI_Recv((void *)i_data,
|
|
sizeof(data_t) * (points / cluster_size), MPI_CHAR,
|
|
i, MPI_ANY_TAG, MPI_COMM_WORLD, MPI_STATUS_IGNORE);
|
|
std::copy_n(i_data, points / cluster_size,
|
|
data + (points / cluster_size) * i + remainder);
|
|
}
|
|
}
|
|
std::stringstream outfile;
|
|
outfile << "output/phase_transition/size_" << L << ".txt";
|
|
utils::mkpath(utils::dirname(outfile.str()));
|
|
ofile.open(outfile.str());
|
|
|
|
double temp, CV, X;
|
|
|
|
using utils::scientific_format;
|
|
for (size_t i = 0; i < points; i++) {
|
|
temp = start + dt * i;
|
|
CV = (data[i].E2 - data[i].E * data[i].E)
|
|
/ ((double)N * temp * temp);
|
|
X = (data[i].M2 - data[i].M_abs * data[i].M_abs)
|
|
/ ((double)N * temp);
|
|
|
|
ofile << scientific_format(temp) << ','
|
|
<< scientific_format(data[i].E / N) << ','
|
|
<< scientific_format(data[i].M_abs / N) << ','
|
|
<< scientific_format(CV) << ',' << scientific_format(X)
|
|
<< '\n';
|
|
}
|
|
ofile.close();
|
|
}
|
|
else {
|
|
MPI_Send(i_data, i_points * sizeof(data_t), MPI_CHAR, 0, rank,
|
|
MPI_COMM_WORLD);
|
|
}
|
|
}
|
|
|
|
t1 = MPI_Wtime();
|
|
|
|
if (rank == 0) {
|
|
std::cout << "Time: " << t1 - t0 << " seconds\n";
|
|
}
|
|
|
|
MPI_Finalize();
|
|
}
|