split_fd.cc

Go to the documentation of this file.

#include <iostream>

#include <EyeEvent.hh>
#include <EyeEventHeader.hh>
#include <EyeEventFile.hh>

#include <TTree.h>

#include <fstream>

using namespace std;

void
help(char *argv[])
{
  cout << endl
       << " Specify: List of input files, also with \"*\" etc. as well as number of events in output files" << endl;
  
  std::cout << "Usage: " << argv[0] << " <files> [n-events]" << std::endl << std::endl;
}


int
main (int argc, char *argv[])
{
  if (argc<=2) {
    help(argv);    
    return 1;
  }
  
  const int nSplit = atoi(argv[argc-1]);
  cout << "Your n-split is : " << nSplit << endl;

  
  if (nSplit <= 0 || nSplit>1e7) {
    help(argv);
    return 1;
  }
  
  ifstream filelist(argv[1]);
  
  while (filelist.good()) {
    
    string filename;
    getline(filelist, filename);
    
    if (filename=="" || filename[0] == '#')
      continue;
    
    const string nameIn = filename;
    const string nameOut = nameIn.substr(nameIn.rfind('/')+1, nameIn.length() - nameIn.rfind('/') - 6);
    
    cout << "Reading file: " << nameIn << " writing to " << nameOut << endl;
    
    TEyeEvent* event = 0;//new TEyeEvent();
    // TEyeEventFile file(nameIn.c_str(), "READ");
    TFile* file = TFile::Open(nameIn.c_str());
    TTree* data = (TTree*) file->Get("EyeEventTree");
    data->SetBranchAddress("EyeEvent", &event);
    
    //TEyeEventFile *out = 0;
    TFile* out = 0;
    TTree* outTree = 0;
    unsigned int iOut = 0;
    unsigned int iOutFile = 0;
    
    // loop over events on file
    //while (file.ReadEvent(&event)) {
    for (int iEv=0; iEv<data->GetEntries(); ++iEv) {

      data->GetEntry(iEv);
      
      TEyeEventHeader* header = event->GetEventHeader();
      
      cout << " Eye: " << header->GetEyeNo() << " ";
      cout << " Event Id: " << header->GetEventNo() << " ";      
      cout << " sec: " << header->GetTimeStamp()->GetGPSSec();
      cout << endl;
      
      if (((iOut++) % nSplit) == 0 || !out) {
        if (out) {
          out->Close();
          out = 0;
        }
        ostringstream name;
        name << nameOut << "_split" << (iOutFile++) << ".root";
        cout << "Opening new output file: " << name.str() << endl;
        //out = new TEyeEventFile(name.str().c_str(), "RECREATE");
        out = TFile::Open(name.str().c_str(), "CREATE");
        out->cd();
        outTree = new TTree("EyeEventTree", "EventEvent tree");
        outTree->SetDirectory(out);
        outTree->Branch("EyeEvent", &event);
      }
      
      // out->WriteEvent(&event);
      outTree->Fill();

    }// loop on events

    if (out) {
      out->Write();
      out->Close();
      out = 0;
    }

    cout << " closing file " << endl;
    file->Close();

  } // loop input files
  
  filelist.close();

  return 0;
}