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(),
    fTrackCuts(NULL),
    fFilterBits(0)
  {
  }

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

  AliEMCalPtTaskTrackSelectionAOD::AliEMCalPtTaskTrackSelectionAOD(const AliEMCalPtTaskTrackSelectionAOD& ref) :
    AliEMCalPtTaskVTrackSelection(ref),
    fTrackCuts(NULL),
    fFilterBits(ref.fFilterBits)
  {
    if(ref.fTrackCuts) fTrackCuts = new AliESDtrackCuts(*(ref.fTrackCuts));
  }

  AliEMCalPtTaskTrackSelectionAOD& AliEMCalPtTaskTrackSelectionAOD::operator=(const AliEMCalPtTaskTrackSelectionAOD& ref) {
    AliEMCalPtTaskVTrackSelection::operator=(ref);
    if(this != &ref){
      fFilterBits = ref.fFilterBits;
      if(fTrackCuts) {
        delete fTrackCuts;
        fTrackCuts = NULL;
      }
      if(ref.fTrackCuts) fTrackCuts = new AliESDtrackCuts(*(ref.fTrackCuts));
    }
    return *this;
  }

  AliEMCalPtTaskTrackSelectionAOD::~AliEMCalPtTaskTrackSelectionAOD() {
    if(fTrackCuts) delete fTrackCuts;
    if(fListOfTracks) delete fListOfTracks;
  }

  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;
    if(fTrackCuts){
      AliESDtrack copyTrack(aodt);
      return fTrackCuts->AcceptTrack(&copyTrack);
    }
    return kTRUE;
  }


} /* namespace EMCalTriggerPtAnalysis */