TreeSplitter.cc

Go to the documentation of this file.

#ifndef TREESPLITTER_CC
#define TREESPLITTER_CC

#include "TreeSplitter.h"

TreeSplitter::TreeSplitter(const edm::ParameterSet& iConfig) :
  treeFileName_( iConfig.getParameter<std::string>("InputFileName") ),
  outputFileName_( iConfig.getParameter<std::string>("OutputFileName") ),
  maxEvents_( iConfig.getParameter<int32_t>("MaxEvents") ),
  subSampleFirstEvent_( iConfig.getParameter<uint32_t>("SubSampleFirstEvent") ),
  subSampleMaxEvents_( iConfig.getParameter<uint32_t>("SubSampleMaxEvents") )
{
}

TreeSplitter::~TreeSplitter()
{
}

void TreeSplitter::endJob()
{
  std::cout << "Reading muon pairs from Root Tree in " << treeFileName_ << std::endl;
  RootTreeHandler rootTreeHandler;

  typedef std::vector<std::pair<lorentzVector,lorentzVector> > MuonPairVector;
  // MuonPairVector savedPair;
  std::vector<MuonPair> savedPair;
  rootTreeHandler.readTree(maxEvents_, treeFileName_, &savedPair, 0);
  // rootTreeHandler.readTree(maxEvents, inputRootTreeFileName_, &savedPair, &(MuScleFitUtils::genPair));

  // Loop on all the pairs
  std::vector<MuonPair> newSavedPair;
  // MuonPairVector newSavedPair;
  unsigned int i = 0;
  // MuonPairVector::iterator it = savedPair.begin();
  std::vector<MuonPair>::iterator it = savedPair.begin();
  std::cout << "Starting loop on " << savedPair.size() << " muons" << std::endl;
  uint32_t lastEvent = subSampleFirstEvent_ + subSampleMaxEvents_;
  for( ; it != savedPair.end(); ++it, ++i ) {
    // Save only events in the selected range
    if( i >= subSampleFirstEvent_ && i < lastEvent ) {
      newSavedPair.push_back(*it);
    }
  }
  rootTreeHandler.writeTree(outputFileName_, &newSavedPair, 0);
}

//define this as a plug-in
DEFINE_FWK_MODULE(TreeSplitter);

#endif