#!/usr/bin/python
# visit -cli -geometry 320x240 -s DynaVisitRecipe.py
# visit -cli -nowin -s DynaVisitRecipe.py

# print PlotPlugins()

print "VTF VisIt Movie Frame Dumper (One-at-a-time Version) 2006 v.0.1"

HideToolbars()

windowAttributes = SaveWindowAttributes()
windowAttributes.fileName = "/dev/null/shot.vtfdyna.BLANK.png"
windowAttributes.format   = windowAttributes.PNG
windowAttributes.width    = 640
windowAttributes.height   = 480
windowAttributes.family   = 0
SetSaveWindowAttributes(windowAttributes)

viewAttributes = View3DAttributes()
annotations = AnnotationAttributes()
annotations.axesFlag=0
annotations.bboxFlag=0
annotations.triadFlag=0
annotations.gradientBackgroundStyle=annotations.TopToBottom
annotations.gradientColor1  =(255,255,255,255)
annotations.gradientColor2  =(223,223,223,255)
annotations.backgroundMode  =annotations.Gradient
annotations.userInfoFlag    =0
annotations.databaseInfoFlag=0
annotations.legendInfoFlag  =1

SetAnnotationAttributes(annotations)

first =1
viewSetup=View3DAttributes()

solidColorScale=PseudocolorAttributes()
solidColorScale.minFlag=1
solidColorScale.maxFlag=1
solidColorScale.min=0.00
solidColorScale.max=0.01
solidColorScale.colorTableName="calewhite"

fluidColorScale=PseudocolorAttributes()
fluidColorScale.minFlag=1
fluidColorScale.maxFlag=1
solidColorScale.min= 0.00
fluidColorScale.max=15.00
#fluidColorScale.colorTableName="hot"

clipSetup=ClipAttributes()
clipSetup.plane1Status=0
clipSetup.plane2Status=1
clipSetup.plane3Status=1

startFrame        = 1
endFrame          = 153
stepBetweenFrames = 1

for i in range(startFrame,endFrame,stepBetweenFrames):
	# HARDWIRED NOTICE:
	windowAttributes.fileName = "shots/shot.vtfdyna.%04i.png" % i

	# HARDWIRED NOTICE:
	solidfile = "/Volumes/Documents/viz/vtf/2006-dyna/solid/cannister.%04i.vtk" % i
	fluidfile = "/Volumes/Documents/viz/vtf/2006-dyna/fluid/fluid.%04i.vtk" % i

	SetSaveWindowAttributes(windowAttributes)
	print "Opening ",solidfile

	OpenDatabase(solidfile)
	AddPlot("Pseudocolor","Color")
	SetPlotOptions(solidColorScale)
	AddOperator("Clip")
	SetOperatorOptions(clipSetup)

	OpenDatabase(fluidfile)
	AddPlot("Pseudocolor","Density")
	SetPlotOptions(fluidColorScale)
	AddOperator("Clip")
	SetOperatorOptions(clipSetup)

	DrawPlots()
	if first == 1:
		# let visit setup near/far planes and such, then just rotate the cannister
		viewSetup = GetView3D()
		#viewSetup.viewNormal = ( -.707, 0, .707 )
		viewSetup.viewNormal = (-0.480788, 0.397592,  0.781514)
		viewSetup.viewUp =     ( 0.19518 , 0.917453, -0.346676)
		print viewSetup
		first = 0
	SetView3D (viewSetup)
	SaveWindow()
	SetActivePlots(1)
	HideActivePlots()
	SetActivePlots(0)
	HideActivePlots()
	DeleteAllPlots()
	ClearWindow()
	CloseDatabase(solidfile)
	CloseDatabase(fluidfile)

#ShowToolbars()

print "Done!"

CloseComputeEngine()
Close()