plot_view.cpp

/*
 * Copyright (c) 2015, 2017, 2020 Kent A. Vander Velden, kent.vandervelden@gmail.com
 *
 * This file is part of BinVis.
 *
 *     BinVis is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU General Public License as published by
 *     the Free Software Foundation, either version 3 of the License, or
 *     (at your option) any later version.
 *
 *     BinVis is distributed in the hope that it will be useful,
 *     but WITHOUT ANY WARRANTY; without even the implied warranty of
 *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *     GNU General Public License for more details.
 *
 *     You should have received a copy of the GNU General Public License
 *     along with BinVis.  If not, see <https://www.gnu.org/licenses/>.
 */

#include <algorithm>
#include <QtGui>

#include "plot_view.h"

using std::min;
using std::max;

PlotView::PlotView(QWidget *p)
        : QLabel(p),
          m1_(0.), m2_(1.), px_(-1), py_(-1), ind_(0), s_(none), allow_selection_(true) {
}

void PlotView::enableSelection(bool v) {
    allow_selection_ = v;
    update();
}

void PlotView::setImage(int ind, QImage &img) {
    img_[ind] = img;

    update_pix();

    update();
}

void PlotView::set_data(const float *dat, long len, bool normalize) {
    ind_ = 0;
    set_data(0, dat, len, normalize);
}

void PlotView::set_data(int ind, const float *dat, long len, bool normalize) {
    int w = width();
    int h = height();

    float mn = 0.;
    float mx = 1.;
    if (normalize) {
        mn = 99999999.;
        mx = -99999999.;
        for (int i = 0; i < len; i++) {
            mn = min(mn, dat[i]);
            mx = max(mx, dat[i]);
        }
        if (mn == mx) {
            mn -= .5;
            mx += .5;
        }
    }

    {
        QImage img(w, h, QImage::Format_RGB32);
        img.fill(0);

        auto acc = new float[h];
        memset(acc, 0, h * sizeof(float));
        auto cnt = new int[h];
        memset(cnt, 0, h * sizeof(int));

        for (int i = 0; i < len; i++) {
            float v = dat[i];
            int ind2 = int((i / float(len)) * (h - 1) + .5);
            acc[ind2] += (v - mn) / (mx - mn);
            cnt[ind2]++;
        }

        auto p = (unsigned int *) img.bits();
        int px = -1;
        int pc = -1;
        for (int i = 0; i < h; i++) {
            int x = px;
            int c = pc;
            if (cnt[i] == 0 && px == -1) continue;
            if (cnt[i] > 0) {
                float na = acc[i] / cnt[i];
                x = int(na * (w - 4) + .5) + 2; // slight offset so not to interfere with border
                px = x;
                c = 20 + int(na * (255 - 20));
                pc = c;
            }
            unsigned char r = 20;
            unsigned char g = min(c + 60, 255);
            unsigned char b = 20;
            unsigned int v = 0xff000000 | (r << 16) | (g << 8) | (b << 0);
            p[i * w + x] = v;
        }

        delete[] acc;
        delete[] cnt;

        setImage(ind, img);
    }
}

void PlotView::paintEvent(QPaintEvent *e) {
    QLabel::paintEvent(e);

    QPainter p(this);
    if (allow_selection_) {
        int ry1 = m1_ * height();
        int ry2 = m2_ * height();

        QBrush brush(QColor(128, 64, 64, 128 + 32));
        QPen pen(brush, 5, Qt::SolidLine, Qt::RoundCap);
        p.setPen(pen);
        p.drawLine(0 + 3, ry1, width() - 1 - 3, ry1);
        p.drawLine(0 + 3, ry2, width() - 1 - 3, ry2);
    }

    {
        // a border around the image helps to see the border of a dark image
        p.setPen(Qt::darkGray);
        p.drawRect(0, 0, width() - 1, height() - 1);
    }
}

void PlotView::resizeEvent(QResizeEvent *e) {
    QLabel::resizeEvent(e);

    update_pix();
}

void PlotView::update_pix() {
    if (img_[ind_].isNull()) return;

    int vw = width() - 4;
    int vh = height() - 4; // TODO BUG: With QDarkStyle, without the subtraction, the height or width of the application grows without bounds.
    pix_ = QPixmap::fromImage(img_[ind_]).scaled(vw, vh); //, Qt::KeepAspectRatio);
    setPixmap(pix_);
}

void PlotView::mousePressEvent(QMouseEvent *e) {
    e->accept();

    if (!allow_selection_) return;
    if (e->button() != Qt::LeftButton) return;

    int x = e->pos().x();
    int y = e->pos().y();

    if (x < 0) x = 0;
    if (x > width() - 1) x = width() - 1;
    if (y < 0) y = 0;
    if (y > height() - 1) y = height() - 1;

    float yp = y / float(height());

    if (yp > m1_ && (yp - m1_) < .01) {
        s_ = m1_moving;
    } else if (yp < m2_ && (m2_ - yp) < .01) {
        s_ = m2_moving;
    } else if (m1_ < yp && yp < m2_) {
        s_ = m12_moving;
    } else {
        s_ = none;
    }

    px_ = x;
    py_ = y;
}

void PlotView::mouseMoveEvent(QMouseEvent *e) {
    e->accept();

    if (s_ == none) return;

    int x = e->pos().x();
    int y = e->pos().y();

    if (x < 0) x = 0;
    if (x > width() - 1) x = width() - 1;
    if (y < 0) y = 0;
    if (y > height() - 1) y = height() - 1;

    if (y == py_) return;

    int h = height();

    float m1 = m1_;
    float m2 = m2_;
    if (s_ == m1_moving) {
        m1 = y / float(h);
    } else if (s_ == m2_moving) {
        m2 = y / float(h);
    } else if (s_ == m12_moving) {
        float dy = (y - py_) / float(h);
        m1 += dy;
        m2 += dy;
    }
    if (m1 >= m2_ - .01) m1 = m2_ - .01;
    if (m1 < 0.) m1 = 0.;
    if (m2 <= m1_ + .01) m2 = m1_ + .01;
    if (m2 > 1.) m2 = 1.;

    m1_ = m1;
    m2_ = m2;

    px_ = x;
    py_ = y;

    update();

    emit(rangeSelected(m1_, m2_));
}

void PlotView::mouseReleaseEvent(QMouseEvent *e) {
    e->accept();

    if (e->button() == Qt::RightButton) {
        ind_ = (ind_ + 1) % 2;
        update_pix();
        update();
    }

    if (e->button() != Qt::LeftButton) return;

//  int x = e->pos().x();
//  int y = e->pos().y();
//
//  if (x < 0) x = 0;
//  if (x > width() - 1) x = width() - 1;
//  if (y < 0) y = 0;
//  if (y > height() - 1) y = height() - 1;

    px_ = -1;
    py_ = -1;
    s_ = none;
}