mirror of
https://git.salome-platform.org/gitpub/modules/smesh.git
synced 2024-11-14 09:38:33 +05:00
106 lines
2.8 KiB
Python
106 lines
2.8 KiB
Python
|
# contains function to compute a mesh in parallel
|
||
|
|
import salome
|
||
|
|
|
||
|
|
salome.salome_init()
|
||
|
|
import salome_notebook
|
||
|
|
notebook = salome_notebook.NoteBook()
|
||
|
|
|
||
|
|
###
|
||
|
|
### GEOM component
|
||
|
|
###
|
||
|
|
|
||
|
|
import GEOM
|
||
|
|
from salome.geom import geomBuilder
|
||
|
|
from salome.smesh import smeshBuilder
|
||
|
|
import SALOMEDS
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
geompy = geomBuilder.New()
|
||
|
|
|
||
|
|
|
||
|
|
nbox = 2
|
||
|
|
boxsize = 100
|
||
|
|
offset = 0
|
||
|
|
# Create 3D faces
|
||
|
|
boxes = []
|
||
|
|
# First creating all the boxes
|
||
|
|
# for i in range(nbox):
|
||
|
|
# for j in range(nbox):
|
||
|
|
# for k in range(nbox):
|
||
|
|
|
||
|
|
# x_orig = i*(boxsize+offset)
|
||
|
|
# y_orig = j*(boxsize+offset)
|
||
|
|
# z_orig = k*(boxsize+offset)
|
||
|
|
|
||
|
|
# tmp_box = geompy.MakeBoxDXDYDZ(boxsize, boxsize, boxsize)
|
||
|
|
|
||
|
|
# if not i == j == k == 0:
|
||
|
|
# box = geompy.MakeTranslation(tmp_box, x_orig,
|
||
|
|
# y_orig, z_orig)
|
||
|
|
# else:
|
||
|
|
# box = tmp_box
|
||
|
|
|
||
|
|
# geompy.addToStudy(box, 'box_{}:{}:{}'.format(i, j, k))
|
||
|
|
|
||
|
|
# boxes.append(box)
|
||
|
|
|
||
|
|
#With 6 boxes works
|
||
|
|
#But simplify for 2 boxes to also Test possibility of rewritting the
|
||
|
|
# input mesh from other parallel tests. In that case this test will break
|
||
|
|
# because the input mesh will not match the exported/imported box geometry.
|
||
|
|
for i in range(nbox):
|
||
|
|
for j in range(1):
|
||
|
|
for k in range(1):
|
||
|
|
|
||
|
|
x_orig = i*(boxsize+offset)
|
||
|
|
y_orig = j*(boxsize+offset)
|
||
|
|
z_orig = k*(boxsize+offset)
|
||
|
|
|
||
|
|
tmp_box = geompy.MakeBoxDXDYDZ(boxsize, boxsize, boxsize)
|
||
|
|
|
||
|
|
if not i == j == k == 0:
|
||
|
|
box = geompy.MakeTranslation(tmp_box, x_orig,
|
||
|
|
y_orig, z_orig)
|
||
|
|
else:
|
||
|
|
box = tmp_box
|
||
|
|
|
||
|
|
geompy.addToStudy(box, 'box_{}:{}:{}'.format(i, j, k))
|
||
|
|
|
||
|
|
boxes.append(box)
|
||
|
|
|
||
|
|
# Create fuse of all boxes
|
||
|
|
all_boxes = geompy.MakeCompound(boxes)
|
||
|
|
geompy.addToStudy(all_boxes, 'Compound_1')
|
||
|
|
|
||
|
|
# Removing duplicates faces and edges
|
||
|
|
all_boxes = geompy.MakeGlueFaces(all_boxes, 1e-07)
|
||
|
|
geompy.addToStudy(all_boxes, 'Glued_Faces_1')
|
||
|
|
|
||
|
|
rubik_cube = geompy.MakeGlueEdges(all_boxes, 1e-07)
|
||
|
|
geompy.addToStudy(all_boxes, 'rubik_cube')
|
||
|
|
|
||
|
|
|
||
|
|
smesh = smeshBuilder.New()
|
||
|
|
print("Creating Parallel Mesh")
|
||
|
|
par_mesh = smesh.ParallelMesh(rubik_cube, name="par_mesh", mesher2D="NETGEN_2D_Remote")
|
||
|
|
|
||
|
|
print("Setting parallelism method")
|
||
|
|
par_mesh.SetParallelismMethod(smeshBuilder.MULTITHREAD)
|
||
|
|
|
||
|
|
print("Setting parallelism options")
|
||
|
|
param = par_mesh.GetParallelismSettings()
|
||
|
|
param.SetNbThreads(6)
|
||
|
|
|
||
|
|
print("Starting parallel compute")
|
||
|
|
is_done = par_mesh.Compute()
|
||
|
|
if not is_done:
|
||
|
|
raise Exception("Error when computing Mesh")
|
||
|
|
|
||
|
|
print(" Triangle: ", par_mesh.NbTriangles())
|
||
|
|
print(" edge: ", par_mesh.NbEdges())
|
||
|
|
|
||
|
|
assert par_mesh.NbTetras() == 0
|
||
|
|
assert par_mesh.NbTriangles() > 0
|
||
|
|
assert par_mesh.NbEdges() > 0
|