/**************************************************************************
 * 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 */