def makeCorrectTree(vertexsize, cellsize, ntris):
T = CGL.newCGNSTree()
b = CGL.newBase(T, 'Base', 3, 3)
s = NPY.array([[vertexsize, cellsize, 0]], dtype='int32', order='F')
z = CGL.newZone(b, 'Zone', s, CGK.Unstructured_s)
g = CGL.newGridCoordinates(z, 'GridCoordinates')
d = CGL.newDataArray(g, CGK.CoordinateX_s,
NPY.ones((vertexsize), dtype='float64', order='F'))
d = CGL.newDataArray(g, CGK.CoordinateY_s,
NPY.ones((vertexsize), dtype='float64', order='F'))
d = CGL.newDataArray(g, CGK.CoordinateZ_s,
NPY.ones((vertexsize), dtype='float64', order='F'))
tetras = CGL.newElements(z, 'TETRAS', CGK.TETRA_4_s,
NPY.ones((cellsize * 4), dtype='int32'),
NPY.array([[1, cellsize]], 'i', order='F'))
tris = CGL.newElements(
z, 'TRIS', CGK.TRI_3_s, NPY.ones((ntris * 3), dtype='int32'),
NPY.array([[cellsize + 1, cellsize + ntris]], 'i', order='F'))
zbc = CGL.newZoneBC(z)
n = CGL.newBoundary(zbc,
'BC', [range(cellsize + 1, cellsize + ntris + 1)],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointList_s)
g = CGL.newGridLocation(n, value=CGK.FaceCenter_s)
return (T, b, z, zbc, n, g)
TESTS.append((tag, T, diag))
# -------------------------------------------------------------------------
tag = 'indexrange on BC_t'
diag = True
(T, b, z) = makeCorrectTree(vertexsize, cellsize)
tetras = CGL.newElements(z, 'TETRAS', CGK.TETRA_4_s,
NPY.ones((cellsize * 4), dtype='int32'),
NPY.array([[1, cellsize]], 'i', order='F'))
ntris = 11
tris = CGL.newElements(
z, 'TRIS', CGK.TRI_3_s, NPY.ones((ntris * 3), dtype='int32'),
NPY.array([[cellsize + 1, cellsize + ntris]], 'i', order='F'))
n = CGL.newBoundary(z,
'BC', [range(cellsize + 1, cellsize + ntris + 1)],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointList_s)
g = CGL.newGridLocation(n, value=CGK.FaceCenter_s)
CGU.removeChildByName(n, CGK.PointList_s)
CGU.newNode(CGK.PointRange_s,
NPY.array([[cellsize + 1, cellsize + ntris]],
dtype=NPY.int32,
order='F'), [],
CGK.IndexRange_ts,
parent=n)
TESTS.append((tag, T, diag))
# -------------------------------------------------------------------------
tag = 'indexrange on BC_t PointRange index out of range #1'
diag = False
vertexsize = 20
cellsize = 7
ntris = 12
(T, b, z, zbc, n, g) = makeCorrectTree(vertexsize, cellsize, ntris)
TESTS.append((tag, T, diag))
# -------------------------------------------------------------------------
tag = 'bc bad location'
diag = False
(T, b, z, zbc, n, g) = makeCorrectTree(vertexsize, cellsize, ntris)
CGL.newPointRange(n, value=NPY.array([[1, cellsize]], 'i'))
zbc[2] = []
n = CGL.newBoundary(zbc,
'BC', [range(1, cellsize + 1)],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointList_s)
g = CGL.newGridLocation(n, value=CGK.CellCenter_s)
TESTS.append((tag, T, diag))
# -------------------------------------------------------------------------
tag = 'bc both PointList and PointRange'
diag = False
(T, b, z, zbc, n, g) = makeCorrectTree(vertexsize, cellsize, ntris)
CGL.newPointRange(n, value=NPY.array([[cellsize + 1, cellsize + ntris]], 'i'))
TESTS.append((tag, T, diag))
# -------------------------------------------------------------------------
tag = 'bc no PointList or PointRange'
diag = False
NPY.ones((vertexsize), dtype='float64', order='F'))
d = CGL.newDataArray(g, CGK.CoordinateZ_s,
NPY.ones((vertexsize), dtype='float64', order='F'))
tetras = CGL.newElements(z, 'TETRAS', CGK.TETRA_4_s,
NPY.ones((cellsize * 4), dtype='int32'),
NPY.array([[1, cellsize]], 'i', order='F'))
zbc = CGL.newZoneBC(z)
return (T, b, z, zbc)
vertexsize = 20
cellsize = 7
(T, b, z, zbc) = makeCorrectTree(vertexsize, cellsize)
n = CGL.newBoundary(zbc,
'BC', [range(1, cellsize + 1)],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointList_s)
g = CGL.newGridLocation(n, value=CGK.CellCenter_s)
TESTS.append((tag, T, diag))
# -------------------------------------------------------------------------
tag = 'indexarray bad parent'
diag = False
(T, b, z, zbc) = makeCorrectTree(vertexsize, cellsize)
n = CGL.newBoundary(zbc,
'BC', [range(1, cellsize + 1)],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointList_s)
g = CGL.newGridLocation(n, value=CGK.CellCenter_s)
import CGNS.PAT.cgnslib as C
import CGNS.PAT.cgnserrors as E
import CGNS.PAT.cgnskeywords as K
import numpy as N
data = C.newZone(None, '{Zone}',
N.array([[5, 4, 0], [7, 7, 0], [9, 8, 0]], order='F'))
g1 = C.newGridCoordinates(data, "GridCoordinates")
C.newRigidGridMotion(data, "{RigidGridMotion}")
C.newArbitraryGridMotion(data, "{ArbitraryGridMotion}")
C.newFlowSolution(data, "{FlowSolution}")
C.newDiscreteData(data, "{DiscreteData}")
C.newIntegralData(data, "{IntegralData}")
C.newZoneGridConnectivity(data, "{GridConnectivity}")
C.newBoundary(data, "{BC}", N.array([[0, 0, 0], [0, 0, 0]]))
C.newZoneIterativeData(data, "{ZoneIterativeData}")
C.newReferenceState(data)
C.newRotatingCoordinates(data)
C.newDataClass(data)
C.newDimensionalUnits(data)
C.newFlowEquationSet(data)
C.newConvergenceHistory(data, K.ZoneConvergenceHistory_s)
C.newUserDefinedData(data, '{UserDefinedData}')
C.newDescriptor(data, '{Descriptor}')
C.newOrdinal(data)
status = '6.3'
comment = 'Full SIDS with all optionals'
pattern = [data, status, comment]
NPY.ones((cellsize * 4), dtype='int32'),
NPY.array([[1, cellsize]], 'i', order='F'))
TESTS.append((tag, T, diag))
# -------------------------------------------------------------------------
tag = 'elements bad elementsizeboundary'
diag = False
(T, b, z) = makeCorrectTree(vertexsize, cellsize)
tetras = CGL.newElements(z, 'TETRAS', CGK.TETRA_4_s,
NPY.ones((cellsize * 4), dtype='int32'),
NPY.array([[1, cellsize]], 'i', order='F'))
tetras[1][1] = cellsize
zbc = CGL.newZoneBC(z)
n = CGL.newBoundary(zbc,
'BC', [range(1, cellsize + 1)],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointList_s)
g = CGL.newGridLocation(n, value=CGK.CellCenter_s)
TESTS.append((tag, T, diag))
# -------------------------------------------------------------------------
tag = 'elements elementsizeboundary BC correctly defined'
diag = True
(T, b, z) = makeCorrectTree(vertexsize, cellsize)
tetras = CGL.newElements(z, 'TETRAS', CGK.TETRA_4_s,
NPY.ones((cellsize * 4), dtype='int32'),
NPY.array([[1, cellsize]], 'i', order='F'))
tetras[1][1] = cellsize - 1
zbc = CGL.newZoneBC(z)
n = CGL.newBoundary(zbc,
def makeStTree():
T = CGL.newCGNSTree()
b = CGL.newBase(T, '{Base}', 3, 3)
z1 = CGL.newZone(b, '{Zone1}',
NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F'))
g = CGL.newGridCoordinates(z1, 'GridCoordinates')
d = CGL.newDataArray(g, CGK.CoordinateX_s,
NPY.ones((5, 7, 9), dtype='float64', order='F'))
d = CGL.newDataArray(g, CGK.CoordinateY_s,
NPY.ones((5, 7, 9), dtype='float64', order='F'))
d = CGL.newDataArray(g, CGK.CoordinateZ_s,
NPY.ones((5, 7, 9), dtype='float64', order='F'))
z2 = CGU.copyNode(z1, '{Zone2}')
b[2].append(z2)
zgc = CGL.newZoneGridConnectivity(z1)
gc = CGL.newGridConnectivity1to1(zgc, 'join1_2', '{Zone2}',
NPY.array([[1, 1], [1, 4], [1, 9]]),
NPY.array([[5, 5], [3, 7], [1, 9]]),
NPY.array([-1, +2, +3]))
zgc = CGL.newZoneGridConnectivity(z2)
gc = CGL.newGridConnectivity1to1(zgc, 'join2_1', '{Zone1}',
NPY.array([[5, 5], [3, 7], [1, 9]]),
NPY.array([[1, 1], [1, 4], [1, 9]]),
NPY.array([-1, +2, +3]))
zbc = CGL.newZoneBC(z1)
n = CGL.newBoundary(zbc,
'{BC1_1}', [[5, 5], [1, 7], [1, 9]],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointRange_s)
g = CGL.newGridLocation(n, value=CGK.FaceCenter_s)
n = CGL.newBoundary(zbc,
'{BC1_2}', [[1, 5], [1, 1], [1, 9]],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointRange_s)
g = CGL.newGridLocation(n, value=CGK.FaceCenter_s)
n = CGL.newBoundary(zbc,
'{BC1_3}', [[1, 5], [7, 7], [1, 9]],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointRange_s)
g = CGL.newGridLocation(n, value=CGK.FaceCenter_s)
n = CGL.newBoundary(zbc,
'{BC1_4}', [[1, 5], [1, 7], [1, 1]],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointRange_s)
g = CGL.newGridLocation(n, value=CGK.FaceCenter_s)
n = CGL.newBoundary(zbc,
'{BC1_5}', [[1, 5], [1, 7], [9, 9]],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointRange_s)
g = CGL.newGridLocation(n, value=CGK.FaceCenter_s)
n = CGL.newBoundary(zbc,
'{BC1_6}', [[1, 1], [4, 7], [1, 9]],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointRange_s)
g = CGL.newGridLocation(n, value=CGK.FaceCenter_s)
zbc = CGL.newZoneBC(z2)
n = CGL.newBoundary(zbc,
'{BC2_1}', [[1, 1], [1, 7], [1, 9]],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointRange_s)
g = CGL.newGridLocation(n, value=CGK.FaceCenter_s)
n = CGL.newBoundary(zbc,
'{BC2_2}', [[1, 5], [1, 1], [1, 9]],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointRange_s)
g = CGL.newGridLocation(n, value=CGK.FaceCenter_s)
n = CGL.newBoundary(zbc,
'{BC2_3}', [[1, 5], [7, 7], [1, 9]],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointRange_s)
g = CGL.newGridLocation(n, value=CGK.FaceCenter_s)
n = CGL.newBoundary(zbc,
'{BC2_4}', [[1, 5], [1, 7], [1, 1]],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointRange_s)
g = CGL.newGridLocation(n, value=CGK.FaceCenter_s)
n = CGL.newBoundary(zbc,
'{BC2_5}', [[1, 5], [1, 7], [9, 9]],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointRange_s)
g = CGL.newGridLocation(n, value=CGK.FaceCenter_s)
n = CGL.newBoundary(zbc,
'{BC2_6}', [[5, 5], [1, 3], [1, 9]],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointRange_s)
g = CGL.newGridLocation(n, value=CGK.FaceCenter_s)
z = [z1, z2]
return (T, b, z)
pth = CGU.getAllNodesByTypeOrNameList(z[0],
['Zone_t', 'ZoneBC_t', '{BC1_1}'])[0]
CGU.removeChildByName(CGU.getNodeByPath(z[0], CGU.getPathAncestor(pth)),
'{BC1_1}')
pth = CGU.getAllNodesByTypeOrNameList(
z[1], ['Zone_t', 'ZoneGridConnectivity_t', 'join2_1'])[0]
CGU.removeChildByName(CGU.getNodeByPath(z[1], CGU.getPathAncestor(pth)),
'join2_1')
TESTS.append((tag, T, diag))
# -------------------------------------------------------------------------
tag = 'zone structured doubly defined BC and GridConnect (warning)'
diag = False
(T, b, z) = makeStTree()
pth = CGU.getAllNodesByTypeOrNameList(z[0], ['Zone_t', 'ZoneBC_t'])[0]
zbc = CGU.getNodeByPath(z[0], pth)
n = CGL.newBoundary(zbc,
'{BC1_1b}', [[5, 5], [1, 2], [1, 2]],
btype=CGK.Null_s,
family=None,
pttype=CGK.PointRange_s)
g = CGL.newGridLocation(n, value=CGK.FaceCenter_s)
pth = CGU.getAllNodesByTypeOrNameList(z[1],
['Zone_t', 'ZoneGridConnectivity_t'])[0]
zgc = CGU.getNodeByPath(z[1], pth)
gc = CGL.newGridConnectivity1to1(zgc, 'join2_1b', '{Zone1}',
NPY.array([[1, 1], [1, 2], [1, 2]]),
NPY.array([[1, 1], [1, 4], [1, 9]]),
NPY.array([-1, +2, +3]))
TESTS.append((tag, T, diag))
