Skip to content

Generalize mesh tally plotting #134

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
pshriwise merged 10 commits into from
Feb 15, 2024
Merged

Generalize mesh tally plotting #134

Changes from 1 commit
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
7b79df4
Add some type hints
paulromano Dec 14, 2023
506c0bb
Utilize new get_plot_bins method from OpenMC for mesh plotting
paulromano Dec 14, 2023
ab9d7f9
Allow all .h5 files to be opened as statepoints
paulromano Jan 29, 2024
297c2cb
Update Python and OpenMC version dependency
paulromano Jan 29, 2024
3c3a934
Fix VTK export dialog
paulromano Feb 2, 2024
e30672a
Replace delta() with angleDelta() for QWheelEvent
paulromano Feb 2, 2024
b9d9f94
Account for mesh filter translation
paulromano Feb 2, 2024
1d3764b
Add checkbox for mesh volume normalization
paulromano Feb 11, 2024
e771ac2
Make sure volume norm checkbox gets updated
paulromano Feb 11, 2024
deae21a
Change mesh tally plot label when volume normalization is applied
paulromano Feb 15, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Change mesh tally plot label when volume normalization is applied
  • Loading branch information
@paulromano
paulromano committed Feb 15, 2024
commit deae21abbfda47929155d4d11ca0fbdcb29b6b68
48 changes: 32 additions & 16 deletions openmc_plotter/plotmodel.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
import itertools
import pickle
import threading
from typing import Literal, Tuple
from typing import Literal, Tuple, Optional

from PySide6.QtWidgets import QItemDelegate, QColorDialog, QLineEdit, QMessageBox
from PySide6.QtCore import QAbstractTableModel, QModelIndex, Qt, QSize, QEvent
Expand All @@ -28,9 +28,14 @@
_MODEL_PROPERTIES = ('temperature', 'density')
_PROPERTY_INDICES = {'temperature': 0, 'density': 1}

_REACTION_UNITS = 'Reactions per Source Particle'
_PRODUCTION_UNITS = 'Particles Produced per Source Particle'
_ENERGY_UNITS = 'eV per Source Particle'
_REACTION_UNITS = 'reactions/source'
_PRODUCTION_UNITS = 'particles/source'
_ENERGY_UNITS = 'eV/source'

_REACTION_UNITS_VOL = 'reactions/cm³/source'
_PRODUCTION_UNITS_VOL = 'particles/cm³/source'
_ENERGY_UNITS_VOL = 'eV/cm³/source'


_SPATIAL_FILTERS = (openmc.UniverseFilter,
openmc.MaterialFilter,
Expand All @@ -42,19 +47,26 @@
_PRODUCTIONS = ('delayed-nu-fission', 'prompt-nu-fission', 'nu-fission',
'nu-scatter', 'H1-production', 'H2-production',
'H3-production', 'He3-production', 'He4-production')
_ENERGY_SCORES = {'heating', 'heating-local', 'kappa-fission',
'fission-q-prompt', 'fission-q-recoverable',
'damage-energy'}

_SCORE_UNITS = {p: _PRODUCTION_UNITS for p in _PRODUCTIONS}
_SCORE_UNITS['flux'] = 'Particle-cm/Particle'
_SCORE_UNITS['current'] = 'Particles per source Particle'
_SCORE_UNITS['events'] = 'Events per Source Particle'
_SCORE_UNITS['inverse-velocity'] = 'Particle-seconds per Source Particle'
_SCORE_UNITS['heating'] = _ENERGY_UNITS
_SCORE_UNITS['heating-local'] = _ENERGY_UNITS
_SCORE_UNITS['kappa-fission'] = _ENERGY_UNITS
_SCORE_UNITS['fission-q-prompt'] = _ENERGY_UNITS
_SCORE_UNITS['fission-q-recoverable'] = _ENERGY_UNITS
_SCORE_UNITS['decay-rate'] = 'Seconds^-1'
_SCORE_UNITS['damage-energy'] = _ENERGY_UNITS
_SCORE_UNITS['flux'] = 'particle-cm/source'
_SCORE_UNITS['current'] = 'particle/source'
_SCORE_UNITS['events'] = 'events/source'
_SCORE_UNITS['inverse-velocity'] = 'particle-s/source'
_SCORE_UNITS['decay-rate'] = 'particle/s/source'
_SCORE_UNITS.update({s: _ENERGY_UNITS for s in _ENERGY_SCORES})

_SCORE_UNITS_VOL = {p: _PRODUCTION_UNITS_VOL for p in _PRODUCTIONS}
_SCORE_UNITS_VOL['flux'] = 'particle/cm²/source'
_SCORE_UNITS_VOL['current'] = 'particle/cm³/source'
_SCORE_UNITS_VOL['events'] = 'events/cm³/source'
_SCORE_UNITS_VOL['inverse-velocity'] = 'particle-s/cm³/source'
_SCORE_UNITS_VOL['decay-rate'] = 'particle/s/cm³/source'
_SCORE_UNITS.update({s: _ENERGY_UNITS_VOL for s in _ENERGY_SCORES})


_TALLY_VALUES = {'Mean': 'mean',
'Std. Dev.': 'std_dev',
Expand Down Expand Up @@ -384,7 +396,7 @@ def storeCurrent(self):
""" Add current view to previousViews list """
self.previousViews.append(copy.deepcopy(self.currentView))

def create_tally_image(self, view=None):
def create_tally_image(self, view: Optional[PlotView] = None):
"""
Parameters
----------
Expand Down Expand Up @@ -438,6 +450,10 @@ def create_tally_image(self, view=None):
contains_cellinstance = tally.contains_filter(openmc.CellInstanceFilter)

if tally.contains_filter(openmc.MeshFilter):
# Check for volume normalization in order to change units
if view.tallyVolumeNorm:
units_out = _SCORE_UNITS_VOL.get(scores[0], _REACTION_UNITS_VOL)

if tally_value == 'rel_err':
# get both the std. dev. data and mean data
# to create the relative error data
Expand Down