#include "toonzqt/strokesdata.h"

#include "tthreadmessage.h"

#include "tstroke.h"

#include "tpalette.h"

#include "toonzqt/rasterimagedata.h"

#include "toonz/toonzimageutils.h"

#include "toonz/trasterimageutils.h"

#include "toonz/stage.h"

using namespace std;

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

namespace {

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

int findStroke(const TVectorImageP &img, TStroke *stroke, const TAffine &aff) {

  TRectD strokeBBox = aff * stroke->getBBox();

  int count         = img->getStrokeCount();

  for (int i = 0; i < count; i++) {

    TStroke *s  = img->getStroke(i);

    TRectD bbox = s->getBBox();

    if (tdistance2(bbox.getP00(), strokeBBox.getP00()) +

            tdistance2(bbox.getP11(), strokeBBox.getP11()) >

        0.001)

      continue;

    return i;

  }

  return -1;

}

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

TAffine findOffset(const TVectorImageP &srcImg, const TVectorImageP &img) {

  TAffine offset;

  TVectorImageP tarImg = img;

  if (!tarImg) return offset;

  if (tarImg->getStrokeCount() == 0 || srcImg->getStrokeCount() == 0)

    return offset;

  bool done = false;

  int i;

  while (!done)

    for (i = 0; i < (int)srcImg->getStrokeCount(); i++) {

      TStroke *stroke = srcImg->getStroke(i);

      assert(stroke);

      if (findStroke(tarImg, stroke, offset) >= 0) {

        offset = offset * TTranslation(10, -10);

        break;

      }

      done = true;

    }

  return offset;

}

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

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

TStroke getStrokeByRect(TRectD r) {

  TStroke stroke;

  if (r.isEmpty()) return stroke;

  vector<tthickpoint> points;</tthickpoint>

  points.push_back(r.getP00());

  points.push_back((r.getP00() + r.getP01()) * 0.5);

  points.push_back(r.getP01());

  points.push_back((r.getP01() + r.getP11()) * 0.5);

  points.push_back(r.getP11());

  points.push_back((r.getP11() + r.getP10()) * 0.5);

  points.push_back(r.getP10());

  points.push_back((r.getP10() + r.getP00()) * 0.5);

  points.push_back(r.getP00());

  stroke.reshape(&(points[0]), points.size());

  stroke.setSelfLoop(true);

  return stroke;

}

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

}  // namespace

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

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

// StrokesData

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

void StrokesData::setImage(TVectorImageP image, const std::set<int> &indices) {</int>

  if (!image) return;

  if (indices.empty()) return;

  // indices e' un set; splitImage si aspetta un vector

  vector<int> indicesV(indices.begin(), indices.end());</int>

  QMutexLocker lock(image->getMutex());

  m_image = image->splitImage(indicesV, false);

  if (m_image->getPalette() == 0) {

    // nel caso lo stroke sia un path (e quindi senza palette)

    m_image->setPalette(new TPalette());

  }

}

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

void StrokesData::getImage(TVectorImageP image, std::set<int> &indices,</int>

                           bool insert) const {

  if (!m_image) return;

  TVectorImageP srcImg = m_image;

  QMutexLocker lock(image->getMutex());

  if (insert) {

    TAffine offset    = findOffset(srcImg, image);

    UINT oldImageSize = image->getStrokeCount();

    int insertAt      = image->mergeImage(srcImg, offset, false);

    UINT newImageSize = image->getStrokeCount();

    indices.clear();

    if (insertAt == 0)

      for (UINT sI = oldImageSize; sI < newImageSize; sI++) indices.insert(sI);

    else

      for (UINT sI = oldImageSize; sI < newImageSize; sI++)

        indices.insert(sI - oldImageSize + insertAt);

  } else {

    std::vector<int> indicesToInsert(indices.begin(), indices.end());</int>

    if (indicesToInsert.empty()) return;

    image->insertImage(srcImg, indicesToInsert);

  }

}

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

ToonzImageData *StrokesData::toToonzImageData(

    const TToonzImageP &imageToPaste) const {

  double dpix, dpiy;

  imageToPaste->getDpi(dpix, dpiy);

  assert(dpix != 0 && dpiy != 0);

  TScale sc(dpix / Stage::inch, dpiy / Stage::inch);

  TRectD bbox = sc * m_image->getBBox();

  bbox.x0     = tfloor(bbox.x0);

  bbox.y0     = tfloor(bbox.y0);

  bbox.x1     = tceil(bbox.x1);

  bbox.y1     = tceil(bbox.y1);

  TDimension size(bbox.getLx(), bbox.getLy());

  TToonzImageP app = ToonzImageUtils::vectorToToonzImage(

      m_image, sc, m_image->getPalette(), bbox.getP00(), size, 0, true);

  vector<trectd> rects;</trectd>

  vector<tstroke> strokes;</tstroke>

  TStroke stroke = getStrokeByRect(bbox);

  strokes.push_back(stroke);

  ToonzImageData *data = new ToonzImageData();

  data->setData(app->getRaster(), m_image->getPalette(), dpix, dpiy,

                TDimension(), rects, strokes, strokes, TAffine());

  return data;

}

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

FullColorImageData *StrokesData::toFullColorImageData(

    const TRasterImageP &imageToPaste) const {

  double dpix, dpiy;

  imageToPaste->getDpi(dpix, dpiy);

  assert(dpix != 0 && dpiy != 0);

  TScale sc(dpix / Stage::inch, dpiy / Stage::inch);

  TRectD bbox = sc * m_image->getBBox();

  bbox.x0     = tfloor(bbox.x0);

  bbox.y0     = tfloor(bbox.y0);

  bbox.x1     = tceil(bbox.x1);

  bbox.y1     = tceil(bbox.y1);

  TDimension size(bbox.getLx(), bbox.getLy());

  TRasterImageP app = TRasterImageUtils::vectorToFullColorImage(

      m_image, sc, m_image->getPalette(), bbox.getP00(), size, 0, true);

  vector<trectd> rects;</trectd>

  vector<tstroke> strokes;</tstroke>

  TStroke stroke = getStrokeByRect(bbox);

  strokes.push_back(stroke);

  FullColorImageData *data = new FullColorImageData();

  data->setData(app->getRaster(), m_image->getPalette(), dpix, dpiy,

                TDimension(), rects, strokes, strokes, TAffine());

  return data;

}