Salome HOME
ParaMEDMEM: DisjointDEC
[tools/medcoupling.git] / src / ParaMEDMEM / OverlapMapping.hxx
index b30afe4184fbeb8001b4fbfd51fb511972b91845..03d0d7f8a3c37862eab375557994cfa52044f4f0 100644 (file)
@@ -1,4 +1,4 @@
-// Copyright (C) 2007-2015  CEA/DEN, EDF R&D
+// Copyright (C) 2007-2016  CEA/DEN, EDF R&D
 //
 // This library is free software; you can redistribute it and/or
 // modify it under the terms of the GNU Lesser General Public
 #ifndef __OVERLAPMAPPING_HXX__
 #define __OVERLAPMAPPING_HXX__
 
-#include "MEDCouplingAutoRefCountObjectPtr.hxx"
+#include "MCAuto.hxx"
 #include "OverlapElementLocator.hxx"
 
 #include <vector>
 #include <map>
+//#define DEC_DEBUG
 
-namespace ParaMEDMEM
+namespace MEDCoupling
 {
   class ProcessorGroup;
   class DataArrayInt;
   class MEDCouplingFieldDouble;
 
-  typedef std::map<int,double> SparseDoubleVec;
+  using namespace std;
+  typedef map<int,double> SparseDoubleVec;
 
   /*!
    * Internal class, not part of the public API.
    *
    * Used by the impl of OverlapInterpolationMatrix, plays an equivalent role than what the NxM_Mapping
    * does for the InterpolationMatrix.
-   *
    */
   class OverlapMapping
   {
@@ -49,15 +50,16 @@ namespace ParaMEDMEM
     OverlapMapping(const ProcessorGroup& group, const OverlapElementLocator& locator);
     void keepTracksOfSourceIds(int procId, DataArrayInt *ids);
     void keepTracksOfTargetIds(int procId, DataArrayInt *ids);
-    void addContributionST(const std::vector< SparseDoubleVec >& matrixST, const DataArrayInt *srcIds, int srcProcId, const DataArrayInt *trgIds, int trgProcId);
-    void prepare(const std::vector< int >& procsToSendField, int nbOfTrgElems);
+    void addContributionST(const vector< SparseDoubleVec >& matrixST, const DataArrayInt *srcIds, int srcProcId, const DataArrayInt *trgIds, int trgProcId);
+    void prepare(const vector< int >& procsToSendField, int nbOfTrgElems);
     void computeDenoConservativeVolumic(int nbOfTuplesTrg);
-    void computeDenoGlobConstraint();
+//    void computeDenoIntegralGlobConstraint();
+//    void computeDenoIntegral();
+    void computeDenoRevIntegral(const DataArrayDouble & targetAreas);
     //
     void multiply(const MEDCouplingFieldDouble *fieldInput, MEDCouplingFieldDouble *fieldOutput, double default_val) const;
     void transposeMultiply(const MEDCouplingFieldDouble *fieldInput, MEDCouplingFieldDouble *fieldOutput);
   private:
-//    void fillProcToSendRcvForMultiply(const std::vector< int >& procsToSendField);
     void serializeMatrixStep0ST(const int *nbOfElemsSrc, int *&bigArr, int *count, int *offsets,
                                 int *countForRecv, int *offsetsForRecv) const;
     int serializeMatrixStep1ST(const int *nbOfElemsSrc, const int *recvStep0, const int *countStep0, const int *offsStep0,
@@ -66,7 +68,7 @@ namespace ParaMEDMEM
     void unserializationST(int nbOfTrgElems, const int *nbOfElemsSrcPerProc, const int *bigArrRecv, const int *bigArrRecvCounts, const int *bigArrRecvOffs,
                            const int *bigArrRecv2, const double *bigArrDRecv2, const int *bigArrRecv2Count, const int *bigArrRecv2Offs);
     void finishToFillFinalMatrixST();
-    void updateZipSourceIdsForMultiply();
+    void fillSourceIdsZipReceivedForMultiply();
 
 #ifdef DEC_DEBUG
     void printMatrixesST() const;
@@ -77,58 +79,51 @@ namespace ParaMEDMEM
     const ProcessorGroup &_group;
     const OverlapElementLocator& _locator;
 
-    /**! Vector of DAInt of cell identifiers. The 2 following class members work in pair. For a proc ID i,
-     * first member gives an old2new map for the local part of the source mesh that has been sent to proc#i, just based on the
+    /**! Map of DAInt of cell identifiers. For a proc ID i,
+     * gives an old2new map for the local part of the source mesh that has been sent to proc#i, just based on the
      * bounding box computation (this is potentially a larger set than what is finally in the interp matrix).
      * Second member gives proc ID.  */
-    std::vector< MEDCouplingAutoRefCountObjectPtr<DataArrayInt> > _sent_src_ids_st2;
-    //! see above _sent_src_ids_st2
-    std::vector< int > _sent_src_proc_st2;
-
-    //! See _src_ids_st2 and _sent_src_proc_st2. Same for target mesh.
-    std::vector< MEDCouplingAutoRefCountObjectPtr<DataArrayInt> > _sent_trg_ids_st2;
-    //! See _src_ids_st2 and _sent_src_proc_st2. Same for target mesh.
-    std::vector< int > _sent_trg_proc_st2;
+    map < int, MCAuto<DataArrayInt> > _sent_src_ids;
 
+    //! See _sent_src_ids. Same for target mesh.
+    map < int, MCAuto<DataArrayInt> > _sent_trg_ids;
 
     /**! Vector of matrixes (partial interpolation ratios), result of the LOCAL interpolator run.
      * Indexing shared with _source_proc_id_st, and _target_proc_id_st.   */
-    std::vector< std::vector< SparseDoubleVec > > _matrixes_st;
+    vector< vector< SparseDoubleVec > > _matrixes_st;
     //! See _matrixes_st - vec of source proc IDs
-    std::vector< int > _source_proc_id_st;
+    vector< int > _source_proc_id_st;
     //! See _matrixes_st - vec of target proc IDs
-    std::vector< int > _target_proc_id_st;
-
-    /**! Vector of remote proc IDs from which this proc received cell IDs of the source mesh.
-     * Indexing shared with _nb_of_rcv_src_ids_proc_st2 */
-    std::vector< int > _rcv_src_ids_proc_st2;
-    /**! Number of received source mesh IDs at mesh data exchange. See _src_ids_proc_st2 above.
-     Counting the number of IDs suffices, as we just need this to prepare the receive when doing the final vector matrix multiplication */
-    std::vector< int > _nb_of_rcv_src_ids_proc_st2;
-
-    /**! Specifies for each (target) remote proc ID (given in _src_ids_zip_proc_st2 below) the corresponding
-     * source cell IDs to use. Same indexing as _src_ids_zip_proc_st2. Sorted.
-     * On a given proc, and after updateZipSourceIdsForMultiply(), this member contains exactly the same set of source cell IDs as what is given
-     * in the locally held interpolation matrices.
-     * IMPORTANT: as a consequence cell IDs in _src_ids_zip_st2 are *remote* identifiers.   */
-    std::vector< std::vector<int> > _src_ids_zip_st2;
-    //! Vector of remote proc ID to which the local source mapping above corresponds. See _src_ids_zip_st2 above.
-    std::vector< int > _src_ids_zip_proc_st2;
+    vector< int > _target_proc_id_st;
+
+    /**! Number of received source mesh IDs at mesh data exchange.
+     Counting the number of IDs suffices, as we just need this to prepare the receive side, when doing the final vector matrix multiplication.
+     First dimension is the remote proc ID from which we received. */
+    map <int, int > _nb_of_rcv_src_ids;
+
+    /**! Specifies for each (target) remote proc ID (first dim of the map) the corresponding
+     * source cell IDs to use.
+     * This information is stored from the *locally* COMPuted matrices, and corresponds hence to field value that will need to
+     * sent later on, if this matrix bit itself is sent aways.  */
+    map<int, vector<int> > _src_ids_zip_comp;
+
+    /**! Same idea as _src_ids_zip_comp above, but for RECEIVED matrix. */
+    map<int, vector<int> > _src_ids_zip_recv;
 
     /**! THE matrix for matrix-vector product. The first dimension is indexed in the set of target procs
     * that interacts with local source mesh. The second dim is the target cell ID.
-    * Same indexing as _the_matrix_st_source_proc_id  */
-    std::vector< std::vector< SparseDoubleVec > > _the_matrix_st;
+    * Same indexing as _the_matrix_st_source_proc_id and _the_deno_st.
+    * We don't use a map here to be more efficient in the final matrix-vector computation which requires the joint
+    * taversal of _the_matrix_st and _the_deno_st.
+    * This matrix is filled after receival from other procs, contrary to _matrixes_st which contains local computations.*/
+    vector< vector< SparseDoubleVec > > _the_matrix_st;
     //! See _the_matrix_st above. List of source proc IDs contributing to _the_matrix_st
-    std::vector< int > _the_matrix_st_source_proc_id;
+    vector< int > _the_matrix_st_source_proc_id;
+    // Denominators (computed from the numerator matrix). As for _the_matrix_st it is paired with _the_matrix_st_source_proc_id
+    vector< vector< SparseDoubleVec > > _the_deno_st;
 
     //! Proc IDs to which data will be sent (originating this current proc) for matrix-vector computation
-    std::vector< int > _proc_ids_to_send_vector_st;
-    //! Proc IDs from which data will be received (on this current proc) for matrix-vector computation
-    std::vector< int > _proc_ids_to_recv_vector_st;
-
-    // Denominators (computed from the numerator matrix). As for _the_matrix_st it is paired with _the_matrix_st_source_proc_id
-    std::vector< std::vector< SparseDoubleVec > > _the_deno_st;
+    vector< int > _proc_ids_to_send_vector_st;
   };
 }