AliEMCalPtTaskTrackSelectionAOD.cxx

Go to the documentation of this file.

/**************************************************************************
 * Copyright(c) 1998-2014, 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 <TObjArray.h>

#include <AliAODEvent.h>
#include <AliAODTrack.h>
#include <AliESDtrack.h>
#include <AliPicoTrack.h>
#include "AliEMCalPtTaskTrackSelectionAOD.h"

ClassImp(EMCalTriggerPtAnalysis::AliEMCalPtTaskTrackSelectionAOD)

namespace EMCalTriggerPtAnalysis {

AliEMCalPtTaskTrackSelectionAOD::AliEMCalPtTaskTrackSelectionAOD() :
  AliEMCalPtTaskVTrackSelection(),
  fFilterBits(0)
{
}

AliEMCalPtTaskTrackSelectionAOD::AliEMCalPtTaskTrackSelectionAOD(AliESDtrackCuts* cuts, UInt_t filterbits):
  AliEMCalPtTaskVTrackSelection(),
  fFilterBits(filterbits)
{
  AddTrackCuts(cuts);
}

TObjArray* AliEMCalPtTaskTrackSelectionAOD::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* AliEMCalPtTaskTrackSelectionAOD::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 AliEMCalPtTaskTrackSelectionAOD::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;
  }
  if(fFilterBits && !aodt->TestFilterBit(fFilterBits)) return kFALSE;
  Bool_t result = kTRUE;
  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)) result = kFALSE;
      } else{
        if(!trackCuts->IsSelected(aodt)) result = kFALSE;
      }
    }

  }
  return result;
}

} /* namespace EMCalTriggerPtAnalysis */