Program Listing for File tasking_oracle.hpp

↰ Return to documentation for file (cosm/oracle/tasking_oracle.hpp)

#pragma once

/*******************************************************************************
 * Includes
 ******************************************************************************/
#include <boost/optional.hpp>
#include <map>
#include <string>
#include <variant>

#include "rcppsw/er/client.hpp"

#include "cosm/cosm.hpp"
#include "cosm/ta/time_estimate.hpp"

/*******************************************************************************
 * Namespaces
 ******************************************************************************/
namespace cosm::ta {
class bi_tdgraph_executive;
class polled_task;
namespace ds {
class bi_tdgraph;
} /* namespace ds */
} // namespace cosm::ta

namespace cosm::oracle::config {
struct tasking_oracle_config;
} // namespace cosm::oracle::config

namespace cosm::oracle {

/*******************************************************************************
 * Class Definitions
 ******************************************************************************/
class tasking_oracle final : public rer::client<tasking_oracle> {
 public:
  static inline const std::string kExecEstPrefix = "exec_est";
  static inline const std::string kInterfaceEstPrefix = "interface_est";

  using variant_type = std::variant<cta::time_estimate>;

  tasking_oracle(const coconfig::tasking_oracle_config* config,
                 const cta::ds::bi_tdgraph* graph);

  boost::optional<variant_type> ask(const std::string& query) const;

  void listener_add(cta::bi_tdgraph_executive* executive);

  bool update_exec_ests(void) const { return mc_exec_ests; }
  bool update_int_ests(void) const { return mc_int_ests; }

  void task_abort_cb(const cta::polled_task* task);
  void task_finish_cb(const cta::polled_task* task);

 private:
  /* clang-format off */
  const bool                          mc_exec_ests;
  const bool                          mc_int_ests;
  std::map<std::string, variant_type> m_map{};
  /* clang-format on */
};

} /* namespace cosm::oracle */