Move source file parsing logic to a separate file.

2026-08-02 19:41:29 +00:00 · 2021-03-26 02:21:53 +01:00
parent 174bb4d2f1
commit bd4335bace
8 changed files with 211 additions and 142 deletions
--- a/server/TracySourceContents.cpp
+++ b/server/TracySourceContents.cpp
@@ -0,0 +1,89 @@
+#include "TracySourceContents.hpp"
+#include "TracyView.hpp"
+#include "TracyWorker.hpp"
+
+namespace tracy
+{
+
+SourceContents::SourceContents()
+    : m_file( nullptr )
+    , m_fileStringIdx( 0 )
+    , m_data( nullptr )
+    , m_dataBuf( nullptr )
+    , m_dataSize( 0 )
+{
+}
+
+SourceContents::~SourceContents()
+{
+    delete[] m_dataBuf;
+}
+
+void SourceContents::Parse( const char* fileName, const Worker& worker, const View& view )
+{
+    if( m_file == fileName ) return;
+
+    m_file = fileName;
+    m_fileStringIdx = worker.FindStringIdx( fileName );
+    m_lines.clear();
+    if( fileName )
+    {
+        uint32_t sz;
+        const auto srcCache = worker.GetSourceFileFromCache( fileName );
+        if( srcCache.data != nullptr )
+        {
+            m_data = srcCache.data;
+            sz = srcCache.len;
+        }
+        else
+        {
+            FILE* f = fopen( view.SourceSubstitution( fileName ), "rb" );
+            if( f )
+            {
+                fseek( f, 0, SEEK_END );
+                sz = ftell( f );
+                fseek( f, 0, SEEK_SET );
+                if( sz > m_dataSize )
+                {
+                    delete[] m_dataBuf;
+                    m_dataBuf = new char[sz];
+                    m_dataSize = sz;
+                }
+                fread( m_dataBuf, 1, sz, f );
+                m_data = m_dataBuf;
+                fclose( f );
+            }
+            else
+            {
+                m_file = nullptr;
+            }
+        }
+
+        if( m_file )
+        {
+            Tokenizer tokenizer;
+            auto txt = m_data;
+            for(;;)
+            {
+                auto end = txt;
+                while( *end != '\n' && *end != '\r' && end - m_data < sz ) end++;
+                m_lines.emplace_back( Tokenizer::Line { txt, end, tokenizer.Tokenize( txt, end ) } );
+                if( end - m_data == sz ) break;
+                if( *end == '\n' )
+                {
+                    end++;
+                    if( end - m_data < sz && *end == '\r' ) end++;
+                }
+                else if( *end == '\r' )
+                {
+                    end++;
+                    if( end - m_data < sz && *end == '\n' ) end++;
+                }
+                if( end - m_data == sz ) break;
+                txt = end;
+            }
+        }
+    }
+}
+
+}