KVGANILDataReader.cpp

//Created by KVClassFactory on Wed Sep 23 16:07:38 2009
//Author: Chbihi
 
#include "KVGANILDataReader.h"
#include "GTGanilData.h"
#include "GTOneScaler.h"
#include "TSystem.h"
#include "TUrl.h"
#include "TPluginManager.h"
#include "RVersion.h"
#include <TInterpreter.h>
#include "TEnv.h"
 
ClassImp(KVGANILDataReader)
 
 
 
 
KVGANILDataReader::KVGANILDataReader(const Char_t* file, Option_t* dataset)
{
   //Open and initialise a GANIL data file for reading.
   //By default, Scaler buffers are ignored.
   //If file cannot be opened, this object will be made a zombie. Do not use.
   //To test if file is open, use IsOpen().
   //The basename of the file (excluding any path) can be obtained using GetName()
   //The full pathname of the file can be obtained using GetTitle()
 
   init();
   OpenFile(file, dataset);
}
 
 
 
 
KVGANILDataReader::~KVGANILDataReader()
{
   // Destructor
   if (ParVal) delete [] ParVal;
   if (ParNum) delete [] ParNum;
   if (fGanilData) delete fGanilData;
}
 
 
 
 
void KVGANILDataReader::init()
{
   //default initialisations
 
   fParameters.SetOwner();
   fUserTree = nullptr;
   ParVal = nullptr;
   ParNum = nullptr;
   make_arrays = make_leaves = kFALSE;
   fGanilData = nullptr;
}
 
 
 
 
void KVGANILDataReader::SetUserTree(TTree* T, Option_t* opt)
{
   // To fill a TTree with the data in the current file, create a TTree:
   //    TFile* file = new TFile("run1.root","recreate");
   //    TTree* T = new TTree("Run1", "Raw data for Run1");
   // and then call this method: SetUserTree(T)
   // If you read all events of the file, the TTree will be automatically filled
   // with data :
   //    while( runfile->GetNextEvent() ) ;
   //
   // Two different TTree structures are available, depending on the option string:
   //
   //    opt = "arrays": [default]
   //
   // The TTree will have the following structure:
   //
   //    *Br    0 :NbParFired : NbParFired/I    = number of fired parameters in event
   //    *............................................................................*
   //    *Br    1 :ParNum    : ParNum[NbParFired]/i   = array of indices of fired parameters
   //    *............................................................................*
   //    *Br    2 :ParVal    : ParVal[NbParFired]/s   = array of values of fired parameters
   //
   // This structure is the fastest to fill and produces the smallest file sizes.
   // In order to be able to directly access the parameters as if option "leaves" were used
   // (i.e. one branch/leaf for each parameter), we add two aliases for each parameter to
   // the tree:
   //        PARNAME           = value of parameter if present in event
   //        PARNAME_M      = number of times parameter appears in event
   // Assuming that each parameter only appears at most once in each event, i.e. PARNAME_M=0 or 1,
   // then
   //    root[0] T->Draw("PARNAME", "PARNAME_M")
   // will histogram the value of PARNAME for each event in which it is present.
   // (if the selection condition "PARNAME_M" is not used, the histogram will also be filled with a 0
   // for each event in which PARNAME does not appear).
   //          N.B. the PARNAME alias is in fact the sum of the values of PARNAME in each event.
   //          If PARNAME_M>1 in some events, it is not the individual values but their sum which will
   //          be histogrammed in this case.
   //
   // Thus, if the data file has parameters called "PAR_1" and "PAR_2",
   // the following command will work
   //
   //    root[0]  T->Draw("PAR_1:PAR_2", "PAR_1_M&&PAR_2_M", "col")
   //
   // even though no branches "PAR_1" or "PAR_2" exist.
   //
   //
   //
   //    opt = "leaves":
   //
   // The TTree will have a branch/leaf for each parameter. This option is slower and produces
   // larger files.
   //
   // If the option string contains both "arrays" and "leaves", then both structures will be used
   // (in this case there is a high redundancy, as each parameter is stored twice).
   //
   // The full list of parameters is stored in a TObjArray in the list returned by TTree::GetUserInfo().
   // Each parameter is represented by a TNamed object.
   // In order to retrieve the name of the parameter with index 674 (e.g. taken from branch ParNum),
   // do:
   //     TObjArray* parlist = (TObjArray*) T->GetUserInfo()->FindObject("ParameterList");
   //     cout << "Par 674 name = " << (*parlist)[674]->GetName() << endl;
   //
   //
   //  Automatic creation & filling of Scalers TTree
   //
   // give an option string containing "scalers", i.e. "leaves,scalers", or "ARRAYS+SCALERS", etc.
   // a TTree with name 'Scalers' will be created, all scaler buffers will be written in it.
 
 
   TString option = opt;
   option.ToUpper();
   make_arrays = option.Contains("ARRAYS");
   make_leaves = option.Contains("LEAVES");
   Bool_t make_scalers = option.Contains("SCALERS");
   if (make_scalers) {
      fGanilData->SetScalerBuffersManagement(GTGanilData::kAutoWriteScaler);
   }
 
   fUserTree = T;
   if (make_arrays) {
      Int_t maxParFired = GetRawDataParameters().GetEntries();
      ParVal = new UShort_t[maxParFired];
      ParNum = new UInt_t[maxParFired];
      fUserTree->Branch("NbParFired", &NbParFired, "NbParFired/I");
      fUserTree->Branch("ParNum", ParNum, "ParNum[NbParFired]/i");
      fUserTree->Branch("ParVal", ParVal, "ParVal[NbParFired]/s");
   }
   if (make_leaves) {
      TIter next_rawpar(&GetRawDataParameters());
      KVEBYEDAT_ACQParam* acqpar;
      while ((acqpar = (KVEBYEDAT_ACQParam*)next_rawpar())) {
         TString leaf;
         leaf.Form("%s/S", acqpar->GetName());
         // for parameters with <=8 bits only use 1 byte for storage
         if (acqpar->GetNbBits() <= 8) leaf += "1";
         fUserTree->Branch(acqpar->GetName(), *(acqpar->ConnectData()), leaf.Data());
      }
   }
 
#if ROOT_VERSION_CODE > ROOT_VERSION(5,25,4)
#if ROOT_VERSION_CODE < ROOT_VERSION(5,26,1)
   // The TTree::OptimizeBaskets mechanism is disabled, as for ROOT versions < 5.26/00b
   // this lead to a memory leak
   fUserTree->SetAutoFlush(0);
#endif
#endif
 
   // add list of parameter names in fUserTree->GetUserInfos()
   // and if option="arrays" add aliases for each parameter & its multiplicity
 
   // TObjArray has to be as big as the largest parameter number in the list
   // of raw data parameters. So first loop over parameters to find max param number.
   UInt_t maxpar = 0;
   TIter next(&GetRawDataParameters());
   KVEBYEDAT_ACQParam* par;
   while ((par = (KVEBYEDAT_ACQParam*)next())) if (par->GetNumber() > maxpar) maxpar = par->GetNumber();
 
   TObjArray* parlist = new TObjArray(maxpar, 1);
   parlist->SetName("ParameterList");
 
   next.Reset();
   while ((par = (KVEBYEDAT_ACQParam*)next())) {
      parlist->AddAt(new TNamed(par->GetName(), Form("index=%d", par->GetNumber())), par->GetNumber());
      if (make_arrays) {
         fUserTree->SetAlias(par->GetName(), Form("Sum$((ParNum==%d)*ParVal)", par->GetNumber()));
         fUserTree->SetAlias(Form("%s_M", par->GetName()), Form("Sum$(ParNum==%d)", par->GetNumber()));
      }
   }
   fUserTree->GetUserInfo()->Add(parlist);
}
 
 
 
 
void KVGANILDataReader::OpenFile(const Char_t* file, Option_t* dataset)
{
   //Open and initialise a GANIL data file for reading.
   //By default, scaler buffers are ignored.
   //This can be changed by changing the value of
   //
   // KVGANILDataReader.ScalerBuffersManagement:       kSkipScaler
   //
   //If file cannot be opened, this object will be made a zombie. Do not use.
   //To test if file is open, use IsOpen().
   //The basename of the file (excluding any path) can be obtained using GetName()
   //The full pathname of the file can be obtained using GetTitle()
 
   if (fGanilData) delete fGanilData;
   fGanilData = NewGanTapeInterface(dataset);
 
   fGanilData->SetFileName(file);
   SetName(gSystem->BaseName(file));
   SetTitle(file);
 
   // handling scaler buffers
   TString what_scale = gEnv->GetValue("KVGANILDataReader.ScalerBuffersManagement", "kSkipScaler");
   what_scale.Prepend("GTGanilData::");
   Long_t ws = gInterpreter->ProcessLine(what_scale);
   fGanilData->SetScalerBuffersManagement((GTGanilData::ScalerWhat_t)ws);
 
   fGanilData->Open();
 
   //test whether file has been opened
   if (!fGanilData->IsOpen()) {
      //if initial attempt fails, we try to open the file as a 'raw' TFile
      //This may work when we are attempting to open a remote file i.e.
      //via rfio, and a subsequent attempt to open the file using the GanTape
      //routines may then succeed.
      TUrl rawtfile(file, kTRUE);
      rawtfile.SetOptions("filetype=raw");
      TFile* rawfile = TFile::Open(rawtfile.GetUrl());
      if (!rawfile) {
         Error("OpenFile", "File cannot be opened: %s",
               file);
         MakeZombie();
         return;
      }
      //TFile::Open managed to open file! Try again...
      delete rawfile;
      fGanilData->Open();
      if (!fGanilData->IsOpen()) {
         //failed again ??!
         Error("OpenFile", "File cannot be opened: %s",
               file);
         MakeZombie();
         return;
      }
   }
}
 
 
 
 
 
void KVGANILDataReader::ConnectRawDataParameters()
{
   //Generate all required KVEBYEDAT_ACQParam objects corresponding to parameters in file
   //
   //fParameters is filled with a KVEBYEDAT_ACQParam for every acquisition parameter in the file.
   //
   //To access the full list of data parameters in the file after this method has been
   //called (i.e. after the file is opened), use GetRawDataParameters().
 
   TIter next(fGanilData->GetListOfDataParameters());
   KVEBYEDAT_ACQParam* par;
   GTDataPar* daq_par;
   while ((daq_par = (GTDataPar*) next())) {//loop over all parameters
      par = new KVEBYEDAT_ACQParam(daq_par->GetName());
      par->SetNumber(daq_par->Index());
      par->SetNbBits(daq_par->Bits());
      fParameters.Add(par);
      fGanilData->Connect(daq_par->Index(), par->ConnectData());
      fGanilData->ConnectFired(daq_par->Index(), par->ConnectFired());
   }
}
 
 
 
 
 
Bool_t KVGANILDataReader::GetNextEvent()
{
   // Read next event in raw data file.
   // Returns false if no event found (end of file).
   // The list of all fired acquisition parameters is filled, and can be retrieved with
   // GetFiredDataParameters().
   // If SetUserTree(TTree*) has been called, the TTree is filled with the values of all
   // parameters in this event.
 
   Bool_t ok = fGanilData->Next();
   FillFiredParameterList();
   if (fUserTree) {
      if (make_arrays) {
         NbParFired = fFired.GetEntries();
         TIter next(&fFired);
         KVEBYEDAT_ACQParam* par;
         int i = 0;
         while ((par = (KVEBYEDAT_ACQParam*)next())) {
            ParVal[i] = par->GetData();
            ParNum[i] = par->GetNumber();
            i++;
         }
      }
      fUserTree->Fill();
   }
   return ok;
}
 
 
 
 
 
GTGanilData* KVGANILDataReader::NewGanTapeInterface(Option_t* dataset)
{
   // Creates and returns new instance of class derived from GTGanilData used to read GANIL acquisition data
   // Actual class is determined by Plugin.GTGanilData in environment files and name of dataset.
 
   //check and load plugin library
   TPluginHandler* ph = LoadPlugin("GTGanilData", dataset);
   if (!ph) {
      // default: GTGanilData
      return (new GTGanilData);
   }
 
   //execute constructor/macro for plugin
   return ((GTGanilData*) ph->ExecPlugin(0));
}
 
 
 
 
 
GTGanilData* KVGANILDataReader::GetGanTapeInterface()
{
   return fGanilData;
}
 
 
 
 
 
KVGANILDataReader* KVGANILDataReader::Open(const Char_t* filename, Option_t* dataset)
{
   //Static method, used by KVDataSet::Open
   //The name of the dataset is passed in the option string
   return new KVGANILDataReader(filename, dataset);
}
 
 
 
 
 
void KVGANILDataReader::FillFiredParameterList()
{
   // clears and then fills list fFired with all fired acquisition parameters in event
   fFired.Clear();
   TIter next(&fParameters);
   KVEBYEDAT_ACQParam* par;
   while ((par = (KVEBYEDAT_ACQParam*)next())) if (par->Fired()) fFired.Add(par);
}
 
 
 
 
Bool_t KVGANILDataReader::HasScalerBuffer() const
{
   return fGanilData->IsScalerBuffer();
}
 
 
 
 
Int_t KVGANILDataReader::GetNumberOfScalers() const
{
   return fGanilData->GetScalers()->GetNbChannel();
}
 
 
 
 
 
UInt_t KVGANILDataReader::GetScalerCount(Int_t index) const
{
   return fGanilData->GetScalers()->GetScalerPtr(index)->GetCount();
}
 
 
 
 
 
Int_t KVGANILDataReader::GetScalerStatus(Int_t index) const
{
   return fGanilData->GetScalers()->GetScalerPtr(index)->GetStatus();
}
 
 
 
 
 
Int_t KVGANILDataReader::GetEventCount() const
{
   return fGanilData->GetEventCount();
}
 
 
 
 
Int_t KVGANILDataReader::GetRunNumberReadFromFile() const
{
   // Return run number of file being read.
   // Only call when file has been successfully opened.
   return fGanilData->GetRunNumber();
}