[u/mrichter/AliRoot.git] / PWGPP / ZDC / macros / MakeTrendZDC.C

#if !defined(__CINT__) || defined(__MAKECINT__)
#include <stdio.h>
#include <stdlib.h>
#include <TROOT.h>
#include <Riostream.h>
#include <TClassTable.h>
#include <TStyle.h>
#include <TMath.h>
#include <TFile.h>
#include <TCanvas.h>
#include <TH1.h>
#include <TH2.h>
#include <TProfile.h>
#include <TLine.h>
#include <TGrid.h>
#include <TBits.h>
#include <TChain.h>
#include <TNtuple.h>
#include <TTree.h>
#include <TBranch.h>
#include <TFileMerger.h>
#include <TGridResult.h>
#include <TSystem.h>
#include <TGaxis.h>
#endif

int MakeTrendZDC(char *infile, int run) {
  
  TStopwatch timer;
  timer.Start();
  gSystem->SetIncludePath("-I$ROOTSYS/include -I$ALICE_ROOT/include -I$ALICE_ROOT");
  gSystem->Load("libSTAT");
  gSystem->Load("libANALYSIS");
  gSystem->Load("libANALYSISalice");
  gSystem->Load("libANALYSIScalib");
  gSystem->Load("libCORRFW");
  gSystem->Load("libANALYSISalice");
  gSystem->Load("libANALYSIScalib");
  gSystem->Load("libTender");
  gSystem->Load("libPWGPP");
  
  char *outfile = "trending.root";
  
  if (!infile) return -1;
  if (!outfile) return -1;
  TFile *f =0;
  f=TFile::Open(infile,"read");
  if (!f) {
    printf("File %s not available\n", infile);
    return -1;
  }  
   
  //access histograms lists
  char zdcQAdirName[20]="ZDC_Performance";
  char genListName[20]="QAZDCHists";
  
  TDirectoryFile * zdcQAdir=(TDirectoryFile*)f->Get(zdcQAdirName);
  if (!zdcQAdir) {
    Printf("ERROR: ZDC QA directory not present in input file.\n");
    return -1;
  }
  
  TList * generalList=(TList*)zdcQAdir->Get(genListName);

  if (!generalList) Printf("WARNING: general QA histograms absent or not accessible\n");

  TH1D    *fhTDCZNC          = (TH1D*)generalList->FindObject("fhTDCZNC");       //! TDC ZNC sum
  TH1D    *fhTDCZNA          = (TH1D*)generalList->FindObject("fhTDCZNA");       //! TDC DIFF sum
  TH1D    *fhTDCZNSum        = (TH1D*)generalList->FindObject("fhTDCZNSum");     //! TDC ZNC sum
  TH1D    *fhTDCZNDiff       = (TH1D*)generalList->FindObject("fhTDCZNDiff");    //! TDC DIFF sum
  TH1D    *fhZNCSumQ         = (TH1D*)generalList->FindObject("fhZNCSumQ");      //! ZNC sum 4Q
  TH1D    *fhZNASumQ         = (TH1D*)generalList->FindObject("fhZNASumQ");      //! ZNA sum 4Q
  TH1D    *fhZPCSumQ         = (TH1D*)generalList->FindObject("fhZPCSumQ");      //! ZPC sum 4Q
  TH1D    *fhZPASumQ         = (TH1D*)generalList->FindObject("fhZPASumQ");      //! ZPA sum 4Q
  TH1D    *fhZEM1Spectrum    = (TH1D*)generalList->FindObject("fhZEM1Spectrum"); //! ZEM1 spectra
  TH1D    *fhZEM2Spectrum    = (TH1D*)generalList->FindObject("fhZEM2Spectrum"); //! ZEM2 spectra
  TH1D    *fhZNCpmc          = (TH1D*)generalList->FindObject("fhZNCpmc");       //! ZNC PMCs
  TH1D    *fhZNApmc          = (TH1D*)generalList->FindObject("fhZNApmc");       //! ZNA PMCs
  TH1D    *fhZPCpmc          = (TH1D*)generalList->FindObject("fhZPCpmc");       //! ZPC PMCs
  TH1D    *fhZPApmc          = (TH1D*)generalList->FindObject("fhZPApmc");       //! ZPA PMCs
  TH1D    *fhPMCZNCemd       = (TH1D*)generalList->FindObject("fhPMCZNCemd");    //! ZNC PMC low gain chain
  TH1D    *fhPMCZNAemd       = (TH1D*)generalList->FindObject("fhPMCZNAemd");    //! ZNA PMC low gain chain
  TH1D    *fhTDCZNAcorr      = (TH1D*)generalList->FindObject("fhTDCZNAcorr");   //! ZNA corrected TDC
  TH1D    *fhTDCZNCcorr      = (TH1D*)generalList->FindObject("fhTDCZNCcorr");   //! ZNC corrected TDC  
  TH2D    *fhZNCCentroid     = (TH2D*)generalList->FindObject("fhZNCCentroid");  //! ZNC centroid
  TH2D    *fhZNACentroid     = (TH2D*)generalList->FindObject("fhZNACentroid");  //! ZNA centroid
  TH2D    *fDebunch          = (TH2D*)generalList->FindObject("fDebunch");       //! TDC sum vs. diff  
  
  Double_t ZNC_mean = 0.;
  Double_t ZNA_mean = 0.;
  Double_t ZPC_mean = 0.;
  Double_t ZPA_mean = 0.;
  Double_t ZEM1_mean = 0.;
  Double_t ZEM2_mean = 0.;  
  
  Int_t fhZNCpmcEntries = fhZNCpmc->GetEntries();
  Int_t fhZNApmcEntries = fhZNApmc->GetEntries();
  Int_t fhZPCpmcEntries = fhZPCpmc->GetEntries();
  Int_t fhZPApmcEntries = fhZPApmc->GetEntries();
  Int_t fhZEM1SpectrumEntries = fhZEM1Spectrum->GetEntries();
  Int_t fhZEM2SpectrumEntries = fhZEM2Spectrum->GetEntries();  
  
  if (fhZNCpmcEntries>0) ZNC_mean = fhZNCpmc->GetMean();
  if (fhZNApmcEntries>0) ZNA_mean = fhZNApmc->GetMean();
  if (fhZPCpmcEntries>0) ZPC_mean = fhZPCpmc->GetMean();
  if (fhZPApmcEntries>0) ZPA_mean = fhZPApmc->GetMean();
  if (fhZEM1SpectrumEntries>0) ZEM1_mean = fhZEM1Spectrum->GetMean();
  if (fhZEM2SpectrumEntries>0) ZEM2_mean = fhZEM2Spectrum->GetMean(); 
  
  Double_t ZNC_XCent = fhZNCCentroid->GetMean(1);
  Double_t ZNC_YCent = fhZNCCentroid->GetMean(2);    
  Double_t ZNA_XCent = fhZNACentroid->GetMean(1);
  Double_t ZNA_YCent = fhZNACentroid->GetMean(2);
  Double_t ZNC_XCent_err = fhZNCCentroid->GetRMS(1);
  Double_t ZNC_YCent_err = fhZNCCentroid->GetRMS(2);    
  Double_t ZNA_XCent_err = fhZNACentroid->GetRMS(1);
  Double_t ZNA_YCent_err = fhZNACentroid->GetRMS(2);     
  Double_t ZN_TDC_Sum = fhTDCZNSum->GetMean();
  Double_t ZN_TDC_Diff = fhTDCZNDiff->GetMean();  
  Double_t ZN_TDC_Sum_err = fhTDCZNSum->GetRMS();
  Double_t ZN_TDC_Diff_err = fhTDCZNDiff->GetRMS();    
  
  TTree * ttree=new TTree("trending","tree of trending variables");
  ttree->Branch("run",&run,"run/I");
  ttree->Branch("ZNC_mean_value",&ZNC_mean,"ZNC_mean_value/D");
  ttree->Branch("ZNA_mean_value",&ZNA_mean,"ZNA_mean_value/D"); 
  ttree->Branch("ZPC_mean_value",&ZPC_mean,"ZPC_mean_value/D");
  ttree->Branch("ZPA_mean_value",&ZPA_mean,"ZPA_mean_value/D");
  ttree->Branch("ZEM1_mean_value",&ZEM1_mean,"ZEM1_mean_value/D");
  ttree->Branch("ZEM2_mean_value",&ZEM2_mean,"ZEM2_mean_value/D");  
  ttree->Branch("ZNC_X_Centroid",&ZNC_XCent,"ZNC_X_Centroid/D");
  ttree->Branch("ZNC_Y_Centroid",&ZNC_YCent,"ZNC_Y_Centroid/D");   
  ttree->Branch("ZNA_X_Centroid",&ZNA_XCent,"ZNA_X_Centroid/D");
  ttree->Branch("ZNA_Y_Centroid",&ZNA_YCent,"ZNA_Y_Centroid/D");  
  ttree->Branch("ZNC_X_Centroid_Err",&ZNC_XCent_err,"ZNC_X_Centroid_Err/D");
  ttree->Branch("ZNC_Y_Centroid_Err",&ZNC_YCent_err,"ZNC_Y_Centroid_Err/D");   
  ttree->Branch("ZNA_X_Centroid_Err",&ZNA_XCent_err,"ZNA_X_Centroid_Err/D");
  ttree->Branch("ZNA_Y_Centroid_Err",&ZNA_YCent_err,"ZNA_Y_Centroid_Err/D");    
  ttree->Branch("ZN_TDC_Sum",&ZN_TDC_Sum,"ZN_TDC_Sum/D");
  ttree->Branch("ZN_TDC_Diff",&ZN_TDC_Diff,"ZN_TDC_Diff/D");    
  ttree->Branch("ZN_TDC_Sum_Err",&ZN_TDC_Sum_err,"ZN_TDC_Sum_Err/D");
  ttree->Branch("ZN_TDC_Diff_Err",&ZN_TDC_Diff_err,"ZN_TDC_Diff_Err/D");  
    
  Printf(":::: Getting post-analysis info for run %i",run);
  TFile * trendFile = new TFile(outfile,"recreate");

  printf("==============  Saving histograms for run %i ===============\n",run);
  
  fhTDCZNC->Write();      
  fhTDCZNA->Write();      
  fhTDCZNSum->Write();    
  fhTDCZNDiff->Write();   
  fhZNCSumQ->Write();     
  fhZNASumQ->Write();     
  fhZPCSumQ->Write();     
  fhZPASumQ->Write();     
  fhZEM1Spectrum->Write();
  fhZEM2Spectrum->Write();
  fhZNCpmc->Write();      
  fhZNApmc->Write();      
  fhZPCpmc->Write();      
  fhZPApmc->Write();      
  fhZNCCentroid->Write(); 
  fhZNACentroid->Write(); 
  fhPMCZNCemd->Write();   
  fhPMCZNAemd->Write();   
  fDebunch->Write();      
  fhTDCZNAcorr->Write();  
  fhTDCZNCcorr->Write();
    
  ttree->Fill();
  printf("==============  Saving trending quantities in tree for run %i ===============\n",run);
  
  trendFile->cd();
  ttree->Write();
  trendFile->Close();

}
Commit	Line	Data
2845593c	1	#if !defined(__CINT__) \|\| defined(__MAKECINT__)
2845593c	2	#include <stdio.h>
	3	#include <stdlib.h>
	4	#include <TROOT.h>
	5	#include <Riostream.h>
	6	#include <TClassTable.h>
	7	#include <TStyle.h>
	8	#include <TMath.h>
	9	#include <TFile.h>
	10	#include <TCanvas.h>
	11	#include <TH1.h>
	12	#include <TH2.h>
	13	#include <TProfile.h>
	14	#include <TLine.h>
	15	#include <TGrid.h>
	16	#include <TBits.h>
	17	#include <TChain.h>
	18	#include <TNtuple.h>
	19	#include <TTree.h>
	20	#include <TBranch.h>
	21	#include <TFileMerger.h>
	22	#include <TGridResult.h>
	23	#include <TSystem.h>
	24	#include <TGaxis.h>
2845593c	25	#endif
2845593c	26
343d2002	27	int MakeTrendZDC(char *infile, int run) {
2845593c	28
343d2002	29	TStopwatch timer;
	30	timer.Start();
	31	gSystem->SetIncludePath("-I$ROOTSYS/include -I$ALICE_ROOT/include -I$ALICE_ROOT");
	32	gSystem->Load("libSTAT");
	33	gSystem->Load("libANALYSIS");
	34	gSystem->Load("libANALYSISalice");
	35	gSystem->Load("libANALYSIScalib");
	36	gSystem->Load("libCORRFW");
4070f709	37	gSystem->Load("libANALYSISalice");
	38	gSystem->Load("libANALYSIScalib");
	39	gSystem->Load("libTender");
	40	gSystem->Load("libPWGPP");
343d2002	41
343d2002	42	char *outfile = "trending.root";
2845593c	43
343d2002	44	if (!infile) return -1;
	45	if (!outfile) return -1;
	46	TFile *f =0;
	47	f=TFile::Open(infile,"read");
	48	if (!f) {
	49	printf("File %s not available\n", infile);
	50	return -1;
	51	}
	52
2845593c	53	//access histograms lists
	54	char zdcQAdirName[20]="ZDC_Performance";
	55	char genListName[20]="QAZDCHists";
	56
343d2002	57	TDirectoryFile * zdcQAdir=(TDirectoryFile*)f->Get(zdcQAdirName);
2845593c	58	if (!zdcQAdir) {
	59	Printf("ERROR: ZDC QA directory not present in input file.\n");
	60	return -1;
	61	}
	62
	63	TList * generalList=(TList*)zdcQAdir->Get(genListName);
	64
	65	if (!generalList) Printf("WARNING: general QA histograms absent or not accessible\n");
	66
343d2002	67	TH1D fhTDCZNC = (TH1D)generalList->FindObject("fhTDCZNC"); //! TDC ZNC sum
	68	TH1D fhTDCZNA = (TH1D)generalList->FindObject("fhTDCZNA"); //! TDC DIFF sum
	69	TH1D fhTDCZNSum = (TH1D)generalList->FindObject("fhTDCZNSum"); //! TDC ZNC sum
	70	TH1D fhTDCZNDiff = (TH1D)generalList->FindObject("fhTDCZNDiff"); //! TDC DIFF sum
	71	TH1D fhZNCSumQ = (TH1D)generalList->FindObject("fhZNCSumQ"); //! ZNC sum 4Q
	72	TH1D fhZNASumQ = (TH1D)generalList->FindObject("fhZNASumQ"); //! ZNA sum 4Q
	73	TH1D fhZPCSumQ = (TH1D)generalList->FindObject("fhZPCSumQ"); //! ZPC sum 4Q
	74	TH1D fhZPASumQ = (TH1D)generalList->FindObject("fhZPASumQ"); //! ZPA sum 4Q
	75	TH1D fhZEM1Spectrum = (TH1D)generalList->FindObject("fhZEM1Spectrum"); //! ZEM1 spectra
	76	TH1D fhZEM2Spectrum = (TH1D)generalList->FindObject("fhZEM2Spectrum"); //! ZEM2 spectra
	77	TH1D fhZNCpmc = (TH1D)generalList->FindObject("fhZNCpmc"); //! ZNC PMCs
	78	TH1D fhZNApmc = (TH1D)generalList->FindObject("fhZNApmc"); //! ZNA PMCs
	79	TH1D fhZPCpmc = (TH1D)generalList->FindObject("fhZPCpmc"); //! ZPC PMCs
	80	TH1D fhZPApmc = (TH1D)generalList->FindObject("fhZPApmc"); //! ZPA PMCs
	81	TH1D fhPMCZNCemd = (TH1D)generalList->FindObject("fhPMCZNCemd"); //! ZNC PMC low gain chain
	82	TH1D fhPMCZNAemd = (TH1D)generalList->FindObject("fhPMCZNAemd"); //! ZNA PMC low gain chain
	83	TH1D fhTDCZNAcorr = (TH1D)generalList->FindObject("fhTDCZNAcorr"); //! ZNA corrected TDC
	84	TH1D fhTDCZNCcorr = (TH1D)generalList->FindObject("fhTDCZNCcorr"); //! ZNC corrected TDC
	85	TH2D fhZNCCentroid = (TH2D)generalList->FindObject("fhZNCCentroid"); //! ZNC centroid
	86	TH2D fhZNACentroid = (TH2D)generalList->FindObject("fhZNACentroid"); //! ZNA centroid
	87	TH2D fDebunch = (TH2D)generalList->FindObject("fDebunch"); //! TDC sum vs. diff
	88
	89	Double_t ZNC_mean = 0.;
	90	Double_t ZNA_mean = 0.;
	91	Double_t ZPC_mean = 0.;
	92	Double_t ZPA_mean = 0.;
	93	Double_t ZEM1_mean = 0.;
	94	Double_t ZEM2_mean = 0.;
2845593c	95
7460a57d	96	Int_t fhZNCpmcEntries = fhZNCpmc->GetEntries();
	97	Int_t fhZNApmcEntries = fhZNApmc->GetEntries();
	98	Int_t fhZPCpmcEntries = fhZPCpmc->GetEntries();
	99	Int_t fhZPApmcEntries = fhZPApmc->GetEntries();
7460a57d	100	Int_t fhZEM1SpectrumEntries = fhZEM1Spectrum->GetEntries();
343d2002	101	Int_t fhZEM2SpectrumEntries = fhZEM2Spectrum->GetEntries();
	102
	103	if (fhZNCpmcEntries>0) ZNC_mean = fhZNCpmc->GetMean();
	104	if (fhZNApmcEntries>0) ZNA_mean = fhZNApmc->GetMean();
	105	if (fhZPCpmcEntries>0) ZPC_mean = fhZPCpmc->GetMean();
	106	if (fhZPApmcEntries>0) ZPA_mean = fhZPApmc->GetMean();
	107	if (fhZEM1SpectrumEntries>0) ZEM1_mean = fhZEM1Spectrum->GetMean();
	108	if (fhZEM2SpectrumEntries>0) ZEM2_mean = fhZEM2Spectrum->GetMean();
	109
2845593c	110	Double_t ZNC_XCent = fhZNCCentroid->GetMean(1);
	111	Double_t ZNC_YCent = fhZNCCentroid->GetMean(2);
	112	Double_t ZNA_XCent = fhZNACentroid->GetMean(1);
	113	Double_t ZNA_YCent = fhZNACentroid->GetMean(2);
	114	Double_t ZNC_XCent_err = fhZNCCentroid->GetRMS(1);
	115	Double_t ZNC_YCent_err = fhZNCCentroid->GetRMS(2);
	116	Double_t ZNA_XCent_err = fhZNACentroid->GetRMS(1);
	117	Double_t ZNA_YCent_err = fhZNACentroid->GetRMS(2);
	118	Double_t ZN_TDC_Sum = fhTDCZNSum->GetMean();
	119	Double_t ZN_TDC_Diff = fhTDCZNDiff->GetMean();
	120	Double_t ZN_TDC_Sum_err = fhTDCZNSum->GetRMS();
	121	Double_t ZN_TDC_Diff_err = fhTDCZNDiff->GetRMS();
	122
	123	TTree * ttree=new TTree("trending","tree of trending variables");
343d2002	124	ttree->Branch("run",&run,"run/I");
2845593c	125	ttree->Branch("ZNC_mean_value",&ZNC_mean,"ZNC_mean_value/D");
	126	ttree->Branch("ZNA_mean_value",&ZNA_mean,"ZNA_mean_value/D");
	127	ttree->Branch("ZPC_mean_value",&ZPC_mean,"ZPC_mean_value/D");
	128	ttree->Branch("ZPA_mean_value",&ZPA_mean,"ZPA_mean_value/D");
	129	ttree->Branch("ZEM1_mean_value",&ZEM1_mean,"ZEM1_mean_value/D");
	130	ttree->Branch("ZEM2_mean_value",&ZEM2_mean,"ZEM2_mean_value/D");
	131	ttree->Branch("ZNC_X_Centroid",&ZNC_XCent,"ZNC_X_Centroid/D");
	132	ttree->Branch("ZNC_Y_Centroid",&ZNC_YCent,"ZNC_Y_Centroid/D");
	133	ttree->Branch("ZNA_X_Centroid",&ZNA_XCent,"ZNA_X_Centroid/D");
	134	ttree->Branch("ZNA_Y_Centroid",&ZNA_YCent,"ZNA_Y_Centroid/D");
	135	ttree->Branch("ZNC_X_Centroid_Err",&ZNC_XCent_err,"ZNC_X_Centroid_Err/D");
	136	ttree->Branch("ZNC_Y_Centroid_Err",&ZNC_YCent_err,"ZNC_Y_Centroid_Err/D");
	137	ttree->Branch("ZNA_X_Centroid_Err",&ZNA_XCent_err,"ZNA_X_Centroid_Err/D");
	138	ttree->Branch("ZNA_Y_Centroid_Err",&ZNA_YCent_err,"ZNA_Y_Centroid_Err/D");
	139	ttree->Branch("ZN_TDC_Sum",&ZN_TDC_Sum,"ZN_TDC_Sum/D");
	140	ttree->Branch("ZN_TDC_Diff",&ZN_TDC_Diff,"ZN_TDC_Diff/D");
	141	ttree->Branch("ZN_TDC_Sum_Err",&ZN_TDC_Sum_err,"ZN_TDC_Sum_Err/D");
	142	ttree->Branch("ZN_TDC_Diff_Err",&ZN_TDC_Diff_err,"ZN_TDC_Diff_Err/D");
	143
343d2002	144	Printf(":::: Getting post-analysis info for run %i",run);
343d2002	145	TFile * trendFile = new TFile(outfile,"recreate");
2845593c	146
343d2002	147	printf("============== Saving histograms for run %i ===============\n",run);
2845593c	148
	149	fhTDCZNC->Write();
	150	fhTDCZNA->Write();
	151	fhTDCZNSum->Write();
	152	fhTDCZNDiff->Write();
	153	fhZNCSumQ->Write();
	154	fhZNASumQ->Write();
	155	fhZPCSumQ->Write();
	156	fhZPASumQ->Write();
	157	fhZEM1Spectrum->Write();
	158	fhZEM2Spectrum->Write();
	159	fhZNCpmc->Write();
	160	fhZNApmc->Write();
	161	fhZPCpmc->Write();
	162	fhZPApmc->Write();
	163	fhZNCCentroid->Write();
	164	fhZNACentroid->Write();
	165	fhPMCZNCemd->Write();
	166	fhPMCZNAemd->Write();
	167	fDebunch->Write();
	168	fhTDCZNAcorr->Write();
	169	fhTDCZNCcorr->Write();
	170
	171	ttree->Fill();
343d2002	172	printf("============== Saving trending quantities in tree for run %i ===============\n",run);
2845593c	173
	174	trendFile->cd();
	175	ttree->Write();
	176	trendFile->Close();
	177
7460a57d	178	}