JetVertexAssociation.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    JetVertexAssociation
// Class:      JetVertexAssociation
// 
//
// Original Author:  Natalia Ilina
// Modified by Eduardo Luiggi
//
//         Created:  Tue Oct 31 10:52:41 CET 2006
//
//

#include <memory>
#include <iostream>
#include <iomanip>
#include <cmath>


#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"

#include "FWCore/PluginManager/interface/ModuleDef.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "DataFormats/JetReco/interface/CaloJetCollection.h"
#include "DataFormats/JetReco/interface/CaloJet.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "JetMETCorrections/JetVertexAssociation/interface/JetVertexAssociation.h"
#include "JetMETCorrections/JetVertexAssociation/interface/JetVertexMain.h"

using namespace std;
using namespace reco;
namespace cms{

  JetVertexAssociation::JetVertexAssociation(const edm::ParameterSet& iConfig): m_algo(iConfig),
                                                                                jet_algo(iConfig.getParameter<std::string>("JET_ALGO")),
                                                                                track_algo(iConfig.getParameter<std::string>("TRACK_ALGO")),
                                                                                vertex_algo(iConfig.getParameter<std::string>("VERTEX_ALGO")) {



    produces<ResultCollection1>("Var");
    produces<ResultCollection2>("JetType");

    
  }

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

   edm::Handle<CaloJetCollection> jets;  
   iEvent.getByLabel(jet_algo, jets);  
    
   edm::Handle<TrackCollection> tracks;
   iEvent.getByLabel(track_algo, tracks);
  
   edm::Handle<VertexCollection> vertexes;
   iEvent.getByLabel(vertex_algo, vertexes); 
 
   double SIGNAL_V_Z = 0.;
   double SIGNAL_V_Z_ERROR = 0.;
   double ptmax = -100.;

   VertexCollection::const_iterator vert = vertexes->begin ();
   if(vertexes->size() > 0 )   { 
        for (; vert != vertexes->end (); vert++) {

                SIGNAL_V_Z = vert->z();
                double pt = 0.;
                reco::Vertex::trackRef_iterator tr = vert->tracks_begin();
                for (; tr != vert->tracks_end(); tr++)  pt += (*tr)->pt();
                if( pt >= ptmax ){
 
                      ptmax = pt;
                  SIGNAL_V_Z = vert->z();
                          SIGNAL_V_Z_ERROR = vert->zError();
    
        }
         
    }
   }

   pair<double, bool> result;
   std::auto_ptr<ResultCollection1> result1 (new ResultCollection1) ; 
   std::auto_ptr<ResultCollection2> result2 (new ResultCollection2) ; 

   CaloJetCollection::const_iterator jet = jets->begin ();
  
   if(jets->size() > 0 )   { 
        for (; jet != jets->end (); jet++) {
         result = m_algo.Main(*jet, tracks, SIGNAL_V_Z, SIGNAL_V_Z_ERROR);
             result1->push_back(result.first);
             result2->push_back(result.second);
       
    }
   }

   iEvent.put(result1, "Var");
   iEvent.put(result2, "JetType");

  }
}