AliTrackContainerV0.cxx

Go to the documentation of this file.

/*************************************************************************
* Copyright(c) 1998-2016, 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 <iostream>
#include <vector>
#include <TClonesArray.h>
#include <AliLog.h>
#include <AliAODTrack.h>
#include <AliAODv0.h>

#include "AliTrackContainerV0.h"

ClassImp(AliTrackContainerV0);

AliTrackContainerV0::AliTrackContainerV0() :
  AliTrackContainer(),
  fFilterDaughterTracks(0),
  fEvent(0),
  fV0s(0),
  fDaughterVec()
{
  // Constructor.

  fBaseClassName = "AliAODTrack";
  SetClassName("AliAODTrack");
}

AliTrackContainerV0::AliTrackContainerV0(const char *name) :
  AliTrackContainer(name),
  fFilterDaughterTracks(0),
  fEvent(0),
  fV0s(0),
  fDaughterVec()
{
  // Constructor.

  fBaseClassName = "AliAODTrack";
  SetClassName("AliAODTrack");
}

void AliTrackContainerV0::SetArray(const AliVEvent *event)
{
  AliTrackContainer::SetArray(event);

  if (!fFilterDaughterTracks)
    return;

  fEvent = dynamic_cast<const AliAODEvent*>(event);
}

void AliTrackContainerV0::NextEvent()
{
  AliTrackContainer::NextEvent();
  
  if (fFilterDaughterTracks) {
    // V0s daughter tracks will be removed from track sample

    if (!fEvent) {
      AliWarning("fEvent is not valid!");
      return;
    }

    fV0s = dynamic_cast<TClonesArray*>(fEvent->GetV0s());
    if (!fV0s) {
      AliWarning("fV0s is not valid!");
      return;
    }

    fDaughterVec.clear();

    Int_t iNumV0s = fV0s->GetEntriesFast();

    for(Int_t iV0 = 0; iV0 < iNumV0s; iV0++)
      ExtractDaughters(dynamic_cast<AliAODv0*>(fV0s->At(iV0)));
  }
}

void AliTrackContainerV0::ExtractDaughters(AliAODv0* cand)
{
  if (!cand)
    return;

  for (Int_t i = 0; i < 2; i++) {
    AliAODTrack* track = dynamic_cast<AliAODTrack*>(cand->GetDaughter(i));
    
    if(!track)
    {
      AliWarning("Track is not valid! Skipping candidate.");
      return;
    }

    fDaughterVec.push_back(track->GetID());
  }
}

Bool_t AliTrackContainerV0::ApplyTrackCuts(const AliVTrack* vp, UInt_t &rejectionReason) const
{
  const AliAODTrack* track = dynamic_cast<const AliAODTrack*>(vp);

  if (!track)
    return kFALSE;
  
  if (AliTrackContainer::ApplyTrackCuts(vp, rejectionReason)) {
    if (IsV0Daughter(track)) {
      // track is one of the V0 daughter - will be rejected
      return kFALSE;
    } else {
      return kTRUE;
    }
  } else {
    return kFALSE;
  }
}

Bool_t AliTrackContainerV0::IsV0Daughter(const AliAODTrack* track) const
{
  Int_t trackID = track->GetID();
  
  if(track->IsGlobalConstrained()){ // constrained tracks have a changed ID
    trackID = -1-trackID;
  }

  for(Int_t i = 0; i < fDaughterVec.size(); i++) {
    if(trackID == fDaughterVec[i]) 
    {
      return kTRUE;
    }
  }
  return kFALSE;
}