crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-28 14:15:42 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Updating structure of RGB color data.

Browse source
This commit is contained in:
Patrick Shriwise 2018-10-29 09:47:49 -05:00
parent 38a248b4a5
commit 8018b56100
2 changed files with 54 additions and 73 deletions
Download patch file Download diff file Expand all files Collapse all files

19
include/openmc/plot.h
View file

@ -29,7 +29,7 @@ extern int n_plots; //!< number of plots in openmc run
class Plot;
extern std::vector<Plot*> plots; //!< Plot instance container
typedef std::vector<std::vector< std::vector<int>>> ImageData;
typedef std::vector<std::vector<std::vector<int>>> ImageData;
enum class plot_type {
slice = 1,
@ -51,10 +51,8 @@ enum class plot_color_by {
// RGBColor holds color information for plotted objects
//===============================================================================
struct RGBColor {
int rgb[3]; //!< RGB color values
};
typedef std::array<int, 3> RGBColor;
//===============================================================================
// Plot class
//===============================================================================
@ -66,9 +64,6 @@ public:
// Constructor
Plot(pugi::xml_node plot);
// Destructor
~Plot();
// Methods
private:
void set_id();
@ -98,7 +93,7 @@ public:
int index_meshlines_mesh; //!< Index of the mesh to draw on the plot
RGBColor meshlines_color; //!< Color of meshlines on the plot
RGBColor not_found; //!< Plot background color
std::vector<RGBColor*> colors; //!< Plot colors
std::vector<RGBColor> colors; //!< Plot colors
std::string path_plot; //!< Plot output filename
pugi::xml_node _plot_node;
};
@ -111,20 +106,20 @@ public:
//! \param[in] plot object
//! \param[out] image data associated with the plot object
void draw_mesh_lines(Plot* pl,
std::vector< std::vector< std::vector<int> > > &data);
ImageData &data);
//! Write a ppm image to file using a plot object's image data
//! \param[in] plot object
//! \param[out] image data associated with the plot object
void output_ppm(Plot* pl,
const std::vector< std::vector< std::vector<int> > > &data);
const ImageData &data);
//! Get the rgb color for a given particle position in a plot
//! \param[in] particle with position for current pixel
//! \param[in] plot object
//! \param[out] rgb color
//! \param[out] cell or material id for particle position
void position_rgb(Particle* p, Plot* pl, int rgb[3], int &id);
void position_rgb(Particle* p, Plot* pl, RGBColor &rgb, int &id);
//! Initialize a voxel file