MpiBroker.h

/*
Copyright (c) 2017-2019,
Battelle Memorial Institute; Lawrence Livermore National Security, LLC; Alliance for Sustainable
Energy, LLC. See the top-level NOTICE for additional details. All rights reserved.
SPDX-License-Identifier: BSD-3-Clause
*/
#pragma once
 
#include "../../core/CoreBroker.hpp"
#include "../CommsBroker.hpp"
 
#include <memory>
#include <string>
 
namespace helics {
namespace mpi {
    class MpiComms;
    class MpiBroker final: public CommsBroker<MpiComms, CoreBroker> {
      public:
        explicit MpiBroker(bool rootBroker = false) noexcept;
        explicit MpiBroker(std::string_view broker_name);
 
      protected:
        virtual std::shared_ptr<helicsCLI11App> generateCLI() override;
 
      public:
        virtual ~MpiBroker();
 
        virtual std::string generateLocalAddressString() const override;
        static void displayHelp(bool local_only = false);
 
      private:
        virtual bool brokerConnect() override;
        mutable std::mutex dataMutex;  
        std::string brokerAddress;  
        int brokerRank{0};
        int brokerTag{0};
    };
}  // namespace mpi
}  // namespace helics