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 **************************************************************************/
18 /////////////////////////////////////////////////////////////
20 // class used to extract and store reco info of dimu candidate
22 // Author: X-M. Zhang, zhang@clermont.in2p3.fr
23 // zhangxm@iopp.ccnu.edu.cn
24 /////////////////////////////////////////////////////////////
26 #include <TDatabasePDG.h>
27 #include <TLorentzVector.h>
29 #include "AliMuonInfoStoreRD.h"
30 #include "AliDimuInfoStoreRD.h"
32 ClassImp(AliDimuInfoStoreRD)
34 const TString AliDimuInfoStoreRD::fgkStdBranchName("DimuRD");
35 Double_t AliDimuInfoStoreRD::fgCutd[16] = {-999999., 999999.,
44 //-----------------------------------------------------------------------------
45 AliDimuInfoStoreRD::AliDimuInfoStoreRD() :
52 // default constructor
54 for (Int_t i=2; i--;) fMuonRef[i] = 0;
57 //-----------------------------------------------------------------------------
58 AliDimuInfoStoreRD::AliDimuInfoStoreRD(AliMuonInfoStoreRD* const trk0, AliMuonInfoStoreRD* const trk1) :
65 // default constructor
72 //-----------------------------------------------------------------------------
73 AliDimuInfoStoreRD::AliDimuInfoStoreRD(const AliDimuInfoStoreRD &src) :
75 fMomentum(src.fMomentum),
82 for (Int_t i=2; i--;) fMuonRef[i] = src.fMuonRef[i];
85 //-----------------------------------------------------------------------------
86 AliDimuInfoStoreRD& AliDimuInfoStoreRD::operator=(const AliDimuInfoStoreRD &src)
89 // assignment constructor
91 if(&src==this) return *this;
93 fMomentum = src.fMomentum;
94 fCharge = src.fCharge;
96 for (Int_t i=2; i--;) fMuonRef[i] = src.fMuonRef[i];
101 //-----------------------------------------------------------------------------
102 AliDimuInfoStoreRD::~AliDimuInfoStoreRD()
109 //-----------------------------------------------------------------------------
110 void AliDimuInfoStoreRD::FillDimuInfo()
112 // fill dimuon candidate info from the corresponding two muon tracks
114 AliMuonInfoStoreRD *trk0 = (AliMuonInfoStoreRD*)fMuonRef[0].GetObject();
115 AliMuonInfoStoreRD *trk1 = (AliMuonInfoStoreRD*)fMuonRef[1].GetObject();
117 fMomentum = trk0->Momentum() + trk1->Momentum();
118 fCharge = trk0->Charge() + trk1->Charge();
120 Double_t mMu = TDatabasePDG::Instance()->GetParticle(13)->Mass();
121 TLorentzVector lorentzP0, lorentzP1, lorentzP;
122 lorentzP0.SetVectM(trk0->Momentum(), mMu);
123 lorentzP1.SetVectM(trk1->Momentum(), mMu);
124 lorentzP = lorentzP0 + lorentzP1;
125 fInvM = lorentzP.Mag();
132 //-----------------------------------------------------------------------------
133 Bool_t AliDimuInfoStoreRD::IsSelected()
135 // select dimuon candidates according to the corresponding two muon tracks cuts
137 Double_t cutsOld[16];
138 AliMuonInfoStoreRD::SelectionCust(cutsOld);
139 AliMuonInfoStoreRD::SetSelectionCuts(AliDimuInfoStoreRD::fgCutd);
140 AliMuonInfoStoreRD *trk0 = (AliMuonInfoStoreRD*)fMuonRef[0].GetObject();
141 AliMuonInfoStoreRD *trk1 = (AliMuonInfoStoreRD*)fMuonRef[1].GetObject();
142 if (!trk0->IsSelected()) { AliMuonInfoStoreRD::SetSelectionCuts(cutsOld); return kFALSE; }
143 if (!trk1->IsSelected()) { AliMuonInfoStoreRD::SetSelectionCuts(cutsOld); return kFALSE; }
145 AliMuonInfoStoreRD::SetSelectionCuts(cutsOld);