Delete only existent objects.

[u/mrichter/AliRoot.git] / TRD / AliTRDdigitsManager.h

#ifndef TRDdigitsManager_H
#define TRDdigitsManager_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id: AliTRDdigitsManager.h,v */

/////////////////////////////////////////////////////////////
//  Manages the TRD digits                                 //
/////////////////////////////////////////////////////////////

#include "TObject.h"

#include "AliTRDsegmentArray.h"
#include "AliTRDdataArrayI.h"
#include "AliTRDdigit.h"

const Int_t  kNDict = 3;

class AliTRDdigitsManager : public TObject {

 public:

  AliTRDdigitsManager();
  ~AliTRDdigitsManager();

  virtual Bool_t              MakeBranch();
  virtual Bool_t              ReadDigits();
  virtual Bool_t              WriteDigits();

  virtual void                SetRaw();

  virtual Bool_t              IsRaw()                { return fIsRaw;         };
  virtual AliTRDsegmentArray *GetDigits()            { return fDigits;        };
  virtual AliTRDsegmentArray *GetDictionary(Int_t i) { return fDictionary[i]; };

  inline  AliTRDdataArrayI   *GetDigits(Int_t det);
  inline  AliTRDdataArrayI   *GetDictionary(Int_t det, Int_t i);
  inline  AliTRDdigit        *GetDigit(Int_t row, Int_t col, Int_t time, Int_t det);
  inline  Int_t               GetTrack(Int_t track, Int_t row, Int_t col, Int_t time, Int_t det);
  inline  Int_t               GetTrack(Int_t track, AliTRDdigit *Digit);

 protected:

  AliTRDsegmentArray *fDigits;             //! Digits data Array
  AliTRDsegmentArray *fDictionary[kNDict]; //! Track dictionary data array

  Bool_t              fIsRaw;              //  Flag indicating raw digits

  ClassDef(AliTRDdigitsManager,1)          //  Manages the TRD digits

};

//_____________________________________________________________________________
inline AliTRDdataArrayI *AliTRDdigitsManager::GetDigits(Int_t det) 
{
  //
  // Returns the digits array for one detector
  //

  return (AliTRDdataArrayI *) fDigits->At(det);

}


//_____________________________________________________________________________
inline AliTRDdataArrayI *AliTRDdigitsManager::GetDictionary(Int_t det, Int_t i) 
{
  //
  // Returns the dictionary for one detector
  //

  return (AliTRDdataArrayI *) fDictionary[i]->At(det);

}

//_____________________________________________________________________________
inline AliTRDdigit *AliTRDdigitsManager::GetDigit(Int_t row, Int_t col
                                                , Int_t time, Int_t det)
{
  // 
  // Creates a single digit object 
  //

  Int_t digits[5];

  digits[0] = det;
  digits[1] = row;
  digits[2] = col;
  digits[3] = time;
  digits[4] = GetDigits(det)->GetData(row,col,time);
  
  return (new AliTRDdigit(fIsRaw,digits));

}

//_____________________________________________________________________________
inline Int_t AliTRDdigitsManager::GetTrack(Int_t track
                                         , Int_t row, Int_t col, Int_t time
                                         , Int_t det)
{
  // 
  // Returns the MC-track numbers from the dictionary.
  //

  if ((track < 0) || (track >= kNDict)) {
    TObject::Error("GetTracks"
                  ,"track %d out of bounds (size: %d, this: 0x%08x)"
                  ,track,kNDict,this);
    return -1;
  }

  // Array contains index+1 to allow data compression
  return (GetDictionary(det,track)->GetData(row,col,time) - 1);

}

//_____________________________________________________________________________
inline Int_t AliTRDdigitsManager::GetTrack(Int_t track, AliTRDdigit *Digit)
{
  // 
  // Returns the MC-track numbers from the dictionary for a given digit
  //

  Int_t row  = Digit->GetRow();
  Int_t col  = Digit->GetCol();
  Int_t time = Digit->GetTime();
  Int_t det  = Digit->GetDetector();

  return GetTrack(track,row,col,time,det);

}

#endif