RAW/AliRawReaderChain.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 ///////////////////////////////////////////////////////////////////////////////
  17 ///
  18 /// This is a class for reading raw data from a root chain.
  19 /// There are two constructors available - one from a text file containing the
  20 /// list of root raw-data files to be processed and one directly from
  21 /// TFileCollection.
  22 ///
  23 /// cvetan.cheshkov@cern.ch 29/07/2008
  24 ///
  25 ///////////////////////////////////////////////////////////////////////////////
  26
  27 #include <TChain.h>
  28 #include <TFileCollection.h>
  29
  30 #include "AliRawReaderChain.h"
  31 #include "AliRawEvent.h"
  32
  33 ClassImp(AliRawReaderChain)
  34
  35 AliRawReaderChain::AliRawReaderChain() :
  36   AliRawReaderRoot(),
  37   fChain(NULL)
  38 {
  39   // default constructor
  40 }
  41
  42 AliRawReaderChain::AliRawReaderChain(TString listFileName) :
  43   AliRawReaderRoot(),
  44   fChain(NULL)
  45 {
  46 // create raw-reader objects which takes as an input a root chain
  47 // from the file list found in 'listFileName'
  48
  49   TFileCollection collection("RAW",
  50                              "Collection with raw-data files",
  51                              listFileName.Data());
  52
  53   TChain* fChain = new TChain("RAW");
  54   if (!fChain->AddFileInfoList((TCollection*)(collection.GetList()))) {
  55     Error("AliRawReaderChain","Bad file list in collection, the chain is empty");
  56     return;
  57   }
  58
  59   fChain->SetBranchStatus("*",0);
  60   fChain->SetBranchStatus("rawevent",1);
  61
  62   fEvent = new AliRawEvent;
  63   fChain->SetBranchAddress("rawevent", &fEvent);
  64 }
  65
  66 AliRawReaderChain::AliRawReaderChain(TFileCollection *collection) :
  67   AliRawReaderRoot(),
  68   fChain(NULL)
  69 {
  70 // create raw-reader objects which takes as an input a root chain
  71 // from a root file collection
  72
  73   TChain* fChain = new TChain("RAW");
  74   if (!fChain->AddFileInfoList((TCollection*)(collection->GetList()))) {
  75     Error("AliRawReaderChain","Bad file list in collection, the chain is empty");
  76     return;
  77   }
  78
  79   fChain->SetBranchStatus("*",0);
  80   fChain->SetBranchStatus("rawevent",1);
  81
  82   fEvent = new AliRawEvent;
  83   fChain->SetBranchAddress("rawevent", &fEvent);
  84 }
  85
  86 AliRawReaderChain::AliRawReaderChain(const AliRawReaderChain& rawReader) :
  87   AliRawReaderRoot(rawReader),
  88   fChain(rawReader.fChain)
  89 {
  90 // copy constructor
  91 }
  92
  93 AliRawReaderChain& AliRawReaderChain::operator = (const AliRawReaderChain&
  94                                                   rawReader)
  95 {
  96 // assignment operator
  97
  98   this->~AliRawReaderChain();
  99   new(this) AliRawReaderChain(rawReader);
 100   return *this;
 101 }
 102
 103 AliRawReaderChain::~AliRawReaderChain()
 104 {
 105 // delete objects and close root file
 106
 107   if (fChain) {
 108     delete fChain;
 109     fChain = NULL;
 110   }
 111 }
 112
 113 Bool_t AliRawReaderChain::NextEvent()
 114 {
 115 // go to the next event in the root file
 116
 117   if (!fChain || !fChain->GetListOfFiles()->GetEntriesFast()) return kFALSE;
 118
 119   do {
 120     delete fEvent;
 121     fEvent = new AliRawEvent;
 122     if (fChain->GetEntry(fEventIndex+1) <= 0)
 123       return kFALSE;
 124     fEventIndex++;
 125   } while (!IsEventSelected());
 126   fEventNumber++;
 127   return Reset();
 128 }
 129
 130 Bool_t AliRawReaderChain::RewindEvents()
 131 {
 132 // go back to the beginning of the root file
 133
 134   fEventIndex = -1;
 135   delete fEvent;
 136   fEvent = new AliRawEvent;
 137   fEventNumber = -1;
 138   return Reset();
 139 }