3 files changed, 103 insertions, 1 deletions

diff --git a/drumgizmo/Makefile.am b/drumgizmo/Makefile.am
index aaef1ee..687485d 100644
--- a/drumgizmo/Makefile.am
+++ b/drumgizmo/Makefile.am
@@ -25,6 +25,7 @@ drumgizmo_SOURCES = \
 EXTRA_DIST = \
 	audioinputenginedl.h \
 	audiooutputenginedl.h \
+	drumgizmoc.h \
 	jackclient.h
 
 endif
diff --git a/drumgizmo/drumgizmoc.cc b/drumgizmo/drumgizmoc.cc
index 61f8eaa..d8479fa 100644
--- a/drumgizmo/drumgizmoc.cc
+++ b/drumgizmo/drumgizmoc.cc
@@ -30,8 +30,10 @@
 #include <string.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <unistd.h>
 
 #include "drumgizmo.h"
+#include "drumgizmoc.h"
 
 #include "audiooutputenginedl.h"
 #include "audioinputenginedl.h"
@@ -64,7 +66,29 @@ static const char usage_str[] =
 "  -h, --help             Print this message and exit.\n"
 ;
 
-int main(int argc, char *argv[])
+CliMain::CliMain() : MessageReceiver(MSGRCV_UI), loading(false)
+{}
+
+CliMain::~CliMain()
+{}
+
+void CliMain::handleMessage(Message *msg)
+{
+  switch(msg->type()) {
+  case Message::LoadStatus:
+    {
+      LoadStatusMessage *ls = (LoadStatusMessage*)msg;
+      if(ls->numer_of_files_loaded == ls->number_of_files) {
+        loading = false;
+      }
+    }
+    break;
+  default:
+    break;
+  }
+}
+
+int CliMain::run(int argc, char *argv[])
 {
   int c;
 
@@ -243,11 +267,27 @@ int main(int argc, char *argv[])
   printf("Using kitfile: %s\n", kitfile.c_str());
 
   DrumGizmo gizmo(oe, ie);
+
   if(kitfile == "" || !gizmo.loadkit(kitfile)) {
     printf("Failed to load \"%s\".\n", kitfile.c_str());
     return 1;
   }
 
+  printf("Loading drumkit, this may take a while...");
+  fflush(stdout);
+  loading = true;
+  while (loading) {
+#ifdef WIN32
+    SleepEx(500, FALSE);
+#else
+    usleep(500000);
+#endif/*WIN32*/
+    handleMessages();
+    printf(".");
+    fflush(stdout);
+  }
+  printf("done.\n");
+
   gizmo.setSamplerate(oe->samplerate());
 
   if(!gizmo.init(preload)) {
@@ -264,3 +304,19 @@ int main(int argc, char *argv[])
 
   return 0;
 }
+
+int main(int argc, char *argv[])
+{
+
+  CliMain* cli = new CliMain();
+  if (cli == NULL) {
+    printf("Could not initialize command line client\n");
+    return 1;
+  }
+
+  cli->run(argc, argv);
+  delete cli;
+
+  return 0;
+
+}
diff --git a/drumgizmo/drumgizmoc.h b/drumgizmo/drumgizmoc.h
new file mode 100644
index 0000000..4b6d4a1
--- /dev/null
+++ b/drumgizmo/drumgizmoc.h
@@ -0,0 +1,45 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/***************************************************************************
+ *            drumgizmoc.h
+ *
+ *  Mon Jan 12 00:36:37 CET 2015
+ *  Copyright 2015 Sergey 'Jin' Bostandzhyan
+ *  jin@mediatomb.cc
+ ****************************************************************************/
+
+/*
+ *  This file is part of DrumGizmo.
+ *
+ *  DrumGizmo is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; either version 2 of the License, or
+ *  (at your option) any later version.
+ *
+ *  DrumGizmo is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with DrumGizmo; if not, write to the Free Software
+ *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA.
+ */
+
+#ifndef __DRUMGIZMOC_H__
+#define __DRUMGIZMOC_H__
+
+#include "messagereceiver.h"
+
+class CliMain : public MessageReceiver {
+public:
+  CliMain();
+  virtual ~CliMain();
+
+  int run(int argc, char *argv[]);
+  void handleMessage(Message *msg);
+private:
+  bool loading;
+};
+
+#endif//__DRUMGIZMOC_H__
+