JKQtPlotter/examples/styling/build_test_graphs.h

#ifndef BUILD_TEST_GRAPHS_H
#define BUILD_TEST_GRAPHS_H

#include "jkqtplotter/graphs/jkqtpscatter.h"
#include "jkqtplotter/graphs/jkqtpbarchart.h"
#include "jkqtplotter/graphs/jkqtpimage.h"
#include "jkqtplotter/graphs/jkqtpgeometric.h"
#include "jkqtplotter/graphs/jkqtpfinancial.h"
#include "jkqtplotter/graphs/jkqtpgeoannotations.h"
#include "jkqtplotter/graphs/jkqtpboxplot.h"
#include "jkqtplotter/graphs/jkqtpfilledcurve.h"
#include "jkqtplotter/graphs/jkqtpimpulses.h"
#include "jkqtplotter/jkqtpbaseplotter.h"

#define DEFAULT_PLOT_SIZE QSize(4,2)

inline void setupGridPrintingAndDatastores(QVector<JKQTBasePlotter*>& plots, QSize size) {
    // connect plots
    for  (auto plt: plots) {

        for (int x=0; x<size.width(); x++) {
            for (int y=0; y<size.height(); y++) {
                plt->addGridPrintingPlotter(x,y, plots[y*size.width()+x]);
            }
        }
        plt->setGridPrinting(true);

        // use same datastore for each plot
        if (plt!=plots.first()) plt->useExternalDatastore(plots.first()->getDatastore());
    }
}

inline void buildPlots(QVector<JKQTBasePlotter*>& plots) {
    JKQTPFinancialGraph::clearColorAssignStore();

    // 0. get a pointer to the internal datastore (for convenience)
    JKQTPDatastore* ds=plots.first()->getDatastore();

    // 1. clear old graphs/data
    for (auto plt: plots) {
        plt->clearGraphs(true);
    }
    ds->clear();

    // 2. now we create data for a simple plot (a sine curve)
    QVector<double> X, Y1, Y2, img, X3, Y3, Y3err, Xbar, Ybar, Ybar2, Ybar2s1, Ybar2s2, Ybar3, Y4, Y5, XImp, YImp, XBP, MinBP, Q25BP, MeanBP, MedianBP, Q75BP, MaxBP, TFin, OFin1,CFin1,LFin1,HFin1, OFin2,CFin2,LFin2,HFin2;
    const int Ndata=50;
    for (int i=0; i<Ndata; i++) {
        const double x=double(i)/double(Ndata)*8.0*JKQTPSTATISTICS_PI;
        X<<x;
        Y1<<3.1+sin(x)+x/25.0*2.0;
        Y2<<3.1+1.5*sin(x)*exp(-0.2*x)+x/25.0*2.0;
        Y4<<4.5*exp(-x/10.0);
        Y5<<2.0*exp(-x/10.0)+0.8;
        if (i%5==0) {
            X3<<x;
            Y3<<(4.5)+sin(x/25.0*JKQTPSTATISTICS_PI)*2.0-x/12.5;
            Y3err<<(double(i+5)/double(Ndata)*0.5);
        }
        if (i>0 && i%16==0) {
            Xbar<<x;
            Ybar<<(double(i+5)/double(Ndata)*(1.5));
            Ybar2<<(double(i+5)/double(Ndata)*(0.6));
            Ybar2s1 << 0.1+(double(i+5)/double(Ndata)*(0.3));
            Ybar2s2 << 0.5-(double(i+5)/double(Ndata)*(0.4));
            Ybar3<<(3.0-(double(i+5)/double(Ndata)*(3.7)));
        }
        if (i%8==0) {
            XImp<<1+x/1.2;
            YImp<<1.5+1.1*cos(x/2.0);
        }
        if (i<=5) {
            XBP<<(i-1);
            MinBP<<(i+1);
            Q25BP<<(i*1.1+3.0);
            MeanBP<<(i*1.12+4.0);
            MedianBP<<(i*1.2+3.5);
            Q75BP<<(i*1.3+6.0);
            MaxBP<<(i*1.35+7.5);
        }
        if (i<=10) {
            TFin << (i+1);
            OFin1 << CFin1.value(CFin1.size()-1,30)+3.0*sin(i);
            CFin1 << OFin1.value(OFin1.size()-1,30)+3.0*cos(i);
            HFin1 << qMax(OFin1.last(),CFin1.last())+2;
            LFin1 << qMin(OFin1.last(),CFin1.last())-2;
            OFin2 << CFin2.value(CFin2.size()-1,20)+2.0*cos(i);
            CFin2 << OFin2.value(OFin2.size()-1,20)+2.0*sin(i);
            HFin2 << qMax(OFin2.last(),CFin2.last())+2;
            LFin2 << qMin(OFin2.last(),CFin2.last())-2;
        }
    }
    auto fgauss=[](double x, double y, double x0, double y0, double sx, double sy) {
        return exp(-2.0*(x-x0)*(x-x0)/sx/sx-2.0*(y-y0)*(y-y0)/sy/sy);
    };
    for (int i=0; i<Ndata; i++) {
        for (int j=0; j<Ndata; j++) {
            img.push_back(fgauss(j,i,Ndata/3,Ndata/3,Ndata/5,Ndata/3)+fgauss(j,i,Ndata*3/4,Ndata*2/3,Ndata/2,Ndata/4));
        }
    }

    // 3. make data available to JKQTPlotter by adding it to the internal datastore.
    //    Note: In this step the data is copied (of not specified otherwise), so you can
    //          reuse X and Y afterwards!
    //    the variables columnX and columnY will contain the internal column ID of the newly
    //    created columns with names "x" and "y" and the (copied) data from X and Y.
    size_t columnX=ds->addCopiedColumn(X, "x");
    size_t columnY1=ds->addCopiedColumn(Y1, "y1");
    size_t columnY2=ds->addCopiedColumn(Y2, "y2");
    size_t columnImg=ds->addCopiedColumn(img, "img");
    size_t columnX3=ds->addCopiedColumn(X3, "x3");
    size_t columnY3=ds->addCopiedColumn(Y3, "y3");
    size_t columnY3err=ds->addCopiedColumn(Y3err, "y3err");
    size_t columnXbar=ds->addCopiedColumn(Xbar, "xbar");
    size_t columnYbar=ds->addCopiedColumn(Ybar, "ybar");
    size_t columnYbar2=ds->addCopiedColumn(Ybar2, "Ybar2");
    size_t columnYbar2s1=ds->addCopiedColumn(Ybar2s1, "Ybar2, stack1");
    size_t columnYbar2s2=ds->addCopiedColumn(Ybar2s2, "Ybar2, stack2");
    size_t columnYbar3=ds->addCopiedColumn(Ybar3, "Ybar3");
    size_t columnY4=ds->addCopiedColumn(Y4, "y4");
    size_t columnY5=ds->addCopiedColumn(Y5, "y5");
    size_t columnXImp=ds->addCopiedColumn(XImp, "ximp");
    size_t columnYImp=ds->addCopiedColumn(YImp, "yimp");

    size_t columnXBP=ds->addCopiedColumn(XBP, "XBP");
    size_t columnMinBP=ds->addCopiedColumn(MinBP, "MinBP");
    size_t columnQ25BP=ds->addCopiedColumn(Q25BP, "Q25BP");
    size_t columnMeanBP=ds->addCopiedColumn(MeanBP, "MeanBP");
    size_t columnMedianBP=ds->addCopiedColumn(MedianBP, "MedianBP");
    size_t columnQ75BP=ds->addCopiedColumn(Q75BP, "Q75BP");
    size_t columnMaxBP=ds->addCopiedColumn(MaxBP, "MaxBP");

    size_t columnFinT=ds->addCopiedColumn(TFin, "FinT");

    // 4. create diverse graphs in the plot:
    JKQTPFilledCurveXGraph* graphf4=new JKQTPFilledCurveXGraph(plots[2]);
    graphf4->setXColumn(columnX);
    graphf4->setYColumn(columnY4);
    graphf4->setTitle(QObject::tr("filled exp-function"));
    plots[2]->addGraph(graphf4);

    JKQTPFilledCurveXGraph* graphf5=new JKQTPFilledCurveXGraph(plots[2]);
    graphf5->setXColumn(columnX);
    graphf5->setYColumn(columnY5);
    graphf5->setTitle(QObject::tr("other filled exp-function"));
    plots[2]->addGraph(graphf5);


    JKQTPColumnMathImage* graphI=new JKQTPColumnMathImage(plots.last());
    graphI->setImageColumn(columnImg);
    graphI->setNx(Ndata);
    graphI->setNy(Ndata);
    graphI->setColorBarTopVisible(true);
    graphI->setColorBarRightVisible(true);
    graphI->setX(0.5);
    graphI->setY(-2.0);
    graphI->setWidth(5);
    graphI->setHeight(1);
    graphI->setTitle(QObject::tr("$2\\times$ Gauss"));
    plots.last()->addGraph(graphI);

    JKQTPXYLineErrorGraph* graph1=new JKQTPXYLineErrorGraph(plots[0]);
    graph1->setXColumn(columnX3);
    graph1->setYColumn(columnY3);
    graph1->setYErrorColumn(columnY3err);
    graph1->setDrawLine(true);
    graph1->setYErrorStyle(JKQTPErrorBarsLines);
    graph1->setSymbolType(JKQTPNoSymbol);
    graph1->setTitle(QObject::tr("sine \\pm errors graph"));
    plots[0]->addGraph(graph1);

    JKQTPXYLineGraph* graph2=new JKQTPXYLineGraph(plots[0]);
    graph2->setXColumn(columnX);
    graph2->setYColumn(columnY1);
    graph2->setSymbolType(JKQTPNoSymbol);
    graph2->setTitle(QObject::tr("sine graph $f(x)=\\sin(x)$"));
    plots[0]->addGraph(graph2);

    JKQTPXYLineGraph* graph3=new JKQTPXYLineGraph(plots[0]);
    graph3->setXColumn(columnX);
    graph3->setYColumn(columnY2);
    graph3->setSymbolType(JKQTPNoSymbol);
    //graph2->setTitle(QObject::tr("sine{\\cdot}exp graph $f(x)=\\sin(x)\\cdot\\exp\\left(-\\frac{x}{5}\\right)$"));
    graph3->setTitle(QObject::tr("sine*exp graph"));
    plots[0]->addGraph(graph3);


    JKQTPImpulsesVerticalGraph* graphi2=new JKQTPImpulsesVerticalGraph(plots[0]);
    graphi2->setXColumn(columnXImp);
    graphi2->setYColumn(columnYImp);
    graphi2->setTitle(QObject::tr("impulses"));
    plots[0]->addGraph(graphi2);



    JKQTPXYLineErrorGraph* graph1s=new JKQTPXYLineErrorGraph(plots[1]);
    graph1s->setXColumn(columnX3);
    graph1s->setYColumn(columnY3);
    graph1s->setYErrorColumn(columnY3err);
    graph1s->setDrawLine(false);
    graph1s->setYErrorStyle(JKQTPErrorBars);
    graph1s->setTitle(QObject::tr("sine \\pm errors graph"));
    plots[1]->addGraph(graph1s);

    JKQTPXYScatterGraph* graph2s=new JKQTPXYScatterGraph(plots[1]);
    graph2s->setXColumn(columnX);
    graph2s->setYColumn(columnY1);
    graph2s->setTitle(QObject::tr("sine graph $f(x)=\\sin(x)$"));
    plots[1]->addGraph(graph2s);

    JKQTPXYScatterGraph* graph3s=new JKQTPXYScatterGraph(plots[1]);
    graph3s->setXColumn(columnX);
    graph3s->setYColumn(columnY2);
    graph3s->setTitle(QObject::tr("sine*exp graph"));
    plots[1]->addGraph(graph3s);

    JKQTPBarVerticalGraph* graphb=new JKQTPBarVerticalGraph(plots[3]);
    graphb->setXColumn(columnXbar);
    graphb->setYColumn(columnYbar);
    graphb->setTitle(QObject::tr("barchart"));
    plots[3]->addGraph(graphb);

    JKQTPBarVerticalStackableGraph* graphb2=new JKQTPBarVerticalStackableGraph(plots[3]);
    graphb2->setXColumn(columnXbar);
    graphb2->setYColumn(columnYbar2);
    graphb2->setTitle(QObject::tr("2^{nd} barchart"));
    plots[3]->addGraph(graphb2);

    JKQTPBarVerticalStackableGraph* graphb2s1=new JKQTPBarVerticalStackableGraph(plots[3]);
    graphb2s1->setXColumn(columnXbar);
    graphb2s1->setYColumn(columnYbar2s1);
    graphb2s1->setTitle(QObject::tr("3^{rd} barchart"));
    plots[3]->addGraph(graphb2s1);
    graphb2s1->stackUpon(graphb2);

    JKQTPBarVerticalStackableGraph* graphb2s2=new JKQTPBarVerticalStackableGraph(plots[3]);
    graphb2s2->setXColumn(columnXbar);
    graphb2s2->setYColumn(columnYbar2s2);
    graphb2s2->setTitle(QObject::tr("4^{th} barchart"));
    plots[3]->addGraph(graphb2s2);
    graphb2s2->stackUpon(graphb2s1);

    JKQTPBarVerticalGraph* graphb3=new JKQTPBarVerticalGraph(plots[3]);
    graphb3->setXColumn(columnXbar);
    graphb3->setYColumn(columnYbar3);
    graphb3->setTitle(QObject::tr("5^{th} barchart"));
    plots[3]->addGraph(graphb3);
    graphb3->autoscaleBarWidthAndShiftSeparatedGroups();
    plots[3]->getMainKey()->setPosition(JKQTPKeyInsideTopLeft);


    JKQTPBoxplotVerticalGraph* graphBP=new JKQTPBoxplotVerticalGraph(plots[plots.size()/2]);
    graphBP->setPositionColumn(columnXBP);
    graphBP->setMinColumn(columnMinBP);
    graphBP->setPercentile25Column(columnQ25BP);
    graphBP->setMeanColumn(columnMeanBP);
    graphBP->setMedianColumn(columnMedianBP);
    graphBP->setPercentile75Column(columnQ75BP);
    graphBP->setMaxColumn(columnMaxBP);
    plots[plots.size()/2]->addGraph(graphBP);


    JKQTPFinancialGraph* graphF1=new JKQTPFinancialGraph(plots[plots.size()/2+1]);
    graphF1->setXColumn(columnFinT);
    graphF1->setOpenColumn(ds->addCopiedColumn(OFin1, "OFin1"));
    graphF1->setHighColumn(ds->addCopiedColumn(HFin1, "HFin1"));
    graphF1->setLowColumn(ds->addCopiedColumn(LFin1, "LFin1"));
    graphF1->setCloseColumn(ds->addCopiedColumn(CFin1, "CFin1"));
    graphF1->setGraphType(JKQTPFinancialGraph::OHLC);
    graphF1->setTitle("OHLC");
    plots[plots.size()/2+1]->addGraph(graphF1);
    JKQTPFinancialGraph* graphF2=new JKQTPFinancialGraph(plots[plots.size()/2+1]);
    graphF2->setXColumn(columnFinT);
    graphF2->setOpenColumn(ds->addCopiedColumn(OFin2, "OFin2"));
    graphF2->setHighColumn(ds->addCopiedColumn(HFin2, "HFin2"));
    graphF2->setLowColumn(ds->addCopiedColumn(LFin2, "LFin2"));
    graphF2->setCloseColumn(ds->addCopiedColumn(CFin2, "CFin2"));
    graphF2->setGraphType(JKQTPFinancialGraph::CandleStick);
    graphF2->setTitle("CandleStick");
    plots[plots.size()/2+1]->addGraph(graphF2);

    auto plotgeo=*(plots.rbegin()+1);
    JKQTPGeoSymbol* annotSym=new JKQTPGeoSymbol(plotgeo, -1.5, 3);
    plotgeo->addGraph(annotSym);

    JKQTPGeoText* annotTxt=new JKQTPGeoText(plotgeo, annotSym->getX()+0.5, annotSym->getY(), "Annotation $\\int_{0}^{2\\pi}\\sin(x)\\;\\mathrm{d}x$");
    plotgeo->addGraph(annotTxt);

    JKQTPGeoEllipse* geoEll=new JKQTPGeoEllipse(plotgeo, 1.5,0.5,3.5,1.2);
    geoEll->setAngle(-25);
    plotgeo->addGraph(geoEll);

    plotgeo->addGraph(new JKQTPGeoArrow(plotgeo, geoEll->getCenter(),annotSym->getP()-QPointF(0.5,0.5)));

    plotgeo->addGraph(new JKQTPGeoLine(plotgeo, geoEll->getCenter(),QPointF(annotSym->getX(),annotSym->getY()-2.0)));

    // 6. autoscale the plot so the graph is contained
    for (auto plt: plots) {
        plt->zoomToFit();
        plt->getXAxis()->setAxisLabel("x-axis");
        plt->getYAxis()->setAxisLabel("y-axis");
    }
    plots[0]->getYAxis()->setRange(plots[0]->getYAxis()->getMin(), 8);
    plots[1]->getYAxis()->setRange(plots[1]->getYAxis()->getMin(), 8);
    plots[3]->getYAxis()->setRange(plots[3]->getYAxis()->getMin(), 2.5);
    plotgeo->setMaintainAspectRatio(true);
    plotgeo->setMaintainAxisAspectRatio(true);
    plotgeo->getXAxis()->setRange(-2.5,4);
    plotgeo->getYAxis()->setRange(-2.5,4);
    plots[plots.size()/2+1]->getXAxis()->setAxisLabel("time");
    plots[plots.size()/2+1]->getYAxis()->setAxisLabel("price [\\euro]");


    plots[0]->setPlotLabel("Line Graphs");
    plots[1]->setPlotLabel("Scatter Graphs");
    plots[2]->setPlotLabel("Filled Graphs");
    plots[3]->setPlotLabel("Bar Charts");
    plots[plots.size()/2]->setPlotLabel("Boxplots");
    plots[plots.size()/2+1]->setPlotLabel("Financial Graphs");
    plotgeo->setPlotLabel("Geometry & Annotations");
    plots.last()->setPlotLabel("Image Plot");

}

inline void buildLineSymbolSampler(JKQTBasePlotter* plotExtra, JKQTBasePlotter* baseplot) {
    plotExtra->clearGraphs(true);
    plotExtra->setBackgroundColor(baseplot->getBackgroundColor());
    plotExtra->setExportBackgroundColor(baseplot->getExportBackgroundColor());
    plotExtra->setPlotBackgroundColor(baseplot->getPlotBackgroundColor());
    plotExtra->setPlotLabelColor(baseplot->getPlotLabelColor());
    plotExtra->getXAxis()->setDrawMode1(JKQTPCADMnone);
    plotExtra->getXAxis()->setDrawMode2(JKQTPCADMnone);
    plotExtra->getYAxis()->setDrawMode1(JKQTPCADMnone);
    plotExtra->getYAxis()->setDrawMode2(JKQTPCADMnone);
    double x=1;
    double xlineend=6;
    double y=1;
    double xmax=15;
    int cnt=0;
    const QColor defaultColor=baseplot->getCurrentPlotterStyle().graphsStyle.defaultGraphColors.first();
    for (auto s: baseplot->getCurrentPlotterStyle().graphsStyle.defaultGraphSymbols) {
        plotExtra->addGraph(new JKQTPGeoSymbol(plotExtra, x, y, s, 12, defaultColor, defaultColor.lighter() ));
        x+=1;
        xmax=qMax(x, xmax);
        if (cnt>0 && cnt%25==0) {
            x=1;
            y+=1.0;
        }
        cnt++;
    }
    y=y+1;
    double ycoltest=y;
    for (auto s: baseplot->getCurrentPlotterStyle().graphsStyle.defaultGraphPenStyles) {
        JKQTPGeoLine* l;
        plotExtra->addGraph(l=new JKQTPGeoLine(plotExtra, 1,y,xlineend,y)); l->setStyle(defaultColor, 2, s);
        plotExtra->addGraph(l=new JKQTPGeoLine(plotExtra, 1,y-0.25,xlineend,y-0.25)); l->setStyle(defaultColor.darker(), 1, s);
        y+=0.75;
    }
    x=xlineend+1;
    const int cntFillTests=qMax(baseplot->getCurrentPlotterStyle().graphsStyle.defaultGraphColors.size(), baseplot->getCurrentPlotterStyle().graphsStyle.defaultGraphFillStyles.size());
    double dx=(xmax-xlineend-1.0)/static_cast<double>(cntFillTests);
    for (int i=0; i<cntFillTests; i++) {
        const auto col=baseplot->getCurrentPlotterStyle().graphsStyle.defaultGraphColors.value(i%baseplot->getCurrentPlotterStyle().graphsStyle.defaultGraphColors.size(), Qt::black);
        const auto fs=baseplot->getCurrentPlotterStyle().graphsStyle.defaultGraphFillStyles.value(i%baseplot->getCurrentPlotterStyle().graphsStyle.defaultGraphFillStyles.size(), JKQTFillStyleSummmary(Qt::SolidPattern));
        JKQTPGeoLine* l;
        plotExtra->addGraph(l=new JKQTPGeoLine(plotExtra, x,ycoltest,x,y-0.5)); l->setStyle(col,5);
        JKQTPGeoRectangle* r;
        plotExtra->addGraph(r=new JKQTPGeoRectangle(plotExtra, x+dx/2,(ycoltest+y-0.5)/2.0, dx*0.5, y-ycoltest-0.5));
        r->setStyle(col,1, Qt::SolidLine, JKQTPGetDerivedColor(baseplot->getCurrentPlotterStyle().graphsStyle.defaultGraphStyle.fillColorDerivationMode, col), fs);
        x+=dx;
    }
    plotExtra->zoomToFit();
    plotExtra->setX(0, plotExtra->getXMax()+2);
    plotExtra->setY(0, plotExtra->getYMax()+1);
    plotExtra->setGrid(false);
    plotExtra->setPlotLabel("default symbols/lines/colors");

}

#endif // BUILD_TEST_GRAPHS_H