Merge pull request #40 from fusion-energy/fix_gamma

Converting material gamma emission plotting to function

README.md

@@ -14,15 +14,15 @@ pip install openmc_source_plotter
 
 # Features
 
-The package simply extends the default ```openmc.IndependentSourceBase``` and ```openmc.Model``` to provides additional functions that:
+The package provides three plotting functions that can plot source data from openmc objects.
+- ```plot_source_energy```
+- ```plot_source_position```
+- ```plot_source_direction```
+- ```plot_gamma_emission```
+
+Additionally the package provides a convienient method of sampling particles
+- ```sample_initial_particles```
 
-- extract the positions, directions and energy of particles
-- visualise a source with respect to:
- - direction
- - energy
- - position
-
-Or just provide the initial particles with ```sample_initial_particles```
 
 # Example plots
 

examples/example_gamma_spec_plot.py

@@ -1,6 +1,9 @@
 import openmc
 from openmc_source_plotter import plot_gamma_emission
 
+# you will need to install optional dependancies for this example
+# pip install openmc_source_plotter[gammas]
+
 # this path will need changing to point to your chain file
 # openmc.config["chain_file"] = "chain-endf.xml"
 

src/openmc_source_plotter/material.py

@@ -3,7 +3,7 @@
 
 
 def plot_gamma_emission(
- self,
+ material,
  label_top: int = None,
 ):
  """makes a plot of the gamma energy spectra for a material. The
@@ -26,7 +26,7 @@ def plot_gamma_emission(
  possible_energies_to_label = []
  import lineid_plot
 
- atoms = self.get_nuclide_atoms()
+ atoms = material.get_nuclide_atoms()
  for nuc, num_atoms in atoms.items():
  dists = []
  probs = []
@@ -47,7 +47,7 @@ def plot_gamma_emission(
 
  probs = []
  en = []
- energy_dis = self.decay_photon_energy
+ energy_dis = material.get_decay_photon_energy(clip_tolerance=0.0)
  for p in energy_dis.p:
  probs.append(0)
  probs.append(p)
@@ -60,17 +60,17 @@ def plot_gamma_emission(
  # print(probs)
  lineid_plot.plot_line_ids(
  en,
- # self.decay_photon_energy.x,
+ # material.decay_photon_energy.x,
  probs,
- # self.decay_photon_energy.p,
+ # material.decay_photon_energy.p,
  energies_to_label,
  labels,
  )
 
  else:
  probs = []
  en = []
- energy_dis = self.decay_photon_energy
+ energy_dis = material.get_decay_photon_energy(clip_tolerance=0.0)
  for p in energy_dis.p:
  probs.append(0)
  probs.append(p)
@@ -87,6 +87,3 @@ def plot_gamma_emission(
  plt.xlabel("Energy [eV]")
  plt.ylabel("Activity [Bq/s]")
  return plt
-
-
-openmc.Material.plot_gamma_emission = plot_gamma_emission