testWrite.cc

Go to the documentation of this file.

#include <iostream>
#include <string>
#include <ctime>
#include "TROOT.h"
#include "TRint.h"
#include "TRandom.h"
#include "EventDataModel/Event.h"
#include "EventDataModel/DataBucket.h"
#include "EventDataModel/EventHandle.h"
#include "IoModules/ReadWriteModule.h"
#include "TFile.h"
#include "DAQStatus.h"

// static TROOT gsTROOT("rootie","tootie");
static TRandom gRAND(time(0));

int main(int /*argc*/, char** argv)
{
  io::ReadWriteModule* iomod = new io::ReadWriteModule();
  iomod->SetOutputFileName(argv[1]);

  edm::EventHandle& evt = iomod->GetEvent();
  evt.DAQ().MakeFolder("./status");
  evt.Raw().MakeFolder("./digits");
  
  DAQStatus           daqStatus;
  std::vector<Digit*> digitList;
  
  unsigned int nevent;
  unsigned int ndigit;
  unsigned int ndigitold = 0;
  unsigned int ndigittot = 0;
  unsigned int minSize = 0;

  nevent = 10 + 10000*gRAND.Uniform();
  time_t t1 = time(0);
  for (unsigned int i=0; i<nevent; ++i) {
    evt.Clear();
    evt.Header().SetRun(1112);
    evt.Header().SetEvent(i+1);
  
    ndigit = 10 + 10000*gRAND.Uniform();
    ndigittot += ndigit;
    digitList.resize(ndigit);
    for (unsigned int j=ndigitold; j<ndigit; ++j) {
      digitList[j] = new Digit();
    }
    // Fill with random data so that compression efficiencies are not
    // over estimated
    long r1 = random();
    long r2 = random();
    for (unsigned int j=0; j<ndigit; ++j) {
      digitList[j]->fCrateId = (r1    )&0xFFFF;
      digitList[j]->fChanId  = (r1<<16)&0xFFFF;
      digitList[j]->fADC     = (r2    )&0xFFFF;
      digitList[j]->fTDC     = (r2<<16)&0xFFFF;
    }
    ndigitold = ndigit;

    evt.DAQ().Put(daqStatus,      "./status");
    evt.Raw().PutVector(digitList,"./digits");
    
    minSize += daqStatus.SizeOf() + ndigit*digitList[0]->SizeOf();

    iomod->WriteEvent();
  }
  delete iomod;
  time_t t2 = time(0);
  
  TFile f(argv[1]);
  unsigned int sz = f.GetSize();
  
  std::cerr << t2-t1 << "\t" 
            << nevent << "\t" 
            << ndigittot << "\t"
            << minSize << "\t"
            << sz << "\t"
            << f.GetCompressionFactor()
            << std::endl;
  return 0;
}

---