#include <FastjetJetProducer.h>

Inheritance diagram for FastjetJetProducer:

Public Member Functions
	FastjetJetProducer (const edm::ParameterSet &iConfig)

virtual void	produce (edm::Event &iEvent, const edm::EventSetup &iSetup)

virtual	~FastjetJetProducer ()

Public Member Functions inherited from VirtualJetProducer
std::string	jetType () const

	VirtualJetProducer (const edm::ParameterSet &iConfig)

virtual	~VirtualJetProducer ()

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

virtual	~EDProducer ()

Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

boost::function< void(const BranchDescription &)>	registrationCallback () const
	used by the fwk to register list of products More...

virtual	~ProducerBase ()

Protected Member Functions
virtual void	produceTrackJets (edm::Event &iEvent, const edm::EventSetup &iSetup)

virtual void	runAlgorithm (edm::Event &iEvent, const edm::EventSetup &iSetup)

Protected Member Functions inherited from VirtualJetProducer
virtual void	copyConstituents (const std::vector< fastjet::PseudoJet > &fjConstituents, reco::Jet *jet)

virtual std::vector < reco::CandidatePtr >	getConstituents (const std::vector< fastjet::PseudoJet > &fjConstituents)

virtual void	inputTowers ()

virtual bool	isAnomalousTower (reco::CandidatePtr input)

bool	makeBasicJet (const JetType::Type &fTag)

bool	makeCaloJet (const JetType::Type &fTag)

bool	makeGenJet (const JetType::Type &fTag)

bool	makePFClusterJet (const JetType::Type &fTag)

bool	makePFJet (const JetType::Type &fTag)

virtual void	makeProduces (std::string s, std::string tag="")

bool	makeTrackJet (const JetType::Type &fTag)

void	offsetCorrectJets (std::vector< fastjet::PseudoJet > &orphanInput)

virtual void	output (edm::Event &iEvent, edm::EventSetup const &iSetup)

template<typename T >
void	writeCompoundJets (edm::Event &iEvent, edm::EventSetup const &iSetup)
	function template to write out the outputs More...

template<typename T >
void	writeJets (edm::Event &iEvent, edm::EventSetup const &iSetup)

Protected Member Functions inherited from edm::EDProducer
CurrentProcessingContext const *	currentContext () const

Protected Member Functions inherited from edm::ProducerBase
template<class TProducer , class TMethod >
void	callWhenNewProductsRegistered (TProducer *iProd, TMethod iMethod)

Private Attributes
float	dxyTrVtxMax_

float	dzTrVtxMax_

float	maxVtxZ_

int	minVtxNdof_

double	muCut_
	Jet pruning technique. More...

int	nFilt_
	for filtering, trimming: dR scale of sub-clustering More...

double	RcutFactor_
	for pruning: constituent minimum pt fraction of parent cluster More...

double	rFilt_
	for mass-drop tagging, symmetry cut: min(pt1^2,pt2^2) * dR(1,2) / mjet > ycut More...

double	trimPtFracMin_
	for filtering, pruning: number of subjets expected More...

bool	useFiltering_
	Mass-drop tagging for boosted Higgs. More...

bool	useMassDropTagger_

bool	useOnlyOnePV_

bool	useOnlyVertexTracks_

bool	usePruning_
	Jet trimming technique. More...

bool	useTrimming_
	Jet filtering technique. More...

double	yCut_
	for mass-drop tagging, m0/mjet (m0 = mass of highest mass subjet) More...

double	zCut_
	for trimming: constituent minimum pt fraction of full jet More...

Additional Inherited Members
Public Types inherited from VirtualJetProducer
typedef boost::shared_ptr < fastjet::GhostedAreaSpec >	ActiveAreaSpecPtr

typedef boost::shared_ptr < fastjet::AreaDefinition >	AreaDefinitionPtr

typedef boost::shared_ptr < fastjet::ClusterSequence >	ClusterSequencePtr

typedef boost::shared_ptr < fastjet::JetDefinition >	JetDefPtr

typedef boost::shared_ptr < fastjet::JetDefinition::Plugin >	PluginPtr

typedef boost::shared_ptr < fastjet::RangeDefinition >	RangeDefPtr

Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

typedef WorkerT< EDProducer >	WorkerType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Attributes inherited from VirtualJetProducer
bool	doAreaDiskApprox_

bool	doAreaFastjet_

bool	doFastJetNonUniform_

bool	doPUOffsetCorr_

bool	doPVCorrection_

bool	doRhoFastjet_

ActiveAreaSpecPtr	fjActiveArea_

AreaDefinitionPtr	fjAreaDefinition_

ClusterSequencePtr	fjClusterSeq_

std::vector< fastjet::PseudoJet >	fjInputs_

JetDefPtr	fjJetDefinition_

std::vector< fastjet::PseudoJet >	fjJets_

PluginPtr	fjPlugin_

RangeDefPtr	fjRangeDef_

double	inputEMin_

double	inputEtMin_

std::vector< edm::Ptr < reco::Candidate > >	inputs_

std::string	jetAlgorithm_

std::string	jetCollInstanceName_

double	jetPtMin_

std::string	jetType_

JetType::Type	jetTypeE

unsigned int	maxInputs_

unsigned int	minSeed_

std::string	moduleLabel_

unsigned int	nExclude_

std::vector< double >	puCenters_

std::string	puSubtractorName_

double	puWidth_

bool	restrictInputs_

double	rParam_

edm::InputTag	src_

edm::InputTag	srcPVs_

boost::shared_ptr < PileUpSubtractor >	subtractor_

bool	useDeterministicSeed_

bool	useExplicitGhosts_

reco::Particle::Point	vertex_

double	voronoiRfact_

bool	writeCompound_

Detailed Description

Definition at line 8 of file FastjetJetProducer.h.

Constructor & Destructor Documentation

FastjetJetProducer::FastjetJetProducer ( const edm::ParameterSet & iConfig )

explicit

Definition at line 58 of file FastjetJetProducer.cc.

References dxyTrVtxMax_, dzTrVtxMax_, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), maxVtxZ_, minVtxNdof_, muCut_, nFilt_, RcutFactor_, rFilt_, trimPtFracMin_, VirtualJetProducer::useExplicitGhosts_, useFiltering_, useMassDropTagger_, useOnlyOnePV_, useOnlyVertexTracks_, usePruning_, useTrimming_, yCut_, and zCut_.

   : VirtualJetProducer( iConfig ),
     useMassDropTagger_(false),
     useFiltering_(false),
     useTrimming_(false),
     usePruning_(false),
     muCut_(-1.0),
     yCut_(-1.0),
     rFilt_(-1.0),
     nFilt_(-1),
     trimPtFracMin_(-1.0),
     zCut_(-1.0),
     RcutFactor_(-1.0)
 {
 
   if ( iConfig.exists("UseOnlyVertexTracks") )
     useOnlyVertexTracks_ = iConfig.getParameter<bool>("UseOnlyVertexTracks");
   else 
     useOnlyVertexTracks_ = false;
   
   if ( iConfig.exists("UseOnlyOnePV") )
     useOnlyOnePV_        = iConfig.getParameter<bool>("UseOnlyOnePV");
   else
     useOnlyOnePV_ = false;
 
   if ( iConfig.exists("DzTrVtxMax") )
     dzTrVtxMax_          = iConfig.getParameter<double>("DzTrVtxMax");
   else
     dzTrVtxMax_ = 999999.;
   if ( iConfig.exists("DxyTrVtxMax") )
     dxyTrVtxMax_          = iConfig.getParameter<double>("DxyTrVtxMax");
   else
     dxyTrVtxMax_ = 999999.;
   if ( iConfig.exists("MinVtxNdof") )
     minVtxNdof_ = iConfig.getParameter<int>("MinVtxNdof");
   else
     minVtxNdof_ = 5;
   if ( iConfig.exists("MaxVtxZ") )
     maxVtxZ_ = iConfig.getParameter<double>("MaxVtxZ");
   else
     maxVtxZ_ = 15;
 
 
   if ( iConfig.exists("useFiltering") ||
        iConfig.exists("useTrimming") ||
        iConfig.exists("usePruning") ||
        iConfig.exists("useMassDropTagger") ) {
     useMassDropTagger_=false;
     useFiltering_=false;
     useTrimming_=false;
     usePruning_=false;
     rFilt_=-1.0;
     nFilt_=-1;
     trimPtFracMin_=-1.0;
     zCut_=-1.0;
     RcutFactor_=-1.0;
     muCut_=-1.0;
     yCut_=-1.0;
     useExplicitGhosts_ = true;
 
 
     if ( iConfig.exists("useMassDropTagger") ) {
       useMassDropTagger_ = true;
       muCut_ = iConfig.getParameter<double>("muCut");
       yCut_ = iConfig.getParameter<double>("yCut");
     }
 
     if ( iConfig.exists("useFiltering") ) {
       useFiltering_ = true;
       rFilt_ = iConfig.getParameter<double>("rFilt");
       nFilt_ = iConfig.getParameter<int>("nFilt");
     }
   
     if ( iConfig.exists("useTrimming") ) {
       useTrimming_ = true;
       rFilt_ = iConfig.getParameter<double>("rFilt");
       trimPtFracMin_ = iConfig.getParameter<double>("trimPtFracMin");
     }
 
     if ( iConfig.exists("usePruning") ) {
       usePruning_ = true;
       zCut_ = iConfig.getParameter<double>("zcut");
       RcutFactor_ = iConfig.getParameter<double>("rcut_factor");
       nFilt_ = iConfig.getParameter<int>("nFilt");
     }
 
   }
 
 }

FastjetJetProducer::~FastjetJetProducer ( )

virtual

Definition at line 150 of file FastjetJetProducer.cc.

151 {

152 }

Member Function Documentation

void FastjetJetProducer::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

virtual

Reimplemented from VirtualJetProducer.

Reimplemented in cms::CATopJetProducer.

Definition at line 159 of file FastjetJetProducer.cc.

References VirtualJetProducer::jetTypeE, VirtualJetProducer::makeTrackJet(), VirtualJetProducer::produce(), and produceTrackJets().

Referenced by cms::CATopJetProducer::produce().

 {
 
   // for everything but track jets
   if (!makeTrackJet(jetTypeE)) {
  
      // use the default production from one collection
      VirtualJetProducer::produce( iEvent, iSetup );
 
   } else { // produce trackjets from tracks grouped per primary vertex
 
     produceTrackJets(iEvent, iSetup);
   
   }
 
 }

void FastjetJetProducer::produceTrackJets	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

protectedvirtual

Definition at line 177 of file FastjetJetProducer.cc.

References dxyTrVtxMax_, dzTrVtxMax_, VirtualJetProducer::fjClusterSeq_, VirtualJetProducer::fjInputs_, VirtualJetProducer::fjJets_, newFWLiteAna::found, edm::Event::getByLabel(), VirtualJetProducer::getConstituents(), i, VirtualJetProducer::inputs_, VirtualJetProducer::inputTowers(), metsig::jet, fwrapper::jets, LogDebug, maxVtxZ_, minVtxNdof_, edm::Event::put(), runAlgorithm(), reco::Jet::setJetArea(), reco::Jet::setPileup(), reco::TrackJet::setPrimaryVertex(), reco::LeafCandidate::setVertex(), VirtualJetProducer::src_, VirtualJetProducer::srcPVs_, useOnlyOnePV_, useOnlyVertexTracks_, VirtualJetProducer::vertex_, and reco::writeSpecific().

Referenced by produce().

 {
 
     // read in the track candidates
     edm::Handle<edm::View<reco::RecoChargedRefCandidate> > inputsHandle;
     iEvent.getByLabel(src_, inputsHandle);
     // make collection with pointers so we can play around with it
     std::vector<edm::Ptr<reco::RecoChargedRefCandidate> > allInputs;
     std::vector<edm::Ptr<reco::Candidate> > origInputs;
     for (size_t i = 0; i < inputsHandle->size(); ++i) {
       allInputs.push_back(inputsHandle->ptrAt(i));
       origInputs.push_back(inputsHandle->ptrAt(i));
     }
 
     // read in the PV collection
     edm::Handle<reco::VertexCollection> pvCollection;
     iEvent.getByLabel(srcPVs_, pvCollection);
     // define the overall output jet container
     std::auto_ptr<std::vector<reco::TrackJet> > jets(new std::vector<reco::TrackJet>() );
 
     // loop over the good vertices, clustering for each vertex separately
     for (reco::VertexCollection::const_iterator itVtx = pvCollection->begin(); itVtx != pvCollection->end(); ++itVtx) {
       if (itVtx->isFake() || itVtx->ndof() < minVtxNdof_ || fabs(itVtx->z()) > maxVtxZ_) continue;
 
       // clear the intermediate containers
       inputs_.clear();
       fjInputs_.clear();
       fjJets_.clear();
 
       // if only vertex-associated tracks should be used
       if (useOnlyVertexTracks_) {
         // loop over the tracks associated to the vertex
         for (reco::Vertex::trackRef_iterator itTr = itVtx->tracks_begin(); itTr != itVtx->tracks_end(); ++itTr) {
           // whether a match was found in the track candidate input
           bool found = false;
           // loop over input track candidates
           for (std::vector<edm::Ptr<reco::RecoChargedRefCandidate> >::iterator itIn = allInputs.begin(); itIn != allInputs.end(); ++itIn) {
             // match the input track candidate to the track from the vertex
             reco::TrackRef trref(itTr->castTo<reco::TrackRef>());
             // check if the tracks match
             if ((*itIn)->track() == trref) {
               found = true;
               // add this track candidate to the input for clustering
               inputs_.push_back(*itIn);
               // erase the track candidate from the total list of input, so we don't reuse it later
               allInputs.erase(itIn);
               // found the candidate track corresponding to the vertex track, so stop the loop
               break;
             } // end if match found
           } // end loop over input tracks
           // give an info message in case no match is found (can happen if candidates are subset of tracks used for clustering)
           if (!found) edm::LogInfo("FastjetTrackJetProducer") << "Ignoring a track at vertex which is not in input track collection!";
         } // end loop over tracks associated to vertex
       // if all inpt track candidates should be used
       } else {
         // loop over input track candidates
         for (std::vector<edm::Ptr<reco::RecoChargedRefCandidate> >::iterator itIn = allInputs.begin(); itIn != allInputs.end(); ++itIn) {
           // check if the track is close enough to the vertex
           float dz = (*itIn)->track()->dz(itVtx->position());
           float dxy = (*itIn)->track()->dxy(itVtx->position());
           if (fabs(dz) > dzTrVtxMax_) continue;
           if (fabs(dxy) > dxyTrVtxMax_) continue;
           bool closervtx = false;
           // now loop over the good vertices a second time
           for (reco::VertexCollection::const_iterator itVtx2 = pvCollection->begin(); itVtx2 != pvCollection->end(); ++itVtx2) {
             if (itVtx->isFake() || itVtx->ndof() < minVtxNdof_ || fabs(itVtx->z()) > maxVtxZ_) continue;
             // and check this track is closer to any other vertex (if more than 1 vertex considered)
             if (!useOnlyOnePV_ &&
                 itVtx != itVtx2 &&
                 fabs((*itIn)->track()->dz(itVtx2->position())) < fabs(dz)) {
               closervtx = true;
               break; // 1 closer vertex makes the track already not matched, so break
             }
           }
           // don't add this track if another vertex is found closer
           if (closervtx) continue;
           // add this track candidate to the input for clustering
           inputs_.push_back(*itIn);
           // erase the track candidate from the total list of input, so we don't reuse it later
           allInputs.erase(itIn);
           // take a step back in the loop since we just erased
           --itIn;
         }
       }
 
       // convert candidates in inputs_ to fastjet::PseudoJets in fjInputs_
       fjInputs_.reserve(inputs_.size());
       inputTowers();
       LogDebug("FastjetTrackJetProducer") << "Inputted towers\n";
 
       // run algorithm, using fjInputs_, modifying fjJets_ and allocating fjClusterSeq_
       runAlgorithm(iEvent, iSetup);
       LogDebug("FastjetTrackJetProducer") << "Ran algorithm\n";
 
       // convert our jets and add to the overall jet vector
       for (unsigned int ijet=0;ijet<fjJets_.size();++ijet) {
         // get the constituents from fastjet
         std::vector<fastjet::PseudoJet> fjConstituents = sorted_by_pt(fjClusterSeq_->constituents(fjJets_[ijet]));
         // convert them to CandidatePtr vector
         std::vector<reco::CandidatePtr> constituents = getConstituents(fjConstituents);
         // fill the trackjet
         reco::TrackJet jet;
         // write the specifics to the jet (simultaneously sets 4-vector, vertex).
         writeSpecific( jet,
                        reco::Particle::LorentzVector(fjJets_[ijet].px(), fjJets_[ijet].py(), fjJets_[ijet].pz(), fjJets_[ijet].E()),
                        vertex_, constituents, iSetup);
         jet.setJetArea(0);
         jet.setPileup(0);
         jet.setPrimaryVertex(edm::Ref<reco::VertexCollection>(pvCollection, (int) (itVtx-pvCollection->begin())));
         jet.setVertex(itVtx->position());
         jets->push_back(jet);
       }
 
       if (useOnlyOnePV_) break; // stop vertex loop if only one vertex asked for
     } // end loop over vertices
 
     // put the jets in the collection
     LogDebug("FastjetTrackJetProducer") << "Put " << jets->size() << " jets in the event.\n";
     iEvent.put(jets);
 
 }

void FastjetJetProducer::runAlgorithm	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

protectedvirtual

Implements VirtualJetProducer.

Reimplemented in cms::CATopJetProducer.

Definition at line 301 of file FastjetJetProducer.cc.

References VirtualJetProducer::doAreaFastjet_, VirtualJetProducer::doRhoFastjet_, Filter_cff::Filter, alcazmumu_cfi::filter, VirtualJetProducer::fjAreaDefinition_, VirtualJetProducer::fjClusterSeq_, VirtualJetProducer::fjInputs_, VirtualJetProducer::fjJetDefinition_, VirtualJetProducer::fjJets_, VirtualJetProducer::jetPtMin_, muCut_, nFilt_, RcutFactor_, rFilt_, trimPtFracMin_, useFiltering_, useMassDropTagger_, usePruning_, useTrimming_, VirtualJetProducer::voronoiRfact_, yCut_, and zCut_.

Referenced by produceTrackJets().

 {
   // run algorithm
   /*
   fjInputs_.clear();
   double px, py , pz, E;
   string line;
   std::ifstream fin("dump3.txt");
   while (getline(fin, line)){
     if (line == "#END") break;
     if (line.substr(0,1) == "#") {continue;}
     istringstream istr(line);
     istr >> px >> py >> pz >> E;
     // create a fastjet::PseudoJet with these components and put it onto
     // back of the input_particles vector
     fastjet::PseudoJet j(px,py,pz,E);
     //if ( fabs(j.rap()) < inputEtaMax )
     fjInputs_.push_back(fastjet::PseudoJet(px,py,pz,E)); 
   }
   fin.close();
   */
 
   if ( !doAreaFastjet_ && !doRhoFastjet_) {
     fjClusterSeq_ = ClusterSequencePtr( new fastjet::ClusterSequence( fjInputs_, *fjJetDefinition_ ) );
   } else if (voronoiRfact_ <= 0) {
     fjClusterSeq_ = ClusterSequencePtr( new fastjet::ClusterSequenceArea( fjInputs_, *fjJetDefinition_ , *fjAreaDefinition_ ) );
   } else {
     fjClusterSeq_ = ClusterSequencePtr( new fastjet::ClusterSequenceVoronoiArea( fjInputs_, *fjJetDefinition_ , fastjet::VoronoiAreaSpec(voronoiRfact_) ) );
   }
 
   if ( !useTrimming_ && !useFiltering_ && !usePruning_ ) {
     fjJets_ = fastjet::sorted_by_pt(fjClusterSeq_->inclusive_jets(jetPtMin_));
   }
   else {
     fjJets_.clear();
     std::vector<fastjet::PseudoJet> tempJets = fastjet::sorted_by_pt(fjClusterSeq_->inclusive_jets(jetPtMin_));
 
     fastjet::MassDropTagger md_tagger( muCut_, yCut_ );
     fastjet::Filter trimmer( fastjet::Filter(fastjet::JetDefinition(fastjet::kt_algorithm, rFilt_), fastjet::SelectorPtFractionMin(trimPtFracMin_)));
     fastjet::Filter filter( fastjet::Filter(fastjet::JetDefinition(fastjet::cambridge_algorithm, rFilt_), fastjet::SelectorNHardest(nFilt_)));
     fastjet::Pruner pruner(fastjet::cambridge_algorithm, zCut_, RcutFactor_);
 
     std::vector<fastjet::Transformer const *> transformers;
 
     if ( useMassDropTagger_ ) {
       transformers.push_back(&md_tagger);
     }
     if ( useTrimming_ ) {
       transformers.push_back(&trimmer);
     } 
     if ( useFiltering_ ) {
       transformers.push_back(&filter);
     } 
     if ( usePruning_ ) {
       transformers.push_back(&pruner);
     }
 
 
 
     for ( std::vector<fastjet::PseudoJet>::const_iterator ijet = tempJets.begin(),
         ijetEnd = tempJets.end(); ijet != ijetEnd; ++ijet ) {
 
       fastjet::PseudoJet transformedJet = *ijet;
       bool passed = true;
       for ( std::vector<fastjet::Transformer const *>::const_iterator itransf = transformers.begin(),
           itransfEnd = transformers.end(); itransf != itransfEnd; ++itransf ) {
     if ( transformedJet != 0 ) {
       transformedJet = (**itransf)(transformedJet);
     }
     else {
       passed=false;
     }
       }
       if ( passed ) 
     fjJets_.push_back( transformedJet );
     }
   }
 
 }