SimCenterGraphPlot.cpp

#include "SimCenterGraphPlot.h"
#include <QLineEdit>
#include <QLabel>
#include <QHBoxLayout>
#include <QVBoxLayout>
#include <QSpinBox>

#include <qcustomplot.h>


SimCenterGraphPlot::SimCenterGraphPlot(
                               QString xLabel,
                               QString yLabel,
                               QWidget *parent)
    : QWidget(parent), numGraphs(0)
{
    yMinValue = 0;
    yMaxValue = 0;
    xMinValue = 0;
    xMaxValue = 0;
    // create a main layout
    QVBoxLayout *mainLayout = new QVBoxLayout();

    //
    // graphic window
    //

    thePlot=new QCustomPlot();
    thePlot->setSizePolicy(QSizePolicy::Preferred, QSizePolicy::Minimum);
    thePlot->setInteractions(QCP::iSelectPlottables);

    QRect rec = QGuiApplication::screens()[0]->geometry();

    int height = 0.2*rec.height();
    int width = 0.5*rec.width();

    thePlot->setMinimumWidth(width);
    thePlot->setMinimumHeight(height);
    mainLayout->addWidget(thePlot);

    thePlot->xAxis->setLabel(xLabel);
    thePlot->yAxis->setLabel(yLabel);
   // thePlot->xAxis->setScaleType(QCPAxis::stLogarithmic);

    this->setLayout(mainLayout);
}

SimCenterGraphPlot::SimCenterGraphPlot(
                               QString xLabel,
                               QString yLabel,
                               int minWidth,
                               int minHeight,
                               QWidget *parent)
    : QWidget(parent), numGraphs(0)
{
    yMinValue = 0;
    yMaxValue = 0;
    xMinValue = 0;
    xMaxValue = 0;
    // create a main layout
    QVBoxLayout *mainLayout = new QVBoxLayout();

    //
    // graphic window
    //

    thePlot=new QCustomPlot();
    thePlot->setSizePolicy(QSizePolicy::Preferred, QSizePolicy::Minimum);
    thePlot->setInteractions(QCP::iSelectPlottables);


    thePlot->setMinimumWidth(minWidth);
    thePlot->setMinimumHeight(minHeight);

    mainLayout->addWidget(thePlot);

    thePlot->xAxis->setLabel(xLabel);
    thePlot->yAxis->setLabel(yLabel);
   // thePlot->xAxis->setScaleType(QCPAxis::stLogarithmic);

    this->setLayout(mainLayout);
}


SimCenterGraphPlot::~SimCenterGraphPlot()
{

}

void
SimCenterGraphPlot::clear() {
   thePlot->clearGraphs();
   thePlot->clearPlottables();
   numGraphs=0;
   yMinValue = 1e10;
   yMaxValue = -1e10;
   xMinValue = 1e10;
   xMaxValue = -1e10;
   thePlot->replot();
}


void
SimCenterGraphPlot::drawPDF(QVector<double> &x, QVector<double> &y) {

  
    addLine(x,y);

    //
    // add shade
    //

    thePlot->graph(numGraphs-1)->setBrush(QBrush(QColor(0, 75, 148,20)));

    //
    // add some margin
    //

    double ymargin =  (yMaxValue - yMinValue)*0.3;
    double xmargin =  (xMaxValue - xMinValue)*0.05;
    thePlot->yAxis->setRange(yMinValue, yMaxValue + ymargin);
    thePlot->xAxis->setRange(xMinValue - xmargin, xMaxValue + xmargin);
    thePlot->replot();

    //
    // righthandside/top axis without ticklabels ("box on")
    //
    
    thePlot->xAxis2->setVisible(true);
    thePlot->xAxis2->setTickLabels(false);
    thePlot->yAxis2->setVisible(true);
    thePlot->yAxis2->setTickLabels(false);

}

void
SimCenterGraphPlot::addLine(QVector<double> &x, QVector<double> &y, int width, int r, int g, int b) {

    //
    // check size of vectors
    //

    int numSteps = y.size();
    if (numSteps != x.size()) {
        qDebug() << "SimCenterGraphPlot - vectors of differing sizes";
        return;
    }

    //
    // plot data
    //

    graph = thePlot->addGraph();
    QPen pen;
    pen.setWidth(width);
    pen.setColor(QColor(r,g,b));
    graph->setPen(pen);

    thePlot->graph(numGraphs)->setData(x, y, true);
    numGraphs++;

    for (int i=0; i<numSteps; i++) {
        double yValue = y.at(i);
        double xValue = x.at(i);
        if (yValue < yMinValue)
            yMinValue = yValue;
        if (yValue > yMaxValue)
            yMaxValue = yValue;
        if (xValue < xMinValue)
            xMinValue = xValue;
        if (xValue > xMaxValue)
            xMaxValue = xValue;
    }

    //
    // resize axis if needed
    //

    thePlot->yAxis->setRange(yMinValue, yMaxValue);
    thePlot->xAxis->setRange(xMinValue, xMaxValue);

    thePlot->replot();
}