ARA ROOT v3.4 Software

utilities/Atri/makeAtriEventTree.cxx

#include <cstdio>
#include <fstream>
#include <iostream>
#include <zlib.h>
#include <libgen.h>     
#include <cstdlib>
 
using namespace std;

#include "TTree.h"
#include "TFile.h"
#include "TSystem.h"

#define HACK_FOR_ROOT

#include "araAtriStructures.h"
#include "RawAtriStationEvent.h"  

void process();
void makeTree(char *inputName, char *outDir);

AraStationEventHeader_t theEventHeader;
char *dataBuffer;
TFile *theFile;
TTree *eventTree;
RawAtriStationEvent *theEvent=0;
char outName[FILENAME_MAX];
UInt_t realTime;
Int_t runNumber;
Int_t lastRunNumber;


int main(int argc, char **argv) {
  dataBuffer = new char[200000];
  theEvent=0;
  if(argc<3) {
    std::cout << "Usage: " << basename(argv[0]) << " <file list> <out dir>" << std::endl;
    return -1;
  }
  if(argc==4) 
    runNumber=atoi(argv[3]);
  makeTree(argv[1],argv[2]);
  delete [] dataBuffer;
  return 0;
}
  

void makeTree(char *inputName, char *outFile) {
  cout << inputName << "\t" << outFile << endl;
  strncpy(outName,outFile,FILENAME_MAX);
  theEvent = new RawAtriStationEvent();
  //    cout << sizeof(AraStationEventHeader_t) << endl;
  ifstream SillyFile(inputName);

  int numBytes=0;
  char fileName[180];
  int error=0;
  //    int eventNumber=1;
  int counter=0;
  while(SillyFile >> fileName) {
    if(counter%100==0) 
      cout << fileName << endl;
    counter++;
    
    //    const char *subDir = gSystem->DirName(fileName);
    //    const char *subSubDir = gSystem->DirName(subDir);
    //    const char *eventDir = gSystem->DirName(subSubDir);
    //    const char *runDir = gSystem->DirName(eventDir);
    //    const char *justRun = gSystem->BaseName(runDir);
    //  cout << justRun << endl;
    //    sscanf(justRun,"run%d",&runNumber);
    
    gzFile infile = gzopen (fileName, "rb");    
    //    std::cout << "gzeof: " << gzeof(infile) << "\n";
    for(int i=0;i<1000;i++) {   
      //      cout << i << endl;
      numBytes=gzread(infile,&theEventHeader,sizeof(AraStationEventHeader_t));
      //      std::cout << numBytes << "\n";
      if(numBytes==0) break;
      if(numBytes!=sizeof(AraStationEventHeader_t)) {
        if(numBytes)
          cerr << "Read problem: " <<numBytes << " of " << sizeof(AraStationEventHeader_t) << endl;
        error=1;
        break;
      }
      if(theEventHeader.gHdr.numBytes>0) {
        //      std::cout << "Num bytes: " << theEventHeader.gHdr.numBytes << "\t" << theEventHeader.numBytes << "\n";
        //      std::cout << "Event number: " << theEventHeader.eventNumber << "\t" << theEventHeader.unixTime << "\t" << theEventHeader.unixTimeUs << "\n";
        
        
        Int_t numDataBytes=theEventHeader.gHdr.numBytes-sizeof(AraStationEventHeader_t);
        numBytes=gzread(infile,dataBuffer,numDataBytes);
        //      std::cout << numBytes << "\n";
        if(numBytes==0) break;
        if(numBytes!=numDataBytes) {
          if(numBytes)
            cerr << "Read problem: " <<numBytes << " of " <<  numDataBytes << endl;
          error=1;
          break;
        }
        process();
        //      exit(0);
      }
      else {
        std::cerr << "How can gHdr.numBytes = " << theEventHeader.gHdr.numBytes << "\n";
        error=1;
        break;
      }

      //      cout << ": " << theEventHeader.unixTime << endl;
     }
    gzclose(infile);
    //  if(error) break;
  }
  if(eventTree)
    eventTree->AutoSave();
  //    theFile->Close();
}


void process() {
  //  cout << "process:\t" << theEventHeader.eventNumber << endl;
  static int doneInit=0;
  if(!doneInit) {
    //    char dirName[FILENAME_MAX];
    //    char fileName[FILENAME_MAX];
    //    sprintf(dirName,"%s/run%d",outDirName,runNumber);
    //    gSystem->mkdir(dirName,kTRUE);
    //    sprintf(fileName,"%s/eventFile%d.root",dirName,runNumber);
    cout << "Creating File: " << outName << endl;
    theFile = new TFile(outName,"RECREATE");
    eventTree = new TTree("eventTree","Tree of ARA Event's");
    eventTree->Branch("run",&runNumber,"run/I");
    eventTree->Branch("event","RawAtriStationEvent",&theEvent);
    
    doneInit=1;
  }  
  //  cout << "Here: "  << theEvent.eventNumber << endl;
  if(theEvent) delete theEvent;
  
  //  theEvent = new RawAtriStationEvent(&theEventHeader,dataBuffer);//BRENDAN -- this line makes an ATRI station event using the ARAAcqd produced stationId (encoded in the raw events in the dat file). The next line with additional argument (in this case 2) forces the station Id in the produced L0 events to be station (2)
  theEvent = new RawAtriStationEvent(&theEventHeader,dataBuffer, 2);//FIXME -- this forces the stationId to 2
  eventTree->Fill();  
  lastRunNumber=runNumber;
  //  delete theEvent;
}

---

UCL HEP Home   

UHE Neutrino Home   

ARA ROOT Software

Main  

Download  

Branches

3.13  

3.12  

3.11  

3.10  

3.9  

3.8  

3.7  

3.6  

3.5  

3.4  

3.3  

3.2  

TestBed