#include "customplotZoom.h"

CustomPlotZoom::CustomPlotZoom(QWidget* parent)
    : QCustomPlot(parent)
    , mZoomMode(false)
    , mRubberBand(new QRubberBand(QRubberBand::Rectangle, this))
{}

CustomPlotZoom::~CustomPlotZoom()
{}
void CustomPlotZoom::setZoomMode(bool mode)
{
    mZoomMode = mode;
}
void CustomPlotZoom::mousePressEvent(QMouseEvent* event)
{
    if (mZoomMode) {
        if (event->button() == Qt::RightButton) {
            mOrigin = event->pos();
            mRubberBand->setGeometry(QRect(mOrigin, QSize()));
            mRubberBand->show();
        }
    }
    QCustomPlot::mousePressEvent(event);
}

void CustomPlotZoom::mouseMoveEvent(QMouseEvent* event)
{
    if (mRubberBand->isVisible()) {
        mRubberBand->setGeometry(QRect(mOrigin, event->pos()).normalized());
    }
    QCustomPlot::mouseMoveEvent(event);
}
void CustomPlotZoom::mouseReleaseEvent(QMouseEvent* event)
{
    if (mRubberBand->isVisible()) {
        const QRect zoomRect = mRubberBand->geometry();
        int         xp1, yp1, xp2, yp2;
        zoomRect.getCoords(&xp1, &yp1, &xp2, &yp2);
        double x1 = xAxis->pixelToCoord(xp1);
        double y1 = xAxis->pixelToCoord(yp1);
        double x2 = xAxis->pixelToCoord(xp2);
        double y2 = xAxis->pixelToCoord(yp2);

        xAxis->setRange(x1, x2);
        yAxis->setRange(y1, y2);

        mRubberBand->hide();
        replot();
    }
    QCustomPlot::mouseReleaseEvent(event);
}