[u/mrichter/AliRoot.git] / TRD / qaRec / AliTRDcalibration.cxx


/**************************************************************************
* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
*                                                                        *
* Author: The ALICE Off-line Project.                                    *
* Contributors are mentioned in the code where appropriate.              *
*                                                                        *
* Permission to use, copy, modify and distribute this software and its   *
* documentation strictly for non-commercial purposes is hereby granted   *
* without fee, provided that the above copyright notice appears in all   *
* copies and that both the copyright notice and this permission notice   *
* appear in the supporting documentation. The authors make no claims     *
* about the suitability of this software for any purpose. It is          *
* provided "as is" without express or implied warranty.                  *
**************************************************************************/

// macro for very simple analysis


#include "Riostream.h"
#include "TChain.h"
#include "TTree.h"
#include "TProfile2D.h"
#include "TH2I.h"
#include "TH1F.h"
#include "TList.h"
#include "TMath.h"
#include "TCanvas.h"
#include "TObject.h"
#include "TFile.h"
#include "TObjArray.h"

#include "AliTRDrecoTask.h"
#include "AliAnalysisManager.h"

#include "AliESDInputHandler.h"
#include "AliTRDtrackV1.h"
#include "AliTRDseedV1.h"
#include "AliTRDcluster.h"
#include "info/AliTRDtrackInfo.h"
#include "AliTRDcalibDB.h"

#include "AliTRDCalibraFillHisto.h"

#include "AliLog.h"

#include "AliTRDcalibration.h"


ClassImp(AliTRDcalibration)

//________________________________________________________________________
AliTRDcalibration::AliTRDcalibration() 
  :AliTRDrecoTask("Calibration", "Calibration on tracks")
  ,fTrackInfo(0)
  ,ftrdTrack(0)
  ,fcl(0)
  ,fTRDCalibraFillHisto(0)
  ,fNbTRDTrackUsed(0)
  ,fNbTimeBin(0x0)
  ,fNbClusters(0)
  ,fPHSum(0)
  ,fCHSum(0)
  ,flow(0)
  ,fhigh(30)
  ,fNbTimeBins(30)
  ,ffillZero(kFALSE)
{
  // Constructor
}  

//________________________________________________________________________
void AliTRDcalibration::CreateOutputObjects() 
{
  OpenFile(0, "RECREATE");

  // Number of time bins
  AliTRDcalibDB *cal = AliTRDcalibDB::Instance();
  fNbTimeBins = cal->GetNumberOfTimeBins();
  
  // instance calibration
  fTRDCalibraFillHisto = AliTRDCalibraFillHisto::Instance();
  fTRDCalibraFillHisto->SetHisto2d(); // choose to use histograms
  fTRDCalibraFillHisto->SetCH2dOn();  // choose to calibrate the gain
  fTRDCalibraFillHisto->SetPH2dOn();  // choose to calibrate the drift velocity
  fTRDCalibraFillHisto->SetPRF2dOn(); // choose to look at the PRF
  fTRDCalibraFillHisto->Init2Dhistos(); // initialise the histos
  fTRDCalibraFillHisto->SetNumberClusters(flow); // At least 11 clusters
  fTRDCalibraFillHisto->SetNumberClustersf(fhigh); // At least 11 clusters
  fTRDCalibraFillHisto->SetFillWithZero(ffillZero); // Fill zeros

  fTRDCalibraFillHisto->SetDebugLevel(fDebugLevel); //debug stuff

  fContainer = new TObjArray();
  fContainer->Add(fTRDCalibraFillHisto->GetCH2d()); //TH2I
  fContainer->Add(fTRDCalibraFillHisto->GetPH2d()); //TProfile2D
  fContainer->Add(fTRDCalibraFillHisto->GetPRF2d()); //TProfile2D

  //printf("create output objects 1\n");

  fNbTRDTrackUsed = new TH1F("TRDTrackUsed","TRDTrackUsed",50,0,50);
  fNbTRDTrackUsed->Sumw2();
  //
  fNbTimeBin = new TH1F("TimeBin","TimeBin",35,0,35);
  fNbTimeBin->Sumw2();
  //
  fNbClusters = new TH1F("NbClusters","",35,0,35);
  fNbClusters->Sumw2();
  //
  fPHSum = new TProfile2D("PH2dSum","Nz0Nrphi0"
      ,30,-0.05,(Double_t)fNbTimeBins/10.0-0.05
      ,540,0,540);
  fPHSum->SetYTitle("Det/pad groups");
  fPHSum->SetXTitle("time [#mus]");
  fPHSum->SetZTitle("<PH> [a.u.]");
  fPHSum->SetStats(0);
  //
  fCHSum = new TH2I("CH2dSum","Nz0Nrphi0",100,0,300,540,0,540);
  fCHSum->SetYTitle("Det/pad groups");
  fCHSum->SetXTitle("charge deposit [a.u]");
  fCHSum->SetZTitle("counts");
  fCHSum->SetStats(0);
  fCHSum->Sumw2();

  fContainer->Add(fNbTRDTrackUsed);
  fContainer->Add(fNbTimeBin);
  fContainer->Add(fNbClusters);
  fContainer->Add(fPHSum);
  fContainer->Add(fCHSum);
  //printf("create output objects 2\n");
  
}

//________________________________________________________________________
void AliTRDcalibration::Exec(Option_t *) 
{
  
  Int_t nbTrdTracksUsed = 0;
  
  for(Int_t itrk=0; itrk < fTracks->GetEntriesFast(); itrk++){
    
    fTrackInfo = (AliTRDtrackInfo*)fTracks->UncheckedAt(itrk);
    ftrdTrack = fTrackInfo->GetTrack();
    if(!ftrdTrack) continue;
    nbTrdTracksUsed++;
    fTRDCalibraFillHisto->UpdateHistogramsV1(ftrdTrack);
    
    const AliTRDseedV1 *tracklet = 0x0;
    for(Int_t itr = 0; itr < AliTRDgeometry::kNlayer; itr++){
      if(!(tracklet = ftrdTrack->GetTracklet(itr))) continue;
      if(!tracklet->IsOK()) continue;
      Int_t nbclusters = 0;
      // For PH
      Double_t phtb[AliTRDseedV1::kNtb];
      memset(phtb, 0, AliTRDseedV1::kNtb*sizeof(Double_t));

      // For CH
      Double_t sum = 0.0;
      // normalisation
      Float_t normalisation = 6.67;
      Int_t detector = 0;
      for(int ic=0; ic<AliTRDseedV1::kNclusters; ic++){
        if(!(fcl = tracklet->GetClusters(ic))) continue;
        nbclusters++;
        Int_t time = fcl->GetPadTime();
        Float_t ch =  tracklet->GetdQdl(ic);
        detector = fcl->GetDetector();	  
        if((time>-1) && (time<fNbTimeBins)) phtb[time]=ch/normalisation;
        sum += ch/normalisation;
        //printf("time %d\n",time);
        fNbTimeBin->Fill(time);
      }
      fNbClusters->Fill(nbclusters);
      if((nbclusters > flow) && (nbclusters < fhigh)){
        fCHSum->Fill(sum/20.0,0.0);
        for(int ic=0; ic<fNbTimeBins; ic++){
          if(ffillZero) fPHSum->Fill((Double_t)ic/10.0,0.0,(Double_t)phtb[ic]);
          else {
            if(phtb[ic] > 0.0) fPHSum->Fill((Double_t)ic/10.0,0.0,(Double_t)phtb[ic]);
          }
        }
      }
    }
  }
  
  //Fill Histos
  fNbTRDTrackUsed->Fill(nbTrdTracksUsed);
  
  
  // Post output data
  PostData(0, fContainer);
  
}      
//________________________________________________________________________
void AliTRDcalibration::Terminate(Option_t *) 
{
  // Draw result to the screen
  // Called once at the end of the query

  //printf("terminate\n");

  if(fTRDCalibraFillHisto) fTRDCalibraFillHisto->DestroyDebugStreamer();

  /*
  fContainer = dynamic_cast<TList*> (GetOutputData(0));
  if (!fContainer) {
    Printf("ERROR: fList not available");
    return;
  }
  */
}
Commit	Line	Data
819f5b0f	1
	2	/**************************************************************************
	3	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* *
	5	* Author: The ALICE Off-line Project. *
	6	* Contributors are mentioned in the code where appropriate. *
	7	* *
	8	* Permission to use, copy, modify and distribute this software and its *
	9	* documentation strictly for non-commercial purposes is hereby granted *
	10	* without fee, provided that the above copyright notice appears in all *
	11	* copies and that both the copyright notice and this permission notice *
	12	* appear in the supporting documentation. The authors make no claims *
	13	* about the suitability of this software for any purpose. It is *
	14	* provided "as is" without express or implied warranty. *
	15	**************************************************************************/
	16
	17	// macro for very simple analysis
	18
	19
	20
	21	#include "Riostream.h"
	22	#include "TChain.h"
	23	#include "TTree.h"
	24	#include "TProfile2D.h"
	25	#include "TH2I.h"
	26	#include "TH1F.h"
	27	#include "TList.h"
	28	#include "TMath.h"
	29	#include "TCanvas.h"
	30	#include "TObject.h"
	31	#include "TFile.h"
	32	#include "TObjArray.h"
	33
3d86166d	34	#include "AliTRDrecoTask.h"
819f5b0f	35	#include "AliAnalysisManager.h"
	36
	37	#include "AliESDInputHandler.h"
	38	#include "AliTRDtrackV1.h"
	39	#include "AliTRDseedV1.h"
	40	#include "AliTRDcluster.h"
873458ab	41	#include "info/AliTRDtrackInfo.h"
819f5b0f	42	#include "AliTRDcalibDB.h"
	43
	44	#include "AliTRDCalibraFillHisto.h"
	45
	46	#include "AliLog.h"
	47
	48	#include "AliTRDcalibration.h"
	49
	50
	51	ClassImp(AliTRDcalibration)
	52
	53	//________________________________________________________________________
3d86166d	54	AliTRDcalibration::AliTRDcalibration()
3d86166d	55	:AliTRDrecoTask("Calibration", "Calibration on tracks")
819f5b0f	56	,fTrackInfo(0)
	57	,ftrdTrack(0)
	58	,fcl(0)
819f5b0f	59	,fTRDCalibraFillHisto(0)
819f5b0f	60	,fNbTRDTrackUsed(0)
3d86166d	61	,fNbTimeBin(0x0)
819f5b0f	62	,fNbClusters(0)
	63	,fPHSum(0)
	64	,fCHSum(0)
	65	,flow(0)
	66	,fhigh(30)
819f5b0f	67	,fNbTimeBins(30)
	68	,ffillZero(kFALSE)
	69	{
	70	// Constructor
ed383798	71	}
ed383798	72
819f5b0f	73	//________________________________________________________________________
	74	void AliTRDcalibration::CreateOutputObjects()
	75	{
	76	OpenFile(0, "RECREATE");
	77
	78	// Number of time bins
	79	AliTRDcalibDB *cal = AliTRDcalibDB::Instance();
	80	fNbTimeBins = cal->GetNumberOfTimeBins();
	81
	82	// instance calibration
	83	fTRDCalibraFillHisto = AliTRDCalibraFillHisto::Instance();
	84	fTRDCalibraFillHisto->SetHisto2d(); // choose to use histograms
	85	fTRDCalibraFillHisto->SetCH2dOn(); // choose to calibrate the gain
	86	fTRDCalibraFillHisto->SetPH2dOn(); // choose to calibrate the drift velocity
	87	fTRDCalibraFillHisto->SetPRF2dOn(); // choose to look at the PRF
	88	fTRDCalibraFillHisto->Init2Dhistos(); // initialise the histos
	89	fTRDCalibraFillHisto->SetNumberClusters(flow); // At least 11 clusters
	90	fTRDCalibraFillHisto->SetNumberClustersf(fhigh); // At least 11 clusters
	91	fTRDCalibraFillHisto->SetFillWithZero(ffillZero); // Fill zeros
	92
	93	fTRDCalibraFillHisto->SetDebugLevel(fDebugLevel); //debug stuff
	94
3d86166d	95	fContainer = new TObjArray();
	96	fContainer->Add(fTRDCalibraFillHisto->GetCH2d()); //TH2I
	97	fContainer->Add(fTRDCalibraFillHisto->GetPH2d()); //TProfile2D
	98	fContainer->Add(fTRDCalibraFillHisto->GetPRF2d()); //TProfile2D
819f5b0f	99
	100	//printf("create output objects 1\n");
	101
	102	fNbTRDTrackUsed = new TH1F("TRDTrackUsed","TRDTrackUsed",50,0,50);
	103	fNbTRDTrackUsed->Sumw2();
	104	//
	105	fNbTimeBin = new TH1F("TimeBin","TimeBin",35,0,35);
	106	fNbTimeBin->Sumw2();
	107	//
	108	fNbClusters = new TH1F("NbClusters","",35,0,35);
	109	fNbClusters->Sumw2();
	110	//
	111	fPHSum = new TProfile2D("PH2dSum","Nz0Nrphi0"
	112	,30,-0.05,(Double_t)fNbTimeBins/10.0-0.05
	113	,540,0,540);
	114	fPHSum->SetYTitle("Det/pad groups");
	115	fPHSum->SetXTitle("time [#mus]");
	116	fPHSum->SetZTitle("<PH> [a.u.]");
	117	fPHSum->SetStats(0);
	118	//
	119	fCHSum = new TH2I("CH2dSum","Nz0Nrphi0",100,0,300,540,0,540);
	120	fCHSum->SetYTitle("Det/pad groups");
	121	fCHSum->SetXTitle("charge deposit [a.u]");
	122	fCHSum->SetZTitle("counts");
	123	fCHSum->SetStats(0);
	124	fCHSum->Sumw2();
	125
3d86166d	126	fContainer->Add(fNbTRDTrackUsed);
	127	fContainer->Add(fNbTimeBin);
	128	fContainer->Add(fNbClusters);
	129	fContainer->Add(fPHSum);
	130	fContainer->Add(fCHSum);
819f5b0f	131	//printf("create output objects 2\n");
	132
	133	}
	134
	135	//________________________________________________________________________
	136	void AliTRDcalibration::Exec(Option_t *)
	137	{
	138
	139	Int_t nbTrdTracksUsed = 0;
	140
	141	for(Int_t itrk=0; itrk < fTracks->GetEntriesFast(); itrk++){
	142
	143	fTrackInfo = (AliTRDtrackInfo*)fTracks->UncheckedAt(itrk);
b718144c	144	ftrdTrack = fTrackInfo->GetTrack();
819f5b0f	145	if(!ftrdTrack) continue;
	146	nbTrdTracksUsed++;
	147	fTRDCalibraFillHisto->UpdateHistogramsV1(ftrdTrack);
	148
	149	const AliTRDseedV1 *tracklet = 0x0;
e3cf3d02	150	for(Int_t itr = 0; itr < AliTRDgeometry::kNlayer; itr++){
819f5b0f	151	if(!(tracklet = ftrdTrack->GetTracklet(itr))) continue;
	152	if(!tracklet->IsOK()) continue;
	153	Int_t nbclusters = 0;
	154	// For PH
e3cf3d02	155	Double_t phtb[AliTRDseedV1::kNtb];
	156	memset(phtb, 0, AliTRDseedV1::kNtb*sizeof(Double_t));
	157
819f5b0f	158	// For CH
	159	Double_t sum = 0.0;
	160	// normalisation
	161	Float_t normalisation = 6.67;
	162	Int_t detector = 0;
8d2bec9e	163	for(int ic=0; ic<AliTRDseedV1::kNclusters; ic++){
819f5b0f	164	if(!(fcl = tracklet->GetClusters(ic))) continue;
	165	nbclusters++;
	166	Int_t time = fcl->GetPadTime();
	167	Float_t ch = tracklet->GetdQdl(ic);
	168	detector = fcl->GetDetector();
	169	if((time>-1) && (time<fNbTimeBins)) phtb[time]=ch/normalisation;
	170	sum += ch/normalisation;
	171	//printf("time %d\n",time);
	172	fNbTimeBin->Fill(time);
	173	}
	174	fNbClusters->Fill(nbclusters);
	175	if((nbclusters > flow) && (nbclusters < fhigh)){
	176	fCHSum->Fill(sum/20.0,0.0);
	177	for(int ic=0; ic<fNbTimeBins; ic++){
	178	if(ffillZero) fPHSum->Fill((Double_t)ic/10.0,0.0,(Double_t)phtb[ic]);
	179	else {
	180	if(phtb[ic] > 0.0) fPHSum->Fill((Double_t)ic/10.0,0.0,(Double_t)phtb[ic]);
	181	}
	182	}
	183	}
	184	}
	185	}
	186
	187	//Fill Histos
	188	fNbTRDTrackUsed->Fill(nbTrdTracksUsed);
	189
	190
	191	// Post output data
3d86166d	192	PostData(0, fContainer);
819f5b0f	193
	194	}
	195	//________________________________________________________________________
	196	void AliTRDcalibration::Terminate(Option_t *)
	197	{
	198	// Draw result to the screen
	199	// Called once at the end of the query
	200
	201	//printf("terminate\n");
	202
	203	if(fTRDCalibraFillHisto) fTRDCalibraFillHisto->DestroyDebugStreamer();
	204
	205	/*
3d86166d	206	fContainer = dynamic_cast<TList*> (GetOutputData(0));
3d86166d	207	if (!fContainer) {
819f5b0f	208	Printf("ERROR: fList not available");
	209	return;
	210	}
	211	*/
	212	}