Added console mode to Find-Object (see "--help"):

> Find-Object.exe -console -objs "Path objects" git-svn-id: http://find-object.googlecode.com/svn/trunk/find_object@278 620bd6b2-0a58-f614-fd9a-1bd335dccda9
2014-05-21 15:24:24 +00:00
parent 17873bacfa
commit b4d04c9460
5 changed files with 168 additions and 25 deletions
@@ -1,18 +1,145 @@
 #include <QtGui/QApplication>
+#include <QtCore/QDir>
+#include <QtCore/QFile>
 #include "MainWindow.h"
+#include "Settings.h"
+
+#include <signal.h>
+
+void my_handler(int s){
+	printf("Ctrl-C catch! Quitting application...\n");
+	QApplication::closeAllWindows();
+	QApplication::quit();
+}
+
+void showUsage()
+{
+	printf("\nUsage:\n"
+#ifdef WIN32
+			"  Find-Object.exe [options]\n"
+#else
+			"  find_object [options]\n"
+#endif
+			"Options:\n"
+			"  -console          Don't use the GUI (by default the camera will be\n"
+			"                    started automatically). Option -objs must also be\n"
+			"                    used with valid objects.\n"
+			"  -objs \"path\"      Directory of the objects to detect.\n"
+			"  -config \"path\"    Path to configuration file (default: %s).\n"
+			"  -help or --help   Show usage.\n", Settings::iniDefaultPath().toStdString().c_str());
+	exit(-1);
+}

 int main(int argc, char* argv[])
 {
-	/* Create tasks */
 	QApplication app(argc, argv);
-	MainWindow mainWindow;

-	mainWindow.showNormal();
+	// parse options
+	bool guiMode = true;
+	QString objectsPath = "";
+	QString configPath = Settings::iniDefaultPath();

-	// Now wait for application to finish
-	app.connect( &app, SIGNAL( lastWindowClosed() ),
-				&app, SLOT( quit() ) );
-	app.exec();// MUST be called by the Main Thread
+	for(int i=1; i<argc; ++i)
+	{
+		if(strcmp(argv[i], "-objs") == 0)
+		{
+			++i;
+			if(i < argc)
+			{
+				objectsPath = argv[i];
+				if(objectsPath.contains('~'))
+				{
+					objectsPath.replace('~', QDir::homePath());
+				}
+				if(!QDir(objectsPath).exists())
+				{
+					printf("[ERROR] Path not valid : %s\n", objectsPath.toStdString().c_str());
+					showUsage();
+				}
+			}
+			else
+			{
+				showUsage();
+			}
+			continue;
+		}
+		if(strcmp(argv[i], "-config") == 0)
+		{
+			++i;
+			if(i < argc)
+			{
+				configPath = argv[i];
+				if(configPath.contains('~'))
+				{
+					configPath.replace('~', QDir::homePath());
+				}
+				if(!QFile::exists(configPath))
+				{
+					printf("[ERROR] Configuration file doesn't exist : %s\n", configPath.toStdString().c_str());
+					showUsage();
+				}
+			}
+			else
+			{
+				showUsage();
+			}
+			continue;
+		}
+		if(strcmp(argv[i], "-console") == 0)
+		{
+			guiMode = false;
+			continue;
+		}
+		if(strcmp(argv[i], "-help") == 0 || strcmp(argv[i], "--help") == 0)
+		{
+			showUsage();
+		}

-    return 0;
+		printf("[ERROR] Unrecognized option : %s\n", argv[i]);
+		showUsage();
+	}
+
+	printf("Options:\n");
+	printf("   GUI mode = %s\n", guiMode?"true":false);
+	printf("   Objects path: \"%s\"\n", objectsPath.toStdString().c_str());
+	printf("   Settings path: \"%s\"\n", configPath.toStdString().c_str());
+
+
+	MainWindow mainWindow(0, configPath);
+
+	int objectsLoaded = 0;
+	if(!objectsPath.isEmpty())
+	{
+		objectsLoaded = mainWindow.loadObjects(objectsPath);
+		if(!objectsLoaded)
+		{
+			printf("[WARNING] No objects loaded from \"%s\"\n", objectsPath.toStdString().c_str());
+		}
+	}
+
+	if(objectsLoaded == 0 && !guiMode)
+	{
+		printf("[ERROR] In console mode, at least one object must be loaded! See -console option.\n");
+		showUsage();
+	}
+
+	if(guiMode)
+	{
+		app.connect( &app, SIGNAL( lastWindowClosed() ), &app, SLOT( quit() ) );
+		mainWindow.show();
+	}
+	else
+	{
+		mainWindow.startProcessing();
+	}
+
+
+	// Catch ctrl-c to close the gui
+	struct sigaction sigIntHandler;
+	sigIntHandler.sa_handler = my_handler;
+	sigemptyset(&sigIntHandler.sa_mask);
+	sigIntHandler.sa_flags = 0;
+	sigaction(SIGINT, &sigIntHandler, NULL);
+
+    return app.exec();
 }