chainevent_looping_cint.C

#include <vector>
#include <TFile.h>
using namespace std;

#if defined(__CINT__) && !defined(__MAKECINT__)
class loadFWLite {
public:
  loadFWLite() {
    gSystem->Load("libFWCoreFWLite");
    FWLiteEnabler::enable();
  }
};

static loadFWLite lfw;
#endif

#include "DataFormats/FWLite/interface/Handle.h"

#if !defined(__CINT__) && !defined(__MAKECINT__)
#include "DataFormats/TestObjects/interface/ThingCollection.h"
#endif

int chainevent_looping_cint() {
  vector<string> files;
  files.push_back("empty_a.root");
  files.push_back("good_a.root");
  files.push_back("empty_a.root");
  files.push_back("good_b.root");
  files.push_back("empty_a.root");
  fwlite::ChainEvent e(files);

  int i = 0;
  int returnValue = 0;
  TFile* f = 0;

  for (; e.isValid(); ++e, ++i) {
    if (e.getTFile() != f) {
      f = e.getTFile();
      cout << "New file " << f->GetName() << endl;
    }

    fwlite::Handle<vector<edmtest::Thing> > pThing;
    //pThing.getByLabel(e,"Thing","","TEST"); //WORKS
    pThing.getByLabel(e, "Thing");

    for (i = 0; i != pThing.ref().size(); ++i) {
      cout << pThing.ref().at(i).a << " ";
    }
    cout << endl;
  }
  if (i == 0) {
    cout << "First loop failed!" << endl;
    returnValue = 1;
  }
  e.toBegin();

  i = 0;
  for (; e; ++e, ++i) {
  }

  if (i == 0) {
    cout << "Second loop failed!" << endl;
    returnValue = 1;
  }

  i = 0;
  for (e.toBegin(); !e.atEnd(); ++e, ++i) {
    fwlite::Handle<vector<edmtest::Thing> > pThing;
    //pThing.getByLabel(e,"Thing","","TEST"); //WORKS
    pThing.getByLabel(e, "Thing");

    for (i = 0; i != pThing.ref().size(); ++i) {
      cout << pThing.ref().at(i).a << " ";
    }
    cout << endl;
    //DOES NOT WORK in CINT
    //for(vector<edmtest::Thing>::const_iterator it = pThing.data()->begin(); it != pThing.data()->end();++it) {
    //   cout <<(*it).a<<endl;
    //}
  }
  if (i == 0) {
    cout << "Third loop failed!" << endl;
    returnValue = 1;
  }
  e.to(0);
  for (int j = 0; j < 20; ++j) {
    int k = rand() % 10;
    if (e.to(k)) {
      edm::EventID id = e.id();
      cout << "Entry " << k << " Run " << id.run() << " event " << id.event() << endl;
    } else {
      cout << "Entry " << k << " is not valid" << endl;
    }
  }

  e.to(0);
  long size = e.size();
  edm::EventID last;
  for (long l = 0; l < size; ++l) {
    e.to(l);
    edm::EventID id = e.id();
    cout << "Entry " << l << " Run " << id.run() << " event " << id.event() << endl;
    if (last == id) {
      returnValue = 1;
      cout << "duplicate event (" << id.run() << "," << id.event() << ") seen at index " << l << " while testing 'to'"
           << endl;
      break;
    }
  }

  return returnValue;
}