#pragma once

#ifndef SPREADSHEETVIEWER_H

#define SPREADSHEETVIEWER_H

#include "tcommon.h"

#include "cellposition.h"

#include "toonz/cellpositionratio.h"

// #include "orientation.h"

#include <qdialog></qdialog>

#include <qscrollarea></qscrollarea>

#undef DVAPI

#undef DVVAR

#ifdef TOONZQT_EXPORTS

#define DVAPI DV_EXPORT_API

#define DVVAR DV_EXPORT_VAR

#else

#define DVAPI DV_IMPORT_API

#define DVVAR DV_IMPORT_VAR

#endif

// forward declaration

class TFrameHandle;

class SpreadsheetViewer;

class Orientation;

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

namespace Spreadsheet {

// forward declaration

class GenericPanel;

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

// Use composition rather than inheritance.

// How this works:

// * scroll area scrollbars sends event to this;

// * it notifies every other FrameScroller with difference;

// * they handle it by adjusting their scrollbars

class DVAPI FrameScroller final : public QObject {

  Q_OBJECT

  const Orientation *m_orientation;

  QScrollArea *m_scrollArea;

  int m_lastX, m_lastY;

  bool m_syncing;

public:

  FrameScroller();

  virtual ~FrameScroller();

  void setFrameScrollArea(QScrollArea *scrollArea);

  QScrollArea *getFrameScrollArea() const { return m_scrollArea; }

  void setOrientation(const Orientation *o) { m_orientation = o; }

  const Orientation *orientation() const { return m_orientation; }

  void registerFrameScroller();

  void unregisterFrameScroller();

  void prepareToScrollOthers(const QPointF &offset);

  void setSyncing(bool s) { m_syncing = s; }

  bool isSyncing() { return m_syncing; }

private:

  void connectScrollbars();

  void disconnectScrollbars();

  void handleScroll(QPoint &offset, int senderMaximum, int senderValue);

  void onScroll(const CellPositionRatio &offset);

  bool exactScroll(const int senderMaximum, const int senderValue);

  void prepareToScrollRatio(const CellPositionRatio &offset);

private slots:

  void onVScroll(int value);

  void onHScroll(int value);

signals:

  void prepareToScrollOffset(const QPointF &offset);

  void zoomScrollAdjust(QPointF &offset, bool toZoom);

};

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

class DVAPI DragTool {

public:

  DragTool() {}

  virtual ~DragTool() {}

  virtual void click(int row, int col, QMouseEvent *e) {}

  virtual void drag(int row, int col, QMouseEvent *e) {}

  virtual void release(int row, int col, QMouseEvent *e) {}

};

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

class DVAPI SetFrameDragTool final : public DragTool {

  TFrameHandle *m_frameHandle;

public:

  SetFrameDragTool(TFrameHandle *frameHandle) : m_frameHandle(frameHandle) {}

  void click(int row, int col, QMouseEvent *e) override;

  void drag(int row, int col, QMouseEvent *e) override;

  void release(int row, int col, QMouseEvent *e) override;

};

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

class DVAPI SelectionDragTool final : public DragTool {

  SpreadsheetViewer *m_viewer;

  int m_firstRow, m_firstCol;

public:

  SelectionDragTool(SpreadsheetViewer *viewer);

  void click(int row, int col, QMouseEvent *e) override;

  void drag(int row, int col, QMouseEvent *e) override;

  void release(int row, int col, QMouseEvent *e) override;

};

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

class DVAPI PanTool final : public DragTool {

  SpreadsheetViewer *m_viewer;

  GenericPanel *m_panel;

  QPoint m_lastPos;

public:

  PanTool(GenericPanel *panel);

  void click(int row, int col, QMouseEvent *e) override;

  void drag(int row, int col, QMouseEvent *e) override;

  void release(int row, int col, QMouseEvent *e) override;

};

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

class DVAPI ScrollArea final : public QScrollArea {

  Q_OBJECT

public:

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

  virtual ~ScrollArea();

protected:

  // keyPressEvent and wheelEvent are ignored by the ScrollArea

  // and therefore they are handled by the parent (Viewer)

  void keyPressEvent(QKeyEvent *e) override;

  void wheelEvent(QWheelEvent *e) override;

};

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

class DVAPI GenericPanel : public QWidget {

  Q_OBJECT

  SpreadsheetViewer *m_viewer;

  DragTool *m_dragTool;

public:

  GenericPanel(SpreadsheetViewer *viewer);

  virtual ~GenericPanel();

  SpreadsheetViewer *getViewer() const { return m_viewer; }

  virtual DragTool *createDragTool(QMouseEvent *) { return 0; };

protected:

  void paintEvent(QPaintEvent *) override;

  void mousePressEvent(QMouseEvent *) override;

  void mouseReleaseEvent(QMouseEvent *) override;

  void mouseMoveEvent(QMouseEvent *) override;

};

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

class DVAPI RowPanel : public GenericPanel {

  Q_OBJECT

  const int m_xa;  // frame cells start at m_xa

public:

  RowPanel(SpreadsheetViewer *viewer);

  virtual ~RowPanel() {}

  DragTool *createDragTool(QMouseEvent *) override;

protected:

  void paintEvent(QPaintEvent *) override;

  void drawRows(QPainter &p, int r0, int r1);

  void drawCurrentRowGadget(QPainter &p, int r0, int r1);

};

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

class DVAPI ColumnPanel : public GenericPanel {

  Q_OBJECT

public:

  ColumnPanel(SpreadsheetViewer *viewer);

  virtual ~ColumnPanel() {}

};

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

class DVAPI CellPanel : public GenericPanel {

  Q_OBJECT

public:

  CellPanel(SpreadsheetViewer *viewer);

  virtual ~CellPanel() {}

  DragTool *createDragTool(QMouseEvent *) override;

protected:

  void paintEvent(QPaintEvent *) override;

  virtual void drawCells(QPainter &p, int r0, int c0, int r1, int c1) {}

};

}  // namespace Spreadsheet

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

class DVAPI SpreadsheetViewer : public QDialog {

  Q_OBJECT

  QColor m_lightLightBgColor;  // RowPanel background (124,124,124)

  QColor m_bgColor;         // RowPanel background in scene range(164,164,164)

  QColor m_lightLineColor;  // horizontal line (146,144,146)

  Q_PROPERTY(QColor LightLightBGColor READ getLightLightBGColor WRITE

                 setLightLightBGColor)

  Q_PROPERTY(QColor BGColor READ getBGColor WRITE setBGColor)

  Q_PROPERTY(

      QColor LightLineColor READ getLightLineColor WRITE setLightLineColor)

  QColor m_currentRowBgColor;    // current frame, column

  QColor m_markerLineColor;      // marker interval (0, 255, 246)

  QColor m_secMarkerLineColor;   // second marker lines

  QColor m_textColor;            // text (black)

  QColor m_currentRowTextColor;  // text color for the current row

  QColor m_verticalLineColor;    // vertical line (black)

  Q_PROPERTY(QColor CurrentRowBgColor READ getCurrentRowBgColor WRITE

                 setCurrentRowBgColor)

  Q_PROPERTY(

      QColor MarkerLineColor READ getMarkerLineColor WRITE setMarkerLineColor)

  Q_PROPERTY(QColor SecMarkerLineColor READ getSecMarkerLineColor WRITE

                 setSecMarkerLineColor)

  Q_PROPERTY(QColor TextColor READ getTextColor WRITE setTextColor)

  Q_PROPERTY(QColor CurrentRowTextColor READ getCurrentRowTextColor WRITE

                 setCurrentRowTextColor)

  Q_PROPERTY(QColor VerticalLineColor READ getVerticalLineColor WRITE

                 setVerticalLineColor)

  // key frame

  QColor m_keyFrameColor;          // (219,139,54)

  QColor m_keyFrameBorderColor;    // (82,51,20)

  QColor m_selectedKeyFrameColor;  // (237,197,155)

  QColor m_ignoredKeyFrameColor;

  QColor m_selectedIgnoredKeyFrameColor;

  // key frame inbetween

  QColor m_inBetweenColor;          // (194,194,176)

  QColor m_inBetweenBorderColor;    // (72,72,65)

  QColor m_selectedInBetweenColor;  // (225,225,216)

  QColor m_ignoredInBetweenColor;

  QColor m_selectedIgnoredInBetweenColor;

  // empty cell

  QColor m_selectedEmptyColor;  // (190,190,190)

  // empty cell in the scene range

  QColor m_selectedSceneRangeEmptyColor;  // (210,210,210)

  Q_PROPERTY(QColor KeyFrameColor READ getKeyFrameColor WRITE setKeyFrameColor)

  Q_PROPERTY(QColor KeyFrameBorderColor READ getKeyFrameBorderColor WRITE

                 setKeyFrameBorderColor)

  Q_PROPERTY(QColor SelectedKeyFrameColor READ getSelectedKeyFrameColor WRITE

                 setSelectedKeyFrameColor)

  Q_PROPERTY(QColor IgnoredKeyFrameColor READ getIgnoredKeyFrameColor WRITE

                 setIgnoredKeyFrameColor)

  Q_PROPERTY(

      QColor SelectedIgnoredKeyFrameColor READ getSelectedIgnoredKeyFrameColor

          WRITE setSelectedIgnoredKeyFrameColor)

  Q_PROPERTY(

      QColor InBetweenColor READ getInBetweenColor WRITE setInBetweenColor)

  Q_PROPERTY(QColor InBetweenBorderColor READ getInBetweenBorderColor WRITE

                 setInBetweenBorderColor)

  Q_PROPERTY(QColor SelectedInBetweenColor READ getSelectedInBetweenColor WRITE

                 setSelectedInBetweenColor)

  Q_PROPERTY(QColor IgnoredInBetweenColor READ getIgnoredInBetweenColor WRITE

                 setIgnoredInBetweenColor)

  Q_PROPERTY(

      QColor SelectedIgnoredInBetweenColor READ getSelectedIgnoredInBetweenColor

          WRITE setSelectedIgnoredInBetweenColor)

  Q_PROPERTY(QColor SelectedEmptyColor READ getSelectedEmptyColor WRITE

                 setSelectedEmptyColor)

  Q_PROPERTY(

      QColor SelectedSceneRangeEmptyColor READ getSelectedSceneRangeEmptyColor

          WRITE setSelectedSceneRangeEmptyColor)

  QColor m_columnHeaderBorderColor;  // column header border lines (46,47,46)

  Q_PROPERTY(QColor ColumnHeaderBorderColor READ getColumnHeaderBorderColor

                 WRITE setColumnHeaderBorderColor)

  Spreadsheet::ScrollArea *m_columnScrollArea;

  Spreadsheet::ScrollArea *m_rowScrollArea;

  Spreadsheet::ScrollArea *m_cellScrollArea;

  TFrameHandle *m_frameHandle;

  int m_columnWidth;

  int m_rowHeight;

  int m_scaleFactor;

  // QPoint m_delta;

  int m_timerId;

  QPoint m_autoPanSpeed;

  QPoint m_lastAutoPanPos;

  int m_rowCount, m_columnCount;

  int m_currentRow;

  int m_markRowDistance, m_markRowOffset, m_markSecRowDistance;

  // QRect m_selectedCells; // x=col, y=row

  bool m_isComputingSize;

  // const Orientation *m_orientation;

protected:

  Spreadsheet::FrameScroller m_frameScroller;

public:

  SpreadsheetViewer(QWidget *parent);

  virtual ~SpreadsheetViewer();

  int getColumnWidth() const { return m_columnWidth; }

  void setColumnWidth(int width) { m_columnWidth = width; }

  int getRowHeight() const { return m_rowHeight; }

  void setRowHeight(int height) { m_rowHeight = height; }

  void setScaleFactor(int factor) { m_scaleFactor = factor; }

  void setRowsPanel(Spreadsheet::RowPanel *rows);

  void setColumnsPanel(Spreadsheet::ColumnPanel *columns);

  void setCellsPanel(Spreadsheet::CellPanel *cells);

  void setButtonAreaWidget(QWidget *widget);

  int getRowCount() const { return m_rowCount; }

  // QProperty

  void setLightLightBGColor(const QColor &color) {

    m_lightLightBgColor = color;

  }

  QColor getLightLightBGColor() const { return m_lightLightBgColor; }

  void setBGColor(const QColor &color) { m_bgColor = color; }

  QColor getBGColor() const { return m_bgColor; }

  void setLightLineColor(const QColor &color) { m_lightLineColor = color; }

  QColor getLightLineColor() const { return m_lightLineColor; }

  void setCurrentRowBgColor(const QColor &color) {

    m_currentRowBgColor = color;

  }

  QColor getCurrentRowBgColor() const { return m_currentRowBgColor; }

  void setMarkerLineColor(const QColor &color) { m_markerLineColor = color; }

  QColor getMarkerLineColor() const { return m_markerLineColor; }

  void setSecMarkerLineColor(const QColor &color) {

    m_secMarkerLineColor = color;

  }

  QColor getSecMarkerLineColor() const { return m_secMarkerLineColor; }

  void setTextColor(const QColor &color) { m_textColor = color; }

  QColor getTextColor() const { return m_textColor; }

  void setCurrentRowTextColor(const QColor &color) {

    m_currentRowTextColor = color;

  }

  QColor getCurrentRowTextColor() const { return m_currentRowTextColor; }

  void setVerticalLineColor(const QColor &color) {

    m_verticalLineColor = color;

  }

  QColor getVerticalLineColor() const { return m_verticalLineColor; }

  void setKeyFrameColor(const QColor &color) { m_keyFrameColor = color; }

  QColor getKeyFrameColor() const { return m_keyFrameColor; }

  void setKeyFrameBorderColor(const QColor &color) {

    m_keyFrameBorderColor = color;

  }

  QColor getKeyFrameBorderColor() const { return m_keyFrameBorderColor; }

  void setSelectedKeyFrameColor(const QColor &color) {

    m_selectedKeyFrameColor = color;

  }

  QColor getSelectedKeyFrameColor() const { return m_selectedKeyFrameColor; }

  void setIgnoredKeyFrameColor(const QColor &color) {

    m_ignoredKeyFrameColor = color;

  }

  QColor getIgnoredKeyFrameColor() const { return m_ignoredKeyFrameColor; }

  void setSelectedIgnoredKeyFrameColor(const QColor &color) {

    m_selectedIgnoredKeyFrameColor = color;

  }

  QColor getSelectedIgnoredKeyFrameColor() const {

    return m_selectedIgnoredKeyFrameColor;

  }

  void setInBetweenColor(const QColor &color) { m_inBetweenColor = color; }

  QColor getInBetweenColor() const { return m_inBetweenColor; }

  void setInBetweenBorderColor(const QColor &color) {

    m_inBetweenBorderColor = color;

  }

  QColor getInBetweenBorderColor() const { return m_inBetweenBorderColor; }

  void setSelectedInBetweenColor(const QColor &color) {

    m_selectedInBetweenColor = color;

  }

  QColor getSelectedInBetweenColor() const { return m_selectedInBetweenColor; }

  void setIgnoredInBetweenColor(const QColor &color) {

    m_ignoredInBetweenColor = color;

  }

  QColor getIgnoredInBetweenColor() const { return m_ignoredInBetweenColor; }

  void setSelectedIgnoredInBetweenColor(const QColor &color) {

    m_selectedIgnoredInBetweenColor = color;

  }

  QColor getSelectedIgnoredInBetweenColor() const {

    return m_selectedIgnoredInBetweenColor;

  }

  void setSelectedEmptyColor(const QColor &color) {

    m_selectedEmptyColor = color;

  }

  QColor getSelectedEmptyColor() const { return m_selectedEmptyColor; }

  void setSelectedSceneRangeEmptyColor(const QColor &color) {

    m_selectedSceneRangeEmptyColor = color;

  }

  QColor getSelectedSceneRangeEmptyColor() const {

    return m_selectedSceneRangeEmptyColor;

  }

  void setColumnHeaderBorderColor(const QColor &color) {

    m_columnHeaderBorderColor = color;

  }

  QColor getColumnHeaderBorderColor() const {

    return m_columnHeaderBorderColor;

  }

  void scroll(QPoint delta);

  void setAutoPanSpeed(const QPoint &speed);

  void setAutoPanSpeed(const QRect &widgetBounds, const QPoint &mousePos);

  void stopAutoPan() { setAutoPanSpeed(QPoint()); }

  bool isAutoPanning() const {

    return m_autoPanSpeed.x() != 0 || m_autoPanSpeed.y() != 0;

  }

  int xToColumn(int x) const;

  int yToRow(int y) const;

  int columnToX(int col) const;

  int rowToY(int row) const;

  CellPosition xyToPosition(const QPoint &point) const;

  QPoint positionToXY(const CellPosition &pos) const;

  CellRange xyRectToRange(const QRect &rect) const;

  // const Orientation *orientation () const { return m_orientation; }

  bool refreshContentSize(int scrollDx, int scrollDy);

  int getCurrentRow() const { return m_currentRow; }

  void setCurrentRow(int row) { m_currentRow = row; }

  virtual QRect getSelectedCells() const               = 0;

  virtual void selectCells(const QRect &selectedCells) = 0;

  bool isSelectedCell(int row, int col) const {

    return getSelectedCells().contains(QPoint(col, row));

  }

  void setMarkRow(int distance, int offset, int secDistance) {

    m_markRowDistance    = distance;  // distance > 0 ? distance : 6;

    m_markRowOffset      = offset;

    m_markSecRowDistance = secDistance;

  }

  void getMarkRow(int &distance, int &offset) const {

    distance = m_markRowDistance;

    offset   = m_markRowOffset;

  }

  bool isMarkRow(int row) const {

    return m_markRowDistance > 0 &&

           ((row - m_markRowOffset) % m_markRowDistance) == 0 && row > 0;

  }

  bool isMarkSecRow(int row) const {

    return m_markSecRowDistance > 0 &&

           ((row - m_markRowOffset) % m_markSecRowDistance) == 0 && row > 0;

  }

  bool isSecMarkerActive() const { return m_markSecRowDistance > 0; }

  void setFrameHandle(TFrameHandle *frameHandle);

  TFrameHandle *getFrameHandle() const { return m_frameHandle; }

  void ensureVisibleCol(int col);

  virtual int getFrameZoomFactor() const { return 100; }

  bool isSmallHeader();

  void updateHeaderHeight();

protected:

  void showEvent(QShowEvent *) override;

  void hideEvent(QHideEvent *) override;

  void resizeEvent(QResizeEvent *event) override;

  void keyPressEvent(QKeyEvent *event) override;

  void wheelEvent(QWheelEvent *event) override;

  void timerEvent(QTimerEvent *) override;

public slots:

  void setRowCount(int rowCount);

  void setColumnCount(int columnCount);

  void frameSwitched();

  void updateAreas();

  void onVSliderChanged(int);

  void onHSliderChanged(int);

  void onPrepareToScrollOffset(const QPointF &offset);

  void onZoomScrollAdjust(QPointF &offset, bool toZoom);

  /*

void updateAllAree();

void updateCellColumnAree();

void updateCellRowAree();

*/

};

#endif