From 6866efad713b35fef16459d866ae96ec28cfe9f0 Mon Sep 17 00:00:00 2001
From: Ishan Khare <ishankhare07@gmail.com>
Date: Mon, 29 Jun 2015 15:53:23 +0530
Subject: [PATCH] issue #174 added a perform snapshot button

---
 rbkit-lib/subscriber.cpp       |  4 ++++
 rbkit-lib/subscriber.h         |  1 +
 rbkit-lib/ui/actiontoolbar.cpp | 12 ++++++++++++
 rbkit-lib/ui/actiontoolbar.h   |  3 +++
 4 files changed, 20 insertions(+)

diff --git a/rbkit-lib/subscriber.cpp b/rbkit-lib/subscriber.cpp
index 1b5b709..fcdb115 100644
--- a/rbkit-lib/subscriber.cpp
+++ b/rbkit-lib/subscriber.cpp
@@ -74,6 +74,10 @@ void Subscriber::takeSnapshot()
    commandSocket->sendCommand(triggerSnapshot);
 }
 
+void Subscriber::startCPUSnapshot() {
+    qDebug() <<"Start CPU Snapshot command received";
+}
+
 void Subscriber::startSubscriber()
 {
     qDebug() << "** (start subscriber) Thread id is : " << QThread::currentThreadId();
diff --git a/rbkit-lib/subscriber.h b/rbkit-lib/subscriber.h
index 1e069b4..2e41665 100644
--- a/rbkit-lib/subscriber.h
+++ b/rbkit-lib/subscriber.h
@@ -83,6 +83,7 @@ public slots:
     void startCPUProfiling();
     void stopCPUProfiling();
     void takeSnapshot();
+    void startCPUSnapshot();
     void startSubscriber();
 
 private:
diff --git a/rbkit-lib/ui/actiontoolbar.cpp b/rbkit-lib/ui/actiontoolbar.cpp
index 1f48402..30970d4 100644
--- a/rbkit-lib/ui/actiontoolbar.cpp
+++ b/rbkit-lib/ui/actiontoolbar.cpp
@@ -70,6 +70,7 @@ void ActionToolbar::setupSubscriber()
     connect(this, SIGNAL(takeSnapshot()), subscriber, SLOT(takeSnapshot()));
     connect(this, SIGNAL(startCPUProfiling()), subscriber, SLOT(startCPUProfiling()));
     connect(this, SIGNAL(stopCPUProfiling()), subscriber, SLOT(stopCPUProfiling()));
+    connect(this, SIGNAL(startCPUSnapshot()), subscriber, SLOT(startCPUSnapshot()));
 
     connect(subscriber, &Subscriber::errored, centralWidget, &CentralWidget::onError);
     connect(subscriber, &Subscriber::connected, this, &ActionToolbar::connectedToSocket);
@@ -156,8 +157,15 @@ void ActionToolbar::setupToolBar()
                              "stop_cpu_profiling",
                              QIcon(":/icons/Compare-32.png"),
                              "cpu_profiling");
+
     connect(stopCPUButton, &QToolButton::clicked, this, &ActionToolbar::performStopCPUAction);
 
+    cpusnapshotButton = toolBar->addRibbonAction("Take Snapshot",
+                                              "start_snapshot",
+                                              QIcon(":/icons/snapshot-32.png"),
+                                                 "cpu_profiling");
+    connect(cpusnapshotButton, &QToolButton::clicked, this, &ActionToolbar::performCPUSnapshot);
+
     toolBar->loadStyleSheet(":/icons/style.css");
 }
 
@@ -174,6 +182,10 @@ void ActionToolbar::performStopCPUAction() {
     emit stopCPUProfiling();
 }
 
+void ActionToolbar::performCPUSnapshot() {
+    emit startCPUSnapshot();
+}
+
 void ActionToolbar::takeSnapshotAction()
 {
     if (centralWidget->attemptMemorySnapshot()) {
diff --git a/rbkit-lib/ui/actiontoolbar.h b/rbkit-lib/ui/actiontoolbar.h
index 47edbac..bfbd14f 100644
--- a/rbkit-lib/ui/actiontoolbar.h
+++ b/rbkit-lib/ui/actiontoolbar.h
@@ -53,6 +53,7 @@ class ActionToolbar : public QObject
     QToolButton *stopCPUButton;
     QToolButton *connectButton;
     QToolButton *snapshotButton;
+    QToolButton *cpusnapshotButton;
     QToolButton *compareSnapshotButton;
 signals:
     void connectToSocket(QString commandSocket, QString eventSocket);
@@ -61,11 +62,13 @@ class ActionToolbar : public QObject
     void stopCPUProfiling();
     void disconnectSubscriber();
     void takeSnapshot();
+    void startCPUSnapshot();
 public slots:
     void performGCAction();
     void performStartCPUAction();
     void performStopCPUAction();
     void takeSnapshotAction();
+    void performCPUSnapshot();
     void attemptConnection();
     void compareSnapshots();
     void useSelectedHost(QString commandSocket, QString eventSocket);