Salome HOME
Adding Multinode method for smesh parallelism + cleanup and doc
[modules/smesh.git] / doc / examples / creating_parallel_mesh.py
1 # contains function to compute a mesh in parallel
2 import salome
3
4 salome.salome_init()
5 import salome_notebook
6 notebook = salome_notebook.NoteBook()
7
8 ###
9 ### GEOM component
10 ###
11
12 import GEOM
13 from salome.geom import geomBuilder
14 from salome.smesh import smeshBuilder
15 import SALOMEDS
16
17 import numpy as np
18
19 geompy = geomBuilder.New()
20
21
22 nbox = 2
23 boxsize = 100
24 offset = 0
25 # Create 3D faces
26 boxes = []
27 # First creating all the boxes
28 for i in range(nbox):
29     for j in range(nbox):
30         for k in range(nbox):
31
32             x_orig = i*(boxsize+offset)
33             y_orig = j*(boxsize+offset)
34             z_orig = k*(boxsize+offset)
35
36             tmp_box = geompy.MakeBoxDXDYDZ(boxsize, boxsize, boxsize)
37
38             if not i == j == k == 0:
39                 box = geompy.MakeTranslation(tmp_box, x_orig,
40                                              y_orig, z_orig)
41             else:
42                 box = tmp_box
43
44             geompy.addToStudy(box, 'box_{}:{}:{}'.format(i, j, k))
45
46             boxes.append(box)
47
48 # Create fuse of all boxes
49 all_boxes = geompy.MakeCompound(boxes)
50 geompy.addToStudy(all_boxes, 'Compound_1')
51
52 # Removing duplicates faces and edges
53 all_boxes = geompy.MakeGlueFaces(all_boxes, 1e-07)
54 geompy.addToStudy(all_boxes, 'Glued_Faces_1')
55
56 rubik_cube = geompy.MakeGlueEdges(all_boxes, 1e-07)
57 geompy.addToStudy(all_boxes, 'rubik_cube')
58
59
60 smesh = smeshBuilder.New()
61 print("Creating Parallel Mesh")
62 par_mesh = smesh.ParallelMesh(rubik_cube, name="par_mesh")
63
64 print("Creating hypoehtesis for netgen")
65 NETGEN_3D_Parameters_1 = smesh.CreateHypothesisByAverageLength( 'NETGEN_Parameters',
66                                          'NETGENEngine', 34.641, 0 )
67 print("Adding hypothesis")
68 par_mesh.AddGlobalHypothesis(NETGEN_3D_Parameters_1)
69
70 print("Setting parallelism method")
71 par_mesh.SetParallelismMethod(smeshBuilder.MULTITHREAD)
72
73 print("Setting parallelism options")
74 param = par_mesh.GetParallelismSettings()
75 param.SetNbThreads(6)
76
77 print("Starting parallel compute")
78 is_done = par_mesh.Compute()
79 if not is_done:
80     raise Exception("Error when computing Mesh")
81
82 print("  Tetrahedron: ",  par_mesh.NbTetras())
83 print("  Triangle: ", par_mesh.NbTriangles())
84 print("  edge: ", par_mesh.NbEdges())
85
86 assert  par_mesh.NbTetras() > 0
87 assert  par_mesh.NbTriangles() > 0
88 assert  par_mesh.NbEdges() > 0