def makeCorrectTree(): 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')) s = NPY.array([[vertexsize, cellsize, 0]], dtype='int32', order='F') z2 = CGL.newZone(b, 'Zone2', s, CGK.Unstructured_s) g = CGL.newGridCoordinates(z2, '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(z2, 'TETRAS', CGK.TETRA_4_s, NPY.ones((cellsize * 4), dtype='int32'), NPY.array([[1, cellsize]], 'i', order='F')) tris = CGL.newElements( z2, 'TRIS', CGK.TRI_3_s, NPY.ones((ntris * 3), dtype='int32'), NPY.array([[cellsize + 1, cellsize + ntris]], 'i', order='F')) z3 = CGU.copyNode(z1, 'Zone3') b[2].append(z3) z4 = CGU.copyNode(z2, 'Zone4') b[2].append(z4) z = [z1, z2, z3, z4] return (T, b, z)
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)
def genTrees(): tree = CGL.newCGNSTree() b = CGL.newBase(tree, '{Base}', 2, 3) z = CGL.newZone(b, '{Zone}', numpy.array([[5, 4, 0], [7, 6, 0]], order='F')) g = CGL.newGridCoordinates(z, 'GridCoordinates') d = CGL.newDataArray(g, CGK.CoordinateX_s, numpy.ones((5, 7), dtype='d', order='F')) d = CGL.newDataArray(g, CGK.CoordinateY_s, numpy.ones((5, 7), dtype='d', order='F')) d = CGL.newDataArray(g, CGK.CoordinateZ_s, numpy.ones((5, 7), dtype='d', order='F')) return (tree,)
def makeStTree(vertexsize, cellsize): T = CGL.newCGNSTree() b = CGL.newBase(T, '{Base}', 3, 3) z = CGL.newZone(b, '{Zone}', NPY.array( [[vertexsize[0], cellsize[0], 0], [vertexsize[1], cellsize[1], 0], [vertexsize[2], cellsize[2], 0]], order='F')) g = CGL.newGridCoordinates(z, 'GridCoordinates') d = CGL.newDataArray(g, CGK.CoordinateX_s, NPY.ones(tuple(vertexsize), dtype='float64', order='F')) d = CGL.newDataArray(g, CGK.CoordinateY_s, NPY.ones(tuple(vertexsize), dtype='float64', order='F')) d = CGL.newDataArray(g, CGK.CoordinateZ_s, NPY.ones(tuple(vertexsize), dtype='float64', order='F')) return (T, b, z)
def makeUnstTree(vertexsize, cellsize): 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')) return (T, b, z)
def makeCorrectTree(vertexsize, cellsize): 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')) return (T, b, z)
def makeCorrectTree(): T = CGL.newCGNSTree() b = CGL.newBase(T, 'Base', 3, 3) z1 = CGL.newZone(b, 'Zone1', NPY.array([[5, 4, 0], [7, 6, 0], [5, 4, 0]], order='F')) g = CGL.newGridCoordinates(z1, 'GridCoordinates') d = CGL.newDataArray(g, CGK.CoordinateX_s, NPY.ones((5, 7, 5), dtype='float64', order='F')) d = CGL.newDataArray(g, CGK.CoordinateY_s, NPY.ones((5, 7, 5), dtype='float64', order='F')) d = CGL.newDataArray(g, CGK.CoordinateZ_s, NPY.ones((5, 7, 5), dtype='float64', order='F')) s = NPY.array([[vertexsize, cellsize, 0]], dtype='int32', order='F') z2 = CGU.copyNode(z1, 'Zone2') b[2].append(z2) z = [z1, z2] return (T, b, z)
#!/usr/bin/env python # ------------------------------------------------------------------------- # pyCGNS.PAT - CFD General Notation System - # See license.txt file in the root directory of this Python module source # ------------------------------------------------------------------------- # from __future__ import print_function import CGNS.PAT.cgnslib as CGL import CGNS.PAT.cgnsutils as CGU T = CGL.newCGNSTree() B = CGL.newBase(T, 'Base', 3, 3) Z = CGL.newZone(B, 'Zone1') Z = CGL.newZone(B, 'Zone2') Z = CGL.newZone(B, 'Zone3') Z = CGL.newZone(B, 'Zone4') l1 = CGU.getAllNodesByTypeList(T, ['CGNSTree_t', 'CGNSBase_t', 'Zone_t']) l2 = CGU.getAllNodesByTypeSet(T, ['Zone_t', 'IndexArray_t']) print(l1) print(l2) print(CGU.getNodeByPath(l1[0], T)) l3 = CGU.getAllNodesByTypeList(B, ['CGNSBase_t', 'Zone_t']) l4 = CGU.getAllNodesByTypeSet(B, ['Zone_t']) print(l3) print(l4)
def test(): T1 = CGL.newCGNSTree() b1 = CGL.newCGNSBase(T1, 'B1', 3, 3) b2 = CGL.newCGNSBase(T1, 'B2', 3, 3) f1 = CGL.newFamily(b1, 'F1') f2 = CGL.newFamily(b1, 'F2') f1 = CGL.newFamily(b2, 'F1') f2 = CGL.newFamily(b2, 'F2') f3 = CGL.newFamily(b2, 'F3') f4 = CGL.newFamily(b2, 'F4') z1 = CGL.newZone(b1, 'Z1', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) z2 = CGL.newZone(b1, 'Z2', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) z3 = CGL.newZone(b1, 'Z3', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) z4 = CGL.newZone(b1, 'Z4', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) z5 = CGL.newZone(b1, 'Z5', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) z1 = CGL.newZone(b2, 'Z1', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) z2 = CGL.newZone(b2, 'Z2', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) z3 = CGL.newZone(b2, 'Z3', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) T2 = CGL.newCGNSTree() b1 = CGL.newCGNSBase(T2, 'B1', 3, 3) b2 = CGL.newCGNSBase(T2, 'B2', 3, 3) b3 = CGL.newCGNSBase(T2, 'B3', 2, 2) f1 = CGL.newFamily(b1, 'F1') f1 = CGL.newFamily(b2, 'F1') f2 = CGL.newFamily(b2, 'F2') f3 = CGL.newFamily(b2, 'F3') f4 = CGL.newFamily(b2, 'F4') f5 = CGL.newFamily(b2, 'F5') f1 = CGL.newFamily(b3, 'F1') f2 = CGL.newFamily(b3, 'F2') z1 = CGL.newZone(b1, 'Z1', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) z2 = CGL.newZone(b1, 'Z2', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) z3 = CGL.newZone(b1, 'Z3', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) z1 = CGL.newZone(b2, 'Z1', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) z2 = CGL.newZone(b2, 'Z2', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) z3 = CGL.newZone(b2, 'Z3', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) z4 = CGL.newZone(b2, 'Z4', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) T3 = CGL.newCGNSTree() b1 = CGL.newCGNSBase(T3, 'B1', 3, 3) b2 = CGL.newCGNSBase(T3, 'B7', 3, 3) b3 = CGL.newCGNSBase(T3, 'B8', 2, 2) f1 = CGL.newFamily(b1, 'F1') f1 = CGL.newFamily(b2, 'F8') Tlist = [] for TT in [T1, T2, T3]: Tlist.append(TT) count = 0 for t in Tlist: CGM.save('merge-T%.2d.hdf' % count, t, flags=CGM.S2P_TRACE) count += 1 T = mergeTrees(Tlist) CGM.save('merge-Result.hdf', T, flags=CGM.S2P_TRACE) return T
# --------------------------------------------------------------------------- # pyCGNS - Python package for CFD General Notation System - # See license.txt file in the root directory of this Python module source # --------------------------------------------------------------------------- # from __future__ import unicode_literals 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}')
tag = 'base dataclass #3' diag = False T = CGL.newCGNSTree() b = CGL.newBase(T, '{Base}', 3, 3) d = CGL.newDataClass(b, CGK.NormalizedByDimensional_s) d[1] = CGU.setStringAsArray("NormalizedByDimensionnal") TESTS.append((tag, T, diag)) # ------------------------------------------------------------------------- tag = 'all levels dataclass' diag = True T = CGL.newCGNSTree() b = CGL.newBase(T, '{Base}', 3, 3) d = CGL.newFamily(b, '{Family}') d = CGL.newDataClass(b, CGK.NondimensionalParameter_s) z = CGL.newZone(b, '{Zone}', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) d = CGL.newDataClass(z, CGK.NondimensionalParameter_s) g = CGL.newGridCoordinates(z, CGK.GridCoordinates_s) w = CGL.newDataArray(g, CGK.CoordinateX_s, NPY.ones((5, 7, 9), dtype='float64', order='F')) w = CGL.newDataArray(g, CGK.CoordinateY_s, NPY.ones((5, 7, 9), dtype='float64', order='F')) w = CGL.newDataArray(g, CGK.CoordinateZ_s, NPY.ones((5, 7, 9), dtype='float64', order='F')) d = CGL.newDataClass(g, CGK.NondimensionalParameter_s) f = CGL.newFlowSolution(z) d = CGL.newDataClass(f, CGK.NondimensionalParameter_s) a = CGL.newDataArray(f, '{DataArray}', value=NPY.ones((4, 6, 8), dtype='float64', order='F')) d = CGL.newDataClass(a, CGK.NondimensionalParameter_s)
# --------------------------------------------------------------------------- # pyCGNS - Python package for CFD General Notation System - # See license.txt file in the root directory of this Python module source # --------------------------------------------------------------------------- # import CGNS.PAT.cgnslib as C import CGNS.PAT.cgnserrors as E import CGNS.PAT.cgnskeywords as K import numpy as N data = C.newBase(None, '{Base}', 3, 3) C.newZone(data, '{Zone}', N.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) C.newSimulationType(data) C.newIntegralData(data, '{IntegralData}') C.newBaseIterativeData(data, '{BaseIterativeData}') C.newConvergenceHistory(data) C.newFamily(data, '{Family}') C.newFlowEquationSet(data) C.newReferenceState(data) C.newAxisymmetry(data) C.newRotatingCoordinates(data) C.newGravity(data) C.newDataClass(data) C.newDimensionalUnits(data) C.newUserDefinedData(data, '{UserDefinedData}') C.newDescriptor(data, '{Descriptor}') status = '6.2' comment = 'Full SIDS with all optionals children' pattern = [data, status, comment]
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)
# See license.txt file in the root directory of this Python module source # ------------------------------------------------------------------------- # import CGNS.PAT.cgnslib as CGL import CGNS.PAT.cgnsutils as CGU import CGNS.PAT.cgnskeywords as CGK import numpy as NPY TESTS = [] # ------------------------------------------------------------------------- tag = 'zone structured' diag = True T = CGL.newCGNSTree() b = CGL.newBase(T, '{Base}', 3, 3) z = CGL.newZone(b, '{Zone}', NPY.array([[5, 4, 0], [7, 6, 0], [9, 8, 0]], order='F')) g = CGL.newGridCoordinates(z, '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')) TESTS.append((tag, T, diag)) # ------------------------------------------------------------------------- tag = 'zone bad zonetype' diag = False T = CGL.newCGNSTree() b = CGL.newBase(T, '{Base}', 3, 3) z = CGL.newZone(b, '{Zone}',
p_arr = var_mat[:,8] # For now, skip the elem connectivity section. Because I do not need it. ############################## # Write to CGNS file ############################## import CGNS.PAT.cgnsutils as CGU import CGNS.PAT.cgnslib as CGL import CGNS.PAT.cgnskeywords as CK import CGNS.MAP as CGM T=CGL.newCGNSTree() B=CGL.newBase(T,'hpMusic_base',2,2) # The shape (3,1) is critical zone_size = np.array([[n_node, n_elem, 0]]) Z=CGL.newZone(B,'Solution',zone_size,CK.Unstructured_s,'') GC=CGL.newGridCoordinates(Z,name='GridCoordinates') FS=CGL.newFlowSolution(Z,name='FlowSolution',gridlocation='Vertex') GL=CGU.getNodeByPath(FS,'GridLocation') CGU.nodeDelete(FS,GL) coordinatex_node = CGL.newDataArray(GC,'CoordinateX',value=x_arr) coordinatey_node = CGL.newDataArray(GC,'CoordinateY',value=y_arr) density_node = CGL.newDataArray(FS,'Density',value=rho_arr) velocityx_node = CGL.newDataArray(FS,'VelocityX',value=u_arr) velocityy_node = CGL.newDataArray(FS,'VelocityY',value=v_arr) p_node = CGL.newDataArray(FS,'Pressure',value=p_arr) CGM.save("sol_tec.cgns",T)