merge changes from the defunct ubot

inc/message.h

@@ -0,0 +1,156 @@
+//
+// YaPB - Counter-Strike Bot based on PODBot by Markus Klinge.
+// Copyright © 2004-2020 YaPB Development Team <team@yapb.ru>.
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU General Public License for more details.
+//
+
+#pragma once
+
+// netmessage functions enum
+CR_DECLARE_SCOPED_ENUM (NetMsg,
+   None = -1,
+   VGUIMenu = 1,
+   ShowMenu = 2,
+   WeaponList = 3,
+   CurWeapon = 4,
+   AmmoX = 5,
+   AmmoPickup = 6,
+   Damage = 7,
+   Money = 8,
+   StatusIcon = 9,
+   DeathMsg = 10,
+   ScreenFade = 11,
+   HLTV = 12,
+   TextMsg = 13,
+   TeamInfo = 14,
+   BarTime = 15,
+   SendAudio = 17,
+   BotVoice = 18,
+   NVGToggle = 19,
+   FlashBat = 20,
+   Fashlight = 21,
+   ItemStatus = 22
+)
+
+// vgui menus (since latest steam updates is obsolete, but left for old cs)
+CR_DECLARE_SCOPED_ENUM (GuiMenu,
+   TeamSelect = 2, // menu select team
+   TerroristSelect = 26, // terrorist select menu
+   CTSelect = 27  // ct select menu
+)
+
+// cache flags for TextMsg message
+CR_DECLARE_SCOPED_ENUM (TextMsgCache,
+   NeedHandle = cr::bit (0),
+   TerroristWin = cr::bit (1),
+   CounterWin = cr::bit (2),
+   Commencing = cr::bit (3),
+   BombPlanted = cr::bit (4),
+   RestartRound = cr::bit (5),
+   BurstOn = cr::bit (6),
+   BurstOff = cr::bit (7)
+)
+
+// cache flags for StatusIcon message
+CR_DECLARE_SCOPED_ENUM (StatusIconCache,
+   NeedHandle = cr::bit (0),
+   BuyZone = cr::bit (1),
+   VipSafety = cr::bit (2),
+   C4 = cr::bit (3)
+)
+
+class MessageDispatcher final : public Singleton <MessageDispatcher> {
+private:
+   using MsgFunc = void (MessageDispatcher::*) ();
+
+private:
+   struct Args {
+      union {
+         float float_;
+         long long_;
+         const char *chars_;
+      };
+
+   public:
+      Args (float value) : float_ (value) { }
+      Args (int value) : long_ (value) { }
+      Args (const char *value) : chars_ (value) { }
+   };
+
+private:
+   Dictionary <String, int32> m_textMsgCache; // cache strings for faster access for textmsg
+   Dictionary <String, int32> m_showMenuCache; // cache for the showmenu message
+   Dictionary <String, int32> m_statusIconCache; // cache for status icon message
+   Dictionary <String, int32> m_teamInfoCache; // cache for teaminfo message
+
+private:
+   Bot *m_bot {}; // owner of a message
+   NetMsg m_current {}; // ongoing message id
+
+   SmallArray <Args> m_args; // args collected from write* functions
+   Dictionary <String, NetMsg> m_wanted; // wanted messages
+
+   Dictionary <NetMsg, int32, IntNoHash <int32>> m_maps; // maps our message to id to engine message id
+   Dictionary <NetMsg, MsgFunc, IntNoHash <int32>> m_handlers; // maps our message id to handler function
+
+private:
+   void netMsgTextMsg ();
+   void netMsgVGUIMenu ();
+   void netMsgShowMenu ();
+   void netMsgWeaponList ();
+   void netMsgCurWeapon ();
+   void netMsgAmmoX ();
+   void netMsgAmmoPickup ();
+   void netMsgDamage ();
+   void netMsgMoney ();
+   void netMsgStatusIcon ();
+   void netMsgDeathMsg ();
+   void netMsgScreenFade ();
+   void netMsgHLTV ();
+   void netMsgTeamInfo ();
+   void netMsgBarTime ();
+   void netMsgItemStatus ();
+   void netMsgNVGToggle ();
+   void netMsgFlashBat ();
+
+public:
+   MessageDispatcher ();
+   ~MessageDispatcher () = default;
+
+public:
+   int32 add (StringRef name, int32 id);
+   int32 id (NetMsg msg);
+
+   void start (edict_t *ent, int32 type);
+   void stop ();
+   void ensureMessages ();
+
+public:
+   template <typename T> void collect (const T &value) {
+      if (m_current == NetMsg::None) {
+         return;
+      }
+      m_args.emplace (value);
+   }
+
+   void stopCollection () {
+      m_current = NetMsg::None;
+   }
+
+private:
+   void reset () {
+      m_current = NetMsg::None;
+      m_bot = nullptr;
+   }
+};
+
+CR_EXPOSE_GLOBAL_SINGLETON (MessageDispatcher, msgs);