Update of the class ESDMuonFilter. New marcros for creating AOD with muon information...

[u/mrichter/AliRoot.git] / PWG3 / muon / AliAnalysisTaskESDMuonFilter.cxx

/**************************************************************************\r
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
 *                                                                        *\r
 * Author: The ALICE Off-line Project.                                    *\r
 * Contributors are mentioned in the code where appropriate.              *\r
 *                                                                        *\r
 * Permission to use, copy, modify and distribute this software and its   *\r
 * documentation strictly for non-commercial purposes is hereby granted   *\r
 * without fee, provided that the above copyright notice appears in all   *\r
 * copies and that both the copyright notice and this permission notice   *\r
 * appear in the supporting documentation. The authors make no claims     *\r
 * about the suitability of this software for any purpose. It is          *\r
 * provided "as is" without express or implied warranty.                  *\r
 **************************************************************************/\r
\r
// Add the muon tracks to the generic AOD track branch during the \r
// filtering of the ESD - R. Arnaldi 5/5/08\r
 \r
#include <TChain.h>\r
#include <TFile.h>\r
\r
#include "AliAnalysisTaskESDMuonFilter.h"\r
#include "AliAnalysisManager.h"\r
#include "AliESDEvent.h"\r
#include "AliAODEvent.h"\r
#include "AliESDInputHandler.h"\r
#include "AliAODHandler.h"\r
#include "AliAnalysisFilter.h"\r
#include "AliESDtrack.h"\r
#include "AliESDMuonTrack.h"\r
#include "AliESDVertex.h"\r
#include "AliMultiplicity.h"\r
#include "AliLog.h"\r
\r
ClassImp(AliAnalysisTaskESDMuonFilter)\r
\r
////////////////////////////////////////////////////////////////////////\r
\r
AliAnalysisTaskESDMuonFilter::AliAnalysisTaskESDMuonFilter():\r
    AliAnalysisTaskSE(),\r
    fTrackFilter(0x0)\r
{\r
  // Default constructor\r
}\r
\r
AliAnalysisTaskESDMuonFilter::AliAnalysisTaskESDMuonFilter(const char* name):\r
    AliAnalysisTaskSE(name),\r
    fTrackFilter(0x0)\r
{\r
  // Constructor\r
}\r
\r
void AliAnalysisTaskESDMuonFilter::UserCreateOutputObjects()\r
{\r
  // Create the output container\r
    OutputTree()->GetUserInfo()->Add(fTrackFilter);\r
}\r
\r
void AliAnalysisTaskESDMuonFilter::Init()\r
{\r
  // Initialization\r
    if (fDebug > 1) AliInfo("Init() \n");\r
}\r
\r
\r
void AliAnalysisTaskESDMuonFilter::UserExec(Option_t */*option*/)\r
{\r
  // Execute analysis for current event					    \r
  Long64_t ientry = Entry();\r
  printf("Muon Filter: Analysing event # %5d\n", (Int_t) ientry);\r
\r
  ConvertESDtoAOD();\r
}\r
\r
void AliAnalysisTaskESDMuonFilter::ConvertESDtoAOD() \r
{\r
  // ESD Muon Filter analysis task executed for each event\r
  AliESDEvent* esd = dynamic_cast<AliESDEvent*>(InputEvent());\r
 \r
  // Define arrays for muons\r
  Double_t pos[3];\r
  Double_t p[3];\r
  Double_t pid[10];\r
 \r
  AliAODHeader* header = AODEvent()->GetHeader();\r
  AliAODTrack *aodTrack = 0x0;\r
 \r
  // Access to the AOD container of tracks\r
  TClonesArray &tracks = *(AODEvent()->GetTracks());\r
  Int_t jTracks = header->GetRefMultiplicity();\r
  \r
  // Read primary vertex from AOD event \r
  AliAODVertex * primary = AODEvent()->GetPrimaryVertex();\r
  primary->Print();\r
 \r
  // Loop on muon tracks to fill the AOD track branch\r
  Int_t nMuTracks = esd->GetNumberOfMuonTracks();\r
  printf("Number of Muon Tracks=%d\n",nMuTracks);\r
 \r
  for (Int_t nMuTrack = 0; nMuTrack < nMuTracks; ++nMuTrack) {\r
\r
       AliESDMuonTrack *esdMuTrack = esd->GetMuonTrack(nMuTrack); \r
     	 \r
       p[0] = esdMuTrack->Px(); \r
       p[1] = esdMuTrack->Py(); \r
       p[2] = esdMuTrack->Pz();\r
       pos[0] = primary->GetX(); \r
       pos[1] = primary->GetY(); \r
       pos[2] = primary->GetZ();\r
\r
       // has to be changed once the muon pid is provided by the ESD\r
       for (Int_t i = 0; i < 10; pid[i++] = 0.); pid[AliAODTrack::kMuon]=1.;\r
       primary->AddDaughter(aodTrack =\r
     	  new(tracks[jTracks++]) AliAODTrack(0, // no ID provided\r
     					     0, // no label provided\r
     					     p,\r
     					     kTRUE,\r
     					     pos,\r
     					     kFALSE,\r
     					     NULL, // no covariance matrix provided\r
     					     (Short_t)-99, // no charge provided\r
     					     0, // no ITSClusterMap\r
     					     pid,\r
     					     primary,\r
     					     kTRUE,  // check if this is right\r
     					     kTRUE,  // not used for vertex fit\r
     					     AliAODTrack::kPrimary)\r
     	 );\r
\r
     aodTrack->ConvertAliPIDtoAODPID();  \r
     aodTrack->SetHitsPatternInTrigCh(esdMuTrack->GetHitsPatternInTrigCh());\r
     Int_t track2Trigger = esdMuTrack->GetMatchTrigger();\r
     aodTrack->SetMatchTrigger(track2Trigger);\r
     if (track2Trigger) \r
       aodTrack->SetChi2MatchTrigger(esdMuTrack->GetChi2MatchTrigger());\r
     else \r
       aodTrack->SetChi2MatchTrigger(0.);\r
 }\r
\r
    return;\r
}\r
\r
void AliAnalysisTaskESDMuonFilter::Terminate(Option_t */*option*/)\r
{\r
// Terminate analysis\r
//\r
    if (fDebug > 1) printf("AnalysisESDfilter: Terminate() \n");\r
}\r