dht_auto_indexing.cpp

Aller à la documentation de ce fichier.

#include "dht_auto_indexing.hpp"

namespace xios
{

  CDHTAutoIndexing::~CDHTAutoIndexing()
  {
  }

  CDHTAutoIndexing::CDHTAutoIndexing(const CArray<size_t,1>& hashValue,
                                     const MPI_Comm& clientIntraComm)
    : CClientClientDHTTemplate<size_t>(clientIntraComm)
  {

    nbIndexOnProc_ = hashValue.size();
    size_t nbIndexAccum;
    MPI_Scan(&nbIndexOnProc_, &nbIndexAccum, 1, MPI_UNSIGNED_LONG, MPI_SUM, clientIntraComm);

    // Broadcasting the total number of indexes
    int rank, size;
    MPI_Comm_rank(clientIntraComm, &rank);
    MPI_Comm_size(clientIntraComm, &size);
    if (rank == (size-1)) nbIndexesGlobal_ = nbIndexAccum;
    MPI_Bcast(&nbIndexesGlobal_, 1, MPI_UNSIGNED_LONG, size-1, clientIntraComm);

    CArray<size_t,1>::const_iterator itbIdx = hashValue.begin(), itIdx,
                                     iteIdx = hashValue.end();

    size_t idx = 0;
    beginIndexOnProc_ = nbIndexAccum - nbIndexOnProc_;
    globalIndex_.resize(nbIndexOnProc_);
    for (itIdx = itbIdx; itIdx != iteIdx; ++itIdx)
    {
      globalIndex_[idx] = beginIndexOnProc_ + idx;
      ++idx ;
    }
  }

  CDHTAutoIndexing::CDHTAutoIndexing(Index2VectorInfoTypeMap& hashInitMap,
                                     const MPI_Comm& clientIntraComm)
    : CClientClientDHTTemplate<size_t>(clientIntraComm)
  {

    nbIndexOnProc_ = hashInitMap.size();
    size_t nbIndexAccum;
    MPI_Scan(&nbIndexOnProc_, &nbIndexAccum, 1, MPI_UNSIGNED_LONG, MPI_SUM, clientIntraComm);

    int rank, size;
    MPI_Comm_rank(clientIntraComm, &rank);
    MPI_Comm_size(clientIntraComm, &size);
    if (rank == (size-1)) nbIndexesGlobal_ = nbIndexAccum;
    MPI_Bcast(&nbIndexesGlobal_, 1, MPI_UNSIGNED_LONG, size-1, clientIntraComm);

    Index2VectorInfoTypeMap::iterator itbIdx = hashInitMap.begin(), itIdx,
                                      iteIdx = hashInitMap.end();
    size_t idx = 0;
    beginIndexOnProc_ = nbIndexAccum - nbIndexOnProc_;
    globalIndex_.resize(nbIndexOnProc_);
    for (itIdx = itbIdx; itIdx != iteIdx; ++itIdx)
    {
//      (itIdx->second)[0] = beginIndexOnProc_ + idx;
      (itIdx->second)[1] = beginIndexOnProc_ + idx;
      globalIndex_[idx] = beginIndexOnProc_ + idx;
      ++idx ;
    }
  }

  size_t CDHTAutoIndexing::getNbIndexesGlobal() const
  {
    return nbIndexesGlobal_;
  }

  size_t CDHTAutoIndexing::getIndexStart() const
  {
    return beginIndexOnProc_;
  }

  size_t CDHTAutoIndexing::getIndexCount() const
  {
    return nbIndexOnProc_;
  }

}