2 # -*- coding: iso-8859-1 -*-
3 # Copyright (C) 2007-2023 CEA/DEN, EDF R&D
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.
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.
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
19 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
22 from medcoupling import *
23 from mpi4py import MPI
27 class ParaMEDMEM_DEC_Tests(unittest.TestCase):
28 def test_NonCoincidentDEC_py(self):
29 size = MPI.COMM_WORLD.size
30 rank = MPI.COMM_WORLD.rank
33 raise RuntimeError("Expect MPI.MPI_COMM_WORLD size == 5")
36 procs_source = list(range(nproc_source))
37 procs_target = list(range(size - nproc_source + 1, size))
39 interface = CommInterface()
41 target_group = MPIProcessorGroup(interface, procs_target)
42 source_group = MPIProcessorGroup(interface, procs_source)
44 dec = NonCoincidentDEC(source_group, target_group)
46 data_dir = os.path.join(os.environ['MEDCOUPLING_ROOT_DIR'], "share", "resources", "med")
47 if not os.path.isdir(data_dir):
48 data_dir = os.environ.get('MED_RESOURCES_DIR',"::").split(":")[1]
49 tmp_dir = os.environ.get('TMP', "")
53 filename_xml1 = os.path.join(data_dir, "square1_split")
54 filename_xml2 = os.path.join(data_dir, "square2_split")
56 MPI.COMM_WORLD.Barrier()
58 if source_group.containsMyRank():
59 filename = filename_xml1 + str(rank+1) + ".med"
60 meshname = "Mesh_2_" + str(rank+1)
62 mesh = MESH(MED_DRIVER, filename, meshname)
63 support = SUPPORT(mesh, "all elements", MED_CELL)
64 paramesh = ParaMESH(mesh, source_group, "source mesh")
66 parasupport = UnstructuredParaSUPPORT( support, source_group)
67 comptopo = ComponentTopology()
69 parafield = ParaFIELD(parasupport, comptopo)
71 nb_local = support.getNumberOfElements(MED_ALL_ELEMENTS);
73 value = [1.0]*nb_local
75 parafield.getField().setValue(value)
76 icocofield = ICoCoMEDDoubleField(paramesh,parafield)
77 dec.attachLocalField(icocofield,'P0')
80 if target_group.containsMyRank():
81 filename = filename_xml2 + str(rank - nproc_source + 1) + ".med"
82 meshname = "Mesh_3_" + str(rank - nproc_source + 1)
84 mesh = MESH(MED_DRIVER, filename, meshname)
85 support = SUPPORT(mesh, "all elements", MED_CELL)
86 paramesh = ParaMESH(mesh, target_group, "target mesh")
88 parasupport = UnstructuredParaSUPPORT( support, target_group)
89 comptopo = ComponentTopology()
90 parafield = ParaFIELD(parasupport, comptopo)
92 nb_local = support.getNumberOfElements(MED_ALL_ELEMENTS)
93 value = [0.0]*nb_local
95 parafield.getField().setValue(value)
96 icocofield = ICoCoMEDDoubleField(paramesh,parafield)
98 dec.attachLocalField(icocofield, 'P0')
101 field_before_int = [0.0]
102 field_after_int = [0.0]
104 if source_group.containsMyRank():
105 field_before_int = [parafield.getVolumeIntegral(1)]
106 MPI.MPI_Bcast(field_before_int, 1, MPI.MPI_DOUBLE, 0, MPI.MPI_COMM_WORLD);
109 dec.setForcedRenormalization(False)
113 if target_group.containsMyRank():
114 MPI.MPI_Bcast(field_before_int, 1, MPI.MPI_DOUBLE, 0, MPI.MPI_COMM_WORLD)
116 dec.setForcedRenormalization(False)
118 field_after_int = [parafield.getVolumeIntegral(1)]
121 MPI.MPI_Bcast(field_before_int, 1, MPI.MPI_DOUBLE, 0, MPI.MPI_COMM_WORLD)
122 MPI.MPI_Bcast(field_after_int , 1, MPI.MPI_DOUBLE, size-1, MPI.MPI_COMM_WORLD)
125 self.assertDoubleEquals(field_before_int[0], field_after_int[0], epsilon)
127 # Some clean up that still needs MPI communication, so to be done before MPI_Finalize()
129 target_group.release()
130 source_group.release()
132 MPI.COMM_WORLD.Barrier()
135 if __name__ == "__main__":