PWG/muon/AliMuonPairCuts.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   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  **************************************************************************/
  15
  16 #include "AliMuonPairCuts.h"
  17
  18 #include "TMath.h"
  19 #include "TList.h"
  20 #include "TLorentzVector.h"
  21 #include "TArrayI.h"
  22
  23 #include "AliLog.h"
  24 #include "AliVParticle.h"
  25 #include "AliInputEventHandler.h"
  26
  27 /// \cond CLASSIMP
  28 ClassImp(AliMuonPairCuts) // Class implementation in ROOT context
  29 /// \endcond
  30
  31 //________________________________________________________________________
  32 AliMuonPairCuts::AliMuonPairCuts() :
  33   AliAnalysisCuts(),
  34   fMuonTrackCuts()
  35 {
  36   /// Default ctor.
  37 }
  38
  39 //________________________________________________________________________
  40 AliMuonPairCuts::AliMuonPairCuts(const char* name, const char* title) :
  41 AliAnalysisCuts(name, title),
  42 fMuonTrackCuts("muonTrackCutInPair","muonTrackCutInPair")
  43 {
  44   /// Constructor
  45   SetDefaultFilterMask();
  46 }
  47
  48
  49 //________________________________________________________________________
  50 AliMuonPairCuts::AliMuonPairCuts(const char* name, const char* title, const AliMuonTrackCuts& trackCuts) :
  51 AliAnalysisCuts(name, title),
  52 fMuonTrackCuts(trackCuts)
  53 {
  54   /// Test Constructor
  55   SetDefaultFilterMask();
  56 }
  57
  58
  59 //________________________________________________________________________
  60 AliMuonPairCuts::AliMuonPairCuts(const AliMuonPairCuts& obj) :
  61   AliAnalysisCuts(obj),
  62   fMuonTrackCuts(obj.fMuonTrackCuts)
  63 {
  64   /// Copy constructor
  65 }
  66
  67
  68 //________________________________________________________________________
  69 AliMuonPairCuts& AliMuonPairCuts::operator=(const AliMuonPairCuts& obj)
  70 {
  71   /// Assignment operator
  72   if ( this != &obj ) {
  73     AliAnalysisCuts::operator=(obj);
  74     fMuonTrackCuts = obj.fMuonTrackCuts;
  75   }
  76   return *this;
  77 }
  78
  79
  80 //________________________________________________________________________
  81 AliMuonPairCuts::~AliMuonPairCuts()
  82 {
  83   /// Destructor
  84 }
  85
  86 //________________________________________________________________________
  87 Bool_t AliMuonPairCuts::SetRun ( const AliInputEventHandler* eventHandler )
  88 {
  89   /// Get parameters from OADB for runNumber
  90   return fMuonTrackCuts.SetRun(eventHandler);
  91 }
  92
  93
  94 //________________________________________________________________________
  95 Bool_t AliMuonPairCuts::IsSelected( TObject* /*obj*/ )
  96 {
  97   /// Not implemented
  98   AliError("Requires a list of two AliVParticle");
  99   return kFALSE;
 100 }
 101
 102 //________________________________________________________________________
 103 Bool_t AliMuonPairCuts::IsSelected( TList* list )
 104 {
 105   /// Pair is selected
 106   UInt_t filterMask = GetFilterMask();
 107   UInt_t selectionMask = GetSelectionMask(list);
 108
 109   return ( ( selectionMask & filterMask ) == filterMask );
 110 }
 111
 112
 113 //________________________________________________________________________
 114 UInt_t AliMuonPairCuts::GetSelectionMask( const TObject* obj )
 115 {
 116   /// Get selection mask (overloaded function)
 117   const TList* list = static_cast<const TList*> ( obj );
 118   if ( list->GetEntries() < 2 ) {
 119     AliError("Requires a list of two AliVParticle");
 120     return 0;
 121   }
 122
 123   return GetSelectionMask(list->At(0), list->At(1));
 124
 125 }
 126
 127
 128 //________________________________________________________________________
 129 UInt_t AliMuonPairCuts::GetSelectionMask( const TObject* track1, const TObject* track2 )
 130 {
 131   /// Get selection mask from AliVParticles
 132
 133   UInt_t selectionMask = 0;
 134
 135   UInt_t maskTrack1 = fMuonTrackCuts.GetSelectionMask(track1);
 136   UInt_t maskTrack2 = fMuonTrackCuts.GetSelectionMask(track2);
 137
 138   UInt_t maskAND = maskTrack1 & maskTrack2;
 139   UInt_t maskOR = maskTrack1 | maskTrack2;
 140   if ( maskAND & AliMuonTrackCuts::kMuEta ) selectionMask |= kBothMuEta;
 141   if ( maskAND & AliMuonTrackCuts::kMuThetaAbs ) selectionMask |= kBothMuThetaAbs;
 142   if ( maskAND & AliMuonTrackCuts::kMuPdca ) selectionMask |= kBothMuPdca;
 143   if ( maskAND & AliMuonTrackCuts::kMuTrackChiSquare ) selectionMask |= kBothMuTrackChiSquare;
 144   if ( maskAND & AliMuonTrackCuts::kMuMatchApt ) selectionMask |= kBothMuMatchApt;
 145   if ( maskAND & AliMuonTrackCuts::kMuMatchLpt ) selectionMask |= kBothMuMatchLpt;
 146   if ( maskAND & AliMuonTrackCuts::kMuMatchHpt ) selectionMask |= kBothMuMatchHpt;
 147   if ( maskOR & AliMuonTrackCuts::kMuMatchApt ) selectionMask |= kOneMuMatchApt;
 148   if ( maskOR & AliMuonTrackCuts::kMuMatchLpt ) selectionMask |= kOneMuMatchLpt;
 149   if ( maskOR & AliMuonTrackCuts::kMuMatchHpt ) selectionMask |= kOneMuMatchHpt;
 150
 151   TLorentzVector vec[2];
 152   Double_t chargeProduct = 1.;
 153   for ( Int_t itrack=0; itrack<2; ++itrack ) {
 154     const AliVParticle* track = static_cast<const AliVParticle*> ( ( itrack == 0 ) ? track1 : track2 );
 155     chargeProduct *= track->Charge();
 156     Double_t trackP = track->P();
 157     Double_t energy = TMath::Sqrt(trackP*trackP + MuonMass2());
 158     vec[itrack].SetPxPyPzE(track->Px(), track->Py(), track->Pz(), energy);
 159   }
 160
 161   if ( chargeProduct < 0. ) selectionMask |= kDimuUnlikeSign;
 162
 163   TLorentzVector vecPair = vec[0] + vec[1];
 164   Double_t rapidity = vecPair.Rapidity();
 165   if ( rapidity > -4. && rapidity < -2.5 ) selectionMask |= kDimuRapidity;
 166
 167   return selectionMask;
 168 }
 169
 170
 171 //_____________________________________________________________________________
 172 Double_t AliMuonPairCuts::MuonMass2() const
 173 {
 174   /// A usefull constant
 175   static Double_t m2 = 1.11636129640000012e-02;
 176   return m2;
 177 }
 178
 179
 180 //________________________________________________________________________
 181 void AliMuonPairCuts::SetDefaultFilterMask ()
 182 {
 183   /// Standard cuts for muon pair
 184   SetFilterMask ( kBothMuEta | kBothMuThetaAbs | kBothMuMatchLpt | kDimuUnlikeSign | kDimuRapidity );
 185 }
 186
 187 //________________________________________________________________________
 188 void AliMuonPairCuts::SetIsMC ( Bool_t isMC )
 189 {
 190   /// Set Is MC
 191   fMuonTrackCuts.SetIsMC(isMC);
 192 }
 193
 194 //________________________________________________________________________
 195 Bool_t AliMuonPairCuts::TrackPtCutMatchTrigClass ( const AliVParticle* track1, const AliVParticle* track2, const TArrayI ptCutFromClass ) const
 196 {
 197   /// Check if track pair passes the trigger pt cut level used in the trigger class
 198   Bool_t matchTrig1 = fMuonTrackCuts.TrackPtCutMatchTrigClass(track1, ptCutFromClass);
 199   Bool_t matchTrig2 = fMuonTrackCuts.TrackPtCutMatchTrigClass(track2, ptCutFromClass);
 200
 201   Bool_t matchTrackerPt1 = kTRUE, matchTrackerPt2 = kTRUE;
 202   if ( IsApplySharpPtCutInMatching() ) {
 203     matchTrackerPt1 = ( track1->Pt() >= fMuonTrackCuts.GetMuonTrackCutsParam().GetSharpPtCut(ptCutFromClass[0]-1,kFALSE) );
 204     matchTrackerPt2 = ( track2->Pt() >= fMuonTrackCuts.GetMuonTrackCutsParam().GetSharpPtCut(ptCutFromClass[0]-1,kFALSE) );
 205   }
 206
 207   matchTrig1 = ( matchTrig1 && matchTrackerPt1 );
 208   matchTrig2 = ( matchTrig2 && matchTrackerPt2 );
 209
 210   Bool_t passCut = ( ptCutFromClass[1]>0 ) ? ( matchTrig1 && matchTrig2 ) : ( matchTrig1 || matchTrig2 );
 211
 212   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));
 213   return passCut;
 214 }
 215
 216 //________________________________________________________________________
 217 void AliMuonPairCuts::Print(Option_t* option) const
 218 {
 219   //
 220   /// Print info
 221   //
 222
 223   TString sopt(option);
 224   sopt.ToLower();
 225   if ( sopt.IsNull() || sopt.Contains("*") || sopt.Contains("all") ) sopt += " pair trackparam trackmask";
 226   if ( sopt.Contains("pair") ) {
 227     UInt_t filterMask = GetFilterMask();
 228     printf(" *** Muon pair filter mask: *** \n");
 229     printf("  0x%x\n", filterMask);
 230     if ( filterMask & kBothMuEta ) printf("  mu1 && mu2 pass eta cut\n");
 231     if ( filterMask & kBothMuThetaAbs ) printf("  mu1 && mu2 pass theta_abs cut\n");
 232     if ( filterMask & kBothMuPdca ) printf("  mu1 && mu2 pass pxDCA cut\n");
 233     if ( filterMask & kBothMuTrackChiSquare ) printf("  Chi2 cut on track\n");
 234     if ( filterMask & kBothMuMatchApt ) printf("  mu1 && mu2 match Apt\n");
 235     if ( filterMask & kBothMuMatchLpt ) printf("  mu1 && mu2 match Lpt\n");
 236     if ( filterMask & kBothMuMatchHpt ) printf("  mu1 && mu2 match Hpt\n");
 237     if ( filterMask & kOneMuMatchApt ) printf("  mu1 || mu2 match Apt\n");
 238     if ( filterMask & kOneMuMatchLpt ) printf("  mu1 || mu2 match Lpt\n");
 239     if ( filterMask & kOneMuMatchHpt ) printf("  mu1 || mu2 match Hpt\n");
 240     if ( filterMask & kDimuUnlikeSign ) printf("  Unlike sign\n");
 241     if ( filterMask & kDimuRapidity ) printf("  -4 < y_{mumu} < -2.5\n");
 242     printf(" ******************** \n");
 243   }
 244
 245   if ( sopt.Contains("track") )
 246     fMuonTrackCuts.Print(sopt.Data());
 247 }