style/Tools.C File Reference

Go to the source code of this file.

Functions
int	Tools ()
bool	DrawText (Double_t posX=0., Double_t posY=0., const char *text="", Double_t size=0.08, Int_t col=1)
bool	DrawNice2DHisto (TH2 *h, const char *opt="", double range=10.)
bool	set_nicer_2d_plot_style ()
bool	BetterStatBox (TPad *pad)
bool	ImproveDefaultStyle ()
bool	LoadPandaStyle (bool)
TH1D	TransformHisto (TH2 *h2, double min, double max)
bool	plothistosfromfile (TString filename="histos.root", TString ext=".pdf", Int_t divx=2, Int_t divy=2, Int_t pix=300)
bool	plotntuplefromfile (TString filename="ntps.root", TString ext=".pdf", Int_t divx=2, Int_t divy=2, Int_t pix=300)
bool	LoadManySimFiles (TString treename="pndsim")
TString	InitDefaultRun (TString filetag)
bool	DrawHistSecondScale (TH1 *hist, int color=kRed)

Function Documentation

bool BetterStatBox

(

TPad *

pad

)

Vorgehensweise:

• Histogramme in ein TPad zeichnen (1. Histogramm: Draw(), ntes Histogramm: Draw("sames"))
• die Update() Funktion vom Canvas aufrufen (die Statistikboxen werden erst erzeugt, wenn das Histogramm wirklich gezeichnet wird)
• der Funktion das aktuelle TPad uebergeben (z.B.: TPad* p=(TPad*)gPad; BetterStatBox(p);)

Definition at line 104 of file style/Tools.C.

References hist, i, name, obj, and TString.

                               {

  int entries  = pad->GetListOfPrimitives()->GetEntries();
  TString name = "";
  TPaveStats *st;
  int obj = 1;

  for(int i=0; i<entries; i++){
    name = pad->GetListOfPrimitives()->At(i)->IsA()->GetName();
    if( name.Contains("TH1") ){
      TH1* hist = (TH1*)pad->GetListOfPrimitives()->At(i);
      st = (TPaveStats*)hist->GetListOfFunctions()->FindObject("stats");
      if(st){
        st->SetOptStat(1);
        st->SetX1NDC(0.80);
        st->SetY1NDC(1-0.16*obj) ;
        st->SetY2NDC( st->GetY1NDC()+0.15 ) ;
        st->SetTextColor( hist->GetLineColor() );
        st->SetLineColor( hist->GetLineColor() );
        st->Draw("same");
        obj++;
      }
    }
  }
  return true;
}

bool DrawHistSecondScale	(	TH1 *	hist,
		int	color = kRed
	)

Definition at line 541 of file style/Tools.C.

{
    // scale hint1 to the pad coordinates
   Float_t rightmax = 1.1*hist->GetMaximum();
   Float_t scale = gPad->GetUymax()/rightmax;
   hist->SetLineColor(color);
   hist->Scale(scale);
   hist->Draw("same");

   // draw an axis on the right side
   TGaxis *axis = new TGaxis(gPad->GetUxmax(),gPad->GetUymin(),
   gPad->GetUxmax(), gPad->GetUymax(),0,rightmax,510,"+L");
   axis->SetLineColor(color);
   axis->SetTextColor(color);
   axis->Draw();
   return true;
}

bool DrawNice2DHisto	(	TH2 *	h,
		const char *	opt = "",
		double	range = 10.
	)

Definition at line 67 of file style/Tools.C.

References range, and TString.

Referenced by plothistosfromfile().

{
  // Draw a 2D histo with the rainbow colors and the palette besides
  TString options = "colz"; options += opt;
  if(gPad->GetLogz() == 1 && h->GetMaximum()<range) h->SetAxisRange(0.,range,"Z");
  else {
    // we squeeze here to see better the mean value
    //range = 10*h->Integral()/(h->GetNbinsX()/h->GetNbinsY());
    cout << "range "<<range<<"   max "<<h->GetMaximum()<<"   min "<<h->GetMinimum()<<endl;
    if( range < h->GetMaximum()) range = h->GetMaximum();
    if(-range > h->GetMinimum()) range = -1*h->GetMinimum();
    range *= 1.1; // zoom a bit out to see the max better
    if(h->GetMinimum() < 0.) h->SetAxisRange(-range,range,"Z");
    else h->SetAxisRange(0,range,"Z");
  }
  h->SetStats(kFALSE);
  h->SetTitleOffset(0.8,"T");
  gPad->SetRightMargin(0.15);
  h->DrawCopy(options.Data());
  return true;
}

bool DrawText	(	Double_t	posX = 0.,
		Double_t	posY = 0.,
		const char *	text = "",
		Double_t	size = 0.08,
		Int_t	col = 1
	)

Definition at line 49 of file style/Tools.C.

References col.

Referenced by GaussFitPlot(), and LineFitPlot().

{
  // Drawing some TLatex text in a root canvas, overlaying the content

  TLatex* pText=new TLatex(posX,posY,text);
  pText->SetNDC(kTRUE);
  //   pText->SetNDC(bNDC);
  pText->SetTextColor(col); pText->SetTextSize(size);
  //   pText->SetTextAngle(angle);
  //   pText->SetTextFont(font); pText->SetTextAlign(align);
  pText->Draw();
  return true;
}

bool ImproveDefaultStyle

(

)

Definition at line 151 of file style/Tools.C.

{
  gStyle->SetOptFit ( 1011 );
  gStyle->SetPaperSize(20,26);
  gStyle->SetPadTopMargin(0.14);
  gStyle->SetPadRightMargin(0.05);
  gStyle->SetPadBottomMargin(0.14);
  gStyle->SetPadLeftMargin(0.14);
  gStyle->SetTextFont(22);//changed to bold (R.K.)
  gStyle->SetTextSize(0.08);
  gStyle->SetLabelFont(22,"x");//changed to bold (R.K.)
  gStyle->SetLabelFont(22,"y");//changed to bold (R.K.)
  gStyle->SetLabelFont(22,"z");//changed to bold (R.K.)
  gStyle->SetLabelSize(0.05,"x");
  gStyle->SetTitleSize(0.06,"x");
  gStyle->SetLabelSize(0.05,"y");
  gStyle->SetTitleSize(0.06,"y");
  gStyle->SetLabelSize(0.05,"z");
  gStyle->SetTitleSize(0.06,"z");
  // use bold lines and markers
  gStyle->SetMarkerStyle(8);
  gStyle->SetHistLineWidth(1);//1.85);
  gStyle->SetLineStyleString(2,"[12 12]"); // postscript dashes

  // do not display any of the standard histogram decorations
  gStyle->SetOptTitle(1);
  gStyle->SetOptStat(1);
  gStyle->SetOptFit(1);

  // put tick marks on top and RHS of plots
  gStyle->SetPadTickX(1);
  gStyle->SetPadTickY(1);

  //R.K. abool clumsy axis lables
  gStyle->SetNdivisions(509); // default root value is 510

  return true;
}

TString InitDefaultRun

(

TString

filetag

)

Definition at line 502 of file style/Tools.C.

References fRun, PndFileNameCreator::GetCustomFileName(), PndFileNameCreator::GetParFileName(), PndFileNameCreator::GetRecoFileName(), PndFileNameCreator::GetSimFileName(), namecreator, outFile, parFile, recoFile, rtdb, simFile, sysFile, and TString.

{
  cout << "-I- Using InitDefaultRun() from macro/run/Tools.C with the sim file " << filetag.Data() << endl;
  FairRunAna* fRun = new FairRunAna();
  FairRuntimeDb* rtdb = fRun->GetRuntimeDb();

  filetag.ReplaceAll("_sim.root",".root");
  PndFileNameCreator namecreator(filetag.Data());
  TString simFile = namecreator.GetSimFileName();
  TString parFile = namecreator.GetParFileName();
  TString recoFile = namecreator.GetRecoFileName();
  TString tracksFile = namecreator.GetCustomFileName("tracks");
  TString pidFile = namecreator.GetCustomFileName("pid");
  TString histoFile = namecreator.GetCustomFileName("histos");
  TString evrdummy = namecreator.GetCustomFileName("evrdummy");
  std::cout<<"simFile="<<simFile.Data()<<std::endl;
  std::cout<<"parFile="<<parFile.Data()<<std::endl;
  std::cout<<"recoFile="<<recoFile.Data()<<std::endl;
  std::cout<<"trkFile="<<tracksFile.Data()<<std::endl;
  std::cout<<"pidFile="<<pidFile.Data()<<std::endl;
  fRun->SetInputFile(simFile);
  fRun->AddFriend(recoFile);//,"rec");
  fRun->AddFriend(tracksFile);//,"trk");
  fRun->AddFriend(pidFile);//,"pid");
  FairParRootFileIo* parIO = new FairParRootFileIo();
  parIO->open(parFile.Data());
  rtdb->setFirstInput(parIO);
  rtdb->setOutput(parIO);

  TString outFile = evrdummy;
  TString sysFile = gSystem->Getenv("VMCWORKDIR");

  fRun->SetOutputFile(outFile.Data());
  FairGeane* geane = new FairGeane();
  fRun->AddTask(geane);
  fRun->Init();
  return histoFile;
}

bool LoadManySimFiles

(

TString

treename = "pndsim"

)

Definition at line 488 of file style/Tools.C.

References fi, next, and R.

{ // to use that method you should have opened some files
  // containing the same tree structure, like splitted files of
  // mass production simulations. (like "root -f data/sim01*.root"
  // tree examination is available via TTree::Draw(...)

  TIter next(gROOT->GetListOfFiles());
  TFile *fi=0;
  TChain *R=new TChain(treename.Data());
  while ( (fi=(TFile*)next()) ) R->Add(fi->GetName());
  cout<<(Int_t)R->GetEntries()<<endl;
  return true;
}

bool LoadPandaStyle

(

bool

)

Definition at line 190 of file style/Tools.C.

References set_nicer_2d_plot_style().

{

  //--------------------------------------------------------------------------
  // File and Version Information:
  //      $Id: PBase.C,v 1.8 2006/09/22 12:04:13 kliemt Exp $
  //
  // Description:
  //      Initialization code executed at the start of a ROOT session.
  //      Set up the Panda style for approved plots.
  //
  // Environment:
  //      Software developed for the PANDA Detector at GSI, Darmstadt
  //
  // Author List:
  //      Sergey Ganzhur                Original Author
  //      Ralf Kliemt (2008)            Small adjustments for PandaRoot use
  //
  // Copyright Information:
  //     Copyright (C) 2001-2002        Ruhr Universitaet Bochum
  //
  //------------------------------------------------------------------------

  if( strcmp(gStyle->GetName(),"PANDA")==0 ) return true;
  // use the 'plain' style for plots (white backgrounds, etc)
  //cout << "...using style 'Plain'" << endl;
  //gROOT->SetStyle("Plain");

  // Create the 'PANDA' style for approved plots. Note that this style may need
  // some fine tuning in your macro depending on what you are plotting, e.g.
  //
  //  gStyle->SetMarkerSize(0.75);  // use smaller markers in a histogram with many bins
  //  gStyle->SetTitleOffset(0.65,"y");  // bring y axis label closer to narrow values

  // Ralf Kliemt:
  // I changed a bit for myself here

  TStyle *pandaStyle= new TStyle("PANDA","PANDA approved plots style, modified by ralfk");

  // use plain black on white colors
  pandaStyle->SetFrameBorderMode(0);
  pandaStyle->SetCanvasBorderMode(0);
  pandaStyle->SetPadBorderMode(0);
  pandaStyle->SetPadColor(0);
  pandaStyle->SetCanvasColor(0);
  pandaStyle->SetTitleColor(1);
  pandaStyle->SetStatColor(0);
  //   pandaStyle->SetFillColor(0);// conflict with 2D plots
  //   pandaStyle->SetPalette(1);
  //R.K: Now remove the title box
  //pandaStyle->SetTitleAlign(33);
  pandaStyle->SetTitleBorderSize(0);
  pandaStyle->SetTitleColor(1);
  pandaStyle->SetTitleFillColor(0);
  pandaStyle->SetTitleFontSize(0.07);

  // set the paper & margin sizes
  pandaStyle->SetPaperSize(20,26);
  pandaStyle->SetPadTopMargin(0.14);
  pandaStyle->SetPadRightMargin(0.05);
  pandaStyle->SetPadBottomMargin(0.14);
  pandaStyle->SetPadLeftMargin(0.14);
//  pandaStyle->SetPaperSize(20,26);
//  pandaStyle->SetPadTopMargin(0.05);
//  pandaStyle->SetPadRightMargin(0.05);
//  pandaStyle->SetPadBottomMargin(0.16);
//  pandaStyle->SetPadLeftMargin(0.12);

  // use large Times-Roman fonts
  //   pandaStyle->SetTextFont(132);
  //   pandaStyle->SetLabelFont(132,"x");
  //   pandaStyle->SetLabelFont(132,"y");
  //   pandaStyle->SetLabelFont(132,"z");
  pandaStyle->SetTextFont(22);//changed to bold (R.K.)
  pandaStyle->SetTextSize(0.08);
  pandaStyle->SetLabelFont(22,"x");//changed to bold (R.K.)
  pandaStyle->SetLabelFont(22,"y");//changed to bold (R.K.)
  pandaStyle->SetLabelFont(22,"z");//changed to bold (R.K.)
  pandaStyle->SetLabelSize(0.05,"x");
  pandaStyle->SetTitleSize(0.06,"x");
  pandaStyle->SetLabelSize(0.05,"y");
  pandaStyle->SetTitleSize(0.06,"y");
  pandaStyle->SetLabelSize(0.05,"z");
  pandaStyle->SetTitleSize(0.06,"z");

  // use bold lines and markers
  pandaStyle->SetMarkerStyle(8);
  pandaStyle->SetHistLineWidth(1);//2
  pandaStyle->SetLineStyleString(2,"[12 12]"); // postscript dashes

  // do not display any of the standard histogram decorations
  pandaStyle->SetOptTitle(1);
  pandaStyle->SetOptStat(1);
  pandaStyle->SetOptFit(1);

  // put tick marks on top and RHS of plots
  pandaStyle->SetPadTickX(1);
  pandaStyle->SetPadTickY(1);

  //R.K. abool clumsy axis lables
  pandaStyle->SetNdivisions(509); // default root value is 510

  //cout <<"    For approved plots use: gROOT->SetStyle(\"PANDA\");"<< endl;

  pandaStyle->cd();
  gROOT->ForceStyle();
  set_nicer_2d_plot_style();//[R.K.] use nicer 2D plots
  return true;
}//bool PBase::LoadPandaStyle(bool)

bool plothistosfromfile	(	TString	filename = "histos.root",
		TString	ext = ".pdf",
		Int_t	divx = 2,
		Int_t	divy = 2,
		Int_t	pix = 300
	)

Definition at line 318 of file style/Tools.C.

References DrawNice2DHisto(), ext, file, filename, i, set_nicer_2d_plot_style(), and TString.

Referenced by plotall().

{ // Plot all histograms into a ps file
  // works with TH1, TH2, & TProfile
  //LoadPandaStyle();
  set_nicer_2d_plot_style();
  TFile* file = new TFile(filename.Data());
  if (!file) {cout<<"File \""<<filename.Data()<<"\" is not there..."<<endl;return true;}
  TCanvas* can = new TCanvas();
  Int_t pixx = ceil(1.4*pix*divx);
  Int_t pixy = pix*divy;
  can->SetCanvasSize(pixx,pixy);
  can->Divide(divx, divy);
  TString picname = filename;
  ext="."+ext;
  ext.ReplaceAll("..",".");
  picname.ReplaceAll(".root",ext); // ps, png, pdf ...
  TString pic = picname + "["; // open empty ps
  cout << "opening: " << pic.Data()<<endl;
  can->Print(pic);
  pic=picname;

  TList* list = file->GetListOfKeys();
  if (!list) {cout<<"List not there..."<<endl;return true;}
  int padcount = 1;
  TString keyclass="";
  for(int i=0;i<list->GetEntries();i++)
  {
    if(padcount > divx*divy)
    {
      can->Print(pic.Data());
      can->Clear();
      can->SetCanvasSize(pixx,pixy);
      can->Divide(divx, divy);
      padcount=1;
    }
    can->cd(padcount);
    TKey* key = (TKey*)list->At(i);
    keyclass = key->GetClassName();
    //cout<<keyclass.Data()<<endl;
    if(keyclass.Contains("TH1"))
    {
      //cout<<"try plotting a TH1"<<endl;
      TH1* his = (TH1*)key->ReadObj();
      his->GetXaxis()->SetNoExponent(); // put exponents to numbers directly
      his->GetYaxis()->SetNoExponent(); // put exponents to numbers directly
      TString options(his->GetOption());
      if(options.Contains("log")){
        gPad->SetLogy();
        options.ReplaceAll("log","");
        his->SetOption(options.Data());
      }
      his->Draw();
    }else if(keyclass.Contains("TH2"))
    {
      //cout<<"try plotting a TH2"<<endl;
      TH2* his2 = (TH2*)key->ReadObj();
      his2->GetXaxis()->SetNoExponent(); // put exponents to numbers directly
      his2->GetYaxis()->SetNoExponent(); // put exponents to numbers directly
      TString options(his2->GetOption());
      if(options.Contains("log")){
        gPad->SetLogz();
        options.ReplaceAll("log","");
        his2->SetOption(options.Data());
      }
      if(options.Contains("nice")){
        options.ReplaceAll("nice","");
        his2->SetOption(options.Data());
        DrawNice2DHisto(his2);
      } else {
        his2->Draw("colz");
      }
    }else if(keyclass.Contains("TProfile"))
    {
      //cout<<"try plotting a TH2"<<endl;
      TProfile* hpro = (TProfile*)key->ReadObj();
      hpro->GetXaxis()->SetNoExponent(); // put exponents to numbers directly
      hpro->GetYaxis()->SetNoExponent(); // put exponents to numbers directly
      TString options(hpro->GetOption());
      if(options.Contains("log")){
        gPad->SetLogy();
        options.ReplaceAll("log","");
        hpro->SetOption(options.Data());
      }
      hpro->Draw();
    } else continue;

    padcount++;
  }

  can->Print(pic.Data());
  pic = picname + "]"; // close ps
  can->Print(pic.Data());
  cout << "closed: " << pic.Data()<<endl;
  //TString convertcmd = "test -r ps2pdf && ps2pdf ";
  //convertcmd += pic.Data();
  //gSystem->Exec(convertcmd.Data());
  delete can;
  return true;
}

bool plotntuplefromfile	(	TString	filename = "ntps.root",
		TString	ext = ".pdf",
		Int_t	divx = 2,
		Int_t	divy = 2,
		Int_t	pix = 300
	)

Definition at line 418 of file style/Tools.C.

References ext, file, filename, i, printf(), and TString.

{
  // Plot all columns of an NTuple into a pdf (or ps file)
  TFile* file = new TFile(filename.Data());
  if (!file) {cout<<"File \""<<filename.Data()<<"\" is not there..."<<endl;return true;}
  TCanvas* can = new TCanvas();
  Int_t pixx = ceil(1.4*pix*divx);
  Int_t pixy = pix*divy;
  can->SetCanvasSize(pixx,pixy);
  can->Divide(divx, divy);
  TString picname = filename;
  ext="."+ext;
  ext.ReplaceAll("..",".");
  picname.ReplaceAll(".root",ext); // ps, png, pdf ...
  TString pic = picname + "["; // open empty ps
  cout << "opening: " << pic.Data()<<endl;
  can->Print(pic);
  pic=picname;

  TList* list = file->GetListOfKeys();
  if (!list) {cout<<"List not there..."<<endl;return true;}
  int padcount = 1;
  TString keyclass="";
  for(int i=0;i<list->GetEntries();i++)
  {
    TKey* key = (TKey*)list->At(i);
    keyclass = key->GetClassName();
    if(keyclass.CompareTo("TTree"))continue;
    TTree* baum = (TTree*)key->ReadObj();
    printf("\t - Tree: %s\n",baum->GetName());
    TObjArray* aeste = baum->GetListOfBranches();
    for(int asti=0;asti<aeste->GetEntriesFast();asti++)
    {
      TBranch* ast = (TBranch*)aeste->At(asti);
      printf("\t\t - Branch: %s\n",ast->GetName());
      TObjArray* bleatter = ast->GetListOfLeaves();
      for(int blatti=0;blatti<bleatter->GetEntriesFast();blatti++)
      {
        TLeaf* blatt = (TLeaf*)bleatter->At(blatti);
        TString blattsorte=blatt->GetTypeName();
        if( !blattsorte.CompareTo("Float_t") || !blattsorte.CompareTo("Int_t") || !blattsorte.CompareTo("Double_t"))
        {
          if(padcount > divx*divy)
          {
            printf("   --- canvasdump ---");
            can->Print(pic.Data());
            can->Clear();
            can->SetCanvasSize(pixx,pixy);
            can->Divide(divx, divy);
            padcount=1;
          }
          can->cd(padcount);
          printf("\t\t\t - Leaf: %s  (%s)\n",blatt->GetName(),blatt->GetTypeName());
          baum->Draw(blatt->GetName());
          padcount++;
        }
      }
    }
  }
  can->Print(pic.Data());
  pic = picname + "]"; // close ps
  can->Print(pic.Data());
  cout << "closed: " << pic.Data()<<endl;
//   TString convertcmd = "test -r ps2pdf && ps2pdf ";
//   convertcmd += pic.Data();
//   gSystem->Exec(convertcmd.Data());
  delete can;
  return true;
}

bool set_nicer_2d_plot_style

(

)

Definition at line 89 of file style/Tools.C.

References Double_t.

Referenced by LoadPandaStyle(), and plothistosfromfile().

{
  const Int_t NRGBs = 5;
  const Int_t NCont = 99;//255;

  Double_t stops[NRGBs] = { 0.00, 0.34, 0.61, 0.84, 1.00 };
  Double_t red[NRGBs]   = { 0.00, 0.00, 0.87, 1.00, 0.51 };
  Double_t green[NRGBs] = { 0.00, 0.81, 1.00, 0.20, 0.00 };
  Double_t blue[NRGBs]  = { 0.51, 1.00, 0.12, 0.00, 0.00 };
  TColor::CreateGradientColorTable(NRGBs, stops, red, green, blue, NCont);
  //     gStyle->CreateGradientColorTable(NRGBs, stops, red, green, blue, NCont);
  gStyle->SetNumberContours(NCont);
  return true;
}

int Tools

(

)

Definition at line 43 of file style/Tools.C.

{
  cout<<"pandaroot/macro/run/Tools.C loaded. Enjoy it."<<endl;
  return 0;
}

TH1D TransformHisto	(	TH2 *	h2,
		double	min,
		double	max
	)

Definition at line 300 of file style/Tools.C.

References i, and nbins.

Referenced by drawMvdHistos().

{
  /*
   *  Created on: Feb 25, 2009
   *      Author: stockman
   */

  TH1D result("h1","h1", 1000, min, max);
  int nbins = h2->GetNbinsX() * h2->GetNbinsY();
  for (int i = 0; i < nbins; i++){
    //std::cout << h2->GetBinContent(i) << std::endl;
    result.Fill(h2->GetBinContent(i));
    if (i == 10)
      cout << h2->GetBinContent(i) << endl;
  }
  return result;
}