From 5f0a3578dc6208fede1dc0da0d35fac03e5474d7 Mon Sep 17 00:00:00 2001
From: Samo Penic <samo.penic@gmail.com>
Date: Sat, 14 Jan 2017 08:52:24 +0000
Subject: [PATCH] Additional functionality in VTU renderer. Added capability to specify optional timestep within the run

---
 trisurf/VTKRendering.py |    8 ++++++--
 1 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/trisurf/VTKRendering.py b/trisurf/VTKRendering.py
index 0ddb5a0..bc07bc2 100644
--- a/trisurf/VTKRendering.py
+++ b/trisurf/VTKRendering.py
@@ -80,10 +80,11 @@
 
 
 class Renderer:
-	def __init__(self,args,host,run):
+	def __init__(self,args,host,run, timestep=-1):
 		self.host=host
 		self.args=args
 		self.run=run
+		self.timestep=timestep
 		self.renderer = vtkRenderer()
 		self.actor=self.lastActor()
 		self.textactor=self.textActor()
@@ -128,7 +129,10 @@
 		return textactor
 
 	def lastActor(self):
-		self.filename=self.lastVTU()
+		if(self.timestep<0):
+			self.filename=self.lastVTU()
+		else:
+			self.filename=os.path.join("./",self.run.Dir.fullpath(),'timestep_{:06d}.vtu'.format(self.timestep))
 		reader=vtkXMLUnstructuredGridReader()
 		reader.SetFileName(self.filename)
 		reader.Update() # Needed because of GetScalarRange

--
Gitblit v1.9.3