BarChart.h

Go to the documentation of this file.

#ifndef BAR_CHART_H
#define BAR_CHART_H
 
#include <unordered_map>
#include <vector>
#include <string>
#include <DataStructure.h>
#include <JSONutil.h>
 
namespace bridges {
    namespace datastructure {
 
        class BarChart : public DataStructure {
            private:
                std::string xLabel;
                std::string yLabel;
                std::string plotTitle;
                std::string plotSubTitle;
                std::string tooltipSuffix;
                std::string alignment;
 
                unordered_map<string, vector<double>> seriesData;
                std::vector<std::string> seriesBins;
 
            public:
                BarChart() {
                    plotTitle = "";
                    plotSubTitle = "";
                    yLabel = "";
                    xLabel = "";
                    tooltipSuffix = "";
                    alignment = "horizontal";
                }
 
                virtual const string getDStype() const override {
                    return "BarChart";
                }
 
                void setTitle(std::string t) {
                    plotTitle = t;
                }
 
                std::string getTitle() const {
                    return plotTitle;
                }
 
                void setSubTitle(std::string s) {
                    plotSubTitle = s;
                }
 
                std::string getSubTitle() const {
                    return plotSubTitle;
                }
 
                void setSeriesLabel(std::string yaxisName) {
                    yLabel = yaxisName;
                }
 
                std::string getSeriesLabel() const {
                    return yLabel;
                }
 
                void setBinsLabel(std::string xaxisName) {
                    xLabel = xaxisName;
                }
 
                std::string getBinsLabel() const {
                    return xLabel;
                }
 
                void setBarAlignment(std::string align) {
                    alignment = align;
                }
 
                std::string getBarAlignment() const {
                    return alignment;
                }
 
                void setTooltipSuffix(std::string suffix) {
                    tooltipSuffix = suffix;
                }
 
                std::string getTooltipSuffix() const {
                    return tooltipSuffix;
                }
 
                void setSeriesBins(std::vector<std::string> bins) {
                    seriesBins = bins;
                }
 
                void addDataSeries(std::string seriesName, std::vector<double> data) {
                    seriesData[seriesName] = data;
                }
 
                virtual const std::string getDataStructureRepresentation() const override {
                    using bridges::JSONUtil::JSONencode;
                    std::string bins = "";
                    bins += JSONencode("xAxis") + COLON + OPEN_CURLY + JSONencode("categories") + COLON + OPEN_BOX;
                    for (auto& entry : seriesBins) {
                        bins += JSONencode(entry) + COMMA;
                    }
                    bins = bins.erase(bins.length() - 1);
                    bins += CLOSE_BOX + CLOSE_CURLY;
 
                    std::string series = "";
                    series += JSONencode("series") + COLON + OPEN_BOX;
                    for (auto& entry : seriesData) {
                        series += OPEN_CURLY;
                        std::string key = entry.first;
                        std::vector<double> value = entry.second;
 
                        series += JSONencode("name") + COLON + JSONencode(key) + COMMA + JSONencode("data") + COLON + OPEN_BOX;
                        for (int i = 0; i < value.size(); i++) {
                            series += JSONencode(value[i]) + COMMA;
                        }
                        series = series.erase(series.length() - 1);
                        series += CLOSE_BOX + CLOSE_CURLY + COMMA;
                    }
                    series = series.erase(series.length() - 1);
                    series += CLOSE_BOX;
 
                    std::string json_str = JSONencode("plot_title") + COLON +  JSONencode(getTitle()) + COMMA +
                        JSONencode("subtitle") + COLON + JSONencode(getSubTitle())  + COMMA +
                        JSONencode("xLabel") + COLON + JSONencode(getBinsLabel()) +  COMMA +
                        JSONencode("yLabel") + COLON + JSONencode(getSeriesLabel()) + COMMA +
                        JSONencode("tooltipSuffix") + COLON + JSONencode(getTooltipSuffix()) + COMMA +
                        JSONencode("alignment") + COLON + JSONencode(getBarAlignment()) + COMMA +
                        JSONencode("xaxis_data") + COLON + OPEN_CURLY + bins + CLOSE_CURLY + COMMA +
                        JSONencode("yaxis_data") + COLON + OPEN_CURLY + series + CLOSE_CURLY + CLOSE_CURLY;
 
                    return json_str;
                }
        };
    }
 
}
#endif