HarpoDccReader.cxx

Go to the documentation of this file.

// 
// File HarpoDccReader.cxx
//
#include "HarpoDccReader.h"
#include "HarpoDet.h"
#include <TSystem.h>

#include <climits>
#include <cstdio>
#include <arpa/inet.h>

#include <iostream>

ClassImp(HarpoDccReader)

#define EXPECTED_NB_DCC_FRAGMENTS 18
#define NB_FEM_FRAGMENTS  72
#define MAX_FEM_DATA_BUFFER 2097152


HarpoDccReader::HarpoDccReader() : HarpoDetReader()
{

  if(gHarpoDebug>0) Info("HarpoDccReader","Create DCC Reader");

  //fRunNo = 0;
  fDone = false;
  fError = false;
  fPlane = -1;

  //  fSaveTree = kFALSE;

  //fInFile = (TString *) NULL;
  fIn = (FILE *) NULL;
  //fNevts = 0;

  //fNewFile = (TFile *) NULL;
  fNewNb = 0;

  //fEvent = (HarpoDccEvent *) NULL;
  fEvent = new HarpoDccEvent();

  fDccAnalyse = (HarpoDccAnalyse *) NULL;
  //fMaxEvents = LONG_MAX;
  //fPrevTimeStamp = 0;
  isEventSaved = false;

}


HarpoDccReader::HarpoDccReader(HarpoConfig *cfg) : HarpoDetReader(cfg)
{
  // Config Function
  fDone = false;
  fError = false;

  //fSaveTree = kFALSE;

  //fRunNo = cfg->GetRunNo();
  //fInFile = cfg->GetInFile();
  fIn = (FILE *) NULL;
  //fNevts = 0;
  //fMaxEvents = cfg->GetMaxEvents();

  TString *FileName = cfg->GetOutFile();
  fNewNb = 0;

  fEvent = new HarpoDccEvent();
  //fEvent = (HarpoDccEvent *) NULL;

  fDccAnalyse = (HarpoDccAnalyse *) NULL;
  
  if ( FileName != (TString *) NULL )
    SetRootFile(FileName);

  //fPrevTimeStamp = 0;
  isEventSaved = false;
}

HarpoDccReader::~HarpoDccReader()
{
  std::fclose(fIn);
}

void HarpoDccReader::SetDccFile(TString *name)
{
  fInFile = name;
}

void HarpoDccReader::SetRootFile(TString *name)
{
  fNewFileName = name; 
  //  fNewFile = new TFile(name->Data(),"RECREATE","HARPO DCC Events");
  //fSaveTree = kTRUE;
}

void HarpoDccReader::Save() // TODO
{
  // if(!fSaveTree) return;   
  // fTree->Write();
  // fNewFile->Close();
}

void HarpoDccReader::SetAnalyseFunction(HarpoDccAnalyse *func)
{
  fDccAnalyse = func;
}
  
Bool_t HarpoDccReader::NextEvent()
{

  if(gHarpoDebug>1) Info("NextEvent"," ");
  // EventHeader_t lHeader; // local buffer 
  //  UShort_t lData[EVENT_BUFFER_SIZE];       // local buffer 
  if ( isEventSaved ) {
    isEventSaved = false;
    fEvent = fSaveEvent;
    return true;
  }

  if (fDone) return false; // EOF after last Event Data read

  fPrevTimeStamp = fTimeStamp;
  
  if (std::fread(&lHeader, sizeof(EventHeader_t), 1, fIn) != 1)
    {
    if(std::feof(fIn))
      {
        fDone = true;
      }
    else
      {
        fError = true;
      }
    return false;
  }
  else
    {
      //      if(fEvent==NULL)
      fEvent = new HarpoDccEvent();
        //else
        //fEvent->ResetData();
      //
      fEvent->SetDet(fDet);
      fEvent->SetHeader(ntohl(lHeader.eventSize),ntohl(lHeader.eventNumb));
      fEvent->SetRunHeader(fRunHeader);
      // Event Data
      size_t dataLen = (fEvent->GetHeader())->eventSize - sizeof(EventHeader_t);
      if (std::fread(&lData,dataLen, 1, fIn) != 1)
        {
          if(std::feof(fIn)) // Read after EOF
            {
              fDone = true;
            }
          else
            {
              fError = true;
            }
          return false;
        }
      else
        {
          //      std::cout << "======= Before SetData " << dataLen << std::endl;
          fEvent->SetData(lData,dataLen);
          fTimeStamp = fEvent->GetTimeStamp();
          fNevts++;
        }
    }
  
  return true;
}

void HarpoDccReader::AddPedestals()
{
  HarpoDccHeader *dh = static_cast<HarpoDccHeader *>(fRunHeader); 
  fEvent->SetPedestals(dh->GetPedestals());
}

HarpoDccEvent *HarpoDccReader::GetEvent()
{
  if(gHarpoDebug>1) Info("GetEvent "," "); 
  fEvent->UnPackData();
  AddPedestals();
  return fEvent;
}

HarpoDccHeader *HarpoDccReader::GetDetHeader()
{
  return static_cast<HarpoDccHeader *>(fRunHeader);
}


Bool_t HarpoDccReader::Init()
{

  // if(fSaveTree){
  //   fTree = new TTree("HarpoDccData","HARPO events");
  //   fTree->Bronch("Events","HarpoDccEvent",&fEvent);
  // }

  if(fPlane==-1) 
    fPlane = fDet;

  const char * fn = fInFile->Data();
  if(gHarpoDebug>0) Info("Init","%s",fn);
  if ( fInFile->BeginsWith("/" )) { // full path 
    if ( gSystem->AccessPathName(fn,kReadPermission) ) {
      perror("Dcc File access");
      return false;
    }
  } else {
    if (  gSystem->AccessPathName(fn,kReadPermission) ) // file not exist
      {
        // TODO
        //  TString *DataDir = gHDccCfg->GetDataDir();
        // Not a best way but i can't use config hire
        TString DataDir(gSystem->Getenv("HARPO_DATA_DIR"));

        if ( ! DataDir.EndsWith("/") ) DataDir.Append("/");
        fInFile->Prepend(DataDir);
        fn = fInFile->Data();
        if ( gSystem->AccessPathName(fn,kReadPermission) ) {
          perror("Dcc File access");
          return false;
        }
      }
  }

  if(gHarpoDebug>0) Info("Init","Guessed %s",fn);
    
  fIn = std::fopen(fn,"r");
  if ( fIn != NULL )
    {
      fRunHeader = new HarpoDccHeader(fIn);
      fRunHeader->SetReaderType(hDccT2K);
      return static_cast<HarpoDccHeader *>(fRunHeader)->check(this);
    }
  else
    { 
      perror("Dcc File open");
      return false;
    }
}
  

Long_t      HarpoDccReader::Loop(Long_t maxevents)
{
  Long_t cnt=0;
  HarpoDccEvent *evt;

  fDccAnalyse->Init();

  while (NextEvent()) {
    evt = (HarpoDccEvent*)GetEvent();
    fDccAnalyse->ProcessEvent(evt);

    //    if(fSaveTree) fTree->Fill();

    evt->Delete();
    //    delete evt;
    if (++cnt >= maxevents) break;
  }
  return cnt;

  fDccAnalyse->Save();
  
}

Long_t      HarpoDccReader::Loop()
{
  Long_t maxevents = fMaxEvents;
  return Loop(maxevents);
}

// Bool_t  HarpoDccReader::Find(Long_t eventNo)
// {
//   Long_t cnt=0;
//   Long_t maxevents = fMaxEvents;
//   Bool_t found = false;

//   while (NextEvent()) {
    
//     if ((fEvent->GetHeader())->eventNumb == eventNo) {
//       found = true;
//       break;
//     }
//     SkipEvent();
//     if (++cnt >= maxevents) break;
//   }
//   return found;
// }
void  HarpoDccReader::SkipEvent()
{
  //isEventSaved = false;
  HarpoDccEvent *evt = GetEvent(); // We need read event data
  if ( evt != NULL ) delete evt ; // We will not process this event,
                                  // return NULL  on read arror
}

void  HarpoDccReader::PushBack()
{
  HarpoDet dt(fDet);
  fSaveEvent = fEvent;
  fEvent = NULL; 
  isEventSaved = true;
  // std::cout << "PushBack " << dt.GetName() << " Evt " << fNevts << std::endl;
  Info("PushBack","%s (%ld), evt %ld",dt.GetDescription(), fDet, fNevts);
  
}