From d55834e161e1f79b7685bb56202f48914b083364 Mon Sep 17 00:00:00 2001
From: Samo Penic <samo.penic@gmail.com>
Date: Sun, 15 Jan 2017 13:42:16 +0000
Subject: [PATCH] Added possibility of setting the VTU color mapper. Color palete is not yet implemented.
---
trisurf/VTKRendering.py | 18 ++++++++++++++++--
1 files changed, 16 insertions(+), 2 deletions(-)
diff --git a/trisurf/VTKRendering.py b/trisurf/VTKRendering.py
index 0ddb5a0..4cca00f 100644
--- a/trisurf/VTKRendering.py
+++ b/trisurf/VTKRendering.py
@@ -80,10 +80,12 @@
class Renderer:
- def __init__(self,args,host,run):
+ def __init__(self,args,host,run, timestep=-1, scalar_field='vertices_idx'):
self.host=host
self.args=args
self.run=run
+ self.timestep=timestep
+ self.scalar_field=scalar_field
self.renderer = vtkRenderer()
self.actor=self.lastActor()
self.textactor=self.textActor()
@@ -128,15 +130,27 @@
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
output = reader.GetOutput()
+ output.GetPointData().SetActiveScalars(self.scalar_field)
scalar_range = output.GetScalarRange()
mapper = vtkDataSetMapper()
mapper.SetInputData(output)
mapper.SetScalarRange(scalar_range)
+
+ #color lookuptables
+ #lut = vtk.vtkLookupTable()
+ #lut.SetHueRange(0.5, 0.6)
+ #lut.SetSaturationRange(0, 1)
+ #lut.SetValueRange(0, 1)
+
+ #mapper.SetLookupTable(lut)
# Create the Actor
actor = vtkActor()
--
Gitblit v1.9.3