Probe.hpp

/*
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 "../application_api/Endpoints.hpp"
#include "../application_api/HelicsPrimaryTypes.hpp"
#include "../application_api/Publications.hpp"
#include "helicsApp.hpp"
 
#include <deque>
#include <map>
#include <memory>
#include <set>
#include <string>
#include <vector>
 
namespace helics::apps {
 
class HELICS_CXX_EXPORT Probe: public App {
  public:
    Probe() = default;
    explicit Probe(std::vector<std::string> args);
    Probe(int argc, char* argv[]);
    Probe(std::string_view name, const FederateInfo& fedInfo);
    Probe(std::string_view name, const std::shared_ptr<Core>& core, const FederateInfo& fedInfo);
 
    Probe(std::string_view name, CoreApp& core, const FederateInfo& fedInfo);
    Probe(std::string_view name, const std::string& configString);
 
    Probe(Probe&& other_source) = default;
 
    Probe& operator=(Probe&& fed) = default;
 
    virtual void initialize() override;
 
    virtual void runTo(Time stopTime_input) override;
    int getConnections() const { return connections; }
    int getMessageCount() const { return messagesReceived; }
 
  private:
    void runProbe();
 
    Endpoint endpoint;  
    int connections{0};  
    int messagesReceived{0};  
};
}  // namespace helics::apps