parallel creation of images example

Author: theodoregoetz

hotwater-grid-stylesheet.py

@@ -60,3 +60,4 @@ def frame_grid(nrows, ncols):
             frame.load_stylesheet('isosurface.sty')
 
 tp.save_png('hotwater-grid-stylesheet.png')
+

hotwater-isosurface-recorded.py

@@ -29,3 +29,4 @@
     1.45886)
 tp.active_frame().plot().view.width=0.339885
 # End Macro.
+

hotwater-parallel.py

@@ -0,0 +1,49 @@
+import atexit
+import glob
+import os
+import multiprocessing
+import sys
+
+import tecplot as tp
+
+def init():
+    # !!! IMPORTANT !!!
+    # Must register stop at exit to ensure Tecplot cleans
+    # up all temporary files and does not create a core dump
+    atexit.register(tp.session.stop)
+
+def work(datafile):
+    tp.new_layout()
+    tp.data.load_tecplot(datafile)
+    tp.active_frame().load_stylesheet('isosurface.sty')
+    imgfile = os.path.basename(datafile).replace('.plt', '.png')
+    tp.save_png(os.path.join('images', imgfile))
+
+if __name__ == '__main__':
+    # !!! IMPORTANT !!!
+    # On Linux systems, Python's multiprocessing start method
+    # defaults to "fork" which is incompatible with PyTecplot
+    # and must be set to "spawn"
+    multiprocessing.set_start_method('spawn')
+
+    # Get the datafiles
+    files = glob.glob('hotwatermixing/HotWaterMixing_Y05YT10*.plt')
+
+    # Set up the pool with initializing function and associated arguments
+    num_workers = min(multiprocessing.cpu_count(), len(files))
+    pool = multiprocessing.Pool(num_workers, initializer=init)
+
+    try:
+        if not os.path.exists('images'):
+            os.makedirs('images')
+
+        # Map the work function to each of the job arguments
+        pool.map(work, files)
+    finally:
+        # !!! IMPORTANT !!!
+        # Must join the process pool before parent script exits
+        # to ensure Tecplot cleans up all temporary files
+        # and does not create a core dump
+        pool.close()
+        pool.join()
+