#include "tstroke.h"

#include "tvectorrenderdata.h"

#include "tgl.h"

#include "tstrokeoutline.h"

#include "tsimplecolorstyles.h"

#include "tpalette.h"

#include "tregion.h"

#include "tregionoutline.h"

#include "tmathutil.h"

#include "tlevel_io.h"

#include "tenv.h"

#include "tvectorbrushstyle.h"

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

//    Vector Brush Style  static members

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

TFilePath TVectorBrushStyle::m_rootDir = TFilePath();

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

//    Vector Brush Prop  declaration

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

class VectorBrushProp final : public TStrokeProp {

  TVectorBrushStyle *m_style;

  TVectorImageP m_brush;

  TRectD m_brushBox;

  std::vector<tstrokeoutline> m_strokeOutlines;</tstrokeoutline>

  std::vector<tregionoutline> m_regionOutlines;</tregionoutline>

  double m_pixelSize;

public:

  VectorBrushProp(const TStroke *stroke, TVectorBrushStyle *style);

  TStrokeProp *clone(const TStroke *stroke) const override;

  void draw(const TVectorRenderData &rd) override;

  const TColorStyle *getColorStyle() const override;

private:

  // not implemented

  VectorBrushProp(const VectorBrushProp &);

  VectorBrushProp &operator=(const VectorBrushProp &);

};

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

//    Vector Brush Prop  implementation

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

VectorBrushProp::VectorBrushProp(const TStroke *stroke,

                                 TVectorBrushStyle *style)

    : TStrokeProp(stroke)

    , m_style(style)

    , m_brush(style->getImage())

    , m_brushBox(m_brush->getBBox())

    , m_pixelSize(.0) {}

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

TStrokeProp *VectorBrushProp::clone(const TStroke *stroke) const {

  return new VectorBrushProp(stroke, m_style);

}

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

void VectorBrushProp::draw(const TVectorRenderData &rd) {

  // Ensure that the stroke overlaps our clipping rect

  if (rd.m_clippingRect != TRect() && !rd.m_is3dView &&

      !convert(rd.m_aff * m_stroke->getBBox()).overlaps(rd.m_clippingRect))

    return;

  TPaletteP palette(m_brush->getPalette());

  if (!palette) return;

  static TOutlineUtil::OutlineParameter param;  // unused, but requested

  // Build a solid color style to draw each m_vi's stroke with.

  TSolidColorStyle colorStyle;

  // Push the specified rd affine before drawing

  glPushMatrix();

  tglMultMatrix(rd.m_aff);

  // 1. If necessary, build the outlines

  double currentPixelSize = sqrt(tglGetPixelSize2());

  bool differentPixelSize = !isAlmostZero(currentPixelSize - m_pixelSize, 1e-5);

  m_pixelSize             = currentPixelSize;

  int i, viRegionsCount = m_brush->getRegionCount(),

         viStrokesCount = m_brush->getStrokeCount();

  if (differentPixelSize || m_strokeChanged) {

    m_strokeChanged = false;

    // 1a. First, the regions

    m_regionOutlines.resize(viRegionsCount);

    for (i = 0; i < viRegionsCount; ++i) {

      TRegionOutline &outline    = m_regionOutlines[i];

      const TRegion *brushRegion = m_brush->getRegion(i);

      // Build the outline

      outline.clear();

      TOutlineUtil::makeOutline(*getStroke(), *brushRegion, m_brushBox,

                                outline);

    }

    // 1b. Then, the strokes

    m_strokeOutlines.resize(viStrokesCount);

    for (i = 0; i < viStrokesCount; ++i) {

      TStrokeOutline &outline    = m_strokeOutlines[i];

      const TStroke *brushStroke = m_brush->getStroke(i);

      outline.getArray().clear();

      TOutlineUtil::makeOutline(*getStroke(), *brushStroke, m_brushBox, outline,

                                param);

    }

  }

  // 2. Draw the outlines

  UINT s, t, r, strokesCount = m_brush->getStrokeCount(),

                regionCount = m_brush->getRegionCount();

  for (s = 0; s < strokesCount; s = t)  // Each cycle draws a group

  {

    // A vector image stores group strokes with consecutive indices.

    // 2a. First, draw regions in the strokeIdx-th stroke's group

    for (r = 0; r < regionCount; ++r) {

      if (m_brush->sameGroupStrokeAndRegion(s, r)) {

        const TRegion *brushRegion = m_brush->getRegion(r);

        const TColorStyle *brushStyle =

            palette->getStyle(brushRegion->getStyle());

        assert(brushStyle);

        // Draw the outline

        colorStyle.setMainColor(brushStyle->getMainColor());

        colorStyle.drawRegion(0, false, m_regionOutlines[r]);

      }

    }

    // 2b. Then, draw all strokes in strokeIdx-th stroke's group

    for (t = s; t < strokesCount && m_brush->sameGroup(s, t); ++t) {

      const TStroke *brushStroke = m_brush->getStroke(t);

      const TColorStyle *brushStyle =

          palette->getStyle(brushStroke->getStyle());

      if (!brushStyle) continue;

      colorStyle.setMainColor(brushStyle->getMainColor());

      colorStyle.drawStroke(0, &m_strokeOutlines[t],

                            brushStroke);  // brushStroke unused but requested

    }

  }

  glPopMatrix();

}

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

const TColorStyle *VectorBrushProp::getColorStyle() const { return m_style; }

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

//    Vector Brush Style  implementation

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

TVectorBrushStyle::TVectorBrushStyle() : m_colorCount(0) {}

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

TVectorBrushStyle::TVectorBrushStyle(const std::string &brushName,

                                     TVectorImageP vi)

    : m_brush(vi) {

  loadBrush(brushName);

}

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

TVectorBrushStyle::~TVectorBrushStyle() {}

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

void TVectorBrushStyle::loadBrush(const std::string &brushName) {

  m_brushName  = brushName;

  m_colorCount = 0;

  if (brushName.empty()) return;

  if (!m_brush) {

    // Load the image associated with fp

    TFilePath fp(m_rootDir + TFilePath(brushName + ".pli"));

    TLevelReaderP lr(fp);

    TLevelP level = lr->loadInfo();

    m_brush = lr->getFrameReader(level->begin()->first)->load();

    assert(m_brush);

    TPalette *palette = level->getPalette();

    m_brush->setPalette(palette);

  }

  assert(m_brush);

  m_colorCount =

      m_brush->getPalette()->getStyleInPagesCount() - 1;  // No transparent

}

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

TColorStyle *TVectorBrushStyle::clone() const {

  TVectorImageP brush;

  if (m_brush) {

    // Clone m_brush and its palette

    brush = m_brush->clone();  // NOTE: This does NOT clone the palette, too.

    brush->setPalette(m_brush->getPalette()->clone());

  }

  TVectorBrushStyle *theClone = new TVectorBrushStyle(m_brushName, brush);

  theClone->assignNames(this);

  theClone->setFlags(getFlags());

  return theClone;

}

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

QString TVectorBrushStyle::getDescription() const { return "VectorBrushStyle"; }

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

TStrokeProp *TVectorBrushStyle::makeStrokeProp(const TStroke *stroke) {

  return new VectorBrushProp(stroke, this);

}

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

void TVectorBrushStyle::saveData(TOutputStreamInterface &osi) const {

  osi << m_brushName;

  // Save palette

  osi << m_colorCount;

  TPalette *pal = m_brush->getPalette();

  assert(pal);

  int p, pagesCount = pal->getPageCount();

  for (p = 0; p < pagesCount; ++p) {

    TPalette::Page *page = pal->getPage(p);

    int s, stylesCount = page->getStyleCount();

    for (s = 0; s < stylesCount; ++s) osi << page->getStyle(s)->getMainColor();

  }

}

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

void TVectorBrushStyle::loadData(TInputStreamInterface &isi) {

  std::string str;

  isi >> str;

  assert(!str.empty());

  loadBrush(str);

  int colorCount;

  isi >> colorCount;

  if (colorCount != m_colorCount)

    return;  // Palette mismatch! Just skip palette loading...

  // WARNING: Is it needed to read all colors nonetheless?

  // Load palette colors

  TPalette *pal = m_brush->getPalette();

  assert(pal);

  TPixel32 color;

  int p, pagesCount = pal->getPageCount();

  for (p = 0; p < pagesCount; ++p) {

    TPalette::Page *page = pal->getPage(p);

    int s, stylesCount = page->getStyleCount();

    for (s = 0; s < stylesCount; ++s) {

      isi >> color;

      page->getStyle(s)->setMainColor(color);

    }

  }

}

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

int TVectorBrushStyle::getColorStyleId(int index) const {

  if (index < 0) return 1;

  ++index;  // Skipping transparent

  TPalette *pal = m_brush->getPalette();

  assert(pal);

  int p, pagesCount = pal->getPageCount();

  for (p = 0; p < pagesCount; ++p) {

    TPalette::Page *page = pal->getPage(p);

    int stylesCount = page->getStyleCount();

    if (index - stylesCount < 0) break;

    index -= stylesCount;

  }

  if (p >= pagesCount) return -1;

  TPalette::Page *page = pal->getPage(p);

  return page->getStyleId(index);

}

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

TPixel32 TVectorBrushStyle::getMainColor() const {

  if (!m_brush) return TPixel32::Transparent;

  TPalette *pal = m_brush->getPalette();

  return pal->getStyle(1)->getMainColor();

}

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

void TVectorBrushStyle::setMainColor(const TPixel32 &color) {

  if (!m_brush) return;

  TPalette *pal = m_brush->getPalette();

  pal->getStyle(1)->setMainColor(color);

}

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

int TVectorBrushStyle::getColorParamCount() const { return m_colorCount; }

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

TPixel32 TVectorBrushStyle::getColorParamValue(int index) const {

  TPalette *pal = m_brush->getPalette();

  assert(pal);

  int styleId              = getColorStyleId(index);

  if (styleId < 0) styleId = 1;

  return pal->getStyle(styleId)->getMainColor();

}

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

void TVectorBrushStyle::setColorParamValue(int index, const TPixel32 &color) {

  TPalette *pal = m_brush->getPalette();

  assert(pal);

  int styleId              = getColorStyleId(index);

  if (styleId < 0) styleId = 1;

  return pal->getStyle(styleId)->setMainColor(color);

}

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

//    Vector Style  global declaration instance

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

namespace {

TColorStyle::Declaration declaration(new TVectorBrushStyle());

}