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

Go to the documentation of this file.

#ifndef LAMA_BLOCKDISTRIBUTION_HPP_
#define LAMA_BLOCKDISTRIBUTION_HPP_

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

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

namespace lama
{

class LAMA_DLL_IMPORTEXPORT BlockDistribution: public Distribution
{
public:

    BlockDistribution(const IndexType globalSize, const CommunicatorPtr communicator);

    virtual ~BlockDistribution();

    static void getRange( IndexType& lb, IndexType& ub,
                          const IndexType n, const PartitionId rank, const PartitionId size );

    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;

    static DistributionPtr create( const IndexType globalSize,
                                   const CommunicatorPtr communicator);

    void printDistributionVector( std::string problem ) const;

protected:

    LAMA_LOG_DECL_STATIC_LOGGER(logger);

private:

    BlockDistribution();  // disable default constructor as it has no size

    IndexType mBlockSize; 
    IndexType lb, ub;     
};

}

#endif // LAMA_BLOCKDISTRIBUTION_HPP_