# ----------------------------------------------------------------------------
#  CLASSES: nightly
#
#  Test Case:  pos_cmfe.py
#
#  Defect ID:  '5528, '5677
#
#  Programmer: Hank Childs
#  Date:       January 9, 2006
#
#  Modifications:
#    Jeremy Meredith, Tue Jul 15 10:43:58 EDT 2008
#    Changed number of vectors in vector plot to match the old behavior.
#    (We now account for how many domains there are.)
#
#    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.
#
# ----------------------------------------------------------------------------




OpenDatabase(silo_data_path("curv2d.silo"))



# Test general capability.
DefineScalarExpression("cmfe", "pos_cmfe(<%s:d>, curvmesh2d, -1.)" % cmfe_silo_data_path("ucd2d.silo"))
AddPlot("Pseudocolor", "cmfe")
DrawPlots()
Test("pos_cmfe_01")

# Zonal vars
DefineScalarExpression("cmfeZ", "pos_cmfe(<%s:p>, curvmesh2d, -1.)" %  cmfe_silo_data_path("ucd2d.silo"))
DeleteAllPlots()
AddPlot("Pseudocolor", "cmfeZ")
DrawPlots()
Test("pos_cmfe_02")

# Test using the expression as a secondary variable.
ChangeActivePlotsVar("d")
AddOperator("Threshold")
t = ThresholdAttributes()
t.lowerBounds = (0.1)
t.listedVarNames = ("cmfe")
SetOperatorOptions(t)
DrawPlots()
Test("pos_cmfe_03")

# Test using cmfe in another expression.
DefineVectorExpression("disp", "{0, 3*cmfe}")
RemoveLastOperator()
d = DisplaceAttributes()
d.variable = "disp"
SetDefaultOperatorOptions(d)
AddOperator("Displace")
DrawPlots()
Test("pos_cmfe_04")

RemoveLastOperator()
ChangeActivePlotsVar("cmfe")
DrawPlots()

Query("MinMax")
t = GetQueryOutputString()
TestText("pos_cmfe_05", t)

# Now test error conditions.
DefineScalarExpression("cmfe2", "pos_cmfe(pressure, quadmesh)")
ChangeActivePlotsVar("cmfe2")
DrawPlots()
t = GetLastError()
TestText("pos_cmfe_06", t)

DefineScalarExpression("cmfe3", "pos_cmfe(<bad_file.silo:a_var>, curvmesh2d, 0.)")
ChangeActivePlotsVar("cmfe3")
DrawPlots()
t = GetLastError()
TestText("pos_cmfe_07", t)

DefineScalarExpression("cmfe4", "pos_cmfe(<%s:pressure>, curvmesh2d)" %  cmfe_silo_data_path("wave0100.silo"))
ChangeActivePlotsVar("cmfe4")
DrawPlots()
t = GetLastError()
TestText("pos_cmfe_08", t)

DefineScalarExpression("cmfe5", "pos_cmfe(<%s:pressure>)" % cmfe_silo_data_path("wave0100.silo"))
ChangeActivePlotsVar("cmfe5")
DrawPlots()
t = GetLastError()
TestText("pos_cmfe_09", t)

# Invalid variable in new database.
DefineScalarExpression("cmfe6", "pos_cmfe(<%s:xyz>, curvmesh2d, 0.)" % cmfe_silo_data_path("ucd2d.silo"))
ChangeActivePlotsVar("cmfe6")
DrawPlots()
t = GetLastError()
TestText("pos_cmfe_10", t)

# Enough error checking, let's do some more testing when it works.

# 3D, multi-block to multi-block.
DeleteAllPlots()
OpenDatabase(silo_data_path("multi_ucd3d.silo"))

DefineScalarExpression("cmfe7", "pos_cmfe(<%s:d>, mesh1, 0.)" %  cmfe_silo_data_path("multi_curv3d.silo"))
AddPlot("Pseudocolor", "cmfe7")
DrawPlots()
Test("pos_cmfe_11")

DeleteAllPlots()

# multi-block to single-block.
DeleteAllPlots()
OpenDatabase(silo_data_path("curv3d.silo"))

DefineScalarExpression("cmfe8", "pos_cmfe(<%s:d>, curvmesh3d, 0.)" %  cmfe_silo_data_path("multi_curv3d.silo"))
AddPlot("Pseudocolor", "cmfe8")
DrawPlots()
Test("pos_cmfe_12")

# single-block to multi-block.
DeleteAllPlots()
OpenDatabase(silo_data_path("multi_curv3d.silo"))

DefineScalarExpression("cmfe9", "pos_cmfe(<%s:d>, mesh1, 0.)" % cmfe_silo_data_path("curv3d.silo"))
AddPlot("Pseudocolor", "cmfe9")
DrawPlots()
Test("pos_cmfe_13")

# single-block to multi-block -- vector
DeleteAllPlots()
OpenDatabase(silo_data_path("multi_ucd3d.silo"))

DefineVectorExpression("cmfe10", "pos_cmfe(gradient(<%s:d>), mesh1, {0,0,0})"% cmfe_silo_data_path("curv3d.silo"))
AddPlot("Vector", "cmfe10")
vec = VectorAttributes()
vec.nVectors = 400*36
SetPlotOptions(vec)
DrawPlots()
Test("pos_cmfe_14")

Exit()