#include <tools replicator.h=""></tools>

#include <tools tool.h=""></tools>

#include <toonz tapplication.h=""></toonz>

#include <toonz txsheet.h=""></toonz>

#include <toonz txsheethandle.h=""></toonz>

#include <toonz txshlevelhandle.h=""></toonz>

#include <toonz tframehandle.h=""></toonz>

#include <toonz tobjecthandle.h=""></toonz>

#include <toonz dpiscale.h=""></toonz>

#include <toonz toonzscene.h=""></toonz>

#include <toonz sceneproperties.h=""></toonz>

#include <tgl.h></tgl.h>

#include <tpixelutils.h></tpixelutils.h>

const int TReplicator::multiplierSoftLimit = 32;

const int TReplicator::multiplierLimit = 256;

//************************************************************************

//    TReplicator implementation

//************************************************************************

TReplicator::TReplicator(TMetaObject &object):

  TAssistantBase(object),

  m_idSkipFirst("skipFirst"),

  m_idSkipLast("skipLast")

{

  addProperty( createSpinProperty(m_idSkipFirst, getSkipFirst(), 0) );

  addProperty( createSpinProperty(m_idSkipLast, getSkipLast(), 0) );

}

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

void

TReplicator::updateTranslation() const {

  TAssistantBase::updateTranslation();

  setTranslation(m_idSkipFirst, tr("Skip First Tracks"));

  setTranslation(m_idSkipLast, tr("Skip Last Tracks"));

}

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

int

TReplicator::getMultipler() const

  { return 1; }

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

void

TReplicator::getModifiers(const TAffine&, TInputModifier::List&) const

  { }

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

void

TReplicator::getPoints(const TAffine&, PointList&) const

  { }

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

TIntProperty*

TReplicator::createCountProperty(const TStringId &id, int def, int min, int max) {

  if (min < 0) min = 0;

  if (def < min) def = min;

  if (max <= 0) max = multiplierSoftLimit;

  return createSpinProperty(id, def, min, max);

}

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

void

TReplicator::transformPoints(const TAffine &aff, PointList &points, int i0, int i1) {

  int cnt = (int)points.size();

  if (i0 < 0) i0 = 0;

  if (i1 > cnt) i1 = cnt;

  if (i0 >= i1) return;

  points.reserve(points.size() + i1 - i0);

  for(int i = i0; i < i1; ++i)

    points.push_back(aff*points[i]);

}

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

void

TReplicator::drawReplicatorPoints(const TPointD *points, int count) {

  TPixelD color = (drawFlags & DRAW_ERROR) ? colorError : colorBase;

  color.m *= 0.3;

  TPixelD colorBack = makeContrastColor(color);

  glPushAttrib(GL_ALL_ATTRIB_BITS);

  tglEnableBlending();

  tglEnableLineSmooth(true, 1.0 * lineWidthScale);

  double pixelSize = sqrt(tglGetPixelSize2());

  TPointD a(5*pixelSize, 0), da(0, 0.5*pixelSize);

  TPointD b(0, 5*pixelSize), db(0.5*pixelSize, 0);

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

    const TPointD &p = points[i];

    tglColor(colorBack);

    tglDrawSegment(p - a - da, p + a - da);

    tglDrawSegment(p - b - db, p + b - db);

    tglColor(color);

    tglDrawSegment(p - a + da, p + a + da);

    tglDrawSegment(p - b + db, p + b + db);

  }

  glPopAttrib();

}

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

int

TReplicator::scanReplicators(

  TTool *tool,

  PointList *inOutPoints,

  TInputModifier::List *outModifiers,

  bool draw,

  bool enabledOnly,

  bool markEnabled,

  bool drawPoints,

  TImage *skipImage )

{

  bool outOfLimit = false;

  long long multiplier = 0;

  int inputPoints = inOutPoints ? (int)inOutPoints->size() : 0;

  if (tool)

  if (TToolViewer *viewer = tool->getViewer())

  if (TApplication *application = tool->getApplication())

  if (TXshLevelHandle *levelHandle = application->getCurrentLevel())

  if (TXshLevel *level = levelHandle->getLevel())

  if (TXshSimpleLevel *simpleLevel = level->getSimpleLevel())

  if (TFrameHandle *frameHandle = application->getCurrentFrame())

  if (TXsheetHandle *XsheetHandle = application->getCurrentXsheet())

  if (TXsheet *Xsheet = XsheetHandle->getXsheet())

  {

    ToonzScene *scene = Xsheet->getScene();

    TSceneProperties *props = scene ? scene->getProperties() : nullptr;

    TPointD dpiScale = getCurrentDpiScale(simpleLevel, tool->getCurrentFid());

    int frame = frameHandle->getFrame();

    int count = Xsheet->getColumnCount();

    TAffine worldToTrack;

    if ( tool->getToolType() & TTool::LevelTool

      && !application->getCurrentObject()->isSpline() )

    {

      worldToTrack.a11 /= dpiScale.x;

      worldToTrack.a22 /= dpiScale.y;

    }

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

      if (TXshColumn *column = Xsheet->getColumn(i))

      if (column->isCamstandVisible())

      if (column->isPreviewVisible())

      if (TImageP image = Xsheet->getCell(frame, i).getImage(false))

      if (image != skipImage)

      if (image->getType() == TImage::META)

      if (TMetaImage *metaImage = dynamic_cast<tmetaimage*>(image.getPointer()))</tmetaimage*>

      {

        TPixelD origColor = TAssistantBase::colorBase;

        TAffine imageToTrack = worldToTrack * tool->getColumnMatrix(i);

        if (draw) {

          if (props)

            TAssistantBase::colorBase = toPixelD(props->getColorFilterColor( column->getColorFilterId() ));

          TAssistantBase::colorBase.m *= column->getOpacity()/255.0;

          glPushMatrix();

          tglMultMatrix(imageToTrack);

        }

        TMetaImage::Reader reader(*metaImage);

        for(TMetaObjectListCW::iterator i = reader->begin(); i != reader->end(); ++i) {

          if (*i)

          if (const TReplicator *replicator = (*i)->getHandler<treplicator>())</treplicator>

          if (!enabledOnly || replicator->getEnabled())

          {

            if (!multiplier) multiplier = 1;

            bool enabled = replicator->getEnabled();

            if (enabled) {

              int m = replicator->getMultipler();

              if (m <= 0) m = 1;

              if (multiplier*m > multiplierLimit) {

                outOfLimit = true;

              } else {

                multiplier *= m;

                if (inOutPoints)

                  replicator->getPoints(imageToTrack, *inOutPoints);

                if (outModifiers)

                  replicator->getModifiers(imageToTrack, *outModifiers);

              }

            }

            if (draw) {

              unsigned int oldFlags = TReplicator::drawFlags;

              if (enabled && outOfLimit) TReplicator::drawFlags |= TReplicator::DRAW_ERROR;

              replicator->draw(viewer, enabled && markEnabled);

              TReplicator::drawFlags = oldFlags;

            }

          }

        } // for each meta object

        if (draw) {

          glPopMatrix();

          TAssistantBase::colorBase = origColor;

        }

      }

    } // for each column

  }

  if (drawPoints && inOutPoints && (int)inOutPoints->size() > inputPoints)

    drawReplicatorPoints(

      inOutPoints->data() + inputPoints,

      (int)inOutPoints->size() - inputPoints );

  return (int)multiplier;

}