#pragma once
 
#ifndef SPREADSHEETVIEWER_H
#define SPREADSHEETVIEWER_H
 
#include "tcommon.h"
#include "cellposition.h"
#include "toonz/cellpositionratio.h"
// #include "orientation.h"
 
#include <QDialog>
#include <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