import os,sys
|
from . import trisurf
|
try:
|
from vtk import *
|
except:
|
print("Vtk rendering works if you manually install vtk7 for python3")
|
# exit(1)
|
|
class MultiRender:
|
def __init__(self,args,host):
|
target_runs=getRargetRunIdxList(args)
|
if target_runs==None:
|
target_runs=list(range(1,len(host['runs'])+1))
|
nruns=len(target_runs)
|
#prepare rendering window
|
self.renderer_window = vtkRenderWindow()
|
self.renderer_window.AddRenderer(self.renderer)
|
self.renderer_window.SetSize(1200,600)
|
interactor = vtkRenderWindowInteractor()
|
interactor.SetRenderWindow(self.renderer_window)
|
interactor.Initialize()
|
interactor.AddObserver("TimerEvent", self.RenderUpdate)
|
timerIDR = interactor.CreateRepeatingTimer(1000)
|
self.filename=[]
|
self.renderer=[]
|
i=0
|
for run in target_runs:
|
#for each target run calculate renderer location
|
r.vtkRenderer()
|
r.SetBackground(0,0,0)
|
p=1.0/float(nruns)
|
x1=i*p
|
x2=(i+1)*p
|
r.SetViewport(x1,0.0,x2,1.0)
|
self.renderer.Append(r)
|
self.renderer_window.AddRenderer(r)
|
i=i+1
|
#call Renderer object with Run, renderer
|
#start endless loop of interactor
|
interactor.Start()
|
|
def lastVTU(self,run):
|
Dir=trisurf.Directory(maindir=run.maindir,simdir=run.subdir)
|
filename=os.path.join("./",Dir.fullpath(),run.getLastVTU())
|
return filename
|
|
def lastActorForRun(self,run):
|
filename=self.lastVTU(run)
|
reader=vtkXMLUnstructuredGridReader()
|
reader.SetFileName(self.filename)
|
reader.Update() # Needed because of GetScalarRange
|
output = reader.GetOutput()
|
scalar_range = output.GetScalarRange()
|
mapper = vtkDataSetMapper()
|
mapper.SetInputData(output)
|
mapper.SetScalarRange(scalar_range)
|
|
# Create the Actor
|
actor = vtkActor()
|
actor.SetMapper(mapper)
|
return actor
|
|
|
|
def RenderUpdate(self, obj, event):
|
i=0
|
for run in runs:
|
if(self.lastVTU(run)!=self.filename[i]):
|
#print("updejt")
|
self.renderer.RemoveActor(self.actor)
|
self.renderer.RemoveActor(self.textactor)
|
self.actor=self.lastActor()
|
self.textactor=self.textActor()
|
self.renderer.AddActor(self.actor)
|
self.renderer.AddActor(self.textactor)
|
self.renderer_window.Render()
|
#self.render.RemoveActor(self.actor)
|
i=i+1
|
return
|
|
|
class Renderer:
|
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(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(800,800)
|
|
# self.renderer.SetViewport(0.0,0.0,0.5,1.0)
|
# rend=vtk.vtkRenderer()
|
# rend.AddActor(self.actor)
|
# rend.SetViewport(0.5,0.0,1.0,1.0)
|
# self.renderer_window.AddRenderer(rend)
|
# Set up a check for aborting rendering.
|
# Create the RendererWindowInteractor and display the vtk_file
|
interactor = vtkRenderWindowInteractor()
|
interactor.SetRenderWindow(self.renderer_window)
|
interactor.Initialize()
|
interactor.AddObserver("TimerEvent", self.RenderUpdate)
|
timerIDR = interactor.CreateRepeatingTimer(10000)
|
interactor.Start()
|
return
|
|
def lastVTU(self):
|
#Dir=trisurf.Directory(maindir=self.host['runs'][self.run].maindir,simdir=self.host['runs'][self.run].subdir)
|
Dir=self.run.Dir
|
#print(self.run.getLastVTU())
|
filename=os.path.join("./",Dir.fullpath(),self.run.getLastVTU())
|
#filename=os.path.join("./",Dir.fullpath(),self.host['runs'][self.run].getLastVTU())
|
return filename
|
|
def textActor(self):
|
textactor=vtkTextActor()
|
textactor.SetInput(self.filename)
|
tp=textactor.GetTextProperty()
|
tp.SetColor(1,1,1)
|
tp.SetFontSize(11)
|
textactor.SetDisplayPosition(20,30)
|
return textactor
|
|
def lastActor(self):
|
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)
|
|
#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)
|
return actor
|
|
|
def RenderUpdate(self, obj, event):
|
if(self.lastVTU()!=self.filename):
|
#print("updejt")
|
self.renderer.RemoveActor(self.actor)
|
self.renderer.RemoveActor(self.textactor)
|
self.actor=self.lastActor()
|
self.textactor=self.textActor()
|
self.renderer.AddActor(self.actor)
|
self.renderer.AddActor(self.textactor)
|
self.renderer_window.Render()
|
#self.render.RemoveActor(self.actor)
|
|
return
|
|
|
class RenderVTUFile(Renderer):
|
def __init__(self,args,filename=None,scalar_field='vertices_idx'):
|
self.filename=filename
|
Renderer.__init__(self,args,host=None, run=None, timestep=-1, scalar_field=scalar_field)
|
|
def lastVTU(self):
|
if(self.filename):
|
return self.filename
|
else:
|
raise Exception('No file specified')
|
