#ifndef HDTMANAGER_HPP
#define HDTMANAGER_HPP

#include <HDT.hpp>
#include <QObject>

#include "hdtcachedinfo.hpp"
#include "triplecomponentmodel.hpp"
#include "searchresultsmodel.hpp"
#include "headermodel.hpp"
#include "predicatestatus.hpp"
#include "regexmodel.hpp"

class TripleComponentModel;
class SearchResultsModel;
class PredicateStatus;
class HeaderModel;
class RegexModel;

class HDTController : public QObject
{
    Q_OBJECT

public:
    explicit HDTController(QObject *parent = 0);
    ~HDTController();

private:
    // HDT
    hdt::HDT *hdt;
    QString fileName;

    // Models / Info
    TripleComponentModel *subjectModel;
    TripleComponentModel *predicateModel;
    TripleComponentModel *objectModel;
    SearchResultsModel *searchResultsModel;
    HeaderModel *headerModel;
    PredicateStatus *predicateStatus;
    HDTCachedInfo *hdtCachedInfo;
    RegexModel *regexModel;

    // Search pattern / results
    hdt::TripleID selectedTriple;
    hdt::TripleID searchPatternID;
    hdt::TripleString searchPatternString;
    hdt::IteratorTripleID *iteratorResults;
    size_t numResults;

public:
    hdt::HDT *getHDT();
    bool hasHDT();
    void openHDTFile(QString file);
    void saveHDTFile(QString file);
    void importRDFFile(QString file, const string &baseUri, hdt::RDFNotation notation, hdt::HDTSpecification &spec);
    void exportRDFFile(QString file, hdt::RDFNotation notation);
    void exportResultsRDFFile(QString file, hdt::RDFNotation notation);
    void closeHDT();

    TripleComponentModel *getSubjectModel();
    TripleComponentModel *getPredicateModel();
    TripleComponentModel *getObjectModel();
    RegexModel *getRegexModel();
    SearchResultsModel *getSearchResultsModel();
    HeaderModel *getHeaderModel();
    PredicateStatus *getPredicateStatus();
    HDTCachedInfo *getHDTCachedInfo();

    void selectNearestTriple(int subject, int predicate, int object);
    void setSelectedTriple(hdt::TripleID &selected);
    hdt::TripleID getSelectedTriple();
    void clearSelectedTriple();

    hdt::TripleID &getSearchPatternID();
    hdt::TripleString &getSearchPatternString();
    void setSearchPattern(hdt::TripleString &pattern);

    vector<hdt::TripleID> &getTriples();

    size_t getNumResults();

    QString getFileName();

signals:
    void datasetChanged();
    void searchPatternChanged();
    void numResultsChanged(int numResults);

private slots:
    void updateOnHDTChanged();
    void updateNumResults();
    void numResultsValueChanged(int numResults);
    void numResultCountFinished();

public slots:

};


#endif // HDTMANAGER_HPP