Plotter.h

Go to the documentation of this file.

#ifndef PLOTTER_HH
#define PLOTTER_HH

#include "TH1.h"
#include "TObject.h"
#include "TLegend.h"
#include "TBox.h"
#include "TArc.h"
#include "TCanvas.h"
#include "TStyle.h"
#include "TEllipse.h"
#include "Mint/MessageService.h"


class Plotter {
  
  protected:


  //This class has ownership of the _objToPlot, but not the _histograms

  TPad*            _canvas;           
  TLegend*         _legend;           
  double           _forcedMax;        
  double           _forcedMin;        
  std::vector<TObject*> _objToPlot;   
  std::vector<TObject*> _histograms;  
  std::vector<int>      _colours;     
  double _lMargin;  
  double _rMargin;  
  double _tMargin;  
  double _bMargin;  
  double _xAxisTitleOffset; 
  double _yAxisTitleOffset; 
  double _xAxisLabelOffset;  
  double _yAxisLabelOffset;  
  double _xAxisTickLength; 
  double _yAxisTickLength; 
  double _xAxisLabelSize;  
  double _yAxisLabelSize;  
  double _xAxisTitleSize;  
  double _yAxisTitleSize;  
  bool _histogramOwnership; 
  bool _objectOwnership;    
  bool _usePresetColours; 
  bool _allImageFormats ; 
  virtual void setCanvasDefaults(TPad* pad); 

  public:
    
  static TString s_imageformat;     
  static TString s_imageformat2;    
  static TString s_legend_position; 
  static int     s_plotterCount;    
  static double  s_forcedMax;       
  static double  s_forcedMin;       
  int getColor(int i);  
  void setColor(int i, int color);  
  void usePresetColours(bool val= true){_usePresetColours = val;}  
  void allImageFormats (bool val= true){_allImageFormats  = val;}  
  void setHistogramOwnership(bool i=1){_histogramOwnership = i;}
  void setObjectOwnership(bool i=1){_objectOwnership = i;}
  Plotter(TString canvasName, double width, double height);

  Plotter(const Plotter& other);
 
  virtual void plot(TString plotDirectory, TString plotOptions = "", TPad* pad = 0, double scaleFactor = 1.0) = 0;
  void add(TObject* histogram);
  void addDot(double xpos, double ypos, double size, int colour = 1, TString shape = "circle", double sizeY = 0.0);
  void logX(bool log = 1);
  void logY(bool log = 1);
  void logZ(bool log = 1);

  //note that any object passed here will have ownership taken by the class
  void addObject(TObject* obj);
  
  int getNumObjects();

  void setImageFormat(TString format);
  

  TPad* getCanvas() { return _canvas; }
  void scaleTextSize(double scale){
    _xAxisLabelSize   *= scale;
    _yAxisLabelSize   *= scale;
    _xAxisTitleSize *= scale;
    _yAxisTitleSize *= scale;
  }
  void scaleAxisTitleSize(double scale){
    _xAxisTitleSize *= scale;
    _yAxisTitleSize *= scale;
  }
  void scaleAxisTitleOffset(double scale){
    _xAxisTitleOffset *= scale;
    _yAxisTitleOffset *= scale;
  }
  void setXAxisLabelSize(double val ){_xAxisLabelSize = val;}          
  void setYAxisLabelSize(double val ){_yAxisLabelSize = val;}      
  void setXAxisTitleSize(double val ){_xAxisTitleOffset = val;}      
  void setYAxisTitleSize(double val ){_yAxisTitleOffset = val;}      
  void setXAxisLabelOffset(double val ){_xAxisLabelOffset = val;}      
  void setYAxisLabelOffset(double val ){_yAxisLabelOffset = val;}      
  void setXAxisTitleOffset(double val ){_xAxisTitleOffset = val;}      
  void setYAxisTitleOffset(double val ){_yAxisTitleOffset = val;}      
  void setXAxisTickLength(double val ){_xAxisTickLength = val;}      
  void setYAxisTickLength(double val ){_yAxisTickLength = val;}      
  void setPropertiesFromTH1(TH1* hist);
  void setMin(double min) { _forcedMin = min; } 
  void setMax(double max) { _forcedMax = max; } 
  void setBMargin(double val){_bMargin = val;} 
  void setLMargin(double val){_lMargin = val;} 
  void setRMargin(double val){_rMargin = val;} 
  void setTMargin(double val){_tMargin = val;} 
  virtual ~Plotter();
  
};

#endif