AliEmcalTrackSelectionAOD.cxx

Go to the documentation of this file.

/**************************************************************************
 * Copyright(c) 1998-2015, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/
#include <TClonesArray.h>
#include <TBits.h>
#include <TObjArray.h>

#include <AliAODEvent.h>
#include <AliAODTrack.h>
#include <AliEmcalTrackSelectionAOD.h>
#include <AliESDtrack.h>
#include <AliPicoTrack.h>

ClassImp(AliEmcalTrackSelectionAOD)

AliEmcalTrackSelectionAOD::AliEmcalTrackSelectionAOD() :
  AliEmcalTrackSelection(),
  fFilterBits(0)
{
}

AliEmcalTrackSelectionAOD::AliEmcalTrackSelectionAOD(AliVCuts* cuts, UInt_t filterbits):
  AliEmcalTrackSelection(),
  fFilterBits(filterbits)
{
  AddTrackCuts(cuts);
}

TObjArray* AliEmcalTrackSelectionAOD::GetAcceptedTracks(const TClonesArray* const tracks) {
  if(!fListOfTracks) fListOfTracks = new TObjArray;
  else fListOfTracks->Clear();
  TIter trackIter(tracks);
  AliVTrack *track(NULL);
  while((track = dynamic_cast<AliVTrack *>(trackIter()))){
    if(IsTrackAccepted(track)) fListOfTracks->AddLast(track);
  }
  return fListOfTracks;
}

  TObjArray* AliEmcalTrackSelectionAOD::GetAcceptedTracks(const AliVEvent* const event) {
    if(!fListOfTracks) fListOfTracks = new TObjArray;
    else fListOfTracks->Clear();
    const AliAODEvent *aod = dynamic_cast<const AliAODEvent *>(event);
    if(!aod){
      AliError("Event not of type AliAODEvent");
      return fListOfTracks;
    }
    for(int itrk = 0; itrk < event->GetNumberOfTracks(); itrk++){
      AliVTrack *trk = dynamic_cast<AliVTrack *>(event->GetTrack(itrk));
      if(IsTrackAccepted(trk)) fListOfTracks->AddLast(trk);
    }
    return fListOfTracks;
  }

bool AliEmcalTrackSelectionAOD::IsTrackAccepted(AliVTrack * const trk){
  AliAODTrack *aodt = dynamic_cast<AliAODTrack *>(trk);
  if(!aodt){
    AliPicoTrack *picotrack = dynamic_cast<AliPicoTrack *>(trk);
    if(picotrack) aodt = dynamic_cast<AliAODTrack *>(picotrack->GetTrack());
    else{
      AliError("Track neither AOD track nor pico track");
      return kFALSE;
    }
  }
  if(!aodt){
    AliError("Failed getting AOD track");
    return kFALSE;
  }

  TBits selectedMap(64);
  Int_t cutcounter(0);
  if(fFilterBits){
    if(aodt->TestFilterBit(fFilterBits)) selectedMap.SetBitNumber(cutcounter++);
  }
  if(fListOfCuts){
    for(TIter cutIter = TIter(fListOfCuts).Begin(); cutIter != TIter::End(); ++cutIter){
      AliVCuts *trackCuts = static_cast<AliVCuts *>(*cutIter);
      if(trackCuts->IsA() == AliESDtrackCuts::Class()){
        // If track cuts are AliESDtrackCuts, the track needs to be converted to an AliESDtrack before
        AliESDtrack copyTrack(aodt);
        if(trackCuts->IsSelected(&copyTrack)) selectedMap.SetBitNumber(cutcounter++);
      } else{
        if(trackCuts->IsSelected(aodt)) selectedMap.SetBitNumber(cutcounter++);
      }
    }
  }
  if(fSelectionModeAny){
    // In case of ANY one of the cuts need to be fulfilled (equivalent to one but set)
    return selectedMap.CountBits() > 0;
  } else {
    // In case of ALL all of the cuts need to be fulfilled (equivalent to all bits set)
    return selectedMap.CountBits() == cutcounter;
  }
}