#pragma once

#ifndef FUNCTIONSHEET_H

#define FUNCTIONSHEET_H

#include "tcommon.h"

#include "functiontreeviewer.h"

#include "spreadsheetviewer.h"

#include "functionselection.h"

#include "toonzqt/lineedit.h"

#include <qwidget></qwidget>

#include <set></set>

#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

class FunctionSheet;

class TDoubleParam;

class TFrameHandle;

class FunctionSelection;

class FunctionSheetRowViewer : public Spreadsheet::RowPanel {

  FunctionSheet *m_sheet;

public:

  FunctionSheetRowViewer(FunctionSheet *parent);

protected:

  void paintEvent(QPaintEvent *);

  void mousePressEvent(QMouseEvent *);

  void mouseReleaseEvent(QMouseEvent *);

  void mouseMoveEvent(QMouseEvent *);

  void contextMenuEvent(QContextMenuEvent *);

};

class FunctionSheetColumnHeadViewer : public Spreadsheet::ColumnPanel {

  FunctionSheet *m_sheet;

  // enable drag and drop the expression arguments

  QPoint m_dragStartPosition;

  FunctionTreeModel::Channel *m_draggingChannel;

public:

  FunctionSheetColumnHeadViewer(FunctionSheet *parent);

protected:

  void paintEvent(QPaintEvent *);

  void mousePressEvent(QMouseEvent *);

  // update the tooltip

  void mouseMoveEvent(QMouseEvent *);

  void contextMenuEvent(QContextMenuEvent *);

};

class FunctionSheetCellViewer : public Spreadsheet::CellPanel {

  Q_OBJECT

  FunctionSheet *m_sheet;

  DVGui::LineEdit *m_lineEdit;

  int m_editRow, m_editCol;

public:

  FunctionSheetCellViewer(FunctionSheet *parent);

  Spreadsheet::DragTool *createDragTool(QMouseEvent *);

protected:

  void drawCells(QPainter &p, int r0, int c0, int r1, int c1);

  void mouseDoubleClickEvent(QMouseEvent *);

  void mousePressEvent(QMouseEvent *);

  void mouseReleaseEvent(QMouseEvent *);

  void mouseMoveEvent(QMouseEvent *);

  void openContextMenu(QMouseEvent *);

private slots:

  void onCellEditorEditingFinished();

};

class FunctionSheet : public SpreadsheetViewer {

  Q_OBJECT

public:

  FunctionSheet(QWidget *parent = 0);

  ~FunctionSheet();

  void setModel(FunctionTreeModel *model);

  FunctionTreeModel *getModel() const { return m_functionTreeModel; }

  void setViewer(FunctionViewer *viewer);

  FunctionViewer *getViewer() const { return m_functionViewer; }

  void setCurrentFrame(int frame);

  int getCurrentFrame() const;

  int getChannelCount();

  TDoubleParam *getCurve(int column);

  FunctionTreeModel::Channel *getChannel(int column);

  QRect getSelectedCells() const;

  void selectCells(const QRect &selectedCells);

  FunctionSelection *getSelection() const { return m_selection; }

  void setSelection(FunctionSelection *selection);  // does NOT get ownership

  QString getSelectedParamName();

  int getColumnIndexByCurve(TDoubleParam *param) const;

  bool anyWidgetHasFocus();

protected:

  void showEvent(QShowEvent *e);

  void hideEvent(QHideEvent *e);

private:

  FunctionSheetRowViewer *m_rowViewer;

  FunctionSheetColumnHeadViewer *m_columnHeadViewer;

  FunctionSheetCellViewer *m_cellViewer;

  FunctionSelection *m_selection;

  FunctionTreeModel *m_functionTreeModel;

  FunctionViewer *m_functionViewer;

  QRect m_selectedCells;

public slots:

  void updateAll();

  void onFrameSwitched();

  /*---

   * カレントChannelが切り替わったら、NumericalColumnsがそのChannelを表示できるようにスクロールする。---*/

  void onCurrentChannelChanged(FunctionTreeModel::Channel *);

};

#endif