CosmicGenFilterHelix.h

Go to the documentation of this file.

#include "FWCore/Framework/interface/EDFilter.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Event.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/EDGetToken.h"

#include "DataFormats/GeometrySurface/interface/Plane.h"
#include "DataFormats/GeometrySurface/interface/Cylinder.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/GeometryVector/interface/GlobalVector.h"

#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"

#include <TObjArray.h>

#include <vector>

class MagneticField;
class Propagator;

class CosmicGenFilterHelix : public edm::EDFilter {
 public:
  explicit CosmicGenFilterHelix(const edm::ParameterSet& config);
  virtual ~CosmicGenFilterHelix();

  virtual void beginJob();
  virtual bool filter(edm::Event &event, const edm::EventSetup &eventSetup);
  virtual void endJob();

 private:
  bool propagateToCutCylinder(const GlobalPoint &vertStart, const GlobalVector &momStart,
                  int charge, const MagneticField *field,
                              const Propagator *propagator); //non-const: monitorEnd
  bool charge(int id, int &charge) const;
  const MagneticField* getMagneticField(const edm::EventSetup &setup) const;
  const Propagator* getPropagator(const edm::EventSetup &setup) const;
// ----------member data ---------------------------

  edm::EDGetTokenT<edm::HepMCProduct> theSrcToken;
  const std::vector<int>  theIds; 
  const std::vector<int>  theCharges; 
  const std::string thePropagatorName; // tag to get propagator from ESetup
  const double      theMinP2; 
  const double      theMinPt2; 

  Cylinder::ConstCylinderPointer theTargetCylinder; 
  Plane::ConstPlanePointer theTargetPlaneMin; 
  Plane::ConstPlanePointer theTargetPlaneMax; 

  unsigned int theNumTotal; 
  unsigned int theNumPass; 

  // for monitoring:
  void createHistsStart(const char *dirName, TObjArray &hists);
  void createHistsEnd(const char *dirName, TObjArray &hists);
  void monitorStart(const GlobalPoint &vert, const GlobalVector &mom, int charge,TObjArray &hists);
  void monitorEnd(const GlobalPoint &endVert, const GlobalVector &endMom,
          const GlobalPoint &vert, const GlobalVector &mom, double path, TObjArray &hists);
  bool equidistLogBins(double* bins, int nBins, double first, double last) const;
  const bool theDoMonitor;   
  TObjArray theHistsBefore; 
  TObjArray theHistsAfter;  

};