#pragma once

#ifndef TCOLUMNFX_H
#define TCOLUMNFX_H

// TnzCore includes
#include "tthreadmessage.h"
#include "trastercm.h"

// TnzBase includes
#include "trasterfx.h"
#include "tbasefx.h"

#undef DVAPI
#undef DVVAR
#ifdef TOONZLIB_EXPORTS
#define DVAPI DV_EXPORT_API
#define DVVAR DV_EXPORT_VAR
#else
#define DVAPI DV_IMPORT_API
#define DVVAR DV_IMPORT_VAR
#endif

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

//    Forward declarations

class TXshLevelColumn;
class TXshPaletteColumn;
class TXshZeraryFxColumn;
class TXsheet;
class TXshColumn;
class TXshSimpleLevel;
class FxDag;
class TImageInfo;
class TOfflineGL;
class TVectorImageP;
class TVectorRenderData;

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

//*******************************************************************************************
//    TColumnFx  declaration
//*******************************************************************************************

class DVAPI TColumnFx : public TRasterFx {
public:
  TColumnFx() : TRasterFx() {}

  virtual int getColumnIndex() const         = 0;
  virtual std::wstring getColumnName() const = 0;
  virtual std::wstring getColumnId() const   = 0;
  virtual TXshColumn *getXshColumn() const   = 0;

  int getReferenceColumnIndex() const { return getColumnIndex(); }
};

//*******************************************************************************************
//    TLevelColumnFx  declaration
//*******************************************************************************************

class TLevelColumnFx : public TColumnFx {
  TXshLevelColumn *m_levelColumn;
  bool m_isCachable;

  TThread::Mutex m_mutex;
  TOfflineGL *m_offlineContext;

public:
  TLevelColumnFx();
  ~TLevelColumnFx();

  TFx *clone(bool recursive = true) const;

  TPalette *getPalette(int frame) const;
  TFilePath getPalettePath(int frame) const;

  void setColumn(TXshLevelColumn *column);
  TXshLevelColumn *getColumn() const { return m_levelColumn; }

  std::wstring getColumnName() const;
  std::wstring getColumnId() const;
  int getColumnIndex() const;
  TXshColumn *getXshColumn() const;

  bool isCachable() const { return m_isCachable; }

  bool canHandle(const TRenderSettings &info, double frame);
  TAffine handledAffine(const TRenderSettings &info, double frame);
  TAffine getDpiAff(int frame);

  TFxTimeRegion getTimeRegion() const;
  bool doGetBBox(double frame, TRectD &bBox, const TRenderSettings &info);
  std::string getAlias(double frame, const TRenderSettings &info) const;
  int getMemoryRequirement(const TRectD &rect, double frame,
                           const TRenderSettings &info);

  void doDryCompute(TRectD &rect, double frame, const TRenderSettings &info);
  void doCompute(TTile &tile, double frame, const TRenderSettings &info);
  void compute(TFlash &flash, int frame);

  void saveData(TOStream &os);
  void loadData(TIStream &is);

  const TPersistDeclaration *getDeclaration() const;
  std::string getPluginId() const;

private:
  void getImageInfo(TImageInfo &imageInfo, TXshSimpleLevel *sl,
                    TFrameId frameId);

  TImageP applyTzpFxs(TToonzImageP &ti, double frame,
                      const TRenderSettings &info);
  void applyTzpFxsOnVector(const TVectorImageP &vi, TTile &tile, double frame,
                           const TRenderSettings &info);

private:
  // not implemented
  TLevelColumnFx(const TLevelColumnFx &);
  TLevelColumnFx &operator=(const TLevelColumnFx &);
};

//*******************************************************************************************
//    TPaletteColumnFx  declaration
//*******************************************************************************************

class TPaletteColumnFx : public TColumnFx {
  TXshPaletteColumn *m_paletteColumn;

public:
  TPaletteColumnFx();
  ~TPaletteColumnFx();

  TFx *clone(bool recursive = true) const;

  TPalette *getPalette(int frame) const;
  TFilePath getPalettePath(int frame) const;

  void setColumn(TXshPaletteColumn *column) { m_paletteColumn = column; }
  TXshPaletteColumn *getColumn() const { return m_paletteColumn; }

  std::wstring getColumnName() const;
  std::wstring getColumnId() const;
  int getColumnIndex() const;
  TXshColumn *getXshColumn() const;

  bool isCachable() const { return false; }

  bool canHandle(const TRenderSettings &info, double frame);

  TFxTimeRegion getTimeRegion() const;
  bool doGetBBox(double frame, TRectD &bBox, const TRenderSettings &info);
  std::string getAlias(double frame, const TRenderSettings &info) const;
  TAffine getDpiAff(int frame);

  void doCompute(TTile &tile, double frame, const TRenderSettings &);
  void compute(TFlash &flash, int frame);

  const TPersistDeclaration *getDeclaration() const;
  std::string getPluginId() const;

private:
  // not implemented
  TPaletteColumnFx(const TPaletteColumnFx &);
  TPaletteColumnFx &operator=(const TPaletteColumnFx &);
};

//*******************************************************************************************
//    TZeraryColumnFx  declaration
//*******************************************************************************************

class DVAPI TZeraryColumnFx : public TColumnFx {
  TXshZeraryFxColumn *m_zeraryFxColumn;
  TZeraryFx *m_fx;

public:
  TZeraryColumnFx();
  ~TZeraryColumnFx();

  TZeraryFx *getZeraryFx() const { return m_fx; }
  void setZeraryFx(TFx *fx);

  void setColumn(TXshZeraryFxColumn *column);
  TXshZeraryFxColumn *getColumn() const { return m_zeraryFxColumn; }

  std::wstring getColumnName() const;
  std::wstring getColumnId() const;
  int getColumnIndex() const;
  TXshColumn *getXshColumn() const;

  bool canHandle(const TRenderSettings &info, double frame) { return true; }

  TFxTimeRegion getTimeRegion() const;
  bool doGetBBox(double frame, TRectD &bBox, const TRenderSettings &info);
  std::string getAlias(double frame, const TRenderSettings &info) const;

  void doCompute(TTile &tile, double frame, const TRenderSettings &);

  void saveData(TOStream &os);
  void loadData(TIStream &is);

  const TPersistDeclaration *getDeclaration() const;
  std::string getPluginId() const;

private:
  // not implemented
  TZeraryColumnFx(const TZeraryColumnFx &);
  TZeraryColumnFx &operator=(const TZeraryColumnFx &);
};

//*******************************************************************************************
//    TXsheetFx  declaration
//*******************************************************************************************

class TXsheetFx : public TRasterFx {
  FxDag *m_fxDag;

public:
  TXsheetFx();

  FxDag *getFxDag() const { return m_fxDag; }

  bool canHandle(const TRenderSettings &info, double frame) { return false; }

  std::string getAlias(double frame, const TRenderSettings &info) const;

  void doCompute(TTile &tile, double frame, const TRenderSettings &);
  bool doGetBBox(double frame, TRectD &bBox, const TRenderSettings &info);

  const TPersistDeclaration *getDeclaration() const;
  std::string getPluginId() const;

private:
  friend class FxDag;
  void setFxDag(FxDag *fxDag);

  // not implemented
  TXsheetFx(const TXsheetFx &);
  TXsheetFx &operator=(const TXsheetFx &);
};

//*******************************************************************************************
//    TOutputFx  declaration
//*******************************************************************************************

class TOutputFx : public TRasterFx {
  TRasterFxPort m_input;

public:
  TOutputFx();

  bool canHandle(const TRenderSettings &info, double frame) { return false; }

  bool doGetBBox(double frame, TRectD &bBox, const TRenderSettings &info);

  void doCompute(TTile &tile, double frame, const TRenderSettings &);

  const TPersistDeclaration *getDeclaration() const;
  std::string getPluginId() const;

private:
  // not implemented
  TOutputFx(const TOutputFx &);
  TOutputFx &operator=(const TOutputFx &);
};

#endif  // TCOLUMNFX_H