#ifndef PROJECTMODEL_H
  #define PROJECTMODEL_H
  
  #include <QAbstractItemModel>
  #include <QSize>
  #include <QMap>
  
  #include <opencv2/core/core.hpp>
  #include <opencv2/highgui/highgui.hpp>
  
  #include <armadillo>
  
  #include "Series.h"
  #include "Segment.h"
  #include "Episode.h"
  #include "VideoFrame.h"
  #include "MovieAnalyzer.h"
  
  class ProjectModel: public QAbstractItemModel
  {
    Q_OBJECT
  
   public:
    ProjectModel(QObject *parent = 0);
    ~ProjectModel();
    bool save(const QString &fName);
    bool load(const QString &fName);
  
    QModelIndex index(int row, int column, const QModelIndex &parent = QModelIndex()) const;
    QModelIndex parent(const QModelIndex &child = QModelIndex()) const;
    QVariant data(const QModelIndex &index, int role) const;
    int rowCount(const QModelIndex &parent = QModelIndex()) const;
    int columnCount(const QModelIndex &parent = QModelIndex()) const;
    QVariant headerData(int section, Qt::Orientation orientation, int role) const;
    Qt::ItemFlags flags(const QModelIndex &index) const;
    QModelIndex indexFromSegment(Segment *segment) const;
    int getDepth() const;
    QModelIndex getShotParentIndex() const;
  
    void setModel(const QString &name, const QString &seriesName, int seasNbr, int epNbr, const QString &epName, const QString &epFName);
    bool appendModel(int seasNbr, int epNbr, const QString &epName, const QString &epFName);
    bool insertSubtitles(const QString &subFName);
    bool speakerDiarization(const QString &subFName, VideoFrame::SpeakerSource source);
    bool localSpkDiar(bool baseline, UtteranceTree::DistType dist, bool norm, UtteranceTree::AgrCrit agr, UtteranceTree::PartMeth partMeth, bool weight, bool sigma);
    bool globalSpkDiar();
    void extractShots(QString fName, int histoType, int nVBins, int nHBins, int nSBins, int metrics, qreal threshold1, qreal threshold2, int nVBlock, int nHBlock, bool iterate);
    void labelSimilarShots(QString fName, int histoType, int nVBins, int nHBins, int nSBins, int metrics, qreal maxDist, int windowSize, int nVBlock, int nHBlock, bool iterate);
    qreal evaluateShotDetection(bool displayResults, qreal thresh1, qreal thresh2) const;
    qreal evaluateSimShotDetection(bool displayResults, qreal thresh1, qreal thresh2) const;
    int retrieveShotPrevPositions(qint64 position, QList<qint64> &shotPositions);
    void reset();
    QString getName() const;
    QString getBaseName() const;
    QString getSeriesName() const;
    void setSpkDiar(const QString &epFName);
  
    public slots:
      void setResolution(const QSize &resolution);
      void setFps(qreal fps);
      void appendVideoFrame(int id, qint64 position);
      void initShotLevel(Segment *segment);
      void insertShot(qint64 position, Segment::Source);
      void insertShot(Segment *segment, Segment::Source);
      void removeShot(Segment *segment, Segment::Source source);
      void processSegmentation(bool checked, bool annot);
      void retrieveSpeakers(bool checked);
      void labelSimShot(qint64 position, int nCamera, Segment::Source source); 
      void setSpeaker(qint64 start, qint64 end, const QString &speaker, VideoFrame::SpeakerSource source);
      void resetSpeaker(qint64 prevStart, qint64 prevEnd, qint64 start, qint64 end, bool resetSub, VideoFrame::SpeakerSource source);
      void exportSubtitles(const QString &fName);
      void improveSpkDiar();
      void retrieveShotSub(qint64 position);
      void playSubtitle(QList<int> utter);
      void playSeg(QList<QPair<qint64, qint64>> segments);
      void currentSub(qint64 position);
      void extractIVectors(bool ubm, const QString &epFName);
      void extractSpkIVectors(const QString &epFName, bool refSpk);
  
   signals:
      void modelChanged();
      void positionChanged(qint64 position);
      void truePositive(qint64 position);
      void falsePositive(qint64 position);
      void getShot(Segment *shot) const;
      void segmentationRetrieved();
      void getSpokenFrame(qint64 position, const QString &sub, const QString &speaker) const;
      void viewSegmentation(bool checked, bool annot);
      void speakersRetrieved(QList<QString> speakers);
      void getCurrentPattern(const QList<QPair<int, qreal>> &subFeatures);
      void getPatternFirstShot(const QList<QPair<int, qreal>> &subFeatures);
      void getPatternSecondShot(const QList<QPair<int, qreal>> &subFeatures);
      void playSegments(QList<QPair<qint64, qint64>> utterances);
      void currentSubtitle(int subIdx);
      void setDiarData(const arma::mat &E, const arma::mat &Sigma, const arma::mat &W);
      void setDiarData(const arma::mat &E, const arma::mat &Sigma, const arma::mat &W, QMap<QString, QList<QPair<qreal, qreal>>> speakers);
  
   private:
      void eraseSubtitles(Segment *segment);
      void clearSpeaker(Segment *segment, VideoFrame::SpeakerSource source);
      bool shotLevelCreated(Segment *segment);
      Segment * getFirstVideoFrame(Segment *segment);
      qreal computePrecision(int tp, int fp) const;
      qreal computeRecall(int tp, int fn) const;
      qreal computeFScore(qreal precision, qreal recall) const;
      qreal computeAccuracy(int tp, int fp, int fn, int tn) const;
      void resetShotsToManual(Segment *segment, QList<Segment *> &toRemove);
      void resetAutoCameraLabels(Segment *segment);
      void depthFirstToShots(Segment *segment) const;
      void depthFirstToSpokenFrames(Segment *segment, VideoFrame::SpeakerSource source) const;
      void retrieveShotPositions(Segment *segment, QList<qint64> &shotPositions) const;
      void retrieveSimCamLabels(Segment *segment, QList<int> &autCamLabels, QList<int> &manCamLabels) const;
      void nFramesByCamLabel(Segment *segment, QMap<QString, QMap<QString, int>> &speakerList, Segment::Source vSource, VideoFrame::SpeakerSource sSource) const;
      void retrieveSpokenVFrames(Segment *segment, QList<VideoFrame *> &spkVFrames, VideoFrame::SpeakerSource source) const;
      void retrieveVFrames(Segment *segment, QList<VideoFrame *> &vFrames) const;
      void retrieveSpeakersList(Segment *segment, QList<QString> &speakers, VideoFrame::SpeakerSource source) const;
      void retrieveShotUtterances();
      void retrieveShotPatterns();
      void retrieveShotLabels(Segment *segment, QList<QString> &shotLabels) const;
      void retrieveSubPositionsLabels(QList<qint64> &subStarts, QList<qint64> &subEnds, QList<QString> &subRefLbl) const;
      bool testShotPattern(const QList<QString> &lblWindow, int pattSize);
      QPair<qint64, qint64> adjustSubBoundaries(qint64 subStart, qint64 subEnd, QList<QPair<qint64, qint64>> shotPattBound);
      QString normalizedPattern(const QString &firstLabel, const QString &secondLabel);
      QString mergePatterns(QString firstLabel, QString secondLabel, QList<QPair<int, qreal>> list1, QList<QPair<int, qreal>> list2, QList<QPair<int, qreal>> &mergedList);
      QString completePatternLabel(const QString &label, QMap<QString, QList<QPair<int, qreal>>> shotPatterns);
      bool interPatterns(QString firstLabel, QString secondLabel);
      QStringList appendStringList(const QStringList &list1, const QStringList &list2);
      void mergePatternBoundaries(QMap<QString, QList<QPair<qint64, qint64>>> &shotPattBound);
      void exportGlobSpkRef(const QString &fName);
      void exportLocSpkRef(const QString &fName);
      arma::mat genWMat(QMap<QString, QList<int>> spkIdx, const arma::mat &X);
  
      QString m_name;
      QString m_baseName;
      Series *m_series;
      Episode *m_episode;
      MovieAnalyzer *m_movieAnalyzer;
      QList<QPair<qint64, qint64>> m_subBound;
      QMap<QString, QList<QPair<int, qreal>>> m_shotUtterances;
      QMap<QString, QList<QPair<int, qreal>>> m_shotPatterns;
      QMap<QString, QList<QPair<qint64, qint64>>> m_shotPattBound;
      QMap<QString, QList<QPair<qint64, qint64>>> m_strictShotPattBound;
      QList<QString> m_subRefLbl;
  
      cv::VideoCapture m_cap;
      arma::mat m_W;
  };
  
  #endif