#pragma once

#ifndef HOOK_SELECTION

#define HOOK_SELECTION

#include "toonzqt/selection.h"

#include "toonzqt/dvmimedata.h"

#include "toonz/txshlevel.h"

#include "toonz/hook.h"

#include "tundo.h"

#include <set></set>

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

// HookUndo

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

class HookUndo final : public TUndo {

  HookSet m_oldHooks, m_newHooks;

  TXshLevelP m_level;

public:

  HookUndo(const TXshLevelP &level);

  ~HookUndo();

  void onAdd() override;

  void assignHookSet(const HookSet &src) const;

  void undo() const override;

  void redo() const override;

  int getSize() const override;

};

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

// HooksData

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

class HooksData final : public DvMimeData {

  struct HookPosition {

    int m_id;

    TPointD m_aPos, m_bPos;

    HookPosition(int id, const TPointD &aPos, const TPointD &bPos)

        : m_id(id), m_aPos(aPos), m_bPos(bPos) {}

  };

  std::vector<hookposition> m_hookPositions;</hookposition>

  TXshLevelP m_level;

public:

  HooksData(const TXshLevelP &level);

  ~HooksData();

  HooksData *clone() const override;

  void storeHookPositions(const std::vector<int> &ids);</int>

  void restoreHookPositions() const;

};

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

// HookSelection

//

// Derived from TSelection. Control which hooks are currently selected

// Note that a hook is made of two different parts (used e.g. to "pass the hook"

// between the feet in a walking cycle) that can be independently selected

// (1=A,2=B)

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

class HookSelection final : public TSelection {

  TXshLevelP m_level;

  std::set<std::pair<int, int="">> m_hooks;  // hookId, side : 1=A 2=B</std::pair<int,>

public:

  HookSelection();

  ~HookSelection();

  TSelection *clone() const;

  void setLevel(const TXshLevelP &level) { m_level = level; }

  void select(int id, int side);

  void unselect(int id, int side);

  bool isSelected(int id, int side) const;

  void invertSelection(int id, int side);

  bool isEmpty() const override;

  void selectNone() override { m_hooks.clear(); }

  HookSet *getHookSet() const;

  bool select(const TSelection *s);

  void enableCommands() override;

  // commands

  void deleteSelectedHooks();

  void copySelectedHooks();

  void cutSelectedHooks();

  void pasteSelectedHooks();

};

#endif  // HOOK_SELECTION