/afs/in2p3.fr/throng/virgo/virgoDev/SoftFlorent/GWOLLUM/v2r2/src/Utils/GwollumPlot.h

Go to the documentation of this file.

//  Author : florent robinet (LAL - Orsay): robinet@lal.in2p3.fr
#ifndef __GwollumPlot__
#define __GwollumPlot__

#include "CUtils.h"
#include "TStyle.h"
#include "TLine.h"
#include "TMultiGraph.h"
#include "TGraph.h"
#include "TGraph2D.h"
#include "TF1.h"
#include "TH1.h"
#include "TH2.h"
#include "TPad.h"
#include "TCanvas.h"
#include "TLegend.h"
#include "TText.h"

using namespace std;

class GwollumPlot{

 public:
  
  GwollumPlot(const string aName, const string aStyleName="GWOLLUM");

  virtual ~GwollumPlot(void);

  void AddText(const string aText, const double aX, const double aY, const double aSize, const int aPadIndex=0);

  void UpdateText(const string aText);

  void AddLegendEntry(const TObject *aObj, const string aLabel, const string aStyle="LPF");
  
  void AddLegendHeader(const string aLabel);

  void DrawLegend(void);

  void ResetLegend(void){ Wleg->Clear(); Wpad->SetPad(0,0,1,1); Wpad->Modified(1); Wpad->Update();};
  
  inline void Draw(TObject *aObj, const string aOptions="", const int aPadIndex=0){ Wpad->cd(aPadIndex); aObj->Draw(aOptions.c_str()); gPad->Modified(); gPad->Update();};   

  inline void Print(const string aFileName){ Wcan->cd(); Wcan->Print(aFileName.c_str()); }; 

  void Print(const string aFileName, const double aScaleFactor);

  inline void SetLogx(const int aValue, const int aPadIndex=0){ Wpad->cd(aPadIndex); gPad->SetLogx(aValue); gPad->Modified(); gPad->Update(); }; 

  inline void SetLogy(const int aValue, const int aPadIndex=0){ Wpad->cd(aPadIndex); gPad->SetLogy(aValue); gPad->Modified(); gPad->Update(); }; 

  inline void SetLogz(const int aValue, const int aPadIndex=0){ Wpad->cd(aPadIndex); gPad->SetLogz(aValue); gPad->Modified(); gPad->Update(); }; 

  inline void SetGridx(const int aValue=1, const int aPadIndex=0){ Wpad->cd(aPadIndex); gPad->SetGridx(aValue);  gPad->Modified(); gPad->Update(); }; 

  inline void SetGridy(const int aValue=1, const int aPadIndex=0){ Wpad->cd(aPadIndex); gPad->SetGridy(aValue);  gPad->Modified(); gPad->Update(); }; 

  inline void RedrawAxis(Option_t* option="", int aPadIndex=0){ Wpad->cd(aPadIndex); gPad->RedrawAxis(option); gPad->Modified(); gPad->Update(); }; 

  inline void ResizePlot(const int aWidth, const int aHeight){ Wcan->SetCanvasSize(aWidth,aHeight);  Wcan->Modified(); Wcan->Update(); }; 
  
  inline void DivideCanvas(const int aNpads){ Wpad->DivideSquare(aNpads); };

  inline void DivideCanvas(const int aNx, const int aNy){ Wpad->Divide(aNx,aNy); };

  inline string GetCurrentStyle(void){ return stylename; };
    

 protected:
  
  int randid;                      
  string srandid;                  

  // STYLE
  TStyle *Wstyle;                  
  string stylename;                

  // CANVAS/PAD
  TCanvas *Wcan;                   
  TPad    *Wpad;                   

  // ELEMENTS
  TLegend *Wleg;                   
  TText   *Wtext;                  

 private:

  void SetGwollumStyle(void);      
  void SetFireStyle(void);         
  void SetStandardStyle(void);     
  


  ClassDef(GwollumPlot,0)
};

#endif

---