#include "iwa_timecodefx.h"

#include "tparamuiconcept.h"

#include <qfont></qfont>

#include <qfontmetrics></qfontmetrics>

#include <qimage></qimage>

#include <qpainter></qpainter>

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

Iwa_TimeCodeFx::Iwa_TimeCodeFx()

    : m_displayType(new TIntEnumParam(TYPE_HHMMSSFF, "HH;MM;SS;FF"))

    , m_frameRate(24)

    , m_startFrame(0)

    , m_position(TPointD(0.0, 0.0))

    , m_size(25)

    , m_textColor(TPixel32::White)

    , m_showBox(true)

    , m_boxColor(TPixel32::Black) {

  m_displayType->addItem(TYPE_HHMMSSFF2, "HH:MM:SS:FF");

  m_displayType->addItem(TYPE_FRAME, "Frame Number");

  m_position->getX()->setMeasureName("fxLength");

  m_position->getY()->setMeasureName("fxLength");

  m_size->setMeasureName("fxLength");

  m_size->setValueRange(0.1, (std::numeric_limits<double>::max)());</double>

  bindParam(this, "displayType", m_displayType);

  bindParam(this, "frameRate", m_frameRate);

  bindParam(this, "startFrame", m_startFrame);

  bindParam(this, "position", m_position);

  bindParam(this, "size", m_size);

  bindParam(this, "textColor", m_textColor);

  bindParam(this, "showBox", m_showBox);

  bindParam(this, "boxColor", m_boxColor);

}

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

bool Iwa_TimeCodeFx::doGetBBox(double frame, TRectD &bBox,

                               const TRenderSettings &ri) {

  bBox = TConsts::infiniteRectD;

  return true;

}

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

void Iwa_TimeCodeFx::doCompute(TTile &tile, double frame,

                               const TRenderSettings &ri) {

  double fac   = sqrt(fabs(ri.m_affine.det()));

  int size     = (int)(fac * fabs(m_size->getValue(frame)));

  TPoint point = convert(

      fac * m_position->getValue(frame) -

      (tile.m_pos + tile.getRaster()->getCenterD()) +

      TPointD(ri.m_cameraBox.getLx() / 2.0, ri.m_cameraBox.getLy() / 2.0));

#ifdef _WIN32

  QFont font("Arial", size);

#else

  QFont font("Helvetica", size);

#endif

  font.setWeight(QFont::Normal);

  QFontMetrics fm(font);

  QString timeCodeStr = getTimeCodeStr(frame, ri);

  int width           = fm.horizontalAdvance(timeCodeStr);

  int height          = fm.height();

  QImage img(width, height, QImage::Format_ARGB32);

  if (m_showBox->getValue()) {

    TPixel32 boxColor = m_boxColor->getValue(frame);

    img.fill(QColor((int)boxColor.r, (int)boxColor.g, (int)boxColor.b,

                    (int)boxColor.m));

  } else

    img.fill(Qt::transparent);

  QPainter painter(&img);

  TPixel32 color = m_textColor->getValue(frame);

  painter.setPen(

      QColor((int)color.r, (int)color.g, (int)color.b, (int)color.m));

  painter.setFont(font);

  painter.drawText(QPoint(0, fm.ascent()), timeCodeStr);

  tile.getRaster()->clear();

  TRaster32P ras32 = (TRaster32P)tile.getRaster();

  TRaster64P ras64 = (TRaster64P)tile.getRaster();

  if (ras32)

    putTimeCodeImage<traster32p, tpixel32="">(ras32, point, img);</traster32p,>

  else if (ras64)

    putTimeCodeImage<traster64p, tpixel64="">(ras64, point, img);</traster64p,>

  else

    throw TException("Iwa_TimeCodeFx: unsupported Pixel Type");

}

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

void Iwa_TimeCodeFx::getParamUIs(TParamUIConcept *&concepts, int &length) {

  concepts = new TParamUIConcept[length = 1];

  concepts[0].m_type  = TParamUIConcept::POINT;

  concepts[0].m_label = "Position";

  concepts[0].m_params.push_back(m_position);

}

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

std::string Iwa_TimeCodeFx::getAlias(double frame,

                                     const TRenderSettings &info) const {

  std::string alias = getFxType();

  alias += "[";

  std::string paramalias("");

  for (int i = 0; i < getParams()->getParamCount(); ++i) {

    TParam *param = getParams()->getParam(i);

    paramalias += param->getName() + "=" + param->getValueAlias(frame, 3);

  }

  return alias + std::to_string(frame) + "," + std::to_string(getIdentifier()) +

         paramalias + "]";

}

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

QString Iwa_TimeCodeFx::getTimeCodeStr(double frame,

                                       const TRenderSettings &ri) {

  int f = (int)frame + m_startFrame->getValue();

  if (m_displayType->getValue() != TYPE_FRAME) {

    QString separator =

        (m_displayType->getValue() == TYPE_HHMMSSFF ? ";" : ":");

    bool neg = (f < 0);

    f        = abs(f);

    int fps  = m_frameRate->getValue();

    int hh   = f / (fps * 60 * 60);

    f -= hh * fps * 60 * 60;

    int mm = f / (fps * 60);

    f -= mm * fps * 60;

    int ss = f / fps;

    int ff = f % fps;

    return QString((neg) ? "-" : "") +

           QString::number(hh).rightJustified(2, '0') + separator +

           QString::number(mm).rightJustified(2, '0') + separator +

           QString::number(ss).rightJustified(2, '0') + separator +

           QString::number(ff).rightJustified(2, '0');

  } else {   // TYPE_FRAMENUMBER

    f += 1;  // starting from "000001" with no frame offset.

    return QString((f < 0) ? "-" : "") +

           QString::number(abs(f)).rightJustified(6, '0');

  }

}

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

template <typename pixel="" raster,="" typename=""></typename>

void Iwa_TimeCodeFx::putTimeCodeImage(const RASTER srcRas, TPoint &pos,

                                      QImage &img) {

  for (int j = 0; j < img.height(); j++) {

    int rasY = pos.y + j;

    if (rasY < 0) continue;

    if (srcRas->getLy() <= rasY) break;

    PIXEL *pix  = srcRas->pixels(rasY);

    QRgb *img_p = (QRgb *)img.scanLine(img.height() - j - 1);

    for (int i = 0; i < img.width(); i++, img_p++) {

      int rasX = pos.x + i;

      if (rasX < 0) continue;

      if (srcRas->getLx() <= rasX) break;

      pix[rasX].r = (typename PIXEL::Channel)(

          qRed(*img_p) * (int)PIXEL::maxChannelValue / (int)UCHAR_MAX);

      pix[rasX].g = (typename PIXEL::Channel)(

          qGreen(*img_p) * (int)PIXEL::maxChannelValue / (int)UCHAR_MAX);

      pix[rasX].b = (typename PIXEL::Channel)(

          qBlue(*img_p) * (int)PIXEL::maxChannelValue / (int)UCHAR_MAX);

      pix[rasX].m = (typename PIXEL::Channel)(

          qAlpha(*img_p) * (int)PIXEL::maxChannelValue / (int)UCHAR_MAX);

    }

  }

}

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

FX_PLUGIN_IDENTIFIER(Iwa_TimeCodeFx, "iwa_TimeCodeFx");