From 8b8845e452b45b765c0c74a388296d10e91787fc Mon Sep 17 00:00:00 2001
From: Samo Penic <samo.penic@gmail.com>
Date: Tue, 28 Nov 2017 21:03:17 +0000
Subject: [PATCH] Changed the executabe to tssystem ;)

---
 trisurf/VTKRendering.py |   27 ++++++++++++++++++++++++---
 1 files changed, 24 insertions(+), 3 deletions(-)

diff --git a/trisurf/VTKRendering.py b/trisurf/VTKRendering.py
index bc07bc2..9080a9a 100644
--- a/trisurf/VTKRendering.py
+++ b/trisurf/VTKRendering.py
@@ -80,21 +80,22 @@
 
 
 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()
 		self.renderer.AddActor(self.actor)
 		self.renderer.AddActor(self.textactor)
-		self.renderer.SetBackground(0, 0, 0) # Set background to white
+		self.renderer.SetBackground(81/255, 87/255, 110/255) # Set background to nicely grey
 		# Create the RendererWindow
 		self.renderer_window = vtkRenderWindow()
 		self.renderer_window.AddRenderer(self.renderer)
-		self.renderer_window.SetSize(1200,600)
+		self.renderer_window.SetSize(800,800)
 	
 #		self.renderer.SetViewport(0.0,0.0,0.5,1.0)
 #		rend=vtk.vtkRenderer()
@@ -137,11 +138,31 @@
 		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)
 
+		#advanced lookuptables...
+		lut = vtk.vtkLookupTable()
+		lutNum = 256
+		lut.SetNumberOfTableValues(lutNum)
+		ctf = vtk.vtkColorTransferFunction()
+		ctf.SetColorSpaceToDiverging()
+		ctf.AddRGBPoint(0.0, 0.230, 0.299, 0.754)
+		ctf.AddRGBPoint(1.0, 0.706, 0.016, 0.150)
+		for ii,ss in enumerate([float(xx)/float(lutNum) for xx in range(lutNum)]):
+			cc = ctf.GetColor(ss)
+			lut.SetTableValue(ii,cc[0],cc[1],cc[2],1.0)
+		mapper.SetLookupTable(lut)
 		# Create the Actor
 		actor = vtkActor()
 		actor.SetMapper(mapper)

--
Gitblit v1.9.3