Mesh annotations#153

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
paulromano merged 7 commits into from
Nov 23, 2024
Merged

Mesh annotations #153

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
ddf56ef
Adding a method for annotating mesh element boundaries
pshriwise Jul 22, 2024
397b920
Adding mesh annotation dock
pshriwise Jul 22, 2024
658483c
Small updates for formatting/behavior
pshriwise Jul 24, 2024
c5503d9
Add shortcut key for mesh annotation
pshriwise Jul 26, 2024
e745fab
Merge branch 'develop' into mesh_annotation
pshriwise Nov 1, 2024
e120abc
Style fixes
paulromano Nov 22, 2024
ecc539c
Update mesh_plot_bins to account for filter translation if needed
pshriwise Nov 23, 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
2 changes: 1 addition & 1 deletion openmc_plotter/__init__.py
View file
Open in desktop
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
__version__ = '0.4.0'
__version__ = '0.4.1'
51 changes: 51 additions & 0 deletions openmc_plotter/docks.py
View file
Open in desktop
Original file line numberDiff line numberDiff line change
Expand Up@@ -33,6 +33,57 @@ def __init__(self, model, font_metric, parent=None):
self.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Expanding)


class MeshAnnotationDock(PlotterDock):
"""Dock for mesh annotation options"""

def __init__(self, model, font_metric, parent=None):
super().__init__(model, font_metric, parent)

self.treeLayout = QVBoxLayout()
self.meshTree = QTreeWidget()
self.treeExpander = Expander("Meshes:", layout=self.treeLayout)
self.treeExpander.expand() # start with meshes expanded

self.meshTree.setColumnCount(1)

self.mesh_items = []
for mesh_id in self.model.cpp_mesh_ids():
mesh_item = QTreeWidgetItem(self.meshTree, (f'Mesh {mesh_id}',))
mesh_item.setFlags(mesh_item.flags() | QtCore.Qt.ItemIsUserCheckable)
mesh_item.setCheckState(0, QtCore.Qt.Unchecked)
self.mesh_items.append((mesh_id, mesh_item))
self.meshTree.addTopLevelItem(mesh_item)

self.meshTree.setHeaderHidden(True)

# Create submit button
self.applyButton = QPushButton("Apply Changes")
# Mac bug fix
self.applyButton.setMinimumHeight(self.font_metric.height() * 1.6)
self.applyButton.clicked.connect(self.main_window.applyChanges)

label = QLabel("Mesh Annotations")
self.treeLayout.addWidget(label)
self.treeLayout.addWidget(self.meshTree)
self.treeLayout.addWidget(HorizontalLine())
self.treeLayout.addWidget(self.applyButton)

self.optionsWidget = QWidget()
self.optionsWidget.setLayout(self.treeLayout)
self.setWidget(self.optionsWidget)

def get_checked_meshes(self):
return [id for id, item in self.mesh_items if item.checkState(0) == QtCore.Qt.Checked]

def update(self):
pass

def resizeEvent(self, event):
self.main_window.resizeEvent(event)

hideEvent = showEvent = moveEvent = resizeEvent


class DomainDock(PlotterDock):
"""
Domain options dock
Expand Down
32 changes: 31 additions & 1 deletion openmc_plotter/main_window.py
View file
Open in desktop
Original file line numberDiff line numberDiff line change
Expand Up@@ -22,7 +22,7 @@

from .plotmodel import PlotModel, DomainTableModel, hash_model
from .plotgui import PlotImage, ColorDialog
from .docks import DomainDock, TallyDock
from .docks import DomainDock, TallyDock, MeshAnnotationDock
from .overlays import ShortcutsOverlay
from .tools import ExportDataDialog, SourceSitesDialog

Expand DownExpand Up@@ -91,6 +91,12 @@ def loadGui(self, use_settings_pkl=True):
self.tallyDock.setObjectName("Tally Options Dock")
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.tallyDock)

# Mesh Annotation Dock
self.meshAnnotationDock = MeshAnnotationDock(self.model, self.font_metric, self)
self.meshAnnotationDock.update()
self.meshAnnotationDock.setObjectName("Mesh Annotation Dock")
self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.meshAnnotationDock)

# Color DialogtallyDock
self.colorDialog = ColorDialog(self.model, self.font_metric, self)
self.colorDialog.hide()
Expand DownExpand Up@@ -386,6 +392,12 @@ def createMenuBar(self):
self.tallyDockAction.setStatusTip('Toggle tally dock visibility')
self.tallyDockAction.triggered.connect(self.toggleTallyDockView)

self.meshAnnotationDockAction = QAction('Mesh &Annotation Dock', self)
self.meshAnnotationDockAction.setShortcut("Ctrl+E")
self.meshAnnotationDockAction.setToolTip('Toggle mesh annotation dock visibility')
self.meshAnnotationDockAction.setStatusTip('Toggle mesh annotation dock visibility')
self.meshAnnotationDockAction.triggered.connect(self.toggleMeshAnnotationDockView)

self.zoomAction = QAction('&Zoom...', self)
self.zoomAction.setShortcut('Alt+Shift+Z')
self.zoomAction.setToolTip('Edit zoom factor')
Expand All@@ -395,6 +407,7 @@ def createMenuBar(self):
self.viewMenu = self.mainMenu.addMenu('&View')
self.viewMenu.addAction(self.dockAction)
self.viewMenu.addAction(self.tallyDockAction)
self.viewMenu.addAction(self.meshAnnotationDockAction)
self.viewMenu.addSeparator()
self.viewMenu.addAction(self.zoomAction)
self.viewMenu.aboutToShow.connect(self.updateViewMenu)
Expand DownExpand Up@@ -624,6 +637,10 @@ def updateDataMenu(self):
elif hasattr(self, "closeStatePointAction"):
self.dataMenu.removeAction(self.closeStatePointAction)


def updateMeshAnnotations(self):
self.model.activeView.mesh_annotations = self.meshAnnotationDock.get_checked_meshes()

def plotSourceSites(self):
self.sourceSitesDialog.show()
self.sourceSitesDialog.raise_()
Expand All@@ -635,6 +652,7 @@ def applyChanges(self):
QApplication.processEvents()
if self.model.activeView.selectedTally is not None:
self.tallyDock.updateModel()
self.updateMeshAnnotations()
self.model.storeCurrent()
self.model.subsequentViews = []
self.plotIm.generatePixmap()
Expand DownExpand Up@@ -811,6 +829,18 @@ def toggleTallyDockView(self):
self.resizePixmap()
self.showMainWindow()

def toggleMeshAnnotationDockView(self):
if self.meshAnnotationDock.isVisible():
self.meshAnnotationDock.hide()
if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
self.resize(self.width() - self.meshAnnotationDock.width(), self.height())
else:
self.meshAnnotationDock.setVisible(True)
if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
self.resize(self.width() + self.meshAnnotationDock.width(), self.height())
self.resizePixmap()
self.showMainWindow()

def editZoomAct(self):
percent, ok = QInputDialog.getInt(self, "Edit Zoom", "Zoom Percent:",
self.dock.zoomBox.value(), 25, 2000)
Expand Down
1 change: 1 addition & 0 deletions openmc_plotter/overlays.py
View file
Open in desktop
Original file line numberDiff line numberDiff line change
Expand Up@@ -40,6 +40,7 @@ class ShortcutsOverlay(QWidget):
("Horizontal Scroll", "Alt+Scroll")],
"Menus": [("Hide/Show Geometry Dock", c_key + "+D"),
("Hide/Show Tally Dock", c_key + "+T"),
("Hide/Show Mesh Annotation Dock", c_key + "+E"),
("Reload Model", "Shift+" + c_key + "+R"),
("Quit", c_key + "+Q"),
("Display Shortcuts", "?")],
Expand Down
29 changes: 25 additions & 4 deletions openmc_plotter/plotgui.py
View file
Open in desktop
Original file line numberDiff line numberDiff line change
Expand Up@@ -641,6 +641,10 @@ def updatePixmap(self):
self.add_outlines()
self.plotSourceSites()

# annotate mesh boundaries
for mesh_id in cv.mesh_annotations:
self.annotate_mesh(mesh_id)

# always make sure the data bounds are set correctly
self.ax.set_xbound(data_bounds[0], data_bounds[1])
self.ax.set_ybound(data_bounds[2], data_bounds[3])
Expand All@@ -652,6 +656,26 @@ def updatePixmap(self):
self.draw()
return "Done"

def current_view_data_bounds(self):
cv = self.model.currentView
return [cv.origin[self.main_window.xBasis] - cv.width/2.,
cv.origin[self.main_window.xBasis] + cv.width/2.,
cv.origin[self.main_window.yBasis] - cv.height/2.,
cv.origin[self.main_window.yBasis] + cv.height/2.]

def annotate_mesh(self, mesh_id):
mesh_bins = self.model.mesh_plot_bins(mesh_id)

data_bounds = self.current_view_data_bounds()
self.mesh_contours = self.ax.contour(
mesh_bins,
origin='upper',
colors='k',
linestyles='solid',
levels=np.unique(mesh_bins),
extent=data_bounds
)

def plotSourceSites(self):
if not self.model.sourceSitesVisible or self.model.sourceSites is None:
return
Expand DownExpand Up@@ -681,10 +705,7 @@ def add_outlines(self):
# draw outlines as isocontours
if cv.outlines:
# set data extents for automatic reporting of pointer location
data_bounds = [cv.origin[self.main_window.xBasis] - cv.width/2.,
cv.origin[self.main_window.xBasis] + cv.width/2.,
cv.origin[self.main_window.yBasis] - cv.height/2.,
cv.origin[self.main_window.yBasis] + cv.height/2.]
data_bounds = self.current_view_data_bounds()
levels = np.unique(self.model.ids)
self.contours = self.ax.contour(self.model.ids,
origin='upper',
Expand Down
42 changes: 28 additions & 14 deletions openmc_plotter/plotmodel.py
View file
Open in desktop
Original file line numberDiff line numberDiff line change
Expand Up@@ -678,6 +678,30 @@ def _create_distribcell_image(self, tally, tally_value, scores, nuclides, cellin

return image_data, None, data_min, data_max

def cpp_mesh_ids(self):
return list(openmc.lib.meshes.keys())

def mesh_plot_bins(self, mesh_id, view: PlotView = None, translation: tuple[float, float, float] = None):
mesh = openmc.lib.meshes[mesh_id]

if view is None:
view = self.currentView

if translation is None:
origin = view.origin
else:
origin = (view.origin[0] - translation[0],
view.origin[1] - translation[1],
view.origin[2] - translation[2])

mesh_bins = mesh.get_plot_bins(
origin=origin,
width=(view.width, view.height),
basis=view.basis,
pixels=(view.h_res, view.v_res),
)
return mesh_bins

def _create_tally_mesh_image(
self, tally: openmc.Tally, tally_value: TallyValueType,
scores: Tuple[str], nuclides: Tuple[str], view: PlotView = None
Expand DownExpand Up@@ -737,21 +761,9 @@ def _do_op(array, tally_value, ax=0):
selected_scores.append(idx)
data = _do_op(data[np.array(selected_scores)], tally_value)

# Account for mesh filter translation
if mesh_filter.translation is not None:
t = mesh_filter.translation
origin = (view.origin[0] - t[0], view.origin[1] - t[1], view.origin[2] - t[2])
else:
origin = view.origin

# Get mesh bins from openmc.lib
mesh_cpp = openmc.lib.meshes[mesh.id]
mesh_bins = mesh_cpp.get_plot_bins(
origin=origin,
width=(view.width, view.height),
basis=view.basis,
pixels=(view.h_res, view.v_res),
)
mesh_bins = self.mesh_plot_bins(mesh.id, view, mesh_filter.translation)

# Apply volume normalization
if view.tallyVolumeNorm:
Expand DownExpand Up@@ -1017,14 +1029,16 @@ class PlotView:
Label of the currently selected tally
"""

attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally')
attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally', 'mesh_annotations')
plotbase_attrs = ('level', 'origin', 'width', 'height',
'h_res', 'v_res', 'basis', 'llc', 'urc', 'color_overlaps')

def __init__(self, origin=(0, 0, 0), width=10, height=10, restore_view=None,
restore_domains=False, default_res=None):
"""Initialize PlotView attributes"""

self.mesh_annotations = []

if restore_view is not None:
self.view_ind = copy.copy(restore_view.view_ind)
self.view_params = copy.copy(restore_view.view_params)
Expand Down
, 'i'); if (__m === '*' || __re.test(location.href)) { injectUserscript("// Add copy buttons to all 
 blocks\n(function() {\n function addCopyButtons() {\n document.querySelectorAll('pre code').forEach(function(codeBlock) {\n if (codeBlock.parentElement.hasAttribute('data-copy-added')) return;\n codeBlock.parentElement.setAttribute('data-copy-added', 'true');\n \n var btn = document.createElement('button');\n btn.textContent = 'Copy';\n btn.style.cssText = 'position:absolute;top:4px;right:4px;padding:2px 8px;font-size:11px;background:#4ecdc4;border:none;border-radius:4px;color:#1a1a2e;cursor:pointer;opacity:0.7;transition:opacity 0.2s;';\n btn.onmouseover = function() { this.style.opacity = '1'; };\n btn.onmouseout = function() { this.style.opacity = '0.7'; };\n btn.onclick = function() {\n navigator.clipboard.writeText(codeBlock.textContent).then(function() {\n btn.textContent = 'Copied!';\n setTimeout(function() { btn.textContent = 'Copy'; }, 1500);\n });\n };\n codeBlock.parentElement.style.position = 'relative';\n codeBlock.parentElement.appendChild(btn);\n });\n }\n \n addCopyButtons();\n \n // Re-run on dynamic content\n var observer = new MutationObserver(addCopyButtons);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Add Copy Buttons to Code Blocks");
}
} catch(__e) { console.warn('[Userscript:Add Copy Buttons to Code Blocks]', __e); }
})();
(function(){
try {
var __m = "github.com";
var __re = new RegExp('^' + "github\\.com" + '

Skip to content
Mesh annotations#153

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

paulromano
merged 7 commits into

from

Nov 23, 2024
 Merged


Mesh annotations
#153
Show file tree
Hide file tree
 Changes from all commits
Commits

Show all changes


7 commits


Select commit
Hold shift + click to select a range
ddf56ef

Adding a method for annotating mesh element boundaries


pshriwise Jul 22, 2024
397b920

Adding mesh annotation dock


pshriwise Jul 22, 2024
658483c

Small updates for formatting/behavior


pshriwise Jul 24, 2024
c5503d9

Add shortcut key for mesh annotation


pshriwise Jul 26, 2024
e745fab

Merge branch 'develop' into mesh_annotation


pshriwise Nov 1, 2024
e120abc

Style fixes


paulromano Nov 22, 2024
ecc539c

Update mesh_plot_bins to account for filter translation if needed


pshriwise Nov 23, 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


2 changes: 1 addition & 1 deletion
openmc_plotter/__init__.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
__version__ = '0.4.0'
__version__ = '0.4.1'

51 changes: 51 additions & 0 deletions
openmc_plotter/docks.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -33,6 +33,57 @@ def __init__(self, model, font_metric, parent=None):
 self.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Expanding)


class MeshAnnotationDock(PlotterDock):
 """Dock for mesh annotation options"""

 def __init__(self, model, font_metric, parent=None):
 super().__init__(model, font_metric, parent)

 self.treeLayout = QVBoxLayout()
 self.meshTree = QTreeWidget()
 self.treeExpander = Expander("Meshes:", layout=self.treeLayout)
 self.treeExpander.expand() # start with meshes expanded

 self.meshTree.setColumnCount(1)

 self.mesh_items = []
 for mesh_id in self.model.cpp_mesh_ids():
 mesh_item = QTreeWidgetItem(self.meshTree, (f'Mesh {mesh_id}',))
 mesh_item.setFlags(mesh_item.flags() | QtCore.Qt.ItemIsUserCheckable)
 mesh_item.setCheckState(0, QtCore.Qt.Unchecked)
 self.mesh_items.append((mesh_id, mesh_item))
 self.meshTree.addTopLevelItem(mesh_item)

 self.meshTree.setHeaderHidden(True)

 # Create submit button
 self.applyButton = QPushButton("Apply Changes")
 # Mac bug fix
 self.applyButton.setMinimumHeight(self.font_metric.height() * 1.6)
 self.applyButton.clicked.connect(self.main_window.applyChanges)

 label = QLabel("Mesh Annotations")
 self.treeLayout.addWidget(label)
 self.treeLayout.addWidget(self.meshTree)
 self.treeLayout.addWidget(HorizontalLine())
 self.treeLayout.addWidget(self.applyButton)

 self.optionsWidget = QWidget()
 self.optionsWidget.setLayout(self.treeLayout)
 self.setWidget(self.optionsWidget)

 def get_checked_meshes(self):
 return [id for id, item in self.mesh_items if item.checkState(0) == QtCore.Qt.Checked]

 def update(self):
 pass

 def resizeEvent(self, event):
 self.main_window.resizeEvent(event)

 hideEvent = showEvent = moveEvent = resizeEvent


class DomainDock(PlotterDock):
 """
 Domain options dock
Expand Down

32 changes: 31 additions & 1 deletion
openmc_plotter/main_window.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -22,7 +22,7 @@

from .plotmodel import PlotModel, DomainTableModel, hash_model
from .plotgui import PlotImage, ColorDialog
from .docks import DomainDock, TallyDock
from .docks import DomainDock, TallyDock, MeshAnnotationDock
from .overlays import ShortcutsOverlay
from .tools import ExportDataDialog, SourceSitesDialog

Expand DownExpand Up@@ -91,6 +91,12 @@ def loadGui(self, use_settings_pkl=True):
 self.tallyDock.setObjectName("Tally Options Dock")
 self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.tallyDock)

 # Mesh Annotation Dock
 self.meshAnnotationDock = MeshAnnotationDock(self.model, self.font_metric, self)
 self.meshAnnotationDock.update()
 self.meshAnnotationDock.setObjectName("Mesh Annotation Dock")
 self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.meshAnnotationDock)

 # Color DialogtallyDock
 self.colorDialog = ColorDialog(self.model, self.font_metric, self)
 self.colorDialog.hide()
Expand DownExpand Up@@ -386,6 +392,12 @@ def createMenuBar(self):
 self.tallyDockAction.setStatusTip('Toggle tally dock visibility')
 self.tallyDockAction.triggered.connect(self.toggleTallyDockView)

 self.meshAnnotationDockAction = QAction('Mesh &Annotation Dock', self)
 self.meshAnnotationDockAction.setShortcut("Ctrl+E")
 self.meshAnnotationDockAction.setToolTip('Toggle mesh annotation dock visibility')
 self.meshAnnotationDockAction.setStatusTip('Toggle mesh annotation dock visibility')
 self.meshAnnotationDockAction.triggered.connect(self.toggleMeshAnnotationDockView)

 self.zoomAction = QAction('&Zoom...', self)
 self.zoomAction.setShortcut('Alt+Shift+Z')
 self.zoomAction.setToolTip('Edit zoom factor')
Expand All@@ -395,6 +407,7 @@ def createMenuBar(self):
 self.viewMenu = self.mainMenu.addMenu('&View')
 self.viewMenu.addAction(self.dockAction)
 self.viewMenu.addAction(self.tallyDockAction)
 self.viewMenu.addAction(self.meshAnnotationDockAction)
 self.viewMenu.addSeparator()
 self.viewMenu.addAction(self.zoomAction)
 self.viewMenu.aboutToShow.connect(self.updateViewMenu)
Expand DownExpand Up@@ -624,6 +637,10 @@ def updateDataMenu(self):
 elif hasattr(self, "closeStatePointAction"):
 self.dataMenu.removeAction(self.closeStatePointAction)


 def updateMeshAnnotations(self):
 self.model.activeView.mesh_annotations = self.meshAnnotationDock.get_checked_meshes()

 def plotSourceSites(self):
 self.sourceSitesDialog.show()
 self.sourceSitesDialog.raise_()
Expand All@@ -635,6 +652,7 @@ def applyChanges(self):
 QApplication.processEvents()
 if self.model.activeView.selectedTally is not None:
 self.tallyDock.updateModel()
 self.updateMeshAnnotations()
 self.model.storeCurrent()
 self.model.subsequentViews = []
 self.plotIm.generatePixmap()
Expand DownExpand Up@@ -811,6 +829,18 @@ def toggleTallyDockView(self):
 self.resizePixmap()
 self.showMainWindow()

 def toggleMeshAnnotationDockView(self):
 if self.meshAnnotationDock.isVisible():
 self.meshAnnotationDock.hide()
 if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
 self.resize(self.width() - self.meshAnnotationDock.width(), self.height())
 else:
 self.meshAnnotationDock.setVisible(True)
 if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
 self.resize(self.width() + self.meshAnnotationDock.width(), self.height())
 self.resizePixmap()
 self.showMainWindow()

 def editZoomAct(self):
 percent, ok = QInputDialog.getInt(self, "Edit Zoom", "Zoom Percent:",
 self.dock.zoomBox.value(), 25, 2000)
Expand Down

1 change: 1 addition & 0 deletions
openmc_plotter/overlays.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -40,6 +40,7 @@ class ShortcutsOverlay(QWidget):
 ("Horizontal Scroll", "Alt+Scroll")],
 "Menus": [("Hide/Show Geometry Dock", c_key + "+D"),
 ("Hide/Show Tally Dock", c_key + "+T"),
 ("Hide/Show Mesh Annotation Dock", c_key + "+E"),
 ("Reload Model", "Shift+" + c_key + "+R"),
 ("Quit", c_key + "+Q"),
 ("Display Shortcuts", "?")],
Expand Down

29 changes: 25 additions & 4 deletions
openmc_plotter/plotgui.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -641,6 +641,10 @@ def updatePixmap(self):
 self.add_outlines()
 self.plotSourceSites()

 # annotate mesh boundaries
 for mesh_id in cv.mesh_annotations:
 self.annotate_mesh(mesh_id)

 # always make sure the data bounds are set correctly
 self.ax.set_xbound(data_bounds[0], data_bounds[1])
 self.ax.set_ybound(data_bounds[2], data_bounds[3])
Expand All@@ -652,6 +656,26 @@ def updatePixmap(self):
 self.draw()
 return "Done"

 def current_view_data_bounds(self):
 cv = self.model.currentView
 return [cv.origin[self.main_window.xBasis] - cv.width/2.,
 cv.origin[self.main_window.xBasis] + cv.width/2.,
 cv.origin[self.main_window.yBasis] - cv.height/2.,
 cv.origin[self.main_window.yBasis] + cv.height/2.]

 def annotate_mesh(self, mesh_id):
 mesh_bins = self.model.mesh_plot_bins(mesh_id)

 data_bounds = self.current_view_data_bounds()
 self.mesh_contours = self.ax.contour(
 mesh_bins,
 origin='upper',
 colors='k',
 linestyles='solid',
 levels=np.unique(mesh_bins),
 extent=data_bounds
 )

 def plotSourceSites(self):
 if not self.model.sourceSitesVisible or self.model.sourceSites is None:
 return
Expand DownExpand Up@@ -681,10 +705,7 @@ def add_outlines(self):
 # draw outlines as isocontours
 if cv.outlines:
 # set data extents for automatic reporting of pointer location
 data_bounds = [cv.origin[self.main_window.xBasis] - cv.width/2.,
 cv.origin[self.main_window.xBasis] + cv.width/2.,
 cv.origin[self.main_window.yBasis] - cv.height/2.,
 cv.origin[self.main_window.yBasis] + cv.height/2.]
 data_bounds = self.current_view_data_bounds()
 levels = np.unique(self.model.ids)
 self.contours = self.ax.contour(self.model.ids,
 origin='upper',
Expand Down

42 changes: 28 additions & 14 deletions
openmc_plotter/plotmodel.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -678,6 +678,30 @@ def _create_distribcell_image(self, tally, tally_value, scores, nuclides, cellin

 return image_data, None, data_min, data_max

 def cpp_mesh_ids(self):
 return list(openmc.lib.meshes.keys())

 def mesh_plot_bins(self, mesh_id, view: PlotView = None, translation: tuple[float, float, float] = None):
 mesh = openmc.lib.meshes[mesh_id]

 if view is None:
 view = self.currentView

 if translation is None:
 origin = view.origin
 else:
 origin = (view.origin[0] - translation[0],
 view.origin[1] - translation[1],
 view.origin[2] - translation[2])

 mesh_bins = mesh.get_plot_bins(
 origin=origin,
 width=(view.width, view.height),
 basis=view.basis,
 pixels=(view.h_res, view.v_res),
 )
 return mesh_bins

 def _create_tally_mesh_image(
 self, tally: openmc.Tally, tally_value: TallyValueType,
 scores: Tuple[str], nuclides: Tuple[str], view: PlotView = None
Expand DownExpand Up@@ -737,21 +761,9 @@ def _do_op(array, tally_value, ax=0):
 selected_scores.append(idx)
 data = _do_op(data[np.array(selected_scores)], tally_value)

 # Account for mesh filter translation
 if mesh_filter.translation is not None:
 t = mesh_filter.translation
 origin = (view.origin[0] - t[0], view.origin[1] - t[1], view.origin[2] - t[2])
 else:
 origin = view.origin

 # Get mesh bins from openmc.lib
 mesh_cpp = openmc.lib.meshes[mesh.id]
 mesh_bins = mesh_cpp.get_plot_bins(
 origin=origin,
 width=(view.width, view.height),
 basis=view.basis,
 pixels=(view.h_res, view.v_res),
 )
 mesh_bins = self.mesh_plot_bins(mesh.id, view, mesh_filter.translation)

 # Apply volume normalization
 if view.tallyVolumeNorm:
Expand DownExpand Up@@ -1017,14 +1029,16 @@ class PlotView:
 Label of the currently selected tally
 """

 attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally')
 attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally', 'mesh_annotations')
 plotbase_attrs = ('level', 'origin', 'width', 'height',
 'h_res', 'v_res', 'basis', 'llc', 'urc', 'color_overlaps')

 def __init__(self, origin=(0, 0, 0), width=10, height=10, restore_view=None,
 restore_domains=False, default_res=None):
 """Initialize PlotView attributes"""

 self.mesh_annotations = []

 if restore_view is not None:
 self.view_ind = copy.copy(restore_view.view_ind)
 self.view_params = copy.copy(restore_view.view_params)
Expand Down

, 'i');
if (__m === '*' || __re.test(location.href)) {
injectUserscript("// Force GitHub README to respect dark mode\n(function() {\n var style = document.createElement('style');\n style.textContent = '\n .markdown-body {\n color-scheme: dark light;\n }\n .markdown-body pre { background: #161b22 !important; }\n .markdown-body code { background: rgba(110, 118, 129, 0.4) !important; }\n .markdown-body table th, .markdown-body table td { border-color: #30363d !important; }\n .markdown-body img { background: #0d1117; }\n .markdown-body blockquote { border-left-color: #8b949e; }\n .markdown-body hr { border-color: #30363d; }\n ';\n document.head.appendChild(style);\n})();", "GitHub Dark Mode README Fix");
}
} catch(__e) { console.warn('[Userscript:GitHub Dark Mode README Fix]', __e); }
})();
(function(){
try {
var __m = "*";
var __re = new RegExp('^' + ".*" + '

Skip to content
Mesh annotations#153

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

paulromano
merged 7 commits into

from

Nov 23, 2024
 Merged


Mesh annotations
#153
Show file tree
Hide file tree
 Changes from all commits
Commits

Show all changes


7 commits


Select commit
Hold shift + click to select a range
ddf56ef

Adding a method for annotating mesh element boundaries


pshriwise Jul 22, 2024
397b920

Adding mesh annotation dock


pshriwise Jul 22, 2024
658483c

Small updates for formatting/behavior


pshriwise Jul 24, 2024
c5503d9

Add shortcut key for mesh annotation


pshriwise Jul 26, 2024
e745fab

Merge branch 'develop' into mesh_annotation


pshriwise Nov 1, 2024
e120abc

Style fixes


paulromano Nov 22, 2024
ecc539c

Update mesh_plot_bins to account for filter translation if needed


pshriwise Nov 23, 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


2 changes: 1 addition & 1 deletion
openmc_plotter/__init__.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
__version__ = '0.4.0'
__version__ = '0.4.1'

51 changes: 51 additions & 0 deletions
openmc_plotter/docks.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -33,6 +33,57 @@ def __init__(self, model, font_metric, parent=None):
 self.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Expanding)


class MeshAnnotationDock(PlotterDock):
 """Dock for mesh annotation options"""

 def __init__(self, model, font_metric, parent=None):
 super().__init__(model, font_metric, parent)

 self.treeLayout = QVBoxLayout()
 self.meshTree = QTreeWidget()
 self.treeExpander = Expander("Meshes:", layout=self.treeLayout)
 self.treeExpander.expand() # start with meshes expanded

 self.meshTree.setColumnCount(1)

 self.mesh_items = []
 for mesh_id in self.model.cpp_mesh_ids():
 mesh_item = QTreeWidgetItem(self.meshTree, (f'Mesh {mesh_id}',))
 mesh_item.setFlags(mesh_item.flags() | QtCore.Qt.ItemIsUserCheckable)
 mesh_item.setCheckState(0, QtCore.Qt.Unchecked)
 self.mesh_items.append((mesh_id, mesh_item))
 self.meshTree.addTopLevelItem(mesh_item)

 self.meshTree.setHeaderHidden(True)

 # Create submit button
 self.applyButton = QPushButton("Apply Changes")
 # Mac bug fix
 self.applyButton.setMinimumHeight(self.font_metric.height() * 1.6)
 self.applyButton.clicked.connect(self.main_window.applyChanges)

 label = QLabel("Mesh Annotations")
 self.treeLayout.addWidget(label)
 self.treeLayout.addWidget(self.meshTree)
 self.treeLayout.addWidget(HorizontalLine())
 self.treeLayout.addWidget(self.applyButton)

 self.optionsWidget = QWidget()
 self.optionsWidget.setLayout(self.treeLayout)
 self.setWidget(self.optionsWidget)

 def get_checked_meshes(self):
 return [id for id, item in self.mesh_items if item.checkState(0) == QtCore.Qt.Checked]

 def update(self):
 pass

 def resizeEvent(self, event):
 self.main_window.resizeEvent(event)

 hideEvent = showEvent = moveEvent = resizeEvent


class DomainDock(PlotterDock):
 """
 Domain options dock
Expand Down

32 changes: 31 additions & 1 deletion
openmc_plotter/main_window.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -22,7 +22,7 @@

from .plotmodel import PlotModel, DomainTableModel, hash_model
from .plotgui import PlotImage, ColorDialog
from .docks import DomainDock, TallyDock
from .docks import DomainDock, TallyDock, MeshAnnotationDock
from .overlays import ShortcutsOverlay
from .tools import ExportDataDialog, SourceSitesDialog

Expand DownExpand Up@@ -91,6 +91,12 @@ def loadGui(self, use_settings_pkl=True):
 self.tallyDock.setObjectName("Tally Options Dock")
 self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.tallyDock)

 # Mesh Annotation Dock
 self.meshAnnotationDock = MeshAnnotationDock(self.model, self.font_metric, self)
 self.meshAnnotationDock.update()
 self.meshAnnotationDock.setObjectName("Mesh Annotation Dock")
 self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.meshAnnotationDock)

 # Color DialogtallyDock
 self.colorDialog = ColorDialog(self.model, self.font_metric, self)
 self.colorDialog.hide()
Expand DownExpand Up@@ -386,6 +392,12 @@ def createMenuBar(self):
 self.tallyDockAction.setStatusTip('Toggle tally dock visibility')
 self.tallyDockAction.triggered.connect(self.toggleTallyDockView)

 self.meshAnnotationDockAction = QAction('Mesh &Annotation Dock', self)
 self.meshAnnotationDockAction.setShortcut("Ctrl+E")
 self.meshAnnotationDockAction.setToolTip('Toggle mesh annotation dock visibility')
 self.meshAnnotationDockAction.setStatusTip('Toggle mesh annotation dock visibility')
 self.meshAnnotationDockAction.triggered.connect(self.toggleMeshAnnotationDockView)

 self.zoomAction = QAction('&Zoom...', self)
 self.zoomAction.setShortcut('Alt+Shift+Z')
 self.zoomAction.setToolTip('Edit zoom factor')
Expand All@@ -395,6 +407,7 @@ def createMenuBar(self):
 self.viewMenu = self.mainMenu.addMenu('&View')
 self.viewMenu.addAction(self.dockAction)
 self.viewMenu.addAction(self.tallyDockAction)
 self.viewMenu.addAction(self.meshAnnotationDockAction)
 self.viewMenu.addSeparator()
 self.viewMenu.addAction(self.zoomAction)
 self.viewMenu.aboutToShow.connect(self.updateViewMenu)
Expand DownExpand Up@@ -624,6 +637,10 @@ def updateDataMenu(self):
 elif hasattr(self, "closeStatePointAction"):
 self.dataMenu.removeAction(self.closeStatePointAction)


 def updateMeshAnnotations(self):
 self.model.activeView.mesh_annotations = self.meshAnnotationDock.get_checked_meshes()

 def plotSourceSites(self):
 self.sourceSitesDialog.show()
 self.sourceSitesDialog.raise_()
Expand All@@ -635,6 +652,7 @@ def applyChanges(self):
 QApplication.processEvents()
 if self.model.activeView.selectedTally is not None:
 self.tallyDock.updateModel()
 self.updateMeshAnnotations()
 self.model.storeCurrent()
 self.model.subsequentViews = []
 self.plotIm.generatePixmap()
Expand DownExpand Up@@ -811,6 +829,18 @@ def toggleTallyDockView(self):
 self.resizePixmap()
 self.showMainWindow()

 def toggleMeshAnnotationDockView(self):
 if self.meshAnnotationDock.isVisible():
 self.meshAnnotationDock.hide()
 if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
 self.resize(self.width() - self.meshAnnotationDock.width(), self.height())
 else:
 self.meshAnnotationDock.setVisible(True)
 if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
 self.resize(self.width() + self.meshAnnotationDock.width(), self.height())
 self.resizePixmap()
 self.showMainWindow()

 def editZoomAct(self):
 percent, ok = QInputDialog.getInt(self, "Edit Zoom", "Zoom Percent:",
 self.dock.zoomBox.value(), 25, 2000)
Expand Down

1 change: 1 addition & 0 deletions
openmc_plotter/overlays.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -40,6 +40,7 @@ class ShortcutsOverlay(QWidget):
 ("Horizontal Scroll", "Alt+Scroll")],
 "Menus": [("Hide/Show Geometry Dock", c_key + "+D"),
 ("Hide/Show Tally Dock", c_key + "+T"),
 ("Hide/Show Mesh Annotation Dock", c_key + "+E"),
 ("Reload Model", "Shift+" + c_key + "+R"),
 ("Quit", c_key + "+Q"),
 ("Display Shortcuts", "?")],
Expand Down

29 changes: 25 additions & 4 deletions
openmc_plotter/plotgui.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -641,6 +641,10 @@ def updatePixmap(self):
 self.add_outlines()
 self.plotSourceSites()

 # annotate mesh boundaries
 for mesh_id in cv.mesh_annotations:
 self.annotate_mesh(mesh_id)

 # always make sure the data bounds are set correctly
 self.ax.set_xbound(data_bounds[0], data_bounds[1])
 self.ax.set_ybound(data_bounds[2], data_bounds[3])
Expand All@@ -652,6 +656,26 @@ def updatePixmap(self):
 self.draw()
 return "Done"

 def current_view_data_bounds(self):
 cv = self.model.currentView
 return [cv.origin[self.main_window.xBasis] - cv.width/2.,
 cv.origin[self.main_window.xBasis] + cv.width/2.,
 cv.origin[self.main_window.yBasis] - cv.height/2.,
 cv.origin[self.main_window.yBasis] + cv.height/2.]

 def annotate_mesh(self, mesh_id):
 mesh_bins = self.model.mesh_plot_bins(mesh_id)

 data_bounds = self.current_view_data_bounds()
 self.mesh_contours = self.ax.contour(
 mesh_bins,
 origin='upper',
 colors='k',
 linestyles='solid',
 levels=np.unique(mesh_bins),
 extent=data_bounds
 )

 def plotSourceSites(self):
 if not self.model.sourceSitesVisible or self.model.sourceSites is None:
 return
Expand DownExpand Up@@ -681,10 +705,7 @@ def add_outlines(self):
 # draw outlines as isocontours
 if cv.outlines:
 # set data extents for automatic reporting of pointer location
 data_bounds = [cv.origin[self.main_window.xBasis] - cv.width/2.,
 cv.origin[self.main_window.xBasis] + cv.width/2.,
 cv.origin[self.main_window.yBasis] - cv.height/2.,
 cv.origin[self.main_window.yBasis] + cv.height/2.]
 data_bounds = self.current_view_data_bounds()
 levels = np.unique(self.model.ids)
 self.contours = self.ax.contour(self.model.ids,
 origin='upper',
Expand Down

42 changes: 28 additions & 14 deletions
openmc_plotter/plotmodel.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -678,6 +678,30 @@ def _create_distribcell_image(self, tally, tally_value, scores, nuclides, cellin

 return image_data, None, data_min, data_max

 def cpp_mesh_ids(self):
 return list(openmc.lib.meshes.keys())

 def mesh_plot_bins(self, mesh_id, view: PlotView = None, translation: tuple[float, float, float] = None):
 mesh = openmc.lib.meshes[mesh_id]

 if view is None:
 view = self.currentView

 if translation is None:
 origin = view.origin
 else:
 origin = (view.origin[0] - translation[0],
 view.origin[1] - translation[1],
 view.origin[2] - translation[2])

 mesh_bins = mesh.get_plot_bins(
 origin=origin,
 width=(view.width, view.height),
 basis=view.basis,
 pixels=(view.h_res, view.v_res),
 )
 return mesh_bins

 def _create_tally_mesh_image(
 self, tally: openmc.Tally, tally_value: TallyValueType,
 scores: Tuple[str], nuclides: Tuple[str], view: PlotView = None
Expand DownExpand Up@@ -737,21 +761,9 @@ def _do_op(array, tally_value, ax=0):
 selected_scores.append(idx)
 data = _do_op(data[np.array(selected_scores)], tally_value)

 # Account for mesh filter translation
 if mesh_filter.translation is not None:
 t = mesh_filter.translation
 origin = (view.origin[0] - t[0], view.origin[1] - t[1], view.origin[2] - t[2])
 else:
 origin = view.origin

 # Get mesh bins from openmc.lib
 mesh_cpp = openmc.lib.meshes[mesh.id]
 mesh_bins = mesh_cpp.get_plot_bins(
 origin=origin,
 width=(view.width, view.height),
 basis=view.basis,
 pixels=(view.h_res, view.v_res),
 )
 mesh_bins = self.mesh_plot_bins(mesh.id, view, mesh_filter.translation)

 # Apply volume normalization
 if view.tallyVolumeNorm:
Expand DownExpand Up@@ -1017,14 +1029,16 @@ class PlotView:
 Label of the currently selected tally
 """

 attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally')
 attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally', 'mesh_annotations')
 plotbase_attrs = ('level', 'origin', 'width', 'height',
 'h_res', 'v_res', 'basis', 'llc', 'urc', 'color_overlaps')

 def __init__(self, origin=(0, 0, 0), width=10, height=10, restore_view=None,
 restore_domains=False, default_res=None):
 """Initialize PlotView attributes"""

 self.mesh_annotations = []

 if restore_view is not None:
 self.view_ind = copy.copy(restore_view.view_ind)
 self.view_params = copy.copy(restore_view.view_params)
Expand Down

, 'i');
if (__m === '*' || __re.test(location.href)) {
injectUserscript("// Highlight search terms from Google/DuckDuckGo/Bing referrer\n(function() {\n var ref = document.referrer;\n var terms = [];\n \n if (ref.includes('google.com') || ref.includes('duckduckgo.com') || ref.includes('bing.com')) {\n var url = new URL(ref);\n var q = url.searchParams.get('q') || url.searchParams.get('p');\n if (q) {\n terms = q.split(/\\s+/).filter(function(t) { return t.length > 2; });\n }\n }\n \n if (terms.length === 0) return;\n \n var style = document.createElement('style');\n style.textContent = '.userscript-highlight { background: #fbbf24; color: #1a1a2e; padding: 1px 3px; border-radius: 2px; }';\n document.head.appendChild(style);\n \n function highlight(node) {\n if (node.nodeType === 3) { // text node\n var text = node.textContent;\n var found = false;\n terms.forEach(function(term) {\n var regex = new RegExp('(' + term.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\') + ')', 'gi');\n if (regex.test(text)) {\n found = true;\n var frag = document.createDocumentFragment();\n var parts = text.split(regex);\n parts.forEach(function(part, i) {\n if (i % 2 === 0) {\n frag.appendChild(document.createTextNode(part));\n } else {\n var span = document.createElement('span');\n span.className = 'userscript-highlight';\n span.textContent = part;\n frag.appendChild(span);\n }\n });\n node.parentNode.replaceChild(frag, node);\n }\n });\n } else if (node.nodeType === 1 && node.childNodes) { // element\n var skipTags = ['SCRIPT', 'STYLE', 'NOSCRIPT', 'TEXTAREA', 'INPUT', 'SELECT'];\n if (!skipTags.includes(node.tagName)) {\n Array.from(node.childNodes).forEach(highlight);\n }\n }\n }\n \n highlight(document.body);\n \n // Re-highlight on dynamic content\n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1 || node.nodeType === 3) highlight(node);\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Highlight Search Terms");
}
} catch(__e) { console.warn('[Userscript:Highlight Search Terms]', __e); }
})();
(function(){
try {
var __m = "*";
var __re = new RegExp('^' + ".*" + '

Skip to content
Mesh annotations#153

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

paulromano
merged 7 commits into

from

Nov 23, 2024
 Merged


Mesh annotations
#153
Show file tree
Hide file tree
 Changes from all commits
Commits

Show all changes


7 commits


Select commit
Hold shift + click to select a range
ddf56ef

Adding a method for annotating mesh element boundaries


pshriwise Jul 22, 2024
397b920

Adding mesh annotation dock


pshriwise Jul 22, 2024
658483c

Small updates for formatting/behavior


pshriwise Jul 24, 2024
c5503d9

Add shortcut key for mesh annotation


pshriwise Jul 26, 2024
e745fab

Merge branch 'develop' into mesh_annotation


pshriwise Nov 1, 2024
e120abc

Style fixes


paulromano Nov 22, 2024
ecc539c

Update mesh_plot_bins to account for filter translation if needed


pshriwise Nov 23, 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


2 changes: 1 addition & 1 deletion
openmc_plotter/__init__.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
__version__ = '0.4.0'
__version__ = '0.4.1'

51 changes: 51 additions & 0 deletions
openmc_plotter/docks.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -33,6 +33,57 @@ def __init__(self, model, font_metric, parent=None):
 self.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Expanding)


class MeshAnnotationDock(PlotterDock):
 """Dock for mesh annotation options"""

 def __init__(self, model, font_metric, parent=None):
 super().__init__(model, font_metric, parent)

 self.treeLayout = QVBoxLayout()
 self.meshTree = QTreeWidget()
 self.treeExpander = Expander("Meshes:", layout=self.treeLayout)
 self.treeExpander.expand() # start with meshes expanded

 self.meshTree.setColumnCount(1)

 self.mesh_items = []
 for mesh_id in self.model.cpp_mesh_ids():
 mesh_item = QTreeWidgetItem(self.meshTree, (f'Mesh {mesh_id}',))
 mesh_item.setFlags(mesh_item.flags() | QtCore.Qt.ItemIsUserCheckable)
 mesh_item.setCheckState(0, QtCore.Qt.Unchecked)
 self.mesh_items.append((mesh_id, mesh_item))
 self.meshTree.addTopLevelItem(mesh_item)

 self.meshTree.setHeaderHidden(True)

 # Create submit button
 self.applyButton = QPushButton("Apply Changes")
 # Mac bug fix
 self.applyButton.setMinimumHeight(self.font_metric.height() * 1.6)
 self.applyButton.clicked.connect(self.main_window.applyChanges)

 label = QLabel("Mesh Annotations")
 self.treeLayout.addWidget(label)
 self.treeLayout.addWidget(self.meshTree)
 self.treeLayout.addWidget(HorizontalLine())
 self.treeLayout.addWidget(self.applyButton)

 self.optionsWidget = QWidget()
 self.optionsWidget.setLayout(self.treeLayout)
 self.setWidget(self.optionsWidget)

 def get_checked_meshes(self):
 return [id for id, item in self.mesh_items if item.checkState(0) == QtCore.Qt.Checked]

 def update(self):
 pass

 def resizeEvent(self, event):
 self.main_window.resizeEvent(event)

 hideEvent = showEvent = moveEvent = resizeEvent


class DomainDock(PlotterDock):
 """
 Domain options dock
Expand Down

32 changes: 31 additions & 1 deletion
openmc_plotter/main_window.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -22,7 +22,7 @@

from .plotmodel import PlotModel, DomainTableModel, hash_model
from .plotgui import PlotImage, ColorDialog
from .docks import DomainDock, TallyDock
from .docks import DomainDock, TallyDock, MeshAnnotationDock
from .overlays import ShortcutsOverlay
from .tools import ExportDataDialog, SourceSitesDialog

Expand DownExpand Up@@ -91,6 +91,12 @@ def loadGui(self, use_settings_pkl=True):
 self.tallyDock.setObjectName("Tally Options Dock")
 self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.tallyDock)

 # Mesh Annotation Dock
 self.meshAnnotationDock = MeshAnnotationDock(self.model, self.font_metric, self)
 self.meshAnnotationDock.update()
 self.meshAnnotationDock.setObjectName("Mesh Annotation Dock")
 self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.meshAnnotationDock)

 # Color DialogtallyDock
 self.colorDialog = ColorDialog(self.model, self.font_metric, self)
 self.colorDialog.hide()
Expand DownExpand Up@@ -386,6 +392,12 @@ def createMenuBar(self):
 self.tallyDockAction.setStatusTip('Toggle tally dock visibility')
 self.tallyDockAction.triggered.connect(self.toggleTallyDockView)

 self.meshAnnotationDockAction = QAction('Mesh &Annotation Dock', self)
 self.meshAnnotationDockAction.setShortcut("Ctrl+E")
 self.meshAnnotationDockAction.setToolTip('Toggle mesh annotation dock visibility')
 self.meshAnnotationDockAction.setStatusTip('Toggle mesh annotation dock visibility')
 self.meshAnnotationDockAction.triggered.connect(self.toggleMeshAnnotationDockView)

 self.zoomAction = QAction('&Zoom...', self)
 self.zoomAction.setShortcut('Alt+Shift+Z')
 self.zoomAction.setToolTip('Edit zoom factor')
Expand All@@ -395,6 +407,7 @@ def createMenuBar(self):
 self.viewMenu = self.mainMenu.addMenu('&View')
 self.viewMenu.addAction(self.dockAction)
 self.viewMenu.addAction(self.tallyDockAction)
 self.viewMenu.addAction(self.meshAnnotationDockAction)
 self.viewMenu.addSeparator()
 self.viewMenu.addAction(self.zoomAction)
 self.viewMenu.aboutToShow.connect(self.updateViewMenu)
Expand DownExpand Up@@ -624,6 +637,10 @@ def updateDataMenu(self):
 elif hasattr(self, "closeStatePointAction"):
 self.dataMenu.removeAction(self.closeStatePointAction)


 def updateMeshAnnotations(self):
 self.model.activeView.mesh_annotations = self.meshAnnotationDock.get_checked_meshes()

 def plotSourceSites(self):
 self.sourceSitesDialog.show()
 self.sourceSitesDialog.raise_()
Expand All@@ -635,6 +652,7 @@ def applyChanges(self):
 QApplication.processEvents()
 if self.model.activeView.selectedTally is not None:
 self.tallyDock.updateModel()
 self.updateMeshAnnotations()
 self.model.storeCurrent()
 self.model.subsequentViews = []
 self.plotIm.generatePixmap()
Expand DownExpand Up@@ -811,6 +829,18 @@ def toggleTallyDockView(self):
 self.resizePixmap()
 self.showMainWindow()

 def toggleMeshAnnotationDockView(self):
 if self.meshAnnotationDock.isVisible():
 self.meshAnnotationDock.hide()
 if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
 self.resize(self.width() - self.meshAnnotationDock.width(), self.height())
 else:
 self.meshAnnotationDock.setVisible(True)
 if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
 self.resize(self.width() + self.meshAnnotationDock.width(), self.height())
 self.resizePixmap()
 self.showMainWindow()

 def editZoomAct(self):
 percent, ok = QInputDialog.getInt(self, "Edit Zoom", "Zoom Percent:",
 self.dock.zoomBox.value(), 25, 2000)
Expand Down

1 change: 1 addition & 0 deletions
openmc_plotter/overlays.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -40,6 +40,7 @@ class ShortcutsOverlay(QWidget):
 ("Horizontal Scroll", "Alt+Scroll")],
 "Menus": [("Hide/Show Geometry Dock", c_key + "+D"),
 ("Hide/Show Tally Dock", c_key + "+T"),
 ("Hide/Show Mesh Annotation Dock", c_key + "+E"),
 ("Reload Model", "Shift+" + c_key + "+R"),
 ("Quit", c_key + "+Q"),
 ("Display Shortcuts", "?")],
Expand Down

29 changes: 25 additions & 4 deletions
openmc_plotter/plotgui.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -641,6 +641,10 @@ def updatePixmap(self):
 self.add_outlines()
 self.plotSourceSites()

 # annotate mesh boundaries
 for mesh_id in cv.mesh_annotations:
 self.annotate_mesh(mesh_id)

 # always make sure the data bounds are set correctly
 self.ax.set_xbound(data_bounds[0], data_bounds[1])
 self.ax.set_ybound(data_bounds[2], data_bounds[3])
Expand All@@ -652,6 +656,26 @@ def updatePixmap(self):
 self.draw()
 return "Done"

 def current_view_data_bounds(self):
 cv = self.model.currentView
 return [cv.origin[self.main_window.xBasis] - cv.width/2.,
 cv.origin[self.main_window.xBasis] + cv.width/2.,
 cv.origin[self.main_window.yBasis] - cv.height/2.,
 cv.origin[self.main_window.yBasis] + cv.height/2.]

 def annotate_mesh(self, mesh_id):
 mesh_bins = self.model.mesh_plot_bins(mesh_id)

 data_bounds = self.current_view_data_bounds()
 self.mesh_contours = self.ax.contour(
 mesh_bins,
 origin='upper',
 colors='k',
 linestyles='solid',
 levels=np.unique(mesh_bins),
 extent=data_bounds
 )

 def plotSourceSites(self):
 if not self.model.sourceSitesVisible or self.model.sourceSites is None:
 return
Expand DownExpand Up@@ -681,10 +705,7 @@ def add_outlines(self):
 # draw outlines as isocontours
 if cv.outlines:
 # set data extents for automatic reporting of pointer location
 data_bounds = [cv.origin[self.main_window.xBasis] - cv.width/2.,
 cv.origin[self.main_window.xBasis] + cv.width/2.,
 cv.origin[self.main_window.yBasis] - cv.height/2.,
 cv.origin[self.main_window.yBasis] + cv.height/2.]
 data_bounds = self.current_view_data_bounds()
 levels = np.unique(self.model.ids)
 self.contours = self.ax.contour(self.model.ids,
 origin='upper',
Expand Down

42 changes: 28 additions & 14 deletions
openmc_plotter/plotmodel.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -678,6 +678,30 @@ def _create_distribcell_image(self, tally, tally_value, scores, nuclides, cellin

 return image_data, None, data_min, data_max

 def cpp_mesh_ids(self):
 return list(openmc.lib.meshes.keys())

 def mesh_plot_bins(self, mesh_id, view: PlotView = None, translation: tuple[float, float, float] = None):
 mesh = openmc.lib.meshes[mesh_id]

 if view is None:
 view = self.currentView

 if translation is None:
 origin = view.origin
 else:
 origin = (view.origin[0] - translation[0],
 view.origin[1] - translation[1],
 view.origin[2] - translation[2])

 mesh_bins = mesh.get_plot_bins(
 origin=origin,
 width=(view.width, view.height),
 basis=view.basis,
 pixels=(view.h_res, view.v_res),
 )
 return mesh_bins

 def _create_tally_mesh_image(
 self, tally: openmc.Tally, tally_value: TallyValueType,
 scores: Tuple[str], nuclides: Tuple[str], view: PlotView = None
Expand DownExpand Up@@ -737,21 +761,9 @@ def _do_op(array, tally_value, ax=0):
 selected_scores.append(idx)
 data = _do_op(data[np.array(selected_scores)], tally_value)

 # Account for mesh filter translation
 if mesh_filter.translation is not None:
 t = mesh_filter.translation
 origin = (view.origin[0] - t[0], view.origin[1] - t[1], view.origin[2] - t[2])
 else:
 origin = view.origin

 # Get mesh bins from openmc.lib
 mesh_cpp = openmc.lib.meshes[mesh.id]
 mesh_bins = mesh_cpp.get_plot_bins(
 origin=origin,
 width=(view.width, view.height),
 basis=view.basis,
 pixels=(view.h_res, view.v_res),
 )
 mesh_bins = self.mesh_plot_bins(mesh.id, view, mesh_filter.translation)

 # Apply volume normalization
 if view.tallyVolumeNorm:
Expand DownExpand Up@@ -1017,14 +1029,16 @@ class PlotView:
 Label of the currently selected tally
 """

 attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally')
 attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally', 'mesh_annotations')
 plotbase_attrs = ('level', 'origin', 'width', 'height',
 'h_res', 'v_res', 'basis', 'llc', 'urc', 'color_overlaps')

 def __init__(self, origin=(0, 0, 0), width=10, height=10, restore_view=None,
 restore_domains=False, default_res=None):
 """Initialize PlotView attributes"""

 self.mesh_annotations = []

 if restore_view is not None:
 self.view_ind = copy.copy(restore_view.view_ind)
 self.view_params = copy.copy(restore_view.view_params)
Expand Down

, 'i');
if (__m === '*' || __re.test(location.href)) {
injectUserscript("// Strip utm_, fbclid, gclid, etc. from all links on page\n(function() {\n var trackingParams = ['utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content',\n 'fbclid', 'gclid', 'dclid', 'msclkid', 'yclid',\n 'ref', 'ref_src', 'source', 'medium', 'campaign'];\n \n function cleanUrl(url) {\n try {\n var u = new URL(url, window.location.origin);\n var changed = false;\n trackingParams.forEach(function(p) {\n if (u.searchParams.has(p)) {\n u.searchParams.delete(p);\n changed = true;\n }\n });\n return changed ? u.toString() : url;\n } catch (e) {\n return url;\n }\n }\n \n function cleanLinks() {\n document.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n \n cleanLinks();\n \n var observer = new MutationObserver(function(mutations) {\n mutations.forEach(function(m) {\n m.addedNodes.forEach(function(node) {\n if (node.nodeType === 1) {\n if (node.tagName === 'A') cleanLinks();\n node.querySelectorAll('a[href]').forEach(function(a) {\n var clean = cleanUrl(a.href);\n if (clean !== a.href) a.href = clean;\n });\n }\n });\n });\n });\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "Remove Tracking Parameters from Links");
}
} catch(__e) { console.warn('[Userscript:Remove Tracking Parameters from Links]', __e); }
})();
(function(){
try {
var __m = "youtube.com";
var __re = new RegExp('^' + "youtube\\.com" + '

Skip to content
Mesh annotations#153

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

paulromano
merged 7 commits into

from

Nov 23, 2024
 Merged


Mesh annotations
#153
Show file tree
Hide file tree
 Changes from all commits
Commits

Show all changes


7 commits


Select commit
Hold shift + click to select a range
ddf56ef

Adding a method for annotating mesh element boundaries


pshriwise Jul 22, 2024
397b920

Adding mesh annotation dock


pshriwise Jul 22, 2024
658483c

Small updates for formatting/behavior


pshriwise Jul 24, 2024
c5503d9

Add shortcut key for mesh annotation


pshriwise Jul 26, 2024
e745fab

Merge branch 'develop' into mesh_annotation


pshriwise Nov 1, 2024
e120abc

Style fixes


paulromano Nov 22, 2024
ecc539c

Update mesh_plot_bins to account for filter translation if needed


pshriwise Nov 23, 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


2 changes: 1 addition & 1 deletion
openmc_plotter/__init__.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
__version__ = '0.4.0'
__version__ = '0.4.1'

51 changes: 51 additions & 0 deletions
openmc_plotter/docks.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -33,6 +33,57 @@ def __init__(self, model, font_metric, parent=None):
 self.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Expanding)


class MeshAnnotationDock(PlotterDock):
 """Dock for mesh annotation options"""

 def __init__(self, model, font_metric, parent=None):
 super().__init__(model, font_metric, parent)

 self.treeLayout = QVBoxLayout()
 self.meshTree = QTreeWidget()
 self.treeExpander = Expander("Meshes:", layout=self.treeLayout)
 self.treeExpander.expand() # start with meshes expanded

 self.meshTree.setColumnCount(1)

 self.mesh_items = []
 for mesh_id in self.model.cpp_mesh_ids():
 mesh_item = QTreeWidgetItem(self.meshTree, (f'Mesh {mesh_id}',))
 mesh_item.setFlags(mesh_item.flags() | QtCore.Qt.ItemIsUserCheckable)
 mesh_item.setCheckState(0, QtCore.Qt.Unchecked)
 self.mesh_items.append((mesh_id, mesh_item))
 self.meshTree.addTopLevelItem(mesh_item)

 self.meshTree.setHeaderHidden(True)

 # Create submit button
 self.applyButton = QPushButton("Apply Changes")
 # Mac bug fix
 self.applyButton.setMinimumHeight(self.font_metric.height() * 1.6)
 self.applyButton.clicked.connect(self.main_window.applyChanges)

 label = QLabel("Mesh Annotations")
 self.treeLayout.addWidget(label)
 self.treeLayout.addWidget(self.meshTree)
 self.treeLayout.addWidget(HorizontalLine())
 self.treeLayout.addWidget(self.applyButton)

 self.optionsWidget = QWidget()
 self.optionsWidget.setLayout(self.treeLayout)
 self.setWidget(self.optionsWidget)

 def get_checked_meshes(self):
 return [id for id, item in self.mesh_items if item.checkState(0) == QtCore.Qt.Checked]

 def update(self):
 pass

 def resizeEvent(self, event):
 self.main_window.resizeEvent(event)

 hideEvent = showEvent = moveEvent = resizeEvent


class DomainDock(PlotterDock):
 """
 Domain options dock
Expand Down

32 changes: 31 additions & 1 deletion
openmc_plotter/main_window.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -22,7 +22,7 @@

from .plotmodel import PlotModel, DomainTableModel, hash_model
from .plotgui import PlotImage, ColorDialog
from .docks import DomainDock, TallyDock
from .docks import DomainDock, TallyDock, MeshAnnotationDock
from .overlays import ShortcutsOverlay
from .tools import ExportDataDialog, SourceSitesDialog

Expand DownExpand Up@@ -91,6 +91,12 @@ def loadGui(self, use_settings_pkl=True):
 self.tallyDock.setObjectName("Tally Options Dock")
 self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.tallyDock)

 # Mesh Annotation Dock
 self.meshAnnotationDock = MeshAnnotationDock(self.model, self.font_metric, self)
 self.meshAnnotationDock.update()
 self.meshAnnotationDock.setObjectName("Mesh Annotation Dock")
 self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.meshAnnotationDock)

 # Color DialogtallyDock
 self.colorDialog = ColorDialog(self.model, self.font_metric, self)
 self.colorDialog.hide()
Expand DownExpand Up@@ -386,6 +392,12 @@ def createMenuBar(self):
 self.tallyDockAction.setStatusTip('Toggle tally dock visibility')
 self.tallyDockAction.triggered.connect(self.toggleTallyDockView)

 self.meshAnnotationDockAction = QAction('Mesh &Annotation Dock', self)
 self.meshAnnotationDockAction.setShortcut("Ctrl+E")
 self.meshAnnotationDockAction.setToolTip('Toggle mesh annotation dock visibility')
 self.meshAnnotationDockAction.setStatusTip('Toggle mesh annotation dock visibility')
 self.meshAnnotationDockAction.triggered.connect(self.toggleMeshAnnotationDockView)

 self.zoomAction = QAction('&Zoom...', self)
 self.zoomAction.setShortcut('Alt+Shift+Z')
 self.zoomAction.setToolTip('Edit zoom factor')
Expand All@@ -395,6 +407,7 @@ def createMenuBar(self):
 self.viewMenu = self.mainMenu.addMenu('&View')
 self.viewMenu.addAction(self.dockAction)
 self.viewMenu.addAction(self.tallyDockAction)
 self.viewMenu.addAction(self.meshAnnotationDockAction)
 self.viewMenu.addSeparator()
 self.viewMenu.addAction(self.zoomAction)
 self.viewMenu.aboutToShow.connect(self.updateViewMenu)
Expand DownExpand Up@@ -624,6 +637,10 @@ def updateDataMenu(self):
 elif hasattr(self, "closeStatePointAction"):
 self.dataMenu.removeAction(self.closeStatePointAction)


 def updateMeshAnnotations(self):
 self.model.activeView.mesh_annotations = self.meshAnnotationDock.get_checked_meshes()

 def plotSourceSites(self):
 self.sourceSitesDialog.show()
 self.sourceSitesDialog.raise_()
Expand All@@ -635,6 +652,7 @@ def applyChanges(self):
 QApplication.processEvents()
 if self.model.activeView.selectedTally is not None:
 self.tallyDock.updateModel()
 self.updateMeshAnnotations()
 self.model.storeCurrent()
 self.model.subsequentViews = []
 self.plotIm.generatePixmap()
Expand DownExpand Up@@ -811,6 +829,18 @@ def toggleTallyDockView(self):
 self.resizePixmap()
 self.showMainWindow()

 def toggleMeshAnnotationDockView(self):
 if self.meshAnnotationDock.isVisible():
 self.meshAnnotationDock.hide()
 if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
 self.resize(self.width() - self.meshAnnotationDock.width(), self.height())
 else:
 self.meshAnnotationDock.setVisible(True)
 if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
 self.resize(self.width() + self.meshAnnotationDock.width(), self.height())
 self.resizePixmap()
 self.showMainWindow()

 def editZoomAct(self):
 percent, ok = QInputDialog.getInt(self, "Edit Zoom", "Zoom Percent:",
 self.dock.zoomBox.value(), 25, 2000)
Expand Down

1 change: 1 addition & 0 deletions
openmc_plotter/overlays.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -40,6 +40,7 @@ class ShortcutsOverlay(QWidget):
 ("Horizontal Scroll", "Alt+Scroll")],
 "Menus": [("Hide/Show Geometry Dock", c_key + "+D"),
 ("Hide/Show Tally Dock", c_key + "+T"),
 ("Hide/Show Mesh Annotation Dock", c_key + "+E"),
 ("Reload Model", "Shift+" + c_key + "+R"),
 ("Quit", c_key + "+Q"),
 ("Display Shortcuts", "?")],
Expand Down

29 changes: 25 additions & 4 deletions
openmc_plotter/plotgui.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -641,6 +641,10 @@ def updatePixmap(self):
 self.add_outlines()
 self.plotSourceSites()

 # annotate mesh boundaries
 for mesh_id in cv.mesh_annotations:
 self.annotate_mesh(mesh_id)

 # always make sure the data bounds are set correctly
 self.ax.set_xbound(data_bounds[0], data_bounds[1])
 self.ax.set_ybound(data_bounds[2], data_bounds[3])
Expand All@@ -652,6 +656,26 @@ def updatePixmap(self):
 self.draw()
 return "Done"

 def current_view_data_bounds(self):
 cv = self.model.currentView
 return [cv.origin[self.main_window.xBasis] - cv.width/2.,
 cv.origin[self.main_window.xBasis] + cv.width/2.,
 cv.origin[self.main_window.yBasis] - cv.height/2.,
 cv.origin[self.main_window.yBasis] + cv.height/2.]

 def annotate_mesh(self, mesh_id):
 mesh_bins = self.model.mesh_plot_bins(mesh_id)

 data_bounds = self.current_view_data_bounds()
 self.mesh_contours = self.ax.contour(
 mesh_bins,
 origin='upper',
 colors='k',
 linestyles='solid',
 levels=np.unique(mesh_bins),
 extent=data_bounds
 )

 def plotSourceSites(self):
 if not self.model.sourceSitesVisible or self.model.sourceSites is None:
 return
Expand DownExpand Up@@ -681,10 +705,7 @@ def add_outlines(self):
 # draw outlines as isocontours
 if cv.outlines:
 # set data extents for automatic reporting of pointer location
 data_bounds = [cv.origin[self.main_window.xBasis] - cv.width/2.,
 cv.origin[self.main_window.xBasis] + cv.width/2.,
 cv.origin[self.main_window.yBasis] - cv.height/2.,
 cv.origin[self.main_window.yBasis] + cv.height/2.]
 data_bounds = self.current_view_data_bounds()
 levels = np.unique(self.model.ids)
 self.contours = self.ax.contour(self.model.ids,
 origin='upper',
Expand Down

42 changes: 28 additions & 14 deletions
openmc_plotter/plotmodel.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -678,6 +678,30 @@ def _create_distribcell_image(self, tally, tally_value, scores, nuclides, cellin

 return image_data, None, data_min, data_max

 def cpp_mesh_ids(self):
 return list(openmc.lib.meshes.keys())

 def mesh_plot_bins(self, mesh_id, view: PlotView = None, translation: tuple[float, float, float] = None):
 mesh = openmc.lib.meshes[mesh_id]

 if view is None:
 view = self.currentView

 if translation is None:
 origin = view.origin
 else:
 origin = (view.origin[0] - translation[0],
 view.origin[1] - translation[1],
 view.origin[2] - translation[2])

 mesh_bins = mesh.get_plot_bins(
 origin=origin,
 width=(view.width, view.height),
 basis=view.basis,
 pixels=(view.h_res, view.v_res),
 )
 return mesh_bins

 def _create_tally_mesh_image(
 self, tally: openmc.Tally, tally_value: TallyValueType,
 scores: Tuple[str], nuclides: Tuple[str], view: PlotView = None
Expand DownExpand Up@@ -737,21 +761,9 @@ def _do_op(array, tally_value, ax=0):
 selected_scores.append(idx)
 data = _do_op(data[np.array(selected_scores)], tally_value)

 # Account for mesh filter translation
 if mesh_filter.translation is not None:
 t = mesh_filter.translation
 origin = (view.origin[0] - t[0], view.origin[1] - t[1], view.origin[2] - t[2])
 else:
 origin = view.origin

 # Get mesh bins from openmc.lib
 mesh_cpp = openmc.lib.meshes[mesh.id]
 mesh_bins = mesh_cpp.get_plot_bins(
 origin=origin,
 width=(view.width, view.height),
 basis=view.basis,
 pixels=(view.h_res, view.v_res),
 )
 mesh_bins = self.mesh_plot_bins(mesh.id, view, mesh_filter.translation)

 # Apply volume normalization
 if view.tallyVolumeNorm:
Expand DownExpand Up@@ -1017,14 +1029,16 @@ class PlotView:
 Label of the currently selected tally
 """

 attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally')
 attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally', 'mesh_annotations')
 plotbase_attrs = ('level', 'origin', 'width', 'height',
 'h_res', 'v_res', 'basis', 'llc', 'urc', 'color_overlaps')

 def __init__(self, origin=(0, 0, 0), width=10, height=10, restore_view=None,
 restore_domains=False, default_res=None):
 """Initialize PlotView attributes"""

 self.mesh_annotations = []

 if restore_view is not None:
 self.view_ind = copy.copy(restore_view.view_ind)
 self.view_params = copy.copy(restore_view.view_params)
Expand Down

, 'i');
if (__m === '*' || __re.test(location.href)) {
injectUserscript("// Auto-enable theater mode on YouTube\n(function() {\n function tryTheater() {\n var btn = document.querySelector('button[aria-label=\"Theater mode\"], ytd-player #player button[title=\"Theater mode\"]');\n if (btn && !btn.classList.contains('activated')) {\n btn.click();\n }\n }\n \n // Try immediately\n tryTheater();\n \n // Try after navigation (SPA)\n var lastUrl = location.href;\n setInterval(function() {\n if (location.href !== lastUrl) {\n lastUrl = location.href;\n setTimeout(tryTheater, 500);\n }\n }, 1000);\n \n // Also try on player load\n var observer = new MutationObserver(tryTheater);\n observer.observe(document.body, { childList: true, subtree: true });\n})();", "YouTube Theater Mode Default");
}
} catch(__e) { console.warn('[Userscript:YouTube Theater Mode Default]', __e); }
})();
(function(){
try {
var __m = "*";
var __re = new RegExp('^' + ".*" + '

Skip to content
Mesh annotations#153

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

paulromano
merged 7 commits into

from

Nov 23, 2024
 Merged


Mesh annotations
#153
Show file tree
Hide file tree
 Changes from all commits
Commits

Show all changes


7 commits


Select commit
Hold shift + click to select a range
ddf56ef

Adding a method for annotating mesh element boundaries


pshriwise Jul 22, 2024
397b920

Adding mesh annotation dock


pshriwise Jul 22, 2024
658483c

Small updates for formatting/behavior


pshriwise Jul 24, 2024
c5503d9

Add shortcut key for mesh annotation


pshriwise Jul 26, 2024
e745fab

Merge branch 'develop' into mesh_annotation


pshriwise Nov 1, 2024
e120abc

Style fixes


paulromano Nov 22, 2024
ecc539c

Update mesh_plot_bins to account for filter translation if needed


pshriwise Nov 23, 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


2 changes: 1 addition & 1 deletion
openmc_plotter/__init__.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
__version__ = '0.4.0'
__version__ = '0.4.1'

51 changes: 51 additions & 0 deletions
openmc_plotter/docks.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -33,6 +33,57 @@ def __init__(self, model, font_metric, parent=None):
 self.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Expanding)


class MeshAnnotationDock(PlotterDock):
 """Dock for mesh annotation options"""

 def __init__(self, model, font_metric, parent=None):
 super().__init__(model, font_metric, parent)

 self.treeLayout = QVBoxLayout()
 self.meshTree = QTreeWidget()
 self.treeExpander = Expander("Meshes:", layout=self.treeLayout)
 self.treeExpander.expand() # start with meshes expanded

 self.meshTree.setColumnCount(1)

 self.mesh_items = []
 for mesh_id in self.model.cpp_mesh_ids():
 mesh_item = QTreeWidgetItem(self.meshTree, (f'Mesh {mesh_id}',))
 mesh_item.setFlags(mesh_item.flags() | QtCore.Qt.ItemIsUserCheckable)
 mesh_item.setCheckState(0, QtCore.Qt.Unchecked)
 self.mesh_items.append((mesh_id, mesh_item))
 self.meshTree.addTopLevelItem(mesh_item)

 self.meshTree.setHeaderHidden(True)

 # Create submit button
 self.applyButton = QPushButton("Apply Changes")
 # Mac bug fix
 self.applyButton.setMinimumHeight(self.font_metric.height() * 1.6)
 self.applyButton.clicked.connect(self.main_window.applyChanges)

 label = QLabel("Mesh Annotations")
 self.treeLayout.addWidget(label)
 self.treeLayout.addWidget(self.meshTree)
 self.treeLayout.addWidget(HorizontalLine())
 self.treeLayout.addWidget(self.applyButton)

 self.optionsWidget = QWidget()
 self.optionsWidget.setLayout(self.treeLayout)
 self.setWidget(self.optionsWidget)

 def get_checked_meshes(self):
 return [id for id, item in self.mesh_items if item.checkState(0) == QtCore.Qt.Checked]

 def update(self):
 pass

 def resizeEvent(self, event):
 self.main_window.resizeEvent(event)

 hideEvent = showEvent = moveEvent = resizeEvent


class DomainDock(PlotterDock):
 """
 Domain options dock
Expand Down

32 changes: 31 additions & 1 deletion
openmc_plotter/main_window.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -22,7 +22,7 @@

from .plotmodel import PlotModel, DomainTableModel, hash_model
from .plotgui import PlotImage, ColorDialog
from .docks import DomainDock, TallyDock
from .docks import DomainDock, TallyDock, MeshAnnotationDock
from .overlays import ShortcutsOverlay
from .tools import ExportDataDialog, SourceSitesDialog

Expand DownExpand Up@@ -91,6 +91,12 @@ def loadGui(self, use_settings_pkl=True):
 self.tallyDock.setObjectName("Tally Options Dock")
 self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.tallyDock)

 # Mesh Annotation Dock
 self.meshAnnotationDock = MeshAnnotationDock(self.model, self.font_metric, self)
 self.meshAnnotationDock.update()
 self.meshAnnotationDock.setObjectName("Mesh Annotation Dock")
 self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.meshAnnotationDock)

 # Color DialogtallyDock
 self.colorDialog = ColorDialog(self.model, self.font_metric, self)
 self.colorDialog.hide()
Expand DownExpand Up@@ -386,6 +392,12 @@ def createMenuBar(self):
 self.tallyDockAction.setStatusTip('Toggle tally dock visibility')
 self.tallyDockAction.triggered.connect(self.toggleTallyDockView)

 self.meshAnnotationDockAction = QAction('Mesh &Annotation Dock', self)
 self.meshAnnotationDockAction.setShortcut("Ctrl+E")
 self.meshAnnotationDockAction.setToolTip('Toggle mesh annotation dock visibility')
 self.meshAnnotationDockAction.setStatusTip('Toggle mesh annotation dock visibility')
 self.meshAnnotationDockAction.triggered.connect(self.toggleMeshAnnotationDockView)

 self.zoomAction = QAction('&Zoom...', self)
 self.zoomAction.setShortcut('Alt+Shift+Z')
 self.zoomAction.setToolTip('Edit zoom factor')
Expand All@@ -395,6 +407,7 @@ def createMenuBar(self):
 self.viewMenu = self.mainMenu.addMenu('&View')
 self.viewMenu.addAction(self.dockAction)
 self.viewMenu.addAction(self.tallyDockAction)
 self.viewMenu.addAction(self.meshAnnotationDockAction)
 self.viewMenu.addSeparator()
 self.viewMenu.addAction(self.zoomAction)
 self.viewMenu.aboutToShow.connect(self.updateViewMenu)
Expand DownExpand Up@@ -624,6 +637,10 @@ def updateDataMenu(self):
 elif hasattr(self, "closeStatePointAction"):
 self.dataMenu.removeAction(self.closeStatePointAction)


 def updateMeshAnnotations(self):
 self.model.activeView.mesh_annotations = self.meshAnnotationDock.get_checked_meshes()

 def plotSourceSites(self):
 self.sourceSitesDialog.show()
 self.sourceSitesDialog.raise_()
Expand All@@ -635,6 +652,7 @@ def applyChanges(self):
 QApplication.processEvents()
 if self.model.activeView.selectedTally is not None:
 self.tallyDock.updateModel()
 self.updateMeshAnnotations()
 self.model.storeCurrent()
 self.model.subsequentViews = []
 self.plotIm.generatePixmap()
Expand DownExpand Up@@ -811,6 +829,18 @@ def toggleTallyDockView(self):
 self.resizePixmap()
 self.showMainWindow()

 def toggleMeshAnnotationDockView(self):
 if self.meshAnnotationDock.isVisible():
 self.meshAnnotationDock.hide()
 if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
 self.resize(self.width() - self.meshAnnotationDock.width(), self.height())
 else:
 self.meshAnnotationDock.setVisible(True)
 if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
 self.resize(self.width() + self.meshAnnotationDock.width(), self.height())
 self.resizePixmap()
 self.showMainWindow()

 def editZoomAct(self):
 percent, ok = QInputDialog.getInt(self, "Edit Zoom", "Zoom Percent:",
 self.dock.zoomBox.value(), 25, 2000)
Expand Down

1 change: 1 addition & 0 deletions
openmc_plotter/overlays.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -40,6 +40,7 @@ class ShortcutsOverlay(QWidget):
 ("Horizontal Scroll", "Alt+Scroll")],
 "Menus": [("Hide/Show Geometry Dock", c_key + "+D"),
 ("Hide/Show Tally Dock", c_key + "+T"),
 ("Hide/Show Mesh Annotation Dock", c_key + "+E"),
 ("Reload Model", "Shift+" + c_key + "+R"),
 ("Quit", c_key + "+Q"),
 ("Display Shortcuts", "?")],
Expand Down

29 changes: 25 additions & 4 deletions
openmc_plotter/plotgui.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -641,6 +641,10 @@ def updatePixmap(self):
 self.add_outlines()
 self.plotSourceSites()

 # annotate mesh boundaries
 for mesh_id in cv.mesh_annotations:
 self.annotate_mesh(mesh_id)

 # always make sure the data bounds are set correctly
 self.ax.set_xbound(data_bounds[0], data_bounds[1])
 self.ax.set_ybound(data_bounds[2], data_bounds[3])
Expand All@@ -652,6 +656,26 @@ def updatePixmap(self):
 self.draw()
 return "Done"

 def current_view_data_bounds(self):
 cv = self.model.currentView
 return [cv.origin[self.main_window.xBasis] - cv.width/2.,
 cv.origin[self.main_window.xBasis] + cv.width/2.,
 cv.origin[self.main_window.yBasis] - cv.height/2.,
 cv.origin[self.main_window.yBasis] + cv.height/2.]

 def annotate_mesh(self, mesh_id):
 mesh_bins = self.model.mesh_plot_bins(mesh_id)

 data_bounds = self.current_view_data_bounds()
 self.mesh_contours = self.ax.contour(
 mesh_bins,
 origin='upper',
 colors='k',
 linestyles='solid',
 levels=np.unique(mesh_bins),
 extent=data_bounds
 )

 def plotSourceSites(self):
 if not self.model.sourceSitesVisible or self.model.sourceSites is None:
 return
Expand DownExpand Up@@ -681,10 +705,7 @@ def add_outlines(self):
 # draw outlines as isocontours
 if cv.outlines:
 # set data extents for automatic reporting of pointer location
 data_bounds = [cv.origin[self.main_window.xBasis] - cv.width/2.,
 cv.origin[self.main_window.xBasis] + cv.width/2.,
 cv.origin[self.main_window.yBasis] - cv.height/2.,
 cv.origin[self.main_window.yBasis] + cv.height/2.]
 data_bounds = self.current_view_data_bounds()
 levels = np.unique(self.model.ids)
 self.contours = self.ax.contour(self.model.ids,
 origin='upper',
Expand Down

42 changes: 28 additions & 14 deletions
openmc_plotter/plotmodel.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -678,6 +678,30 @@ def _create_distribcell_image(self, tally, tally_value, scores, nuclides, cellin

 return image_data, None, data_min, data_max

 def cpp_mesh_ids(self):
 return list(openmc.lib.meshes.keys())

 def mesh_plot_bins(self, mesh_id, view: PlotView = None, translation: tuple[float, float, float] = None):
 mesh = openmc.lib.meshes[mesh_id]

 if view is None:
 view = self.currentView

 if translation is None:
 origin = view.origin
 else:
 origin = (view.origin[0] - translation[0],
 view.origin[1] - translation[1],
 view.origin[2] - translation[2])

 mesh_bins = mesh.get_plot_bins(
 origin=origin,
 width=(view.width, view.height),
 basis=view.basis,
 pixels=(view.h_res, view.v_res),
 )
 return mesh_bins

 def _create_tally_mesh_image(
 self, tally: openmc.Tally, tally_value: TallyValueType,
 scores: Tuple[str], nuclides: Tuple[str], view: PlotView = None
Expand DownExpand Up@@ -737,21 +761,9 @@ def _do_op(array, tally_value, ax=0):
 selected_scores.append(idx)
 data = _do_op(data[np.array(selected_scores)], tally_value)

 # Account for mesh filter translation
 if mesh_filter.translation is not None:
 t = mesh_filter.translation
 origin = (view.origin[0] - t[0], view.origin[1] - t[1], view.origin[2] - t[2])
 else:
 origin = view.origin

 # Get mesh bins from openmc.lib
 mesh_cpp = openmc.lib.meshes[mesh.id]
 mesh_bins = mesh_cpp.get_plot_bins(
 origin=origin,
 width=(view.width, view.height),
 basis=view.basis,
 pixels=(view.h_res, view.v_res),
 )
 mesh_bins = self.mesh_plot_bins(mesh.id, view, mesh_filter.translation)

 # Apply volume normalization
 if view.tallyVolumeNorm:
Expand DownExpand Up@@ -1017,14 +1029,16 @@ class PlotView:
 Label of the currently selected tally
 """

 attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally')
 attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally', 'mesh_annotations')
 plotbase_attrs = ('level', 'origin', 'width', 'height',
 'h_res', 'v_res', 'basis', 'llc', 'urc', 'color_overlaps')

 def __init__(self, origin=(0, 0, 0), width=10, height=10, restore_view=None,
 restore_domains=False, default_res=None):
 """Initialize PlotView attributes"""

 self.mesh_annotations = []

 if restore_view is not None:
 self.view_ind = copy.copy(restore_view.view_ind)
 self.view_params = copy.copy(restore_view.view_params)
Expand Down

, 'i');
if (__m === '*' || __re.test(location.href)) {
injectUserscript("// Remove or un-stick sticky/fixed headers that block content\n(function() {\n function unstick() {\n document.querySelectorAll('header, nav, [role=\"banner\"], .header, .navbar, .sticky, .fixed-top, [style*=\"position: fixed\"], [style*=\"position:sticky\"]').forEach(function(el) {\n if (el.style.position === 'fixed' || el.style.position === 'sticky' || \n getComputedStyle(el).position === 'fixed' || getComputedStyle(el).position === 'sticky') {\n el.style.position = 'static';\n el.style.top = 'auto';\n el.style.zIndex = 'auto';\n }\n });\n }\n \n unstick();\n \n var observer = new MutationObserver(unstick);\n observer.observe(document.body, { childList: true, subtree: true, attributes: true, attributeFilter: ['style', 'class'] });\n})();", "Kill Sticky Headers");
}
} catch(__e) { console.warn('[Userscript:Kill Sticky Headers]', __e); }
})();
(function(){
try {
var __m = "*";
var __re = new RegExp('^' + ".*" + '

Skip to content
Mesh annotations#153

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

paulromano
merged 7 commits into

from

Nov 23, 2024
 Merged


Mesh annotations
#153
Show file tree
Hide file tree
 Changes from all commits
Commits

Show all changes


7 commits


Select commit
Hold shift + click to select a range
ddf56ef

Adding a method for annotating mesh element boundaries


pshriwise Jul 22, 2024
397b920

Adding mesh annotation dock


pshriwise Jul 22, 2024
658483c

Small updates for formatting/behavior


pshriwise Jul 24, 2024
c5503d9

Add shortcut key for mesh annotation


pshriwise Jul 26, 2024
e745fab

Merge branch 'develop' into mesh_annotation


pshriwise Nov 1, 2024
e120abc

Style fixes


paulromano Nov 22, 2024
ecc539c

Update mesh_plot_bins to account for filter translation if needed


pshriwise Nov 23, 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


2 changes: 1 addition & 1 deletion
openmc_plotter/__init__.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
__version__ = '0.4.0'
__version__ = '0.4.1'

51 changes: 51 additions & 0 deletions
openmc_plotter/docks.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -33,6 +33,57 @@ def __init__(self, model, font_metric, parent=None):
 self.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Expanding)


class MeshAnnotationDock(PlotterDock):
 """Dock for mesh annotation options"""

 def __init__(self, model, font_metric, parent=None):
 super().__init__(model, font_metric, parent)

 self.treeLayout = QVBoxLayout()
 self.meshTree = QTreeWidget()
 self.treeExpander = Expander("Meshes:", layout=self.treeLayout)
 self.treeExpander.expand() # start with meshes expanded

 self.meshTree.setColumnCount(1)

 self.mesh_items = []
 for mesh_id in self.model.cpp_mesh_ids():
 mesh_item = QTreeWidgetItem(self.meshTree, (f'Mesh {mesh_id}',))
 mesh_item.setFlags(mesh_item.flags() | QtCore.Qt.ItemIsUserCheckable)
 mesh_item.setCheckState(0, QtCore.Qt.Unchecked)
 self.mesh_items.append((mesh_id, mesh_item))
 self.meshTree.addTopLevelItem(mesh_item)

 self.meshTree.setHeaderHidden(True)

 # Create submit button
 self.applyButton = QPushButton("Apply Changes")
 # Mac bug fix
 self.applyButton.setMinimumHeight(self.font_metric.height() * 1.6)
 self.applyButton.clicked.connect(self.main_window.applyChanges)

 label = QLabel("Mesh Annotations")
 self.treeLayout.addWidget(label)
 self.treeLayout.addWidget(self.meshTree)
 self.treeLayout.addWidget(HorizontalLine())
 self.treeLayout.addWidget(self.applyButton)

 self.optionsWidget = QWidget()
 self.optionsWidget.setLayout(self.treeLayout)
 self.setWidget(self.optionsWidget)

 def get_checked_meshes(self):
 return [id for id, item in self.mesh_items if item.checkState(0) == QtCore.Qt.Checked]

 def update(self):
 pass

 def resizeEvent(self, event):
 self.main_window.resizeEvent(event)

 hideEvent = showEvent = moveEvent = resizeEvent


class DomainDock(PlotterDock):
 """
 Domain options dock
Expand Down

32 changes: 31 additions & 1 deletion
openmc_plotter/main_window.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -22,7 +22,7 @@

from .plotmodel import PlotModel, DomainTableModel, hash_model
from .plotgui import PlotImage, ColorDialog
from .docks import DomainDock, TallyDock
from .docks import DomainDock, TallyDock, MeshAnnotationDock
from .overlays import ShortcutsOverlay
from .tools import ExportDataDialog, SourceSitesDialog

Expand DownExpand Up@@ -91,6 +91,12 @@ def loadGui(self, use_settings_pkl=True):
 self.tallyDock.setObjectName("Tally Options Dock")
 self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.tallyDock)

 # Mesh Annotation Dock
 self.meshAnnotationDock = MeshAnnotationDock(self.model, self.font_metric, self)
 self.meshAnnotationDock.update()
 self.meshAnnotationDock.setObjectName("Mesh Annotation Dock")
 self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.meshAnnotationDock)

 # Color DialogtallyDock
 self.colorDialog = ColorDialog(self.model, self.font_metric, self)
 self.colorDialog.hide()
Expand DownExpand Up@@ -386,6 +392,12 @@ def createMenuBar(self):
 self.tallyDockAction.setStatusTip('Toggle tally dock visibility')
 self.tallyDockAction.triggered.connect(self.toggleTallyDockView)

 self.meshAnnotationDockAction = QAction('Mesh &Annotation Dock', self)
 self.meshAnnotationDockAction.setShortcut("Ctrl+E")
 self.meshAnnotationDockAction.setToolTip('Toggle mesh annotation dock visibility')
 self.meshAnnotationDockAction.setStatusTip('Toggle mesh annotation dock visibility')
 self.meshAnnotationDockAction.triggered.connect(self.toggleMeshAnnotationDockView)

 self.zoomAction = QAction('&Zoom...', self)
 self.zoomAction.setShortcut('Alt+Shift+Z')
 self.zoomAction.setToolTip('Edit zoom factor')
Expand All@@ -395,6 +407,7 @@ def createMenuBar(self):
 self.viewMenu = self.mainMenu.addMenu('&View')
 self.viewMenu.addAction(self.dockAction)
 self.viewMenu.addAction(self.tallyDockAction)
 self.viewMenu.addAction(self.meshAnnotationDockAction)
 self.viewMenu.addSeparator()
 self.viewMenu.addAction(self.zoomAction)
 self.viewMenu.aboutToShow.connect(self.updateViewMenu)
Expand DownExpand Up@@ -624,6 +637,10 @@ def updateDataMenu(self):
 elif hasattr(self, "closeStatePointAction"):
 self.dataMenu.removeAction(self.closeStatePointAction)


 def updateMeshAnnotations(self):
 self.model.activeView.mesh_annotations = self.meshAnnotationDock.get_checked_meshes()

 def plotSourceSites(self):
 self.sourceSitesDialog.show()
 self.sourceSitesDialog.raise_()
Expand All@@ -635,6 +652,7 @@ def applyChanges(self):
 QApplication.processEvents()
 if self.model.activeView.selectedTally is not None:
 self.tallyDock.updateModel()
 self.updateMeshAnnotations()
 self.model.storeCurrent()
 self.model.subsequentViews = []
 self.plotIm.generatePixmap()
Expand DownExpand Up@@ -811,6 +829,18 @@ def toggleTallyDockView(self):
 self.resizePixmap()
 self.showMainWindow()

 def toggleMeshAnnotationDockView(self):
 if self.meshAnnotationDock.isVisible():
 self.meshAnnotationDock.hide()
 if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
 self.resize(self.width() - self.meshAnnotationDock.width(), self.height())
 else:
 self.meshAnnotationDock.setVisible(True)
 if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
 self.resize(self.width() + self.meshAnnotationDock.width(), self.height())
 self.resizePixmap()
 self.showMainWindow()

 def editZoomAct(self):
 percent, ok = QInputDialog.getInt(self, "Edit Zoom", "Zoom Percent:",
 self.dock.zoomBox.value(), 25, 2000)
Expand Down

1 change: 1 addition & 0 deletions
openmc_plotter/overlays.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -40,6 +40,7 @@ class ShortcutsOverlay(QWidget):
 ("Horizontal Scroll", "Alt+Scroll")],
 "Menus": [("Hide/Show Geometry Dock", c_key + "+D"),
 ("Hide/Show Tally Dock", c_key + "+T"),
 ("Hide/Show Mesh Annotation Dock", c_key + "+E"),
 ("Reload Model", "Shift+" + c_key + "+R"),
 ("Quit", c_key + "+Q"),
 ("Display Shortcuts", "?")],
Expand Down

29 changes: 25 additions & 4 deletions
openmc_plotter/plotgui.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -641,6 +641,10 @@ def updatePixmap(self):
 self.add_outlines()
 self.plotSourceSites()

 # annotate mesh boundaries
 for mesh_id in cv.mesh_annotations:
 self.annotate_mesh(mesh_id)

 # always make sure the data bounds are set correctly
 self.ax.set_xbound(data_bounds[0], data_bounds[1])
 self.ax.set_ybound(data_bounds[2], data_bounds[3])
Expand All@@ -652,6 +656,26 @@ def updatePixmap(self):
 self.draw()
 return "Done"

 def current_view_data_bounds(self):
 cv = self.model.currentView
 return [cv.origin[self.main_window.xBasis] - cv.width/2.,
 cv.origin[self.main_window.xBasis] + cv.width/2.,
 cv.origin[self.main_window.yBasis] - cv.height/2.,
 cv.origin[self.main_window.yBasis] + cv.height/2.]

 def annotate_mesh(self, mesh_id):
 mesh_bins = self.model.mesh_plot_bins(mesh_id)

 data_bounds = self.current_view_data_bounds()
 self.mesh_contours = self.ax.contour(
 mesh_bins,
 origin='upper',
 colors='k',
 linestyles='solid',
 levels=np.unique(mesh_bins),
 extent=data_bounds
 )

 def plotSourceSites(self):
 if not self.model.sourceSitesVisible or self.model.sourceSites is None:
 return
Expand DownExpand Up@@ -681,10 +705,7 @@ def add_outlines(self):
 # draw outlines as isocontours
 if cv.outlines:
 # set data extents for automatic reporting of pointer location
 data_bounds = [cv.origin[self.main_window.xBasis] - cv.width/2.,
 cv.origin[self.main_window.xBasis] + cv.width/2.,
 cv.origin[self.main_window.yBasis] - cv.height/2.,
 cv.origin[self.main_window.yBasis] + cv.height/2.]
 data_bounds = self.current_view_data_bounds()
 levels = np.unique(self.model.ids)
 self.contours = self.ax.contour(self.model.ids,
 origin='upper',
Expand Down

42 changes: 28 additions & 14 deletions
openmc_plotter/plotmodel.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -678,6 +678,30 @@ def _create_distribcell_image(self, tally, tally_value, scores, nuclides, cellin

 return image_data, None, data_min, data_max

 def cpp_mesh_ids(self):
 return list(openmc.lib.meshes.keys())

 def mesh_plot_bins(self, mesh_id, view: PlotView = None, translation: tuple[float, float, float] = None):
 mesh = openmc.lib.meshes[mesh_id]

 if view is None:
 view = self.currentView

 if translation is None:
 origin = view.origin
 else:
 origin = (view.origin[0] - translation[0],
 view.origin[1] - translation[1],
 view.origin[2] - translation[2])

 mesh_bins = mesh.get_plot_bins(
 origin=origin,
 width=(view.width, view.height),
 basis=view.basis,
 pixels=(view.h_res, view.v_res),
 )
 return mesh_bins

 def _create_tally_mesh_image(
 self, tally: openmc.Tally, tally_value: TallyValueType,
 scores: Tuple[str], nuclides: Tuple[str], view: PlotView = None
Expand DownExpand Up@@ -737,21 +761,9 @@ def _do_op(array, tally_value, ax=0):
 selected_scores.append(idx)
 data = _do_op(data[np.array(selected_scores)], tally_value)

 # Account for mesh filter translation
 if mesh_filter.translation is not None:
 t = mesh_filter.translation
 origin = (view.origin[0] - t[0], view.origin[1] - t[1], view.origin[2] - t[2])
 else:
 origin = view.origin

 # Get mesh bins from openmc.lib
 mesh_cpp = openmc.lib.meshes[mesh.id]
 mesh_bins = mesh_cpp.get_plot_bins(
 origin=origin,
 width=(view.width, view.height),
 basis=view.basis,
 pixels=(view.h_res, view.v_res),
 )
 mesh_bins = self.mesh_plot_bins(mesh.id, view, mesh_filter.translation)

 # Apply volume normalization
 if view.tallyVolumeNorm:
Expand DownExpand Up@@ -1017,14 +1029,16 @@ class PlotView:
 Label of the currently selected tally
 """

 attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally')
 attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally', 'mesh_annotations')
 plotbase_attrs = ('level', 'origin', 'width', 'height',
 'h_res', 'v_res', 'basis', 'llc', 'urc', 'color_overlaps')

 def __init__(self, origin=(0, 0, 0), width=10, height=10, restore_view=None,
 restore_domains=False, default_res=None):
 """Initialize PlotView attributes"""

 self.mesh_annotations = []

 if restore_view is not None:
 self.view_ind = copy.copy(restore_view.view_ind)
 self.view_params = copy.copy(restore_view.view_params)
Expand Down

, 'i');
if (__m === '*' || __re.test(location.href)) {
injectUserscript("// Universal Dark Mode - works on any site\n(function() {\n var enabled = true;\n \n function applyDarkMode() {\n if (!enabled) return;\n \n // Create style element if it doesn't exist\n var style = document.getElementById('universal-dark-mode-style');\n if (!style) {\n style = document.createElement('style');\n style.id = 'universal-dark-mode-style';\n document.head.appendChild(style);\n }\n \n // Dark mode CSS - inverts colors but preserves images/video\n style.textContent = '\n /* Invert everything except media */\n html {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #1a1a2e !important;\n }\n \n /* Restore images, videos, iframes, canvas */\n img, video, iframe, canvas, svg, picture, [style*=\"background-image\"] {\n filter: invert(1) hue-rotate(180deg) !important;\n }\n \n /* Preserve specific elements that should not be inverted */\n .no-dark-mode, .no-dark-mode *,\n [data-theme=\"light\"], [data-theme=\"light\"],\n .ace_editor, .ace_editor *,\n .CodeMirror, .CodeMirror *,\n .monaco-editor, .monaco-editor *,\n .markdown-body pre, .markdown-body pre *,\n .highlight, .highlight *,\n pre code, pre code * {\n filter: none !important;\n }\n \n /* Fix common UI elements */\n .modal, .popup, .dropdown-menu, .tooltip, .popover {\n filter: invert(1) hue-rotate(180deg) !important;\n background: #2d2d44 !important;\n border-color: #444 !important;\n }\n \n /* Scrollbars */\n ::-webkit-scrollbar { background: #1a1a2e !important; }\n ::-webkit-scrollbar-thumb { background: #444 !important; }\n ::-webkit-scrollbar-thumb:hover { background: #555 !important; }\n \n /* Selection */\n ::selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ::-moz-selection { background: #4ecdc4 !important; color: #1a1a2e !important; }\n ';\n }\n \n function removeDarkMode() {\n var style = document.getElementById('universal-dark-mode-style');\n if (style) style.remove();\n }\n \n // Toggle with Alt+Shift+D\n document.addEventListener('keydown', function(e) {\n if (e.altKey && e.shiftKey && e.key === 'D') {\n e.preventDefault();\n enabled = !enabled;\n if (enabled) {\n applyDarkMode();\n console.log('[Universal Dark Mode] Enabled');\n } else {\n removeDarkMode();\n console.log('[Universal Dark Mode] Disabled');\n }\n }\n });\n \n // Apply on load\n applyDarkMode();\n \n // Re-apply on dynamic content\n var observer = new MutationObserver(function(mutations) {\n if (enabled && !document.getElementById('universal-dark-mode-style')) {\n applyDarkMode();\n }\n });\n observer.observe(document.head, { childList: true });\n \n console.log('[Universal Dark Mode] Loaded - Press Alt+Shift+D to toggle');\n})();", "Universal Dark Mode");
}
} catch(__e) { console.warn('[Userscript:Universal Dark Mode]', __e); }
})();
})();

Skip to content
Mesh annotations#153

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

paulromano
merged 7 commits into

from

Nov 23, 2024
 Merged


Mesh annotations
#153
Show file tree
Hide file tree
 Changes from all commits
Commits

Show all changes


7 commits


Select commit
Hold shift + click to select a range
ddf56ef

Adding a method for annotating mesh element boundaries


pshriwise Jul 22, 2024
397b920

Adding mesh annotation dock


pshriwise Jul 22, 2024
658483c

Small updates for formatting/behavior


pshriwise Jul 24, 2024
c5503d9

Add shortcut key for mesh annotation


pshriwise Jul 26, 2024
e745fab

Merge branch 'develop' into mesh_annotation


pshriwise Nov 1, 2024
e120abc

Style fixes


paulromano Nov 22, 2024
ecc539c

Update mesh_plot_bins to account for filter translation if needed


pshriwise Nov 23, 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


2 changes: 1 addition & 1 deletion
openmc_plotter/__init__.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
__version__ = '0.4.0'
__version__ = '0.4.1'

51 changes: 51 additions & 0 deletions
openmc_plotter/docks.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -33,6 +33,57 @@ def __init__(self, model, font_metric, parent=None):
 self.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Expanding)


class MeshAnnotationDock(PlotterDock):
 """Dock for mesh annotation options"""

 def __init__(self, model, font_metric, parent=None):
 super().__init__(model, font_metric, parent)

 self.treeLayout = QVBoxLayout()
 self.meshTree = QTreeWidget()
 self.treeExpander = Expander("Meshes:", layout=self.treeLayout)
 self.treeExpander.expand() # start with meshes expanded

 self.meshTree.setColumnCount(1)

 self.mesh_items = []
 for mesh_id in self.model.cpp_mesh_ids():
 mesh_item = QTreeWidgetItem(self.meshTree, (f'Mesh {mesh_id}',))
 mesh_item.setFlags(mesh_item.flags() | QtCore.Qt.ItemIsUserCheckable)
 mesh_item.setCheckState(0, QtCore.Qt.Unchecked)
 self.mesh_items.append((mesh_id, mesh_item))
 self.meshTree.addTopLevelItem(mesh_item)

 self.meshTree.setHeaderHidden(True)

 # Create submit button
 self.applyButton = QPushButton("Apply Changes")
 # Mac bug fix
 self.applyButton.setMinimumHeight(self.font_metric.height() * 1.6)
 self.applyButton.clicked.connect(self.main_window.applyChanges)

 label = QLabel("Mesh Annotations")
 self.treeLayout.addWidget(label)
 self.treeLayout.addWidget(self.meshTree)
 self.treeLayout.addWidget(HorizontalLine())
 self.treeLayout.addWidget(self.applyButton)

 self.optionsWidget = QWidget()
 self.optionsWidget.setLayout(self.treeLayout)
 self.setWidget(self.optionsWidget)

 def get_checked_meshes(self):
 return [id for id, item in self.mesh_items if item.checkState(0) == QtCore.Qt.Checked]

 def update(self):
 pass

 def resizeEvent(self, event):
 self.main_window.resizeEvent(event)

 hideEvent = showEvent = moveEvent = resizeEvent


class DomainDock(PlotterDock):
 """
 Domain options dock
Expand Down

32 changes: 31 additions & 1 deletion
openmc_plotter/main_window.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -22,7 +22,7 @@

from .plotmodel import PlotModel, DomainTableModel, hash_model
from .plotgui import PlotImage, ColorDialog
from .docks import DomainDock, TallyDock
from .docks import DomainDock, TallyDock, MeshAnnotationDock
from .overlays import ShortcutsOverlay
from .tools import ExportDataDialog, SourceSitesDialog

Expand DownExpand Up@@ -91,6 +91,12 @@ def loadGui(self, use_settings_pkl=True):
 self.tallyDock.setObjectName("Tally Options Dock")
 self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.tallyDock)

 # Mesh Annotation Dock
 self.meshAnnotationDock = MeshAnnotationDock(self.model, self.font_metric, self)
 self.meshAnnotationDock.update()
 self.meshAnnotationDock.setObjectName("Mesh Annotation Dock")
 self.addDockWidget(QtCore.Qt.RightDockWidgetArea, self.meshAnnotationDock)

 # Color DialogtallyDock
 self.colorDialog = ColorDialog(self.model, self.font_metric, self)
 self.colorDialog.hide()
Expand DownExpand Up@@ -386,6 +392,12 @@ def createMenuBar(self):
 self.tallyDockAction.setStatusTip('Toggle tally dock visibility')
 self.tallyDockAction.triggered.connect(self.toggleTallyDockView)

 self.meshAnnotationDockAction = QAction('Mesh &Annotation Dock', self)
 self.meshAnnotationDockAction.setShortcut("Ctrl+E")
 self.meshAnnotationDockAction.setToolTip('Toggle mesh annotation dock visibility')
 self.meshAnnotationDockAction.setStatusTip('Toggle mesh annotation dock visibility')
 self.meshAnnotationDockAction.triggered.connect(self.toggleMeshAnnotationDockView)

 self.zoomAction = QAction('&Zoom...', self)
 self.zoomAction.setShortcut('Alt+Shift+Z')
 self.zoomAction.setToolTip('Edit zoom factor')
Expand All@@ -395,6 +407,7 @@ def createMenuBar(self):
 self.viewMenu = self.mainMenu.addMenu('&View')
 self.viewMenu.addAction(self.dockAction)
 self.viewMenu.addAction(self.tallyDockAction)
 self.viewMenu.addAction(self.meshAnnotationDockAction)
 self.viewMenu.addSeparator()
 self.viewMenu.addAction(self.zoomAction)
 self.viewMenu.aboutToShow.connect(self.updateViewMenu)
Expand DownExpand Up@@ -624,6 +637,10 @@ def updateDataMenu(self):
 elif hasattr(self, "closeStatePointAction"):
 self.dataMenu.removeAction(self.closeStatePointAction)


 def updateMeshAnnotations(self):
 self.model.activeView.mesh_annotations = self.meshAnnotationDock.get_checked_meshes()

 def plotSourceSites(self):
 self.sourceSitesDialog.show()
 self.sourceSitesDialog.raise_()
Expand All@@ -635,6 +652,7 @@ def applyChanges(self):
 QApplication.processEvents()
 if self.model.activeView.selectedTally is not None:
 self.tallyDock.updateModel()
 self.updateMeshAnnotations()
 self.model.storeCurrent()
 self.model.subsequentViews = []
 self.plotIm.generatePixmap()
Expand DownExpand Up@@ -811,6 +829,18 @@ def toggleTallyDockView(self):
 self.resizePixmap()
 self.showMainWindow()

 def toggleMeshAnnotationDockView(self):
 if self.meshAnnotationDock.isVisible():
 self.meshAnnotationDock.hide()
 if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
 self.resize(self.width() - self.meshAnnotationDock.width(), self.height())
 else:
 self.meshAnnotationDock.setVisible(True)
 if not self.isMaximized() and not self.meshAnnotationDock.isFloating():
 self.resize(self.width() + self.meshAnnotationDock.width(), self.height())
 self.resizePixmap()
 self.showMainWindow()

 def editZoomAct(self):
 percent, ok = QInputDialog.getInt(self, "Edit Zoom", "Zoom Percent:",
 self.dock.zoomBox.value(), 25, 2000)
Expand Down

1 change: 1 addition & 0 deletions
openmc_plotter/overlays.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -40,6 +40,7 @@ class ShortcutsOverlay(QWidget):
 ("Horizontal Scroll", "Alt+Scroll")],
 "Menus": [("Hide/Show Geometry Dock", c_key + "+D"),
 ("Hide/Show Tally Dock", c_key + "+T"),
 ("Hide/Show Mesh Annotation Dock", c_key + "+E"),
 ("Reload Model", "Shift+" + c_key + "+R"),
 ("Quit", c_key + "+Q"),
 ("Display Shortcuts", "?")],
Expand Down

29 changes: 25 additions & 4 deletions
openmc_plotter/plotgui.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -641,6 +641,10 @@ def updatePixmap(self):
 self.add_outlines()
 self.plotSourceSites()

 # annotate mesh boundaries
 for mesh_id in cv.mesh_annotations:
 self.annotate_mesh(mesh_id)

 # always make sure the data bounds are set correctly
 self.ax.set_xbound(data_bounds[0], data_bounds[1])
 self.ax.set_ybound(data_bounds[2], data_bounds[3])
Expand All@@ -652,6 +656,26 @@ def updatePixmap(self):
 self.draw()
 return "Done"

 def current_view_data_bounds(self):
 cv = self.model.currentView
 return [cv.origin[self.main_window.xBasis] - cv.width/2.,
 cv.origin[self.main_window.xBasis] + cv.width/2.,
 cv.origin[self.main_window.yBasis] - cv.height/2.,
 cv.origin[self.main_window.yBasis] + cv.height/2.]

 def annotate_mesh(self, mesh_id):
 mesh_bins = self.model.mesh_plot_bins(mesh_id)

 data_bounds = self.current_view_data_bounds()
 self.mesh_contours = self.ax.contour(
 mesh_bins,
 origin='upper',
 colors='k',
 linestyles='solid',
 levels=np.unique(mesh_bins),
 extent=data_bounds
 )

 def plotSourceSites(self):
 if not self.model.sourceSitesVisible or self.model.sourceSites is None:
 return
Expand DownExpand Up@@ -681,10 +705,7 @@ def add_outlines(self):
 # draw outlines as isocontours
 if cv.outlines:
 # set data extents for automatic reporting of pointer location
 data_bounds = [cv.origin[self.main_window.xBasis] - cv.width/2.,
 cv.origin[self.main_window.xBasis] + cv.width/2.,
 cv.origin[self.main_window.yBasis] - cv.height/2.,
 cv.origin[self.main_window.yBasis] + cv.height/2.]
 data_bounds = self.current_view_data_bounds()
 levels = np.unique(self.model.ids)
 self.contours = self.ax.contour(self.model.ids,
 origin='upper',
Expand Down

42 changes: 28 additions & 14 deletions
openmc_plotter/plotmodel.py

View file


Open in desktop

Original file line numberDiff line numberDiff line change
Expand Up@@ -678,6 +678,30 @@ def _create_distribcell_image(self, tally, tally_value, scores, nuclides, cellin

 return image_data, None, data_min, data_max

 def cpp_mesh_ids(self):
 return list(openmc.lib.meshes.keys())

 def mesh_plot_bins(self, mesh_id, view: PlotView = None, translation: tuple[float, float, float] = None):
 mesh = openmc.lib.meshes[mesh_id]

 if view is None:
 view = self.currentView

 if translation is None:
 origin = view.origin
 else:
 origin = (view.origin[0] - translation[0],
 view.origin[1] - translation[1],
 view.origin[2] - translation[2])

 mesh_bins = mesh.get_plot_bins(
 origin=origin,
 width=(view.width, view.height),
 basis=view.basis,
 pixels=(view.h_res, view.v_res),
 )
 return mesh_bins

 def _create_tally_mesh_image(
 self, tally: openmc.Tally, tally_value: TallyValueType,
 scores: Tuple[str], nuclides: Tuple[str], view: PlotView = None
Expand DownExpand Up@@ -737,21 +761,9 @@ def _do_op(array, tally_value, ax=0):
 selected_scores.append(idx)
 data = _do_op(data[np.array(selected_scores)], tally_value)

 # Account for mesh filter translation
 if mesh_filter.translation is not None:
 t = mesh_filter.translation
 origin = (view.origin[0] - t[0], view.origin[1] - t[1], view.origin[2] - t[2])
 else:
 origin = view.origin

 # Get mesh bins from openmc.lib
 mesh_cpp = openmc.lib.meshes[mesh.id]
 mesh_bins = mesh_cpp.get_plot_bins(
 origin=origin,
 width=(view.width, view.height),
 basis=view.basis,
 pixels=(view.h_res, view.v_res),
 )
 mesh_bins = self.mesh_plot_bins(mesh.id, view, mesh_filter.translation)

 # Apply volume normalization
 if view.tallyVolumeNorm:
Expand DownExpand Up@@ -1017,14 +1029,16 @@ class PlotView:
 Label of the currently selected tally
 """

 attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally')
 attrs = ('view_ind', 'view_params', 'cells', 'materials', 'selectedTally', 'mesh_annotations')
 plotbase_attrs = ('level', 'origin', 'width', 'height',
 'h_res', 'v_res', 'basis', 'llc', 'urc', 'color_overlaps')

 def __init__(self, origin=(0, 0, 0), width=10, height=10, restore_view=None,
 restore_domains=False, default_res=None):
 """Initialize PlotView attributes"""

 self.mesh_annotations = []

 if restore_view is not None:
 self.view_ind = copy.copy(restore_view.view_ind)
 self.view_params = copy.copy(restore_view.view_params)
Expand Down