mirror/tracy
1
0
Fork 0
mirror of https://github.com/wolfpld/tracy.git synced 2024-09-20 05:42:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Process client broadcasts.

Browse Source
This commit is contained in:
Bartosz Taudul 2019-06-17 19:34:48 +02:00
parent e0bbb41976
commit 265913d969
1 changed files with 66 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
profiler/src/main.cpp
View File

@ -1,5 +1,6 @@
#include <algorithm>
#include <assert.h>
#include <chrono>
#include <inttypes.h>
#include <imgui.h>
#include "imgui_impl_glfw.h"
@ -25,6 +26,7 @@
#define STBI_ONLY_PNG
#include "stb_image.h"
#include "../../server/tracy_flat_hash_map.hpp"
#include "../../server/tracy_pdqsort.h"
#include "../../server/TracyBadVersion.hpp"
#include "../../server/TracyFileRead.hpp"
@ -82,8 +84,17 @@ std::vector<std::unordered_map<std::string, uint64_t>::const_iterator> RebuildCo
return ret;
}
struct ClientData
{
int64_t time;
uint32_t protocolVersion;
std::string procName;
};
int main( int argc, char** argv )
{
tracy::flat_hash_map<uint32_t, ClientData> clients;
std::unique_ptr<tracy::View> view;
int badVer = 0;
@ -236,6 +247,7 @@ int main( int argc, char** argv )
char addr[1024] = { "127.0.0.1" };
std::thread loadThread;
tracy::UdpListen* broadcastListen = nullptr;
glfwShowWindow( window );
@ -266,6 +278,54 @@ int main( int argc, char** argv )
glfwSetWindowTitle( window, title );
}
if( !broadcastListen )
{
broadcastListen = new tracy::UdpListen();
if( !broadcastListen->Listen( 8087 ) )
{
delete broadcastListen;
broadcastListen = nullptr;
}
}
else
{
tracy::IpAddress addr;
int len;
auto msg = broadcastListen->Read( len, addr );
const auto t = std::chrono::duration_cast<std::chrono::milliseconds>( std::chrono::system_clock::now().time_since_epoch() ).count();
if( msg )
{
uint32_t protoVer;
memcpy( &protoVer, msg, sizeof( uint32_t ) );
auto procname = msg + sizeof( uint32_t );
auto it = clients.find( addr.GetNumber() );
if( it == clients.end() )
{
clients.emplace( addr.GetNumber(), ClientData { t, protoVer, procname } );
}
else
{
it->second.time = t;
if( it->second.protocolVersion != protoVer ) it->second.protocolVersion = protoVer;
if( strcmp( it->second.procName.c_str(), procname ) != 0 ) it->second.procName = procname;
}
}
auto it = clients.begin();
while( it != clients.end() )
{
const auto diff = t - it->second.time;
if( diff > 10000 ) // 10s
{
it = clients.erase( it );
}
else
{
++it;
}
}
}
setlocale( LC_NUMERIC, "C" );
style.Colors[ImGuiCol_WindowBg] = ImVec4( 0.129f, 0.137f, 0.11f, 1.f );
ImGui::Begin( "Get started", nullptr, ImGuiWindowFlags_AlwaysAutoResize );
@ -399,6 +459,12 @@ int main( int argc, char** argv )
}
else
{
if( broadcastListen )
{
delete broadcastListen;
broadcastListen = nullptr;
clients.clear();
}
if( loadThread.joinable() ) loadThread.join();
view->NotifyRootWindowSize( display_w, display_h );
if( !view->Draw() )