# ----------------------------------------------------------------------------
#  CLASSES: nightly
#
#  Test Case:  subset.py
#
#  Tests:      meshes    - 2D curvilinear, multiple domain
#              plots     - subset
#
#  Defect ID:  '6105, '6762
#
#  Programmer: Hank Childs
#  Date:       March 31, 2005
#
#  Modifications:
#
#    Hank Childs, Tue Nov 15 07:28:43 PST 2005
#    Added test for subset plus macro expression.
#
#    Mark C. Miller, Wed Jan 20 07:37:11 PST 2010
#    Added ability to swtich between Silo's HDF5 and PDB data.
#
#    Cyrus Harrison, Thu Mar 25 09:57:34 PDT 2010
#    Added call(s) to DrawPlots() b/c of changes to the default plot state
#    behavior when an operator is added.
#
#    Hank Childs, Mon Mar 26 12:00:23 PDT 2012
#    Add test for Subset plot + material selection
#
#    Kathleen Biagas, Fri Jun  5 08:35:31 PDT 2020
#    Added tests for point glyhphing/sizing for data that doesn't declare
#    itself a point mesh and for data with mixed topology.
#
# ----------------------------------------------------------------------------


OpenDatabase(silo_data_path("multi_curv2d.silo"))

AddPlot("Subset", "domains")
DrawPlots()

v = GetView2D()
v.viewportCoords = (0, 1, 0, 1)
v.windowCoords = (-5, 5, 0, 5)
SetView2D(v)

Test("subset_01")

s = SubsetAttributes()
s.wireframe = 1
SetPlotOptions(s)
Test("subset_02")

i = IsovolumeAttributes()
i.variable = "d"
i.lbound = 2.7
i.ubound = 3.6
SetDefaultOperatorOptions(i)
AddOperator("Isovolume")
DrawPlots()
Test("subset_03")

s.wireframe = 0
SetPlotOptions(s)
Test("subset_04")

RemoveLastOperator()
op = OnionPeelAttributes()
op.categoryName = "domains"
op.subsetName = "domain3"
op.index = (100)
SetDefaultOperatorOptions(op)
AddOperator("OnionPeel")
DrawPlots()

v.windowCoords = (-0.7, -0.3, 3.1, 3.25)
SetView2D(v)

Test("subset_05")

# This is a different wireframe path than the other tests, because
# we have special handling for single cell wireframes.
# This corresponds to ticket '6105.
s.wireframe = 1
SetPlotOptions(s)
Test("subset_06")

DeleteAllPlots()
CloseDatabase(silo_data_path("multi_curv2d.silo"))

OpenDatabase(data_path("boxlib_test_data/2D/plt0822/Header"))

AddPlot("Subset", "levels")
AddOperator("Isovolume")
DefineScalarExpression("vort", "curl(vel)")
iso_atts = IsovolumeAttributes()
iso_atts.variable = "vort"
iso_atts.lbound = -10000000
iso_atts.ubound = 0
SetOperatorOptions(iso_atts)
DrawPlots()
ResetView()
Test("subset_07")

RemoveLastOperator()
s = SILRestriction()
mats = s.SetsInCategory("materials")
s.TurnOffSet(mats[0])
s.TurnOffSet(mats[1])
SetPlotSILRestriction(s)
Test("subset_08")

ChangeActivePlotsVar("patches")
Test("subset_09")

DeleteAllPlots()
CloseDatabase(data_path("boxlib_test_data/2D/plt0822/Header"))

OpenDatabase(data_path("silo_amr_test_data/amr2d_wmrgtree.silo"))

AddPlot("Subset", "blocks(amr_mesh)")
DrawPlots()

s = SubsetAttributes()
s.wireframe = 1
SetPlotOptions(s)

Test("subset_10")

DeleteAllPlots()
CloseDatabase(data_path("silo_amr_test_data/amr2d_wmrgtree.silo"))

TestSection("point mesh")
OpenDatabase(data_path("blueprint_v0.3.1_test_data/braid_3d_examples_json.root"))
AddPlot("Subset", "domains(points_mesh)")

s = SubsetAttributes()
s.colorType = s.ColorBySingleColor
s.singleColor = (0, 170, 255, 255)
SetPlotOptions(s)
DrawPlots()
ResetView()
v = GetView3D()
v.viewNormal = (-0.605449, 0.469667, 0.642529)
v.viewUp = (0.169201, 0.864818, -0.472716)
SetView3D(v)
Test("subset_11")

s.pointSizePixels = 5
SetPlotOptions(s)
Test("subset_12")


s.pointType = s.Tetrahedron
s.pointSize = 3
SetPlotOptions(s)
Test("subset_13")

DeleteAllPlots()
CloseDatabase(data_path("blueprint_v0.3.1_test_data/braid_3d_examples_json.root"))


OpenDatabase(data_path("vtk_test_data/ugrid_mixed_cells.vtk"))

AddPlot("Subset", "mesh")
s = SubsetAttributes()
s.colorType = s.ColorBySingleColor
s.singleColor = (0, 170, 255, 255)
s.lineWidth = 3
s.pointType = s.Point
s.pointSizePixels = 2
SetPlotOptions(s)
DrawPlots()
ResetView()

Test("subset_14")

s.pointSizePixels = 5
SetPlotOptions(s)
Test("subset_15")

s.pointType = s.Icosahedron
s.pointSize = 0.5
SetPlotOptions(s)
Test("subset_16")

DeleteAllPlots()
CloseDatabase(data_path("vtk_test_data/ugrid_mixed_cells.vtk"))

Exit()