Salome HOME
MEDCoupling Memory output from SMESH engine now deal with fields too.
[modules/smesh.git] / doc / salome / examples / basic_smesh_output_with_mc_field.py
1 #!/usr/bin/env python
2
3 import sys
4 import salome
5
6 import unittest
7
8 class SMESHExportOfFieldsInMemory(unittest.TestCase):
9
10   def testMEDCouplingFieldOnCells(self):
11     """
12     Test focuses on ExportMEDCoupling method in the context of MED_CELL field output.
13     """
14     salome.standalone()
15     salome.salome_init()
16     ###
17     ### SHAPER component
18     ###
19
20     from salome.shaper import model
21
22     model.begin()
23     partSet = model.moduleDocument()
24
25     ### Create Part
26     Part_1 = model.addPart(partSet)
27     Part_1_doc = Part_1.document()
28
29     ### Create Box
30     Box_1 = model.addBox(Part_1_doc, 10, 10, 10)
31
32     ### Create Plane
33     Plane_4 = model.addPlane(Part_1_doc, model.selection("FACE", "PartSet/XOY"), 5, False)
34
35     ### Create Plane
36     Plane_5 = model.addPlane(Part_1_doc, model.selection("FACE", "PartSet/YOZ"), 5, False)
37
38     ### Create Partition
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)
43
44     ### Create Field
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]])
51
52
53     model.end()
54
55     ###
56     ### SHAPERSTUDY component
57     ###
58
59     model.publishToShaperStudy()
60     import SHAPERSTUDY
61     Partition_1_1, Field_1_1 = SHAPERSTUDY.shape(model.featureStringId(Partition_1))
62     ###
63     ### SMESH component
64     ###
65
66     import  SMESH, SALOMEDS
67     from salome.smesh import smeshBuilder
68
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)
72
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()
79     smesh.SetName(Mesh_1, 'Mesh_1')
80
81     #### Mesh_1.ExportMED( r'Mesh_with_one_field_on_cells.med', 0, 41, 1, Mesh_1.GetMesh(), 1, [ Field_1_1 ], '',-1 )
82     mfd = Mesh_1.ExportMEDCoupling(0, Mesh_1.GetMesh(), 1, [ Field_1_1 ], '',-1 )#### <- important line of test is here !
83
84     self.assertEqual(len(mfd.getMeshes()),1)
85     self.assertEqual(len(mfd.getFields()),1)
86     f = mfd.getFields()[0][0].field(mfd.getMeshes()[0])
87     f.checkConsistencyLight()
88     import medcoupling
89     self.assertEqual(f.getDiscretization().getEnum(),medcoupling.ON_CELLS)
90     self.assertTrue(f.getMesh().getNumberOfCells()>1)
91     pass
92
93
94   def testMEDCouplingFieldOnNodes(self):
95     """
96     Test focuses on ExportMEDCoupling method in the context of MED_NODES field output.
97     """
98     salome.standalone()
99     salome.salome_init()
100
101     ###
102     ### SHAPER component
103     ###
104
105     from salome.shaper import model
106
107     model.begin()
108     partSet = model.moduleDocument()
109
110     ### Create Part
111     Part_1 = model.addPart(partSet)
112     Part_1_doc = Part_1.document()
113
114     ### Create Box
115     Box_1 = model.addBox(Part_1_doc, 10, 10, 10)
116
117     ### Create Plane
118     Plane_4 = model.addPlane(Part_1_doc, model.selection("FACE", "PartSet/XOY"), 5, False)
119
120     ### Create Plane
121     Plane_5 = model.addPlane(Part_1_doc, model.selection("FACE", "PartSet/YOZ"), 5, False)
122
123     ### Create Partition
124     Partition_1_objects = [model.selection("FACE", "Plane_1"),
125                           model.selection("FACE", "Plane_2"),
126                           model.selection("SOLID", "Box_1_1")]
127     Partition_1 = model.addPartition(Part_1_doc, Partition_1_objects, keepSubResults = True)
128
129     ### Create Field
130     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]"),
131                       model.selection("VERTEX", "Partition_1_1_2/Generated_Vertex&Plane_2/Plane_2&new_weak_name_1"),
132                       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]"),
133                       model.selection("VERTEX", "Partition_1_1_1/Generated_Vertex&Plane_1/Plane_1&new_weak_name_1"),
134                       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]"),
135                       model.selection("VERTEX", "Partition_1_1_3/Generated_Vertex&Plane_1/Plane_1&new_weak_name_1"),
136                       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]"),
137                       model.selection("VERTEX", "Partition_1_1_1/Generated_Vertex&Plane_2/Plane_2&new_weak_name_1"),
138                       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]")]
139     Field_2 = model.addField(Part_1_doc, 1, "DOUBLE", 1, ["Comp 1"], Field_2_objects)
140     Field_2.addStep(0, 0, [[0], [1], [2], [3], [4], [5], [6], [7], [8], [9]])
141
142     model.end()
143
144     ###
145     ### SHAPERSTUDY component
146     ###
147
148     model.publishToShaperStudy()
149     import SHAPERSTUDY
150     Partition_1_1, Field_2_1 = SHAPERSTUDY.shape(model.featureStringId(Partition_1))
151     ###
152     ### SMESH component
153     ###
154
155     import  SMESH, SALOMEDS
156     from salome.smesh import smeshBuilder
157
158     smesh = smeshBuilder.New()
159     #smesh.SetEnablePublish( False ) # Set to False to avoid publish in study if not needed or in some particular situations:
160                                     # multiples meshes built in parallel, complex and numerous mesh edition (performance)
161
162     Mesh_1 = smesh.Mesh(Partition_1_1)
163     Regular_1D = Mesh_1.Segment()
164     Local_Length_1 = Regular_1D.LocalLength(5,None,1e-07)
165     Quadrangle_2D = Mesh_1.Quadrangle(algo=smeshBuilder.QUADRANGLE)
166     Hexa_3D = Mesh_1.Hexahedron(algo=smeshBuilder.Hexa)
167     isDone = Mesh_1.Compute()
168     smesh.SetName(Mesh_1, 'Mesh_1')
169
170     # 23th of june 2021 : Bug both in ExportMED and in ExportMEDCoupling
171     #Mesh_1.ExportMED( r'/tmp/Mesh_with_one_field_on_nodes.med', 0, 41, 1, Mesh_1.GetMesh(), 1, [ Field_2_1 ], '',-1 )
172     #mfd = Mesh_1.ExportMEDCoupling(0,Mesh_1.GetMesh(), 1, [ Field_2_1 ], '',-1)
173
174 if __name__ == '__main__':
175     unittest.main()