1 /**************************************************************************
2 * Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
16 /////////////////////////////////////////////////////////////
18 // class used to extract and store reco info of dimu candidate
20 // Author: X-M. Zhang, zhang@clermont.in2p3.fr
21 // zhangxm@iopp.ccnu.edu.cn
22 /////////////////////////////////////////////////////////////
24 #include <TDatabasePDG.h>
25 #include <TLorentzVector.h>
27 #include "AliMuonInfoStoreRD.h"
28 #include "AliDimuInfoStoreRD.h"
30 ClassImp(AliDimuInfoStoreRD)
32 const TString AliDimuInfoStoreRD::fgkStdBranchName("DimuRD");
33 Double_t AliDimuInfoStoreRD::fgCutd[12] = {-999999., 999999.,
40 //-----------------------------------------------------------------------------
41 AliDimuInfoStoreRD::AliDimuInfoStoreRD() :
48 // default constructor
50 for (Int_t i=2; i--;) fMuonRef[i] = 0;
53 //-----------------------------------------------------------------------------
54 AliDimuInfoStoreRD::AliDimuInfoStoreRD(AliMuonInfoStoreRD* const trk0, AliMuonInfoStoreRD* const trk1) :
61 // default constructor
68 //-----------------------------------------------------------------------------
69 AliDimuInfoStoreRD::AliDimuInfoStoreRD(const AliDimuInfoStoreRD &src) :
71 fMomentum(src.fMomentum),
78 for (Int_t i=2; i--;) fMuonRef[i] = src.fMuonRef[i];
81 //-----------------------------------------------------------------------------
82 AliDimuInfoStoreRD& AliDimuInfoStoreRD::operator=(const AliDimuInfoStoreRD &src)
85 // assignment constructor
87 if(&src==this) return *this;
89 fMomentum = src.fMomentum;
90 fCharge = src.fCharge;
92 for (Int_t i=2; i--;) fMuonRef[i] = src.fMuonRef[i];
97 //-----------------------------------------------------------------------------
98 AliDimuInfoStoreRD::~AliDimuInfoStoreRD()
105 //-----------------------------------------------------------------------------
106 void AliDimuInfoStoreRD::FillDimuInfo()
108 // fill dimuon candidate info from the corresponding two muon tracks
110 AliMuonInfoStoreRD *trk0 = (AliMuonInfoStoreRD*)fMuonRef[0].GetObject();
111 AliMuonInfoStoreRD *trk1 = (AliMuonInfoStoreRD*)fMuonRef[1].GetObject();
113 fMomentum = trk0->Momentum() + trk1->Momentum();
114 fCharge = trk0->Charge() + trk1->Charge();
116 Double_t mMu = TDatabasePDG::Instance()->GetParticle(13)->Mass();
117 TLorentzVector lorentzP0, lorentzP1, lorentzP;
118 lorentzP0.SetVectM(trk0->Momentum(), mMu);
119 lorentzP1.SetVectM(trk1->Momentum(), mMu);
120 lorentzP = lorentzP0 + lorentzP1;
121 fInvM = lorentzP.Mag();
128 //-----------------------------------------------------------------------------
129 Bool_t AliDimuInfoStoreRD::DimuSelection()
131 // select dimuon candidates according to the corresponding two muon tracks cuts
133 Double_t cutsOld[12];
134 AliMuonInfoStoreRD::SelectionCust(cutsOld);
135 AliMuonInfoStoreRD::SetSelectionCuts(AliDimuInfoStoreRD::fgCutd);
136 AliMuonInfoStoreRD *trk0 = (AliMuonInfoStoreRD*)fMuonRef[0].GetObject();
137 AliMuonInfoStoreRD *trk1 = (AliMuonInfoStoreRD*)fMuonRef[1].GetObject();
138 if (!trk0->MuonSelection()) { AliMuonInfoStoreRD::SetSelectionCuts(cutsOld); return kFALSE; }
139 if (!trk1->MuonSelection()) { AliMuonInfoStoreRD::SetSelectionCuts(cutsOld); return kFALSE; }
141 AliMuonInfoStoreRD::SetSelectionCuts(cutsOld);