testEvent.cc

Go to the documentation of this file.

// $Id: testEvent.cc,v 1.1 2007/02/09 04:43:48 fmwk Exp $
//
// Test making and event and writing it out, then reading it back into
// memory
//
// messier@indiana.edu
extern "C" {
#include <unistd.h>
#include <sys/time.h>
}
#include <string>
#include <iostream>
#include <cassert>
#include "TRint.h"
#include "TROOT.h"
#include "TFile.h"
#include "TTree.h"
#include "TBranch.h"
#include "TBrowser.h"
#include "EventDataModel/Event.h"
#include "EventDataModel/DataBucket.h"
#include "EventDataModel/test/EDMTestDataObject.h"

static void testMakeOne() {
  std::cerr << "\ntestMakeOne::Start\n";
  // Test that I can make one event
  struct timeval  tv;
  struct timezone tz;
  gettimeofday(&tv, &tz);
  edm::Event e1;

  edm::TestDataObject testDigit;
  edm::DataBucket detsim(e1.fDetSim);
  detsim.MakeFolder("./hits");
  for (int i=0; i<10; ++i) detsim.Put(testDigit,"./hits");
  std::cerr << "\nFull MC event contains\n";
  detsim.List();
  
  e1.Clear();
  std::cerr << "\nAfter clear:\n";
  detsim.List();
  std::cerr << "\ntestMakeOne::Done\n";
}

//......................................................................

void testWrite()
{
  int bSz  = 16000;
  int sLvl = 99; // 99 gives core dump on root >=3.03.09
  
  std::cerr << "\ntestWrite::Start\n";
  
  TFile* f       = new TFile("events.root","RECREATE");
  TTree* evtTree = new TTree("event_tree","EDMEvent tree");
  struct timeval  tv;
  struct timezone tz;
  gettimeofday(&tv, &tz);
  edm::Event* evt  = new edm::Event;

  assert(f);
  assert(evtTree);
  assert(evt);

  evtTree->Branch("fHeader", "edm::EventHeader",&evt->fHeader, bSz,sLvl);
  evtTree->Branch("fTop",    "TFolder",         &evt->fTop,    bSz,sLvl);
  evtTree->Branch("fMC",     "TFolder",         &evt->fMC,     bSz,sLvl);
  evtTree->Branch("fDetSim", "TFolder",         &evt->fDetSim, bSz,sLvl);
  evtTree->Branch("fDAQ",    "TFolder",         &evt->fDAQ,    bSz,sLvl);
  evtTree->Branch("fRaw",    "TFolder",         &evt->fRaw,    bSz,sLvl);
  evtTree->Branch("fCal",    "TFolder",         &evt->fCal,    bSz,sLvl);
  evtTree->Branch("fReco",   "TFolder",         &evt->fReco,   bSz,sLvl);
  evtTree->Branch("fUser",   "TFolder",         &evt->fUser,   bSz,sLvl);
  evtTree->Branch("fSummary","TFolder",         &evt->fSummary,bSz,sLvl);
  
  // evt->fHeader->SetRun(100);
  // evt->fHeader->SetSubRun(0);
  for (int i=1; i<1001; ++i) {
    evt->Clear();
    struct timeval  tv;
    struct timezone tz;
    gettimeofday(&tv, &tz);
    // evt->fHeader->SetEvent(i);
    // evt->fHeader->SetTimeStamp(tv.tv_sec,tv.tv_usec);

    edm::TestDataObject tobj;
    edm::DataBucket detsim(evt->fDetSim);
    edm::DataBucket raw   (evt->fRaw);
    detsim.MakeFolder("hits");
    raw.   MakeFolder("tobj");
    for (int j=0; j<1000; ++j) {
      tobj.fChannel = j;
      tobj.fADC     = 10-j;
      tobj.fTDC     = 10-j;
      detsim.Put(tobj,"./hits");
      raw.   Put(tobj,"./tobj");
    }
    evtTree->Fill();
    if (i%10 == 0) {
      std::cerr << i << " events..." << std::endl;
      detsim.List();
      raw.List();
    }
  }
  f->Write();
  f->Close();
  // delete evt;
  // Delete of tree causes crash. Presumably ownership is in the file
  // delete evtTree
  delete f;
  std::cerr << "\ntestWrite::Done\n";
}

//......................................................................

static void testRead() 
{
  TFile*    f = new TFile("events.root");
  TTree*    evtTree;
  TBranch*  bDetSim;
  TBranch*  bRaw;
  edm::Event* evt = new edm::Event;
  
  evtTree   = (TTree*)f->Get("event_tree");
  assert(evtTree);
  evtTree->Print();

  bDetSim = evtTree->GetBranch("fDetSim");
  bDetSim->SetAddress(&evt->fDetSim);

  bRaw = evtTree->GetBranch("fRaw");
  bRaw->SetAutoDelete(kTRUE);

  unsigned int nevt = (int)evtTree->GetEntries();
  std::vector<const edm::TestDataObject*> testDigits;
  std::cerr << "Nevents = " << nevt << std::endl;
  for (unsigned int i=0; i<nevt; ++i) {
    evt->Clear();
    bDetSim->GetEntry(i);
    bRaw->GetEntry(i);
    edm::DataBucket detsim(evt->fDetSim);
    detsim.Get("./hits/edm::TestDataObjects",testDigits);
    for (unsigned int i=0; i<testDigits.size(); ++i) {
      if (i%10==0)
        std::cerr << "TestObjects: "
                  << testDigits[i]->fChannel << ":" <<testDigits[i]->fADC 
                  <<"\n";
    }
    testDigits.clear();
  }
  delete f;
}

//......................................................................

static void testBrowse() 
{
#ifdef TESTBROWSE
//
// Open file, read first event, start browser, then return
//
  TFile*    f = new TFile("events.root");
  TTree*    evtTree;
  TBranch*  evtBranch;
  EDMEvent* evt;
  
  evtTree   = (TTree*)f->Get("event_tree");
  assert(evtTree);
  evtBranch = evtTree->GetBranch("event_branch");
  assert(evtBranch);
  evt       = new EDMEvent();
  evtBranch->SetAddress(&evt);

  evt->Clear();
  evtBranch->GetEntry(0);

  TBrowser* b = new TBrowser();
#endif
}

//......................................................................

int main(int argc, char** argv) 
{
  TRint* app = new TRint("testEvent", &argc, argv);
  testMakeOne();
  std::string wopt = "-w";
  std::string ropt = "-r";
  std::string bopt = "-i";
  std::string oopt = "-1";
  if (argc<2) {
    std::cerr << "Need option: Please use -w, -r, or -i.\n";
  }
  else {
    for (int i=0; i<argc; ++i) {
      if      (wopt == argv[i]) testWrite();
      else if (ropt == argv[i]) testRead();
      else if (bopt == argv[i]) testBrowse();
      else if (oopt == argv[i]) testMakeOne();
      else std::cerr << "Unknown option: Please use -w, -r or -i.\n";
    }
  }
  app->Run();
  return 0;
}

---