PileUpProducer.h

Go to the documentation of this file.

#ifndef FastSimulation_PileUpProducer_PileUpProducer_H
#define FastSimulation_PileUpProducer_PileUpProducer_H

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "SimDataFormats/PileupSummaryInfo/interface/PileupMixingContent.h"
#include "SimDataFormats/PileupSummaryInfo/interface/PileupVertexContent.h"

#include <vector>
#include <string>
#include <fstream>
#include "TH1F.h"

namespace edm {
  class Event;
  class EventSetup;
  class LuminosityBlock;
  class ParameterSet;
  class Run;
}

class TFile;
class TTree;
class TBranch;
class PUEvent;

class PrimaryVertexGenerator;

class PileUpProducer : public edm::stream::EDProducer <>
{

 public:

  explicit PileUpProducer(edm::ParameterSet const & p);
  virtual ~PileUpProducer();
  virtual void beginRun(edm::Run const&, edm::EventSetup const&) override;
  virtual void beginLuminosityBlock(edm::LuminosityBlock const&, edm::EventSetup const&) override;
  virtual void endRun(edm::Run const&, edm::EventSetup const&) override;
  virtual void produce(edm::Event & e, const edm::EventSetup & c) override;

 private:

  void save();

  bool read(std::string inputFile);

 private:

  PrimaryVertexGenerator* theVertexGenerator;

  double averageNumber_;
  std::vector<std::string> theFileNames;
  std::string inputFile;
  unsigned theNumberOfFiles;
  bool usePoisson_;

  std::vector<TFile*> theFiles;
  std::vector<TTree*> theTrees;
  std::vector<TBranch*> theBranches;
  std::vector<PUEvent*> thePUEvents;
  std::vector<unsigned> theCurrentEntry;
  std::vector<unsigned> theCurrentMinBiasEvt;
  std::vector<unsigned> theNumberOfEntries;
  std::vector<unsigned> theNumberOfMinBiasEvts;

  std::ofstream myOutputFile;
  unsigned myOutputBuffer;

  TH1F * hprob;
  std::vector<int> dataProbFunctionVar;
  std::vector<double> dataProb;
  int varSize;
  int probSize;

  bool currentValuesWereSet;
};

#endif