THUAI6/CAPI/API/include/logic.h

#pragma once

#ifndef LOGIC_H
#define LOGIC_H

#ifdef _MSC_VER
#pragma warning(disable : 4996)
#endif

#include <iostream>
#include <vector>
#include <thread>
#include <mutex>
#include <condition_variable>
#include <atomic>

#include "Message2Server.pb.h"
#include "Message2Clients.pb.h"
#include "MessageType.pb.h"
#include "Services.grpc.pb.h"
#include "Services.pb.h"
#include "API.h"
#include "AI.h"
#include "structures.h"
#include "state.h"

// 封装了通信组件和对AI对象进行操作
class Logic : public ILogic
{
private:
    // gRPC客户端的stub，所有与服务端之间的通信操作都需要基于stub完成。
    std::unique_ptr<protobuf::AvailableService::Stub> THUAI6Stub;
    // ID、阵营记录
    int64_t playerID;
    THUAI6::PlayerType playerType;

    // 类型记录
    THUAI6::HumanType humanType;
    THUAI6::ButcherType butcherType;

    // GUID信息
    std::vector<int64_t> playerGUIDs;

    // THUAI5中的通信组件可以完全被我们的stub取代，故无须再写

    std::unique_ptr<IAI> pAI;

    std::unique_ptr<IGameTimer> timer;

    std::thread tAI;  // 用于运行AI的线程

    mutable std::mutex mtxAI;
    mutable std::mutex mtxState;
    mutable std::mutex mtxBuffer;

    std::condition_variable cvBuffer;
    std::condition_variable cvAI;

    // 信息队列目前可能会不用？具体待定

    // 存储状态，分别是现在的状态和缓冲区的状态。
    State state[2];
    State* currentState;
    State* bufferState;

    // 是否应该执行player()
    std::atomic_bool AILoop = true;

    // buffer是否更新完毕
    bool bufferUpdated = true;

    // 是否可以启用当前状态
    bool currentStateAccessed = false;

    // 是否应当启动AI
    bool AIStart = false;

    // 控制内容更新的变量
    std::atomic_bool freshed = false;

    // 提供给API使用的函数

    // 获取服务器发来的消息

    std::vector<std::shared_ptr<const THUAI6::Butcher>> GetButchers() const override;
    std::vector<std::shared_ptr<const THUAI6::Human>> GetHumans() const override;
    std::vector<std::shared_ptr<const THUAI6::Prop>> GetProps() const override;
    std::shared_ptr<const THUAI6::Human> HumanGetSelfInfo() const override;
    std::shared_ptr<const THUAI6::Butcher> ButcherGetSelfInfo() const override;

    std::vector<std::vector<THUAI6::PlaceType>> GetFullMap() const override;

    // 供IAPI使用的操作相关的部分
    bool Move(int64_t time, double angle) override;
    bool PickProp(THUAI6::PropType prop) override;
    bool UseProp() override;
    bool UseSkill() override;
    bool SendMessage(int64_t toID, std::string message) override;
    bool HaveMessage() override;
    std::pair<std::string, int64_t> GetMessage() override;

    bool Escape() override;

    // 说明：双向stream由三个函数共同实现，两个记录开始和结束，结果由Logic里的私有的成员变量记录，获得返回值则另调函数
    bool StartFixMachine() override
    {
    }
    bool EndFixMachine() override
    {
    }
    bool GetFixStatus() override
    {
    }

    bool StartSaveHuman() override
    {
    }
    bool EndSaveHuman() override
    {
    }
    bool GetSaveStatus() override
    {
    }

    bool Attack(double angle) override
    {
    }
    bool CarryHuman() override
    {
    }
    bool ReleaseHuman() override
    {
    }
    bool HangHuman() override
    {
    }

    bool WaitThread() override
    {
    }

    int GetCounter() override
    {
    }

    bool TryConnection();

    // 执行AI线程
    void PlayerWrapper(std::function<void()> player);

    // THUAI5中的一系列用于处理信息的函数可能也不会再用

    void ProcessMessage();

    // 将信息加载到buffer
    void LoadBuffer(protobuf::MessageToClient&);

    // 解锁状态更新线程
    void UnBlockBuffer();

    // 解锁AI线程
    void UnBlockAI();

    // 更新状态
    void Update() noexcept;

    // 等待
    void Wait() noexcept;

public:
    // 构造函数还需要传更多参数，有待补充
    Logic(THUAI6::PlayerType type, int64_t ID, THUAI6::ButcherType butcher, THUAI6::HumanType human);

    ~Logic()
    {
    }

    // Main函数同上
    void Main(CreateAIFunc createAI, std::string IP, std::string port);
};

#endif