CanvasHelper.h

#ifndef CanvasHelper_HH_
#define CanvasHelper_HH_
 
#include <RtypesCore.h>
#include <Rtypes.h>
#include <TColor.h>
#include <TVirtualPad.h>
#include <THStack.h>
#include <TPaveStats.h>
#include <TLegend.h>
#include <TMap.h>
#include <TNamed.h>
#include <TLine.h>
#include <TQObject.h>
 
#include <utility>
#include <map>
#include <string>
 
namespace Round {
  int getFirstDigit();
  std::pair<double, double> valueError(const double value, const double error);
  void paveTextValueErrors(TPaveText *pave);
}
 
enum EPaveAlignBits {
  kPaveAlignLeft = BIT(14),    
  kPaveAlignRight = BIT(15),   
  kPaveAlignTop = BIT(16),     
  kPaveAlignBottom = BIT(17)   
};
 
enum ECanvasFormatBits {
  kFormatPng = BIT(14),      
  kFormatPs = BIT(15),       
  kFormatROOT = BIT(16),     
  kFormatC = BIT(17)         
};
 
class TNamedLine: public TLine {
  protected:
    TString fName;
 
  public:
    TNamedLine(const char *name, Double_t x1, Double_t y1, Double_t x2, Double_t y2);
    ~TNamedLine();
 
    const char *GetName() const override { return fName.Data(); }
 
  ClassDef(TNamedLine, 1)
};
 
class CanvasHelper: public TObject {
  public:
    static CanvasHelper* getInstance();
 
    virtual ~CanvasHelper();
 
    void addCanvas(TCanvas *canvas);
 
    static void addSubtitle(TVirtualPad *pad, const char *text);
 
    static void setPaveAlignment(TPave *pave, UInt_t align);
 
    static TPaveStats* getDefaultPaveStats(TVirtualPad *pad);
 
    static TLegend* getDefaultLegend(TVirtualPad *pad);
 
    static void addTextToStats(const char *text, TVirtualPad *pad);
 
    static void addTextToStats(const char *text, TPaveStats *stats, TVirtualPad *pad);
 
    static void addMultiCanvasTitle(TCanvas *canvas, const char *title, const char *subtitle = "");
 
    static void saveCanvas(TCanvas *canvas, UInt_t format);
 
  protected:
    CanvasHelper();
    static CanvasHelper *fgInstance;
 
    struct Margin {
        Double_t left;
        Double_t right;
        Double_t bottom;
        Double_t top;
    };
 
    enum EFontFace {
      TimesItalic = 1,
      TimesBold = 2,
      TimesBoldItalic = 3,
      Helvetica = 4,
      HelveticaItalic = 5,
      HelveticaBold = 6,
      HelveticaBoldItalic = 7,
      Courier = 8,
      CourierItalic = 9,
      CourierBold = 10,
      CourierBoldItalic = 11,
      Symbol = 12,
      Times = 13,
      Wingdings = 14,
      SymbolItalic = 15
    };
 
    // Font sizes in pixels
    static const Int_t FONT_SIZE_NORMAL;
    static const Int_t FONT_SIZE_SMALL;
    static const Int_t FONT_SIZE_SMALLER;
    static const Int_t FONT_SIZE_LARGE;
    static const Int_t FONT_SIZE_LARGER;
    static const Int_t FONT_SIZE_HUGE;
 
    static const Double_t LINE_HEIGHT;
 
//    static const Int_t AXIS_TITLE_OFFSET;
    static const Int_t AXIS_LABEL_OFFSET;
    static const Int_t AXIS_TICK_LENGTH;
 
    static const Int_t MARGIN_LEFT;
    static const Int_t MARGIN_TOP;
    static const Int_t MARGIN_RIGHT;
    static const Int_t MARGIN_BOTTOM;
 
    static const Int_t TITLE_VSPACE;
    static const Int_t SUBTITLE_VSPACE;
    static const Int_t PAVELINE_VSPACE;
    static const Int_t AXISTITLE_VSPACE;
 
    static Style_t getFont(EFontFace fontFace = EFontFace::Helvetica);
    static UInt_t getPaveLines(TPave *pave);
    static UInt_t getPaveTextWidthPx(TPaveText *paveText);
    static UInt_t getLegendWidthPx(TLegend *paveText);
 
    // TMap *canvasesToBeExported;
    static std::pair<Double_t, Double_t> getSubtitleYNDCCoordinates(TVirtualPad *pad);
 
    std::map<TCanvas*, std::pair<unsigned int, unsigned int>> registeredCanvases;
//    std::map<std::string, double> defaultPadLeftMargins;
//    static TGraph* findTGraphOnPad(TVirtualPad* pad);
 
    static TObject* findObjectOnPad(TClass *c, TVirtualPad *pad);
    // static TObject* findObjectOnPad(const char* name, TVirtualPad* pad);
 
    static void alignChildPad(TVirtualPad *pad);
 
    static Int_t getFrameLeftMarginPx(TVirtualPad *pad);
    static Int_t getFrameRightMarginPx(/*TVirtualPad* pad*/);
    static Int_t getFrameTopMarginPx(TVirtualPad *pad);
    static Int_t getFrameBottomMarginPx(TVirtualPad *pad);
 
    static Bool_t hasXAxisTitle(TVirtualPad *pad);
    static Bool_t hasYAxisTitle(TVirtualPad *pad);
 
    static Bool_t isChildPad(TVirtualPad *pad);
 
    static void alignTitle(TVirtualPad *pad);
    static void alignSubtitle(TVirtualPad *pad);
    static void alignSubtitle(TPaveText *subtitle, TVirtualPad *pad);
    static void alignAllPaves(TVirtualPad *pad);
    // static void alignAxisTitles(TVirtualPad* pad);
 
    void processCanvas(TCanvas *canvas);
    void processPad(TVirtualPad *pad);
    static void setPadMargins(TVirtualPad *pad);
 
    static void setPadNDivisions(TVirtualPad *pad);
    void convertAxisToPxSize(TAxis *axis, const char type, TVirtualPad *pad);
 
    static void setPadCustomFrameBorder(TVirtualPad *pad);
 
    static Double_t getPadWidthPx(TVirtualPad *pad);
    static Double_t getPadHeightPx(TVirtualPad *pad);
 
    static std::pair<TAxis*, TAxis*> getPadXYAxis(TVirtualPad *pad);
    static Double_t getYAxisMaxLabelWidthPx(TVirtualPad *pad);
//    Double_t getLabelHeigthPx();
 
    static Double_t pxToNdcHorizontal(Int_t px, TVirtualPad *pad);
    static Double_t pxToNdcVertical(Int_t py, TVirtualPad *pad);
 
    static constexpr char subtitleObjectName[] = "subtitle";
 
    static TFrame* getPadFrame(TVirtualPad *pad);
 
  public:
    // Slot for canvas resizing (need to be public)
    void onCanvasResized();
 
  ClassDef(CanvasHelper, 0)
};
 
#endif