adding PythiaHT filter

cms-sw · Dec 15, 2015 · 063cdb1 · 063cdb1
1 parent 0c54573
commit 063cdb1
Show file tree

Hide file tree

Showing 3 changed files with 181 additions and 0 deletions.
diff --git a/GeneratorInterface/GenFilters/interface/PythiaFilterHT.h b/GeneratorInterface/GenFilters/interface/PythiaFilterHT.h
@@ -0,0 +1,71 @@
+#ifndef PYTHIAFILTERHT_h
+#define PYTHIAFILTERHT_h
+// -*- C++ -*-
+//
+// Package:    PythiaFilterHT
+// Class:      PythiaFilterHT
+// 
+/**\class PythiaFilterHT PythiaFilterHT.cc IOMC/PythiaFilterHT/src/PythiaFilterHT.cc
+
+ Description: <one line class summary>
+
+ Implementation:
+     <Notes on implementation>
+*/
+//
+// Original Author:  Alejandro Gomez Espinosa
+//
+//
+
+
+// system include files
+#include <memory>
+
+// user include files
+#include "FWCore/Framework/interface/Frameworkfwd.h"
+#include "FWCore/Framework/interface/EDFilter.h"
+
+#include "FWCore/Framework/interface/Event.h"
+#include "FWCore/Framework/interface/MakerMacros.h"
+
+#include "FWCore/ParameterSet/interface/ParameterSet.h"
+
+
+//
+// class decleration
+//
+
+class PythiaFilterHT : public edm::EDFilter {
+   public:
+      explicit PythiaFilterHT(const edm::ParameterSet&);
+      ~PythiaFilterHT();
+
+
+      virtual bool filter(edm::Event&, const edm::EventSetup&);
+   private:
+      // ----------member data ---------------------------
+
+       std::string label_;
+       int particleID;
+       double minpcut;
+       double maxpcut;
+       double minptcut;
+       double minhtcut;
+       double maxptcut;
+       double minetacut;
+       double maxetacut;
+       double minrapcut;
+       double maxrapcut;
+       double maxphicut;
+       double minphicut;
+
+       double rapidity;
+
+       int status; 
+       int motherID;   
+       int processID;    
+       int theNumberOfTestedEvt;
+       int theNumberOfSelected;
+       int maxnumberofeventsinrun;
+};
+#endif
diff --git a/GeneratorInterface/GenFilters/src/PythiaFilterHT.cc b/GeneratorInterface/GenFilters/src/PythiaFilterHT.cc
@@ -0,0 +1,108 @@
+#include "GeneratorInterface/GenFilters/interface/PythiaFilterHT.h"
+#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
+#include <iostream>
+
+using namespace edm;
+using namespace std;
+
+
+PythiaFilterHT::PythiaFilterHT(const edm::ParameterSet& iConfig) :
+	label_(iConfig.getUntrackedParameter("moduleLabel",std::string("generator"))),
+	/*minpcut(iConfig.getUntrackedParameter("MinP", 0.)),
+	maxpcut(iConfig.getUntrackedParameter("MaxP", 10000.)),
+	minptcut(iConfig.getUntrackedParameter("MinPt", 0.)),
+	maxptcut(iConfig.getUntrackedParameter("MaxPt", 10000.)),
+	minetacut(iConfig.getUntrackedParameter("MinEta", -10.)),
+	maxetacut(iConfig.getUntrackedParameter("MaxEta", 10.)),
+	minrapcut(iConfig.getUntrackedParameter("MinRapidity", -20.)),
+	maxrapcut(iConfig.getUntrackedParameter("MaxRapidity", 20.)),
+	minphicut(iConfig.getUntrackedParameter("MinPhi", -3.5)),
+	maxphicut(iConfig.getUntrackedParameter("MaxPhi", 3.5)),*/
+	minhtcut(iConfig.getUntrackedParameter("MinHT", 0.)),
+	motherID(iConfig.getUntrackedParameter("MotherID", 0)) {
+
+		theNumberOfTestedEvt = 0;
+		theNumberOfSelected = 0;
+
+		cout << " Cut Definition: " << endl;
+		/*cout << " MinP  = " << minpcut << endl;
+		cout << " MaxP = " << maxpcut << endl;
+		cout << " MinPt = " << minptcut << endl;
+		cout << " MaxPt = " << maxptcut << endl;
+		cout << " MinEta = " << minetacut << endl;
+		cout << " MaxEta = " << maxetacut << endl;
+		cout << " MinRapidity = " << minrapcut << endl;
+		cout << " MaxRapidity = " << maxrapcut << endl;
+		cout << " MinPhi = " << minphicut << endl;
+		cout << " MaxPhi = " << maxphicut << endl;*/
+		cout << " MinHT = " << minhtcut << endl;
+		cout << " MotherID = " << motherID << endl;
+
+}
+
+
+PythiaFilterHT::~PythiaFilterHT() {
+	std::cout << "Total number of tested events = " << theNumberOfTestedEvt << std::endl;
+	std::cout << "Total number of accepted events = " << theNumberOfSelected << std::endl;
+}
+
+
+//
+// member functions
+//
+
+// ------------ method called to produce the data  ------------
+bool PythiaFilterHT::filter(edm::Event& iEvent, const edm::EventSetup& iSetup) {
+
+	theNumberOfTestedEvt++;
+	if(theNumberOfTestedEvt%1000 == 0) cout << "Number of tested events = " << theNumberOfTestedEvt <<  endl;
+
+	bool accepted = false;
+	Handle<HepMCProduct> evt;
+	iEvent.getByLabel(label_, evt);
+
+	const HepMC::GenEvent * myGenEvent = evt->GetEvent();
+
+	double HT = 0;
+
+	for ( HepMC::GenEvent::particle_const_iterator p = myGenEvent->particles_begin(); p != myGenEvent->particles_end(); ++p ) {
+
+		if ( ( (*p)->status() == 23 ) && ( ( abs( (*p)->pdg_id() ) < 6 ) || ( (*p)->pdg_id() == 21 ) ) ) {
+
+			/*rapidity = 0.5*log( ((*p)->momentum().e()+(*p)->momentum().pz()) / ((*p)->momentum().e()-(*p)->momentum().pz()) );
+
+			if ( (*p)->momentum().rho() > minpcut 
+					&& (*p)->momentum().rho() < maxpcut
+					&& (*p)->momentum().perp() > minptcut 
+					&& (*p)->momentum().perp() < maxptcut
+					&& (*p)->momentum().eta() > minetacut
+					&& (*p)->momentum().eta() < maxetacut 
+					&& rapidity > minrapcut
+					&& rapidity < maxrapcut 
+					&& (*p)->momentum().phi() > minphicut
+					&& (*p)->momentum().phi() < maxphicut ) {*/
+
+				if ( motherID == 0 ) {
+				   HT += (*p)->momentum().perp();
+				} else {
+					HepMC::GenParticle* mother = (*((*p)->production_vertex()->particles_in_const_begin()));
+					if (abs(mother->pdg_id()) == abs(motherID)) {
+						HT += (*p)->momentum().perp();
+					}
+				}
+			//}			
+		}
+	}
+	if ( HT > minhtcut ) accepted = true;
+
+	if (accepted){
+		theNumberOfSelected++;
+		cout << "========>  Event preselected " << theNumberOfSelected << " HT = " << HT << endl;
+		return true; 
+	} else {
+		//cout << "========>  Event rejected HT = " << HT << endl;
+		return false;
+	}
+
+}
+
diff --git a/GeneratorInterface/GenFilters/src/SealModule.cc b/GeneratorInterface/GenFilters/src/SealModule.cc
@@ -1,6 +1,7 @@
 #include "FWCore/Framework/interface/InputSourceMacros.h"
 #include "FWCore/Framework/interface/MakerMacros.h"
 #include "GeneratorInterface/GenFilters/interface/PythiaFilter.h"
+#include "GeneratorInterface/GenFilters/interface/PythiaFilterHT.h"
 #include "GeneratorInterface/GenFilters/interface/PythiaFilterMultiMother.h"
 #include "GeneratorInterface/GenFilters/interface/PythiaDauFilter.h"
 #include "GeneratorInterface/GenFilters/interface/PythiaFilterGammaJet.h"
@@ -48,6 +49,7 @@
   using cms::BHFilter;
   DEFINE_FWK_MODULE(LQGenFilter);
   DEFINE_FWK_MODULE(PythiaFilter);
+  DEFINE_FWK_MODULE(PythiaFilterHT);
   DEFINE_FWK_MODULE(PythiaFilterMultiMother);
   DEFINE_FWK_MODULE(PythiaDauFilter);
   DEFINE_FWK_MODULE(PythiaFilterGammaJet);