def write_window(f, prefix):
# Window setup
f.write('width, height = %d, %d\n' %
visit.GetWindowInformation().windowSize)
f.write(
'win = GetGlobalAttributes().windows[GetGlobalAttributes().activeWindow]\n'
)
f.write('ResizeWindow(win, width, height)\n')
f.write('SetActiveWindow(win) # Synchronize\n')
f.write('size = GetWindowInformation().windowSize\n')
f.write('if width < size[0] or height < size[1]:\n')
f.write(
' ResizeWindow(win, width + (size[0] - width), height + (size[1] - height))\n'
)
f.write('DeleteAllPlots()\n')
f.write('for name in GetAnnotationObjectNames():\n')
f.write(' DeleteAnnotationObject(name)\n')
f.write('\n')
# Write the plots.
write_plots(f)
f.write('DrawPlots()\n\n')
# Set the view
f.write('# Set the view\n')
if visit.GetWindowInformation().viewDimension == 1:
view = visit.GetViewCurve()
elif visit.GetWindowInformation().viewDimension == 2:
view = visit.GetView2D()
else:
view = visit.GetView3D()
write_state_object(f, view, "view")
if visit.GetWindowInformation().viewDimension == 1:
f.write('SetViewCurve(view)\n')
elif visit.GetWindowInformation().viewDimension == 2:
f.write('SetView2D(view)\n')
else:
f.write('SetView3D(view)\n')
f.write('\n')
# Get the annotation attributes
f.write('# Set the annotation attributes\n')
annot = visit.GetAnnotationAttributes()
if write_state_object_diffs(f, annot, visit.AnnotationAttributes(),
"annot"):
f.write('SetAnnotationAttributes(annot)\n\n')
else:
f.write('#SetAnnotationAttributes(annot)\n\n')
set_annotation_objects(f, prefix)
def plot(dbfile=None):
if not dbfile is None:
visit.OpenDatabase(dbfile)
else:
wi = visit.GetWindowInformation()
dbfile = wi.activeSource
md = visit.GetMetaData(dbfile)
ncrvs = md.GetNumCurves()
cnames = [md.GetCurves(i).name for i in xrange(ncrvs)]
cnames.sort()
for cname in cnames:
visit.AddPlot("Curve", cname)
visit.DrawPlots()