small fix

[proxmark3-svn] / client / proxgui.cpp

diff --git a/client/proxgui.cpp b/client/proxgui.cpp
index 7dcb0da0fec3cbe8765049d6ca69a2af40c3fa0f..e899174c419870e2e78014e609f0f3faa2f38f04 100644 (file)
--- a/client/proxgui.cpp
+++ b/client/proxgui.cpp
@@ -10,8 +10,22 @@
 
 #include "proxgui.h"
 #include "proxguiqt.h"
+#include "proxmark3.h"
 
 static ProxGuiQT *gui = NULL;
+static WorkerThread *main_loop_thread = NULL;
+
+WorkerThread::WorkerThread(char *script_cmds_file, char *script_cmd, bool usb_present) : script_cmds_file(script_cmds_file), script_cmd(script_cmd), usb_present(usb_present)
+{
+}
+
+WorkerThread::~WorkerThread() 
+{
+}
+
+void WorkerThread::run() {
+       main_loop(script_cmds_file, script_cmd, usb_present);
+}
 
 extern "C" void ShowGraphWindow(void)
 {
@@ -39,31 +53,31 @@ extern "C" void RepaintGraphWindow(void)
 
 extern "C" void MainGraphics(void)
 {
-  if (!gui)
-    return;
+       if (!gui)
+               return;
 
-  gui->MainLoop();
+       gui->MainLoop();
 }
 
-extern "C" void InitGraphics(int argc, char **argv)
+extern "C" void InitGraphics(int argc, char **argv, char *script_cmds_file, char *script_cmd, bool usb_present)
 {
 #ifdef Q_WS_X11
-  bool useGUI = getenv("DISPLAY") != 0;
+       bool useGUI = getenv("DISPLAY") != 0;
 #else
-  bool useGUI = true;
+       bool useGUI = true;
 #endif
-  if (!useGUI)
-    return;
+       if (!useGUI)
+               return;
 
-  gui = new ProxGuiQT(argc, argv);
+       main_loop_thread = new WorkerThread(script_cmds_file, script_cmd, usb_present);
+       gui = new ProxGuiQT(argc, argv, main_loop_thread);
 }
 
 extern "C" void ExitGraphics(void)
 {
-  if (!gui)
-    return;
+       if (!gui)
+               return;
 
-  gui->Exit();
-  //delete gui;
-  gui = NULL;
+       gui->Exit();
+       gui = NULL;
 }