GeometryComparisonPlotter.h

Go to the documentation of this file.

#include <TROOT.h>
#include <iostream>
#include <vector>

#include <TString.h>
#include <TStyle.h>
#include <TAxis.h>
#include <TGraph.h>
#include <TH2.h>
#include <TMultiGraph.h>
#include <TFile.h>
#include <TCanvas.h>
#include <TLegend.h>
#include <TTree.h>
#include <TDirectory.h>
#include <TMath.h>
#include <TPaveText.h>
#include <TLatex.h>
#include <TList.h>


class GeometryComparisonPlotter
{
    // internal variables
#ifndef NB_SUBLEVELS
#define NB_SUBLEVELS 6
#endif
    TString _sublevel_names[NB_SUBLEVELS],
            _output_directory,
            _output_filename,
            _print_option,
            _module_plot_option,
            _alignment_name,
            _reference_name,
            _print_only_global;
    bool _print,
         _legend,
         _write,
         _batchMode,
         _1dModule,
         _2dModule;
    int _levelCut,
        _grid_x,
        _grid_y,
        _window_width,
        _window_height;

    // branches
    map<TString, int> branch_i;
    map<TString, float> branch_f,
                        _max, _min, _SF;
    map<TString, TString> _units;

    // variables of external objects
    TFile * tree_file;
    TFile * output;
    TTree * data;

    // methods
    TString LateXstyle (TString);
    TString ExtensionFromPrintOption (TString);
    TLegend * MakeLegend (double x1,
                          double y1,
                          double x2,  
                          double y2,
                          const TString title = "");

public:

    static int canvas_index; // to append to the name of the canvases in case of duplication

    // constructor and destructor
    GeometryComparisonPlotter(TString tree_file_name,
                              TString outputDirname = "output/",
                              TString modulesToPlot="all",
                              TString referenceName="Ideal",
                              TString alignmentName="Alignment",
                              TString plotOnlyGlobal="false");
    ~GeometryComparisonPlotter ();

    // main methods
    void MakePlots (const vector<TString>,
                    const vector<TString>,
                    const vector<float>,
                    const vector<float>
                    );

    // option methods
    void SetPrint               (const bool);           // activates the printing of the individual and global pdf
    void SetLegend              (const bool);           // activates the legends
    void SetWrite               (const bool);           // activates the writing into a Root file
    void Set1dModule            (const bool);           // cut to include 1D modules
    void Set2dModule            (const bool);           // cut to include 2D modules
#define DEFAULT_LEVEL 1
    void SetLevelCut            (const int);            // module level: level=1 (default)
    void SetBatchMode           (const bool);           // activates the display of the canvases
    void SetGrid                (const int,             // activates the display of the grids
                                 const int);
    void SetBranchMax           (const TString,         // sets a max value for the variable
                                 const float);          // by giving the name and the value
    void SetBranchMin           (const TString,         // sets a min value for the variable
                                 const float);          // by giving the name and the value
    void SetBranchSF            (const TString,         // sets a rescaling factor for the variable
                                 const float);          // by giving the name and the value
    void SetBranchUnits         (const TString,         // writes de units next on the axis
                                 const TString);
    void SetOutputDirectoryName (const TString);        // sets the output name of the directory
    void SetOutputFileName      (const TString);        // sets the name of the root file (if applicable)
    void SetPrintOption         (const Option_t *);     // litteraly the print option of the TPad::Print()
#define DEFAULT_WINDOW_WIDTH  3508
#define DEFAULT_WINDOW_HEIGHT 2480
    void SetCanvasSize          (const int window_width  = DEFAULT_WINDOW_WIDTH,
                                 const int window_height = DEFAULT_WINDOW_HEIGHT);
};