Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F103447143
dumpers.cc
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Sun, Mar 2, 03:09
Size
4 KB
Mime Type
text/x-c
Expires
Tue, Mar 4, 03:09 (1 d, 23 h)
Engine
blob
Format
Raw Data
Handle
24520827
Attached To
R7871 phys-743-exercises
dumpers.cc
View Options
/* -------------------------------------------------------------------------- */
#include "dumpers.hh"
#include "grid.hh"
/* -------------------------------------------------------------------------- */
#include <array>
#include <fstream>
#include <iomanip>
#include <sstream>
/* -------------------------------------------------------------------------- */
/* -------------------------------------------------------------------------- */
void Dumper::set_min(float min) { m_min = min; }
void Dumper::set_max(float max) { m_max = max; }
float Dumper::min() const { return m_min; }
float Dumper::max() const { return m_max; }
/* -------------------------------------------------------------------------- */
void DumperASCII::dump(int step) {
std::ofstream fout;
std::stringstream sfilename;
sfilename << "out_" << std::setfill('0') << std::setw(5) << step << ".pgm";
fout.open(sfilename.str());
int m = m_grid.m();
int n = m_grid.n();
fout << "P2" << std::endl << "# CREATOR: Poisson program" << std::endl;
fout << m << " " << n << std::endl;
fout << 255 << std::endl;
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
int v = 255. * (m_grid(i, j) - m_min) / (m_max - m_min);
v = std::min(v, 255);
fout << v << std::endl;
}
}
}
/* -------------------------------------------------------------------------- */
void DumperBinary::dump(int step) {
std::ofstream fout;
std::stringstream sfilename;
int prank, psize;
MPI_Comm_rank(m_communicator, &prank);
MPI_Comm_size(m_communicator, &psize);
sfilename << "out_" << std::setfill('0') << std::setw(5) << step << ".bmp";
fout.open(sfilename.str(), std::ios_base::binary);
int h = m_grid.m();
int w = m_grid.n();
// removing the ghosts from the height
if (psize > 1)
h = (prank == 0 || prank == psize - 1 ? h - 1 : h - 2);
int start = (prank == 0 ? 0 : 1);
// Gathering the size of every processors, this could be done as in the
// constructor of the Simulation instead
std::vector<int> size_per_proc(psize);
MPI_Allgather(&h, 1, MPI_INT, size_per_proc.data(), 1, MPI_INT,
m_communicator);
// determining the local offset
int offset_h = 0;
for (int i = 0; i < prank; ++i) {
offset_h += size_per_proc[i];
}
int total_h = offset_h;
for (int i = prank; i < psize; ++i) {
total_h += size_per_proc[i];
}
// gathering all the data on the root processor
std::vector<float> buffer;
if (prank == 0) {
std::vector<int> displs(psize + 1);
displs[0] = 0;
for (int i = 0; i < psize; ++i) {
size_per_proc[i] *= w;
displs[i + 1] = displs[i] + size_per_proc[i];
}
buffer.resize(displs[psize]);
MPI_Gatherv(&m_grid(start, 0), h * w, MPI_FLOAT,
buffer.data(), size_per_proc.data(), displs.data(), MPI_FLOAT, 0, m_communicator);
} else {
MPI_Gatherv(&m_grid(start, 0), h * w, MPI_FLOAT,
NULL, NULL, NULL, MPI_FLOAT, 0, m_communicator);
return;
}
// only processor 0 continues from here
int row_size = 3 * w;
// if the file width (3*w) is not a multiple of 4 adds enough bytes to make it
// a multiple of 4
int padding = (4 - (row_size) % 4) % 4;
row_size += padding;
int filesize = 54 + (row_size)*total_h;
std::vector<char> img(row_size * total_h);
std::fill(img.begin(), img.end(), 0);
for (int i = 0; i < total_h; i++) {
for (int j = 0; j < w; j++) {
float v = ((buffer[(total_h - 1 - i)*w + j] - m_min) / (m_max - m_min));
float r = v * 255; // Red channel
float g = v * 255; // Green channel
float b = v * 255; // Red channel
r = std::min(r, 255.f);
g = std::min(g, 255.f);
b = std::min(b, 255.f);
img[row_size * i + 3 * j + 2] = r;
img[row_size * i + 3 * j + 1] = g;
img[row_size * i + 3 * j + 0] = b;
}
}
std::array<char, 14> bmpfileheader = {'B', 'M', 0, 0, 0, 0, 0,
0, 0, 0, 54, 0, 0, 0};
std::array<char, 40> bmpinfoheader = {40, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 1, 0, 24, 0};
bmpfileheader[2] = filesize;
bmpfileheader[3] = filesize >> 8;
bmpfileheader[4] = filesize >> 16;
bmpfileheader[5] = filesize >> 24;
bmpinfoheader[4] = w;
bmpinfoheader[5] = w >> 8;
bmpinfoheader[6] = w >> 16;
bmpinfoheader[7] = w >> 24;
bmpinfoheader[8] = total_h;
bmpinfoheader[9] = total_h >> 8;
bmpinfoheader[10] = total_h >> 16;
bmpinfoheader[11] = total_h >> 24;
bmpinfoheader[20] = (filesize - 54);
bmpinfoheader[21] = (filesize - 54) >> 8;
bmpinfoheader[22] = (filesize - 54) >> 16;
bmpinfoheader[23] = (filesize - 54) >> 24;
fout.write(bmpfileheader.data(), 14);
fout.write(bmpinfoheader.data(), 40);
fout.write(img.data(), total_h * row_size);
}
Event Timeline
Log In to Comment