In This Package:

RootRawWriter.cc

Go to the documentation of this file.

/*
 *   Convert ".data" raw data file to ".rraw" root compitable raw data file
 *
 */
const char* usage_string = "\n\
   Usage: \n\
        RootRawWriter.exe [options] [aRawData.data|aRawData.list] ...\n\
        In the \"aRawData.list\" each line is a \".data\" file name.\n\
        By default the \".rraw\" file is made next to the \".data\" file.\n\
\n\
   options:\n\
        -d outdir: specify an output directory\n\
\n\
   return codes:\n\
         0: standard unix code for success\n\
         1: error, no output produced\n\
         2: error, partial output may have been produced\n\
";
/*   22 Jan, 2010
 *   Zhe Wang
 *
 *    Thu Jun 17 17:23:53 2010 bv@bnl.gov
 *    Let multiple .list or .data files be given.
 *
 *    Sun Jul 4 12:08:19 EDT 2010 bv@bnl.gov
 *    Let the user specify an output directory, fix return codes
 */

#include <iostream>
#include <fstream>
#include <vector>
#include <string>

#include <unistd.h>

#include "RawData/RawRecord.h"
#include "RawData/RawRecordFormat.h"
#include "RawData/RawDataReader.h"
#include "RootRawWriter/RRawTree.h"

using namespace std;
using namespace DayaBay::RawRecord_v06;  

// for getopt
extern char *optarg;
extern int optind, opterr, optopt;

int usage(int code, const string extra_message = "");

int main( int argc, char ** argv ) {
  int c=0;
  const char* outdir = 0;
  while ((c=getopt(argc,argv,"d:h?")) != EOF) {
    switch(c) {
    case 'd':
      outdir = optarg;
      break;
    case 'h':
    case '?':
      return usage(0);
    }
  }

  if( argc == 1 ) {
    return usage(1,"Require at least one input file.");
  }

  vector<string> filenames;

  for (int ind = optind; ind < argc; ++ind) {

    string parameter = argv[ind];
    string suffix = parameter.substr(parameter.size()-5,5);
    if( suffix != ".data" && suffix != ".list") {
      string msg = "Error: File extension \"" +suffix+ "\" not recognized.";
      return usage(1,msg);
    }

    if( suffix == ".data" ) {
      filenames.push_back(parameter.c_str());
      continue;
    }

    if( suffix == ".list" ) {
      ifstream filelist;
      filelist.open(parameter.c_str(), ios::in);
    
      string aLine;
      while(filelist >> aLine) {
        filenames.push_back(aLine);
      }
    
      filelist.close();
    }
  }


  for(unsigned int ci=0; ci<filenames.size(); ci++) {
    cout<<"RRaw::Processing "<<filenames[ci]<<endl;

    DayaBay::RawDataReader rr;

    bool ok = rr.open( filenames[ci].c_str() );
    if(!ok){
      cerr << "Failed to open " << filenames[ci] << endl;
      string msg = "Failed to open: \"" + filenames[ci] + "\"";
      return usage(1, msg);
    }

    RRawTree rrtree;

    string newfile = filenames[ci];
    newfile.replace(newfile.size()-4,4,"rraw");

    // But, if user wants output to go to some specific directory,
    // munge the filename to match
    if (outdir) {
      string::size_type ind = newfile.rfind("/");
      if (ind != string::npos) {
        newfile = newfile.substr(ind);
      }
      else {
        newfile = "/" + newfile;
      }
      newfile = outdir + newfile;
    }

    // Do not modify this line, scripts rely on it!
    cout << "RRaw::Processing to output file: " << newfile<< endl;

    int status = rrtree.open(newfile.c_str());
    if (!status) {
      cout <<"failed to open file: " << newfile << endl;
      return 2;
    }

    DayaBay::RawRecord* record;
    unsigned int nRecord;
    for( nRecord=1; /*empty field*/ ;nRecord++ ) {
      record = rr.nextRecord();

      if(!record) {
        return usage(2,"Error: Failed to load one record in raw data file.");
      }
      //cout<<record->size()<<endl;
      //record->dump( cout );

      unsigned int bufferSize = record->size();

      if( record->type() == RecordFormat::DataSeparator::type ) {
        DayaBay::RawDataSeparator* pSeparator 
          = dynamic_cast<DayaBay::RawDataSeparator*> (record);
        unsigned int bytes = pSeparator->blockSize();    
        unsigned int lines = (bytes+sizeof(unsigned int)-1)/sizeof(unsigned int);
        bufferSize = bufferSize + lines;
      }

      rrtree.fill( (Int32*)(record->buffer()), (Int32)bufferSize );

      if( record->type() == RecordFormat::FileEnd::type ) break;
    } // end of each record

    rrtree.close();

    cout<<"Read in "<<nRecord<<" records"<<endl;

  } // end of every file

  
  return 0;                     // 0 is unix standard for successful
} // end of main


int usage(int code, string extra_message)
{
  cout << usage_string << endl;
  if (extra_message != "") {
    cout << extra_message << endl << endl;
  }
  return code;
}



// Local Variables: **
// c-basic-offset:2 **
// indent-tabs-mode:nil **
// End: **

---