[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 "AliTRDtrackInfo/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 < 6; itr++){
      if(!(tracklet = ftrdTrack->GetTracklet(itr))) continue;
      if(!tracklet->IsOK()) continue;
      Int_t nbclusters = 0;
      // For PH
      Double_t phtb[35];
      for(Int_t k=0; k < 35; k++){
  phtb[k] = 0.0;
      }
      // For CH
      Double_t sum = 0.0;
      // normalisation
      Float_t normalisation = 6.67;
      Int_t detector = 0;
      for(int ic=0; ic<AliTRDseed::knTimebins; 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"
	41	#include "AliTRDtrackInfo/AliTRDtrackInfo.h"
	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;
	150	for(Int_t itr = 0; itr < 6; itr++){
	151	if(!(tracklet = ftrdTrack->GetTracklet(itr))) continue;
	152	if(!tracklet->IsOK()) continue;
	153	Int_t nbclusters = 0;
	154	// For PH
	155	Double_t phtb[35];
	156	for(Int_t k=0; k < 35; k++){
	157	phtb[k] = 0.0;
	158	}
	159	// For CH
	160	Double_t sum = 0.0;
	161	// normalisation
	162	Float_t normalisation = 6.67;
	163	Int_t detector = 0;
	164	for(int ic=0; ic<AliTRDseed::knTimebins; ic++){
	165	if(!(fcl = tracklet->GetClusters(ic))) continue;
	166	nbclusters++;
	167	Int_t time = fcl->GetPadTime();
	168	Float_t ch = tracklet->GetdQdl(ic);
	169	detector = fcl->GetDetector();
	170	if((time>-1) && (time<fNbTimeBins)) phtb[time]=ch/normalisation;
	171	sum += ch/normalisation;
	172	//printf("time %d\n",time);
	173	fNbTimeBin->Fill(time);
	174	}
	175	fNbClusters->Fill(nbclusters);
	176	if((nbclusters > flow) && (nbclusters < fhigh)){
	177	fCHSum->Fill(sum/20.0,0.0);
	178	for(int ic=0; ic<fNbTimeBins; ic++){
	179	if(ffillZero) fPHSum->Fill((Double_t)ic/10.0,0.0,(Double_t)phtb[ic]);
	180	else {
	181	if(phtb[ic] > 0.0) fPHSum->Fill((Double_t)ic/10.0,0.0,(Double_t)phtb[ic]);
	182	}
	183	}
	184	}
	185	}
	186	}
	187
	188	//Fill Histos
	189	fNbTRDTrackUsed->Fill(nbTrdTracksUsed);
	190
	191
	192	// Post output data
3d86166d	193	PostData(0, fContainer);
819f5b0f	194
	195	}
	196	//________________________________________________________________________
	197	void AliTRDcalibration::Terminate(Option_t *)
	198	{
	199	// Draw result to the screen
	200	// Called once at the end of the query
	201
	202	//printf("terminate\n");
	203
	204	if(fTRDCalibraFillHisto) fTRDCalibraFillHisto->DestroyDebugStreamer();
	205
	206	/*
3d86166d	207	fContainer = dynamic_cast<TList*> (GetOutputData(0));
3d86166d	208	if (!fContainer) {
819f5b0f	209	Printf("ERROR: fList not available");
	210	return;
	211	}
	212	*/
	213	}