#pragma once

#ifndef IMAGEVIEWER_INCLUDE

#define IMAGEVIEWER_INCLUDE

#include "toonz/imagepainter.h"

#include "toonzqt/glwidget_for_highdpi.h"

#include <qtouchdevice></qtouchdevice>

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

//  Forward declarations

class FlipBook;

class HistogramPopup;

class QOpenGLFramebufferObject;

class LutCalibrator;

class QTouchEvent;

class QGestureEvent;

class QElapsedTimer;

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

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

//    ImageViewer

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

class ImageViewer final : public GLWidgetForHighDpi {

  Q_OBJECT

  enum DragType {

    eNone,

    eDrawRect  = 0x1,

    eMoveRect  = 0x2,

    eMoveLeft  = 0x4,

    eMoveRight = 0x8,

    eMoveUp    = 0x10,

    eMoveDown  = 0x20

  };

  int m_dragType;

  FlipBook *m_flipbook;

  TPoint m_pressedMousePos;

  // Modifying rect-picking position offset for vector image.

  // For unknown reasons, glReadPixels is covering the entire window not the

  // OpenGL widget.

  TPointD m_winPosMousePosOffset;

  Qt::MouseButton m_mouseButton;

  bool m_draggingZoomSelection;

  bool m_rectRGBPick;

  int m_FPS;

  ImagePainter::VisualSettings m_visualSettings;

  ImagePainter::CompareSettings m_compareSettings;

  TImageP m_image;     // displayed image

  TImageP m_orgImage;  // image without gain adjustment

  TAffine m_viewAff;

  QPoint m_pos;

  bool m_isHistogramEnable;

  HistogramPopup *m_histogramPopup;

  bool m_firstImage;

  bool m_isColorModel;

  // when fx parameter is modified with showing the fx preview,

  // a flipbook shows a red border line before the rendered result is shown.

  bool m_isRemakingPreviewFx;

  // used for color calibration with 3DLUT

  QOpenGLFramebufferObject *m_fbo = NULL;

  LutCalibrator *m_lutCalibrator  = NULL;

  bool m_touchActive                     = false;

  bool m_gestureActive                   = false;

  QTouchDevice::DeviceType m_touchDevice = QTouchDevice::TouchScreen;

  bool m_zooming                         = false;

  bool m_panning                         = false;

  double m_scaleFactor;  // used for zoom gesture

  bool m_stylusUsed       = false;

  bool m_firstInitialized = true;

  // passed from PlaybackExecutor

  QElapsedTimer *m_timer;

  qint64 m_targetInstant;

  int getDragType(const TPoint &pos, const TRect &loadBox);

  void updateLoadbox(const TPoint &curPos);

  void updateCursor(const TPoint &curPos);

  // for RGB color picker

  void pickColor(QMouseEvent *event, bool putValueToStyleEditor = false);

  void rectPickColor(bool putValueToStyleEditor = false);

  void setPickedColorToStyleEditor(const TPixel32 &color);

  // get the image (m_image or the snapshot) to be picked.

  TImageP getPickedImage(QPointF mousePos);

public:

  ImageViewer(QWidget *parent, FlipBook *flipbook, bool showHistogram);

  ~ImageViewer();

  void setIsColorModel(bool isColorModel) { m_isColorModel = isColorModel; }

  bool isColorModel() { return m_isColorModel; }

  void setVisual(const ImagePainter::VisualSettings &settings);

  void setImage(TImageP image, TImageP orgImage = TImageP());

  TImageP getImage() { return m_image; }

  TAffine getViewAff() { return m_viewAff; }

  void setViewAff(TAffine viewAff);

  TAffine getImgToWidgetAffine(const TRectD &imgBounds) const;

  TAffine getImgToWidgetAffine() const;

  /*! If histogram popup exist and is open set its title to "Histogram: level

     name"

                  if level Name is not empty. */

  void setHistogramTitle(QString levelName);

  void setHistogramEnable(bool enable);

  /*! If histogram popup exist and is open, it is set to default and is hidden.

   */

  void hideHistogram();

  void zoomQt(bool forward, bool reset);

  void resetZoom();

  void setIsRemakingPreviewFx(bool on) {

    m_isRemakingPreviewFx = on;

    update();

  }

  bool isRemakingPreviewFx() { return m_isRemakingPreviewFx; }

  void adaptView(const TRect &imgRect, const TRect &viewRect);

  void adaptView(const QRect &viewerRect);

  void doSwapBuffers();

  void changeSwapBehavior(bool enable);

  void invalidateCompHisto();

  void setTimerAndTargetInstant(QElapsedTimer *timer, qint64 target) {

    m_timer         = timer;

    m_targetInstant = target;

  }

protected:

  void contextMenuEvent(QContextMenuEvent *event) override;

  void initializeGL() override;

  void resizeGL(int width, int height) override;

  void paintGL() override;

  void showEvent(QShowEvent *) override;

  void hideEvent(QHideEvent *) override;

  void mouseMoveEvent(QMouseEvent *event) override;

  void mousePressEvent(QMouseEvent *event) override;

  void mouseReleaseEvent(QMouseEvent *event) override;

  void keyPressEvent(QKeyEvent *event) override;

  void mouseDoubleClickEvent(QMouseEvent *event) override;

  void wheelEvent(QWheelEvent *) override;

  void panQt(const QPoint &delta);

  void zoomQt(const QPoint ¢er, double factor);

  void dragCompare(const QPoint &dp);

  void tabletEvent(QTabletEvent *e) override;

  void touchEvent(QTouchEvent *e, int type);

  void gestureEvent(QGestureEvent *e);

  bool event(QEvent *e) override;

public slots:

  void updateImageViewer();

  void resetView();

  void fitView();

  void showHistogram();

  void swapCompared();

  void onContextAboutToBeDestroyed();

  void onPreferenceChanged(const QString &prefName);

private:

  QPointF m_firstPanPoint;

};

#endif  // IMAGEVIEWER_INCLUDE