server_distribution_description.hpp

Aller à la documentation de ce fichier.

#ifndef __XIOS_SERVER_DISTRIBUTION_DESCRIPTION_HPP
#define __XIOS_SERVER_DISTRIBUTION_DESCRIPTION_HPP

#include "xios_spl.hpp"
#include "array_new.hpp"
#include <unordered_map>

namespace xios
{
class CServerDistributionDescription
{
  public:
    enum ServerDistributionType
    {
      BAND_DISTRIBUTION, PLAN_DISTRIBUTION, ROOT_DISTRIBUTION
    };

    CServerDistributionDescription(const std::vector<int>& globalDimensionSize,
                                   int nServer,
                                   ServerDistributionType serType=BAND_DISTRIBUTION);

    virtual ~CServerDistributionDescription();

    void computeServerDistribution(bool doComputeGlobalIndex = false, int positionDimensionDistributed = 1);
    std::vector<int> computeServerGlobalIndexInRange(const std::pair<size_t, size_t>& indexBeginEnd, int positionDimensionDistributed = 1);
    std::vector<int> computeServerGlobalByElement(std::vector<std::unordered_map<size_t,std::vector<int> > >& indexServerOnElement,
                                                  int rank,
                                                  int clientSize,
                                                  const CArray<int,1>& axisDomainOrder,
                                                  int positionDimensionDistributed = 1);

    std::vector<std::vector<int> > getServerIndexBegin() const;
    std::vector<std::vector<int> > getServerDimensionSizes() const;
    const std::vector<CArray<size_t,1> >& getGlobalIndex() const;
    const std::unordered_map<size_t,int>& getGlobalIndexRange() const;
    int getDimensionDistributed();

    static int defaultDistributedDimension(int gridDimension,                                   
                                           ServerDistributionType serType=BAND_DISTRIBUTION);

  protected:
    int computeBandDistribution(int nServer, int positionDimensionDistributed = 1);
    int computeRootDistribution(int nServer, int positionDimensionDistributed = 1);
    void computePlanDistribution(int nServer);
    void computeRangeProcIndex(int clientRank,
                               int clientSize,
                               int rangeProcSize,
                               int& rangeBegin,
                               int& rangeSize);

  private:
    std::vector<std::vector<int> > indexBegin_;     
    std::vector<std::vector<int> > dimensionSizes_; 
    std::vector<int> nGlobal_; 

    std::vector<CArray<size_t,1> > vecGlobalIndex_;

    std::unordered_map<size_t,int> globalIndex_;

    ServerDistributionType serverType_;

    int nServer_;

    int positionDimensionDistributed_;
};

} // namespace xios
#endif // __XIOS_SERVER_DISTRIBUTION_DESCRIPTION_HPP