tknuclear_chart.h

#ifndef __TKNUCLEAR_CHART_H
#define __TKNUCLEAR_CHART_H
 
#include "Rtypes.h"
 
#include "TMultiGraph.h"
#include "TCanvas.h"
#include "TAxis.h"
#include "TGaxis.h"
#include "TH2.h"
 
namespace tkn{
class tknuclear_chart
{
    TMultiGraph mgrn, mgrz;
    TH2F hzn, hz, hn;
    TString chart_title;
    TList linesp1, linesp2, linesp3;
 
    int width = 900;
    int height = 900;
    double xratio = 1;
    double xoffset = 0.;
 
    double min_val = 9999999.;
    double max_val = -9999999.;
    double min_z = 9999999.;
    double min_n = 9999999.;
    double max_z = -9999999.;
    double max_n = -9999999.;
 
    bool fset_min_val=false;
    double min_val_histo=0.;
 
    bool autofit;
    TCanvas* cc{};
    TPad *p1{}, *p2{}, *p3{}, *p4{};
 
    TGaxis* axX{}, *axY{}, *axZz{}, *axZn{};
    int ndivx, ndivy, ndivz;
 
    int odd_col = kRed;
    int even_col = kBlack;
 
    TLatex *ftitletex = nullptr;
 
    int fsaved_emptybins_color{};
 
    int kLineWidth = 1;
 
public:
 
    enum chart_types {
        kAll,
        kEven,
        kOdd
    };
 
   tknuclear_chart(const char* title, int type=kAll, bool autofit=false);
   tknuclear_chart(const char* title, int type=kAll, double _min_val_histo=0, bool autofit=false);
   virtual ~tknuclear_chart() {}
 
   void set_size(int _width, int _height);
 
   void set_value(int _zz, int _nn, double _val);
   double get_value(int _zz, int _nn);
 
   void fill(int _zz, int _nn, double _val);
   void fill(int _zz, int _nn) {fill(_zz,_nn,1.);}
 
   void update() {draw(true);}
   void draw(bool _update_only=false);
 
   void set_odd_color(int _col) {odd_col = _col;}
   void set_even_color(int _col){even_col = _col;}
 
   void set_log_scale(bool _log=true);
   void set_editables(bool _ed=true);
 
   void set_ndivisions_x(int _ndiv);
   void set_ndivisions_y(int _ndiv);
   void set_ndivisions_z(int _ndiv);
 
   void reset();
 
   void set_title(TString _title) {chart_title = _title;}
 
   void save_as(const char* file_name);
 
   void hide_empty_bins();
 
   void show_empty_bins();
 
   void set_line_width(int _width){kLineWidth = _width;}
 
   TH2 *get_histo() {return &hzn;}
 
protected:
   void draw_lines(int nz);
   void draw_line(int nz, double min, double max, bool onx=true);
   void draw_canvas();
   void draw_axes();
   void set_axis_att(TAxis* _ax, bool _show_labels=false);
   void style_axis(TGaxis *_ax);
 
   void calculate_projections();
 
   ClassDef(tknuclear_chart,1)//nuclear chart
};
}
#endif