#include "CCNodeGrid.h"
#include "CCGrid.h"

#include "GroupCommand.h"
#include "Renderer.h"
#include "CustomCommand.h"


NS_CC_BEGIN

NodeGrid* NodeGrid::create()
{
    NodeGrid * pRet = new NodeGrid();
    if (pRet && pRet->init())
    {
        pRet->autorelease();
    }
    else
    {
        CC_SAFE_DELETE(pRet);
    }
    return pRet;
}

NodeGrid::NodeGrid()
:Node()
,_nodeGrid(nullptr)
,_gridTarget(nullptr)
{

}

void NodeGrid::setTarget(Node* target)
{
    CC_SAFE_RELEASE(_gridTarget);
    CC_SAFE_RETAIN(target);
    _gridTarget = target;
}

NodeGrid::~NodeGrid()
{
    CC_SAFE_RELEASE(_nodeGrid);
    CC_SAFE_RELEASE(_gridTarget);
}

bool NodeGrid::init()
{
    return Node::init();
}

void NodeGrid::onGridBeginDraw()
{
    if (_nodeGrid && _nodeGrid->isActive())
    {
        _nodeGrid->beforeDraw();
    }
}

void NodeGrid::onGridEndDraw()
{
    if(_nodeGrid && _nodeGrid->isActive())
    {
        _nodeGrid->afterDraw(this);
    }
}

void NodeGrid::visit()
{
    // quick return if not visible. children won't be drawn.
    if (!_visible)
    {
        return;
    }
    
    Renderer* renderer = Renderer::getInstance();

    GroupCommand* groupCommand = GroupCommand::getCommandPool().generateCommand();
    groupCommand->init(0,_vertexZ);
    renderer->addCommand(groupCommand);
    
    renderer->pushGroup(groupCommand->getRenderQueueID());

    kmGLPushMatrix();


    this->transform();
    
    kmGLGetMatrix(KM_GL_MODELVIEW, &_cachedMVmat);
    
    CustomCommand* gridBeginCmd = CustomCommand::getCommandPool().generateCommand();
    gridBeginCmd->init(0,_vertexZ);
    gridBeginCmd->func = CC_CALLBACK_0(NodeGrid::onGridBeginDraw, this);
    renderer->addCommand(gridBeginCmd);


    if(_gridTarget)
    {
        _gridTarget->visit();
    }
    
    int i = 0;

    if(!_children.empty())
    {
        sortAllChildren();
        // draw children zOrder < 0
        for( ; i < _children.size(); i++ )
        {
            auto node = _children.at(i);

            if ( node && node->getZOrder() < 0 )
                node->visit();
            else
                break;
        }
        // self draw,currently we have nothing to draw on NodeGrid, so there is no need to add render command
        this->draw();

        // Uses std::for_each to improve performance.
        std::for_each(_children.cbegin()+i, _children.cend(), [](Node* node){
            node->visit();
        });
    }
    else
    {
        this->draw();
    }
    
    // reset for next frame
    _orderOfArrival = 0;

    CustomCommand* gridEndCmd = CustomCommand::getCommandPool().generateCommand();
    gridEndCmd->init(0,_vertexZ);
    gridEndCmd->func = CC_CALLBACK_0(NodeGrid::onGridEndDraw, this);
    renderer->addCommand(gridEndCmd);

    renderer->popGroup();
 
    kmGLPopMatrix();
}

void NodeGrid::setGrid(GridBase *grid)
{
    CC_SAFE_RELEASE(_nodeGrid);
    CC_SAFE_RETAIN(grid);
    _nodeGrid = grid;
}

NS_CC_END