Adding track references at decay points (M.Ivanov)

[u/mrichter/AliRoot.git] / RICH / AliRICHTransientDigit.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, 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.                  *
 **************************************************************************/

/* $Id$ */

#include "AliRICHTransientDigit.h"
#include <TObjArray.h>
#include "TVector.h"

ClassImp(AliRICHTransientDigit)
    
//____________________________________________________________________________
AliRICHTransientDigit::AliRICHTransientDigit(Int_t ich, Int_t *digits): 
    AliRICHDigit(digits)
{
    //
    // Creates a RICH digit list object
    //
    
    fChamber = ich;
    fTrackList   = new TObjArray;
    
}
//_____________________________________________________________________________

AliRICHTransientDigit::~AliRICHTransientDigit()
{
  fTrackList->Delete();
  delete fTrackList;
}

////////////////////////////////////////////////////////////////////////
void AliRICHTransientDigit::AddToTrackList(Int_t track, Int_t charge)
{
  TVector *pTrInfo = new TVector(2);
  TVector &trInfo = *pTrInfo;
  trInfo(0) = track;
  trInfo(1) = charge;
  fTrackList->Add(pTrInfo);
}

////////////////////////////////////////////////////////////////////////
void AliRICHTransientDigit::UpdateTrackList(Int_t track, Int_t charge)
{
  Int_t lastEntry = fTrackList->GetLast();
  TVector *pVect = static_cast<TVector*>(fTrackList->At(lastEntry));
  if ( static_cast<Int_t>((*pVect)(0)) == track) {
    (*pVect)(1) += charge;  // update charge
  } else {
    AddToTrackList(track,charge);
  }
}

////////////////////////////////////////////////////////////////////////
Int_t AliRICHTransientDigit::GetTrack(Int_t i) const
{
  if (i > fTrackList->GetEntriesFast()) return 0;
  TVector *pVect = static_cast<TVector*>(fTrackList->At(i));
  return static_cast<Int_t>((*pVect)(0));
}


////////////////////////////////////////////////////////////////////////
Int_t AliRICHTransientDigit::GetCharge(Int_t i) const
{
  if (i > fTrackList->GetEntriesFast()) return 0;
  TVector *pVect = static_cast<TVector*>(fTrackList->At(i));
  return static_cast<Int_t>((*pVect)(1));
}