#include "selectionutils.h"

// Tnz6 includes
#include "tapp.h"
#include "cellselection.h"
#include "castselection.h"
#include "filmstripselection.h"
#include "columnselection.h"

// TnzQt includes
#include "toonzqt/selection.h"
#include "toonzqt/tselectionhandle.h"

// TnzLib includes
#include "toonz/txsheet.h"
#include "toonz/txshsimplelevel.h"
#include "toonz/txshcell.h"
#include "toonz/txsheethandle.h"
#include "toonz/txshlevelhandle.h"

// tcg includes
#include "tcg/boost/range_utility.h"

// Boost includes
#include <boost/range/counting_range.hpp>
#include <boost/range/adaptor/transformed.hpp>

//*********************************************************************************
//    Local namespace
//*********************************************************************************

namespace {

template <typename LevelType>
LevelType *getLevel(const TXshCell &cell) {
  return dynamic_cast<LevelType *>(cell.m_level.getPointer());
}

template <>
TXshLevel *getLevel<TXshLevel>(const TXshCell &cell) {
  return cell.m_level.getPointer();
}

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

template <typename LevelType>
void getSelectedFrames(CastSelection *castSelection,
                       std::map<LevelType *, std::set<TFrameId>> &frames) {}

template <>
void getSelectedFrames<TXshSimpleLevel>(
    CastSelection *castSelection,
    std::map<TXshSimpleLevel *, std::set<TFrameId>> &frames) {
  std::vector<TXshLevel *> levels;
  castSelection->getSelectedLevels(levels);

  int l, lSize = levels.size();
  for (l = 0; l != lSize; ++l) {
    TXshSimpleLevel *sl = dynamic_cast<TXshSimpleLevel *>(levels[l]);
    if (!sl) continue;

    tcg::substitute(frames[sl], boost::counting_range(0, sl->getFrameCount()) |
      boost::adaptors::transformed([&sl](int index){ return sl->getFrameId(index); }));
  }
}

}  // namespace

//*********************************************************************************
//    Selection utility functions
//*********************************************************************************

template <typename LevelType>
void getSelectedFrames(const TXsheet &xsh, int r0, int c0, int r1, int c1,
                       std::map<LevelType *, std::set<TFrameId>> &frames) {
  for (int c = c0; c <= c1; ++c) {
    for (int r = r0; r <= r1; ++r) {
      const TXshCell &cell = xsh.getCell(r, c);

      if (LevelType *level = ::getLevel<LevelType>(cell))
        frames[level].insert(cell.getFrameId());
    }
  }
}

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

template <typename LevelType>
void getSelectedFrames(std::map<LevelType *, std::set<TFrameId>> &frames) {
  TSelection *selection = TSelection::getCurrent();

  if (TCellSelection *cellSelection =
          dynamic_cast<TCellSelection *>(selection)) {
    TXsheet *xsh = TApp::instance()->getCurrentXsheet()->getXsheet();

    int r0, c0, r1, c1;
    cellSelection->getSelectedCells(r0, c0, r1, c1);

    getSelectedFrames(*xsh, r0, c0, r1, c1, frames);
    return;
  }

  if (TColumnSelection *columnSelection =
          dynamic_cast<TColumnSelection *>(selection)) {
    TXsheet *xsh = TApp::instance()->getCurrentXsheet()->getXsheet();
    const std::set<int> &indices = columnSelection->getIndices();

    std::set<int>::const_iterator it, iEnd(indices.end());
    for (it = indices.begin(); it != iEnd; ++it) {
      TXshCellColumn *column =
          dynamic_cast<TXshCellColumn *>(xsh->getColumn(*it));

      int r0, r1;
      column->getRange(r0, r1);

      for (int r = r0; r <= r1; ++r) {
        const TXshCell &cell = column->getCell(r);

        LevelType *level    = ::getLevel<LevelType>(cell);
        const TFrameId &fid = cell.getFrameId();

        if (level) frames[level].insert(fid);
      }
    }

    return;
  }

  if (CastSelection *castSelection = dynamic_cast<CastSelection *>(selection)) {
    getSelectedFrames<LevelType>(castSelection, frames);
    return;
  }

  if (TFilmstripSelection *filmstripSelection =
          dynamic_cast<TFilmstripSelection *>(selection)) {
    LevelType *level = dynamic_cast<LevelType *>(
        TApp::instance()->getCurrentLevel()->getLevel());
    if (level) frames[level] = filmstripSelection->getSelectedFids();

    return;
  }
}

//*********************************************************************************
//    Selection utility  explicit instantiations
//*********************************************************************************

// template void getSelectedFrames<TXshLevel>(const TXsheet&, int, int, int,
// int,
//                                                 std::map<TXshLevel*,
//                                                 std::set<TFrameId> >&);
template void getSelectedFrames<TXshSimpleLevel>(
    const TXsheet &, int, int, int, int,
    std::map<TXshSimpleLevel *, std::set<TFrameId>> &);
// template void getSelectedFrames<TXshChildLevel>(const TXsheet&, int, int,
// int, int,
//                                                 std::map<TXshChildLevel*,
//                                                 std::set<TFrameId> >&);

template void getSelectedFrames<TXshLevel>(
    std::map<TXshLevel *, std::set<TFrameId>> &);
// template void getSelectedFrames<TXshSimpleLevel>(std::map<TXshSimpleLevel*,
// std::set<TFrameId> >&);
// template void getSelectedFrames<TXshChildLevel>(std::map<TXshChildLevel*,
// std::set<TFrameId> >&);