From 6e476b683a05e4b8177d3cbc11ee6d175e2ea551 Mon Sep 17 00:00:00 2001
From: Christian Hergert <chergert@redhat.com>
Date: Thu, 29 Jun 2023 11:02:35 -0700
Subject: [PATCH] libsysprof-gtk: allow finding matching sub-tracks by counter

---
 src/libsysprof-gtk/sysprof-session-discover.c | 74 +++++++++++++++++--
 1 file changed, 67 insertions(+), 7 deletions(-)

diff --git a/src/libsysprof-gtk/sysprof-session-discover.c b/src/libsysprof-gtk/sysprof-session-discover.c
index b4af6be0..b3fca200 100644
--- a/src/libsysprof-gtk/sysprof-session-discover.c
+++ b/src/libsysprof-gtk/sysprof-session-discover.c
@@ -25,24 +25,84 @@
 #include "sysprof-session-private.h"
 #include "sysprof-track.h"
 
+typedef struct _SysprofTrackCounter
+{
+  const char *category;
+  const char *name;
+  const char *subtracks_name_glob;
+  const char *track_name;
+} SysprofTrackCounter;
+
+static const SysprofTrackCounter discovery_counters[] = {
+  { "CPU Percent", "Combined", "Total CPU *", N_("CPU Usage") },
+};
+
+static GListModel *
+filter_counters (GListModel *model,
+                 const char *category,
+                 const char *name_glob)
+{
+  g_autoptr(GListStore) store = NULL;
+  g_autoptr(GPatternSpec) spec = NULL;
+  guint n_items;
+
+  g_assert (G_IS_LIST_MODEL (model));
+  g_assert (category != NULL);
+  g_assert (name_glob != NULL);
+
+  store = g_list_store_new (SYSPROF_TYPE_DOCUMENT_COUNTER);
+  spec = g_pattern_spec_new (name_glob);
+  n_items = g_list_model_get_n_items (model);
+
+  for (guint i = 0; i < n_items; i++)
+    {
+      g_autoptr(SysprofDocumentCounter) counter = g_list_model_get_item (model, i);
+      const char *ctrcat = sysprof_document_counter_get_category (counter);
+      const char *ctrname = sysprof_document_counter_get_name (counter);
+
+      if (g_strcmp0 (category, ctrcat) == 0 &&
+          g_pattern_spec_match (spec, strlen (ctrname), ctrname, NULL))
+        g_list_store_append (store, counter);
+    }
+
+  if (g_list_model_get_n_items (G_LIST_MODEL (store)) == 0)
+    return NULL;
+
+  return G_LIST_MODEL (g_steal_pointer (&store));
+}
+
 void
 _sysprof_session_discover_tracks (SysprofSession  *self,
                                   SysprofDocument *document,
                                   GListStore      *tracks)
 {
-  g_autoptr(SysprofDocumentCounter) cpu = NULL;
+  g_autoptr(GListModel) counters = NULL;
 
   g_assert (SYSPROF_IS_SESSION (self));
   g_assert (SYSPROF_IS_DOCUMENT (document));
   g_assert (G_IS_LIST_STORE (tracks));
 
-  if ((cpu = sysprof_document_find_counter (document, "CPU Percent", "Combined")))
-    {
-      g_autoptr(SysprofTrack) cpu_track = NULL;
+  counters = sysprof_document_list_counters (document);
 
-      cpu_track = g_object_new (SYSPROF_TYPE_TRACK,
-                                "title", _("CPU Usage"),
+  for (guint i = 0; i < G_N_ELEMENTS (discovery_counters); i++)
+    {
+      const SysprofTrackCounter *info = &discovery_counters[i];
+      g_autoptr(SysprofDocumentCounter) counter = NULL;
+
+      if ((counter = sysprof_document_find_counter (document, info->category, info->name)))
+        {
+          g_autoptr(SysprofTrack) track = NULL;
+          g_autoptr(GListModel) subcounters = NULL;
+
+          track = g_object_new (SYSPROF_TYPE_TRACK,
+                                "title", g_dgettext (GETTEXT_PACKAGE, info->track_name),
                                 NULL);
-      g_list_store_append (tracks, cpu_track);
+
+          if ((subcounters = filter_counters (counters, info->category, info->subtracks_name_glob)))
+            {
+            }
+
+          g_list_store_append (tracks, track);
+        }
     }
 }