#ifndef UTTERANCETREEWIDGET_H
  #define UTTERANCETREEWIDGET_H
  
  #include <QWidget>
  #include <QRectF>
  #include <QMap>
  
  #include <armadillo>
  
  #include "UtteranceTree.h"
  
  class UtteranceTreeWidget: public QWidget
  {
    Q_OBJECT
    
   public:
    UtteranceTreeWidget(int width = 480, int height = 270, QWidget *parent = 0);
    QList<QList<int>> getPartition();
  
    public slots:
      void updateUtteranceTree(const arma::mat &S, const arma::mat &W, const arma::umat &map, const arma::mat &SigmaInv);
      void setDistance(UtteranceTree::DistType dist);
      void setCovInv(const arma::mat &CovInv);
      void setAgrCrit(UtteranceTree::AgrCrit agr);
      void setPartitionMethod(UtteranceTree::PartMeth partMeth);
      void setWeight(const arma::mat &W);
      void setCurrSubtitle(int subIdx);
      void normalizeVectors(bool checked);
      void setDiff(const arma::mat &Diff);
      void releasePos(bool released);
      
   signals:
      void playSubtitle(QList<int> utter);
      void setSpeakers(QList<int> spkIdx);
      
   protected:
    void paintEvent(QPaintEvent * event);
    void mousePressEvent(QMouseEvent * event);
    void mouseDoubleClickEvent(QMouseEvent * event);
    void mouseMoveEvent(QMouseEvent * event);
  
   private:
    int m_width;
    int m_height;
    arma::mat m_S;
    arma::mat m_unnormS;
    arma::mat m_W;
    arma::mat m_SigmaInv;
    arma::umat m_map;
    UtteranceTree *m_tree;
    QString m_currSub;
    QVector<qreal> m_cutDist;
    QList<QList<int>> m_partition;
    qreal m_yPad;
    qreal m_YRatio;
    QPoint m_mousePos;
    QList<int> m_selectedUtter;
    bool m_posReleased;
  };
  
  #endif