]> SALOME platform Git repositories - tools/medcoupling.git/blob - src/ParaMEDMEM_Swig/test_BasicOperation.py
Salome HOME
Merge branch 'V9_11_BR'
[tools/medcoupling.git] / src / ParaMEDMEM_Swig / test_BasicOperation.py
1 #!/usr/bin/env python
2 #  -*- coding: iso-8859-1 -*-
3 # Copyright (C) 2020-2023  CEA, EDF
4 #
5 # This library is free software; you can redistribute it and/or
6 # modify it under the terms of the GNU Lesser General Public
7 # License as published by the Free Software Foundation; either
8 # version 2.1 of the License, or (at your option) any later version.
9 #
10 # This library is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13 # Lesser General Public License for more details.
14 #
15 # You should have received a copy of the GNU Lesser General Public
16 # License along with this library; if not, write to the Free Software
17 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
18 #
19 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 #
21
22 __doc__ = """Here a mesh is initially loaded into two parts in the 2 procs along Y axis ( low part and upper part ).
23 Then a redistribution of cells is performed along X axis ( left part and right part )
24 """
25
26 import medcoupling as mc
27
28 from mpi4py import MPI
29
30 def GenerateGlobalMesh():
31     mesh = mc.MEDCouplingCMesh()
32     arr = mc.DataArrayDouble(11) ; arr.iota()
33     mesh.setCoords(arr,arr)
34     mesh = mesh.buildUnstructured()
35     return mesh
36
37 def MeshPart0():
38     mesh = GenerateGlobalMesh()
39     p0 = mesh[list(range(50))]
40     nodeIds = p0.computeFetchedNodeIds()
41     p0.zipCoords()
42     return mc.ParaUMesh(p0,mc.DataArrayInt(list(range(50))),nodeIds)
43     
44 def MeshPart1():
45     mesh = GenerateGlobalMesh()
46     cellIds = mc.DataArrayInt(list(range(50,100,1)))
47     p1 = mesh[cellIds]
48     nodeIds = p1.computeFetchedNodeIds()
49     p1.zipCoords()
50     return mc.ParaUMesh(p1,cellIds,nodeIds)
51
52 workPerProc = {0 : MeshPart0 , 1 : MeshPart1}
53 distribPerProc = { 0 : [0, 1, 2, 3, 4, 5, 11, 12, 13, 14, 15, 16, 22, 23, 24, 25, 26, 27, 33, 34, 35, 36, 37, 38, 44, 45, 46, 47, 48, 49, 55, 56, 57, 58, 59, 60, 66, 67, 68, 69, 70, 71, 77, 78, 79, 80, 81, 82, 88, 89, 90, 91, 92, 93, 99, 100, 101, 102, 103, 104, 110, 111, 112, 113, 114, 115],
54 1 : [6, 7, 8, 9, 10, 17, 18, 19, 20, 21, 28, 29, 30, 31, 32, 39, 40, 41, 42, 43, 50, 51, 52, 53, 54, 61, 62, 63, 64, 65, 72, 73, 74, 75, 76, 83, 84, 85, 86, 87, 94, 95, 96, 97, 98, 105, 106, 107, 108, 109, 116, 117, 118, 119, 120]}
55
56 cellsExpected = {0 : [0, 1, 2, 3, 4, 10, 11, 12, 13, 14, 20, 21, 22, 23, 24, 30, 31, 32, 33, 34, 40, 41, 42, 43, 44, 50, 51, 52, 53, 54, 60, 61, 62, 63, 64, 70, 71, 72, 73, 74, 80, 81, 82, 83, 84, 90, 91, 92, 93, 94],
57 1 : [6, 7, 8, 9, 16, 17, 18, 19, 26, 27, 28, 29, 36, 37, 38, 39, 46, 47, 48, 49, 56, 57, 58, 59, 66, 67, 68, 69, 76, 77, 78, 79, 86, 87, 88, 89, 96, 97, 98, 99]}
58
59 if MPI.COMM_WORLD.size != 2 :
60     raise RuntimeError("Expected to be lanched with 2 procs !")
61
62 def test():
63     pmesh = workPerProc[MPI.COMM_WORLD.rank]()
64     pnodeids = mc.DataArrayInt(distribPerProc[MPI.COMM_WORLD.rank])
65     cells = pmesh.getCellIdsLyingOnNodes(pnodeids,True)
66     assert(cells.isEqual( mc.DataArrayInt(cellsExpected[MPI.COMM_WORLD.rank]) ))
67     pmesh_red = pmesh.redistributeCells(cells)
68     expected_mesh = GenerateGlobalMesh()[cells] ; expected_mesh.zipCoords()
69     assert( expected_mesh.isEqual(pmesh_red.getMesh(),1e-12) )
70
71 if __name__ == "__main__":
72     test()