GwollumPlot.h

Go to the documentation of this file.

 
#ifndef __GwollumPlot__
#define __GwollumPlot__
 
#include "Monitor.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 UnDrawLegend(void);
 
  inline 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 UnDraw(TObject *aObj, const int aPadIndex=0){
    Wpad->cd(aPadIndex);
    gPad->GetListOfPrimitives()->Remove(aObj);
    gPad->Modified();
    gPad->Update();
  };
  
  inline void Clear(const int aPadIndex=0){
    Wpad->cd(aPadIndex);
    gPad->Clear();
    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);
  
  void Print(const string aFileName,
             const unsigned int aNewWidth, const unsigned int aNewHeight);
 
  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 unsigned int aWidth, const unsigned int aHeight){
    Wcan->SetWindowSize(aWidth + (aWidth- Wcan->GetWw()), aHeight + (aHeight - Wcan->GetWh()));
    Wcan->SetCanvasSize(aWidth,aHeight);
    Wcan->Modified(); Wcan->Update();
  }; 
  
  inline int GetWidth(void){ return Wcan->GetWw(); }; 
  
  inline int GetHeight(void){ return Wcan->GetWh(); }; 
  
  inline void DivideCanvas(const unsigned int aNpads){ Wpad->DivideSquare((int)aNpads); };
 
  inline void DivideCanvas(const unsigned int aNx, const unsigned int aNy){ Wpad->Divide((int)aNx, (int)aNy); };
 
  inline string GetCurrentStyle(void){ return stylename; };
    
  inline unsigned int GetColorPalette(const unsigned int i){ return (unsigned int)gStyle->GetColorPalette((int)i); };
    
  inline unsigned int GetNumberOfColors(void){ return (unsigned int)gStyle->GetNumberOfColors(); };
    
 protected:
  
  int randid;                      
  string srandid;                  
  string name;                     
  
  // STYLE
  string stylename;                
 
  // CANVAS/PAD
  TCanvas *Wcan;                   
  TPad    *Wpad;                   
 
  // ELEMENTS
  TLegend *Wleg;                   
  TText   *Wtext;                  
 
 private:
 
  Monitor *mon;                    
 
  void SetGwollumStyle(void);
 
  void SetPinkStyle(void);
 
  void SetFireStyle(void);
 
  void SetStandardStyle(void);
  
  void SetOxStyle(void);
  
  ClassDef(GwollumPlot,0)
};
 
#endif