/************************************************************************
* file name         : blocks_graphics_view.cpp
* ----------------- :
* creation time     : 2016/06/26
* copyright         : (c) 2016 Victor Zarubkin
* author            : Victor Zarubkin
* email             : v.s.zarubkin@gmail.com
* ----------------- :
* description       : The file contains implementation of GraphicsScene and GraphicsView and
*                   : it's auxiliary classes for displyaing easy_profiler blocks tree.
* ----------------- :
* change log        : * 2016/06/26 Victor Zarubkin: moved sources from graphics_view.h
*                   :       and renamed classes from My* to Prof*.
*                   : *
* ----------------- :
* license           : TODO: add license text
************************************************************************/

#include <QWheelEvent>
#include "blocks_graphics_view.h"

/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

const ProfViewGlobalSignals GLOBALS;

/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

ProfGraphicsPolygonItem::ProfGraphicsPolygonItem(QGraphicsItem* _parent) : QGraphicsPolygonItem(_parent)
{
}

ProfGraphicsPolygonItem::~ProfGraphicsPolygonItem()
{
}

/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

ProfGraphicsTextItem::ProfGraphicsTextItem(const char* _text, QGraphicsItem* _parent) : QGraphicsSimpleTextItem(_text, _parent), QObject()
{
    connect(&GLOBALS, &ProfViewGlobalSignals::scaleIncreased, this, &ProfGraphicsTextItem::onScaleIncrease);
    connect(&GLOBALS, &ProfViewGlobalSignals::scaleDecreased, this, &ProfGraphicsTextItem::onScaleDecrease);
}

ProfGraphicsTextItem::~ProfGraphicsTextItem()
{
}

void ProfGraphicsTextItem::onScaleIncrease(qreal _scale)
{
    setScale(100.0 / _scale);

    if (!isVisible())
    {
        if (boundingRect().width() * 100.0 < parentItem()->boundingRect().width() * _scale)
        {
            show();
        }
    }
}

void ProfGraphicsTextItem::onScaleDecrease(qreal _scale)
{
    setScale(100.0 / _scale);

    if (isVisible())
    {
        if (boundingRect().width() * 100.0 > parentItem()->boundingRect().width() * _scale)
        {
            hide();
        }
    }
}

/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

const qreal LEFT = -2500000;
const qreal WIDTH = 5000000;
const int N_ITEMS = 200000;

ProfGraphicsScene::ProfGraphicsScene(QGraphicsView* _parent) : QGraphicsScene(_parent), m_start(0)
{
    setSceneRect(QRectF(LEFT, -200, WIDTH, 220));
    test();
}

ProfGraphicsScene::ProfGraphicsScene(const thread_blocks_tree_t& _blocksTree, QGraphicsView* _parent) : QGraphicsScene(_parent), m_start(0)
{
    setTree(_blocksTree);
}

ProfGraphicsScene::~ProfGraphicsScene()
{
}

void ProfGraphicsScene::test()
{
    QPolygonF poly(QRectF(-5, -180, 10, 180));
    for (int i = 0; i < N_ITEMS; ++i)
    {
        ProfGraphicsPolygonItem* item = new ProfGraphicsPolygonItem();
        int h = 50 + rand() % 131;
        item->setPolygon(QRectF(-5, -h, 10, h));
        item->setPos(LEFT + i * 10, 0);
        item->setBrush(QBrush(QColor(30 + rand() % 225, 30 + rand() % 225, 30 + rand() % 225)));
        addItem(item);
    }
}

void ProfGraphicsScene::setTree(const thread_blocks_tree_t& _blocksTree)
{
    m_start = -1;
    profiler::timestamp_t finish = 0;
    for (const auto& threadTree : _blocksTree)
    {
        ++items_couinter;
        const auto timestart = threadTree.second.children.front().node->block()->getBegin();
        const auto timefinish = threadTree.second.children.back().node->block()->getEnd();
        if (m_start > timestart) m_start = timestart;
        if (finish < timefinish) finish = timefinish;
    }

    const qreal endX = time2position(finish + 1000000);
    setSceneRect(QRectF(0, 0, endX, 110 * _blocksTree.size()));

    qreal y = 0;
    for (const auto& threadTree : _blocksTree)
    {
        setTree(threadTree.second.children, y);
        y += 110;
    }
}

void ProfGraphicsScene::setTree(const BlocksTree::children_t& _children, qreal _y, int _level)
{
    for (const auto& child : _children)
    {
        ++items_couinter;
        ProfGraphicsPolygonItem* item = new ProfGraphicsPolygonItem();

        const qreal xbegin = time2position(child.node->block()->getBegin());
        const qreal height = 100 - _level * 5;
        qreal duration = time2position(child.node->block()->getEnd()) - xbegin;
        if (duration < 1)
            duration = 1;

        item->setPolygon(QRectF(0, _level * 5, duration, height));
        item->setPos(xbegin, _y);
        item->setZValue(_level);

        const auto color = child.node->block()->getColor();
        item->setBrush(QBrush(QColor(profiler::colors::get_red(color), profiler::colors::get_green(color), profiler::colors::get_blue(color))));
        item->setPen(QPen(Qt::NoPen)); // disable borders painting

        addItem(item);
        ProfGraphicsTextItem* text = new ProfGraphicsTextItem(child.node->getBlockName(), item);
        QRectF textRect = text->boundingRect();
        text->setPos(0, _level * 5);

        QRectF itemRect = item->boundingRect();
        if (textRect.width() > itemRect.width())
            text->hide();

        setTree(child.children, _y, _level + 1);
    }
}

/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

ProfGraphicsView::ProfGraphicsView() : QGraphicsView(), m_scale(100), m_scaleCoeff(1.25)
{
    initMode();
    setScene(new ProfGraphicsScene(this));
    centerOn(0, 0);
}

ProfGraphicsView::ProfGraphicsView(const thread_blocks_tree_t& _blocksTree) : QGraphicsView(), m_scale(100), m_scaleCoeff(1.25)
{
    initMode();
    setScene(new ProfGraphicsScene(_blocksTree, this));
    centerOn(0, 0);
}

ProfGraphicsView::~ProfGraphicsView()
{
}

void ProfGraphicsView::initMode()
{
    setCacheMode(QGraphicsView::CacheBackground);
    setTransformationAnchor(QGraphicsView::AnchorUnderMouse);
    setViewportUpdateMode(QGraphicsView::BoundingRectViewportUpdate);
}

void ProfGraphicsView::wheelEvent(QWheelEvent* _event)
{
    //ProfGraphicsScene* myscene = (ProfGraphicsScene*)scene();
    if (_event->delta() > 0)
    {
        scale(m_scaleCoeff, m_scaleCoeff);
        m_scale *= m_scaleCoeff;
        emit GLOBALS.scaleIncreased(m_scale);
    }
    else
    {
        const qreal scaleCoeff = 1.0 / m_scaleCoeff;
        scale(scaleCoeff, scaleCoeff);
        m_scale *= scaleCoeff;
        emit GLOBALS.scaleDecreased(m_scale);
    }
}

/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////