1 files changed, 43 insertions, 0 deletions
diff --git a/src/selection.cpp b/src/selection.cpp
new file mode 100644
index 0000000..6f293a6
--- /dev/null
+++ b/src/selection.cpp
@@ -0,0 +1,43 @@
+#include <config.h>
+
+#include "selection.h"
+#include "proctable.h"
+#include "util.h"
+
+namespace procman
+{
+  void SelectionMemento::add_to_selected(GtkTreeModel* model, GtkTreePath*, GtkTreeIter* iter, gpointer data)
+  {
+    guint pid = 0;
+    gtk_tree_model_get(model, iter, COL_PID, &pid, -1);
+    if (pid)
+      static_cast<SelectionMemento*>(data)->pids.push_back(pid);
+  }
+
+
+  void SelectionMemento::save(GtkWidget* tree)
+  {
+    GtkTreeSelection* selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(tree));
+    gtk_tree_selection_selected_foreach(selection, &SelectionMemento::add_to_selected, this);
+  }
+
+
+  void SelectionMemento::restore(GtkWidget* tree)
+  {
+    if (not this->pids.empty())
+      {
+	GtkTreeSelection* selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(tree));
+	typedef std::vector<pid_t>::iterator iterator;
+	for (iterator it(this->pids.begin()); it != this->pids.end(); ++it)
+	  {
+	    if (ProcInfo* proc = ProcInfo::find(*it))
+	      {
+		gtk_tree_selection_select_iter(selection, &proc->node);
+		procman_debug("Re-selected process %u", unsigned(*it));
+	      }
+	    else
+		procman_debug("Could not find process %u, cannot re-select it", unsigned(*it));
+	  }
+      }
+  }
+}