1 // Copyright (C) 2007-2024 CEA, EDF
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 // Lesser General Public License for more details.
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 // Author : Anthony Geay (CEA/DEN)
21 #ifndef __OVERLAPELEMENTLOCATOR_HXX__
22 #define __OVERLAPELEMENTLOCATOR_HXX__
24 #include "InterpolationOptions.hxx"
25 #include "MEDCouplingNatureOfField.hxx"
26 #include "MEDCouplingPointSet.hxx"
27 #include "MEDCouplingMemArray.hxx"
41 class OverlapInterpolationMatrix;
43 typedef std::pair<int,int> ProcCouple; // a couple of proc IDs, typically used to define a exchange betw 2 procs
45 class OverlapElementLocator : public INTERP_KERNEL::InterpolationOptions
48 OverlapElementLocator(const ParaFIELD *sourceField, const ParaFIELD *targetField, const ProcessorGroup& group,
49 double epsAbs, int workSharingAlgo);
50 virtual ~OverlapElementLocator();
51 const MPI_Comm *getCommunicator() const;
52 void exchangeMeshes(OverlapInterpolationMatrix& matrix);
53 std::vector< std::pair<int,int> > getToDoList() const { return _to_do_list; }
54 std::vector< int > getProcsToSendFieldData() const { return _procs_to_send_field; } // same set as the set of procs we sent mesh data to
55 std::string getSourceMethod() const;
56 std::string getTargetMethod() const;
57 const MEDCouplingPointSet *getSourceMesh(int procId) const;
58 const DataArrayIdType *getSourceIds(int procId) const;
59 const MEDCouplingPointSet *getTargetMesh(int procId) const;
60 const DataArrayIdType *getTargetIds(int procId) const;
61 bool isInMyTodoList(int i, int j) const;
62 void debugPrintWorkSharing(std::ostream & ostr) const;
64 void computeBoundingBoxesAndInteractionList();
65 void computeTodoList_original();
66 void computeTodoList_new(bool revertIter);
67 void fillProcToSend();
68 bool intersectsBoundingBox(int i, int j) const;
69 void sendLocalMeshTo(int procId, bool sourceOrTarget, OverlapInterpolationMatrix& matrix) const;
70 void receiveRemoteMeshFrom(int procId, bool sourceOrTarget);
71 void sendMesh(int procId, const MEDCouplingPointSet *mesh, const DataArrayIdType *idsToSend) const;
72 void receiveMesh(int procId, MEDCouplingPointSet* &mesh, DataArrayIdType *&ids) const;
74 typedef MCAuto< MEDCouplingPointSet > AutoMCPointSet;
75 typedef MCAuto< DataArrayIdType > AutoDAInt;
76 typedef std::pair<int,bool> Proc_SrcOrTgt; // a key indicating a proc ID and whether the data is for source mesh/field or target mesh/field
78 static const int START_TAG_MESH_XCH;
80 const ParaFIELD *_local_source_field;
81 const ParaFIELD *_local_target_field;
84 MEDCouplingPointSet *_local_source_mesh;
85 MEDCouplingPointSet *_local_target_mesh;
87 /*! of size _group.size(). Contains for each source proc i, the ids of proc j the targets interact with.
88 This vector is common for all procs in _group. This is the full list of jobs to do the interp. */
89 std::vector< std::vector< int > > _proc_pairs;
90 //! todo lists per proc
91 std::vector< std::vector< ProcCouple > > _all_todo_lists;
92 //! list of interpolation couples to be done by this proc only. This is a simple extraction of the member above _all_todo_lists
93 std::vector< ProcCouple > _to_do_list;
94 //! list of procs the local proc will have to send mesh data to:
95 std::vector< Proc_SrcOrTgt > _procs_to_send_mesh;
96 // /*! list of procs the local proc will have to send field data to for the final matrix-vector computation:
97 // * This can be different from _procs_to_send_mesh (restricted to Source) because interpolation matrix bits are computed on a potentially
98 // * different proc than the target one. */
99 std::vector< int > _procs_to_send_field;
100 //! Set of distant meshes
101 std::map< Proc_SrcOrTgt, AutoMCPointSet > _remote_meshes;
102 //! Set of cell ID mappings for the above distant meshes (because only part of the meshes are exchanged)
103 std::map< Proc_SrcOrTgt, AutoDAInt > _remote_elems;
104 double* _domain_bounding_boxes;
105 //! bounding box absolute adjustment
108 std::vector<int> _distant_proc_ids;
110 const ProcessorGroup& _group;
111 const MPI_Comm *_comm;