MHO_BasicPlotVisitor.hh

Go to the documentation of this file.

#ifndef MHO_BasicPlotVisitor_HH__
#define MHO_BasicPlotVisitor_HH__
 
#include "MHO_FringePlotVisitor.hh"
#include "MHO_JSONHeaderWrapper.hh"
#include "MHO_PlotDataExtractor.hh"
 
#include <algorithm>
#include <cmath>
#include <iomanip>
#include <matplot/matplot.h>
#include <numeric>
#include <sstream>
#include <string>
#include <utility>
#include <vector>
 
namespace hops
{
 
class MHO_BasicPlotVisitor: public MHO_FringePlotVisitor
{
    public:
        MHO_BasicPlotVisitor();
        virtual ~MHO_BasicPlotVisitor();
 
        virtual void Plot(MHO_FringeData* data) override;
 
    protected:
    private:
        class subplot_parameters
        {
            public:
                subplot_parameters(): total_rows(0), total_cols(0), start_row(0), start_col(0), rowspan(0), colspan(0){};
 
                subplot_parameters(int a, int b, int c, int d, int e, int f)
                    : total_rows(a), total_cols(b), start_row(c), start_col(d), rowspan(e), colspan(f){};
 
                virtual ~subplot_parameters(){};
 
                //data
                int total_rows;
                int total_cols;
                int start_row;
                int start_col;
                int rowspan;
                int colspan;
        };
 
        std::map< std::string, subplot_parameters > fSubplotConfig;
 
        void ConfigureSubplots();
 
        void ConstructXTitle(const subplot_parameters& params, std::string title, std::string font_color, int font_size,
                             double x_coord, double y_coord, bool center = false);
        void ConstructYTitle(const subplot_parameters& params, std::string title, std::string font_color, int font_size,
                             bool is_y2 = false); // double x_coord, double y_coord, bool center)
 
        void ConstructPlot(const mho_json& plot_data);
        void DirectSavePlot(std::string filename);
 
        void make_dr_mbd_plot(const mho_json& plot_dict);
 
        void make_sbd_dtec_plot(const mho_json& plot_dict);
 
        void make_xpower_plot(const mho_json& plot_dict);
 
        void make_channel_segment_plots(const mho_json& plot_dict);
 
        void make_channel_segment_validity_plots(const mho_json& plot_dict);
 
        void make_pcal_plots(const mho_json& plot_dict);
 
        void make_top_info_text(const mho_json& plot_dict);
 
        void make_basic_info_text(const mho_json& plot_dict);
 
        void make_model_resid_info_text(const mho_json& plot_dict);
 
        void make_rms_table(const mho_json& plot_dict);
 
        void make_coord_text(const mho_json& plot_dict);
 
        void make_amplitude_table(const mho_json& plot_dict);
 
        void make_window_table(const mho_json& plot_dict);
 
        void make_data_stats_text(const mho_json& plot_dict);
 
        void make_channel_info_table(const mho_json& plot_dict);
 
        void setup_figure_layout();
 
        double determine_desired_tick_spacing(int n_ticks, double lower_limit, double upper_limit,
                                              const std::vector< double >& desired_intervals)
        {
            // Calculate tick marks based on span and desired spacing intervals
            double span = std::fabs(upper_limit - lower_limit);
            double target = span / (double)n_ticks;
            //example target spacing:
            //std::vector<double> tspace = {0.1, 0.5, 1, 5, 10, 25, 50, 100, 200, 500};
            int idx = 0;
            double tmin = 1e3 * upper_limit;
            for(std::size_t i = 0; i < desired_intervals.size(); i++)
            {
                double delta = std::fabs(target - desired_intervals[i]);
                if(delta < tmin)
                {
                    tmin = delta;
                    idx = i;
                }
            }
            double tick_spacing = desired_intervals[idx];
            return tick_spacing;
        }
 
        matplot::axes_handle subplot2grid(const std::pair< int, int >& shape, const std::pair< int, int >& loc, int rowspan = 1,
                                          int colspan = 1, float left_margin = 0.0, float right_margin = 0.0);
 
        matplot::axes_handle subplot2grid_wrapper(const subplot_parameters sp, double left_margin = 0.0,
                                                  double right_margin = 0.0);
 
        // Member variables for plot state
        matplot::figure_handle fCurrentFigure;
        bool fShowPlot;
        std::string fFilename;
        mho_json fPlotData;
 
        //place to stash all the axes so they stay part of the figure
        std::vector< matplot::axes_handle > fAxes;
        matplot::axes_handle fLastAxis;
 
        int fPageWidth;
        int fPageHeight;
        float fLeftMargin;
        float fRightMargin;
        int fNRows;
        int fNCols;
};
 
} // namespace hops
 
#endif /* end of include guard: MHO_BasicPlotVisitor_HH__ */