From 1527e7bc1006e26556b95fff098108bacd6eadf8 Mon Sep 17 00:00:00 2001
From: Bartosz Taudul <wolf.pld@gmail.com>
Date: Sun, 13 Oct 2019 15:50:37 +0200
Subject: [PATCH] Add annotation modification window.

---
 server/TracyView.cpp | 67 ++++++++++++++++++++++++++++++++++++++++++--
 server/TracyView.hpp |  2 ++
 2 files changed, 67 insertions(+), 2 deletions(-)

diff --git a/server/TracyView.cpp b/server/TracyView.cpp
index e8dbedbc..f4fd0835 100644
--- a/server/TracyView.cpp
+++ b/server/TracyView.cpp
@@ -662,6 +662,7 @@ bool View::DrawImpl()
     if( m_lockInfoWindow != InvalidId ) DrawLockInfoWindow();
     if( m_showPlayback ) DrawPlayback();
     if( m_showCpuDataWindow ) DrawCpuDataWindow();
+    if( m_selectedAnnotation ) DrawSelectedAnnotation();
 
     if( m_zoomAnim.active )
     {
@@ -1427,6 +1428,7 @@ void View::HandleZoneViewMouse( int64_t timespan, const ImVec2& wpos, float w, d
             const auto e = std::max( m_highlight.start, m_highlight.end );
             ann->start = s;
             ann->end = e;
+            m_selectedAnnotation = ann.get();
             m_annotations.emplace_back( std::move( ann ) );
         }
         m_highlight.active = false;
@@ -2556,8 +2558,8 @@ void View::DrawZones()
     {
         if( ann->start < m_vd.zvEnd && ann->end > m_vd.zvStart )
         {
-            draw->AddRectFilled( linepos + ImVec2( ( ann->start - m_vd.zvStart ) * pxns, 0 ), linepos + ImVec2( ( ann->end - m_vd.zvStart ) * pxns, lineh ), 0x22888888 );
-            draw->AddRect( linepos + ImVec2( ( ann->start - m_vd.zvStart ) * pxns, 0 ), linepos + ImVec2( ( ann->end - m_vd.zvStart ) * pxns, lineh ), 0x44888888 );
+            draw->AddRectFilled( linepos + ImVec2( ( ann->start - m_vd.zvStart ) * pxns, 0 ), linepos + ImVec2( ( ann->end - m_vd.zvStart ) * pxns, lineh ), m_selectedAnnotation == ann.get() ? 0x44888888 : 0x22888888 );
+            draw->AddRect( linepos + ImVec2( ( ann->start - m_vd.zvStart ) * pxns, 0 ), linepos + ImVec2( ( ann->end - m_vd.zvStart ) * pxns, lineh ), m_selectedAnnotation == ann.get() ? 0x66888888 : 0x44888888 );
             if( ImGui::IsMouseHoveringRect( linepos + ImVec2( ( ann->start - m_vd.zvStart ) * pxns, 0 ), linepos + ImVec2( ( ann->end - m_vd.zvStart ) * pxns, lineh ) ) )
             {
                 ImGui::BeginTooltip();
@@ -2575,6 +2577,15 @@ void View::DrawZones()
                 TextFocused( "Annotation length:", TimeToString( ann->end - ann->start ) );
                 ImGui::EndTooltip();
             }
+            if( ( ann->end - ann->start ) * pxns > th * 4 )
+            {
+                draw->AddCircleFilled( linepos + ImVec2( ( ann->start - m_vd.zvStart ) * pxns + th * 2, th * 2 ), th, 0x88AABB22 );
+                draw->AddCircle( linepos + ImVec2( ( ann->start - m_vd.zvStart ) * pxns + th * 2, th * 2 ), th, 0xAAAABB22 );
+                if( ImGui::IsMouseClicked( 0 ) && ImGui::IsMouseHoveringRect( linepos + ImVec2( ( ann->start - m_vd.zvStart ) * pxns + th, th ), linepos + ImVec2( ( ann->start - m_vd.zvStart ) * pxns + th * 3, th * 3 ) ) )
+                {
+                    m_selectedAnnotation = ann.get();
+                }
+            }
         }
     }
 
@@ -11504,6 +11515,58 @@ void View::DrawCpuDataWindow()
     ImGui::End();
 }
 
+void View::DrawSelectedAnnotation()
+{
+    assert( m_selectedAnnotation );
+    bool show = true;
+    ImGui::Begin( "Annotation", &show, ImGuiWindowFlags_AlwaysAutoResize );
+#ifdef TRACY_EXTENDED_FONT
+    if( ImGui::Button( ICON_FA_MICROSCOPE " Zoom to annotation" ) )
+#else
+    if( ImGui::Button( "Zoom to annotation" ) )
+#endif
+    {
+        ZoomToRange( m_selectedAnnotation->start, m_selectedAnnotation->end );
+    }
+    ImGui::SameLine();
+#ifdef TRACY_EXTENDED_FONT
+    if( ImGui::Button( ICON_FA_TRASH_ALT " Remove" ) )
+#else
+    if( ImGui::Button( "Remove" ) )
+#endif
+    {
+        for( auto it = m_annotations.begin(); it != m_annotations.end(); ++it )
+        {
+            if( it->get() == m_selectedAnnotation )
+            {
+                m_annotations.erase( it );
+                break;
+            }
+        }
+        ImGui::End();
+        m_selectedAnnotation = nullptr;
+        return;
+    }
+    ImGui::Separator();
+    {
+        const auto desc = m_selectedAnnotation->text.c_str();
+        const auto descsz = std::min<size_t>( 1023, m_selectedAnnotation->text.size() );
+        char buf[1024];
+        buf[descsz] = '\0';
+        memcpy( buf, desc, descsz );
+        if( ImGui::InputTextWithHint( "", "Describe annotation", buf, 256 ) )
+        {
+            m_selectedAnnotation->text.assign( buf );
+        }
+    }
+    ImGui::Separator();
+    TextFocused( "Annotation begin:", TimeToString( m_selectedAnnotation->start ) );
+    TextFocused( "Annotation end:", TimeToString( m_selectedAnnotation->end ) );
+    TextFocused( "Annotation length:", TimeToString( m_selectedAnnotation->end - m_selectedAnnotation->start ) );
+    ImGui::End();
+    if( !show ) m_selectedAnnotation = nullptr;
+}
+
 template<class T>
 void View::ListMemData( T ptr, T end, std::function<void(T&)> DrawAddress, const char* id, int64_t startTime )
 {
diff --git a/server/TracyView.hpp b/server/TracyView.hpp
index 75f57b96..7ae4bbfd 100644
--- a/server/TracyView.hpp
+++ b/server/TracyView.hpp
@@ -139,6 +139,7 @@ private:
     void DrawLockInfoWindow();
     void DrawPlayback();
     void DrawCpuDataWindow();
+    void DrawSelectedAnnotation();
 
     template<class T>
     void ListMemData( T ptr, T end, std::function<void(T&)> DrawAddress, const char* id = nullptr, int64_t startTime = -1 );
@@ -282,6 +283,7 @@ private:
     bool m_disconnectIssued = false;
     DecayValue<uint64_t> m_drawThreadMigrations = 0;
     DecayValue<uint64_t> m_drawThreadHighlight = 0;
+    Annotation* m_selectedAnnotation = nullptr;
 
     Region m_highlight;
     Region m_highlightZoom;