AliVTrackSelection.cxx

Go to the documentation of this file.

/**************************************************************************
 * Copyright(c) 1998-2015, 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 <TObjArray.h>
#include <TClonesArray.h>
#include <AliVTrack.h>
#include <AliVEvent.h>
#include <AliVTrackSelection.h>
#include "AliVCuts.h"

ClassImp(AliVTrackSelection)


AliVTrackSelection::AliVTrackSelection() :
  TObject(),
  fListOfTracks(NULL),
  fListOfTrackBitmaps(NULL),
  fTrackBitmap(64),
  fListOfCuts(NULL),
  fSelectionModeAny(kFALSE)
{
}

AliVTrackSelection::AliVTrackSelection(const AliVTrackSelection& ref):
  TObject(ref),
  fListOfTracks(NULL),
  fListOfTrackBitmaps(NULL),
  fTrackBitmap(64),
  fListOfCuts(NULL),
  fSelectionModeAny(kFALSE)
{
  if(ref.fListOfTracks) fListOfTracks = new TObjArray(*(ref.fListOfTracks));
  if(ref.fListOfTrackBitmaps) fListOfTrackBitmaps = new TClonesArray(*(ref.fListOfTrackBitmaps));
  if(ref.fListOfCuts){
    fListOfCuts = new TObjArray;
    fListOfCuts->SetOwner(false);
    for(TIter cutIter = TIter(ref.fListOfCuts).Begin(); cutIter != TIter::End(); ++cutIter)
      fListOfCuts->Add(*cutIter);
  }
}

AliVTrackSelection& AliVTrackSelection::operator=(const AliVTrackSelection& ref) {
  TObject::operator=(ref);
  if(this != &ref){
    this->~AliVTrackSelection();
    if(ref.fListOfTracks) fListOfTracks = new TObjArray(*(ref.fListOfTracks));
    if(ref.fListOfTrackBitmaps) fListOfTrackBitmaps = new TClonesArray(*(ref.fListOfTrackBitmaps));
    if(ref.fListOfCuts){
      fListOfCuts = new TObjArray;
      fListOfCuts->SetOwner(false);
      for(TIter cutIter = TIter(ref.fListOfCuts).Begin(); cutIter != TIter::End(); ++cutIter)
        fListOfCuts->Add(*cutIter);
    } else fListOfCuts = NULL;
  }
  return *this;
}

AliVTrackSelection::~AliVTrackSelection() {
  if(fListOfTracks) delete fListOfTracks;
  if(fListOfTrackBitmaps) delete fListOfTrackBitmaps;
  if(fListOfCuts) delete fListOfCuts;
}

void AliVTrackSelection::AddTrackCuts(AliVCuts *cuts){
  if(!fListOfCuts){
    fListOfCuts = new TObjArray;
    fListOfCuts->SetOwner(true);
  }
  fListOfCuts->Add(cuts);
}

void AliVTrackSelection::AddTrackCuts(TObjArray *cuts){
  TIter next(cuts);
  AliVCuts* item = 0;
  while ((item = static_cast<AliVCuts*>(next())))
  {
    AddTrackCuts(item);
  }
}

Int_t AliVTrackSelection::GetNumberOfCutObjects() const {
  if(!fListOfCuts) return 0;
  return fListOfCuts->GetEntries();
}

AliVCuts* AliVTrackSelection::GetTrackCuts(Int_t icut) {
  if(!fListOfCuts) return NULL;
  if(icut < fListOfCuts->GetEntries())
    return static_cast<AliVCuts *>(fListOfCuts->At(icut));
  return NULL;
}

TObjArray* AliVTrackSelection::GetAcceptedTracks(const TClonesArray* const tracks)
{
  if (!fListOfTracks) {
    fListOfTracks = new TObjArray;
  }
  else {
    fListOfTracks->Clear();
  }

  if (!fListOfTrackBitmaps) {
    fListOfTrackBitmaps = new TClonesArray("TBits", 1000);
    fListOfTrackBitmaps->SetOwner(kTRUE);
  }
  else {
    fListOfTrackBitmaps->Clear();
  }

  TIter next(tracks);
  AliVTrack* track = 0;
  Int_t i = 0;
  while((track = static_cast<AliVTrack*>(next()))) {
    if (IsTrackAccepted(track)) {
      fListOfTracks->AddLast(track);
    }
    else {
      fListOfTracks->AddLast(0);
    }
    new ((*fListOfTrackBitmaps)[i]) TBits(fTrackBitmap);
    i++;
  }
  return fListOfTracks;
}

TObjArray* AliVTrackSelection::GetAcceptedTracks(const AliVEvent* const event)
{
  if (!fListOfTracks) {
    fListOfTracks = new TObjArray;
  }
  else {
    fListOfTracks->Clear();
  }

  if (!fListOfTrackBitmaps) {
    fListOfTrackBitmaps = new TClonesArray("TBits", 1000);
    fListOfTrackBitmaps->SetOwner(kTRUE);
  }
  else {
    fListOfTrackBitmaps->Clear();
  }

  for(int itrk = 0; itrk < event->GetNumberOfTracks(); itrk++){
    AliVTrack *trk = static_cast<AliVTrack*>(event->GetTrack(itrk));
    if (IsTrackAccepted(trk)) {
      fListOfTracks->AddLast(trk);
    }
    else {
      fListOfTracks->AddLast(trk);
    }
    new ((*fListOfTrackBitmaps)[itrk]) TBits(fTrackBitmap);
  }
  return fListOfTracks;
}