plots/volumePlot.py# ----------------------------------------------------------------------------
# CLASSES: nightly
#
# Test Case: volumePlot.py
#
# Tests: mesh - 3D rectilinear, one domain
# plots - volume
# operators - none
# selection - no
#
# Programmer: Kathleen Bonnell
# Date: March 4, 2005
#
# Modifications:
# Kathleen Bonnell, Fri Oct 14 10:12:06 PDT 2005
# Added InitAnnotationsLegendOn and TestVolumeOpacity per '6670.
#
# Hank Childs, Fri May 26 10:11:52 PDT 2006
# Add test for extremely high aspect ratio meshes. ['7250]
#
# Brad Whitlock, Wed Mar 14 17:18:05 PST 2007
# Added tests for color control points and gaussian control points.
#
# Brad Whitlock, Thu Mar 12 09:17:57 PDT 2009
# I changed freeformFlag to opacityMode to reflect changes to the plot
#
# Mark C. Miller, Wed Jan 20 07:37:11 PST 2010
# Added ability to swtich between Silo's HDF5 and PDB data.
#
# Hank Childs, Sun Aug 29 16:07:56 PDT 2010
# Change Log10 to Log, to reflect change in naming.
#
# ----------------------------------------------------------------------------
def InitAnnotations():
# Turn off all annotation
TurnOffAllAnnotations()
def InitAnnotationsLegendOn():
# Turn off all annotation
a = AnnotationAttributes()
TurnOffAllAnnotations(a)
a.legendInfoFlag = 1
SetAnnotationAttributes(a)
def TestVolumeScaling():
OpenDatabase(silo_data_path("rect3d.silo"))
AddPlot("Volume", "t")
volAtts = VolumeAttributes()
volAtts.rendererType = volAtts.Splatting
SetPlotOptions(volAtts)
DrawPlots()
v = GetView3D()
v.viewNormal = (1, 0, 0)
SetView3D(v)
Test("volumeScaling_01")
volAtts.scaling = volAtts.Log
SetPlotOptions(volAtts)
Test("volumeScaling_02")
volAtts.scaling = volAtts.Skew
volAtts.skewFactor = 0.0001
SetPlotOptions(volAtts)
Test("volumeScaling_03")
volAtts.skewFactor = 1000
SetPlotOptions(volAtts)
Test("volumeScaling_04")
volAtts.rendererType = volAtts.Texture3D
volAtts.scaling = volAtts.Linear
SetPlotOptions(volAtts)
Test("volumeScaling_05")
volAtts.scaling = volAtts.Log
SetPlotOptions(volAtts)
Test("volumeScaling_06")
volAtts.scaling = volAtts.Skew
volAtts.skewFactor = 0.0001
SetPlotOptions(volAtts)
Test("volumeScaling_07")
volAtts.skewFactor = 1000
SetPlotOptions(volAtts)
Test("volumeScaling_08")
volAtts.rendererType = volAtts.RayCasting
volAtts.scaling = volAtts.Linear
SetPlotOptions(volAtts)
Test("volumeScaling_09")
volAtts.scaling = volAtts.Log
SetPlotOptions(volAtts)
Test("volumeScaling_10")
volAtts.scaling = volAtts.Skew
volAtts.skewFactor = 0.0001
SetPlotOptions(volAtts)
Test("volumeScaling_11")
volAtts.skewFactor = 1000
SetPlotOptions(volAtts)
Test("volumeScaling_12")
DeleteAllPlots()
def TestVolumeOpacity():
OpenDatabase(silo_data_path("noise.silo"))
AddPlot("Volume", "hardyglobal")
volAtts = VolumeAttributes()
SetPlotOptions(volAtts)
DrawPlots()
ResetView()
# default opacity
Test("volumeOpacity_01")
# setting opacity via individual index and value
for i in range(150):
volAtts.SetFreeformOpacity(i, 0)
SetPlotOptions(volAtts)
Test("volumeOpacity_02")
# setting opacity via creation of a tuple
opac1 = (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 8, 16, 24, 32, 40, 48, 56, 64, 72, 80, 88, 96, 104, 112, 120, 128, 136, 144, 152, 160, 168, 176, 184, 192, 200, 208, 216, 224, 232, 240, 248, 256, 248, 240, 232, 224, 216, 208, 200, 192, 184, 176, 168, 160, 152, 144, 136, 128, 120, 112, 104, 96, 88, 80, 72, 64, 56, 48, 40, 32, 24, 16, 8, 0, 8, 16, 24, 32, 40, 48, 56, 64, 72, 80, 88, 96, 104, 112, 120, 128, 136, 144, 152, 160, 168, 176, 184, 192, 200, 208, 216, 224, 232, 240, 248, 256, 248, 240, 232, 224, 216, 208, 200, 192, 184, 176, 168, 160, 152, 144, 136, 128, 120, 112, 104, 96, 88, 80, 72, 64, 56, 48, 40, 32, 24, 16, 8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)
volAtts.SetFreeformOpacity(opac1)
SetPlotOptions(volAtts)
Test("volumeOpacity_03")
# setting opacity via creation of a list
opac2 = []
for j in range(32):
opac2 = opac2 + [j*8]
for j in range(32):
opac2 = opac2 + [256-(j*8)]
for j in range(128):
opac2 = opac2 + [0]
for j in range(32):
opac2 = opac2 + [j*8]
for j in range(32):
opac2 = opac2 + [256-(j*8)]
volAtts.SetFreeformOpacity(opac2)
SetPlotOptions(volAtts)
Test("volumeOpacity_04")
DeleteAllPlots()
def TestVolumeAspect():
OpenDatabase(silo_data_path("noise.silo"))
AddPlot("Volume", "hardyglobal")
DefineVectorExpression("disp", "{0,0,-0.9999*coord(Mesh)[2]}")
AddOperator("Displace")
d = DisplaceAttributes()
d.variable = "disp"
SetOperatorOptions(d)
DrawPlots()
ResetView()
v = GetView3D()
v.viewNormal = (-0.324974, 0.839345, 0.435765)
SetView3D(v)
Test("volumeAspect_01")
DeleteAllPlots()
def TestVolumeColorControlPoints():
OpenDatabase(silo_data_path("noise.silo"))
AddPlot("Volume", "hardyglobal")
# Modify colors. The default color table has 5 control points. Delete
# all but 2 of them and then change their colors.
v = VolumeAttributes()
v.colorControlPoints.RemoveControlPoints(4)
v.colorControlPoints.RemoveControlPoints(3)
v.colorControlPoints.RemoveControlPoints(2)
v.colorControlPoints.GetControlPoints(0).colors = (255,0,0,255)
v.colorControlPoints.GetControlPoints(0).position = 0.
v.colorControlPoints.GetControlPoints(1).colors = (0,0,255,255)
v.colorControlPoints.GetControlPoints(1).position = 1.
SetPlotOptions(v)
DrawPlots()
ResetView()
Test("volumeColors_01")
# Start over with the colors.
v.GetColorControlPoints().ClearControlPoints()
tmp = ColorControlPoint()
npts = 10
for i in range(npts):
t = 0.
if i < npts/2:
t = float(i) / float(npts/2 - 1)
r = 255
g = int(t * 255.)
b = 0
else:
t = 1. - (float(i-(npts/2)) / float(npts/2 - 1))
r = int((1-t) * 255.)
g = int((1-t) * 255.)
b = int(t * 255.)
tmp.colors = (r,g,b,255)
tmp.position = float(i) / float(npts-1)
v.GetColorControlPoints().AddControlPoints(tmp)
SetPlotOptions(v)
Test("volumeColors_02")
DeleteAllPlots()
def TestVolumeGaussianControlPoints():
OpenDatabase(silo_data_path("noise.silo"))
AddPlot("Volume", "hardyglobal")
v = VolumeAttributes()
v.opacityMode = v.GaussianMode
v.opacityControlPoints.ClearControlPoints()
tmp = GaussianControlPoint()
# Design gaussians
tmp.x = 0.818182
tmp.height = 0.787879
tmp.width = 0.147059
tmp.xBias = 0
tmp.yBias = 0
v.GetOpacityControlPoints().AddControlPoints(tmp)
tmp.x = 0.203209
tmp.height = 0.590909
tmp.width = 0.0588235
tmp.xBias = 0.0026738
tmp.yBias = 2
v.GetOpacityControlPoints().AddControlPoints(tmp)
tmp.x = 0.47861
tmp.height = 0.19697
tmp.width = 0.0721925
tmp.xBias = -0.00802138
tmp.yBias = 0
v.GetOpacityControlPoints().AddControlPoints(tmp)
SetPlotOptions(v)
DrawPlots()
ResetView()
Test("volumeGaussian_01")
DeleteAllPlots()
def TestVolumeSLIVR():
OpenDatabase(silo_data_path("noise.silo"))
AddPlot("Volume", "hardyglobal")
# Test SLIVR with Gaussian
v = VolumeAttributes()
v.rendererType = v.SLIVR
v.opacityAttenuation = 0.5 #default for SLIVR
v.rendererSamples = 3
v.opacityMode = v.GaussianMode
v.opacityControlPoints.ClearControlPoints()
tmp = GaussianControlPoint()
# Design gaussians
tmp.x = 0.858136
tmp.height = 0.431579
tmp.width = 0.184979
tmp.xBias = 0.0166898
tmp.yBias = 2
v.GetOpacityControlPoints().AddControlPoints(tmp)
tmp.x = 0.557719
tmp.height = 0.631579
tmp.width = 0.0445063
tmp.xBias = 0.00417244
tmp.yBias = 1.13333
v.GetOpacityControlPoints().AddControlPoints(tmp)
tmp.x = 0.215577
tmp.height = 0.631579
tmp.width = 0.0890125
tmp.xBias = -0.00556327
tmp.yBias = 0.944445
v.GetOpacityControlPoints().AddControlPoints(tmp)
# SLIVR with No Lighting
v.lightingFlag = 0
v.resampleFlag = 0
SetPlotOptions(v)
DrawPlots()
Test("volumeSLIVR_01")
# SLIVR with Lighting
v.lightingFlag = 1
v.SetMaterialProperties(0.2, 0.6, 0.7, 50)
SetPlotOptions(v)
DrawPlots()
Test("volumeSLIVR_02")
# SLIVR with Lighting and Transluscency
v.opacityAttenuation = 0.25
SetPlotOptions(v)
DrawPlots()
Test("volumeSLIVR_03")
"""
# Test with a 2D transfer Fn
v.transferFunctionDim = 2
temp = TransferFunctionWidget()
#temp.Type = v.GetTransferFunction2DWidgets(0).Ellipsoid # Rectangle, Triangle, Paraboloid, Ellipsoid
temp.Type = 3 # Rectangle, Triangle, Paraboloid, Ellipsoid
temp.Name = "ellipse0001"
temp.BaseColor = (0.2, 0.4, 1, 0.58)
temp.Position = (0.665608, 0.0550725, 0.1, 0.2, 1.47216, 0, 0, 0)
+
temp.Type = 0 # Rectangle, Triangle, Paraboloid, Ellipsoid
temp.Name = "rect0002"
temp.BaseColor = (1, 1, 0, 0.8)
temp.Position = (0.0761905, 0.157246, 0.370899, 0.487681, 0.5, 0, 0, 0)
+
SetPlotOptions(v)
DrawPlots()
ResetView()
Test("volumeSLIVR_04")
"""
# Freeform opacity with VisIt resampling
v.transferFunctionDim = 1
v.opacityMode = v.FreeformMode
v.freeformOpacity = (0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 72, 101, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254)
v.opacityAttenuation = 1
v.rendererSamples = 2
v.resampleFlag = 0
SetPlotOptions(v)
DrawPlots()
ResetView()
Test("volumeSLIVR_05")
DeleteAllPlots()
InitAnnotationsLegendOn()
TestVolumeColorControlPoints()
TestVolumeGaussianControlPoints()
TestVolumeAspect()
TestVolumeOpacity()
TestVolumeSLIVR()
InitAnnotations()
TestVolumeScaling()
Exit()