ai.h

#pragma once
 
#include <functional>
 
#include "proto/play_info_msg.pb.h"
#include "software/ai/hl/stp/play/play.h"
#include "software/ai/play_selection_fsm.h"
#include "software/time/timestamp.h"
#include "software/world/world.h"
 
class Ai final
{
   public:
    Ai() = delete;
 
    explicit Ai(const TbotsProto::AiConfig& ai_config);
 
    void overridePlay(std::unique_ptr<Play> play);
 
    std::unique_ptr<TbotsProto::PrimitiveSet> getPrimitives(const WorldPtr& world_ptr);
 
    TbotsProto::PlayInfo getPlayInfo() const;
 
    void overridePlayFromProto(TbotsProto::Play play_proto);
 
    void updateAiConfig(TbotsProto::AiConfig& ai_config);
 
   private:
    void checkAiConfig();
 
    TbotsProto::AiConfig ai_config_;
    std::unique_ptr<FSM<PlaySelectionFSM>> fsm;
    std::unique_ptr<Play> override_play;
    std::unique_ptr<Play> current_play;
    TbotsProto::Play current_override_play_proto;
    bool ai_config_changed;
 
    // inter play communication
    InterPlayCommunication inter_play_communication;
};