added sqlitestorage, refs #12

Author: chriamue

source/annotatorlib/include/AnnotatorLib/Storage/MySQLStorage.h

@@ -115,7 +115,7 @@ class ANNOTATORLIB_API MySQLStorage : public AbstractStorage {
    * @brief createTables
    * creates tables in mysql database if they not exist
    */
-  void createTables();
+  virtual void createTables();
   Poco::Data::Statement getStatement();
 };
 /************************************************************/

source/annotatorlib/include/AnnotatorLib/Storage/SQLiteStorage.h

@@ -0,0 +1,55 @@
+// Copyright 2016 Annotator Team
+#ifndef ANNOTATOR_ANNOTATORLIB_STORAGE_SQLITESTORAGE_H
+#define ANNOTATOR_ANNOTATORLIB_STORAGE_SQLITESTORAGE_H
+
+/************************************************************
+ MySQLStorage class header
+ ************************************************************/
+#include <AnnotatorLib/Storage/AbstractStorage.h>
+#include <AnnotatorLib/Storage/Pkg_Storage.h>
+#include <AnnotatorLib/annotatorlib_api.h>
+#include <AnnotatorLib/Storage/MySQLStorage.h>
+#include <QJsonObject>
+
+#include <Poco/Data/SessionPool.h>
+
+namespace AnnotatorLib {
+namespace Storage {
+
+/************************************************************/
+/**
+ * @brief The SQLiteStorage class
+ */
+class ANNOTATORLIB_API SQLiteStorage : public MySQLStorage {
+
+ public:
+  ~SQLiteStorage();
+
+  bool open();
+
+  /**
+   * @brief setPath
+   * @param path
+   * Path should be like:
+   * "annotator.db";
+   */
+  void setPath(std::string path) override;
+
+  StorageType getType() override;
+
+  void createTables() override;
+};
+/************************************************************/
+/* External declarations (package visibility)               */
+/************************************************************/
+
+/* Inline functions                                         */
+
+}  // of namespace Storage
+}  // of namespace AnnotatorLib
+
+/************************************************************
+ End of SQLiteStorage class header
+ ************************************************************/
+
+#endif

source/annotatorlib/source/Storage/SQLiteStorage.cpp

@@ -0,0 +1,96 @@
+// Copyright 2016 Annotator Team
+#define Annotator_AnnotatorLib_Storage_SQLiteStorage_BODY
+
+/************************************************************
+ SQLiteStorage class body
+ ************************************************************/
+
+// include associated header file
+#include <AnnotatorLib/Loader/SQLiteLoader.h>
+#include <AnnotatorLib/Saver/SQLiteSaver.h>
+#include <AnnotatorLib/Storage/SQLiteStorage.h>
+
+#include <Poco/Data/SQLite/Connector.h>
+#include <Poco/Data/RecordSet.h>
+#include <Poco/Data/Session.h>
+#include <Poco/Data/Statement.h>
+
+using Poco::Data::Session;
+using Poco::Data::Statement;
+using namespace Poco::Data::Keywords;
+
+namespace AnnotatorLib {
+namespace Storage {
+
+void SQLiteStorage::setPath(std::string path) { this->path = path; }
+
+StorageType SQLiteStorage::getType() { return AnnotatorLib::StorageType::SQLITE; }
+
+SQLiteStorage::~SQLiteStorage() {
+  if (pool) delete pool;
+}
+
+bool SQLiteStorage::open() {
+  Poco::Data::SQLite::Connector::registerConnector();
+  pool = new Poco::Data::SessionPool("SQLite", this->path);
+  createTables();
+
+  AnnotatorLib::Loader::SQLiteLoader loader;
+  loader.setPath(this->path);
+  loader.loadSession(this);
+
+  this->_open = true;
+  return _open;
+}
+
+void SQLiteStorage::createTables() {
+  Poco::Data::Statement statement = getStatement();
+
+  statement = getStatement();
+
+  statement << "PRAGMA encoding = \"UTF-8\";";
+  statement.execute();
+
+  statement << "CREATE TABLE IF NOT EXISTS `annotations` ( \
+           `id` char(16) NOT NULL, \
+            `next` char(16) NOT NULL, \
+            `previous` char(16) NOT NULL, \
+            `object` char(16) NOT NULL, \
+            `frame` char(16) NOT NULL, \
+            `x` float NOT NULL default 0, \
+            `y` float NOT NULL default 0, \
+           `width` float NOT NULL default 1, \
+            `height` float NOT NULL default 1, \
+            `type` char(16) NOT NULL, \
+            PRIMARY KEY (`id`) \
+            ) DEFAULT CHARSET=utf8;";
+  statement.execute();
+
+  statement = getStatement();
+
+  statement << "CREATE TABLE IF NOT EXISTS `classes` ("
+            << "`id` char(16) NOT NULL, "
+            << "`name` varchar(256) NOT NULL,"
+            << "PRIMARY KEY (`id`)"
+            << ") DEFAULT CHARSET=utf8;";
+  statement.execute();
+
+  statement = getStatement();
+
+  statement << "CREATE TABLE IF NOT EXISTS `objects` ("
+            << "`id` char(16) NOT NULL, "
+            << "`name` varchar(256) NOT NULL,"
+            << "`class` char(16) NOT NULL,"
+            << "PRIMARY KEY (`id`)"
+            << ") DEFAULT CHARSET=utf8;";
+  statement.execute();
+}
+
+// static attributes (if any)
+
+}  // of namespace Storage
+}  // of namespace AnnotatorLib
+
+/************************************************************
+ End of SQLiteStorage class body
+ ************************************************************/

source/annotatorlib/source/Storage/StorageFactory.cpp

@@ -10,6 +10,7 @@
 // include associated header file
 #include "AnnotatorLib/Storage/JSONStorage.h"
 #include "AnnotatorLib/Storage/MySQLStorage.h"
+#include "AnnotatorLib/Storage/SQLiteStorage.h"
 #include "AnnotatorLib/Storage/StorageFactory.h"
 #include "AnnotatorLib/Storage/XMLStorage.h"
 
@@ -38,6 +39,7 @@ shared_ptr<AbstractStorage> StorageFactory::createStorage(
     case StorageType::MYSQL:
       return std::make_shared<MySQLStorage>();
     case StorageType::SQLITE:
+      return std::make_shared<SQLiteStorage>();
     default:
       return nullptr;
   }