timeline.cpp

Go to the documentation of this file.

#include "gui/timeline.h"
#include "keyframe_management/keyframe_manager.h"
#include <QComboBox>
#include <QHBoxLayout>
#include <QMessageBox>
#include <QMouseEvent>
#include <QPaintEvent>
#include <QPainter>
 
namespace Gui
{
 
    Timeline::Timeline(QWidget* parent)
        : QWidget(parent), m_timeStamp(0.0f), m_activeAgentId(1),
          m_pixelsPerSecond(11.75)
    {
        setObjectName("Timeline");
        setMinimumHeight(100);
 
        QObject::connect(&KeyframeManagement::KeyframeManager::Instance(),
                         SIGNAL(KeyframeAdded()), this, SLOT(update()));
    }
 
    void
    Timeline::paintEvent(QPaintEvent* event)
    {
        QPainter painter(this);
        painter.setBrush(Qt::black);
        painter.setPen(Qt::black);
        painter.drawRect(0, 0, width(), height());
        int numDivisions = 20;
        float increment = width() / (float)numDivisions;
        for (int i = 0; i <= numDivisions; ++i) {
            float xPos = i * increment;
            painter.setPen(Qt::lightGray);
            painter.drawLine(xPos, 0, xPos, height());
            QString timestampText = QString::number(i * 5);
            painter.setPen(Qt::lightGray);
            painter.drawText(QPointF(xPos + 2, height() - 5), timestampText);
        }
        float xPos = (m_timeStamp / 100.0f) * width();
        m_pixelsPerSecond = width() / 100.0f;
        painter.setBrush(Qt::red);
        painter.setPen(Qt::red);
        painter.drawLine(xPos, 0, xPos, height());
 
        int squareSize = 10;
 
        const std::vector<Core::Keyframe>& keyframes =
            KeyframeManagement::KeyframeManager::Instance().GetKeyframes();
        float secondsPerPixel = 1.0f / m_pixelsPerSecond;
        for (const Core::Keyframe& kf : keyframes) {
            float timeStamp = kf.TimeStamp;
 
            int x = static_cast<int>((timeStamp / secondsPerPixel) -
                                     (squareSize / 2));
            int y = height() / 2 - squareSize / 2;
 
            painter.setPen(Qt::NoPen);
            painter.setBrush(Qt::red);
            painter.drawRect(x, y, squareSize, squareSize);
        }
    }
 
    void
    Timeline::mouseReleaseEvent(QMouseEvent* event)
    {
        float xPosition = event->position().x();
        float timeStamp = (xPosition / width()) * 100.0f;
 
        bool keyframeClicked = false;
 
        if (event->button() == Qt::RightButton) {
            int squareSize = 10;
            float secondsPerPixel = 1.0f / m_pixelsPerSecond;
 
            const std::vector<Core::Keyframe>& keyframes =
                KeyframeManagement::KeyframeManager::Instance().GetKeyframes();
 
            for (size_t i = 0; i < keyframes.size(); ++i) {
                const Core::Keyframe& kf = keyframes[i];
                float kfTimeStamp = kf.TimeStamp;
 
                int x = static_cast<int>((kfTimeStamp / secondsPerPixel) -
                                         (squareSize / 2));
                int y = height() / 2 - squareSize / 2;
 
                QRect keyframeRect(x, y, squareSize, squareSize);
 
                if (keyframeRect.contains(event->pos())) {
                    QMessageBox::StandardButton reply = QMessageBox::question(
                        this, "Delete keyframe",
                        "Do you want to delete this keyframe?",
                        QMessageBox::Yes | QMessageBox::No);
 
                    if (reply == QMessageBox::Yes) {
                        KeyframeManagement::KeyframeManager::Instance()
                            .RemoveKeyframe(kf);
                        update();
                    }
 
                    keyframeClicked = true;
                    break;
                }
            }
        }
 
        if (!keyframeClicked) {
            m_timeStamp = timeStamp;
            update();
            emit timeStampSelected(m_timeStamp);
        }
    }
 
    void
    Timeline::resizeEvent(QResizeEvent* event)
    {
        update();
        QWidget::resizeEvent(event);
    }
 
} // namespace Gui