AliEMCalPtTaskTrackSelectionESD.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.                  *
 **************************************************************************/
/*
 * Implementation of the track selection for the analysis on ESDs using
 * AliESDtrackCuts as underlying structure
 *
 * Author:
 *    Markus Fasel
 */
#include <TClonesArray.h>
#include "AliEMCalPtTaskTrackSelectionESD.h"
#include <memory>

#include "AliESDEvent.h"
#include "AliESDtrack.h"
#include "AliESDtrackCuts.h"
#include "AliLog.h"
#include "AliPicoTrack.h"


ClassImp(EMCalTriggerPtAnalysis::AliEMCalPtTaskTrackSelectionESD)

namespace EMCalTriggerPtAnalysis {

//______________________________________________________________________________
AliEMCalPtTaskTrackSelectionESD::AliEMCalPtTaskTrackSelectionESD():
    AliEMCalPtTaskVTrackSelection(),
    fTrackCuts(NULL)
{
  /*
   * Default constructor
   */
}

//______________________________________________________________________________
AliEMCalPtTaskTrackSelectionESD::AliEMCalPtTaskTrackSelectionESD(AliESDtrackCuts* cuts):
    AliEMCalPtTaskVTrackSelection(),
    fTrackCuts(cuts)
{
  /*
   * Constructor with cuts
   */
}

//______________________________________________________________________________
AliEMCalPtTaskTrackSelectionESD::AliEMCalPtTaskTrackSelectionESD(
    const AliEMCalPtTaskTrackSelectionESD& ref):
    AliEMCalPtTaskVTrackSelection(ref),
    fTrackCuts(NULL)
{
  /*
   * Copy constructor, creating a new cut object
   */
  if(ref.fTrackCuts) fTrackCuts = new AliESDtrackCuts(*(ref.fTrackCuts));
}

//______________________________________________________________________________
AliEMCalPtTaskTrackSelectionESD& AliEMCalPtTaskTrackSelectionESD::operator=(
    const AliEMCalPtTaskTrackSelectionESD& ref)
{
  /*
   * Assignment operator
   */
  AliEMCalPtTaskVTrackSelection::operator=(ref);
  if(&ref != this){
    if(fTrackCuts) {
      delete fTrackCuts;
      fTrackCuts = NULL;
    }
    if(ref.fTrackCuts) fTrackCuts = new AliESDtrackCuts(*(ref.fTrackCuts));
  }
  return *this;
}

//______________________________________________________________________________
AliEMCalPtTaskTrackSelectionESD::~AliEMCalPtTaskTrackSelectionESD() {
  /*
   * Destructor, deleting track cuts
   */
  if(fTrackCuts) delete fTrackCuts;
}

//______________________________________________________________________________
TObjArray* AliEMCalPtTaskTrackSelectionESD::GetAcceptedTracks(
    const TClonesArray* const tracks) {
  /*
   * Select tracks from a TClonesArray of input tracks
   *
   * @param tracks: TClonesArray of tracks (must not be null)
   * @return: TObjArray of selected tracks
   */
  if(!fListOfTracks) fListOfTracks = new TObjArray;
  else fListOfTracks->Clear();
  if(!fTrackCuts){
    AliError("Track cuts not provided");
    return fListOfTracks;
  }
  TIter trackIter(tracks);
  AliESDtrack *track(NULL);
  AliPicoTrack *picoTrack(NULL);
  TObject *containerObject(NULL);
  while((containerObject = dynamic_cast<TObject *>(trackIter()))){
    // Handle pico tracks
    if((picoTrack = dynamic_cast<AliPicoTrack *>(containerObject)))
      track = dynamic_cast<AliESDtrack *>(picoTrack->GetTrack());
    else
      track = dynamic_cast<AliESDtrack *>(containerObject);
    if(fTrackCuts->AcceptTrack(track)) fListOfTracks->AddLast(track);
  }
  return fListOfTracks;
}



//______________________________________________________________________________
TObjArray* AliEMCalPtTaskTrackSelectionESD::GetAcceptedTracks(const AliVEvent* const event) {
  /*
   * Select tracks from a virtual event
   *
   * @param event: AliESDEvent, via interface of virtual event (must not be null)
   * @return: TObjArray of selected tracks
   */
  if(!fListOfTracks) fListOfTracks = new TObjArray;
  else fListOfTracks->Clear();
  if(!fTrackCuts){
    AliError("Track cuts not provided");
    return fListOfTracks;
  }
  const AliESDEvent *esd = dynamic_cast<const AliESDEvent *>(event);
  if(!esd){
    AliError("Event not of type AliESDEvent");
    return fListOfTracks;
  }
  std::auto_ptr<TObjArray> accepted(fTrackCuts->GetAcceptedTracks(esd));
  TIter trackIter(accepted.get());
  AliESDtrack *track(NULL);
  while((track = dynamic_cast<AliESDtrack *>(trackIter()))){
    fListOfTracks->AddLast(track);
  }
  return fListOfTracks;
}

//______________________________________________________________________________
bool AliEMCalPtTaskTrackSelectionESD::IsTrackAccepted(AliVTrack* const trk) {
  /*
   * Check whether track is accepted
   *
   * @param trk: Track to check
   * @return: true if selected, false otherwise
   */
  AliESDtrack *esdt = dynamic_cast<AliESDtrack *>(trk);
  if(!esdt){
    AliPicoTrack *picoTrack = dynamic_cast<AliPicoTrack *>(trk);
    if(picoTrack)
      esdt = dynamic_cast<AliESDtrack*>(picoTrack->GetTrack());
    else{
      AliError("Neither Pico nor ESD track");
      return kFALSE;
    }
  }
  return fTrackCuts->AcceptTrack(esdt);
}

} /* namespace EMCalTriggerPtAnalysis */