[u/mrichter/AliRoot.git] / MFT / AliMFTClusterQA.cxx

#include "TObject.h"
#include "AliRunLoader.h"
#include "AliRun.h"
#include "AliLoader.h"
#include "AliMFT.h"
#include "TClonesArray.h"
#include "AliMFTCluster.h"
#include "AliMFTSegmentation.h"
#include "TFile.h"
#include "TH1D.h"
#include "AliLog.h"
#include "TString.h"

#include "AliMFTClusterQA.h"

//====================================================================================================================================================
//
// Class for the analysis of the MFT clusters (a.k.a. rec points). Few QA histograms are created
//
// Contact author: antonio.uras@cern.ch
//
//====================================================================================================================================================

ClassImp(AliMFTClusterQA)

//====================================================================================================================================================

AliMFTClusterQA::AliMFTClusterQA():
  TObject(),
  fMFTLoader(0),
  fRunLoader(0),
  fMFT(0),
  fNPlanes(0),
  fNEvents(0),
  fEv(0),
  fFileOut(0),
  fReadDir(0),
  fOutDir(0)
{
  
  // default constructor

  for (Int_t iPlane=0; iPlane<fMaxNPlanesMFT; iPlane++) {
    fHistNClustersPerEvent[iPlane] = 0;
    fHistNPixelsPerCluster[iPlane] = 0;
    fHistClusterSizeX[iPlane] = 0; 
    fHistClusterSizeY[iPlane] = 0;
  }

}

//====================================================================================================================================================

void AliMFTClusterQA::Init(Char_t *readDir, Char_t *outDir, Int_t nEventsToAnalyze) {

  fReadDir = readDir;
  fOutDir  = outDir;

  fRunLoader = AliRunLoader::Open(Form("%s/galice.root", fReadDir.Data()));
  gAlice = fRunLoader->GetAliRun();
  if (!gAlice) fRunLoader->LoadgAlice();
  fNEvents = fRunLoader->GetNumberOfEvents();
  if (nEventsToAnalyze>0) fNEvents = TMath::Min(fNEvents, nEventsToAnalyze);

  fMFT = (AliMFT*) gAlice->GetDetector("MFT"); 
  fNPlanes = fMFT->GetSegmentation()->GetNPlanes();

  BookHistos();

  fMFTLoader = fRunLoader->GetDetectorLoader("MFT");
  fMFTLoader -> LoadRecPoints("READ");

}

//====================================================================================================================================================

Bool_t AliMFTClusterQA::LoadNextEvent() {

  if (fEv>=fNEvents) return kFALSE;
  AliDebug(1, Form("event %5d",fEv));
  
  fRunLoader->GetEvent(fEv);
  fEv++;

  if (!fMFTLoader->TreeR()->GetEvent()) return kTRUE;

  for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
    Int_t nClusters = fMFT->GetRecPointsList(iPlane)->GetEntries();
    fHistNClustersPerEvent[iPlane] -> Fill(nClusters);
    AliDebug(1,Form("nClusters = %5d", nClusters));
    for (Int_t iCluster=0; iCluster<nClusters; iCluster++) {
      AliMFTCluster *cluster = (AliMFTCluster*) (fMFT->GetRecPointsList(iPlane))->At(iCluster);
      fHistNPixelsPerCluster[iPlane] -> Fill(cluster->GetSize());
      fHistClusterSizeX[iPlane]      -> Fill(cluster->GetErrX()*1.e4);   // converted in microns
      fHistClusterSizeY[iPlane]      -> Fill(cluster->GetErrY()*1.e4);   // converted in microns
    }
  }

  return kTRUE;

}  

//====================================================================================================================================================

void AliMFTClusterQA::BookHistos() {

  fFileOut = new TFile(Form("%s/MFT.RecPoints.QA.root",fOutDir.Data()), "recreate");

  for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {

    fHistNClustersPerEvent[iPlane] = new TH1D(Form("fHistNClustersPerEvent_Pl%02d",iPlane), 
					      Form("Number of clusters per event in Plane%02d",iPlane),
					      10000, -0.5, 9999.5);

    fHistNPixelsPerCluster[iPlane] = new TH1D(Form("fHistNPixelsPerCluster_Pl%02d",iPlane), 
					      Form("Number of pixels per cluster in Plane%02d",iPlane),
					      15, -0.5, 14.5);

    fHistClusterSizeX[iPlane]      = new TH1D(Form("fHistClusterSizeX_Pl%02d",iPlane), 
					      Form("#Deltax for clusters in Plane%02d",iPlane), 
					      100, 0., 100.);
    
    fHistClusterSizeY[iPlane]      = new TH1D(Form("fHistClusterSizeY_Pl%02d",iPlane), 
					      Form("#Deltay for clusters in Plane%02d",iPlane), 
					      100, 0., 100.);
    
    fHistNClustersPerEvent[iPlane] -> SetXTitle("N_{clusters} per Event");
    fHistNPixelsPerCluster[iPlane] -> SetXTitle("N_{pixels} per Cluster");
    fHistClusterSizeX[iPlane]      -> SetXTitle("#Deltax  [#mum]");
    fHistClusterSizeY[iPlane]      -> SetXTitle("#Deltay  [#mum]");

    fHistNClustersPerEvent[iPlane] -> Sumw2();
    fHistNPixelsPerCluster[iPlane] -> Sumw2();
    fHistClusterSizeX[iPlane]      -> Sumw2();
    fHistClusterSizeY[iPlane]      -> Sumw2();
    
  }
  
}

//====================================================================================================================================================

void AliMFTClusterQA::Terminate() {

  AliInfo("Writing QA histos...");

  fFileOut->cd();

  for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
    fHistNClustersPerEvent[iPlane] -> Write();
    fHistNPixelsPerCluster[iPlane] -> Write();
    fHistClusterSizeX[iPlane]      -> Write();
    fHistClusterSizeY[iPlane]      -> Write();
  }

  fFileOut -> Close();

}

//====================================================================================================================================================
Commit	Line	Data
820b4d9e	1	#include "TObject.h"
	2	#include "AliRunLoader.h"
	3	#include "AliRun.h"
	4	#include "AliLoader.h"
	5	#include "AliMFT.h"
	6	#include "TClonesArray.h"
	7	#include "AliMFTCluster.h"
	8	#include "AliMFTSegmentation.h"
	9	#include "TFile.h"
	10	#include "TH1D.h"
	11	#include "AliLog.h"
	12	#include "TString.h"
	13
	14	#include "AliMFTClusterQA.h"
	15
	16	//====================================================================================================================================================
	17	//
	18	// Class for the analysis of the MFT clusters (a.k.a. rec points). Few QA histograms are created
	19	//
	20	// Contact author: antonio.uras@cern.ch
	21	//
	22	//====================================================================================================================================================
	23
	24	ClassImp(AliMFTClusterQA)
	25
	26	//====================================================================================================================================================
	27
	28	AliMFTClusterQA::AliMFTClusterQA():
	29	TObject(),
	30	fMFTLoader(0),
	31	fRunLoader(0),
	32	fMFT(0),
	33	fNPlanes(0),
	34	fNEvents(0),
	35	fEv(0),
	36	fFileOut(0),
	37	fReadDir(0),
	38	fOutDir(0)
	39	{
	40
	41	// default constructor
	42
7230691d	43	for (Int_t iPlane=0; iPlane<fMaxNPlanesMFT; iPlane++) {
820b4d9e	44	fHistNClustersPerEvent[iPlane] = 0;
	45	fHistNPixelsPerCluster[iPlane] = 0;
	46	fHistClusterSizeX[iPlane] = 0;
	47	fHistClusterSizeY[iPlane] = 0;
	48	}
	49
	50	}
	51
	52	//====================================================================================================================================================
	53
	54	void AliMFTClusterQA::Init(Char_t readDir, Char_t outDir, Int_t nEventsToAnalyze) {
	55
	56	fReadDir = readDir;
	57	fOutDir = outDir;
	58
	59	fRunLoader = AliRunLoader::Open(Form("%s/galice.root", fReadDir.Data()));
	60	gAlice = fRunLoader->GetAliRun();
	61	if (!gAlice) fRunLoader->LoadgAlice();
	62	fNEvents = fRunLoader->GetNumberOfEvents();
	63	if (nEventsToAnalyze>0) fNEvents = TMath::Min(fNEvents, nEventsToAnalyze);
	64
	65	fMFT = (AliMFT*) gAlice->GetDetector("MFT");
	66	fNPlanes = fMFT->GetSegmentation()->GetNPlanes();
	67
	68	BookHistos();
	69
	70	fMFTLoader = fRunLoader->GetDetectorLoader("MFT");
	71	fMFTLoader -> LoadRecPoints("READ");
	72
	73	}
	74
	75	//====================================================================================================================================================
	76
	77	Bool_t AliMFTClusterQA::LoadNextEvent() {
	78
	79	if (fEv>=fNEvents) return kFALSE;
	80	AliDebug(1, Form("event %5d",fEv));
	81
	82	fRunLoader->GetEvent(fEv);
	83	fEv++;
	84
	85	if (!fMFTLoader->TreeR()->GetEvent()) return kTRUE;
	86
	87	for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
	88	Int_t nClusters = fMFT->GetRecPointsList(iPlane)->GetEntries();
	89	fHistNClustersPerEvent[iPlane] -> Fill(nClusters);
	90	AliDebug(1,Form("nClusters = %5d", nClusters));
	91	for (Int_t iCluster=0; iCluster<nClusters; iCluster++) {
	92	AliMFTCluster cluster = (AliMFTCluster) (fMFT->GetRecPointsList(iPlane))->At(iCluster);
	93	fHistNPixelsPerCluster[iPlane] -> Fill(cluster->GetSize());
	94	fHistClusterSizeX[iPlane] -> Fill(cluster->GetErrX()*1.e4); // converted in microns
	95	fHistClusterSizeY[iPlane] -> Fill(cluster->GetErrY()*1.e4); // converted in microns
	96	}
	97	}
	98
	99	return kTRUE;
	100
	101	}
	102
	103	//====================================================================================================================================================
	104
	105	void AliMFTClusterQA::BookHistos() {
	106
	107	fFileOut = new TFile(Form("%s/MFT.RecPoints.QA.root",fOutDir.Data()), "recreate");
108
109	for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
110
111	fHistNClustersPerEvent[iPlane] = new TH1D(Form("fHistNClustersPerEvent_Pl%02d",iPlane),
112	Form("Number of clusters per event in Plane%02d",iPlane),
113	10000, -0.5, 9999.5);
114
115	fHistNPixelsPerCluster[iPlane] = new TH1D(Form("fHistNPixelsPerCluster_Pl%02d",iPlane),
116	Form("Number of pixels per cluster in Plane%02d",iPlane),
117	15, -0.5, 14.5);
118
119	fHistClusterSizeX[iPlane] = new TH1D(Form("fHistClusterSizeX_Pl%02d",iPlane),
120	Form("#Deltax for clusters in Plane%02d",iPlane),
121	100, 0., 100.);
122
123	fHistClusterSizeY[iPlane] = new TH1D(Form("fHistClusterSizeY_Pl%02d",iPlane),
124	Form("#Deltay for clusters in Plane%02d",iPlane),
125	100, 0., 100.);
126
127	fHistNClustersPerEvent[iPlane] -> SetXTitle("N_{clusters} per Event");
128	fHistNPixelsPerCluster[iPlane] -> SetXTitle("N_{pixels} per Cluster");
129	fHistClusterSizeX[iPlane] -> SetXTitle("#Deltax [#mum]");
130	fHistClusterSizeY[iPlane] -> SetXTitle("#Deltay [#mum]");
131
132	fHistNClustersPerEvent[iPlane] -> Sumw2();
133	fHistNPixelsPerCluster[iPlane] -> Sumw2();
134	fHistClusterSizeX[iPlane] -> Sumw2();
135	fHistClusterSizeY[iPlane] -> Sumw2();
136
137	}
138
139	}
140
141	//====================================================================================================================================================
142
143	void AliMFTClusterQA::Terminate() {
144
145	AliInfo("Writing QA histos...");
146
147	fFileOut->cd();
148
149	for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
150	fHistNClustersPerEvent[iPlane] -> Write();
151	fHistNPixelsPerCluster[iPlane] -> Write();
152	fHistClusterSizeX[iPlane] -> Write();
153	fHistClusterSizeY[iPlane] -> Write();
154	}
155
156	fFileOut -> Close();
157
158	}
159
160	//====================================================================================================================================================