#include "graphwidget.h"
#include <QDebug>
#include "node.h"

//#include <QGraphicsTextItem>
#include "edge.h"
#include "math.h"

GraphWidget::GraphWidget(QWidget *parent) :
    QGraphicsView(parent),
    m_activeNode(0),
    m_showingNodeNumbers(false),
    m_followNode(0)
{
    qDebug() << __PRETTY_FUNCTION__;

    m_scene = new QGraphicsScene(this);
    m_scene->setItemIndexMethod(QGraphicsScene::NoIndex);
    m_scene->setSceneRect(-400, -400, 800, 800);
    setScene(m_scene);
    setCacheMode(CacheBackground);
    setViewportUpdateMode(BoundingRectViewportUpdate);
    setRenderHint(QPainter::Antialiasing);
    setTransformationAnchor(AnchorUnderMouse);
    setMinimumSize(400, 400);

    Node *node1 = new Node(this);
    node1->setHtml(QString("me"));
    m_scene->addItem(node1);
    node1->setPos(-10, -10);
    m_nodeList.append(node1);

    Node *node2 = new Node(this);
    node2->setHtml(QString("work"));
    m_scene->addItem(node2);
    node2->setPos(60, -10);
    m_nodeList.append(node2);

    Node *node3 = new Node(this);
    node3->setHtml(QString("read"));
    m_scene->addItem(node3);
    node3->setPos(-70, -10);
    m_nodeList.append(node3);

    Node *node4 = new Node(this);
    node4->setHtml(QString("pragmatic programmer"));
    m_scene->addItem(node4);
    node4->setPos(-120, -80);
    m_nodeList.append(node4);

    Node *node5 = new Node(this);
    node5->setHtml(QString("joy"));
    m_scene->addItem(node5);
    node5->setPos(-10, 50);
    m_nodeList.append(node5);

    Node *node6 = new Node(this);
    node6->setHtml(QString("rape goats"));
    m_scene->addItem(node6);
    node6->setPos(-10, 100);
    m_nodeList.append(node6);

    Node *node7 = new Node(this);
    node7->setHtml(QString("node <h1>important</h1> shit"));
    m_scene->addItem(node7);
    node7->setPos(-200, 50);
    m_nodeList.append(node7);

    Node *node8 = new Node(this);
    node8->setHtml(QString("more than\none lins"));
    m_scene->addItem(node8);
    node8->setPos(50, -80);
    m_nodeList.append(node8);

    Node *node9 = new Node(this);
    node9->setHtml(QString("iam a <b>bald</b> and <i>italian</i> guy"));
    m_scene->addItem(node9);
    node9->setPos(50, 40);
    m_nodeList.append(node9);

    Node *node10 = new Node(this);
    node10->setHtml(QString("no joke"));
    m_scene->addItem(node10);
    node10->setPos(-160, -10);
    m_nodeList.append(node10);

    Node *node11 = new Node(this);
    node11->setHtml(QString("salalal"));
    m_scene->addItem(node11);
    node11->setPos(-120, -120);
    m_nodeList.append(node11);

    Node *node12 = new Node(this);
    node12->setHtml(QString("lalalala"));
    m_scene->addItem(node12);
    node12->setPos(130, -10);
    m_nodeList.append(node12);;

    m_scene->addItem(new Edge(node1, node2));
    m_scene->addItem(new Edge(node1, node3));
    m_scene->addItem(new Edge(node3, node4));
    m_scene->addItem(new Edge(node1, node5));
    m_scene->addItem(new Edge(node5, node6));
    m_scene->addItem(new Edge(node4, node11));
    m_scene->addItem(new Edge(node2, node12));
    m_scene->addItem(new Edge(node3, node10));
    m_scene->addItem(new Edge(node10, node7));
    m_scene->addItem(new Edge(node5, node7));
    m_scene->addItem(new Edge(node2, node8));
    m_scene->addItem(new Edge(node2, node9));


    m_activeNode = m_nodeList.first();
    m_activeNode->setActive();
}

QGraphicsScene *GraphWidget::getScene()
{
    return m_scene;
}

void GraphWidget::keyPressEvent(QKeyEvent *event)
 {
    qDebug() << __PRETTY_FUNCTION__;
    qDebug() << event->key();

     switch (event->key()) {
     case Qt::Key_Up:
         if (m_activeNode)
            m_activeNode->moveBy(0, -20);
         break;
     case Qt::Key_Down:
         if (m_activeNode)
            m_activeNode->moveBy(0, 20);
         break;
     case Qt::Key_Left:
         if (m_activeNode)
            m_activeNode->moveBy(-20, 0);
         break;
     case Qt::Key_Right:
         if (m_activeNode)
            m_activeNode->moveBy(20, 0);
         break;
     case Qt::Key_Plus:
         scaleView(qreal(1.2));
         break;
     case Qt::Key_Minus:
         scaleView(1 / qreal(1.2));
         break;
     case Qt::Key_F:
         m_showingNodeNumbers = !m_showingNodeNumbers;
         if (m_showingNodeNumbers)
             m_followNumber.clear();

         showingAllNodeNumbers(m_showingNodeNumbers);
         if (m_showingNodeNumbers)
         {
             m_nodeList.first()->showNumber(0,true,true);
             m_followNode = m_nodeList.first();
         }
         break;

     case Qt::Key_Insert:
         insertNode();
         break;

     case Qt::Key_0:
     case Qt::Key_1:
     case Qt::Key_2:
     case Qt::Key_3:
     case Qt::Key_4:
     case Qt::Key_5:
     case Qt::Key_6:
     case Qt::Key_7:
     case Qt::Key_8:
     case Qt::Key_9:
         if (!m_showingNodeNumbers)
             break;

         m_followNumber.append(QString::number(event->key()-48));
         showingAllNodeNumbers(false);
         showingNodeNumbersBeginWithNumber(m_followNumber.toInt(), true);

         break;

     case Qt::Key_Backspace:
         if (!m_showingNodeNumbers)
             break;

         if (!m_followNumber.isEmpty())
         {
            m_followNumber.remove(m_followNumber.length()-1,1);
            if (m_followNumber.isEmpty())
            {
                showingAllNodeNumbers(true);
                m_nodeList.first()->showNumber(0,true,true);
                m_followNode = m_nodeList.first();
            }
            else
            {
                showingAllNodeNumbers(false);
                showingNodeNumbersBeginWithNumber(m_followNumber.toInt(), true);
            }
         }
         break;

     case Qt::Key_Return:
     case Qt::Key_Enter:
         if (m_followNode && m_showingNodeNumbers)
         {
             qDebug() << m_activeNode;
             qDebug() << m_followNode;
             showingAllNodeNumbers(false);
             if (m_activeNode)
                m_activeNode->setActive(false);
             m_activeNode = m_followNode;
             m_activeNode->setActive();
             m_showingNodeNumbers = false;
         }
         break;

     case Qt::Key_Delete:

         if (m_activeNode)
         {
             if (m_followNode==m_activeNode)
                 m_followNode=0;

             m_nodeList.removeAll(m_activeNode);
             delete m_activeNode;
             m_activeNode = 0;

             /// @bug
             if (m_showingNodeNumbers)
             {
                 m_showingNodeNumbers = false;
                 showingAllNodeNumbers(false);
             }
         }

         break;

     default:
         QGraphicsView::keyPressEvent(event);
     }
 }

void GraphWidget::wheelEvent(QWheelEvent *event)
{
    scaleView(pow((double)2, -event->delta() / 240.0));
}

void GraphWidget::drawBackground(QPainter *painter, const QRectF &rect)
 {
     Q_UNUSED(rect);

     QRectF sceneRect = this->sceneRect();

     // Fill
     QLinearGradient gradient(sceneRect.topLeft(), sceneRect.bottomRight());
     gradient.setColorAt(0, Qt::white);
     gradient.setColorAt(1, Qt::lightGray);
     painter->fillRect(rect.intersect(sceneRect), gradient);
     painter->setBrush(Qt::NoBrush);
     painter->drawRect(sceneRect);
 }


void GraphWidget::scaleView(qreal scaleFactor)
{
    qreal factor = transform().scale(scaleFactor, scaleFactor).mapRect(QRectF(0, 0, 1, 1)).width();
    if (factor < 0.2 || factor > 10) return;

    scale(scaleFactor, scaleFactor);
}

void GraphWidget::setActiveNode(Node *node)
{
    qDebug() << __PRETTY_FUNCTION__;

    if (m_activeNode!=0)
        m_activeNode->setActive(false);

    m_activeNode = node;
    m_activeNode->setActive();
}

void GraphWidget::insertNode()
{
    qDebug() << __PRETTY_FUNCTION__;
}

void GraphWidget::showingAllNodeNumbers(const bool &show)
{
    int i =0;
    for (QList<Node *>::const_iterator it = m_nodeList.begin(); it != m_nodeList.end(); it++, i++)
        dynamic_cast<Node*>(*it)->showNumber(i,show);
}

void GraphWidget::showingNodeNumbersBeginWithNumber(const int &number, const bool &show)
{
    int i(0);
    int hit(0);
    for (QList<Node *>::const_iterator it = m_nodeList.begin(); it != m_nodeList.end(); it++, i++)
    {
        if (i == number)
        {
            hit++;
            dynamic_cast<Node*>(*it)->showNumber(i,show,true);
            m_followNode = dynamic_cast<Node*>(*it);
            continue;
        }
        if (numberStartsWithNumber(i, number))
        {
            hit++;
            dynamic_cast<Node*>(*it)->showNumber(i,show);
        }
    }
    if (hit==1)
    {
        qDebug() << "set active";
        showingAllNodeNumbers(false);
        if (m_activeNode)
            m_activeNode->setActive(false);

        m_activeNode = m_followNode;
        m_activeNode->setActive();
        m_showingNodeNumbers = false;
    }

    /// @bug handle this case
//    else if (hit == 0)
//    {


//    }
}

bool GraphWidget::numberStartsWithNumber(const int &number, const int &prefix)
{
    return (QString::number(number)).startsWith(QString::number(prefix));
}