split_ad.cc

Go to the documentation of this file.

#include <iostream>
#include <IoAuger.h>

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

#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]);
  
  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;
    
    AugerFile file(nameIn.c_str(), AugerFile::eRead);
    
    AugerFile *out = 0;
    unsigned int iOut = 0;
    unsigned int iOutFile = 0;
    
    // loop over events on file
    AugerEvent event;
    while (file.ReadNext(event) != AugerFile::eEOF) {
    
      unsigned int nstations = 0;
      if (event.HasSd()) {
        IoSdEvent sd = event.Sd();
        nstations = sd.NumberOfErrorZeroStation;
      }
      
      cout << " Event Id " << event.Id() << " ";
      cout << " Nstations: " << setw(2) << nstations;
      cout << " NEyes: " << event.Eyes.size( ) << endl;
      
      if (event.Eyes.size() == 0)
        continue;
      
      // loop on eye events
      bool hasHEorCO = false;
      AugerEvent::EyeIterator eye;
      for(eye= event.EyesBegin(); eye!=event.EyesEnd(); ++eye) {
        
        if (eye->GetEventHeader()->GetEyeNo() >= 4)
          hasHEorCO = true;
        
        cout << "                            ";
        cout << "          eye: "<<eye->GetEventHeader()->GetEyeNo()
             << " sec: " << eye->GetEventHeader()->GetTimeStamp()->GetGPSSec()
             << endl;
        // add here cut on TIMES, if you want   
      } // loop on eyes

      if (!hasHEorCO)
        continue;
      
      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 AugerFile(name.str().c_str(), AugerFile::eWrite);
      }
    
      //    out->Write(*(event.RawAugerEvent()), "VERBOSE");
      out->Write(event, "VERBOSE");

    }// loop on events

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

    cout << " closing file " << endl;
    file.Close();

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

  return 0;
}