#include <algorithm>

#include <glm/gtc/matrix_transform.hpp>

#include "defines.h"
#include "DrawVisitor.h"
#include "Scene.h"


DrawVisitor::DrawVisitor(Node *nodetodraw, glm::mat4 projection, bool force): force_(force)
{
    targets_.push_back(nodetodraw);
    modelview_ = glm::identity<glm::mat4>();
    projection_ = projection;
    num_duplicat_ = 1;
    transform_duplicat_ = glm::identity<glm::mat4>();
}

DrawVisitor::DrawVisitor(const std::vector<Node *> &nodestodraw, glm::mat4 projection, bool force):
    force_(force), targets_(nodestodraw), modelview_(glm::identity<glm::mat4>()),
    projection_(projection), num_duplicat_(1), transform_duplicat_(glm::identity<glm::mat4>())
{

}

void DrawVisitor::loop(int num, glm::mat4 transform)
{
    num_duplicat_ = CLAMP(num, 1, 10000);
    transform_duplicat_ = transform;
}

void DrawVisitor::visit(Node &n)
{
    // force visible status if required
    bool v = n.visible_;
    if (force_)
        n.visible_ = true;

    // find the node with this id
    std::vector<Node *>::iterator it = std::find_if(targets_.begin(), targets_.end(), hasId(n.id()));

    // found this node in the list of targets: draw it
    if (it != targets_.end()) {

        targets_.erase(it);

        for (int i = 0; i < num_duplicat_; ++i) {
            // draw multiple copies if requested
            n.draw(modelview_, projection_);
            modelview_ *= transform_duplicat_;
        }
    }

    // restore visibility
    n.visible_ = v;

    if (targets_.empty()) return;

    // update transform
    modelview_ *= n.transform_;
}


void DrawVisitor::visit(Group &n)
{
    // no need to traverse deeper if this node was drawn already
    if (targets_.empty()) return;

    // traverse children
    glm::mat4 mv = modelview_;
    for (NodeSet::iterator node = n.begin(); !targets_.empty() && node != n.end(); ++node) {
        if ( (*node)->visible_ || force_)
            (*node)->accept(*this);
        modelview_ = mv;
    }
}

void DrawVisitor::visit(Scene &n)
{
    modelview_ = glm::identity<glm::mat4>();
    n.root()->accept(*this);
}

void DrawVisitor::visit(Switch &n)
{
    // no need to traverse deeper if this node was drawn already
    if (targets_.empty()) return;

    // traverse acive child
    glm::mat4 mv = modelview_;
    if ( n.activeChild()->visible_ || force_)
        n.activeChild()->accept(*this);
    modelview_ = mv;
}

void DrawVisitor::visit(Primitive &n)
{
}