source: XIOS/dev/dev_olga/src/distribution_server.hpp @ 1132

/*!
   \file distribution_server.hpp
   \author Ha NGUYEN
   \since 13 Jan 2015
   \date 04 Feb 2015

   \brief Index distribution on server side.
 */

#ifndef __XIOS_DISTRIBUTION_SERVER_HPP__
#define __XIOS_DISTRIBUTION_SERVER_HPP__

#include "distribution.hpp"

namespace xios {

/*!
  \class CDistributionServer
  The class, for now, plays a role of computing local index for writing data on server
*/
class CDistributionServer : public CDistribution
{
  public:
    /** Default constructor */
    CDistributionServer(int rank, int dims, const CArray<size_t,1>& globalIndex = CArray<size_t,1>());
    CDistributionServer(int rank, const std::vector<int>& nZoomBeginServer,
                        const std::vector<int>& nZoomSizeServer, const std::vector<int>& nGlobal);
    CDistributionServer(int rank, const std::vector<int>& nZoomBeginServer,
                        const std::vector<int>& nZoomSizeServer,
                        const std::vector<int>& nZoomBeginGlobal,
                        const std::vector<int>& nGlobal);
    CDistributionServer(int rank, 
                        const std::vector<CArray<int,1> >& globalIndexEachDimension,
                        const std::vector<int>& nGlobal);

    /** Default destructor */
    virtual ~CDistributionServer();

    const std::vector<int>& getZoomBeginGlobal() const;
    const std::vector<int>& getZoomBeginServer() const;
    const std::vector<int>& getZoomSizeServer() const;
    const GlobalLocalMap& getGlobalLocalIndex() const { return globalLocalIndexMap_; }
    const std::vector<CArray<int,1> >& getGlobalIndexEachDimension() const {return globalIndexEachDimension_;}
    int getGridSize() const;

    virtual CArray<size_t,1> computeLocalIndex(const CArray<size_t,1>& globalIndex);
    virtual void computeLocalIndex(CArray<size_t,1>& globalIndex);
    virtual void computeGlobalIndex(CArray<int,1>& indexes) const;

  protected:
    virtual void createGlobalIndex();

    void createGlobalIndexFromIndex(const std::vector<CArray<int,1> >& globalIndexOnEachDimension,
                                    const std::vector<int>& nbGlobal);

  protected:
    GlobalLocalMap globalLocalIndexMap_;

  private:
    std::vector<int> nGlobal_;
    std::vector<int> nZoomBeginGlobal_;
    std::vector<int> nZoomSize_;
    std::vector<int> nZoomBegin_;
    std::vector<CArray<int, 1> > globalIndexEachDimension_;

};

} // namespace xios
#endif // __XIOS_DISTRIBUTION_SERVER_HPP__