#include "toonz/scriptbinding_renderer.h"

#include "toonz/scriptbinding_scene.h"

#include "toonz/scriptbinding_level.h"

#include "toonz/txsheet.h"

#include "toonz/txshsimplelevel.h"

#include "toonz/toonzscene.h"

#include "trenderer.h"

#include "toonz/scenefx.h"

#include "toonz/sceneproperties.h"

#include "toonz/tcamera.h"

#include "toutputproperties.h"

#include <qeventloop></qeventloop>

#include <qwaitcondition></qwaitcondition>

#include "timagecache.h"

namespace TScriptBinding

{

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

QScriptValue getScene(QScriptContext *context, const QScriptValue &sceneArg, Scene *&scene)

{

	scene = qscriptvalue_cast<scene *="">(sceneArg);</scene>

	if (!scene)

		return context->throwError(QObject::tr("First argument must be a scene : %1").arg(sceneArg.toString()));

	if (scene->getToonzScene() == 0)

		return context->throwError(QObject::tr("Can't render empty scene"));

	return QScriptValue();

}

void valueToIntList(const QScriptValue &arr, QList<int> &list)</int>

{

	list.clear();

	if (arr.isArray()) {

		int length = arr.property("length").toInteger();

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

			list.append(arr.property(i).toInteger());

	}

}

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

class Renderer::Imp : public TRenderPort

{

public:

	TScriptBinding::Image *m_outputImage;

	TScriptBinding::Level *m_outputLevel;

	TPointD m_cameraDpi;

	bool m_completed;

	TRenderer m_renderer;

	QList<int> m_columnList;</int>

	QList<int> m_frameList;</int>

	Imp() : m_completed(false)

	{

		m_renderer.setThreadsCount(1);

		m_renderer.addPort(this);

		m_outputImage = 0;

		m_outputLevel = 0;

	}

	~Imp()

	{

	}

	void setRenderArea(ToonzScene *scene)

	{

		TDimension cameraRes = scene->getCurrentCamera()->getRes();

		double rx = cameraRes.lx * 0.5, ry = cameraRes.ly * 0.5;

		TRectD renderArea(-rx, -ry, rx, ry);

		TRenderPort::setRenderArea(renderArea);

		m_cameraDpi = scene->getCurrentCamera()->getDpi();

	}

	void enableColumns(ToonzScene *scene, QList<bool> &oldStatus)</bool>

	{

		if (m_columnList.empty())

			return;

		QList<bool> newStatus;</bool>

		TXsheet *xsh = scene->getXsheet();

		for (int i = 0; i < xsh->getColumnCount(); i++) {

			oldStatus.append(xsh->getColumn(i)->isPreviewVisible());

			newStatus.append(false);

		}

		foreach (int i, m_columnList) {

			if (0 <= i && i < xsh->getColumnCount())

				newStatus[i] = true;

		}

		for (int i = 0; i < newStatus.length(); i++) {

			xsh->getColumn(i)->setPreviewVisible(newStatus[i]);

		}

	}

	void restoreColumns(ToonzScene *scene, const QList<bool> &oldStatus)</bool>

	{

		TXsheet *xsh = scene->getXsheet();

		for (int i = 0; i < oldStatus.length(); i++) {

			xsh->getColumn(i)->setPreviewVisible(oldStatus[i]);

		}

	}

	std::vector<trenderer::renderdata> *makeRenderData(ToonzScene *scene, const std::vector<int> &rows)</int></trenderer::renderdata>

	{

		TRenderSettings settings = scene

									   ->getProperties()

									   ->getOutputProperties()

									   ->getRenderSettings();

		QList<bool> oldColumnStates;</bool>

		enableColumns(scene, oldColumnStates);

		std::vector<trenderer::renderdata> *rds = new std::vector<trenderer::renderdata>;</trenderer::renderdata></trenderer::renderdata>

		for (int i = 0; i < (int)rows.size(); i++) {

			double frame = rows[i];

			TFxP sceneFx = buildSceneFx(scene, frame, 1, false);

			TFxPair fxPair;

			fxPair.m_frameA = sceneFx;

			rds->push_back(TRenderer::RenderData(frame, settings, fxPair));

		}

		restoreColumns(scene, oldColumnStates);

		return rds;

	}

	void render(std::vector<trenderer::renderdata> *rds)</trenderer::renderdata>

	{

		QMutex mutex;

		mutex.lock();

		m_completed = false;

		m_renderer.startRendering(rds);

		while (!m_completed) {

			QEventLoop loop;

			loop.processEvents();

			QWaitCondition waitCondition;

			waitCondition.wait(&mutex, 100);

		}

		mutex.unlock();

	}

	void renderFrame(ToonzScene *scene, int row, Image *outputImage)

	{

		setRenderArea(scene);

		std::vector<int> rows;</int>

		rows.push_back(row);

		m_outputImage = outputImage;

		m_outputLevel = 0;

		render(makeRenderData(scene, rows));

	}

	void renderScene(ToonzScene *scene, Level *outputLevel)

	{

		setRenderArea(scene);

		std::vector<int> rows;</int>

		if (m_frameList.empty()) {

			for (int i = 0; i < scene->getFrameCount(); i++)

				rows.push_back(i);

		} else {

			for (int i = 0; i < m_frameList.length(); i++)

				rows.push_back(m_frameList[i]);

		}

		m_outputImage = 0;

		m_outputLevel = outputLevel;

		render(makeRenderData(scene, rows));

	}

	void onRenderRasterStarted(const RenderData &renderData)

	{

		int a = 1;

	}

	void onRenderRasterCompleted(const RenderData &renderData)

	{

		TRasterP outputRaster = renderData.m_rasA;

		TRasterImageP img(outputRaster->clone());

		img->setDpi(m_cameraDpi.x, m_cameraDpi.y);

		if (m_outputImage)

			m_outputImage->setImg(img);

		else if (m_outputLevel) {

			std::vector<std::string> ids;</std::string>

			for (int i = 0; i < (int)renderData.m_frames.size(); i++) {

				TFrameId fid((int)(renderData.m_frames[i]) + 1);

				m_outputLevel->setFrame(fid, img);

				std::string id = m_outputLevel->getSimpleLevel()->getImageId(fid);

				ids.push_back(id);

			}

			img = TImageP();

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

				TImageCache::instance()->compress(ids[i]);

		}

	}

	void onRenderFailure(const RenderData &renderData, TException &e)

	{

	}

	void onRenderFinished()

	{

		m_completed = true;

	}

}; // class RenderEngine

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

Renderer::Renderer()

	: m_imp(new Imp())

{

}

Renderer::~Renderer()

{

}

QScriptValue Renderer::ctor(QScriptContext *context, QScriptEngine *engine)

{

	QScriptValue r = create(engine, new Renderer());

	r.setProperty("frames", engine->newArray());

	r.setProperty("columns", engine->newArray());

	return r;

}

QScriptValue Renderer::toString()

{

	return "Renderer";

}

QScriptValue Renderer::renderScene(const QScriptValue &sceneArg)

{

	QScriptValue obj = context()->thisObject();

	valueToIntList(obj.property("frames"), m_imp->m_frameList);

	valueToIntList(obj.property("columns"), m_imp->m_columnList);

	Scene *scene = 0;

	QScriptValue err = getScene(context(), sceneArg, scene);

	if (err.isError())

		return err;

	Level *outputLevel = new Level();

	// engine()->collectGarbage();

	m_imp->renderScene(scene->getToonzScene(), outputLevel);

	return create(engine(), outputLevel);

	/*

    for(int row=0;row<scene->getToonzScene()->getFrameCount();row++)</scene->

    {

      engine()->collectGarbage();

      TImageP img = renderEngine.renderFrame(row);

      if(img)

      {

        QScriptValue frame = create(new Image(img));

        QScriptValueList args; args << QString::number(row+1) << frame;

        newLevel.property("setFrame").call(newLevel, args);

      }

      else

      {

        return context()->throwError(tr("Render failed"));

      }

    }

    return newLevel;

    */

}

Q_INVOKABLE QScriptValue Renderer::renderFrame(const QScriptValue &sceneArg, int frame)

{

	QScriptValue obj = context()->thisObject();

	valueToIntList(obj.property("columns"), m_imp->m_columnList);

	Scene *scene = 0;

	QScriptValue err = getScene(context(), sceneArg, scene);

	if (err.isError())

		return err;

	Image *outputImage = new Image();

	engine()->collectGarbage();

	m_imp->renderFrame(scene->getToonzScene(), frame, outputImage);

	return create(engine(), outputImage);

	/*

    Scene *scene = 0;

    QScriptValue err = getScene(context(), sceneArg, scene);

    if(err.isError()) return err;

    engine()->collectGarbage();

    RenderEngine renderEngine(scene->getToonzScene());

    TImageP img = renderEngine.renderFrame(frame);

    for(int i=0;i

      xsh->getColumn(i)->setPreviewVisible(oldStatus[i]);

    if(img)

    {

      return create(engine(), new Image(img));

    }    

    else

    {

      return context()->throwError(tr("Render failed"));

    }

    */

}

/*

  QScriptValue Renderer::renderColumns(const QScriptValue &sceneArg, const QScriptValue &columnListArg)

  {

    Scene *scene = 0;

    QScriptValue err = getScene(context(), sceneArg, scene);

    if(err.isError()) return err;

    QList<bool> oldStatus;</bool>

    QList<bool> newStatus;</bool>

    TXsheet *xsh = scene->getToonzScene()->getXsheet();

    for(int i=0;i<xsh->getColumnCount();i++)</xsh->

    {

      oldStatus.append(xsh->getColumn(i)->isPreviewVisible());

      newStatus.append(false);

    }

    if(!columnListArg.isArray())

      return context()->throwError(tr("Second argument must be an array of column indices : ").arg(columnListArg.toString()));

    int m = columnListArg.property("length").toInt32();

    for(quint32 i=0;i<(int)m;i++)

    {

      QScriptValue c = columnListArg.property(i);

      if(!c.isNumber())

      {

        return context()->throwError(tr("Second argument must be an array of integer numbers : %1 (#%2)")

          .arg(columnListArg.toString())

          .arg(i));

      }

      int index = c.toInteger();

      if(0<=index && index

    }

    for(int i=0;i

      xsh->getColumn(i)->setPreviewVisible(newStatus[i]);

    err = QScriptValue();

    QScriptValue newLevel = create(new Level());

    RenderEngine renderEngine(scene->getToonzScene());

    for(int row=0;row<scene->getToonzScene()->getFrameCount();row++)</scene->

    {

      engine()->collectGarbage();

      TImageP img = renderEngine.renderFrame(row);

      if(img)

      {

        QScriptValue frame = create(new Image(img));

        QScriptValueList args; args << QString::number(row+1) << frame;

        newLevel.property("setFrame").call(newLevel, args);

      }

      else

      {

        err = context()->throwError(tr("Render failed"));

        break;

      }

    }

    for(int i=0;i

      xsh->getColumn(i)->setPreviewVisible(oldStatus[i]);

    if(err.isError()) return err;

    else return newLevel;

  }

  */

void Renderer::dumpCache()

{

	TImageCache::instance()->outputMap(0, "C:\\Users\\gmt\\PLI\\cache.log");

}

} // namespace TScriptBinding