/****************************************************************************
Copyright (c) 2009      On-Core
Copyright (c) 2010-2012 cocos2d-x.org
Copyright (c) 2013-2016 Chukong Technologies Inc.
Copyright (c) 2017-2018 Xiamen Yaji Software Co., Ltd.
Copyright (c) 2019-present Axmol Engine contributors (see AUTHORS.md).

 https://axmolengine.github.io/

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.
****************************************************************************/
#pragma once

#include "base/Ref.h"
#include "base/Types.h"
#include "base/Director.h"
#include "renderer/GroupCommand.h"
#include "renderer/CustomCommand.h"
#include "renderer/CallbackCommand.h"

NS_AX_BEGIN

class Texture2D;
// class Grabber;
// class GLProgram;
class Node;

namespace backend
{
class RenderTarget;
}

/**
 * @addtogroup _2d
 * @{
 */

/** Base class for Other grid.
 */
class AX_DLL GridBase : public Ref
{
public:
    /**
    Destructor.
     * @js NA
     * @lua NA
     */
    virtual ~GridBase();

    /**@{
     Interface for custom action when before or after draw.
     @js NA
     */
    virtual void beforeBlit() {}
    virtual void afterBlit() {}
    /**@}*/

    /**Interface used to blit the texture with grid to screen.*/
    virtual void blit() = 0;
    /**Interface, Reuse the grid vertices.*/
    virtual void reuse() = 0;
    /**Interface, Calculate the vertices used for the blit.*/
    virtual void calculateVertexPoints() = 0;

    /**@{
     Init the Grid base.
     @param gridSize the size of the grid.
     @param texture The texture used for grab.
     @param flipped whether or not the grab texture should be flip by Y or not.
     @param rect The effective grid rect.
    */
    bool initWithSize(const Vec2& gridSize);
    bool initWithSize(const Vec2& gridSize, const Rect& rect);
    bool initWithSize(const Vec2& gridSize, Texture2D* texture, bool flipped);
    bool initWithSize(const Vec2& gridSize, Texture2D* texture, bool flipped, const Rect& rect);

    /**@}*/
    /** @{
    Getter and setter of the active state of the grid.
    */
    bool isActive() const { return _active; }
    void setActive(bool active);
    /**@}*/

    /** Get number of times that the grid will be reused. */
    int getReuseGrid() const { return _reuseGrid; }
    /** Set number of times that the grid will be reused. */
    void setReuseGrid(int reuseGrid) { _reuseGrid = reuseGrid; }

    /** The size of the grid. */
    const Vec2& getGridSize() const { return _gridSize; }
    /**Set the size of the grid.*/
    void setGridSize(const Vec2& gridSize) { _gridSize = gridSize; }

    /** Pixels between the grids. */
    const Vec2& getStep() const { return _step; }
    /**Get the pixels between the grids.*/
    void setStep(const Vec2& step) { _step = step; }

    /** is texture flipped. */
    bool isTextureFlipped() const { return _isTextureFlipped; }
    /**Set the texture flipped or not.*/
    void setTextureFlipped(bool flipped);

    /**@{
     Init and reset the status when render effects by using the grid.
     */
    void beforeDraw();
    void afterDraw(Node* target);
    /**@}*/

    /**Change projection to 2D for grabbing.*/
    void set2DProjection();

    /**
     * @brief Set the effect grid rect.
     * @param rect The effect grid rect.
     */
    void setGridRect(const Rect& rect);
    /**
     * @brief Get the effect grid rect.
     * @return Return the effect grid rect.
     */
    const Rect& getGridRect() const { return _gridRect; }

protected:
    void updateBlendState();

    bool _active   = false;
    int _reuseGrid = 0;
    Vec2 _gridSize;
    Texture2D* _texture = nullptr;
    Vec2 _step;
    bool _isTextureFlipped                   = false;
    Director::Projection _directorProjection = Director::Projection::_2D;
    Rect _gridRect;

    Color4F _clearColor = {0, 0, 0, 0};

    CustomCommand _drawCommand;
    //CallbackCommand _beforeDrawCommand;
    //CallbackCommand _afterDrawCommand;
    //CallbackCommand _beforeBlitCommand;
    //CallbackCommand _afterBlitCommand;

    // New
    backend::RenderTarget* _oldRenderTarget = nullptr;
    backend::RenderTarget* _renderTarget    = nullptr;

    backend::UniformLocation _mvpMatrixLocation;
    backend::UniformLocation _textureLocation;
    backend::ProgramState* _programState = nullptr;

    BlendFunc _blendFunc;
};

/**
 Grid3D is a 3D grid implementation. Each vertex has 3 dimensions: x,y,z
 */
class AX_DLL Grid3D : public GridBase
{
public:
    /** create one Grid. */
    static Grid3D* create(const Vec2& gridSize);
    /** create one Grid. */
    static Grid3D* create(const Vec2& gridSize, const Rect& rect);
    /** create one Grid. */
    static Grid3D* create(const Vec2& gridSize, Texture2D* texture, bool flipped);
    /** create one Grid. */
    static Grid3D* create(const Vec2& gridSize, Texture2D* texture, bool flipped, const Rect& rect);
    /**
     Constructor.
     * @js ctor
     */
    Grid3D();
    /**
     Destructor.
     * @js NA
     * @lua NA
     */
    ~Grid3D();

    /** Returns the vertex at a given position.
     * @js NA
     * @lua NA
     */
    Vec3 getVertex(const Vec2& pos) const;

    /** Returns the original (non-transformed) vertex at a given position.
     * @js NA
     * @lua NA
     */
    Vec3 getOriginalVertex(const Vec2& pos) const;

    /** Sets a new vertex at a given position.
     * @lua NA
     */
    void setVertex(const Vec2& pos, const Vec3& vertex);
    /**@{
     Implementations for interfaces in base class.
     */
    virtual void beforeBlit() override;
    virtual void afterBlit() override;
    virtual void blit() override;
    virtual void reuse() override;
    virtual void calculateVertexPoints() override;
    /**@}*/

    /**@{
     Getter and Setter for depth test state when blit.
     @js NA
     */
    void setNeedDepthTestForBlit(bool neededDepthTest) { _needDepthTestForBlit = neededDepthTest; }
    bool getNeedDepthTestForBlit() const { return _needDepthTestForBlit; }
    /**@}*/
protected:
    void updateVertexBuffer();
    void updateVertexAndTexCoordinate();

    void* _vertexBuffer        = nullptr;
    void* _texCoordinates      = nullptr;
    void* _vertices            = nullptr;
    void* _originalVertices    = nullptr;
    unsigned short* _indices   = nullptr;
    bool _needDepthTestForBlit = false;
    bool _oldDepthTest         = false;
    bool _oldDepthWrite        = false;
};

/**
 TiledGrid3D is a 3D grid implementation. It differs from Grid3D in that
 the tiles can be separated from the grid.
*/
class AX_DLL TiledGrid3D : public GridBase
{
public:
    /** Create one Grid. */
    static TiledGrid3D* create(const Vec2& gridSize);
    /** Create one Grid. */
    static TiledGrid3D* create(const Vec2& gridSize, const Rect& rect);
    /** Create one Grid. */
    static TiledGrid3D* create(const Vec2& gridSize, Texture2D* texture, bool flipped);
    /** Create one Grid. */
    static TiledGrid3D* create(const Vec2& gridSize, Texture2D* texture, bool flipped, const Rect& rect);

    /**@{
     Implementations for interfaces in base class.
     */
    virtual void blit() override;
    virtual void reuse() override;
    virtual void calculateVertexPoints() override;
    /**@}*/

    /** Returns the tile at the given position.
     * @js NA
     * @lua NA
     */
    Quad3 getTile(const Vec2& pos) const;
    /** Returns the original tile (untransformed) at the given position.
     * @js NA
     * @lua NA
     */
    Quad3 getOriginalTile(const Vec2& pos) const;

    /** Sets a new tile.
     * @lua NA
     */
    void setTile(const Vec2& pos, const Quad3& coords);

protected:
    /**
     Constructor.
     * @js ctor
     */
    TiledGrid3D() = default;
    /**
     Destructor.
     * @js NA
     * @lua NA
     */
    ~TiledGrid3D();

    void updateVertexBuffer();
    void updateVertexAndTexCoordinate();

    void* _texCoordinates    = nullptr;
    void* _vertices          = nullptr;
    void* _originalVertices  = nullptr;
    unsigned short* _indices = nullptr;
    void* _vertexBuffer      = nullptr;
};

// end of effects group
/// @}

NS_AX_END