/home/brandes/workspace/LAMA/src/lama/distribution/GenBlockDistribution.hpp

Go to the documentation of this file.

#ifndef LAMA_GENBLOCKDISTRIBUTION_HPP_
#define LAMA_GENBLOCKDISTRIBUTION_HPP_

// for dll_import
#include <lama/config.hpp>

// base classes
#include <lama/distribution/Distribution.hpp>

// boost
#include <boost/scoped_array.hpp>

namespace lama
{

class LAMA_DLL_IMPORTEXPORT GenBlockDistribution: public lama::Distribution
{
public:

    GenBlockDistribution(const IndexType globalSize,
                         const std::vector<IndexType>& localSizes,
                         const CommunicatorPtr communicator);

     GenBlockDistribution(const IndexType globalSize,
                          const IndexType firstGlobalIdx,
                          const IndexType lastGlobalIdx,
                          const CommunicatorPtr communicator);

    GenBlockDistribution(const IndexType globalSize,
                         const IndexType localSize,
                         const CommunicatorPtr communicator);

    GenBlockDistribution(const IndexType globalSize,
                         const float weight,
                         const CommunicatorPtr communicator);

    virtual ~GenBlockDistribution();

    void getLocalRange( IndexType& lb, IndexType& ub ) const;

    // Implementation of abstract methods, override default virtual methods

    virtual bool isLocal(const IndexType index) const;

    virtual PartitionId getOwner( const IndexType globalIndex ) const;

    virtual IndexType getLocalSize() const;

    virtual IndexType local2global(const IndexType localIndex) const;

    virtual IndexType global2local(const IndexType globalIndex) const;

    virtual bool isEqual(const Distribution& other) const;

    virtual void writeAt(std::ostream& stream) const;

    virtual void computeOwners(
        const std::vector<IndexType>& requiredIndexes,
        std::vector<PartitionId>& owners) const;

    void printDistributionVector( std::string name ) const;

protected:

    LAMA_LOG_DECL_STATIC_LOGGER(logger);

private:

    void setOffsets( const IndexType rank, const IndexType numPartitions, const IndexType localSizes[] );

    void setOffsets( const IndexType rank, const IndexType numPartitions, const IndexType mySize );

    GenBlockDistribution();  // disable default destructor

    boost::scoped_array<IndexType> mOffsets; 

    IndexType mLB, mUB;     
};

}

#endif // LAMA_GENBLOCKDISTRIBUTION_HPP_