//-----------------------------------------------------------------------------
// Copyright (C) 2009 Michael Gernoth <michael at gernoth.net>
//
// This code is licensed to you under the terms of the GNU GPL, version 2 or,
// at your option, any later version. See the LICENSE.txt file for the text of
// the license.
//-----------------------------------------------------------------------------
// GUI (QT)
//-----------------------------------------------------------------------------
#define __STDC_FORMAT_MACROS
#include "proxguiqt.h"
#include <inttypes.h>
#include <stdbool.h>
#include <iostream>
//#include <QtCore>
#include <QPainterPath>
#include <QBrush>
#include <QPen>
#include <QTimer>
#include <QCloseEvent>
#include <QMouseEvent>
#include <QKeyEvent>
#include <math.h>
#include <limits.h>
#include <stdio.h>
#include <QSlider>
#include <QHBoxLayout>
#include <string.h>
#include <QtGui>
#include "proxgui.h"
#include "ui.h"
#include "comms.h"
#include "graph.h"
#include "cmddata.h"
#include "util_darwin.h"
//#include "fileutils.h"

extern "C" int preferences_save(void);

static int s_Buff[MAX_GRAPH_TRACE_LEN];
static bool g_useOverlays = false;
static int g_absVMax = 0;
static uint32_t startMax; // Maximum offset in the graph (right side of graph)
static uint32_t PageWidth; // How many samples are currently visible on this 'page' / graph
static int unlockStart = 0;

void ProxGuiQT::ShowGraphWindow(void) {
    emit ShowGraphWindowSignal();
}

void ProxGuiQT::RepaintGraphWindow(void) {
    emit RepaintGraphWindowSignal();
}

void ProxGuiQT::HideGraphWindow(void) {
    emit HideGraphWindowSignal();
}

// emit picture viewer signals
void ProxGuiQT::ShowPictureWindow(char *fn) {
    emit ShowPictureWindowSignal(fn);
}

void ProxGuiQT::RepaintPictureWindow(void) {
    emit RepaintPictureWindowSignal();
}

void ProxGuiQT::HidePictureWindow(void) {
    emit HidePictureWindowSignal();
}

void ProxGuiQT::Exit(void) {
    emit ExitSignal();
}

void ProxGuiQT::_ShowGraphWindow(void) {
    if (!plotapp)
        return;

    if (!plotwidget) {

#if defined(__MACH__) && defined(__APPLE__)
        makeFocusable();
#endif

        plotwidget = new ProxWidget();
    }
    plotwidget->show();

}

void ProxGuiQT::_RepaintGraphWindow(void) {
    if (!plotapp || !plotwidget)
        return;

    plotwidget->update();
}

void ProxGuiQT::_HideGraphWindow(void) {
    if (!plotapp || !plotwidget)
        return;

    plotwidget->hide();
}

// picture viewer
void ProxGuiQT::_ShowPictureWindow(char *fn) {

    if (!plotapp)
        return;

    if (fn == NULL)
        return;

    size_t slen = strlen(fn);
    if (slen == 0)
        return;

    char *myfn = (char *)calloc(slen + 1, sizeof(uint8_t));
    if (myfn == NULL)
        return;

    memcpy(myfn, fn, slen);

    if (!pictureWidget) {

#if defined(__MACH__) && defined(__APPLE__)
        makeFocusable();
#endif

        pictureWidget = new PictureWidget();
    }

    QPixmap pm;
    if (pm.load(myfn) == false) {
        qWarning("Failed to load %s", myfn);
    }
    free(myfn);
    free(fn);

    //QPixmap newPixmap = pm.scaled(QSize(50,50),  Qt::KeepAspectRatio);
    //pm = pm.scaled(pictureController->lbl_pm->size(),  Qt::KeepAspectRatio);

    pictureController->lbl_pm->setPixmap(pm);
    pictureController->lbl_pm->setScaledContents(false);
    pictureController->lbl_pm->setAlignment(Qt::AlignCenter);

    QString s = QString("w: %1  h: %2")
                .arg(pm.size().width())
                .arg(pm.size().height()
                    );
    pictureController->lbl_sz->setText(s);
    pictureWidget->show();
}

void ProxGuiQT::_RepaintPictureWindow(void) {
    if (!plotapp || !pictureWidget)
        return;

    pictureWidget->update();
}

void ProxGuiQT::_HidePictureWindow(void) {
    if (!plotapp || !pictureWidget)
        return;

    pictureWidget->hide();
}

void ProxGuiQT::_Exit(void) {
    delete this;
}

void ProxGuiQT::_StartProxmarkThread(void) {
    if (!proxmarkThread)
        return;

    // if thread finished delete self and delete application
    QObject::connect(proxmarkThread, SIGNAL(finished()), proxmarkThread, SLOT(deleteLater()));
    QObject::connect(proxmarkThread, SIGNAL(finished()), this, SLOT(_Exit()));
    // start proxmark thread
    proxmarkThread->start();
}

void ProxGuiQT::MainLoop() {
    plotapp = new QApplication(argc, argv);

    // Setup the picture widget
    pictureWidget = new PictureWidget();
    pictureController = new Ui::PictureForm();
    pictureController->setupUi(pictureWidget);
//    pictureWidget->setAttribute(Qt::WA_DeleteOnClose,true);

    // Set picture widget position if no settings.
    if (g_session.preferences_loaded == false) {
        // Move controller widget below plot
        //pictureController->move(x(), y() + frameSize().height());
        //pictureController->resize(size().width(), 200);
    }

    connect(this, SIGNAL(ShowGraphWindowSignal()), this, SLOT(_ShowGraphWindow()));
    connect(this, SIGNAL(RepaintGraphWindowSignal()), this, SLOT(_RepaintGraphWindow()));
    connect(this, SIGNAL(HideGraphWindowSignal()), this, SLOT(_HideGraphWindow()));

    connect(this, SIGNAL(ExitSignal()), this, SLOT(_Exit()));

    // hook up picture viewer signals
    connect(this, SIGNAL(ShowPictureWindowSignal(char *)), this, SLOT(_ShowPictureWindow(char *)));
    connect(this, SIGNAL(RepaintPictureWindowSignal()), this, SLOT(_RepaintPictureWindow()));
    connect(this, SIGNAL(HidePictureWindowSignal()), this, SLOT(_HidePictureWindow()));

    //start proxmark thread after starting event loop
    QTimer::singleShot(200, this, SLOT(_StartProxmarkThread()));

#if defined(__MACH__) && defined(__APPLE__)
    //Prevent the terminal from loosing focus during launch by making the client unfocusable
    makeUnfocusable();
#endif

    plotapp->exec();
}

ProxGuiQT::ProxGuiQT(int argc, char **argv, WorkerThread *wthread) :
    plotapp(NULL), plotwidget(NULL), pictureController(NULL), pictureWidget(NULL), argc(argc), argv(argv), proxmarkThread(wthread) {

}

ProxGuiQT::~ProxGuiQT(void) {

    if (pictureController) {
        delete pictureController;
        pictureController = NULL;
    }

    if (pictureWidget) {
        pictureWidget->close();
        delete pictureWidget;
        pictureWidget = NULL;
    }

    if (plotapp) {
        plotapp->quit();
        plotapp = NULL;
    }
}

// -------------------------------------------------
// Slider Widget form based on a class to enable
// Event override functions
// -------------------------------------------------
PictureWidget::PictureWidget() {
    // Set the initail postion and size from settings
//    if (g_session.preferences_loaded)
//        setGeometry(g_session.pw.x, g_session.pw.y, g_session.pw.w, g_session.pw.h);
//    else
    resize(400, 400);
}

void PictureWidget::closeEvent(QCloseEvent *event) {
    this->hide();
    event->ignore();
}


// -------------------------------------------------
// Slider Widget form based on a class to enable
// Event override functions
// -------------------------------------------------

SliderWidget::SliderWidget() {
    // Set the initail postion and size from settings
    if (g_session.preferences_loaded)
        setGeometry(g_session.overlay.x, g_session.overlay.y, g_session.overlay.w, g_session.overlay.h);
    else
        resize(800, 400);
}

void SliderWidget::resizeEvent(QResizeEvent *event) {
    g_session.overlay.h = event->size().height();
    g_session.overlay.w = event->size().width();
    g_session.window_changed = true;

}

void SliderWidget::moveEvent(QMoveEvent *event) {
    g_session.overlay.x = event->pos().x();
    g_session.overlay.y = event->pos().y();
    g_session.window_changed = true;
}

//--------------------
void ProxWidget::applyOperation() {
    //printf("ApplyOperation()");
    save_restoreGB(GRAPH_SAVE);
    memcpy(g_GraphBuffer, s_Buff, sizeof(int) * g_GraphTraceLen);
    RepaintGraphWindow();
}
void ProxWidget::stickOperation() {
    save_restoreGB(GRAPH_RESTORE);
    //printf("stickOperation()");
}
void ProxWidget::vchange_autocorr(int v) {
    int ans = AutoCorrelate(g_GraphBuffer, s_Buff, g_GraphTraceLen, v, true, false);
    if (g_debugMode) printf("vchange_autocorr(w:%d): %d\n", v, ans);
    g_useOverlays = true;
    RepaintGraphWindow();
}
void ProxWidget::vchange_askedge(int v) {
    //extern int AskEdgeDetect(const int *in, int *out, int len, int threshold);
    int ans = AskEdgeDetect(g_GraphBuffer, s_Buff, g_GraphTraceLen, v);
    if (g_debugMode) printf("vchange_askedge(w:%d)%d\n", v, ans);
    g_useOverlays = true;
    RepaintGraphWindow();
}
void ProxWidget::vchange_dthr_up(int v) {
    int down = opsController->horizontalSlider_dirthr_down->value();
    directionalThreshold(g_GraphBuffer, s_Buff, g_GraphTraceLen, v, down);
    //printf("vchange_dthr_up(%d)", v);
    g_useOverlays = true;
    RepaintGraphWindow();
}
void ProxWidget::vchange_dthr_down(int v) {
    //printf("vchange_dthr_down(%d)", v);
    int up = opsController->horizontalSlider_dirthr_up->value();
    directionalThreshold(g_GraphBuffer, s_Buff, g_GraphTraceLen, v, up);
    g_useOverlays = true;
    RepaintGraphWindow();
}


ProxWidget::ProxWidget(QWidget *parent, ProxGuiQT *master) : QWidget(parent) {
    this->master = master;
    // Set the initail postion and size from settings
    if (g_session.preferences_loaded)
        setGeometry(g_session.plot.x, g_session.plot.y, g_session.plot.w, g_session.plot.h);
    else
        resize(800, 400);

    // Setup the controller widget
    controlWidget = new SliderWidget();
    opsController = new Ui::Form();
    opsController->setupUi(controlWidget);
    //Due to quirks in QT Designer, we need to fiddle a bit
    opsController->horizontalSlider_dirthr_down->setMinimum(-128);
    opsController->horizontalSlider_dirthr_down->setMaximum(0);
    opsController->horizontalSlider_dirthr_down->setValue(-20);
    opsController->horizontalSlider_dirthr_up->setMinimum(-40);
    opsController->horizontalSlider_dirthr_up->setMaximum(128);
    opsController->horizontalSlider_dirthr_up->setValue(20);
    opsController->horizontalSlider_askedge->setValue(25);
    opsController->horizontalSlider_window->setValue(4000);

    QObject::connect(opsController->pushButton_apply, SIGNAL(clicked()), this, SLOT(applyOperation()));
    QObject::connect(opsController->pushButton_sticky, SIGNAL(clicked()), this, SLOT(stickOperation()));
    QObject::connect(opsController->horizontalSlider_window, SIGNAL(valueChanged(int)), this, SLOT(vchange_autocorr(int)));
    QObject::connect(opsController->horizontalSlider_dirthr_up, SIGNAL(valueChanged(int)), this, SLOT(vchange_dthr_up(int)));
    QObject::connect(opsController->horizontalSlider_dirthr_down, SIGNAL(valueChanged(int)), this, SLOT(vchange_dthr_down(int)));
    QObject::connect(opsController->horizontalSlider_askedge, SIGNAL(valueChanged(int)), this, SLOT(vchange_askedge(int)));

    controlWidget->setGeometry(g_session.overlay.x, g_session.overlay.y, g_session.overlay.w, g_session.overlay.h);

    // Set up the plot widget, which does the actual plotting
    plot = new Plot(this);
    QVBoxLayout *layout = new QVBoxLayout;
    layout->addWidget(plot);
    setLayout(layout);

    // plot window title
    QString pt = QString("[*]Plot [ %1 ]").arg(g_conn.serial_port_name);
    setWindowTitle(pt);

    // shows plot window on the screen.
    show();

    // Set Slider/Overlay position if no settings.
    if (g_session.preferences_loaded == false) {
        // Move controller widget below plot
        controlWidget->move(x(), y() + frameSize().height());
        controlWidget->resize(size().width(), 200);
    }

    // Olverlays / slider window title
    QString ct = QString("[*]Slider [ %1 ]").arg(g_conn.serial_port_name);
    controlWidget->setWindowTitle(ct);

    // The hide/show event functions should take care of this.
    //  controlWidget->show();

    // now that is up, reset pos/size change flags
    g_session.window_changed = false;

}

// not 100% sure what i need in this block
// feel free to fix - marshmellow...
ProxWidget::~ProxWidget(void) {
    if (controlWidget) {
        controlWidget->close();
        delete controlWidget;
        controlWidget = NULL;
    }

    if (opsController) {
        delete opsController;
        opsController = NULL;
    }

    if (plot) {
        plot->close();
        delete plot;
        plot = NULL;
    }
}

void ProxWidget::closeEvent(QCloseEvent *event) {
    event->ignore();
    this->hide();
    g_useOverlays = false;
}
void ProxWidget::hideEvent(QHideEvent *event) {
    controlWidget->hide();
    plot->hide();
}
void ProxWidget::showEvent(QShowEvent *event) {
    if (g_session.overlay_sliders)
        controlWidget->show();
    else
        controlWidget->hide();

    plot->show();
}
void ProxWidget::moveEvent(QMoveEvent *event) {
    g_session.plot.x = event->pos().x();
    g_session.plot.y = event->pos().y();
    g_session.window_changed = true;
}
void ProxWidget::resizeEvent(QResizeEvent *event) {
    g_session.plot.h = event->size().height();
    g_session.plot.w = event->size().width();
    g_session.window_changed = true;
}

//----------- Plotting

int Plot::xCoordOf(int i, QRect r) {
    return r.left() + (int)((i - g_GraphStart) * g_GraphPixelsPerPoint);
}

int Plot::yCoordOf(int v, QRect r, int maxVal) {
    int z = (r.bottom() - r.top()) / 2;
    if (maxVal == 0) ++maxVal;
    return -(z * v) / maxVal + z;
}

int Plot::valueOf_yCoord(int y, QRect r, int maxVal) {
    int z = (r.bottom() - r.top()) / 2;
    return (y - z) * maxVal / z;
}

static const QColor BLACK     = QColor(0, 0, 0);
static const QColor GRAY60    = QColor(60, 60, 60);
static const QColor GRAY100   = QColor(100, 100, 100);
static const QColor GRAY240   = QColor(240, 240, 240);
static const QColor WHITE     = QColor(255, 255, 255);
static const QColor GREEN     = QColor(100, 255, 100);
static const QColor RED       = QColor(255, 100, 100);
static const QColor BLUE      = QColor(100, 100, 255);
static const QColor YELLOW    = QColor(255, 255, 0);
static const QColor PINK      = QColor(255, 0, 255);
static const QColor ORANGE    = QColor(255, 153, 0);
static const QColor LIGHTBLUE = QColor(100, 209, 246);

QColor Plot::getColor(int graphNum) {
    switch (graphNum) {
        case 0:
            return GREEN;
        case 1:
            return RED;
        case 2:
            return BLUE;
        default:
            return GRAY240;
    }
}

void Plot::setMaxAndStart(int *buffer, size_t len, QRect plotRect) {
    if (len == 0) return;
    startMax = 0;
    if (plotRect.right() >= plotRect.left() + 40) {
        uint32_t t = (plotRect.right() - plotRect.left() - 40) / g_GraphPixelsPerPoint;
        if (len >= t)
            startMax = len - t;
    }
    if (g_GraphStart > startMax) {
        g_GraphStart = startMax;
    }
    if (g_GraphStart > len) return;
    int vMin = INT_MAX, vMax = INT_MIN;
    uint32_t sample_index = g_GraphStart ;
    for (; sample_index < len && xCoordOf(sample_index, plotRect) < plotRect.right() ; sample_index++) {

        int v = buffer[sample_index];
        if (v < vMin) vMin = v;
        if (v > vMax) vMax = v;
    }

    g_absVMax = 0;
    if (fabs((double) vMin) > g_absVMax) g_absVMax = (int)fabs((double) vMin);
    if (fabs((double) vMax) > g_absVMax) g_absVMax = (int)fabs((double) vMax);
    g_absVMax = (int)(g_absVMax * 1.25 + 1);
}

void Plot::PlotDemod(uint8_t *buffer, size_t len, QRect plotRect, QRect annotationRect, QPainter *painter, int graphNum, uint32_t plotOffset) {
    if (len == 0 || g_PlotGridX <= 0) return;
    //clock_t begin = clock();
    QPainterPath penPath;

    int grid_delta_x = g_PlotGridX;
    int first_delta_x = grid_delta_x; //(plotOffset > 0) ? g_PlotGridX : (g_PlotGridX +);
    if (g_GraphStart > plotOffset) first_delta_x -= (g_GraphStart - plotOffset);
    int DemodStart = g_GraphStart;
    if (plotOffset > g_GraphStart) DemodStart = plotOffset;

    int BitStart = 0;
    // round down
    if (DemodStart - plotOffset > 0) BitStart = (int)(((DemodStart - plotOffset) + (g_PlotGridX - 1)) / g_PlotGridX) - 1;
    first_delta_x += BitStart * g_PlotGridX;
    if (BitStart > (int)len) return;
    int delta_x = 0;
//    int v = 0;
    //printf("first_delta_x %i, grid_delta_x %i, DemodStart %i, BitStart %i\n",first_delta_x,grid_delta_x,DemodStart, BitStart);

    painter->setPen(getColor(graphNum));
    char str[5];
    int absVMax = (int)(100 * 1.05 + 1);
    delta_x = 0;
    int clk = first_delta_x;
    for (int i = BitStart; i < (int)len && xCoordOf(delta_x + DemodStart, plotRect) < plotRect.right(); i++) {
        for (int j = 0; j < (clk) && i < (int)len && xCoordOf(DemodStart + delta_x + j, plotRect) < plotRect.right() ; j++) {
            int x = xCoordOf(DemodStart + delta_x + j, plotRect);
            int v = buffer[i] * 200 - 100;

            int y = yCoordOf(v, plotRect, absVMax);
            if ((i == BitStart) && (j == 0)) { // First point
                penPath.moveTo(x, y);
            } else {
                penPath.lineTo(x, y);
            }
            if (g_GraphPixelsPerPoint > 10) {
                QRect f(QPoint(x - 3, y - 3), QPoint(x + 3, y + 3));
                painter->fillRect(f, getColor(graphNum));
            }
            if (j == (int)clk / 2) {
                //print label
                sprintf(str, "%u", buffer[i]);
                painter->drawText(x - 8, y + ((buffer[i] > 0) ? 18 : -6), str);
            }
        }
        delta_x += clk;
        clk = grid_delta_x;
    }

    // Graph annotations
    painter->drawPath(penPath);
}

void Plot::PlotGraph(int *buffer, size_t len, QRect plotRect, QRect annotationRect, QPainter *painter, int graphNum) {
    if (len == 0) return;
    // clock_t begin = clock();
    QPainterPath penPath;
    int vMin = INT_MAX, vMax = INT_MIN, v = 0;
    int64_t vMean = 0;
    uint32_t i = 0;
    int x = xCoordOf(g_GraphStart, plotRect);
    int y = yCoordOf(buffer[g_GraphStart], plotRect, g_absVMax);
    penPath.moveTo(x, y);
    for (i = g_GraphStart; i < len && xCoordOf(i, plotRect) < plotRect.right(); i++) {

        x = xCoordOf(i, plotRect);
        v = buffer[i];

        y = yCoordOf(v, plotRect, g_absVMax);

        penPath.lineTo(x, y);

        if (g_GraphPixelsPerPoint > 10) {
            QRect f(QPoint(x - 3, y - 3), QPoint(x + 3, y + 3));
            painter->fillRect(f, GREEN);
        }
        // catch stats
        if (v < vMin) vMin = v;
        if (v > vMax) vMax = v;
        vMean += v;
    }
    g_GraphStop = i;
    vMean /= (g_GraphStop - g_GraphStart);

    painter->setPen(getColor(graphNum));

    // Draw y-axis
    int xo = 5 + (graphNum * 40);
    painter->drawLine(xo, plotRect.top(), xo, plotRect.bottom());

    int vMarkers = (g_absVMax - (g_absVMax % 10)) / 5;
    int minYDist = 40; // Minimum pixel-distance between markers

    char yLbl[20];

    int n = 0;
    int lasty0 = 65535;

    for (v = vMarkers; yCoordOf(v, plotRect, g_absVMax) > plotRect.top() && n < 20; v += vMarkers, n++) {
        int y0 = yCoordOf(v, plotRect, g_absVMax);
        int y1 = yCoordOf(-v, plotRect, g_absVMax);

        if (lasty0 - y0 < minYDist) continue;

        painter->drawLine(xo - 5, y0, xo + 5, y0);

        sprintf(yLbl, "%d", v);
        painter->drawText(xo + 8, y0 + 7, yLbl);

        painter->drawLine(xo - 5, y1, xo + 5, y1);
        sprintf(yLbl, "%d", -v);
        painter->drawText(xo + 8, y1 + 5, yLbl);
        lasty0 = y0;
    }

    //Graph annotations
    painter->drawPath(penPath);
    char str[200];
    sprintf(str, "max=%d  min=%d  mean=%" PRId64 "  n=%u/%zu  CursorAVal=[%d]  CursorBVal=[%d]",
            vMax, vMin, vMean, g_GraphStop - g_GraphStart, len, buffer[CursorAPos], buffer[CursorBPos]);
    painter->drawText(20, annotationRect.bottom() - 23 - 20 * graphNum, str);
    //clock_t end = clock();
    //double elapsed_secs = double(end - begin) / CLOCKS_PER_SEC;
    //printf("Plot time %f\n", elapsed_secs);
}

void Plot::plotGridLines(QPainter *painter, QRect r) {

    // set g_GridOffset
    if (g_PlotGridX <= 0) return;

    double offset = g_GridOffset;
    if (g_GridLocked && g_PlotGridX) {
        offset = g_GridOffset + g_PlotGridX - fmod(g_GraphStart, g_PlotGridX);
    } else if (!g_GridLocked && g_GraphStart > 0 && g_PlotGridX) {
        offset = g_PlotGridX - fmod(g_GraphStart - offset, g_PlotGridX) + g_GraphStart - unlockStart;
    }
    offset = fmod(offset, g_PlotGridX);
    if (offset < 0) offset += g_PlotGridX;

    double i;
    double grid_delta_x = g_PlotGridX * g_GraphPixelsPerPoint;
    int grid_delta_y = g_PlotGridY;

    if ((g_PlotGridX > 0) && ((g_PlotGridX * g_GraphPixelsPerPoint) > 1)) {
        for (i = (offset * g_GraphPixelsPerPoint); i < r.right(); i += grid_delta_x) {
            painter->drawLine(r.left() + i, r.top(), r.left() + i, r.bottom());
        }
    }

    if (g_PlotGridY > 0) {
        for (i = 0; yCoordOf(i, r, g_absVMax) > r.top(); i += grid_delta_y) {
            // line above mid
            painter->drawLine(r.left(), yCoordOf(i, r, g_absVMax), r.right(), yCoordOf(i, r, g_absVMax));
            // line below mid
            painter->drawLine(r.left(), yCoordOf(-i, r, g_absVMax), r.right(), yCoordOf(-i, r, g_absVMax));
        }
    }
}

#define HEIGHT_INFO 60
#define WIDTH_AXES 80

void Plot::paintEvent(QPaintEvent *event) {
    QPainter painter(this);
    QBrush brush(GREEN);
    QPen pen(GREEN);

    painter.setFont(QFont("Courier New", 10));

    if (CursorAPos > g_GraphTraceLen)
        CursorAPos = 0;
    if (CursorBPos > g_GraphTraceLen)
        CursorBPos = 0;
    if (g_CursorCPos > g_GraphTraceLen)
        g_CursorCPos = 0;
    if (g_CursorDPos > g_GraphTraceLen)
        g_CursorDPos = 0;

    QRect plotRect(WIDTH_AXES, 0, width() - WIDTH_AXES, height() - HEIGHT_INFO);
    QRect infoRect(0, height() - HEIGHT_INFO, width(), HEIGHT_INFO);
    PageWidth = plotRect.width() / g_GraphPixelsPerPoint;

    //Grey background
    painter.fillRect(rect(), GRAY60);
    //Black foreground
    painter.fillRect(plotRect, BLACK);

    //init graph variables
    setMaxAndStart(g_GraphBuffer, g_GraphTraceLen, plotRect);

    // center line
    int zeroHeight = plotRect.top() + (plotRect.bottom() - plotRect.top()) / 2;
    painter.setPen(GRAY100);
    painter.drawLine(plotRect.left(), zeroHeight, plotRect.right(), zeroHeight);
    // plot X and Y grid lines
    plotGridLines(&painter, plotRect);

    //Start painting graph
    PlotGraph(g_GraphBuffer, g_GraphTraceLen, plotRect, infoRect, &painter, 0);
    if (g_DemodBufferLen > 8) {
        PlotDemod(g_DemodBuffer, g_DemodBufferLen, plotRect, infoRect, &painter, 2, g_DemodStartIdx);
    }
    if (g_useOverlays) {
        //init graph variables
        setMaxAndStart(s_Buff, g_GraphTraceLen, plotRect);
        PlotGraph(s_Buff, g_GraphTraceLen, plotRect, infoRect, &painter, 1);
    }
    // End graph drawing

    //Draw the cursors
    if (CursorAPos > g_GraphStart && xCoordOf(CursorAPos, plotRect) < plotRect.right()) {
        painter.setPen(YELLOW);
        painter.drawLine(xCoordOf(CursorAPos, plotRect), plotRect.top(), xCoordOf(CursorAPos, plotRect), plotRect.bottom());
    }
    if (CursorBPos > g_GraphStart && xCoordOf(CursorBPos, plotRect) < plotRect.right()) {
        painter.setPen(PINK);
        painter.drawLine(xCoordOf(CursorBPos, plotRect), plotRect.top(), xCoordOf(CursorBPos, plotRect), plotRect.bottom());
    }
    if (g_CursorCPos > g_GraphStart && xCoordOf(g_CursorCPos, plotRect) < plotRect.right()) {
        painter.setPen(ORANGE);
        painter.drawLine(xCoordOf(g_CursorCPos, plotRect), plotRect.top(), xCoordOf(g_CursorCPos, plotRect), plotRect.bottom());
    }
    if (g_CursorDPos > g_GraphStart && xCoordOf(g_CursorDPos, plotRect) < plotRect.right()) {
        painter.setPen(LIGHTBLUE);
        painter.drawLine(xCoordOf(g_CursorDPos, plotRect), plotRect.top(), xCoordOf(g_CursorDPos, plotRect), plotRect.bottom());
    }

    //Draw annotations
    char str[200];
    char scalestr[30] = {0};
    if (g_CursorScaleFactor != 1) {
        if (g_CursorScaleFactorUnit[0] == '\x00') {
            sprintf(scalestr, "[%2.2f] ", ((int32_t)(CursorBPos - CursorAPos)) / g_CursorScaleFactor);
        } else {
            sprintf(scalestr, "[%2.2f %s] ", ((int32_t)(CursorBPos - CursorAPos)) / g_CursorScaleFactor, g_CursorScaleFactorUnit);
        }
    }
    sprintf(str, "@%u..%u  dt=%i %szoom=%2.2f  CursorAPos=%u  CursorBPos=%u  GridX=%lf  GridY=%lf (%s) GridXoffset=%lf",
            g_GraphStart,
            g_GraphStop,
            CursorBPos - CursorAPos,
            scalestr,
            g_GraphPixelsPerPoint,
            CursorAPos,
            CursorBPos,
            g_PlotGridXdefault,
            g_PlotGridYdefault,
            g_GridLocked ? "Locked" : "Unlocked",
            g_GridOffset
           );
    painter.setPen(WHITE);
    painter.drawText(20, infoRect.bottom() - 3, str);
}

Plot::Plot(QWidget *parent) : QWidget(parent), g_GraphPixelsPerPoint(1) {
    //Need to set this, otherwise we don't receive keypress events
    setFocusPolicy(Qt::StrongFocus);
    resize(400, 200);

    QPalette palette(QColor(0, 0, 0, 0));
    palette.setColor(QPalette::WindowText, WHITE);
    palette.setColor(QPalette::Text, WHITE);
    palette.setColor(QPalette::Button, GRAY100);
    setPalette(palette);
    setAutoFillBackground(true);

    CursorAPos = 0;
    CursorBPos = 0;
    g_GraphStart = 0;
    g_GraphStop = 0;

    setWindowTitle(tr("Sliders"));
    master = parent;
}

void Plot::closeEvent(QCloseEvent *event) {
    event->ignore();
    this->hide();
    g_useOverlays = false;
}

void Plot::Zoom(double factor, uint32_t refX) {
    if (factor >= 1) { // Zoom in
        if (g_GraphPixelsPerPoint <= 25 * factor) {
            g_GraphPixelsPerPoint *= factor;
            if (refX > g_GraphStart) {
                g_GraphStart += (refX - g_GraphStart) - ((refX - g_GraphStart) / factor);
            }
        }
    } else {          // Zoom out
        if (g_GraphPixelsPerPoint >= 0.01 / factor) {
            g_GraphPixelsPerPoint *= factor;
            if (refX > g_GraphStart) {
                if (g_GraphStart >= ((refX - g_GraphStart) / factor) - (refX - g_GraphStart)) {
                    g_GraphStart -= ((refX - g_GraphStart) / factor) - (refX - g_GraphStart);
                } else {
                    g_GraphStart = 0;
                }
            }
        }
    }
}

void Plot::Move(int offset) {
    if (g_GraphTraceLen == 0) return;
    if (offset > 0) { // Move right
        if (g_GraphPixelsPerPoint < 20) {
            g_GraphStart += offset;
        } else {
            g_GraphStart++;
        }
    } else { // Move left
        if (g_GraphPixelsPerPoint < 20) {
            if (g_GraphStart >= (uint) - offset) {
                g_GraphStart += offset;
            } else {
                g_GraphStart = 0;
            }
        } else {
            if (g_GraphStart > 0) {
                g_GraphStart--;
            }
        }
    }
}

void Plot::Trim(void) {
    uint32_t lref, rref;
    const double zoom_offset = 1.148698354997035; // 2**(1/5)
    if ((CursorAPos == 0) || (CursorBPos == 0)) { // if we don't have both cursors set
        lref = g_GraphStart;
        rref = g_GraphStop;
        if (CursorAPos >= lref) {
            CursorAPos -= lref;
        } else {
            CursorAPos = 0;
        }
        if (CursorBPos >= lref) {
            CursorBPos -= lref;
        } else {
            CursorBPos = 0;
        }
    } else {
        lref = CursorAPos < CursorBPos ? CursorAPos : CursorBPos;
        rref = CursorAPos < CursorBPos ? CursorBPos : CursorAPos;
        // g_GraphPixelsPerPoint mush remain a power of zoom_offset
        double GPPPtarget = g_GraphPixelsPerPoint * (g_GraphStop - g_GraphStart) / (rref - lref);
        while (g_GraphPixelsPerPoint < GPPPtarget) {
            g_GraphPixelsPerPoint *= zoom_offset;
        }
        g_GraphPixelsPerPoint /= zoom_offset;
        CursorAPos -= lref;
        CursorBPos -= lref;
    }
    g_DemodStartIdx -= lref;
    for (uint32_t i = lref; i < rref; ++i)
        g_GraphBuffer[i - lref] = g_GraphBuffer[i];
    g_GraphTraceLen = rref - lref;
    g_GraphStart = 0;
}

void Plot::wheelEvent(QWheelEvent *event) {
    // event->delta()
    //  120 => shift right 5%
    // -120 => shift left 5%
    const float move_offset = 0.05;
    // -120+shift => zoom in  (5 times = *2)
    //  120+shift => zoom out (5 times = /2)
    const double zoom_offset = 1.148698354997035; // 2**(1/5)
    if (event->modifiers() & Qt::ShiftModifier) {
// event->position doesn't exist in QT5.12.8, both exist in 5.14.2 and event->x doesn't exist in 5.15.0
#if QT_VERSION >= 0x050d00
        uint32_t x = event->position().x();
#else
        uint32_t x = event->x();
#endif
        x -= WIDTH_AXES;
        x = (int)(x / g_GraphPixelsPerPoint);
        x += g_GraphStart;
// event->angleDelta doesn't exist in QT4, both exist in 5.12.8 and 5.14.2 and event->delta doesn't exist in 5.15.0
#if QT_VERSION >= 0x050d00
        float delta = event->angleDelta().y();
#else
        float delta = event->delta();
#endif
        if (delta < 0) {
            Zoom(zoom_offset, x);
        } else {
            Zoom(1.0 / zoom_offset, x);
        }
    } else {
#if QT_VERSION >= 0x050d00
        Move(PageWidth * (-(float)event->angleDelta().y() / (120 / move_offset)));
#else
        Move(PageWidth * (-(float)event->delta() / (120 / move_offset)));
#endif
    }
    this->update();
}

void Plot::mouseMoveEvent(QMouseEvent *event) {
    int x = event->x();
    x -= WIDTH_AXES;
    x = (int)(x / g_GraphPixelsPerPoint);
    x += g_GraphStart;
    if ((event->buttons() & Qt::LeftButton)) {
        CursorAPos = x;
    } else if (event->buttons() & Qt::RightButton) {
        CursorBPos = x;
    }
    this->update();
}

void Plot::keyPressEvent(QKeyEvent *event) {
    uint32_t offset; // Left/right movement offset (in sample size)
    const double zoom_offset = 1.148698354997035; // 2**(1/5)

    if (event->modifiers() & Qt::ShiftModifier) {
        if (g_PlotGridX)
            offset = PageWidth - fmod(PageWidth, g_PlotGridX);
        else
            offset = PageWidth;
    } else {
        if (event->modifiers() & Qt::ControlModifier)
            offset = 1;
        else
            offset = (int)(20 / g_GraphPixelsPerPoint);
    }

    switch (event->key()) {
        case Qt::Key_Down:
            if (event->modifiers() & Qt::ShiftModifier) {
                if (event->modifiers() & Qt::ControlModifier) {
                    Zoom(zoom_offset, CursorBPos);
                } else {
                    Zoom(2, CursorBPos);
                }
            } else {
                if (event->modifiers() & Qt::ControlModifier) {
                    Zoom(zoom_offset, CursorAPos);
                } else {
                    Zoom(2, CursorAPos);
                }
            }
            break;

        case Qt::Key_Up:
            if (event->modifiers() & Qt::ShiftModifier) {
                if (event->modifiers() & Qt::ControlModifier) {
                    Zoom(1.0 / zoom_offset, CursorBPos);
                } else {
                    Zoom(0.5, CursorBPos);
                }
            } else {
                if (event->modifiers() & Qt::ControlModifier) {
                    Zoom(1.0 / zoom_offset, CursorAPos);
                } else {
                    Zoom(0.5, CursorAPos);
                }
            }
            break;

        case Qt::Key_Right:
            Move(offset);
            break;

        case Qt::Key_Left:
            Move(-offset);
            break;

        case Qt::Key_Greater:
            g_DemodStartIdx += 1;
            break;

        case Qt::Key_Less:
            g_DemodStartIdx -= 1;
            break;

        case Qt::Key_G:
            if (g_PlotGridX || g_PlotGridY) {
                g_PlotGridX = 0;
                g_PlotGridY = 0;
            } else {
                if (g_PlotGridXdefault < 0)
                    g_PlotGridXdefault = 64;
                if (g_PlotGridYdefault < 0)
                    g_PlotGridYdefault = 0;

                g_PlotGridX = g_PlotGridXdefault;
                g_PlotGridY = g_PlotGridYdefault;
            }
            break;

        case Qt::Key_H:
            g_printAndLog = PRINTANDLOG_PRINT;
            PrintAndLogEx(NORMAL, "\n\n" _CYAN_("PLOT window keystrokes and mouse events"));
            PrintAndLogEx(NORMAL, "\n" _GREEN_("Move:"));
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9 + 9, _RED_("Home") "/" _RED_("End"), "Move to the start/end of the graph");
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9, _YELLOW_("Mouse wheel"), "Move left/right");
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9 + 9, _RED_("Left") "/" _RED_("Right"), "Move left/right");
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9, " + " _RED_("Ctrl"), "... by 1 sample");
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9, " + " _RED_("Shift"), "... by 1 window");
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9 + 9, _RED_("PgUp") "/" _RED_("PgDown"), "Move left/right by 1 window");
            PrintAndLogEx(NORMAL, "\n" _GREEN_("Zoom:"));
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9 + 9, _RED_("Shift") " + " _YELLOW_("Mouse wheel"), "Zoom in/out around mouse cursor");
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9 + 9, _RED_("Down") "/" _RED_("Up"), "Zoom in/out around yellow cursor");
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9, " + " _RED_("Ctrl"), "... with smaller increment");
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9, " + " _RED_("Shift"), "... around purple cursor");
            PrintAndLogEx(NORMAL, "\n" _GREEN_("Trim:"));
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9, _RED_("t"), "Trim data on window or on cursors if defined");
            PrintAndLogEx(NORMAL, "\n" _GREEN_("Grid and demod:"));
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9, _RED_("g"), "Toggle grid and demodulation plot display");
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9, _RED_("l"), "Toggle lock grid relative to samples");
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9 + 9, _RED_("<") "/" _RED_(">"), "Move demodulation left/right relative to samples");
            PrintAndLogEx(NORMAL, "\n" _GREEN_("Misc:"));
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9, _YELLOW_("Left mouse click"), "Set yellow cursor");
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9, _YELLOW_("Right mouse click"), "Set purple cursor");
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9, _RED_("h"), "Show this help");
            PrintAndLogEx(NORMAL, "    %-*s%s", 25 + 9, _RED_("q"), "Close plot window");
            g_printAndLog = PRINTANDLOG_PRINT | PRINTANDLOG_LOG;
            break;

        case Qt::Key_L:
            g_GridLocked = !g_GridLocked;
            if (g_GridLocked)
                g_GridOffset += (g_GraphStart - unlockStart);
            else
                unlockStart = g_GraphStart;
            break;

        case Qt::Key_Q:
            master->hide();
            break;

        case Qt::Key_T:
            Trim();
            break;

        case Qt::Key_Home:
            g_GraphStart = 0;
            break;

        case Qt::Key_End:
            g_GraphStart = startMax;
            break;

        case Qt::Key_PageUp:
            if (g_GraphStart >= PageWidth) {
                g_GraphStart -= PageWidth;
            } else {
                g_GraphStart = 0;
            }
            break;

        case Qt::Key_PageDown:
            g_GraphStart += PageWidth;
            if (g_GraphStart > startMax)
                g_GraphStart = startMax;
            break;

        default:
            QWidget::keyPressEvent(event);
            return;
            break;
    }

    this->update();
}