def isosurfacePlot(self, stay=False, samples=100, contours=50):
visit.AddWindow()
returned = []
if len(visit.ListPlots()) > 0:
visit.SetActivePlots(0)
visit.HideActivePlots()
p = visit.PseudocolorAttributes()
q = visit.ResampleAttributes()
r = visit.IsosurfaceAttributes()
q.samplesX = samples
q.samplesY = samples
q.samplesZ = samples
r.contourNLevels = contours
plot = visit.AddPlot("Pseudocolor", "Heat")
visit.AddOperator("Resample")
# visit.AddOperator("Isosurface")
visit.SetOperatorOptions(q)
visit.SetOperatorOptions(r)
visit.SetPlotOptions(p)
visit.DrawPlots()
returned.append(visit.SaveWindow())
count = 0
visit.AddOperator("Isosurface")
visit.DrawPlots()
time.sleep(60)
returned.append(visit.SaveWindow())
return returned
def setup_visit(plot_variable_data, db_name):
print 'Opening database...'
# Get the complete database name including directory path, db file
# pattern and extension, followed by the word database
# Example: db_filename = './output*.vtk database'
visit.OpenDatabase(db_name)
# Getting the list of current variables
db_meta_data = visit.GetMetaData(db_name)
scalar_list = [db_meta_data.GetScalars(i).name for i in range(db_meta_data.GetNumScalars())]
vector_list = [db_meta_data.GetVectors(i).name for i in range(db_meta_data.GetNumVectors())]
for vec in vector_list:
scalar_list.append(vec + '_magnitude')
# Check if current variable name exists in the database. If no, then
# res becomes 0. Hence, we need to define a scalar expression for the
# current variable
if plot_variable_data['plot variable'] in scalar_list or \
plot_variable_data['plot variable'] in vector_list:
visit.AddPlot(plot_variable_data['plot type'], \
plot_variable_data['plot variable'])
else:
print 'No existing variable of that name'
visit.DefineScalarExpression(plot_variable_data['plot variable'], \
plot_variable_data['plot variable definition'])
visit.AddPlot(plot_variable_data['plot type'], \
plot_variable_data['plot variable'])
if plot_variable_data['plot type'] == 'Pseudocolor':
p = visit.PseudocolorAttributes()
p.SetCentering(1)
p.colorTableName = 'hot_desaturated'
# p.colorTableName = 'rainbow'
visit.SetPlotOptions(p)
def SetPseudocolorAttributes(valMin, valMax):
# The minimum and maximum values of color box for the variable values
# are set as constant values valMin and valMax (i.e. not varying in time).
pa = visit.PseudocolorAttributes()
# options I like to set
pa.minFlag = 1
pa.min = valMin
pa.maxFlag = 1
pa.max = valMax
# other options with default values
pa.scaling = pa.Linear # Linear, Log, Skew
pa.skewFactor = 1
pa.limitsMode = pa.OriginalData # OriginalData, CurrentPlot
pa.centering = pa.Natural # Natural, Nodal, Zonal
pa.colorTableName = "hot"
pa.invertColorTable = 0
pa.opacityType = pa.FullyOpaque # ColorTable, FullyOpaque, Constant, Ramp, VariableRange
pa.opacityVariable = ""
pa.opacity = 1
pa.opacityVarMin = 0
pa.opacityVarMax = 1
pa.opacityVarMinFlag = 0
pa.opacityVarMaxFlag = 0
pa.pointSize = 0.05
pa.pointType = pa.Point # Box, Axis, Icosahedron, Octahedron, Tetrahedron, SphereGeometry, Point, Sphere
pa.pointSizeVarEnabled = 0
pa.pointSizeVar = "default"
pa.pointSizePixels = 2
pa.lineType = pa.Line # Line, Tube, Ribbon
pa.lineStyle = pa.SOLID # SOLID, DASH, DOT, DOTDASH
pa.lineWidth = 0
pa.tubeDisplayDensity = 10
pa.tubeRadiusSizeType = pa.FractionOfBBox # Absolute, FractionOfBBox
pa.tubeRadiusAbsolute = 0.125
pa.tubeRadiusBBox = 0.005
pa.varyTubeRadius = 0
pa.varyTubeRadiusVariable = ""
pa.varyTubeRadiusFactor = 10
pa.endPointType = pa.None # None, Tails, Heads, Both
pa.endPointStyle = pa.Spheres # Spheres, Cones
pa.endPointRadiusSizeType = pa.FractionOfBBox # Absolute, FractionOfBBox
pa.endPointRadiusAbsolute = 1
pa.endPointRadiusBBox = 0.005
pa.endPointRatio = 2
pa.renderSurfaces = 1
pa.renderWireframe = 0
pa.renderPoints = 0
pa.smoothingLevel = 0
pa.legendFlag = 1
pa.lightingFlag = 1
visit.SetPlotOptions(pa)
def binPhase(eDB):
visit.OpenDatabase(eDB, 0)
visit.AddPlot("Pseudocolor", "operators/DataBinning/2D/electrons", 1, 1)
DataBinningAtts = visit.DataBinningAttributes()
DataBinningAtts.numDimensions = DataBinningAtts.Two # One, Two, Three
DataBinningAtts.dim1BinBasedOn = DataBinningAtts.Variable # X, Y, Z, Variable
DataBinningAtts.dim1Var = "electrons_zSI"
DataBinningAtts.dim1SpecifyRange = 1
DataBinningAtts.dim1MinRange = 0
DataBinningAtts.dim1MaxRange = 1.0E-7
DataBinningAtts.dim1NumBins = 100
DataBinningAtts.dim2BinBasedOn = DataBinningAtts.Variable # X, Y, Z, Variable
DataBinningAtts.dim2Var = "electrons_gamma"
DataBinningAtts.dim2SpecifyRange = 1
DataBinningAtts.dim2MinRange = 0.991
DataBinningAtts.dim2MaxRange = 1.005
DataBinningAtts.dim2NumBins = 100
DataBinningAtts.outOfBoundsBehavior = DataBinningAtts.Clamp # Clamp, Discard
DataBinningAtts.reductionOperator = DataBinningAtts.Sum # Average, Minimum, Maximum, StandardDeviation, Variance, Sum, Count, RMS, PDF
DataBinningAtts.varForReduction = "electrons_chargeSI"
DataBinningAtts.emptyVal = 0
DataBinningAtts.outputType = DataBinningAtts.OutputOnBins # OutputOnBins, OutputOnInputMesh
DataBinningAtts.removeEmptyValFromCurve = 1
visit.SetOperatorOptions(DataBinningAtts, 1)
visit.SetTimeSliderState(0)
visit.SetTimeSliderState(1)
PseudocolorAtts = visit.PseudocolorAttributes()
PseudocolorAtts.centering = PseudocolorAtts.Nodal # Natural, Nodal, Zonal
PseudocolorAtts.opacityType = PseudocolorAtts.Ramp # ColorTable, FullyOpaque, Constant, Ramp, VariableRange
PseudocolorAtts.colorTableName = "hot_desaturated"
visit.SetPlotOptions(PseudocolorAtts)
View2DAtts = visit.View2DAttributes()
View2DAtts.fullFrameActivationMode = View2DAtts.On
# View2DAtts.fullFrameAutoThreshold = 100
visit.SetView2D(View2DAtts)
visit.ResetView()
AnnotationAtts = visit.AnnotationAttributes()
AnnotationAtts.axes2D.yAxis.grid = 0
AnnotationAtts.axes2D.xAxis.grid = 0
AnnotationAtts.axes2D.xAxis.title.userTitle = 1
AnnotationAtts.axes2D.xAxis.title.userUnits = 1
AnnotationAtts.axes2D.xAxis.title.title = "ct-z"
AnnotationAtts.axes2D.xAxis.title.units = "m"
AnnotationAtts.axes2D.yAxis.title.userTitle = 1
AnnotationAtts.axes2D.yAxis.title.userUnits = 1
AnnotationAtts.axes2D.yAxis.title.title = "gamma"
AnnotationAtts.axes2D.yAxis.title.units = ""
AnnotationAtts.userInfoFlag = 0
AnnotationAtts.databaseInfoFlag = 0
# AnnotationAtts.legendInfoFlag = 0
visit.SetAnnotationAttributes(AnnotationAtts)
visit.DrawPlots()
def pseudocolorPlot(self):
visit.AddWindow()
returned = []
if len(visit.ListPlots()) > 0:
visit.SetActivePlots(0)
visit.HideActivePlots()
p = visit.PseudocolorAttributes()
plot = visit.AddPlot("Pseudocolor", "Heat")
visit.SetPlotOptions(p)
visit.DrawPlots()
return visit.SaveWindow()
def test_visit():
import visit
VISIT_ARGS = os.environ.get('VISIT_ARGS', ["-nosplash"])
if isinstance(VISIT_ARGS, str):
VISIT_ARGS = VISIT_ARGS.split()
for arg in VISIT_ARGS:
visit.AddArgument(arg)
visit.Launch()
print "visit is installed in", os.path.dirname(visit.__file__)
print "visit version", visit.Version()
visit.ResetView()
v3D = visit.GetView3D()
v3D.SetViewUp(0, 0, 1)
v3D.SetViewNormal(-0.5, -0.8, 0.4)
v3D.SetImageZoom(1.0)
visit.SetView3D(v3D)
aa = visit.AnnotationAttributes()
aa.SetAxesType(2) # outside edges
aa.SetDatabaseInfoFlag(False)
aa.SetUserInfoFlag(False)
visit.SetAnnotationAttributes(aa)
t = visit.CreateAnnotationObject("Text2D")
t.SetText("Hello VisIt!")
t.SetPosition(0.4, 0.9) # (0,0) is lower left corner
t.SetFontFamily(0) # 0: Arial, 1: Courier, 2: Times
t.SetWidth(0.25) # 25%
t.SetTextColor((0, 0, 0))
t.SetUseForegroundForTextColor(False)
t.SetVisible(True)
visit.OpenDatabase("tmp_.vtk")
visit.AddPlot("Mesh", "scalars")
ma = visit.MeshAttributes()
visit.SetPlotOptions(ma)
visit.AddPlot("Pseudocolor", "scalars")
pa = visit.PseudocolorAttributes()
visit.SetPlotOptions(pa)
visit.RedrawWindow()
visit.DrawPlots()
wait()
def _plot_pseudocolor(self):
"""Plots the data on a pseudocolor plot to use."""
# add the pseudocolor plot to contour
v.AddPlot("Pseudocolor", self.data)
att = v.PseudocolorAttributes()
# min/max for the pseudocolor plot
att.minFlag = True
att.min = self.minN
att.maxFlag = True
att.max = self.maxN
# plot
v.SetPlotOptions(att)
v.DrawPlots()
def _getDefaultPseudoColorAttributes(self):
pseudocolorAttributes = visit.PseudocolorAttributes()
pseudocolorAttributes.scaling = pseudocolorAttributes.Linear # Linear, Log, Skew
pseudocolorAttributes.skewFactor = 1
pseudocolorAttributes.limitsMode = pseudocolorAttributes.OriginalData # OriginalData, CurrentPlot
pseudocolorAttributes.minFlag = 0
pseudocolorAttributes.min = 0
pseudocolorAttributes.maxFlag = 0
pseudocolorAttributes.max = 0
pseudocolorAttributes.centering = pseudocolorAttributes.Natural # Natural, Nodal, Zonal
pseudocolorAttributes.colorTableName = "hot"
pseudocolorAttributes.invertColorTable = 0
pseudocolorAttributes.opacityType = pseudocolorAttributes.FullyOpaque # ColorTable, FullyOpaque, Constant, Ramp, VariableRange
pseudocolorAttributes.opacityVariable = ""
pseudocolorAttributes.opacity = 1
pseudocolorAttributes.opacityVarMin = 0
pseudocolorAttributes.opacityVarMax = 1
pseudocolorAttributes.opacityVarMinFlag = 0
pseudocolorAttributes.opacityVarMaxFlag = 0
pseudocolorAttributes.pointSize = 0.05
pseudocolorAttributes.pointType = pseudocolorAttributes.Point # Box, Axis, Icosahedron, Octahedron, Tetrahedron, SphereGeometry, Point, Sphere
pseudocolorAttributes.pointSizeVarEnabled = 0
pseudocolorAttributes.pointSizeVar = "default"
pseudocolorAttributes.pointSizePixels = 2
pseudocolorAttributes.lineType = pseudocolorAttributes.Line # Line, Tube, Ribbon
pseudocolorAttributes.lineStyle = pseudocolorAttributes.SOLID # SOLID, DASH, DOT, DOTDASH
pseudocolorAttributes.lineWidth = 0
# pseudocolorAttributes.tubeDisplayDensity = 10
pseudocolorAttributes.tubeRadiusSizeType = pseudocolorAttributes.FractionOfBBox # Absolute, FractionOfBBox
pseudocolorAttributes.tubeRadiusAbsolute = 0.125
pseudocolorAttributes.tubeRadiusBBox = 0.005
# pseudocolorAttributes.varyTubeRadius = 0
# pseudocolorAttributes.varyTubeRadiusVariable = ""
# pseudocolorAttributes.varyTubeRadiusFactor = 10
# pseudocolorAttributes.endPointType = pseudocolorAttributes.None # None, Tails, Heads, Both
# pseudocolorAttributes.endPointStyle = pseudocolorAttributes.Spheres # Spheres, Cones
pseudocolorAttributes.endPointRadiusSizeType = pseudocolorAttributes.FractionOfBBox # Absolute, FractionOfBBox
pseudocolorAttributes.endPointRadiusAbsolute = 1
pseudocolorAttributes.endPointRadiusBBox = 0.005
pseudocolorAttributes.endPointRatio = 2
pseudocolorAttributes.renderSurfaces = 1
pseudocolorAttributes.renderWireframe = 0
pseudocolorAttributes.renderPoints = 0
pseudocolorAttributes.smoothingLevel = 0
pseudocolorAttributes.legendFlag = 1
pseudocolorAttributes.lightingFlag = 1
return pseudocolorAttributes
def surfIntens(fDB):
visit.OpenDatabase(fDB, 0)
visit.AddPlot("Pseudocolor", "operators/DataBinning/2D/meshScaled", 1, 1)
DataBinningAtts = visit.DataBinningAttributes()
DataBinningAtts.numDimensions = DataBinningAtts.Two # One, Two, Three
DataBinningAtts.dim1BinBasedOn = DataBinningAtts.X # X, Y, Z, Variable
DataBinningAtts.dim1Var = "default"
DataBinningAtts.dim1SpecifyRange = 0
DataBinningAtts.dim1MinRange = 0
DataBinningAtts.dim1MaxRange = 1
DataBinningAtts.dim1NumBins = 85
DataBinningAtts.dim2BinBasedOn = DataBinningAtts.Y # X, Y, Z, Variable
DataBinningAtts.dim2Var = "default"
DataBinningAtts.dim2SpecifyRange = 0
DataBinningAtts.dim2MinRange = 0
DataBinningAtts.dim2MaxRange = 1
DataBinningAtts.dim2NumBins = 85
DataBinningAtts.dim3BinBasedOn = DataBinningAtts.Variable # X, Y, Z, Variable
DataBinningAtts.dim3Var = "default"
DataBinningAtts.dim3SpecifyRange = 0
DataBinningAtts.dim3MinRange = 0
DataBinningAtts.dim3MaxRange = 1
DataBinningAtts.dim3NumBins = 50
DataBinningAtts.outOfBoundsBehavior = DataBinningAtts.Clamp # Clamp, Discard
DataBinningAtts.reductionOperator = DataBinningAtts.Average # Average, Minimum, Maximum, StandardDeviation, Variance, Sum, Count, RMS, PDF
DataBinningAtts.varForReduction = "intensityScaled"
DataBinningAtts.emptyVal = 0
DataBinningAtts.outputType = DataBinningAtts.OutputOnBins # OutputOnBins, OutputOnInputMesh
DataBinningAtts.removeEmptyValFromCurve = 1
visit.SetOperatorOptions(DataBinningAtts, 1)
View2DAtts = visit.View2DAttributes()
View2DAtts.fullFrameActivationMode = View2DAtts.On
# View2DAtts.fullFrameAutoThreshold = 100
visit.SetView2D(View2DAtts)
visit.ResetView()
PseudocolorAtts = visit.PseudocolorAttributes()
PseudocolorAtts.centering = PseudocolorAtts.Nodal # Natural, Nodal, Zonal
visit.SetPlotOptions(PseudocolorAtts)
visit.DrawPlots()
def draw_point_picture(variableName,
minValue,
maxValue,
inputDirectory,
inputFileName,
coordinate,
outputDirectory,
outputFileName,
colorTable="hot_desaturated"):
'''
Function for making a visit plot with a point
Arguments:
:param variableName Name of the variable
:param minValue Minimum value of the variable
:param maxValue Maximum value of the variable
:param inputDirectory Path to input vlsv/silo files
:param inputFileName Name of the file, for example \"bulk.00000.silo\"
:param coordinates Coordinates corresponding to the files so for example [ [[0,0,0], [0,1,0]], [[2,1,2], [2,1,4]] ]
:param outputDirectory Path to output directory
:param outputFileName Name of the output file
:param colorTable="hot_desaturated" Color table for the plots
'''
# OPTIONS
#################################################################
# Input variable
_variableName = variableName
minVariableValue = minValue
maxVariableValue = maxValue
colorTableName = colorTable
# Input directory and file names
_outputDir = outputDirectory
_outputFileName = outputFileName # The file names for the png files.
databaseName = "localhost:" + inputDirectory + inputFileName # For navigating to the silo files
# Note: a slice of the plot in z-axis is taken automatically
#################################################################
inputFileName2 = "point.vtk"
databaseName2 = "localhost:" + os.getcwd() + "/" + inputFileName2
currentPlot = 0
vis.OpenDatabase(databaseName, 0)
#vis.ActiveDatabase("localhost:" + inputDirectory + inputFileName)
#Load settings
visSettings.load_visit_settings()
vis.AddPlot("Pseudocolor", _variableName, 1, 1) #CONTINUE
vis.SetActivePlots(currentPlot)
vis.PseudocolorAtts = vis.PseudocolorAttributes()
vis.PseudocolorAtts.legendFlag = 1
vis.PseudocolorAtts.lightingFlag = 1
vis.PseudocolorAtts.minFlag = 1
vis.PseudocolorAtts.maxFlag = 1
vis.PseudocolorAtts.centering = vis.PseudocolorAtts.Natural # Natural, Nodal, Zonal
vis.PseudocolorAtts.scaling = vis.PseudocolorAtts.Linear # Linear, Log, Skew
vis.PseudocolorAtts.limitsMode = vis.PseudocolorAtts.CurrentPlot # OriginalData, CurrentPlot
vis.PseudocolorAtts.min = minVariableValue
vis.PseudocolorAtts.max = maxVariableValue
vis.PseudocolorAtts.pointSize = 0.05
vis.PseudocolorAtts.pointType = vis.PseudocolorAtts.Point # Box, Axis, Icosahedron, Point, Sphere
vis.PseudocolorAtts.skewFactor = 1
vis.PseudocolorAtts.opacity = 1
vis.PseudocolorAtts.colorTableName = colorTableName
vis.PseudocolorAtts.invertColorTable = 0
vis.PseudocolorAtts.smoothingLevel = 0
vis.PseudocolorAtts.pointSizeVarEnabled = 0
vis.PseudocolorAtts.pointSizeVar = "default"
vis.PseudocolorAtts.pointSizePixels = 2
vis.PseudocolorAtts.lineStyle = vis.PseudocolorAtts.SOLID # SOLID, DASH, DOT, DOTDASH
vis.PseudocolorAtts.lineWidth = 0
vis.PseudocolorAtts.opacityType = vis.PseudocolorAtts.Explicit # Explicit, ColorTable
vis.SetPlotOptions(vis.PseudocolorAtts)
vis.AddOperator("Slice", 1)
vis.SliceAtts = vis.SliceAttributes()
vis.SliceAtts.originType = vis.SliceAtts.Intercept # Point, Intercept, Percent, Zone, Node
vis.SliceAtts.originPoint = (0, 0, 0)
vis.SliceAtts.originIntercept = 0
vis.SliceAtts.originPercent = 0
vis.SliceAtts.originZone = 0
vis.SliceAtts.originNode = 0
vis.SliceAtts.normal = (0, 0, 1)
vis.SliceAtts.axisType = vis.SliceAtts.ZAxis # XAxis, YAxis, ZAxis, Arbitrary, ThetaPhi
vis.SliceAtts.upAxis = (0, 1, 0)
vis.SliceAtts.project2d = 1
vis.SliceAtts.interactive = 1
vis.SliceAtts.flip = 0
vis.SliceAtts.originZoneDomain = 0
vis.SliceAtts.originNodeDomain = 0
vis.SliceAtts.meshName = "SpatialGrid"
vis.SliceAtts.theta = 0
vis.SliceAtts.phi = 90
vis.SetOperatorOptions(vis.SliceAtts, 1)
vis.DrawPlots()
create_point_vtk(fileName=inputFileName2, coordinates=coordinate)
vis.OpenDatabase(databaseName2, 0)
currentPlot = currentPlot + 1
vis.SetActivePlots(currentPlot)
vis.AddPlot("Mesh", "mesh", 1, 1)
vis.MeshAtts = vis.MeshAttributes()
vis.MeshAtts.legendFlag = 1
vis.MeshAtts.lineStyle = vis.MeshAtts.SOLID # SOLID, DASH, DOT, DOTDASH
vis.MeshAtts.lineWidth = 0
vis.MeshAtts.meshColor = (0, 0, 0, 255)
vis.MeshAtts.outlineOnlyFlag = 0
vis.MeshAtts.errorTolerance = 0.01
vis.MeshAtts.meshColorSource = vis.MeshAtts.Foreground # Foreground, MeshCustom
vis.MeshAtts.opaqueColorSource = vis.MeshAtts.Background # Background, OpaqueCustom
vis.MeshAtts.opaqueMode = vis.MeshAtts.Auto # Auto, On, Off
vis.MeshAtts.pointSize = 0.05
vis.MeshAtts.opaqueColor = (255, 255, 255, 255)
vis.MeshAtts.smoothingLevel = vis.MeshAtts.None # None, Fast, High
vis.MeshAtts.pointSizeVarEnabled = 0
vis.MeshAtts.pointSizeVar = "default"
vis.MeshAtts.pointType = vis.MeshAtts.Point # Box, Axis, Icosahedron, Point, Sphere
vis.MeshAtts.showInternal = 0
vis.MeshAtts.pointSizePixels = 25
vis.MeshAtts.opacity = 1
vis.SetPlotOptions(vis.MeshAtts)
vis.DrawPlots()
vis.SaveWindowAtts = vis.SaveWindowAttributes()
vis.SaveWindowAtts.outputToCurrentDirectory = 0
vis.SaveWindowAtts.outputDirectory = _outputDir
vis.SaveWindowAtts.fileName = _outputFileName
vis.SaveWindowAtts.family = 1
vis.SaveWindowAtts.format = vis.SaveWindowAtts.PNG # BMP, CURVE, JPEG, OBJ, PNG, POSTSCRIPT, POVRAY, PPM, RGB, STL, TIFF, ULTRA, VTK, PLY
vis.SaveWindowAtts.width = 3000
vis.SaveWindowAtts.height = 3000
vis.SaveWindowAtts.screenCapture = 0
vis.SaveWindowAtts.saveTiled = 0
vis.SaveWindowAtts.quality = 100
vis.SaveWindowAtts.progressive = 0
vis.SaveWindowAtts.binary = 0
vis.SaveWindowAtts.stereo = 0
vis.SaveWindowAtts.compression = vis.SaveWindowAtts.PackBits # None, PackBits, Jpeg, Deflate
vis.SaveWindowAtts.forceMerge = 0
vis.SaveWindowAtts.resConstraint = vis.SaveWindowAtts.ScreenProportions # NoConstraint, EqualWidthHeight, ScreenProportions
vis.SaveWindowAtts.advancedMultiWindowSave = 0
vis.SetSaveWindowAttributes(vis.SaveWindowAtts)
vis.SaveWindow()
vis.SetActivePlots((0, 1))
vis.DeleteActivePlots()
vis.CloseDatabase(databaseName2)
vis.CloseDatabase(databaseName)
def create_visit_point_movie(variableName,
minValue,
maxValue,
inputDirectory,
inputFileNames,
coordinates,
outputDirectory,
outputFileName,
colorTable="hot_desaturated"):
'''
Function for making a movie
Arguments:
:param variableName Name of the variable
:param minValue Minimum value of the variable
:param maxValue Maximum value of the variable
:param inputDirectory Path to input vlsv/silo files
:param inputFileNames Name of the files for example [\"bulk.00000.silo\", \"bulk.00001.silo\"]
:param coordinates Coordinates corresponding to the files so for example [ [[0,0,0], [0,1,0]], [[2,1,2], [2,1,4]] ]
:param outputDirectory Path to output directory
:param outputFileName Name of the output file
:param colorTable="hot_desaturated" Color table for the plots
'''
coordinates = [coordinates]
for i in range(len(inputFileNames)):
# OPTIONS
#################################################################
# Input variable
_variableName = variableName
minVariableValue = minValue
maxVariableValue = maxValue
colorTableName = colorTable
# Input directory and file names
#_outputDir = "/home/hannukse/MOVINGFRAME_MOVIES/AAJ_BZ_REMAKE/" # Set the output directory (Where .png s are saved)
_outputDir = outputDirectory
#_outputFileName = "BZ_FORESHOCK_2_" # The file names for the png files. These for ex. will be saved visit0000.png, visit0001.png, ..
_outputFileName = outputFileName # The file names for the png files.
#databaseName = "localhost:/home/hannukse/meteo/stornext/field/vlasiator/2D/AAJ/silo_files/bulk.*.silo database" # For navigating to the silo files
inputFileName = inputFileNames[i]
databaseName = "localhost:" + inputDirectory + inputFileName # For navigating to the silo files
# Note: a slice of the plot in z-axis is taken automatically
#################################################################
# LaunchNowin(vdir=visitBinDirectory)
#dx = speedX * frameInSeconds # Note: This is in meters per frame!
#dy = speedY * frameInSeconds # Note: This is in meters per frame!
#LaunchNowin(vdir="/usr/local/visit/bin")
#Set up window and annotations
#vis.LaunchNowin(vdir="/usr/local/visit/bin")
inputFileName2 = "point.vtk"
databaseName2 = "localhost:" + os.getcwd() + "/" + inputFileName2
vis.OpenDatabase(databaseName, 0)
#vis.ActiveDatabase("localhost:" + inputDirectory + inputFileName)
#Load settings
visSettings.load_visit_settings()
vis.AddPlot("Pseudocolor", _variableName, 1, 1) #CONTINUE
vis.SetActivePlots(1)
vis.PseudocolorAtts = vis.PseudocolorAttributes()
vis.PseudocolorAtts.legendFlag = 1
vis.PseudocolorAtts.lightingFlag = 1
vis.PseudocolorAtts.minFlag = 1
vis.PseudocolorAtts.maxFlag = 1
vis.PseudocolorAtts.centering = vis.PseudocolorAtts.Natural # Natural, Nodal, Zonal
vis.PseudocolorAtts.scaling = vis.PseudocolorAtts.Linear # Linear, Log, Skew
vis.PseudocolorAtts.limitsMode = vis.PseudocolorAtts.CurrentPlot # OriginalData, CurrentPlot
vis.PseudocolorAtts.min = minVariableValue
vis.PseudocolorAtts.max = maxVariableValue
vis.PseudocolorAtts.pointSize = 0.05
vis.PseudocolorAtts.pointType = vis.PseudocolorAtts.Point # Box, Axis, Icosahedron, Point, Sphere
vis.PseudocolorAtts.skewFactor = 1
vis.PseudocolorAtts.opacity = 1
vis.PseudocolorAtts.colorTableName = colorTableName
vis.PseudocolorAtts.invertColorTable = 0
vis.PseudocolorAtts.smoothingLevel = 0
vis.PseudocolorAtts.pointSizeVarEnabled = 0
vis.PseudocolorAtts.pointSizeVar = "default"
vis.PseudocolorAtts.pointSizePixels = 2
vis.PseudocolorAtts.lineStyle = vis.PseudocolorAtts.SOLID # SOLID, DASH, DOT, DOTDASH
vis.PseudocolorAtts.lineWidth = 0
vis.PseudocolorAtts.opacityType = vis.PseudocolorAtts.Explicit # Explicit, ColorTable
vis.SetPlotOptions(vis.PseudocolorAtts)
vis.SetActivePlots(1)
vis.AddOperator("Slice", 1)
vis.SetActivePlots(1)
vis.SliceAtts = vis.SliceAttributes()
vis.SliceAtts.originType = vis.SliceAtts.Intercept # Point, Intercept, Percent, Zone, Node
vis.SliceAtts.originPoint = (0, 0, 0)
vis.SliceAtts.originIntercept = 0
vis.SliceAtts.originPercent = 0
vis.SliceAtts.originZone = 0
vis.SliceAtts.originNode = 0
vis.SliceAtts.normal = (0, 0, 1)
vis.SliceAtts.axisType = vis.SliceAtts.ZAxis # XAxis, YAxis, ZAxis, Arbitrary, ThetaPhi
vis.SliceAtts.upAxis = (0, 1, 0)
vis.SliceAtts.project2d = 1
vis.SliceAtts.interactive = 1
vis.SliceAtts.flip = 0
vis.SliceAtts.originZoneDomain = 0
vis.SliceAtts.originNodeDomain = 0
vis.SliceAtts.meshName = "SpatialGrid"
vis.SliceAtts.theta = 0
vis.SliceAtts.phi = 90
vis.SetOperatorOptions(vis.SliceAtts, 1)
vis.DrawPlots()
vis.SetActivePlots(0)
for coordinate in coordinates[i]:
print(str(coordinate))
create_point_vtk(fileName=inputFileName2, coordinates=coordinate)
vis.OpenDatabase(databaseName2, 0)
vis.AddPlot("Mesh", "mesh", 1, 1)
vis.SetActivePlots(vis.GetNumPlots())
vis.MeshAtts = vis.MeshAttributes()
vis.MeshAtts.legendFlag = 1
vis.MeshAtts.lineStyle = vis.MeshAtts.SOLID # SOLID, DASH, DOT, DOTDASH
vis.MeshAtts.lineWidth = 0
vis.MeshAtts.meshColor = (0, 0, 0, 255)
vis.MeshAtts.outlineOnlyFlag = 0
vis.MeshAtts.errorTolerance = 0.01
vis.MeshAtts.meshColorSource = vis.MeshAtts.Foreground # Foreground, MeshCustom
vis.MeshAtts.opaqueColorSource = vis.MeshAtts.Background # Background, OpaqueCustom
vis.MeshAtts.opaqueMode = vis.MeshAtts.Auto # Auto, On, Off
vis.MeshAtts.pointSize = 0.05
vis.MeshAtts.opaqueColor = (255, 255, 255, 255)
vis.MeshAtts.smoothingLevel = vis.MeshAtts.None # None, Fast, High
vis.MeshAtts.pointSizeVarEnabled = 0
vis.MeshAtts.pointSizeVar = "default"
vis.MeshAtts.pointType = vis.MeshAtts.Point # Box, Axis, Icosahedron, Point, Sphere
vis.MeshAtts.showInternal = 0
vis.MeshAtts.pointSizePixels = 10
vis.MeshAtts.opacity = 1
vis.SetPlotOptions(vis.MeshAtts)
vis.DrawPlots()
# Iterate through frames
vis.SaveWindowAtts = vis.SaveWindowAttributes()
vis.SaveWindowAtts.outputToCurrentDirectory = 0
vis.SaveWindowAtts.outputDirectory = _outputDir
vis.SaveWindowAtts.fileName = _outputFileName
vis.SaveWindowAtts.family = 1
vis.SaveWindowAtts.format = vis.SaveWindowAtts.PNG # BMP, CURVE, JPEG, OBJ, PNG, POSTSCRIPT, POVRAY, PPM, RGB, STL, TIFF, ULTRA, VTK, PLY
vis.SaveWindowAtts.width = 3000
vis.SaveWindowAtts.height = 3000
vis.SaveWindowAtts.screenCapture = 0
vis.SaveWindowAtts.saveTiled = 0
vis.SaveWindowAtts.quality = 100
vis.SaveWindowAtts.progressive = 0
vis.SaveWindowAtts.binary = 0
vis.SaveWindowAtts.stereo = 0
vis.SaveWindowAtts.compression = vis.SaveWindowAtts.PackBits # None, PackBits, Jpeg, Deflate
vis.SaveWindowAtts.forceMerge = 0
vis.SaveWindowAtts.resConstraint = vis.SaveWindowAtts.ScreenProportions # NoConstraint, EqualWidthHeight, ScreenProportions
vis.SaveWindowAtts.advancedMultiWindowSave = 0
vis.SetSaveWindowAttributes(vis.SaveWindowAtts)
vis.SaveWindow()
vis.DeleteActivePlots()
vis.DeleteActivePlots()
vis.CloseDatabase(databaseName)
vis.CloseDatabase(databaseName2)
# Make the movie:
#subprocess.call("./moviecompilescript.sh " + _outputDir + " " + _outputFileName)
pyVisitPath = "pyVisit/"
#subprocess.call(pythonLibDirectoryPath + pyVisitPath + "moviecompilescript.sh")
#subprocess.call(pythonLibDirectoryPath + pyVisitPath + "moviecompilescript.sh " + _outputDir + " " + _outputFileName)
framerate = "10"
subprocess.call([
pythonLibDirectoryPath + pyVisitPath + "moviecompilescript.sh",
_outputDir, _outputFileName, framerate
])
# Delete the point vtk file:
os.remove(os.getcwd() + "/" + inputFileName2)
def make_movie_auto(variableName,
boundaryBox,
vlsvFileName,
inputDirectory,
inputFileName,
outputDirectory,
outputFileName,
colorTableName="hot_desaturated",
startFrame=-1,
endFrame=-1,
thresholdCoefficient=0.6):
'''
Function for making a movie
Arguments:
:param variableName Name of the variable
:param boundaryBox Box for collecting min and max threshold (The movie will focus on that area)
:param vlsvFileName Name of a vlsv file where the function collects the threshold for the boundary box
:param inputDirectory Path to input vlsv/silo files
:param inputFileName Name of the file(s) so for example if the filenames are bulk.0000.silo, bulk.0001.silo, .. then inputFileName=\"bulk.*.silo\""
:param outputDirectory Path to output directory
:param outputFileName Name of the output file
:param colorTableName="hot_desaturated" Color table for the plots
:param thresholdCoefficient Sets the coefficient for a covariant collected from the values from boundary box. The lower this is, the more focused the movie will be on the boundary box area
:param startFrame=-1 Starting frame of the movie (-1 equals 0)
:param endFrame=-1 Starting frame of the movie (-1 equals last frame)
'''
if thresholdCoefficient < 0:
print("thresholdCoefficient must be non-negative!")
return
# OPTIONS
#################################################################
# Input frame properties
_startFrame = startFrame # Note: if _startFrame is set to -1 the start frame gets set to 0
_endFrame = endFrame # Note: if _endFrame is set to -1 the _endFrame is automatically the number of frames in the database
# Input variable
_variableName = variableName
# Input directory and file names
#_outputDir = "/home/hannukse/MOVINGFRAME_MOVIES/AAJ_BZ_REMAKE/" # Set the output directory (Where .png s are saved)
_outputDir = outputDirectory
#_outputFileName = "BZ_FORESHOCK_2_" # The file names for the png files. These for ex. will be saved visit0000.png, visit0001.png, ..
_outputFileName = outputFileName # The file names for the png files.
#databaseName = "localhost:/home/hannukse/meteo/stornext/field/vlasiator/2D/AAJ/silo_files/bulk.*.silo database" # For navigating to the silo files
databaseName = "localhost:" + inputDirectory + inputFileName + " database" # For navigating to the silo files
# Note: a slice of the plot in z-axis is taken automatically
#################################################################
# Get the min and max values:
# Get all cell ids within the boundary box:
vlsvReader = VlsvReader(vlsvFileName)
# Get global boundaries:
# Get xmax, xmin and xcells_ini
xmax = vlsvReader.read_parameter(name="xmax")
xmin = vlsvReader.read_parameter(name="xmin")
xcells = vlsvReader.read_parameter(name="xcells_ini")
# Do the same for y
ymax = vlsvReader.read_parameter(name="ymax")
ymin = vlsvReader.read_parameter(name="ymin")
ycells = vlsvReader.read_parameter(name="ycells_ini")
# And for z
zmax = vlsvReader.read_parameter(name="zmax")
zmin = vlsvReader.read_parameter(name="zmin")
zcells = vlsvReader.read_parameter(name="zcells_ini")
#Calculate cell lengths:
cell_lengths = np.array([(xmax - xmin) / (float)(xcells),
(ymax - ymin) / (float)(ycells),
(zmax - zmin) / (float)(zcells)])
# Get cell indices:
cell_indice_bounds = np.array([
(int)(((float)(boundaryBox[0]) - xmin) / (float)(cell_lengths[0])),
(int)(((float)(boundaryBox[1]) - xmin) / (float)(cell_lengths[0])),
(int)(((float)(boundaryBox[2]) - ymin) / (float)(cell_lengths[1])),
(int)(((float)(boundaryBox[3]) - ymin) / (float)(cell_lengths[1])),
(int)(((float)(boundaryBox[4]) - zmin) / (float)(cell_lengths[2])),
(int)(((float)(boundaryBox[5]) - zmin) / (float)(cell_lengths[2]))
])
# Get every cell id within the boundary box:
cellids = []
cell_indice = np.array(
[cell_indice_bounds[0], cell_indice_bounds[2], cell_indice_bounds[4]])
while True:
cellids.append(cell_indice[0] + cell_indice[1] * xcells +
cell_indice[2] * xcells * ycells + 1)
if cell_indice[0] < cell_indice_bounds[1]:
cell_indice[0] = cell_indice[0] + 1
elif cell_indice[1] < cell_indice_bounds[3]:
cell_indice[1] = cell_indice[1] + 1
cell_indice[0] = cell_indice_bounds[0]
elif cell_indice[2] < cell_indice_bounds[5]:
cell_indice[2] = cell_indice[2] + 1
cell_indice[1] = cell_indice_bounds[1]
cell_indice[0] = cell_indice_bounds[0]
else:
# Indice out of bounds -- got all cell ids
break
# Convert cell ids into set:
cellids = Set(cellids)
cellidlocations = []
# Get all of the cell ids locations:
allcellids = vlsvReader.read(name="SpatialGrid", tag="MESH")
for i in range(len(allcellids)):
if allcellids[i] in cellids:
#This cell id is within the user-given boundary
cellidlocations.append(allcellids[i])
# Get all of the values:
allvalues = vlsvReader.read_variables(name=_variableName)
values = []
# Get the values of the cell ids within the boundary
for i in cellidlocations:
values.append(allvalues[i])
# We now have all the cell ids (and their locations in the arrays) from the area, set min and max thresholds:
meanValue = np.mean(values)
standardDeviationValue = np.std(values)
maxValue = meanValue + (
float)(thresholdCoefficient) * standardDeviationValue
minValue = meanValue - (
float)(thresholdCoefficient) * standardDeviationValue
# Put threshold values:
minVariableValue = minValue
maxVariableValue = maxValue
# LaunchNowin(vdir=visitBinDirectory)
#dx = speedX * frameInSeconds # Note: This is in meters per frame!
#dy = speedY * frameInSeconds # Note: This is in meters per frame!
#LaunchNowin(vdir="/usr/local/visit/bin")
#Set up window and annotations
#vis.LaunchNowin(vdir="/usr/local/visit/bin")
vis.OpenDatabase(databaseName, 0)
#Load settings
visSettings.load_visit_settings()
vis.AddPlot("Pseudocolor", _variableName, 1, 1) #CONTINUE
vis.SetActivePlots(0)
vis.PseudocolorAtts = vis.PseudocolorAttributes()
vis.PseudocolorAtts.legendFlag = 1
vis.PseudocolorAtts.lightingFlag = 1
vis.PseudocolorAtts.minFlag = 1
vis.PseudocolorAtts.maxFlag = 1
vis.PseudocolorAtts.centering = vis.PseudocolorAtts.Natural # Natural, Nodal, Zonal
vis.PseudocolorAtts.scaling = vis.PseudocolorAtts.Linear # Linear, Log, Skew
vis.PseudocolorAtts.limitsMode = vis.PseudocolorAtts.CurrentPlot # OriginalData, CurrentPlot
vis.PseudocolorAtts.min = minVariableValue
vis.PseudocolorAtts.max = maxVariableValue
vis.PseudocolorAtts.pointSize = 0.05
vis.PseudocolorAtts.pointType = vis.PseudocolorAtts.Point # Box, Axis, Icosahedron, Point, Sphere
vis.PseudocolorAtts.skewFactor = 1
vis.PseudocolorAtts.opacity = 1
vis.PseudocolorAtts.colorTableName = colorTableName
vis.PseudocolorAtts.invertColorTable = 0
vis.PseudocolorAtts.smoothingLevel = 0
vis.PseudocolorAtts.pointSizeVarEnabled = 0
vis.PseudocolorAtts.pointSizeVar = "default"
vis.PseudocolorAtts.pointSizePixels = 2
vis.PseudocolorAtts.lineStyle = vis.PseudocolorAtts.SOLID # SOLID, DASH, DOT, DOTDASH
vis.PseudocolorAtts.lineWidth = 0
vis.PseudocolorAtts.opacityType = vis.PseudocolorAtts.Explicit # Explicit, ColorTable
vis.SetPlotOptions(vis.PseudocolorAtts)
vis.SetActivePlots(0)
vis.AddOperator("Slice", 1)
vis.SetActivePlots(0)
vis.SliceAtts = vis.SliceAttributes()
vis.SliceAtts.originType = vis.SliceAtts.Intercept # Point, Intercept, Percent, Zone, Node
vis.SliceAtts.originPoint = (0, 0, 0)
vis.SliceAtts.originIntercept = 0
vis.SliceAtts.originPercent = 0
vis.SliceAtts.originZone = 0
vis.SliceAtts.originNode = 0
vis.SliceAtts.normal = (0, 0, 1)
vis.SliceAtts.axisType = vis.SliceAtts.ZAxis # XAxis, YAxis, ZAxis, Arbitrary, ThetaPhi
vis.SliceAtts.upAxis = (0, 1, 0)
vis.SliceAtts.project2d = 1
vis.SliceAtts.interactive = 1
vis.SliceAtts.flip = 0
vis.SliceAtts.originZoneDomain = 0
vis.SliceAtts.originNodeDomain = 0
vis.SliceAtts.meshName = "SpatialGrid"
vis.SliceAtts.theta = 0
vis.SliceAtts.phi = 90
vis.SetOperatorOptions(vis.SliceAtts, 1)
vis.DrawPlots()
if _endFrame == -1:
_endFrame = vis.TimeSliderGetNStates() - 1
if _startFrame == -1:
_startFrame = 0
# Iterate through frames
for i in range(_startFrame, _endFrame + 1):
vis.SetTimeSliderState(i)
frame = i - _startFrame
vis.SaveWindowAtts = vis.SaveWindowAttributes()
vis.SaveWindowAtts.outputToCurrentDirectory = 0
vis.SaveWindowAtts.outputDirectory = _outputDir
vis.SaveWindowAtts.fileName = _outputFileName
vis.SaveWindowAtts.family = 1
vis.SaveWindowAtts.format = vis.SaveWindowAtts.PNG # BMP, CURVE, JPEG, OBJ, PNG, POSTSCRIPT, POVRAY, PPM, RGB, STL, TIFF, ULTRA, VTK, PLY
vis.SaveWindowAtts.width = 3000
vis.SaveWindowAtts.height = 300
vis.SaveWindowAtts.screenCapture = 0
vis.SaveWindowAtts.saveTiled = 0
vis.SaveWindowAtts.quality = 100
vis.SaveWindowAtts.progressive = 0
vis.SaveWindowAtts.binary = 0
vis.SaveWindowAtts.stereo = 0
vis.SaveWindowAtts.compression = vis.SaveWindowAtts.PackBits # None, PackBits, Jpeg, Deflate
vis.SaveWindowAtts.forceMerge = 0
vis.SaveWindowAtts.resConstraint = vis.SaveWindowAtts.ScreenProportions # NoConstraint, EqualWidthHeight, ScreenProportions
vis.SaveWindowAtts.advancedMultiWindowSave = 0
vis.SetSaveWindowAttributes(vis.SaveWindowAtts)
vis.SaveWindow()
vis.DeleteActivePlots()
vis.CloseDatabase(databaseName)
# Make the movie:
#subprocess.call("./moviecompilescript.sh " + _outputDir + " " + _outputFileName)
pyVisitPath = "pyVisit/"
#subprocess.call(pythonLibDirectoryPath + pyVisitPath + "moviecompilescript.sh")
#subprocess.call(pythonLibDirectoryPath + pyVisitPath + "moviecompilescript.sh " + _outputDir + " " + _outputFileName)
frameRate = "10"
subprocess.call([
pythonLibDirectoryPath + pyVisitPath + "moviecompilescript.sh",
_outputDir, _outputFileName, frameRate
])
def visit_plot_pseudocolor_2d(xdmf_path,
name,
value_range=(-5.0, 5.0),
curve2d_paths=None,
config_view=None,
out_dir=os.getcwd(),
out_prefix='wake2d_',
figsize=(1024, 1024),
visit_dir=None,
visit_arch='linux-x86_64',
state=None,
states=None,
states_range=[0, None, 1]):
# Import VisIt package.
if visit_dir is None:
visit_dir = os.environ.get('VISIT_DIR')
if visit_dir is None:
raise ValueError('Provide VisIt installation path or '
'set env variable VISIT_DIR')
sys.path.append(os.path.join(visit_dir, visit_arch, 'lib',
'site-packages'))
import visit
visit.LaunchNowin()
# Check version of VisIt.
visit_check_version(visit.Version())
# Create database correlation with optional Curve2D files.
num_bodies = 0
databases = [str(xdmf_path)]
if curve2d_paths is not None:
num_bodies = len(curve2d_paths)
databases = [str(path) for path in curve2d_paths]
databases.append(str(xdmf_path))
visit.CreateDatabaseCorrelation('common', databases[num_bodies:], 0)
# Open the file with the coordinates of the immersed boundary.
if num_bodies > 0:
for i in range(num_bodies):
visit.OpenDatabase(databases[i], 0)
# Add plot the mesh points.
visit.AddPlot('Curve', 'curve', 1, 1)
# Set attributes of the curve.
CurveAtts = visit.CurveAttributes()
CurveAtts.lineWidth = 1
CurveAtts.curveColorSource = CurveAtts.Custom
CurveAtts.curveColor = (0, 0, 0, 255)
CurveAtts.showLegend = 0
CurveAtts.showLabels = 0
visit.SetPlotOptions(CurveAtts)
# Open the XMF file for the spanwise-averaged z-component of the vorticity.
visit.OpenDatabase(databases[-1], 0)
# Add a pseudocolor plot of the scalar field.
visit.AddPlot('Pseudocolor', name, 1, 1)
# Set attributes of the pseudocolor.
PseudocolorAtts = visit.PseudocolorAttributes()
PseudocolorAtts.minFlag = 1
PseudocolorAtts.min = value_range[0]
PseudocolorAtts.maxFlag = 1
PseudocolorAtts.max = value_range[1]
PseudocolorAtts.colorTableName = 'viridis'
visit.SetPlotOptions(PseudocolorAtts)
# Parse the 2D view configuration file.
if config_view is not None:
with open(str(config_view), 'r') as infile:
config_view = yaml.load(infile, Loader=yaml.FullLoader)
config_view = config_view['View2DAtts']
# Set attributes of the view.
View2DAtts = visit.View2DAttributes()
for key, value in config_view.items():
if type(value) is list:
value = tuple(value)
setattr(View2DAtts, key, value)
visit.SetView2D(View2DAtts)
# Remove time and user info.
AnnotationAtts = visit.AnnotationAttributes()
AnnotationAtts.userInfoFlag = 0
AnnotationAtts.timeInfoFlag = 1
visit.SetAnnotationAttributes(AnnotationAtts)
visit.SetActiveWindow(1)
visit.Source(os.path.join(visit_dir, visit_arch, 'bin', 'makemovie.py'))
visit.ToggleCameraViewMode()
# Create output directory if necessary.
if not os.path.isdir(str(out_dir)):
os.makedirs(str(out_dir))
# Loop over the states to render and save the plots.
if state is not None:
states = [state]
elif states is None:
if states_range[1] is None:
states_range[1] = visit.TimeSliderGetNStates()
else:
states_range[1] += 1
states = range(*states_range)
for i, state in enumerate(states):
print('[state {}] Rendering and saving figure ...'.format(state))
visit.SetTimeSliderState(state)
if i == 0:
visit.DrawPlots()
RenderingAtts = visit.RenderingAttributes()
visit.SetRenderingAttributes(RenderingAtts)
SaveWindowAtts = visit.SaveWindowAttributes()
SaveWindowAtts.outputToCurrentDirectory = 0
SaveWindowAtts.outputDirectory = str(out_dir)
SaveWindowAtts.fileName = '{}{:0>4}'.format(out_prefix, state)
SaveWindowAtts.family = 0
SaveWindowAtts.format = SaveWindowAtts.PNG
SaveWindowAtts.width = figsize[0]
SaveWindowAtts.height = figsize[1]
SaveWindowAtts.quality = 100
SaveWindowAtts.resConstraint = SaveWindowAtts.NoConstraint
visit.SetSaveWindowAttributes(SaveWindowAtts)
visit.SaveWindow()
os.remove('visitlog.py')
visit.Close()
return
def visit_plot_qcrit_wx_3d(xdmf_dir,
wx_range=(-5.0, 5.0),
q_value=0.1,
config_view=None,
out_dir=os.getcwd(),
out_prefix='qcrit_wx_',
figsize=(1024, 1024),
state=None,
states=None,
states_range=[0, None, 1]):
visit_initialize()
# Define some variables to get the q_crit and wx_cc.
p_xdmf_path = os.path.join(str(xdmf_dir), 'p.xmf')
visit.OpenDatabase(p_xdmf_path, 0)
# Define cell-centered velocity vector field.
ux_xdmf_path = os.path.join(str(xdmf_dir), 'u.xmf')
uy_xdmf_path = os.path.join(str(xdmf_dir), 'v.xmf')
uz_xdmf_path = os.path.join(str(xdmf_dir), 'w.xmf')
vel_expr = ('{' +
'pos_cmfe(<{}[0]id:u>, <p Grid>, 1.0),'.format(ux_xdmf_path) +
'pos_cmfe(<{}[0]id:v>, <p Grid>, 0.0),'.format(uy_xdmf_path) +
'pos_cmfe(<{}[0]id:w>, <p Grid>, 0.0)'.format(uz_xdmf_path) +
'}')
visit.DefineVectorExpression('velocity', vel_expr)
# Define Q-criterion.
qcrit_expr = ('q_criterion(' + 'gradient(velocity[0]),' +
'gradient(velocity[1]),' + 'gradient(velocity[2])' + ')')
visit.DefineScalarExpression('q_crit', qcrit_expr)
# Define cell-centered streamwise vorticity.
wx_xdmf_path = os.path.join(str(xdmf_dir), 'wx.xmf')
wx_exp = 'pos_cmfe(<{}[0]id:wx>, <p Grid>, 0.0)'.format(wx_xdmf_path)
visit.DefineScalarExpression('wx_cc', wx_exp)
# Add a pseudocolor of the cell-centered streamwise vorticity.
visit.AddPlot('Pseudocolor', 'wx_cc', 1, 1)
PseudocolorAtts = visit.PseudocolorAttributes()
PseudocolorAtts.minFlag = 1
PseudocolorAtts.min = wx_range[0]
PseudocolorAtts.maxFlag = 1
PseudocolorAtts.max = wx_range[1]
PseudocolorAtts.colorTableName = 'viridis'
PseudocolorAtts.invertColorTable = 0
PseudocolorAtts.opacityType = PseudocolorAtts.Constant
PseudocolorAtts.opacity = 0.8
PseudocolorAtts.legendFlag = 0
visit.SetPlotOptions(PseudocolorAtts)
# Add an isosurface of the Q-criterion.
visit.AddOperator('Isosurface', 1)
IsosurfaceAtts = visit.IsosurfaceAttributes()
IsosurfaceAtts.variable = 'q_crit'
IsosurfaceAtts.contourMethod = IsosurfaceAtts.Value
IsosurfaceAtts.contourValue = (q_value)
IsosurfaceAtts.scaling = IsosurfaceAtts.Linear
visit.SetOperatorOptions(IsosurfaceAtts, 1)
# Remove info about user, time, database, and legend.
AnnotationAtts = visit.AnnotationAttributes()
AnnotationAtts.userInfoFlag = 0
AnnotationAtts.databaseInfoFlag = 0
AnnotationAtts.timeInfoFlag = 0
AnnotationAtts.legendInfoFlag = 0
AnnotationAtts.axes3D.visible = 0
AnnotationAtts.axes3D.triadFlag = 1
AnnotationAtts.axes3D.bboxFlag = 0
visit.SetAnnotationAttributes(AnnotationAtts)
# Parse the 3D view configuration file.
if config_view is not None:
View3DAtts = visit_get_view(config_view, '3D')
visit.SetView3D(View3DAtts)
visit.SetActiveWindow(1)
states = visit_get_states(state=state,
states=states,
states_range=states_range)
visit_render_save_states(states,
out_dir=out_dir,
out_prefix=out_prefix,
figsize=figsize)
visit_finalize()
return
def visit_plot_pseudocolor_2d(xdmf_path,
name,
value_range=(-5.0, 5.0),
curve2d_paths=None,
config_view=None,
out_dir=os.getcwd(),
out_prefix=None,
figsize=(1024, 1024),
state=None,
states=None,
states_range=[0, None, 1]):
visit_initialize()
# Create database correlation with optional Curve2D files.
num_bodies = 0
databases = [str(xdmf_path)]
if curve2d_paths is not None:
num_bodies = len(curve2d_paths)
databases = [str(path) for path in curve2d_paths]
databases.append(str(xdmf_path))
visit.CreateDatabaseCorrelation('common', databases[num_bodies:], 0)
# Open the file with the coordinates of the immersed boundary.
if num_bodies > 0:
for i in range(num_bodies):
visit.OpenDatabase(databases[i], 0)
# Add plot the mesh points.
visit.AddPlot('Curve', 'curve', 1, 1)
# Set attributes of the curve.
CurveAtts = visit.CurveAttributes()
CurveAtts.lineWidth = 1
CurveAtts.curveColorSource = CurveAtts.Custom
CurveAtts.curveColor = (0, 0, 0, 255)
CurveAtts.showLegend = 0
CurveAtts.showLabels = 0
visit.SetPlotOptions(CurveAtts)
# Open the XMF file for the spanwise-averaged z-component of the vorticity.
visit.OpenDatabase(databases[-1], 0)
# Add a pseudocolor plot of the scalar field.
visit.AddPlot('Pseudocolor', name, 1, 1)
# Set attributes of the pseudocolor.
PseudocolorAtts = visit.PseudocolorAttributes()
PseudocolorAtts.minFlag = 1
PseudocolorAtts.min = value_range[0]
PseudocolorAtts.maxFlag = 1
PseudocolorAtts.max = value_range[1]
PseudocolorAtts.colorTableName = 'viridis'
visit.SetPlotOptions(PseudocolorAtts)
# Parse the 2D view configuration file.
if config_view is not None:
View2DAtts = visit_get_view(config_view, '2D')
visit.SetView2D(View2DAtts)
# Remove time and user info.
AnnotationAtts = visit.AnnotationAttributes()
AnnotationAtts.userInfoFlag = 0
AnnotationAtts.timeInfoFlag = 1
visit.SetAnnotationAttributes(AnnotationAtts)
visit.SetActiveWindow(1)
states = visit_get_states(state=state,
states=states,
states_range=states_range)
if out_prefix is None:
out_prefix = name + '_'
visit_render_save_states(states,
out_dir=out_dir,
out_prefix=out_prefix,
figsize=figsize)
visit_finalize()
return
def visit_plot_qcrit_wx_3d(xdmf_path,
qcrit_vals=(6.0, 1.0),
wx_lims=(-5.0, 5.0),
config_view=None,
out_dir=os.getcwd(),
prefix='qcrit_wx_3d_',
figsize=(1024, 1024),
state=None,
states=None,
states_range=[0, None, 1]):
"""Plot the 3D isosurface of the Q-criterion at 2 values.
The first isosurface is colored by the streamwise vorticity.
The second isosurface is colored with a single color (grey).
Parameters
----------
xdmf_path : str
Path of the XDMF file with information about the Q-criterion and
the streamwise vorticity.
qcrit_vals : tuple, optional
Values of the Q-criterion to display as a tuple of 2 floats;
default is (6.0, 1.0).
wx_lims : tuple, optional
Limits of the color range for the streamwise vorticity
as a tuple of 2 floats; default is (-5.0, 5.0).
config_view : str, optional
Path of the YAML file with the configuration of the view;
default is None (use default VisIt view).
out_dir : str, optional
Output directory in which figures will be saved;
default is the present working directory.
prefix : str, optional
Output filename prefix; default is "qcrit_wx_3d_".
figsize : tuple
Figure width and height (in pixels); default is (1024, 1024).
state : int, optional
Single state index to render;
default is None (i.e., render multiple states).
states : list, optional
List of states to render; default is None (i.e., render all states).
states_range : list, optional
Start, end, and step indices for states to render;
default is [0, None, 1] (i.e., render all states).
"""
visit_initialize()
# Open database from XDMF file.
visit.OpenDatabase(xdmf_path, 0)
# Add a pseudocolor of the streamwise vorticity.
visit.AddPlot('Pseudocolor', 'wx_cc', 1, 0)
PseudocolorAtts = visit.PseudocolorAttributes()
PseudocolorAtts.minFlag, PseudocolorAtts.maxFlag = 1, 1
PseudocolorAtts.min, PseudocolorAtts.max = wx_lims
PseudocolorAtts.colorTableName = 'viridis'
PseudocolorAtts.invertColorTable = 0
visit.SetPlotOptions(PseudocolorAtts)
# Add isosurface of the Q-criterion (colored by streamwise vorticity).
visit.AddOperator('Isosurface', 0)
IsosurfaceAtts = visit.IsosurfaceAttributes()
IsosurfaceAtts.contourMethod = IsosurfaceAtts.Value
IsosurfaceAtts.contourValue = (qcrit_vals[0])
IsosurfaceAtts.variable = 'qcrit'
visit.SetOperatorOptions(IsosurfaceAtts, 0)
# Add single-value contour of the Q-criterion.
visit.AddPlot('Contour', 'qcrit', 1, 0)
ContourAtts = visit.ContourAttributes()
ContourAtts.colorType = ContourAtts.ColorBySingleColor
ContourAtts.legendFlag = 0
ContourAtts.singleColor = (128, 128, 128, 153) # grey
ContourAtts.contourNLevels = 1
ContourAtts.minFlag, ContourAtts.maxFlag = 1, 1
ContourAtts.min, ContourAtts.max = qcrit_vals[1], qcrit_vals[1]
visit.SetPlotOptions(ContourAtts)
# Remove some annotations; keep triad.
AnnotationAtts = visit.AnnotationAttributes()
AnnotationAtts.userInfoFlag = 0
AnnotationAtts.databaseInfoFlag = 0
AnnotationAtts.timeInfoFlag = 0
AnnotationAtts.legendInfoFlag = 0
AnnotationAtts.axes3D.visible = 0
AnnotationAtts.axes3D.triadFlag = 1
AnnotationAtts.axes3D.bboxFlag = 0
visit.SetAnnotationAttributes(AnnotationAtts)
# Define state indices to render.
states = visit_get_states(state=state,
states=states,
states_range=states_range)
# Render states and save figures to files.
visit_render_save_states(states,
config_view=config_view,
out_dir=out_dir,
prefix=prefix,
figsize=figsize)
visit_finalize()
pi = math.pi
kxlo = -pi / dx * float(nx - 1) / nx
kxhi = pi / dx * float(nx + 1) / nx
kylo = -pi / dy * float(ny - 1) / ny
kyhi = pi / dy * float(ny + 1) / ny
# visit: open datafile and add pseudocolor plot of S(k)
visit.OpenDatabase(datafile)
visit.AddPlot("Pseudocolor", "S_k")
visit.HideToolbars()
# colorbar range
pa = visit.PseudocolorAttributes()
pa.minFlag = 1
pa.min = minSk
pa.maxFlag = 1
pa.max = maxSk
visit.SetPlotOptions(pa)
# x- and y-ranges of plot and position of plot
va = visit.View2DAttributes()
va.windowCoords = (kxlo, kxhi, kylo, kyhi)
va.viewportCoords = (0.06, 0.94, 0.11, 0.99)
visit.SetView2D(va)
# legend(colorbox) and user/database info on/off | axes
def visit_plot_qcrit_wx_3d_direct(xdmf_path,
wx_range=(-5.0, 5.0),
q_value=0.1,
config_view=None,
out_dir=os.getcwd(),
out_prefix='qcrit_wx_',
figsize=(1024, 1024),
state=None,
states=None,
states_range=[0, None, 1]):
visit_initialize()
visit.OpenDatabase(str(xdmf_path), 0)
# Add a pseudocolor of the cell-centered streamwise vorticity.
visit.AddPlot('Pseudocolor', 'wx_cc', 1, 1)
PseudocolorAtts = visit.PseudocolorAttributes()
PseudocolorAtts.minFlag = 1
PseudocolorAtts.min = wx_range[0]
PseudocolorAtts.maxFlag = 1
PseudocolorAtts.max = wx_range[1]
PseudocolorAtts.colorTableName = 'viridis'
PseudocolorAtts.invertColorTable = 0
PseudocolorAtts.opacityType = PseudocolorAtts.Constant
PseudocolorAtts.opacity = 0.8
PseudocolorAtts.legendFlag = 0
visit.SetPlotOptions(PseudocolorAtts)
# Add an isosurface of the Q-criterion.
visit.AddOperator('Isosurface', 1)
IsosurfaceAtts = visit.IsosurfaceAttributes()
IsosurfaceAtts.variable = 'qcrit'
IsosurfaceAtts.contourMethod = IsosurfaceAtts.Value
IsosurfaceAtts.contourValue = (q_value)
IsosurfaceAtts.scaling = IsosurfaceAtts.Linear
visit.SetOperatorOptions(IsosurfaceAtts, 1)
# Remove info about user, time, database, and legend.
AnnotationAtts = visit.AnnotationAttributes()
AnnotationAtts.userInfoFlag = 0
AnnotationAtts.databaseInfoFlag = 0
AnnotationAtts.timeInfoFlag = 0
AnnotationAtts.legendInfoFlag = 0
AnnotationAtts.axes3D.visible = 0
AnnotationAtts.axes3D.triadFlag = 1
AnnotationAtts.axes3D.bboxFlag = 0
visit.SetAnnotationAttributes(AnnotationAtts)
# Parse the 3D view configuration file.
if config_view is not None:
View3DAtts = visit_get_view(config_view, '3D')
visit.SetView3D(View3DAtts)
visit.SetActiveWindow(1)
states = visit_get_states(state=state,
states=states,
states_range=states_range)
visit_render_save_states(states,
out_dir=out_dir,
out_prefix=out_prefix,
figsize=figsize)
visit_finalize()
return
def visit_plot_qcrit_wx_3d(xdmf_dir,
wx_range=(-5.0, 5.0),
q_value=0.1,
config_view=None,
out_dir=os.getcwd(),
out_prefix='wake3d_',
figsize=(1024, 1024),
visit_dir=None,
visit_arch='linux-x86_64',
state=None,
states=None,
states_range=[0, None, 1]):
# Import VisIt package.
if visit_dir is None:
visit_dir = os.environ.get('VISIT_DIR')
if visit_dir is None:
raise ValueError('Provide VisIt installation path or '
'set env variable VISIT_DIR')
sys.path.append(os.path.join(visit_dir, visit_arch, 'lib',
'site-packages'))
import visit
visit.LaunchNowin()
# Check version of VisIt.
visit_check_version(visit.Version())
# Define some variables to get the q_crit and wx_cc.
p_xdmf_path = os.path.join(str(xdmf_dir), 'p.xmf')
visit.OpenDatabase(p_xdmf_path, 0)
visit.DefineScalarExpression("operators/ConnectedComponents/p Grid",
"cell_constant(<p Grid>, 0.)")
visit.DefineCurveExpression("operators/DataBinning/1D/p Grid",
"cell_constant(<p Grid>, 0)")
visit.DefineScalarExpression("operators/DataBinning/2D/p Grid",
"cell_constant(<p Grid>, 0)")
visit.DefineScalarExpression("operators/DataBinning/3D/p Grid",
"cell_constant(<p Grid>, 0)")
visit.DefineScalarExpression("operators/Flux/p Grid",
"cell_constant(<p Grid>, 0.)")
visit.DefineCurveExpression("operators/Lineout/p",
"cell_constant(<p>, 0.)")
visit.DefineCurveExpression(
"operators/Lineout/time_derivative/p Grid_time",
"cell_constant(time_derivative/p Grid_time, 0.)")
visit.DefineCurveExpression(
"operators/Lineout/time_derivative/p Grid_lasttime",
"cell_constant(time_derivative/p Grid_lasttime, 0.)")
visit.DefineCurveExpression("operators/Lineout/time_derivative/p",
"cell_constant(time_derivative/p, 0.)")
visit.DefineScalarExpression("operators/ModelFit/model",
"point_constant(<p Grid>, 0)")
visit.DefineScalarExpression("operators/ModelFit/distance",
"point_constant(<p Grid>, 0)")
visit.DefineScalarExpression("operators/StatisticalTrends/Sum/p",
"cell_constant(<p>, 0.)")
visit.DefineScalarExpression("operators/StatisticalTrends/Mean/p",
"cell_constant(<p>, 0.)")
visit.DefineScalarExpression("operators/StatisticalTrends/Variance/p",
"cell_constant(<p>, 0.)")
visit.DefineScalarExpression("operators/StatisticalTrends/Std. Dev./p",
"cell_constant(<p>, 0.)")
visit.DefineScalarExpression("operators/StatisticalTrends/Slope/p",
"cell_constant(<p>, 0.)")
visit.DefineScalarExpression("operators/StatisticalTrends/Residuals/p",
"cell_constant(<p>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Sum/time_derivative/p Grid_time",
"cell_constant(<time_derivative/p Grid_time>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Sum/time_derivative/p Grid_lasttime",
"cell_constant(<time_derivative/p Grid_lasttime>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Sum/time_derivative/p",
"cell_constant(<time_derivative/p>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Mean/time_derivative/p Grid_time",
"cell_constant(<time_derivative/p Grid_time>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Mean/time_derivative/p Grid_lasttime",
"cell_constant(<time_derivative/p Grid_lasttime>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Mean/time_derivative/p",
"cell_constant(<time_derivative/p>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Variance/time_derivative/p Grid_time",
"cell_constant(<time_derivative/p Grid_time>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Variance/time_derivative/p Grid_lasttime",
"cell_constant(<time_derivative/p Grid_lasttime>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Variance/time_derivative/p",
"cell_constant(<time_derivative/p>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Std. Dev./time_derivative/p Grid_time",
"cell_constant(<time_derivative/p Grid_time>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Std. Dev./time_derivative/p Grid_lasttime",
"cell_constant(<time_derivative/p Grid_lasttime>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Std. Dev./time_derivative/p",
"cell_constant(<time_derivative/p>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Slope/time_derivative/p Grid_time",
"cell_constant(<time_derivative/p Grid_time>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Slope/time_derivative/p Grid_lasttime",
"cell_constant(<time_derivative/p Grid_lasttime>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Slope/time_derivative/p",
"cell_constant(<time_derivative/p>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Residuals/time_derivative/p Grid_time",
"cell_constant(<time_derivative/p Grid_time>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Residuals/time_derivative/p Grid_lasttime",
"cell_constant(<time_derivative/p Grid_lasttime>, 0.)")
visit.DefineScalarExpression(
"operators/StatisticalTrends/Residuals/time_derivative/p",
"cell_constant(<time_derivative/p>, 0.)")
visit.DefineVectorExpression("operators/SurfaceNormal/p Grid",
"cell_constant(<p Grid>, 0.)")
# Define cell-centered velocity vector field.
ux_xdmf_path = os.path.join(str(xdmf_dir), 'u.xmf')
uy_xdmf_path = os.path.join(str(xdmf_dir), 'v.xmf')
uz_xdmf_path = os.path.join(str(xdmf_dir), 'w.xmf')
vel_exp = ('{' +
'pos_cmfe(<{}[0]id:u>, <p Grid>, 1.0),'.format(ux_xdmf_path) +
'pos_cmfe(<{}[0]id:v>, <p Grid>, 0.0),'.format(uy_xdmf_path) +
'pos_cmfe(<{}[0]id:w>, <p Grid>, 0.0)'.format(uz_xdmf_path) +
'}')
visit.DefineVectorExpression('velocity', vel_exp)
# Define Q-criterion.
qcrit_exp = ('q_criterion(' + 'gradient(velocity[0]),' +
'gradient(velocity[1]),' + 'gradient(velocity[2])' + ')')
visit.DefineScalarExpression('q_crit', qcrit_exp)
# Define cell-centered streamwise vorticity.
wx_xdmf_path = os.path.join(str(xdmf_dir), 'wx.xmf')
wx_exp = 'pos_cmfe(<{}[0]id:wx>, <p Grid>, 0.0)'.format(wx_xdmf_path)
visit.DefineScalarExpression('wx_cc', wx_exp)
# Add a pseudocolor of the cell-centered streamwise vorticity.
visit.AddPlot('Pseudocolor', 'wx_cc', 1, 1)
PseudocolorAtts = visit.PseudocolorAttributes()
PseudocolorAtts.minFlag = 1
PseudocolorAtts.min = wx_range[0]
PseudocolorAtts.maxFlag = 1
PseudocolorAtts.max = wx_range[1]
PseudocolorAtts.colorTableName = 'viridis'
PseudocolorAtts.invertColorTable = 1
PseudocolorAtts.opacityType = PseudocolorAtts.Constant
PseudocolorAtts.opacity = 0.8
PseudocolorAtts.legendFlag = 0
visit.SetPlotOptions(PseudocolorAtts)
# Add an isosurface of the Q-criterion.
visit.AddOperator('Isosurface', 1)
IsosurfaceAtts = visit.IsosurfaceAttributes()
IsosurfaceAtts.variable = 'q_crit'
IsosurfaceAtts.contourMethod = IsosurfaceAtts.Value
IsosurfaceAtts.contourValue = (q_value)
IsosurfaceAtts.scaling = IsosurfaceAtts.Linear
visit.SetOperatorOptions(IsosurfaceAtts, 1)
# Remove info about user, time, database, and legend.
AnnotationAtts = visit.AnnotationAttributes()
AnnotationAtts.userInfoFlag = 0
AnnotationAtts.databaseInfoFlag = 0
AnnotationAtts.timeInfoFlag = 0
AnnotationAtts.legendInfoFlag = 0
AnnotationAtts.axes3D.visible = 0
AnnotationAtts.axes3D.triadFlag = 1
AnnotationAtts.axes3D.bboxFlag = 0
visit.SetAnnotationAttributes(AnnotationAtts)
# Parse the 3D view configuration file.
if config_view is not None:
with open(str(config_view), 'r') as infile:
config_view = yaml.load(infile, Loader=yaml.FullLoader)
config_view = config_view['View3DAtts']
# Set attributes of the view.
View3DAtts = visit.View3DAttributes()
for key, value in config_view.items():
if type(value) is list:
value = tuple(value)
setattr(View3DAtts, key, value)
visit.SetView3D(View3DAtts)
visit.SetActiveWindow(1)
visit.Source(os.path.join(visit_dir, visit_arch, 'bin', 'makemovie.py'))
visit.ToggleCameraViewMode()
# Create output directory if necessary.
if not os.path.isdir(str(out_dir)):
os.makedirs(str(out_dir))
# Loop over the states to render and save the plots.
if state is not None:
states = [state]
elif states is None:
if states_range[1] is None:
states_range[1] = visit.TimeSliderGetNStates()
else:
states_range[1] += 1
states = range(*states_range)
for i, state in enumerate(states):
print('[state {}] Rendering and saving figure ...'.format(state))
visit.SetTimeSliderState(state)
if i == 0:
visit.DrawPlots()
RenderingAtts = visit.RenderingAttributes()
visit.SetRenderingAttributes(RenderingAtts)
SaveWindowAtts = visit.SaveWindowAttributes()
SaveWindowAtts.outputToCurrentDirectory = 0
SaveWindowAtts.outputDirectory = str(out_dir)
SaveWindowAtts.fileName = '{}{:0>4}'.format(out_prefix, state)
SaveWindowAtts.family = 0
SaveWindowAtts.format = SaveWindowAtts.PNG
SaveWindowAtts.width = figsize[0]
SaveWindowAtts.height = figsize[1]
SaveWindowAtts.quality = 100
SaveWindowAtts.resConstraint = SaveWindowAtts.NoConstraint
visit.SetSaveWindowAttributes(SaveWindowAtts)
visit.SaveWindow()
os.remove('visitlog.py')
visit.CloseComputeEngine()
visit.Close()
return
def make_movie( variableName, minValue, maxValue, inputDirectory, inputFileName, outputDirectory, outputFileName, colorTable="hot_desaturated", startFrame=-1, endFrame=-1 ):
'''
Function for making a movie
Arguments:
:param variableName Name of the variable
:param minValue Minimum value of the variable
:param maxValue Maximum value of the variable
:param inputDirectory Path to input vlsv/silo files
:param inputFileName Name of the file(s) so for example if the filenames are bulk.0000.silo, bulk.0001.silo, .. then inputFileName=\"bulk.*.silo\""
:param outputDirectory Path to output directory
:param outputFileName Name of the output file
:param colorTable="hot_desaturated" Color table for the plots
:param startFrame=-1 Starting frame of the movie (-1 equals 0)
:param endFrame=-1 Starting frame of the movie (-1 equals last frame)
'''
# OPTIONS
#################################################################
# Input frame properties
_startFrame = startFrame # Note: if _startFrame is set to -1 the start frame gets set to 0
_endFrame = endFrame # Note: if _endFrame is set to -1 the _endFrame is automatically the number of frames in the database
# Input variable
_variableName = variableName
minVariableValue = minValue
maxVariableValue = maxValue
colorTableName = colorTable
# Input directory and file names
#_outputDir = "/home/hannukse/MOVINGFRAME_MOVIES/AAJ_BZ_REMAKE/" # Set the output directory (Where .png s are saved)
_outputDir = outputDirectory
#_outputFileName = "BZ_FORESHOCK_2_" # The file names for the png files. These for ex. will be saved visit0000.png, visit0001.png, ..
_outputFileName = outputFileName # The file names for the png files.
#databaseName = "localhost:/home/hannukse/meteo/stornext/field/vlasiator/2D/AAJ/silo_files/bulk.*.silo database" # For navigating to the silo files
databaseName = "localhost:" + inputDirectory + inputFileName + " database" # For navigating to the silo files
# Note: a slice of the plot in z-axis is taken automatically
#################################################################
# LaunchNowin(vdir=visitBinDirectory)
#dx = speedX * frameInSeconds # Note: This is in meters per frame!
#dy = speedY * frameInSeconds # Note: This is in meters per frame!
#LaunchNowin(vdir="/usr/local/visit/bin")
#Set up window and annotations
#vis.LaunchNowin(vdir="/usr/local/visit/bin")
vis.OpenDatabase(databaseName, 0)
#Load settings
visSettings.load_visit_settings()
vis.AddPlot("Pseudocolor", _variableName, 1, 1) #CONTINUE
vis.SetActivePlots(0)
vis.PseudocolorAtts = vis.PseudocolorAttributes()
vis.PseudocolorAtts.legendFlag = 1
vis.PseudocolorAtts.lightingFlag = 1
vis.PseudocolorAtts.minFlag = 1
vis.PseudocolorAtts.maxFlag = 1
vis.PseudocolorAtts.centering = vis.PseudocolorAtts.Natural # Natural, Nodal, Zonal
vis.PseudocolorAtts.scaling = vis.PseudocolorAtts.Linear # Linear, Log, Skew
vis.PseudocolorAtts.limitsMode = vis.PseudocolorAtts.CurrentPlot # OriginalData, CurrentPlot
vis.PseudocolorAtts.min = minVariableValue
vis.PseudocolorAtts.max = maxVariableValue
vis.PseudocolorAtts.pointSize = 0.05
vis.PseudocolorAtts.pointType = vis.PseudocolorAtts.Point # Box, Axis, Icosahedron, Point, Sphere
vis.PseudocolorAtts.skewFactor = 1
vis.PseudocolorAtts.opacity = 1
vis.PseudocolorAtts.colorTableName = colorTableName
vis.PseudocolorAtts.invertColorTable = 0
vis.PseudocolorAtts.smoothingLevel = 0
vis.PseudocolorAtts.pointSizeVarEnabled = 0
vis.PseudocolorAtts.pointSizeVar = "default"
vis.PseudocolorAtts.pointSizePixels = 2
vis.PseudocolorAtts.lineStyle = vis.PseudocolorAtts.SOLID # SOLID, DASH, DOT, DOTDASH
vis.PseudocolorAtts.lineWidth = 0
vis.PseudocolorAtts.opacityType = vis.PseudocolorAtts.Explicit # Explicit, ColorTable
vis.SetPlotOptions(vis.PseudocolorAtts)
vis.SetActivePlots(0)
vis.AddOperator("Slice", 1)
vis.SetActivePlots(0)
vis.SliceAtts = vis.SliceAttributes()
vis.SliceAtts.originType = vis.SliceAtts.Intercept # Point, Intercept, Percent, Zone, Node
vis.SliceAtts.originPoint = (0, 0, 0)
vis.SliceAtts.originIntercept = 0
vis.SliceAtts.originPercent = 0
vis.SliceAtts.originZone = 0
vis.SliceAtts.originNode = 0
vis.SliceAtts.normal = (0, 0, 1)
vis.SliceAtts.axisType = vis.SliceAtts.ZAxis # XAxis, YAxis, ZAxis, Arbitrary, ThetaPhi
vis.SliceAtts.upAxis = (0, 1, 0)
vis.SliceAtts.project2d = 1
vis.SliceAtts.interactive = 1
vis.SliceAtts.flip = 0
vis.SliceAtts.originZoneDomain = 0
vis.SliceAtts.originNodeDomain = 0
vis.SliceAtts.meshName = "SpatialGrid"
vis.SliceAtts.theta = 0
vis.SliceAtts.phi = 90
vis.SetOperatorOptions(vis.SliceAtts, 1)
vis.DrawPlots()
if _endFrame == -1:
_endFrame = vis.TimeSliderGetNStates() - 1
if _startFrame == -1:
_startFrame = 0
# Iterate through frames
for i in range(_startFrame, _endFrame+1):
vis.SetTimeSliderState(i)
frame = i - _startFrame
vis.SaveWindowAtts = vis.SaveWindowAttributes()
vis.SaveWindowAtts.outputToCurrentDirectory = 0
vis.SaveWindowAtts.outputDirectory = _outputDir
vis.SaveWindowAtts.fileName = _outputFileName
vis.SaveWindowAtts.family = 1
vis.SaveWindowAtts.format = vis.SaveWindowAtts.PNG # BMP, CURVE, JPEG, OBJ, PNG, POSTSCRIPT, POVRAY, PPM, RGB, STL, TIFF, ULTRA, VTK, PLY
vis.SaveWindowAtts.width = 3000
vis.SaveWindowAtts.height = 3000
vis.SaveWindowAtts.screenCapture = 0
vis.SaveWindowAtts.saveTiled = 0
vis.SaveWindowAtts.quality = 100
vis.SaveWindowAtts.progressive = 0
vis.SaveWindowAtts.binary = 0
vis.SaveWindowAtts.stereo = 0
vis.SaveWindowAtts.compression = vis.SaveWindowAtts.PackBits # None, PackBits, Jpeg, Deflate
vis.SaveWindowAtts.forceMerge = 0
vis.SaveWindowAtts.resConstraint = vis.SaveWindowAtts.ScreenProportions # NoConstraint, EqualWidthHeight, ScreenProportions
vis.SaveWindowAtts.advancedMultiWindowSave = 0
vis.SetSaveWindowAttributes(vis.SaveWindowAtts)
vis.SaveWindow()
vis.DeleteActivePlots()
vis.CloseDatabase(databaseName)
# Make the movie:
#subprocess.call("./moviecompilescript.sh " + _outputDir + " " + _outputFileName)
pyVisitPath = "pyVisit/"
#subprocess.call(pythonLibDirectoryPath + pyVisitPath + "moviecompilescript.sh")
#subprocess.call(pythonLibDirectoryPath + pyVisitPath + "moviecompilescript.sh " + _outputDir + " " + _outputFileName)
framerate = "10"
subprocess.call([pythonLibDirectoryPath + pyVisitPath + "moviecompilescript.sh", _outputDir, _outputFileName, framerate])
def createPseudocolor(self, varname, display_name=None, cmap=None,
limits=None, linewidth=None, legend=True, alpha=False):
"""Generic creation of pseudocolor"""
if display_name is None:
display_name = vrc.renameScalar(varname)
if "temperature" in display_name:
display_name = display_name.replace("[K]", "[C]")
print "defining alias: %s = %s"%(display_name, varname)
v.DefineScalarExpression(display_name, "%s - 273.15"%varname)
elif display_name != varname:
print "defining alias: %s = %s"%(display_name, varname)
v.DefineScalarExpression(display_name, varname)
v.AddPlot('Pseudocolor', display_name)
pa = v.PseudocolorAttributes()
# limits
if limits is None:
limits = vrc.getLimits(varname)
if limits is not None:
min = limits[0]
max = limits[1]
if min is not None:
pa.minFlag = 1
pa.min = min
if max is not None:
pa.maxFlag = 1
pa.max = max
# opacity
if alpha:
pa.opacity = 0
pa.opacityType = pa.ColorTable
# colormap
if cmap is not None:
reverse = cmap.endswith("_r")
if reverse:
cmap = cmap.strip("_r")
pa.invertColorTable = 1
pa.colorTableName = cmap
# linewidth for 2D
if linewidth is None:
linewidth = vrc.rcParams['pseudocolor.linewidth']
pa.lineWidth = linewidth
# turn off legend for 2D surf
if not legend:
pa.legendFlag = 0
v.SetPlotOptions(pa)
pname = v.GetPlotList().GetPlots(v.GetNumPlots()-1).plotName
if legend:
plot = Plot(pname, 'Pseudocolor', pa, display_name)
else:
plot = Plot(pname, 'Pseudocolor', pa)
self.plots.append(plot)
return plot
def make_moving_frame_of_reference_movie(x_begin,
x_end,
y_begin,
y_end,
speed_x,
speed_y,
variable_name,
minThreshold,
maxThreshold,
input_directory,
input_file_name,
output_directory,
output_file_name,
color_table="hot_desaturated",
start_frame=-1,
end_frame=-1,
frame_skip_dt=1.0):
'''
Function for making a movie with a moving frame of reference.
:param x_begin The starting frame's beginning x-coordinate
:param x_end The starting frame's ending x-coordinate
:param y_begin The starting frame's beginning x-coordinate
:param y_end The starting frame's ending y-coordinate
:param speed_x The speed at which the frame moves in the x direction
:param speed_y The speed at which the frame moves in the y direction
:param variable_name Name of the variable (For ex \"rho\")
:param minThreshold Minimum threshold for the variable
:param maxThreshold Maximum threshold for the variable
:param input_directory The path to the directory where the files are
:param input_file_name Name of the files (For ex \"bulk.*.silo\")
:param output_directory Directory where to output the movie
:param output_file_name Name of the outputted file (For ex \"RHOMOVIE\")
:param color_table Name of the color table (\"hot_desaturated\" by default)
:param start_frame Starting frame for the movie (if -1, equals 0, -1 by default)
:param end_frame Ending frame for the movie (if -1, equals the last frame, -1 by default)
:param frame_skip_dt The number of seconds one skip in frame equals (1.0 by default) (Note: This may change depending on the run and should always be checked)
'''
# OPTIONS
#################################################################
# Input the boundary box for starting coordinates (Starting values)
startX = x_begin # The left x-boundary of the box
endX = x_end # The right x-boundary of the box
startY = y_begin # The bottom y-boundary of the box
endY = y_end # The upper y-boundary of the box
# Input frame properties
startFrame = start_frame # Note: if startFrame is set to -1 the start frame gets set to 0
endFrame = end_frame # Note: if endFrame is set to -1 the endFrame is automatically the number of frames in the database
frameInSeconds = frame_skip_dt # Set how many seconds one frame skip is
# Input speed in x and y direction
speedX = speed_x # Meters per second
speedY = speed_y # Meters per second
# Input variable
variableName = variable_name
minVariableValue = minThreshold
maxVariableValue = maxThreshold
colorTableName = color_table
# Input directory and file names
outputDir = output_directory # Set the output directory (Where .png s are saved)
outputFileName = output_file_name # The file names for the png files. These for ex. will be saved visit0000.png, visit0001.png, .
databaseName = "localhost:" + input_directory + input_file_name + " database" # For navigating to the silo files
# visitBinDirectory = "/usr/local/visit/bin" #Nevermind this
# Note: a slice of the plot in z-axis is taken automatically
#################################################################
# Launch visit
visitBinDirectory = '/home/htest/visit/bin'
vis.LaunchNowin(vdir=visitBinDirectory)
dx = speedX * frameInSeconds # Note: This is in meters per frame!
dy = speedY * frameInSeconds # Note: This is in meters per frame!
#Set up window and annotations
vis.OpenDatabase(databaseName, 0)
#Load settings
visSettings.load_visit_settings()
vis.AddPlot("Pseudocolor", variableName, 1, 1)
vis.SetActivePlots(0)
vis.PseudocolorAtts = vis.PseudocolorAttributes()
vis.PseudocolorAtts.legendFlag = 1
vis.PseudocolorAtts.lightingFlag = 1
vis.PseudocolorAtts.minFlag = 1
vis.PseudocolorAtts.maxFlag = 1
vis.PseudocolorAtts.centering = vis.PseudocolorAtts.Natural # Natural, Nodal, Zonal
vis.PseudocolorAtts.scaling = vis.PseudocolorAtts.Linear # Linear, Log, Skew
vis.PseudocolorAtts.limitsMode = vis.PseudocolorAtts.CurrentPlot # OriginalData, CurrentPlot
vis.PseudocolorAtts.min = minVariableValue
vis.PseudocolorAtts.max = maxVariableValue
vis.PseudocolorAtts.pointSize = 0.05
vis.PseudocolorAtts.pointType = vis.PseudocolorAtts.Point # Box, Axis, Icosahedron, Point, Sphere
vis.PseudocolorAtts.skewFactor = 1
vis.PseudocolorAtts.opacity = 1
vis.PseudocolorAtts.colorTableName = color_table
vis.PseudocolorAtts.invertColorTable = 0
vis.PseudocolorAtts.smoothingLevel = 0
vis.PseudocolorAtts.pointSizeVarEnabled = 0
vis.PseudocolorAtts.pointSizeVar = "default"
vis.PseudocolorAtts.pointSizePixels = 2
vis.PseudocolorAtts.lineStyle = vis.PseudocolorAtts.SOLID # SOLID, DASH, DOT, DOTDASH
vis.PseudocolorAtts.lineWidth = 0
vis.PseudocolorAtts.opacityType = vis.PseudocolorAtts.Explicit # Explicit, ColorTable
vis.SetPlotOptions(vis.PseudocolorAtts)
vis.SetActivePlots(0)
vis.AddOperator("Slice", 1)
vis.AddOperator("Threshold", 1)
vis.ThresholdAtts = vis.ThresholdAttributes()
vis.ThresholdAtts.outputMeshType = 0
vis.ThresholdAtts.listedVarNames = ("Boundary_type")
vis.ThresholdAtts.zonePortions = (1)
vis.ThresholdAtts.lowerBounds = (1)
vis.ThresholdAtts.upperBounds = (1)
vis.ThresholdAtts.defaultVarName = variableName
vis.ThresholdAtts.defaultVarIsScalar = 1
vis.SetOperatorOptions(vis.ThresholdAtts, 1)
vis.ThresholdAtts = vis.ThresholdAttributes()
vis.ThresholdAtts.outputMeshType = 0
vis.ThresholdAtts.listedVarNames = ("Boundary_type")
vis.ThresholdAtts.zonePortions = (1)
vis.ThresholdAtts.lowerBounds = (1)
vis.ThresholdAtts.upperBounds = (1)
vis.ThresholdAtts.defaultVarName = variableName
vis.ThresholdAtts.defaultVarIsScalar = 1
vis.SetOperatorOptions(vis.ThresholdAtts, 1)
vis.SetActivePlots(0)
vis.SliceAtts = vis.SliceAttributes()
vis.SliceAtts.originType = vis.SliceAtts.Intercept # Point, Intercept, Percent, Zone, Node
vis.SliceAtts.originPoint = (0, 0, 0)
vis.SliceAtts.originIntercept = 0
vis.SliceAtts.originPercent = 0
vis.SliceAtts.originZone = 0
vis.SliceAtts.originNode = 0
vis.SliceAtts.normal = (0, 0, 1)
vis.SliceAtts.axisType = vis.SliceAtts.ZAxis # XAxis, YAxis, ZAxis, Arbitrary, ThetaPhi
vis.SliceAtts.upAxis = (0, 1, 0)
vis.SliceAtts.project2d = 1
vis.SliceAtts.interactive = 1
vis.SliceAtts.flip = 0
vis.SliceAtts.originZoneDomain = 0
vis.SliceAtts.originNodeDomain = 0
vis.SliceAtts.meshName = "SpatialGrid"
vis.SliceAtts.theta = 0
vis.SliceAtts.phi = 90
vis.SetOperatorOptions(vis.SliceAtts, 1)
vis.DrawPlots()
if endFrame == -1:
endFrame = vis.TimeSliderGetNStates() - 1
if startFrame == -1:
startFrame = 0
# Iterate through frames
for i in xrange(startFrame, endFrame + 1):
vis.SetTimeSliderState(i)
frame = i - startFrame
vis.View2DAtts = vis.View2DAttributes()
vis.View2DAtts.windowCoords = (startX + frame * dx, endX + frame * dx,
startY + frame * dy, endY + frame * dy)
vis.View2DAtts.viewportCoords = (0.2, 0.95, 0.15, 0.95)
vis.View2DAtts.fullFrameActivationMode = vis.View2DAtts.Auto # On, Off, Auto
vis.View2DAtts.fullFrameAutoThreshold = 100
vis.View2DAtts.xScale = vis.View2DAtts.LINEAR # LINEAR, LOG
vis.View2DAtts.yScale = vis.View2DAtts.LINEAR # LINEAR, LOG
vis.View2DAtts.windowValid = 1
vis.SetView2D(vis.View2DAtts)
vis.SaveWindowAtts = vis.SaveWindowAttributes()
vis.SaveWindowAtts.outputToCurrentDirectory = 0
vis.SaveWindowAtts.outputDirectory = outputDir
vis.SaveWindowAtts.fileName = outputFileName
vis.SaveWindowAtts.family = 1
vis.SaveWindowAtts.format = vis.SaveWindowAtts.PNG # BMP, CURVE, JPEG, OBJ, PNG, POSTSCRIPT, POVRAY, PPM, RGB, STL, TIFF, ULTRA, VTK, PLY
vis.SaveWindowAtts.width = 1024
vis.SaveWindowAtts.height = 1024
vis.SaveWindowAtts.screenCapture = 0
vis.SaveWindowAtts.saveTiled = 0
vis.SaveWindowAtts.quality = 100
vis.SaveWindowAtts.progressive = 0
vis.SaveWindowAtts.binary = 0
vis.SaveWindowAtts.stereo = 0
vis.SaveWindowAtts.compression = vis.SaveWindowAtts.PackBits # None, PackBits, Jpeg, Deflate
vis.SaveWindowAtts.forceMerge = 0
vis.SaveWindowAtts.resConstraint = vis.SaveWindowAtts.ScreenProportions # NoConstraint, EqualWidthHeight, ScreenProportions
vis.SaveWindowAtts.advancedMultiWindowSave = 0
vis.SetSaveWindowAttributes(vis.SaveWindowAtts)
vis.SaveWindow()
vis.DeleteActivePlots()
vis.CloseDatabase(databaseName)
# Make the movie:
framerate = 7
subprocess.call([
pythonLibDirectoryPath + pyVisitPath + "moviecompilescript.sh",
outputDir, outputFileName, framerate
])
