+++ /dev/null
-/**************************************************************************
- * 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. *
- **************************************************************************/
-
-/* $Id$ */
-
-//-----------------------------------------------------------------------
-// Example of task (running locally, on AliEn and CAF),
-// which provides standard way of calculating acceptance and efficiency
-// between different steps of the procedure.
-// The ouptut of the task is a AliCFContainer from which the efficiencies
-// can be calculated
-//-----------------------------------------------------------------------
-// Author : R. Vernet, Consorzio Cometa - Catania (it)
-//-----------------------------------------------------------------------
-// Modification done by X. Lopez - LPC Clermont (fr)
-//-----------------------------------------------------------------------
-
-#ifndef ALICFMUONSINGLETASK1_CXX
-#define ALICFMUONSINGLETASK1_CXX
-
-#include "AliCFMuonSingleTask1.h"
-#include "AliHeader.h"
-#include "AliESDHeader.h"
-#include "AliStack.h"
-#include "TParticle.h"
-#include "TLorentzVector.h"
-#include "TH1I.h"
-#include "AliMCEvent.h"
-#include "AliAnalysisManager.h"
-#include "AliESDEvent.h"
-#include "AliAODEvent.h"
-#include "AliCFManager.h"
-#include "AliCFCutBase.h"
-#include "AliCFContainer.h"
-#include "TChain.h"
-#include "AliESDtrack.h"
-#include "AliLog.h"
-#include "AliESDMuonTrack.h"
-#include "AliESDtrack.h"
-#include "AliESDInputHandler.h"
-#include "TCanvas.h"
-
-ClassImp(AliCFMuonSingleTask1)
-
-//__________________________________________________________________________
-AliCFMuonSingleTask1::AliCFMuonSingleTask1() :
- fReadAODData(0),
- fCFManager(0x0),
- fQAHistList(0x0),
- fHistEventsProcessed(0x0),
- fNevt(0),
- fIsMC(kFALSE)
-{
- //
- //Default ctor
- //
-}
-//___________________________________________________________________________
-AliCFMuonSingleTask1::AliCFMuonSingleTask1(const Char_t* name) :
- AliAnalysisTaskSE(name),
- fReadAODData(0),
- fCFManager(0x0),
- fQAHistList(0x0),
- fHistEventsProcessed(0x0),
- fNevt(0),
- fIsMC(kFALSE)
-{
- //
- // Constructor. Initialization of Inputs and Outputs
- //
- Info("AliCFMuonSingleTask1","Calling Constructor");
-
- fHistEventsProcessed = new TH1I("fHistEventsProcessed","",1,0,1) ;
-
- DefineOutput(1,TH1I::Class());
- DefineOutput(2,AliCFContainer::Class());
-
-}
-
-//___________________________________________________________________________
-AliCFMuonSingleTask1& AliCFMuonSingleTask1::operator=(const AliCFMuonSingleTask1& c)
-{
- //
- // Assignment operator
- //
- if (this!=&c) {
- AliAnalysisTaskSE::operator=(c) ;
- fReadAODData = c.fReadAODData ;
- fCFManager = c.fCFManager;
- fQAHistList = c.fQAHistList ;
- fHistEventsProcessed = c.fHistEventsProcessed;
- fNevt = c.fNevt ;
- }
- return *this;
-}
-
-//___________________________________________________________________________
-AliCFMuonSingleTask1::AliCFMuonSingleTask1(const AliCFMuonSingleTask1& c) :
- AliAnalysisTaskSE(c),
- fReadAODData(c.fReadAODData),
- fCFManager(c.fCFManager),
- fQAHistList(c.fQAHistList),
- fHistEventsProcessed(c.fHistEventsProcessed),
- fNevt(c.fNevt),
- fIsMC(kFALSE)
-{
- //
- // Copy Constructor
- //
-}
-
-//___________________________________________________________________________
-AliCFMuonSingleTask1::~AliCFMuonSingleTask1() {
- //
- //destructor
- //
- Info("~AliCFMuonSingleTask1","Calling Destructor");
- if (fCFManager) delete fCFManager ;
- if (fHistEventsProcessed) delete fHistEventsProcessed ;
- if (fQAHistList) {fQAHistList->Clear(); delete fQAHistList;}
-}
-
-//_________________________________________________
-void AliCFMuonSingleTask1::UserExec(Option_t *)
-{
- //
- // Main loop function
- //
-
- Info("UserExec"," ") ;
- if (!fMCEvent && fIsMC) {
- Error("UserExec","NO MC EVENT FOUND!");
- return;
- }
-
- fNevt++;
- fCFManager->SetEventInfo(fMCEvent);
- Double_t containerInput[18] ;
-
-////////
-//// MC
-////////
-
-// loop on the MC part
-
- if(fIsMC){
-
- for (Int_t ipart=0; ipart<fMCEvent->GetNumberOfTracks(); ipart++) {
-
- AliMCParticle *mcPart = (AliMCParticle*) fMCEvent->GetTrack(ipart);
- TParticle *part = mcPart->Particle();
-
- // rapidity and Pt cuts (default -4<y<-2.5 et 0<pt<20)
- if (!fCFManager->CheckParticleCuts(AliCFManager::kPartAccCuts,mcPart)) continue;
-
- // Selection of mu
- Int_t pdg=part->GetPdgCode();
- if(TMath::Abs(pdg)==13){
-
- Float_t emc = part->Energy();
- Float_t pzmc = part->Pz();
- Float_t rapmc = Rap(emc,pzmc);
- Float_t phimc = part->Phi();
- phimc = Phideg(phimc);
- Float_t ptmc = part->Pt();
- Float_t pmc = part->P();
- Float_t zmc = part->Vz();
- Float_t etamc = part->Eta();
- Float_t chargemc=0;
- if(pdg==13) chargemc=-1;
- if(pdg==-13) chargemc=1;
- Float_t thetamc;
- thetamc = TMath::Abs(-TMath::Pi()+TMath::ASin(ptmc/pmc)); // convention for EVGEN
- Float_t ymc = part->Vy();
- Float_t xmc = part->Vx();
- Float_t rmc = TMath::Sqrt(xmc*xmc+ymc*ymc);
-
- containerInput[0] = etamc ;
- containerInput[1] = rapmc ;
- containerInput[2] = phimc ;
- containerInput[3] = ptmc ;
- containerInput[4] = pmc ;
- containerInput[14] = zmc ;
- containerInput[16] = chargemc ;
- containerInput[17] = thetamc ;
-
- // 10 var calculated only for ESD .i.e set at 1 in MC step
- for(Int_t i=5; i<=13; i++){
- containerInput[i] = 1;
- }
-
- containerInput[7] = 3;
- containerInput[10] = rmc ; // radius at production
- containerInput[15] = 175. ; // rabsmc
-
- // fill the container at the first step
- fCFManager->GetParticleContainer()->Fill(containerInput,0);
- }
- }
- }
-
-
-////////
-//// ESD
-////////
-
- Int_t trig1MuL = 1 ,trig1MuH = 2, trigUSL = 4, trigUSH = 8, trigLSL = 16, trigLSH = 32;
- Int_t trig = 0;
-
- AliESDEvent *fESD;
- AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*>
- (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
- if ( ! esdH) {
- AliError("Cannot get input event handler");
- return;
- }
- fESD = esdH->GetEvent();
-
-// trigger
-
- if (fESD->GetTriggerMask() & trig1MuL) trig = 1;
- if (fESD->GetTriggerMask() & trig1MuH) trig = 2;
- if (fESD->GetTriggerMask() & trigUSL) trig = 3;
- if (fESD->GetTriggerMask() & trigUSH) trig = 4;
- if (fESD->GetTriggerMask() & trigLSL) trig = 5;
- if (fESD->GetTriggerMask() & trigLSH) trig = 6;
-
-// vertex
-
- Float_t vx = -200 , vy = -200 , vz = -200, vt=-200;
-
- Double_t Ncont = 0.;
- AliESDVertex* vertex = (AliESDVertex*) fESD->GetVertex();
- if (vertex->GetNContributors()) {
- vz = vertex->GetZ();
- vy = vertex->GetY();
- vx = vertex->GetX();
- vt=TMath::Sqrt(vx*vx+vy*vy);
- Ncont=vertex->GetNContributors();
- }
-
-// tracks
-
- Int_t mult1 = fESD->GetNumberOfMuonTracks() ;
-
- for (Int_t j = 0; j<mult1; j++) {
- AliESDMuonTrack* mu1 = new AliESDMuonTrack(*(fESD->GetMuonTrack(j)));
-
- if (!mu1->ContainTrackerData()) { mu1=0; continue; }
-
- Float_t charger = mu1->Charge();
- Float_t er = mu1->E();
- Float_t pzr = mu1->Pz();
- Float_t rapr=Rap(er,pzr);
- Float_t phir = mu1->Phi();
- phir = Phideg(phir);
- Float_t ptr = mu1->Pt();
- Float_t pr = mu1->P();
- Float_t hit = mu1->GetNHit();
- Float_t chi2 = mu1->GetChi2() / (2.0 * hit - 5);
- Int_t matchtrig = mu1->GetMatchTrigger();
- Float_t chi2match = mu1->GetChi2MatchTrigger();
- Float_t dcar = mu1->GetDCA();
- Float_t zr = mu1->GetZ();
- Float_t etar = mu1->Eta();
- Float_t rabs = mu1->GetRAtAbsorberEnd();
- Float_t thetar ;
- thetar = TMath::Abs(-TMath::Pi()+TMath::ASin(ptr/pr));
-
-// rapidity and Pt cuts (default -4<y<-2.5 et 0<pt<20)
- if (!fCFManager->CheckParticleCuts(AliCFManager::kPartAccCuts,mu1)) continue;
-
- Float_t thetaAbs = 180.*(1.-TMath::ATan(rabs/505.)/TMath::Pi());
- if (thetaAbs<171 || thetaAbs>178) continue;
-
- containerInput[0] = etar ;
- containerInput[1] = rapr ;
- containerInput[2] = phir ;
- containerInput[3] = ptr ;
- containerInput[4] = pr ;
- containerInput[5] = hit ;
- containerInput[6] = chi2 ;
- containerInput[7] = matchtrig ;
- containerInput[8] = chi2match ;
- containerInput[9] = Ncont ;
- containerInput[10] = vt ;
- containerInput[11] = vz ;
- containerInput[12] = trig ;
- containerInput[13] = dcar ;
- containerInput[14] = zr ;
- containerInput[15] = thetaAbs;
- containerInput[16] = charger ;
- containerInput[17] = thetar ;
-
-// fill the container at the second step (for simu, first for data)
- if(fIsMC){
- fCFManager->GetParticleContainer()->Fill(containerInput,1);
- }
- else{
- fCFManager->GetParticleContainer()->Fill(containerInput,0);
- }
-
-
- } // mu Loop
-
-// ----------
- fHistEventsProcessed->Fill(0);
- PostData(1,fHistEventsProcessed) ;
- PostData(2,fCFManager->GetParticleContainer()) ;
-
-}
-//________________________________________________________________________
-Float_t AliCFMuonSingleTask1::Rap(Float_t e, Float_t pz)
-{
-// calculate rapidity
- Float_t rap;
- if(e!=pz){
- rap = 0.5*TMath::Log((e+pz)/(e-pz));
- return rap;
- }
- else{
- rap = -200;
- return rap;
- }
-}
-//________________________________________________________________________
-Float_t AliCFMuonSingleTask1::Phideg(Float_t phi)
-{
-// calculate Phi in range [-180,180]
- Float_t phideg;
-
- phideg = phi-TMath::Pi();
- phideg = phideg*57.296;
- return phideg;
-}
-//________________________________________________________________________
-void AliCFMuonSingleTask1::Terminate(Option_t *)
-{
- // project pt (var[3]) from the two steps MC(0) and ESD(1)
-
- AliCFContainer *cont = dynamic_cast<AliCFContainer*> (GetOutputData(2));
- if ( ! cont ) {
- AliError("Cannot find container in file");
- return;
- }
-
- TH1D *kpt = cont->ShowProjection(16,0);
- TH1D *rpt = cont->ShowProjection(16,1);
-
- TCanvas *c1 = new TCanvas("AliCFMuonSingleTask1"," MC & ESD",10,10,510,510);
- c1->Divide(1,2);
- c1->cd(1);
- kpt->Draw("HIST");
- c1->cd(2);
- rpt->Draw("HIST");
-
-}
-//________________________________________________________________________
-
-#endif
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff