MpiCommunicationStrategy.h

#pragma once
 
#include "antares-xpansion/benders/benders_core/ICommunicationStrategy.h"
#include "common_mpi.h"
 
class MpiCommunicationStrategy: public ICommunicationStrategy
{
public:
    explicit MpiCommunicationStrategy(mpi::communicator& world):
        world_(world)
    {
    }
 
    [[nodiscard]] int Rank() const override
    {
        return world_.rank();
    }
 
    [[nodiscard]] int WorldSize() const override
    {
        return world_.size();
    }
 
    void Barrier() const override
    {
        world_.barrier();
    }
 
    [[nodiscard]] bool ShouldParallelize() const override
    {
        return false;
    }
 
    void BroadcastBool(bool& value) const override
    {
        mpi::broadcast(world_, value, 0);
    }
 
    mpi::communicator& World()
    {
        return world_;
    }
 
    const mpi::communicator& World() const
    {
        return world_;
    }
 
private:
    mpi::communicator& world_;
};