#pragma once

#ifndef STYLESELECTION_INCLUDED

#define STYLESELECTION_INCLUDED

#include "toonzqt/selection.h"

#include "toonz/tpalettehandle.h"

#include "tpalette.h"

#include <set></set>

#include <qstring></qstring>

class QByteArray;

#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 TXsheetHandle;

class TXshLevelHandle;

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

// TStyleSelection

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

class DVAPI TStyleSelection final : public TSelection {

  TPaletteHandle *m_paletteHandle;

  // Used to change level palette; in other palette (cleanup, ...) xsheetHandle

  // is not necessary.

  TXsheetHandle *m_xsheetHandle;

  // for clearing cache when the pastestyle command is executed

  TXshLevelHandle *m_levelHandle;

  int m_pageIndex;

  std::set<int> m_styleIndicesInPage;</int>

public:

  TStyleSelection();

  TStyleSelection(TStyleSelection *styleSelection)

      : m_paletteHandle(styleSelection->getPaletteHandle())

      , m_pageIndex(styleSelection->getPageIndex())

      , m_styleIndicesInPage(styleSelection->getIndicesInPage()) {}

  ~TStyleSelection();

  void select(int pageIndex);

  void select(int pageIndex, int styleIndexInPage, bool on);

  bool isSelected(int pageIndex, int styleIndexInPage) const;

  bool isPageSelected(int pageIndex) const;

  bool canHandleStyles();

  void selectNone() override;

  bool isEmpty() const override;

  int getStyleCount() const;

  TPaletteHandle *getPaletteHandle() const { return m_paletteHandle; }

  void setPaletteHandle(TPaletteHandle *paletteHandle) {

    m_paletteHandle = paletteHandle;

  }

  TPalette *getPalette() const { return m_paletteHandle->getPalette(); }

  int getPageIndex() const { return m_pageIndex; }

  const std::set<int> &getIndicesInPage() const { return m_styleIndicesInPage; }</int>

  void getIndices(std::set<int> &indices) const;</int>

  // Used to change level palette; in other palette (cleanup, ...) xsheetHandle

  // is not necessary.

  void setXsheetHandle(TXsheetHandle *xsheetHandle) {

    m_xsheetHandle = xsheetHandle;

  }

  TXsheetHandle *getXsheetHandle() const { return m_xsheetHandle; }

  void setLevelHandle(TXshLevelHandle *levelHandle) {

    m_levelHandle = levelHandle;

  }

  // commands

  void cutStyles();

  void copyStyles();

  void pasteStyles();

  void pasteStylesValues(bool pasteName, bool pasteColor = true);

  void pasteStylesValue();

  void pasteStylesColor();

  void pasteStylesName();

  void deleteStyles();

  void eraseUnsedStyle();

  void blendStyles();

  void toggleLink();

  void eraseToggleLink();

  void enableCommands() override;

  void toggleKeyframe(int frame);

  // remove link from the studio palette (if linked)

  bool removeLink();

  // get back the style from the studio palette (if linked)

  void getBackOriginalStyle();

};

#endif  // STYLESELECTION_INCLUDED