crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-25 20:45:35 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Adding output_ppm skeleton.

Browse source
This commit is contained in:
Patrick Shriwise 2018-10-09 14:25:13 -05:00
parent 2a3b432859
commit 05192cc2d7
1 changed files with 38 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

39
src/plot.cpp
View file

@ -1,3 +1,5 @@
#include <fstream>
#include "openmc/plot.h"
#include "openmc/constants.h"
#include "openmc/settings.h"
@ -64,7 +66,9 @@ void create_ppm(ObjectPlot* pl) {
data.resize(width);
for (auto i : data) {
i.resize(height);
for (auto j : i) { j.resize(3); }
for (auto j : i) {
j.resize(3);
}
}
int in_i, out_i;
@ -174,6 +178,39 @@ void position_rgb(Particle* p, ObjectPlot* pl, int rgb[3], int &id) {
} // endif found_cell
}
//===============================================================================
// OUTPUT_PPM writes out a previously generated image to a PPM file
//===============================================================================
void output_ppm(ObjectPlot* pl, int* data, int data_size)
{
// check incoming data size
assert(data_size % 3 == 0);
// Open PPM file for writing
std::ofstream of(pl->path_plot);
// Write header
of << "P6" << std::endl;
of << pl->pixels[0] << " " << pl->pixels[1] << std::endl;
of << "255" << std::endl;
int* rgb;
// Write color for each pixel
for (int y = 0; y < pl->pixels[1]; y++) {
for (int x = 0; x < pl->pixels[0]; x++) {
int idx = 3*(x + y*width);
rgb = &data[idx];
of << itoa(rgb[0]) << itoa(rgb[1]) << itoa(rgb[2]);
}
}
// Close file
of.close();
}
void
voxel_init(hid_t file_id, const hsize_t* dims, hid_t* dspace, hid_t* dset,
hid_t* memspace)