8 class SMESHExportOfFieldsInMemory(unittest.TestCase):
10 def testMEDCouplingFieldOnCells(self):
12 Test focuses on ExportMEDCoupling method in the context of MED_CELL field output.
20 from salome.shaper import model
23 partSet = model.moduleDocument()
26 Part_1 = model.addPart(partSet)
27 Part_1_doc = Part_1.document()
30 Box_1 = model.addBox(Part_1_doc, 10, 10, 10)
33 Plane_4 = model.addPlane(Part_1_doc, model.selection("FACE", "PartSet/XOY"), 5, False)
36 Plane_5 = model.addPlane(Part_1_doc, model.selection("FACE", "PartSet/YOZ"), 5, False)
39 Partition_1_objects = [model.selection("FACE", "Plane_1"),
40 model.selection("FACE", "Plane_2"),
41 model.selection("SOLID", "Box_1_1")]
42 Partition_1 = model.addPartition(Part_1_doc, Partition_1_objects, keepSubResults = True)
45 Field_1_objects = [model.selection("SOLID", "Partition_1_1_2"),
46 model.selection("SOLID", "Partition_1_1_4"),
47 model.selection("SOLID", "Partition_1_1_1"),
48 model.selection("SOLID", "Partition_1_1_3")]
49 Field_1 = model.addField(Part_1_doc, 1, "DOUBLE", 1, ["Comp 1"], Field_1_objects)
50 Field_1.addStep(0, 0, [[0], [1], [2], [3], [4]])
56 ### SHAPERSTUDY component
59 model.publishToShaperStudy()
61 Partition_1_1, Field_1_1 = SHAPERSTUDY.shape(model.featureStringId(Partition_1))
66 import SMESH, SALOMEDS
67 from salome.smesh import smeshBuilder
69 smesh = smeshBuilder.New()
70 #smesh.SetEnablePublish( False ) # Set to False to avoid publish in study if not needed or in some particular situations:
71 # multiples meshes built in parallel, complex and numerous mesh edition (performance)
73 Mesh_1 = smesh.Mesh(Partition_1_1)
74 Regular_1D = Mesh_1.Segment()
75 Local_Length_1 = Regular_1D.LocalLength(5,None,1e-07)
76 Quadrangle_2D = Mesh_1.Quadrangle(algo=smeshBuilder.QUADRANGLE)
77 Hexa_3D = Mesh_1.Hexahedron(algo=smeshBuilder.Hexa)
78 isDone = Mesh_1.Compute()
80 raise Exception("Error when computing Mesh")
81 smesh.SetName(Mesh_1, 'Mesh_1')
83 #### Mesh_1.ExportMED( r'Mesh_with_one_field_on_cells.med', 0, 41, 1, Mesh_1.GetMesh(), 1, [ Field_1_1 ], '',-1 )
84 mfd = Mesh_1.ExportMEDCoupling(0, Mesh_1.GetMesh(), 1, [ Field_1_1 ], '',-1 )#### <- important line of test is here !
86 self.assertEqual(len(mfd.getMeshes()),1)
87 self.assertEqual(len(mfd.getFields()),1)
88 f = mfd.getFields()[0][0].field(mfd.getMeshes()[0])
89 f.checkConsistencyLight()
91 self.assertEqual(f.getDiscretization().getEnum(),medcoupling.ON_CELLS)
92 self.assertTrue(f.getMesh().getNumberOfCells()>1)
96 def testMEDCouplingFieldOnNodes(self):
98 Test focuses on ExportMEDCoupling method in the context of MED_NODES field output.
107 from salome.shaper import model
110 partSet = model.moduleDocument()
113 Part_1 = model.addPart(partSet)
114 Part_1_doc = Part_1.document()
117 Box_1 = model.addBox(Part_1_doc, 10, 10, 10)
120 Plane_4 = model.addPlane(Part_1_doc, model.selection("FACE", "PartSet/XOY"), 5, False)
123 Plane_5 = model.addPlane(Part_1_doc, model.selection("FACE", "PartSet/YOZ"), 5, False)
126 Partition_1_objects = [model.selection("FACE", "Plane_1"),
127 model.selection("FACE", "Plane_2"),
128 model.selection("SOLID", "Box_1_1")]
129 Partition_1 = model.addPartition(Part_1_doc, Partition_1_objects, keepSubResults = True)
132 Field_2_objects = [model.selection("VERTEX", "[Partition_1_1_2/Modified_Face&Box_1_1/Back][Partition_1_1_2/Modified_Face&Box_1_1/Left][Partition_1_1_2/Modified_Face&Box_1_1/Top]"),
133 model.selection("VERTEX", "Partition_1_1_2/Generated_Vertex&Plane_2/Plane_2&new_weak_name_1"),
134 model.selection("VERTEX", "[Partition_1_1_4/Modified_Face&Box_1_1/Front][Partition_1_1_4/Modified_Face&Box_1_1/Left][Partition_1_1_4/Modified_Face&Box_1_1/Top]"),
135 model.selection("VERTEX", "Partition_1_1_1/Generated_Vertex&Plane_1/Plane_1&new_weak_name_1"),
136 model.selection("VERTEX", "[Partition_1_1_1/Modified_Face&Box_1_1/Left][Partition_1_1_1/Modified_Face&Plane_1/Plane_1][Partition_1_1_1/Modified_Face&Plane_2/Plane_2]"),
137 model.selection("VERTEX", "Partition_1_1_3/Generated_Vertex&Plane_1/Plane_1&new_weak_name_1"),
138 model.selection("VERTEX", "[Partition_1_1_1/Modified_Face&Box_1_1/Back][Partition_1_1_1/Modified_Face&Box_1_1/Left][Partition_1_1_1/Modified_Face&Box_1_1/Bottom]"),
139 model.selection("VERTEX", "Partition_1_1_1/Generated_Vertex&Plane_2/Plane_2&new_weak_name_1"),
140 model.selection("VERTEX", "[Partition_1_1_3/Modified_Face&Box_1_1/Left][Partition_1_1_3/Modified_Face&Box_1_1/Bottom][Partition_1_1_3/Modified_Face&Box_1_1/Front]")]
141 Field_2 = model.addField(Part_1_doc, 1, "DOUBLE", 1, ["Comp 1"], Field_2_objects)
142 Field_2.addStep(0, 0, [[0], [1], [2], [3], [4], [5], [6], [7], [8], [9]])
147 ### SHAPERSTUDY component
150 model.publishToShaperStudy()
152 Partition_1_1, Field_2_1 = SHAPERSTUDY.shape(model.featureStringId(Partition_1))
157 import SMESH, SALOMEDS
158 from salome.smesh import smeshBuilder
160 smesh = smeshBuilder.New()
161 #smesh.SetEnablePublish( False ) # Set to False to avoid publish in study if not needed or in some particular situations:
162 # multiples meshes built in parallel, complex and numerous mesh edition (performance)
164 Mesh_1 = smesh.Mesh(Partition_1_1)
165 Regular_1D = Mesh_1.Segment()
166 Local_Length_1 = Regular_1D.LocalLength(5,None,1e-07)
167 Quadrangle_2D = Mesh_1.Quadrangle(algo=smeshBuilder.QUADRANGLE)
168 Hexa_3D = Mesh_1.Hexahedron(algo=smeshBuilder.Hexa)
169 isDone = Mesh_1.Compute()
171 raise Exception("Error when computing Mesh")
172 smesh.SetName(Mesh_1, 'Mesh_1')
174 # 23th of june 2021 : Bug both in ExportMED and in ExportMEDCoupling
175 #Mesh_1.ExportMED( r'/tmp/Mesh_with_one_field_on_nodes.med', 0, 41, 1, Mesh_1.GetMesh(), 1, [ Field_2_1 ], '',-1 )
176 #mfd = Mesh_1.ExportMEDCoupling(0,Mesh_1.GetMesh(), 1, [ Field_2_1 ], '',-1)
178 if __name__ == '__main__':