From 5b00127d7de8a3c107f475373b12dbaf2f56dbc3 Mon Sep 17 00:00:00 2001
From: Christian Hergert <chergert@redhat.com>
Date: Wed, 24 May 2023 14:30:13 -0700
Subject: [PATCH] libsysprof-analyze: add API to get only samples

We'll probably want this for allocations too at some point.
---
 src/libsysprof-analyze/sysprof-document.c | 26 ++++++++++++++++++++++-
 src/libsysprof-analyze/sysprof-document.h |  2 ++
 2 files changed, 27 insertions(+), 1 deletion(-)

diff --git a/src/libsysprof-analyze/sysprof-document.c b/src/libsysprof-analyze/sysprof-document.c
index 64e76ea5..99095e9b 100644
--- a/src/libsysprof-analyze/sysprof-document.c
+++ b/src/libsysprof-analyze/sysprof-document.c
@@ -62,6 +62,7 @@ struct _SysprofDocument
   SysprofStrings           *strings;
 
   GtkBitset                *file_chunks;
+  GtkBitset                *samples;
   GtkBitset                *traceables;
   GtkBitset                *processes;
   GtkBitset                *mmaps;
@@ -216,6 +217,7 @@ sysprof_document_finalize (GObject *object)
   g_clear_pointer (&self->overlays, gtk_bitset_unref);
   g_clear_pointer (&self->pids, gtk_bitset_unref);
   g_clear_pointer (&self->processes, gtk_bitset_unref);
+  g_clear_pointer (&self->samples, gtk_bitset_unref);
   g_clear_pointer (&self->traceables, gtk_bitset_unref);
 
   g_clear_object (&self->counters);
@@ -255,6 +257,7 @@ sysprof_document_init (SysprofDocument *self)
   self->overlays = gtk_bitset_new_empty ();
   self->pids = gtk_bitset_new_empty ();
   self->processes = gtk_bitset_new_empty ();
+  self->samples = gtk_bitset_new_empty ();
   self->traceables = gtk_bitset_new_empty ();
 
   self->files_first_position = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, NULL);
@@ -628,7 +631,11 @@ sysprof_document_load_worker (GTask        *task,
       switch ((int)tainted->type)
         {
         case SYSPROF_CAPTURE_FRAME_ALLOCATION:
+          gtk_bitset_add (self->traceables, self->frames->len);
+          break;
+
         case SYSPROF_CAPTURE_FRAME_SAMPLE:
+          gtk_bitset_add (self->samples, self->frames->len);
           gtk_bitset_add (self->traceables, self->frames->len);
           break;
 
@@ -950,7 +957,7 @@ sysprof_document_list_files (SysprofDocument *self)
  * Gets a #GListModel containing #SysprofDocumentTraceable found within
  * the #SysprofDocument.
  *
- * Returns: (transfer full): a #GListModel of #SysprofTraceable
+ * Returns: (transfer full): a #GListModel of #SysprofDocumentTraceable
  */
 GListModel *
 sysprof_document_list_traceables (SysprofDocument *self)
@@ -960,6 +967,23 @@ sysprof_document_list_traceables (SysprofDocument *self)
   return _sysprof_document_bitset_index_new (G_LIST_MODEL (self), self->traceables);
 }
 
+/**
+ * sysprof_document_list_samples:
+ * @self: a #SysprofDocument
+ *
+ * Gets a #GListModel containing #SysprofDocumentSample found within
+ * the #SysprofDocument.
+ *
+ * Returns: (transfer full): a #GListModel of #SysprofDocumentSample
+ */
+GListModel *
+sysprof_document_list_samples (SysprofDocument *self)
+{
+  g_return_val_if_fail (SYSPROF_IS_DOCUMENT (self), NULL);
+
+  return _sysprof_document_bitset_index_new (G_LIST_MODEL (self), self->samples);
+}
+
 /**
  * sysprof_document_list_processes:
  * @self: a #SysprofDocument
diff --git a/src/libsysprof-analyze/sysprof-document.h b/src/libsysprof-analyze/sysprof-document.h
index 26cd392c..6413065e 100644
--- a/src/libsysprof-analyze/sysprof-document.h
+++ b/src/libsysprof-analyze/sysprof-document.h
@@ -44,6 +44,8 @@ GListModel             *sysprof_document_list_files          (SysprofDocument
 SYSPROF_AVAILABLE_IN_ALL
 GListModel             *sysprof_document_list_traceables     (SysprofDocument           *self);
 SYSPROF_AVAILABLE_IN_ALL
+GListModel             *sysprof_document_list_samples        (SysprofDocument           *self);
+SYSPROF_AVAILABLE_IN_ALL
 GListModel             *sysprof_document_list_processes      (SysprofDocument           *self);
 SYSPROF_AVAILABLE_IN_ALL
 GListModel             *sysprof_document_list_jitmaps        (SysprofDocument           *self);