TableHelper.hpp

Go to the documentation of this file.

#pragma once

#include <boost/unordered_map.hpp>
#include <boost/shared_ptr.hpp>
#include <string>
#include <map>
#include <set>

#include "soci/src/core/soci.h"
#include "soci/src/backends/sqlite3/soci-sqlite3.h"
#include "common_type/Object.hpp"
#include "common_type/ObjectSet.hpp"
#include "common_type/RecordedPattern.hpp"
#include "common_type/VoteSpecifier.hpp"
#include "common_type/Pattern.hpp"

using namespace soci;
namespace ISM {
  typedef std::map<std::string, std::vector<VoteSpecifierPtr> > ObjectTypeToVoteMap;
  typedef std::map<std::string, PatternPtr> PatternNameToPatternMap;
  //Access votes for a specific object (given as combination of object type and id).
  typedef std::map<std::string, std::map<std::string, std::vector<VoteSpecifierPtr> > > ObjectToVoteMap;
  //Access votes for a specific object (given as combination of object type and id) in a specific scene.
  typedef std::map<std::string, std::map<std::string, std::map<std::string, std::vector<VoteSpecifierPtr> > > > PatternToObjectToVoteMap;

  class TableHelper {
    boost::shared_ptr<session> sqlite;
  public:
    /* util */
    TableHelper(std::string dbfilename = "record.sqlite");

    ~TableHelper();

    void createTablesIfNecessary() const;
    void createColumnsIfNecessary();
    void createTable(const std::string& tablename, const std::string& sql) const;
    void dropTable(const std::string& tablename) const;
    void dropRecordTables() const;
    void dropTables() const;
    void dropModelTables() const;

    bool recordDataExists() const;
    bool modelDataExists() const;

    int getLastInsertId(const std::string& tablename) const;

    const std::set<std::string> getObjectsInPattern(std::string pattern_name);


    /* Functions that are used by recorder to insert object data into table. */
    int insertRecordedObject(const boost::shared_ptr<Object>& o, int setId);
    int insertRecordedObjectSet(const boost::shared_ptr<ObjectSet>& os, const std::string& patternName);
    int insertRecordedPattern(const std::string& patternName) const;
    int insertRecordedPattern(const RecordedPatternPtr& pattern) const;

    int getRecordedPatternId(const std::string& patternName) const;
    int ensureRecordedPatternName(const std::string& patternName) const;
      
    std::vector<std::string> getRecordedPatternNames() const;

    const RecordedPatternPtr getRecordedPattern(const std::string& patternName) const;
    const std::vector<int> getSetIds() const;
    const ObjectSetPtr getRecordedObjectSet(int setId) const;

    /* model */
    int insertModelVoteSpecifier(const VoteSpecifierPtr& vote) const;
    int insertModelPattern(const std::string& patternName) const;
    int upsertModelPattern(const std::string& patternName, int expectedMaxWeight) const;
    int insertModelObjectType(const std::string& objectType) const;

    int getModelPatternId(const std::string& patternName) const;
    int getModelObjectTypeId(const std::string& objectType) const;
    int ensureModelPatternName(const std::string& patternName) const;
    int ensureModelObjectType(const std::string& objectType) const;

    std::vector<std::string> getModelPatternNames() const;

    const PatternNameToPatternMap getPatternDefinitionsByName(const std::set<std::string>& patternNames) const;

    const std::set<std::string> getObjectTypes() const;

    const std::map<std::string, boost::filesystem::path> getRessourcePaths() const;

    const std::map<std::string, std::map<std::string, std::string>> getModelWeightsPerTypeAndId() const;

    const std::set<std::pair<std::string, std::string> > getObjectTypesAndIdsFromModelObjects() const;
    const std::set<std::pair<std::string, std::string> > getObjectTypesAndIdsFromRecordedObjects() const;
    const ObjectTypeToVoteMap getVoteSpecifiersForObjectTypes(const std::set<std::string>& objectTypes) const;

    const std::set<std::pair<std::string, std::string> > getObjectTypesAndIdsBelongingToPattern(const std::string& patternName) const;
      
    const ObjectToVoteMap getVoteSpecifiersForPatternAndObjects(const std::string& patternName, std::set<std::pair<std::string, std::string> > objects) const;

    const std::vector<std::pair<int, ObjectPtr>> getAllMarkerObjects();

    int updateObjectQuaternion(int dbId, ObjectPtr object);
private:
    int insertRecordedObjectIfNecessary(ObjectPtr object);
    boost::unordered_map<std::pair<std::string, std::string>, int> objectIdMap_;
  };

  typedef boost::shared_ptr<TableHelper> TableHelperPtr;
}