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 |   12 +++++++++++-
 1 files changed, 11 insertions(+), 1 deletions(-)

diff --git a/trisurf/VTKRendering.py b/trisurf/VTKRendering.py
index bc07bc2..4cca00f 100644
--- a/trisurf/VTKRendering.py
+++ b/trisurf/VTKRendering.py
@@ -80,11 +80,12 @@
 
 
 class Renderer:
-	def __init__(self,args,host,run, timestep=-1):
+	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()
@@ -137,10 +138,19 @@
 		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