1 /**************************************************************************
2 * Copyright(c) 1998-1999, 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 #include "AliMuonPairCuts.h"
20 #include "TLorentzVector.h"
24 #include "AliVParticle.h"
25 #include "AliInputEventHandler.h"
28 ClassImp(AliMuonPairCuts) // Class implementation in ROOT context
31 //________________________________________________________________________
32 AliMuonPairCuts::AliMuonPairCuts() :
39 //________________________________________________________________________
40 AliMuonPairCuts::AliMuonPairCuts(const char* name, const char* title) :
41 AliAnalysisCuts(name, title),
42 fMuonTrackCuts("muonTrackCutInPair","muonTrackCutInPair")
45 SetDefaultFilterMask();
49 //________________________________________________________________________
50 AliMuonPairCuts::AliMuonPairCuts(const char* name, const char* title, const AliMuonTrackCuts& trackCuts) :
51 AliAnalysisCuts(name, title),
52 fMuonTrackCuts(trackCuts)
55 SetDefaultFilterMask();
59 //________________________________________________________________________
60 AliMuonPairCuts::AliMuonPairCuts(const AliMuonPairCuts& obj) :
62 fMuonTrackCuts(obj.fMuonTrackCuts)
68 //________________________________________________________________________
69 AliMuonPairCuts& AliMuonPairCuts::operator=(const AliMuonPairCuts& obj)
71 /// Assignment operator
73 AliAnalysisCuts::operator=(obj);
74 fMuonTrackCuts = obj.fMuonTrackCuts;
80 //________________________________________________________________________
81 AliMuonPairCuts::~AliMuonPairCuts()
86 //________________________________________________________________________
87 Bool_t AliMuonPairCuts::SetRun ( const AliInputEventHandler* eventHandler )
89 /// Get parameters from OADB for runNumber
90 return fMuonTrackCuts.SetRun(eventHandler);
94 //________________________________________________________________________
95 Bool_t AliMuonPairCuts::IsSelected( TObject* /*obj*/ )
98 AliError("Requires a list of two AliVParticle");
102 //________________________________________________________________________
103 Bool_t AliMuonPairCuts::IsSelected( TList* list )
106 UInt_t filterMask = GetFilterMask();
107 UInt_t selectionMask = GetSelectionMask(list);
109 return ( ( selectionMask & filterMask ) == filterMask );
112 //________________________________________________________________________
113 Bool_t AliMuonPairCuts::IsSelected( TObject* track1, TObject* track2 )
116 UInt_t filterMask = GetFilterMask();
117 UInt_t selectionMask = GetSelectionMask(track1, track2);
119 return ( ( selectionMask & filterMask ) == filterMask );
123 //________________________________________________________________________
124 UInt_t AliMuonPairCuts::GetSelectionMask( const TObject* obj )
126 /// Get selection mask (overloaded function)
127 const TList* list = static_cast<const TList*> ( obj );
128 if ( list->GetEntries() < 2 ) {
129 AliError("Requires a list of two AliVParticle");
133 return GetSelectionMask(list->At(0), list->At(1));
138 //________________________________________________________________________
139 UInt_t AliMuonPairCuts::GetSelectionMask( const TObject* track1, const TObject* track2 )
141 /// Get selection mask from AliVParticles
143 UInt_t selectionMask = 0;
145 UInt_t maskTrack1 = fMuonTrackCuts.GetSelectionMask(track1);
146 UInt_t maskTrack2 = fMuonTrackCuts.GetSelectionMask(track2);
148 UInt_t maskAND = maskTrack1 & maskTrack2;
149 UInt_t maskOR = maskTrack1 | maskTrack2;
150 if ( maskAND & AliMuonTrackCuts::kMuEta ) selectionMask |= kBothMuEta;
151 if ( maskAND & AliMuonTrackCuts::kMuThetaAbs ) selectionMask |= kBothMuThetaAbs;
152 if ( maskAND & AliMuonTrackCuts::kMuPdca ) selectionMask |= kBothMuPdca;
153 if ( maskAND & AliMuonTrackCuts::kMuTrackChiSquare ) selectionMask |= kBothMuTrackChiSquare;
154 if ( maskAND & AliMuonTrackCuts::kMuMatchApt ) selectionMask |= kBothMuMatchApt;
155 if ( maskAND & AliMuonTrackCuts::kMuMatchLpt ) selectionMask |= kBothMuMatchLpt;
156 if ( maskAND & AliMuonTrackCuts::kMuMatchHpt ) selectionMask |= kBothMuMatchHpt;
157 if ( maskOR & AliMuonTrackCuts::kMuMatchApt ) selectionMask |= kOneMuMatchApt;
158 if ( maskOR & AliMuonTrackCuts::kMuMatchLpt ) selectionMask |= kOneMuMatchLpt;
159 if ( maskOR & AliMuonTrackCuts::kMuMatchHpt ) selectionMask |= kOneMuMatchHpt;
161 TLorentzVector vec[2];
162 Double_t chargeProduct = 1.;
163 for ( Int_t itrack=0; itrack<2; ++itrack ) {
164 const AliVParticle* track = static_cast<const AliVParticle*> ( ( itrack == 0 ) ? track1 : track2 );
165 chargeProduct *= track->Charge();
166 Double_t trackP = track->P();
167 Double_t energy = TMath::Sqrt(trackP*trackP + MuonMass2());
168 vec[itrack].SetPxPyPzE(track->Px(), track->Py(), track->Pz(), energy);
171 if ( chargeProduct < 0. ) selectionMask |= kDimuUnlikeSign;
173 TLorentzVector vecPair = vec[0] + vec[1];
174 Double_t rapidity = vecPair.Rapidity();
175 if ( rapidity > -4. && rapidity < -2.5 ) selectionMask |= kDimuRapidity;
177 return selectionMask;
181 //_____________________________________________________________________________
182 Double_t AliMuonPairCuts::MuonMass2() const
184 /// A usefull constant
185 static Double_t m2 = 1.11636129640000012e-02;
190 //________________________________________________________________________
191 void AliMuonPairCuts::SetDefaultFilterMask ()
193 /// Standard cuts for muon pair
194 SetFilterMask ( kBothMuEta | kBothMuThetaAbs | kBothMuMatchLpt | kDimuUnlikeSign | kDimuRapidity );
197 //________________________________________________________________________
198 void AliMuonPairCuts::SetIsMC ( Bool_t isMC )
201 fMuonTrackCuts.SetIsMC(isMC);
204 //________________________________________________________________________
205 Bool_t AliMuonPairCuts::TrackPtCutMatchTrigClass ( const AliVParticle* track1, const AliVParticle* track2, const TArrayI ptCutFromClass ) const
207 /// Check if track pair passes the trigger pt cut level used in the trigger class
208 Bool_t matchTrig1 = fMuonTrackCuts.TrackPtCutMatchTrigClass(track1, ptCutFromClass);
209 Bool_t matchTrig2 = fMuonTrackCuts.TrackPtCutMatchTrigClass(track2, ptCutFromClass);
211 Bool_t matchTrackerPt1 = kTRUE, matchTrackerPt2 = kTRUE;
212 if ( IsApplySharpPtCutInMatching() ) {
213 matchTrackerPt1 = ( track1->Pt() >= fMuonTrackCuts.GetMuonTrackCutsParam().GetSharpPtCut(ptCutFromClass[0]-1,kFALSE) );
214 matchTrackerPt2 = ( track2->Pt() >= fMuonTrackCuts.GetMuonTrackCutsParam().GetSharpPtCut(ptCutFromClass[0]-1,kFALSE) );
217 matchTrig1 = ( matchTrig1 && matchTrackerPt1 );
218 matchTrig2 = ( matchTrig2 && matchTrackerPt2 );
220 Bool_t passCut = ( ptCutFromClass[1]>0 ) ? ( matchTrig1 && matchTrig2 ) : ( matchTrig1 || matchTrig2 );
222 AliDebug(1,Form("Class matchTrig %i %i trackMatchTrig %i %i trackPt %g %g (required %i) passCut %i", ptCutFromClass[0], ptCutFromClass[1], matchTrig1, matchTrig2, track1->Pt(), track2->Pt(), IsApplySharpPtCutInMatching(),passCut));
226 //________________________________________________________________________
227 void AliMuonPairCuts::Print(Option_t* option) const
233 TString sopt(option);
235 if ( sopt.IsNull() || sopt.Contains("*") || sopt.Contains("all") ) sopt += " pair trackparam trackmask";
236 if ( sopt.Contains("pair") ) {
237 UInt_t filterMask = GetFilterMask();
238 printf(" *** Muon pair filter mask: *** \n");
239 printf(" 0x%x\n", filterMask);
240 if ( filterMask & kBothMuEta ) printf(" mu1 && mu2 pass eta cut\n");
241 if ( filterMask & kBothMuThetaAbs ) printf(" mu1 && mu2 pass theta_abs cut\n");
242 if ( filterMask & kBothMuPdca ) printf(" mu1 && mu2 pass pxDCA cut\n");
243 if ( filterMask & kBothMuTrackChiSquare ) printf(" Chi2 cut on track\n");
244 if ( filterMask & kBothMuMatchApt ) printf(" mu1 && mu2 match Apt\n");
245 if ( filterMask & kBothMuMatchLpt ) printf(" mu1 && mu2 match Lpt\n");
246 if ( filterMask & kBothMuMatchHpt ) printf(" mu1 && mu2 match Hpt\n");
247 if ( filterMask & kOneMuMatchApt ) printf(" mu1 || mu2 match Apt\n");
248 if ( filterMask & kOneMuMatchLpt ) printf(" mu1 || mu2 match Lpt\n");
249 if ( filterMask & kOneMuMatchHpt ) printf(" mu1 || mu2 match Hpt\n");
250 if ( filterMask & kDimuUnlikeSign ) printf(" Unlike sign\n");
251 if ( filterMask & kDimuRapidity ) printf(" -4 < y_{mumu} < -2.5\n");
252 printf(" ******************** \n");
255 if ( sopt.Contains("track") )
256 fMuonTrackCuts.Print(sopt.Data());