Improvements

core/src/main/java/com/github/grishberg/profiler/chart/CallTracePanel.java

@@ -93,6 +93,8 @@ public class CallTracePanel extends JPanel
     private int currentFocusedFoundElement = -1;
     private int currentSelectedElement = -1;
     private final ArrayList<ProfileRectangle> foundItems = new ArrayList<>();
+    private double foundTotalGlobalDuration;
+    private double foundTotalThreadDuration;
     private final CalledStacktrace calledStacktrace;
     private Grid scale;
     private final SettingsFacade settings;
@@ -945,10 +947,15 @@ public void renderFoundItems(Finder.ThreadFindResult threadFindResult) {
 
         currentFocusedFoundElement = 0;
         ProfileRectangle element = foundItems.get(currentFocusedFoundElement);
+
+        foundTotalGlobalDuration = threadFindResult.getTotalGlobalDuration();
+        foundTotalThreadDuration = threadFindResult.getTotalThreadDuration();
         foundNavigationListener.onSelected(
             foundItems.size(),
             currentFocusedFoundElement,
-            element.profileData
+            element.profileData,
+            foundTotalGlobalDuration,
+            foundTotalThreadDuration
         );
         zoomAndPanDelegate.fitZoom(element, FIT_PADDING, ZoomAndPanDelegate.VerticalAlign.ENABLED);
 
@@ -1070,7 +1077,9 @@ private void focusFoundItem(int currentFocusedFoundElement) {
         foundNavigationListener.onSelected(
             foundItems.size(),
             currentFocusedFoundElement,
-            found.profileData
+            found.profileData,
+            foundTotalGlobalDuration,
+            foundTotalThreadDuration
         );
     }
 

core/src/main/java/com/github/grishberg/profiler/chart/Finder.kt

@@ -134,6 +134,8 @@ class Finder(
             val shouldEndsWithText: Boolean = textToFind.endsWith("()")
             val targetString = prepareTextToFind(shouldEndsWithText, textToFind, ignoreCase)
 
+            var totalGlobalDuration = 0.0
+            var totalThreadDuration = 0.0
             val foundMethods = mutableSetOf<ProfileData>()
             for (i in profileList.indices) {
                 val currentMethod = profileList[i]
@@ -144,11 +146,13 @@ class Finder(
                     )
                 if (isEquals) {
                     foundMethods.add(currentMethod)
+                    totalGlobalDuration = currentMethod.globalEndTimeInMillisecond - currentMethod.globalStartTimeInMillisecond
+                    totalThreadDuration = currentMethod.threadEndTimeInMillisecond - currentMethod.threadStartTimeInMillisecond
                 }
             }
             if (foundMethods.isNotEmpty()) {
                 val threadItem = analyzerResult.threads.find { it.threadId == threadId }!!
-                result.add(ThreadFindResult(foundMethods, threadId, threadItem))
+                result.add(ThreadFindResult(foundMethods, threadId, threadItem, totalGlobalDuration, totalThreadDuration))
             }
         }
 
@@ -191,7 +195,9 @@ class Finder(
     data class ThreadFindResult(
         val foundResult: Set<ProfileData>,
         val threadId: Int,
-        val threadItem: ThreadItem
+        val threadItem: ThreadItem,
+        val totalGlobalDuration: Double,
+        val totalThreadDuration: Double
     ) {
         fun hasMethod(method: ProfileData) = foundResult.contains(method)
     }

core/src/main/java/com/github/grishberg/profiler/chart/FoundNavigationListener.kt

@@ -1,7 +1,8 @@
 package com.github.grishberg.profiler.chart
 
 interface FoundNavigationListener<T> {
-    fun onSelected(count: Int, selectedIndex: Int, selectedElement: T)
+    fun onSelected(count: Int, selectedIndex: Int, selectedElement: T,
+        totalGlobalDuration: Double, totalThreadDuration: Double)
     fun onNavigatedOverLastItem()
     fun onNavigatedOverFirstItem()
 }

core/src/main/java/com/github/grishberg/profiler/chart/threads/ThreadsViewDialog.kt

@@ -9,14 +9,19 @@ import com.github.grishberg.profiler.ui.dialogs.CloseByEscapeDialog
 import java.awt.Color
 import java.awt.Dimension
 import java.awt.Frame
+import java.awt.Point
 import java.awt.event.ActionEvent
 import java.awt.event.KeyEvent
 import java.awt.event.MouseAdapter
 import java.awt.event.MouseEvent
-import javax.swing.*
+import javax.swing.AbstractAction
+import javax.swing.JScrollPane
+import javax.swing.JTable
+import javax.swing.JViewport
+import javax.swing.KeyStroke
+import javax.swing.ListSelectionModel
 import kotlin.math.max
 
-
 private const val THREAD_NAME_WIDTH = 180
 private const val SELECT_ACTION = "Select"
 
@@ -32,6 +37,15 @@ class ThreadsViewDialog(
     var selectedThreadItem: ThreadItem? = null
         private set
 
+    private var _selectedRow: Int = -1
+    private var _verticalScrollPosition: Int = 0
+
+    val selectedRow: Int
+        get() = _selectedRow
+
+    val scrollPosition: Int
+        get() = _verticalScrollPosition
+
     private val table = object : JTable(model) {
         override fun getRowHeight(): Int {
             return max(super.getRowHeight(), PREVIEW_IMAGE_HEIGHT)
@@ -41,6 +55,7 @@ class ThreadsViewDialog(
             return false
         }
     }
+    private val scrollPane: JScrollPane
 
     init {
         controller.view = this
@@ -66,13 +81,14 @@ class ThreadsViewDialog(
         table.showHorizontalLines = true
         table.gridColor = Color.GRAY
 
-        val scrollPane = JScrollPane(table)
+        scrollPane = JScrollPane(table)
         scrollPane.preferredSize = Dimension(PREVIEW_IMAGE_WIDTH + THREAD_NAME_WIDTH + 32, 600)
         add(scrollPane)
         pack()
     }
 
     private inner class EnterAction : AbstractAction() {
+
         override fun actionPerformed(e: ActionEvent) {
             val selected = table.selectedRow
             if (selected < 0) {
@@ -83,6 +99,9 @@ class ThreadsViewDialog(
     }
 
     private fun setResultAndClose(selected: Int) {
+        val viewport = scrollPane.viewport
+        _selectedRow = selected
+        _verticalScrollPosition = viewport.viewPosition.y;
         val modelRowIndex = table.convertRowIndexToModel(selected)
         selectedThreadItem = model.getThreadInfo(modelRowIndex)
         isVisible = false
@@ -91,4 +110,13 @@ class ThreadsViewDialog(
     override fun showThreads(threads: List<ThreadItem>) {
         model.setData(threads)
     }
+
+    fun restoreSelection(selectedRow: Int, verticalScrollPosition: Int) {
+        if (selectedRow < 0) {
+            return
+        }
+        table.setRowSelectionInterval(selectedRow, selectedRow)
+        val viewport: JViewport = scrollPane.viewport
+        viewport.viewPosition = Point(0, verticalScrollPosition)
+    }
 }