#pragma once

#ifndef TRACKERPOPUP_H

#define TRACKERPOPUP_H

#include <qwidget></qwidget>

#include <qthread></qthread>

#include "tlevel.h"

#include "toonzqt/dvdialog.h"

#include "toonz/txshsimplelevel.h"

// per il tracker

#include "dummyprocessor.h"

// tracker.lib

#include "ObjectTracker.h"

// forward declaration

class Tracker;

class MyThread;

class QComboBox;

class QPushButton;

class QLabel;

class QSlider;

class TXshSimpleLevel;

// forward declaration

namespace DVGui {

class DoubleField;

class IntField;

// class LineEdit;

class CheckBox;

}  // namespace DVGui

//=============================================================================

// TrackerPopup

//-----------------------------------------------------------------------------

//! La classe si occupa di visualizzare il popup "convert-To-Vector Settings".

/*!Consente di cambiare modalita' di vettorizzazione e quindi in funzione

   di questa di cambiare i parametri da settare.*/

class TrackerPopup final : public DVGui::Dialog {

  Q_OBJECT

  DVGui::DoubleField *m_threshold;

  DVGui::DoubleField *m_sensitivity;

  DVGui::CheckBox *m_activeBackground;

  DVGui::CheckBox *m_variationWindow;

  /*DVGui::LineEdit*    m_frameStart;

  DVGui::LineEdit*    m_frameEnd;*/

  QPushButton *m_trackBtn;

  DVGui::IntField *m_from;

  DVGui::IntField *m_to;

public:

  TrackerPopup(QWidget *parent = 0, Qt::WindowFlags flags = Qt::Tool);

  bool apply();

private:

  DVGui::ProgressDialog *m_progress;

  MyThread *m_myThread;

  Tracker *m_tracker;

  int m_progressBarIndex;

  bool m_stoppedProcess;

public slots:

  //! Se la vettorializzazione e' andata a buon fine chiude il popup.

  void onTrack();

  //! Inizia il tracking in un nuovo thread

  void startNewThread();

  //! Ferma il processo di tracking dopo il frame corrente

  void stopProcessing();

};

//=============================================================================

// Tracker

//-----------------------------------------------------------------------------

class Tracker {

  float m_threshold;

  float m_sensitivity;

  int m_activeBackground;

  int m_manageOcclusion;

  int m_variationWindow;

  int m_indexFrameStart;

  int m_framesNumber;

  short m_trackerCount;

  int *m_numstart;

  // TLevel::Iterator m_currentFrameIt;

  int m_currentFrame;

  int m_num;

  short m_oldObjectId;

  short m_numobjactive;

  int m_trackerRegionIndex;

  TPointD m_rasterOrigin;

  CObjectTracker *m_pObjectTracker[30];

  TRaster32P m_raster_prev;

  TRaster32P m_raster;

  TRaster32P *m_raster_template;

  TrackerObjectsSet *m_trackerObjectsSet;

  DummyProcessor *m_processor;

  TLevelP m_level;

  TAffine m_affine;

public:

  Tracker(double threshold, double sensitivity, int activeBackground,

          int manageOcclusion, int variationWindow, int frameStart,

          int framesNumber);

  ~Tracker();

  // ritorna true se è tutto ok

  bool trackCurrentFrame();

  int getFramesNumber() { return m_framesNumber; }

  int getStartFrameIndex() { return m_indexFrameStart; }

  QString getLastError() { return getErrorMessage(m_lastErrorCode); }

private:

  // Inizializzazione Variabili

  // ritorna true se è tutto ok

  bool setup();

  int m_lastErrorCode;

  // Restituisce il messaggio di errore dal codice di errore che viene dato dal

  // metodo apply()

  QString getErrorMessage(int errorCode);

};

class MyThread final : public QThread {

  Tracker *m_tracker;

public:

  MyThread(Tracker *tracker);

  void run() override;

  // Ritorna 0 se non c'è stato alcun errore nell'esecuzione del thread

  // altrimenti ritorna il codice di errore del tracker

  int getLastError() { return m_trackerError; };

private:

  int m_trackerError;

};

#endif  // TRACKERPOPUP_H