tvtk module optional - sphere - GPU-based 3D discrete element method algorithm … | |
git clone git://src.adamsgaard.dk/sphere | |
Log | |
Files | |
Refs | |
LICENSE | |
--- | |
commit 3b79a5ef5d587c082856f945e4aacb8247ae20b1 | |
parent e37744f1b9fdd39f371ad2930b341ba01b37e894 | |
Author: Anders Damsgaard <[email protected]> | |
Date: Sun, 22 Jun 2014 10:40:08 +0200 | |
vtk module optional | |
Diffstat: | |
M python/sphere.py | 14 +++++++++++++- | |
1 file changed, 13 insertions(+), 1 deletion(-) | |
--- | |
diff --git a/python/sphere.py b/python/sphere.py | |
t@@ -6,7 +6,13 @@ matplotlib.use('Agg') | |
import matplotlib.pyplot as plt | |
from matplotlib.font_manager import FontProperties | |
import subprocess | |
-import vtk | |
+try: | |
+ import vtk | |
+except ImportError: | |
+ print('Warning: Could not find "vtk" python module. VTK calls unavailable') | |
+ py_vtk = False | |
+else: | |
+ py_vtk = True | |
numpy.seterr(all='warn', over='raise') | |
t@@ -1158,6 +1164,8 @@ class sim: | |
:param verbose: Show diagnostic information (default = True) | |
:type verbose: bool | |
''' | |
+ if py_vtk == False: | |
+ raise Exception('Error: vtk module not found.') | |
lastfile = status(self.sid) | |
sb = sim(fluid = self.fluid) | |
t@@ -1218,6 +1226,8 @@ class sim: | |
:param verbose: Show diagnostic information (default = True) | |
:type verbose: bool | |
''' | |
+ if py_vtk == False: | |
+ raise Exception('Error: vtk module not found.') | |
fh = None | |
try : | |
t@@ -1475,6 +1485,8 @@ class sim: | |
:param verbose: Show diagnostic information (default = True) | |
:type verbose: bool | |
''' | |
+ if py_vtk == False: | |
+ raise Exception('Error: vtk module not found.') | |
filename = folder + '/fluid-' + self.sid + '.vti' # image grid | |