#include "toonz/txshzeraryfxcolumn.h"

#include "toonz/tcolumnfx.h"

#include "toonz/toonzscene.h"

#include "toonz/txshcell.h"

#include "toonz/txsheet.h"

#include "toonz/fxdag.h"

#include "toonz/txshzeraryfxlevel.h"

#include "tstream.h"

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

// TXshZeraryFxColumn

TXshZeraryFxColumn::TXshZeraryFxColumn(int frameCount)

    : m_zeraryColumnFx(new TZeraryColumnFx())

    , m_zeraryFxLevel(new TXshZeraryFxLevel()) {

  m_zeraryColumnFx->addRef();

  m_zeraryColumnFx->setColumn(this);

  m_zeraryFxLevel->addRef();

  m_zeraryFxLevel->setColumn(this);

  for (int i = 0; i < frameCount; i++)

    setCell(i, TXshCell(m_zeraryFxLevel, TFrameId(1)));

}

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

TXshZeraryFxColumn::TXshZeraryFxColumn(const TXshZeraryFxColumn &src)

    : m_zeraryColumnFx(new TZeraryColumnFx())

    , m_zeraryFxLevel(new TXshZeraryFxLevel()) {

  m_zeraryColumnFx->addRef();

  m_zeraryColumnFx->setColumn(this);

  m_zeraryFxLevel->addRef();

  m_zeraryFxLevel->setColumn(this);

  m_first = src.m_first;

  int i;

  for (i = 0; i < (int)src.m_cells.size(); i++)

    m_cells.push_back(TXshCell(m_zeraryFxLevel, src.m_cells[i].getFrameId()));

  assert((int)src.m_cells.size() == (int)m_cells.size());

  TFx *fx = src.getZeraryColumnFx()->getZeraryFx();

  if (fx) {

    std::wstring fxName = fx->getName();

    fx                  = fx->clone(false);

    fx->setName(fxName);

    m_zeraryColumnFx->setZeraryFx(fx);

  }

}

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

TXshZeraryFxColumn::~TXshZeraryFxColumn() {

  m_zeraryColumnFx->setColumn(0);

  m_zeraryColumnFx->release();

  m_zeraryFxLevel->release();

}

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

TXshColumn::ColumnType TXshZeraryFxColumn::getColumnType() const {

  return eZeraryFxType;

}

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

bool TXshZeraryFxColumn::canSetCell(const TXshCell &cell) const {

  return cell.isEmpty() || cell.m_level->getZeraryFxLevel() != 0;

}

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

TXshColumn *TXshZeraryFxColumn::clone() const {

  return new TXshZeraryFxColumn(*this);

}

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

TFx *TXshZeraryFxColumn::getFx() const { return m_zeraryColumnFx; }

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

bool TXshZeraryFxColumn::setCell(int row, const TXshCell &cell) {

  if (cell.isEmpty()) return false;

  TXshCell newCell = cell;

  // Sto settando delle celle in una colonna nuova, devo settare anche

  // l'effetto.

  if (isEmpty() && getZeraryColumnFx()->getZeraryFx() == 0) {

    newCell                      = TXshCell(m_zeraryFxLevel, cell.getFrameId());

    TXshZeraryFxLevel *fxLevel   = cell.m_level->getZeraryFxLevel();

    TXshZeraryFxColumn *fxColumn = fxLevel->getColumn();

    m_zeraryColumnFx->setZeraryFx(fxColumn->getZeraryColumnFx()->getZeraryFx());

  }

  return TXshCellColumn::setCell(row, newCell);

}

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

bool TXshZeraryFxColumn::setCells(int row, int rowCount,

                                  const TXshCell cells[]) {

  std::vector<txshcell> newCells;</txshcell>

  bool isEmptyColumn = isEmpty() && getZeraryColumnFx()->getZeraryFx() == 0;

  int i;

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

    if (isEmptyColumn)

      newCells.push_back(TXshCell(m_zeraryFxLevel, cells[i].getFrameId()));

    else

      newCells.push_back(cells[i]);

  }

  // Sto settando delle celle in una colonna nuova, devo settare anche

  // l'effetto.

  if (isEmptyColumn) {

    i = 0;

    while (i < rowCount && cells[i].isEmpty()) i++;

    if (i >= rowCount) return false;

    TXshZeraryFxLevel *fxLevel =

        dynamic_cast<txshzeraryfxlevel *="">(cells[i].m_level.getPointer());</txshzeraryfxlevel>

    TXshZeraryFxColumn *fxColumn = fxLevel->getColumn();

    m_zeraryColumnFx->setZeraryFx(fxColumn->getZeraryColumnFx()->getZeraryFx());

  }

  return TXshCellColumn::setCells(row, rowCount, &newCells[0]);

}

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

void TXshZeraryFxColumn::loadData(TIStream &is) {

  TPersist *p = 0;

  is >> p;

  if (!p) return;

  TZeraryColumnFx *fx = dynamic_cast<tzerarycolumnfx *="">(p);</tzerarycolumnfx>

  fx->addRef();

  if (m_zeraryColumnFx) {

    m_zeraryColumnFx->setColumn(0);

    m_zeraryColumnFx->release();

  }

  m_zeraryColumnFx = fx;

  m_zeraryColumnFx->setColumn(this);

  int r0, r1;

  bool touched = false;

  const TXshCell cell(m_zeraryFxLevel, TFrameId(1));

  std::string tagName;

  while (is.matchTag(tagName)) {

    if (tagName == "status") {

      int status;

      is >> status;

      setStatusWord(status);

    } else if (tagName == "cells") {

      while (is.matchTag(tagName)) {

        if (tagName == "cell") {

          if (!touched) {

            touched = true;

            if (getRange(r0, r1)) removeCells(r0, r1 - r0 + 1);

          }

          int r, n;

          is >> r >> n;

          for (int i = 0; i < n; i++) setCell(r++, cell);

        } else

          throw TException("expected <cell>");</cell>

        is.closeChild();

      }

    } else

      throw TException("expected <status> or <cells>");</cells></status>

    is.closeChild();

  }

}

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

void TXshZeraryFxColumn::saveData(TOStream &os) {

  os << m_zeraryColumnFx;

  os.child("status") << getStatusWord();

  int r0, r1;

  if (getRange(r0, r1)) {

    os.openChild("cells");

    for (int r = r0; r <= r1; r++) {

      TXshCell cell = getCell(r);

      if (cell.isEmpty()) continue;

      int n = 1;

      while (r + n <= r1 && !getCell(r + n).isEmpty()) n++;

      os.child("cell") << r << n;

      r += n - 1;

    }

    os.closeChild();

  }

}

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

PERSIST_IDENTIFIER(TXshZeraryFxColumn, "zeraryFxColumn")