src/ParaMEDMEM_Swig/ParaMEDMEMCommon.i

   1 // Copyright (C) 2017-2021  CEA/DEN, EDF R&D
   2 //
   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.
   7 //
   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.
  12 //
  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
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 %include std_set.i
  21
  22 %template() std::set<int>;
  23
  24 %{
  25 #include "CommInterface.hxx"
  26 #include "ProcessorGroup.hxx"
  27 #include "Topology.hxx"
  28 #include "MPIProcessorGroup.hxx"
  29 #include "DEC.hxx"
  30 #include "InterpKernelDEC.hxx"
  31 #include "NonCoincidentDEC.hxx"
  32 #include "StructuredCoincidentDEC.hxx"
  33 #include "OverlapDEC.hxx"
  34 #include "ParaMESH.hxx"
  35 #include "ParaFIELD.hxx"
  36 #include "ICoCoMEDField.hxx"
  37 #include "ComponentTopology.hxx"
  38 #include "ParaUMesh.hxx"
  39 #include "ParaSkyLineArray.hxx"
  40 #include "ParaDataArray.hxx"
  41
  42 using namespace INTERP_KERNEL;
  43 using namespace MEDCoupling;
  44 using namespace ICoCo;
  45 %}
  46
  47 %include "InterpolationOptions.hxx"
  48 %include "ProcessorGroup.hxx"
  49 %include "DECOptions.hxx"
  50 %include "ParaMESH.hxx"
  51 %include "ParaFIELD.hxx"
  52 %include "MPIProcessorGroup.hxx"
  53 %include "ComponentTopology.hxx"
  54 %include "DEC.hxx"
  55 %include "DisjointDEC.hxx"
  56 %include "InterpKernelDEC.hxx"
  57 %include "StructuredCoincidentDEC.hxx"
  58 %include "OverlapDEC.hxx"
  59
  60 %include "ICoCoField.hxx"
  61 %rename(ICoCoMEDField) ICoCo::MEDField;
  62 %include "ICoCoMEDField.hxx"
  63
  64 %newobject MEDCoupling::ParaUMesh::New;
  65 %newobject MEDCoupling::ParaUMesh::getMesh;
  66 %newobject MEDCoupling::ParaUMesh::getGlobalCellIds;
  67 %newobject MEDCoupling::ParaUMesh::getGlobalNodeIds;
  68 %newobject MEDCoupling::ParaUMesh::getCellIdsLyingOnNodes;
  69 %newobject MEDCoupling::ParaUMesh::redistributeCells;
  70 %newobject MEDCoupling::ParaUMesh::redistributeCellField;
  71 %newobject MEDCoupling::ParaUMesh::redistributeNodeField;
  72 %newobject MEDCoupling::ParaDataArrayInt32::New;
  73 %newobject MEDCoupling::ParaDataArrayInt32::buildComplement;
  74 %newobject MEDCoupling::ParaDataArrayInt64::New;
  75 %newobject MEDCoupling::ParaDataArrayInt64::buildComplement;
  76 %newobject MEDCoupling::ParaSkyLineArray::New;
  77 %newobject MEDCoupling::ParaSkyLineArray::equiRedistribute;
  78 %newobject MEDCoupling::ParaSkyLineArray::getSkyLineArray;
  79 %newobject MEDCoupling::ParaSkyLineArray::getGlobalIdsArray;
  80
  81 %feature("unref") ParaSkyLineArray "$this->decrRef();"
  82 %feature("unref") ParaUMesh "$this->decrRef();"
  83 %feature("unref") ParaDataArrayInt32 "$this->decrRef();"
  84 %feature("unref") ParaDataArrayInt64 "$this->decrRef();"
  85
  86 %nodefaultctor;
  87
  88 namespace MEDCoupling
  89 {
  90   class CommInterface
  91   {
  92   public:
  93     CommInterface();
  94     virtual ~CommInterface();
  95     int worldSize() const;
  96     int commSize(MPI_Comm comm, int* size) const;
  97     int commRank(MPI_Comm comm, int* rank) const;
  98     int commGroup(MPI_Comm comm, MPI_Group* group) const;
  99     int groupIncl(MPI_Group group, int size, int* ranks, MPI_Group* group_output) const;
 100     int commCreate(MPI_Comm comm, MPI_Group group, MPI_Comm* comm_output) const;
 101     int groupFree(MPI_Group* group) const;
 102     int commFree(MPI_Comm* comm) const;
 103
 104     int send(void* buffer, int count, MPI_Datatype datatype, int target, int tag, MPI_Comm comm) const;
 105     int recv(void* buffer, int count, MPI_Datatype datatype, int source, int tag, MPI_Comm comm, MPI_Status* status) const;
 106     int sendRecv(void* sendbuf, int sendcount, MPI_Datatype sendtype,
 107                  int dest, int sendtag, void* recvbuf, int recvcount,
 108                  MPI_Datatype recvtype, int source, int recvtag, MPI_Comm comm,
 109                  MPI_Status* status);
 110
 111     int Isend(void* buffer, int count, MPI_Datatype datatype, int target,
 112               int tag, MPI_Comm comm, MPI_Request *request) const;
 113     int Irecv(void* buffer, int count, MPI_Datatype datatype, int source,
 114               int tag, MPI_Comm comm, MPI_Request* request) const;
 115
 116     int wait(MPI_Request *request, MPI_Status *status) const;
 117     int test(MPI_Request *request, int *flag, MPI_Status *status) const;
 118     int requestFree(MPI_Request *request) const;
 119     int waitany(int count, MPI_Request *array_of_requests, int *index, MPI_Status *status) const;
 120     int testany(int count, MPI_Request *array_of_requests, int *index, int *flag, MPI_Status *status) const;
 121     int waitall(int count, MPI_Request *array_of_requests, MPI_Status *array_of_status) const { return MPI_Waitall(count, array_of_requests, array_of_status); }
 122     int testall(int count, MPI_Request *array_of_requests, int *flag, MPI_Status *array_of_status) const;
 123     int waitsome(int incount, MPI_Request *array_of_requests,int *outcount, int *array_of_indices, MPI_Status *array_of_status) const;
 124     int testsome(int incount, MPI_Request *array_of_requests, int *outcount,
 125                  int *array_of_indices, MPI_Status *array_of_status) const;
 126     int probe(int source, int tag, MPI_Comm comm, MPI_Status *status) const;
 127     int Iprobe(int source, int tag, MPI_Comm comm, int *flag, MPI_Status *status) const;
 128     int cancel(MPI_Request *request) const;
 129     int testCancelled(MPI_Status *status, int *flag) const;
 130     int barrier(MPI_Comm comm) const;
 131     int errorString(int errorcode, char *string, int *resultlen) const;
 132     int getCount(MPI_Status *status, MPI_Datatype datatype, int *count) const;
 133
 134     int broadcast(void* buffer, int count, MPI_Datatype datatype, int root, MPI_Comm comm) const;
 135     int allGather(void* sendbuf, int sendcount, MPI_Datatype sendtype,
 136                   void* recvbuf, int recvcount, MPI_Datatype recvtype,
 137                   MPI_Comm comm) const;
 138     int allToAll(void* sendbuf, int sendcount, MPI_Datatype sendtype,
 139                  void* recvbuf, int recvcount, MPI_Datatype recvtype,
 140                  MPI_Comm comm) const;
 141     int allToAllV(void* sendbuf, int* sendcounts, int* senddispls,
 142                   MPI_Datatype sendtype, void* recvbuf, int* recvcounts,
 143                   int* recvdispls, MPI_Datatype recvtype,
 144                   MPI_Comm comm) const;
 145
 146     int reduce(void* sendbuf, void* recvbuf, int count, MPI_Datatype datatype, MPI_Op op, int root, MPI_Comm comm) const;
 147     int allReduce(void* sendbuf, void* recvbuf, int count, MPI_Datatype datatype, MPI_Op op, MPI_Comm comm) const;
 148     %extend
 149     {
 150       PyObject *allGatherArrays(const DataArrayIdType *array) const
 151       {
 152         std::vector< MCAuto<DataArrayIdType> > ret;
 153         self->allGatherArrays(MPI_COMM_WORLD,array,ret);
 154         return convertFromVectorAutoObjToPyObj<DataArrayIdType>(ret,SWIGTITraits<mcIdType>::TI);
 155       }
 156
 157       PyObject *allToAllArrays(PyObject *arrays) const
 158       {
 159         std::vector< DataArrayIdType * > arraysIn;
 160         std::vector< MCAuto<DataArrayIdType> > arrayOut;
 161         convertFromPyObjVectorOfObj<MEDCoupling::DataArrayIdType*>(arrays,SWIGTITraits<mcIdType>::TI,"DataArrayIdType",arraysIn);
 162         std::vector< MCAuto<DataArrayIdType> > arraysIn2(FromVecToVecAuto<DataArrayIdType>(arraysIn));
 163         self->allToAllArrays(MPI_COMM_WORLD,arraysIn2,arrayOut);
 164         return convertFromVectorAutoObjToPyObj<DataArrayIdType>(arrayOut,SWIGTITraits<mcIdType>::TI);
 165       }
 166     }
 167   };
 168
 169   class ParaUMesh : public RefCountObject
 170   {
 171   public:
 172     static ParaUMesh *New(MEDCouplingUMesh *mesh, DataArrayIdType *globalCellIds, DataArrayIdType *globalNodeIds);
 173     ParaUMesh *redistributeCells(const DataArrayIdType *globalCellIds) const;
 174     DataArrayIdType *redistributeCellField(const DataArrayIdType *globalCellIds, const DataArrayIdType *fieldValueToRed) const;
 175     DataArrayDouble *redistributeCellField(const DataArrayIdType *globalCellIds, const DataArrayDouble *fieldValueToRed) const;
 176     DataArrayIdType *redistributeNodeField(const DataArrayIdType *globalCellIds, const DataArrayIdType *fieldValueToRed) const;
 177     DataArrayDouble *redistributeNodeField(const DataArrayIdType *globalCellIds, const DataArrayDouble *fieldValueToRed) const;
 178     %extend
 179     {
 180       ParaUMesh(MEDCouplingUMesh *mesh, DataArrayIdType *globalCellIds, DataArrayIdType *globalNodeIds)
 181       {
 182         return ParaUMesh::New(mesh,globalCellIds,globalNodeIds);
 183       }
 184
 185       MEDCouplingUMesh *getMesh()
 186       {
 187         MEDCouplingUMesh *ret(self->getMesh());
 188         if(ret) ret->incrRef();
 189         return ret;
 190       }
 191
 192       DataArrayIdType *getGlobalCellIds()
 193       {
 194         DataArrayIdType *ret(self->getGlobalCellIds());
 195         if(ret) ret->incrRef();
 196         return ret;
 197       }
 198
 199       DataArrayIdType *getGlobalNodeIds()
 200       {
 201         DataArrayIdType *ret(self->getGlobalNodeIds());
 202         if(ret) ret->incrRef();
 203         return ret;
 204       }
 205
 206       DataArrayIdType *getCellIdsLyingOnNodes(const DataArrayIdType *globalNodeIds, bool fullyIn) const
 207       {
 208         MCAuto<DataArrayIdType> ret(self->getCellIdsLyingOnNodes(globalNodeIds,fullyIn));
 209         return ret.retn();
 210       }
 211     }
 212   };
 213
 214   class ParaDataArray : public RefCountObject
 215   {
 216   };
 217
 218   class ParaDataArrayInt32 : public ParaDataArray
 219   {
 220   public:
 221     static ParaDataArrayInt32 *New(DataArrayInt32 *seqDa);
 222     DataArrayIdType *buildComplement(int nbOfElems) const;
 223     %extend
 224     {
 225       ParaDataArrayInt32(DataArrayInt32 *seqDa)
 226       {
 227         return ParaDataArrayInt32::New(seqDa);
 228       }
 229     }
 230   };
 231
 232   class ParaDataArrayInt64 : public ParaDataArray
 233   {
 234   public:
 235     static ParaDataArrayInt64 *New(DataArrayInt64 *seqDa);
 236     DataArrayIdType *buildComplement(long nbOfElems) const;
 237     %extend
 238     {
 239       ParaDataArrayInt64(DataArrayInt64 *seqDa)
 240       {
 241         return ParaDataArrayInt64::New(seqDa);
 242       }
 243     }
 244   };
 245
 246   class ParaSkyLineArray : public RefCountObject
 247   {
 248   public:
 249     static ParaSkyLineArray *New(MEDCouplingSkyLineArray *ska, DataArrayIdType *globalIds);
 250     %extend
 251     {
 252       ParaSkyLineArray(MEDCouplingSkyLineArray *ska, DataArrayIdType *globalIds)
 253       {
 254         return ParaSkyLineArray::New(ska,globalIds);
 255       }
 256
 257       ParaSkyLineArray *equiRedistribute(mcIdType nbOfEntities) const
 258       {
 259         MCAuto<ParaSkyLineArray> ret(self->equiRedistribute(nbOfEntities));
 260         return ret.retn();
 261       }
 262
 263       MEDCouplingSkyLineArray *getSkyLineArray() const
 264       {
 265         MEDCouplingSkyLineArray *ret(self->getSkyLineArray());
 266         if(ret)
 267           ret->incrRef();
 268         return ret;
 269       }
 270
 271       DataArrayIdType *getGlobalIdsArray() const
 272       {
 273         DataArrayIdType *ret(self->getGlobalIdsArray());
 274         if(ret)
 275           ret->incrRef();
 276         return ret;
 277       }
 278     }
 279   };
 280 }
 281
 282 /* This object can be used only if MED_ENABLE_FVM is defined*/
 283 #ifdef MED_ENABLE_FVM
 284 class NonCoincidentDEC : public DEC
 285 {
 286 public:
 287   NonCoincidentDEC(ProcessorGroup& source, ProcessorGroup& target);
 288 };
 289 #endif
 290
 291 %extend MEDCoupling::ParaMESH
 292 {
 293   PyObject *getGlobalNumberingCell2() const
 294   {
 295     const mcIdType *tmp=self->getGlobalNumberingCell();
 296     mcIdType size=self->getCellMesh()->getNumberOfCells();
 297     PyObject *ret=PyList_New(size);
 298     for(mcIdType i=0;i<size;i++)
 299       PyList_SetItem(ret,i,PyInt_FromLong(tmp[i]));
 300     return ret;
 301   }
 302
 303   PyObject *getGlobalNumberingFace2() const
 304   {
 305     const mcIdType *tmp=self->getGlobalNumberingFace();
 306     mcIdType size=self->getFaceMesh()->getNumberOfCells();
 307     PyObject *ret=PyList_New(size);
 308     for(mcIdType i=0;i<size;i++)
 309       PyList_SetItem(ret,i,PyInt_FromLong(tmp[i]));
 310     return ret;
 311   }
 312
 313   PyObject *getGlobalNumberingNode2() const
 314   {
 315     const mcIdType *tmp=self->getGlobalNumberingNode();
 316     mcIdType size=self->getCellMesh()->getNumberOfNodes();
 317     PyObject *ret=PyList_New(size);
 318     for(mcIdType i=0;i<size;i++)
 319       PyList_SetItem(ret,i,PyInt_FromLong(tmp[i]));
 320     return ret;
 321   }
 322 }
 323
 324 %pythoncode %{
 325 if MEDCouplingUse64BitIDs():
 326   ParaDataArrayInt = ParaDataArrayInt64
 327 else:
 328   ParaDataArrayInt = ParaDataArrayInt32
 329 %}