Brought over LY legacy imgui code

Signed-off-by: AMZN-Olex <5432499+AMZN-Olex@users.noreply.github.com>
2021-07-26 18:01:14 -04:00
parent 90e45a5bbd
commit 46777b537e
7 changed files with 812 additions and 0 deletions
@@ -0,0 +1,90 @@
+/*
+* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
+* its licensors.
+*
+* For complete copyright and license terms please see the LICENSE at the root of this
+* distribution (the "License"). All use of this software is governed by the License,
+* or, if provided, by the license below or the license accompanying this file. Do not
+* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+*
+*/
+#pragma once
+#include <GridMate/Drillers/ReplicaDriller.h>
+#include "MultiplayerDebugByteReporter.h"
+#include <GridMate/Drillers/CarrierDriller.h>
+
+namespace MultiplayerDiagnostics
+{
+    /**
+     * \brief GridMate network live analysis tool via ImGui.
+     */
+    class MultiplayerDebugPerEntityReporter
+        : public GridMate::Debug::ReplicaDrillerBus::Handler
+        , public GridMate::Debug::CarrierDrillerBus::Handler
+    {
+    public:
+        MultiplayerDebugPerEntityReporter();
+        virtual ~MultiplayerDebugPerEntityReporter();
+
+        // main update loop
+        void OnImGuiUpdate();
+
+        // ReplicaDrillerBus - receive
+
+        void OnReceiveReplicaBegin(GridMate::Replica* replica, const void* data, size_t len) override;
+        void OnReceiveReplicaEnd(GridMate::Replica* replica) override;
+        void OnReceiveReplicaChunkEnd(GridMate::ReplicaChunkBase* chunk, AZ::u32 chunkIndex) override;
+        void OnReceiveDataSet(GridMate::ReplicaChunkBase* chunk, AZ::u32 chunkIndex, GridMate::DataSetBase* dataSet, GridMate::PeerId from, GridMate::PeerId to, const void* data, size_t len) override;
+        void OnReceiveRpc(GridMate::ReplicaChunkBase* chunk, AZ::u32 chunkIndex, GridMate::Internal::RpcRequest* rpc, GridMate::PeerId from, GridMate::PeerId to, const void* data, size_t len) override;
+
+        // ReplicaDrillerBus - sending
+
+        void OnSendReplicaBegin(GridMate::Replica* replica) override;
+        void OnSendReplicaEnd(GridMate::Replica* replica, const void* data, size_t len) override;
+        void OnSendReplicaChunkEnd(GridMate::ReplicaChunkBase* chunk, AZ::u32 chunkIndex, const void* data, size_t len) override;
+        void OnSendDataSet(GridMate::ReplicaChunkBase* chunk, AZ::u32 chunkIndex, GridMate::DataSetBase* dataSet, GridMate::PeerId from, GridMate::PeerId to, const void* data, size_t len) override;
+        void OnSendRpc(GridMate::ReplicaChunkBase* chunk, AZ::u32 chunkIndex, GridMate::Internal::RpcRequest* rpc, GridMate::PeerId from, GridMate::PeerId to, const void* data, size_t len) override;
+
+        // CarrierDrillerBus
+        void OnIncomingConnection (GridMate::Carrier* carrier, GridMate::ConnectionID id) override;
+        void OnFailedToConnect (GridMate::Carrier* carrier,
+            GridMate::ConnectionID id,
+            GridMate::CarrierDisconnectReason reason) override;
+        void OnConnectionEstablished (GridMate::Carrier* carrier, GridMate::ConnectionID id) override;
+        void OnDisconnect (GridMate::Carrier* carrier,
+            GridMate::ConnectionID id,
+            GridMate::CarrierDisconnectReason reason) override;
+        void OnDriverError (GridMate::Carrier* carrier,
+            GridMate::ConnectionID id,
+            const GridMate::DriverError& error) override;
+        void OnSecurityError (GridMate::Carrier* carrier,
+            GridMate::ConnectionID id,
+            const GridMate::SecurityError& error) override;
+        void OnUpdateStatistics (const GridMate::string& address,
+            const GridMate::TrafficControl::Statistics& lastSecond,
+            const GridMate::TrafficControl::Statistics& lifeTime,
+            const GridMate::TrafficControl::Statistics& effectiveLastSecond,
+            const GridMate::TrafficControl::Statistics& effectiveLifeTime) override;
+        void OnConnectionStateChanged (GridMate::Carrier* carrier,
+            GridMate::ConnectionID id,
+            GridMate::Carrier::ConnectionStates newState) override;
+
+    private:
+
+        bool m_showServerReportWindow = false;
+        bool m_isTrackingMessages = false;
+
+        AZStd::map<AZStd::string, EntityReporter> m_sendingEntityReports{};
+        EntityReporter m_currentSendingEntityReport;
+
+        AZStd::map<AZStd::string, EntityReporter> m_receivingEntityReports{};
+        EntityReporter m_currentReceivingEntityReport;
+
+        float m_replicatedStateKbpsWarn = 10.f;
+        float m_replicatedStateMaxSizeWarn = 30.f;
+
+        void UpdateTrafficStatistics();
+        AZStd::map<GridMate::string, GridMate::TrafficControl::Statistics> m_lastSecondStats;
+    };
+}