MpiComms.h

/*
Copyright (c) 2017-2024,
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 "../CommsInterface.hpp"
#include "gmlc/containers/BlockingQueue.hpp"
#include "helics/helics-config.h"
 
#include <atomic>
#include <future>
#include <mutex>
#include <set>
#include <string>
#include <utility>
#include <vector>
 
namespace helics {
namespace mpi {
    class MpiComms final: public CommsInterface {
      public:
        MpiComms();
        ~MpiComms();
 
      private:
        std::atomic<bool> shutdown{false};
        virtual void queue_rx_function() override;  
        virtual void queue_tx_function() override;  
 
        int processIncomingMessage(ActionMessage& cmd);
 
        gmlc::containers::BlockingQueue<ActionMessage> rxMessageQueue;
        gmlc::containers::BlockingQueue<std::pair<std::pair<int, int>, std::vector<char>>>
            txMessageQueue;
 
        std::atomic<bool> hasBroker{false};
        virtual void closeReceiver() override;  
 
      public:
        void setBrokerAddress(const std::string& address);
 
        std::string getAddress() { return localTargetAddress; }
        gmlc::containers::BlockingQueue<ActionMessage>& getRxMessageQueue()
        {
            return rxMessageQueue;
        }
        gmlc::containers::BlockingQueue<std::pair<std::pair<int, int>, std::vector<char>>>&
            getTxMessageQueue()
        {
            return txMessageQueue;
        }
    };
 
}  // namespace mpi
}  // namespace helics