Adds a batch mode to the plotter

Author: pshriwise

openmc_plotter/__main__.py

@@ -24,6 +24,8 @@ def main():
     ap.add_argument('model_path', nargs='?', default=os.curdir,
                     help='Location of model XML file or a directory containing '
                     'XML files (default is current dir)')
+    ap.add_argument('-b', '--batch-mode', nargs='+', default=False,
+                    help='View files used to generate plots in batch mode')
 
     args = ap.parse_args()
 
@@ -68,9 +70,18 @@ def run_app(user_args):
     mainWindow = MainWindow(font_metric, screen_size, user_args.model_path)
     # connect splashscreen to main window, close when main window opens
     mainWindow.loadGui(use_settings_pkl=user_args.ignore_settings)
-    mainWindow.show()
+
     splash.close()
 
+    if user_args.batch_mode:
+        for view_file in user_args.batch_mode:
+            print(view_file)
+            mainWindow.saveBatchImage(view_file)
+        mainWindow.close()
+        sys.exit()
+    else:
+        mainWindow.show()
+
     # connect interrupt signal to close call
     signal.signal(signal.SIGINT, lambda *args: mainWindow.close())
     # create timer that interrupts the Qt event loop

openmc_plotter/main_window.py

@@ -449,6 +449,13 @@ def updateWindowMenu(self):
         self.colorDialogAction.setChecked(self.colorDialog.isActiveWindow())
         self.mainWindowAction.setChecked(self.isActiveWindow())
 
+    def saveBatchImage(self, view_file):
+        # store the
+        cv = self.model.currentView
+        # load the view from file
+        self.loadViewFile(view_file)
+        self.plotIm.saveImage(view_file.replace('.pltvw', ''))
+
     # Menu and shared methods
     def loadModel(self, reload=False, use_settings_pkl=True):
         if reload:
@@ -478,10 +485,8 @@ def saveImage(self):
                                                     "untitled",
                                                     "Images (*.png)")
         if filename:
-            if "." not in filename:
-                filename += ".png"
-            self.plotIm.figure.savefig(filename, transparent=True)
-            self.statusBar().showMessage('Plot Image Saved', 5000)
+            self.plotIm.saveImage(filename)
+        self.statusBar().showMessage('Plot Image Saved', 5000)
 
     def saveView(self):
         filename, ext = QFileDialog.getSaveFileName(self,
@@ -497,26 +502,29 @@ def saveView(self):
             with open(filename, 'wb') as file:
                 pickle.dump(saved, file)
 
+    def loadViewFile(self, filename):
+        try:
+            with open(filename, 'rb') as file:
+                saved = pickle.load(file)
+        except Exception:
+            message = 'Error loading plot settings'
+            saved = {'version': None,
+                        'current': None}
+        if saved['version'] == self.model.version:
+            self.model.activeView = saved['current']
+            self.dock.updateDock()
+            self.colorDialog.updateDialogValues()
+            self.applyChanges()
+            message = '{} settings loaded'.format(filename)
+        else:
+            message = 'Error loading plot settings. Incompatible model.'
+        self.statusBar().showMessage(message, 5000)
+
     def openView(self):
         filename, ext = QFileDialog.getOpenFileName(self, "Open View Settings",
                                                     ".", "*.pltvw")
         if filename:
-            try:
-                with open(filename, 'rb') as file:
-                    saved = pickle.load(file)
-            except Exception:
-                message = 'Error loading plot settings'
-                saved = {'version': None,
-                         'current': None}
-            if saved['version'] == self.model.version:
-                self.model.activeView = saved['current']
-                self.dock.updateDock()
-                self.colorDialog.updateDialogValues()
-                self.applyChanges()
-                message = '{} settings loaded'.format(filename)
-            else:
-                message = 'Error loading plot settings. Incompatible model.'
-            self.statusBar().showMessage(message, 5000)
+            self.loadViewFile(filename)
 
     def openStatePoint(self):
         # check for an alread-open statepoint

openmc_plotter/plotgui.py

@@ -111,6 +111,18 @@ def _resize(self):
         self.resize(self.parent.width() * z,
                     self.parent.height() * z)
 
+    def saveImage(self, filename):
+        """Save an image of the current view
+
+        Parameters
+        ----------
+        filename : str or pathlib.Path
+            Name of the image to save
+        """
+        if "." not in filename:
+            filename += ".png"
+        self.figure.savefig(filename, transparent=True)
+
     def getDataIndices(self, event):
         cv = self.model.currentView