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

Go to the documentation of this file.

#ifndef LAMA_GENERALDISTRIBUTION_HPP_
#define LAMA_GENERALDISTRIBUTION_HPP_

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

// others
#include <lama/LAMATypes.hpp>

#include <lama/distribution/Distribution.hpp>

#include <vector>

#if (BOOST_VERSION < 103600)
#include <map>
#else //(BOOST_VERSION >= 103600)
#include <boost/unordered_map.hpp>
#endif //(BOOST_VERSION < 103600)


namespace lama
{

class LAMA_DLL_IMPORTEXPORT GeneralDistribution: public Distribution
{
public:

    GeneralDistribution(
        const IndexType globalSize,
        const std::vector<IndexType>& myGlobalIndexes,
        const CommunicatorPtr communicator);

    explicit GeneralDistribution( const Distribution& other );

//    GeneralDistribution(const GeneralDistribution& other);

    virtual ~GeneralDistribution();

    virtual bool isLocal(const IndexType index) const;

    virtual IndexType getLocalSize() const;

    virtual std::vector<IndexType>& getLocalRows();

    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;

    void getDistributionVector( std::vector<IndexType>& row2Partition ) const;

    void printDistributionVector( std::string name ) const;

protected:

    GeneralDistribution(const IndexType globalSize,
        const CommunicatorPtr communicator);

    //TODO: Evaluate which one is faster
#if (BOOST_VERSION < 103600)
    typedef std::map<IndexType, IndexType> Global2LocalMapType;
#else //(BOOST_VERSION >= 103600)
    typedef boost::unordered_map<IndexType,IndexType> Global2LocalMapType;
#endif //(BOOST_VERSION < 103600)

    Global2LocalMapType mGlobal2Local;
    std::vector<IndexType> mLocal2Global;

private:

    GeneralDistribution();

    GeneralDistribution& operator=( const GeneralDistribution& other);

    LAMA_LOG_DECL_STATIC_LOGGER(logger);
};

typedef boost::shared_ptr<GeneralDistribution> GeneralDistributionPtr;

}

#endif // LAMA_GENERALDISTRIBUTION_HPP_