PFCand_AssoMap.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    PFCand_AssoMap
// Class:      PFCand_AssoMap
// 
//
// Original Author:  Matthias Geisler
//         Created:  Wed Apr 18 14:48:37 CEST 2012
// $Id: PFCand_AssoMap.cc,v 1.6 2012/12/06 14:02:13 mgeisler Exp $
//
//
#include "CommonTools/RecoUtils/interface/PFCand_AssoMap.h"

// system include files
#include <memory>
#include <vector>
#include <string>

// user include files

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/Common/interface/View.h"

//
// constructors and destructor
//
PFCand_AssoMap::PFCand_AssoMap(const edm::ParameterSet& iConfig):PFCand_AssoMapAlgos(iConfig)
{

   //now do what ever other initialization is needed

   input_AssociationType_ = iConfig.getParameter<edm::InputTag>("AssociationType");

   input_PFCandidates_ = iConfig.getParameter<edm::InputTag>("PFCandidateCollection");

   //register your products

   if ( input_AssociationType_.label() == "PFCandsToVertex" ) {
     produces<PFCandToVertexAssMap>();
   } else {
     if ( input_AssociationType_.label() == "VertexToPFCands" ) {
       produces<VertexToPFCandAssMap>();
     } else {
       if ( input_AssociationType_.label() == "Both" ) {
         produces<PFCandToVertexAssMap>();
     produces<VertexToPFCandAssMap>();
       } else {
         edm::LogWarning("Prtcl2VtxAssociation") << "No correct InputTag for AssociationType!" << std::endl
         << "Won't produce any AssociationMap!" << std::endl;
       }
     }
   }
  
}


PFCand_AssoMap::~PFCand_AssoMap()
{
}


//
// member functions
//

// ------------ method called to produce the data  ------------
void
PFCand_AssoMap::produce(edm::Event& iEvent, const edm::EventSetup& iSetup)
{

  using namespace edm;
  using namespace std;
  using namespace reco;
  
  //get the input pfCandidateCollection
  Handle<PFCandidateCollection> pfCandH;
  iEvent.getByLabel(input_PFCandidates_,pfCandH);
    
  string asstype = input_AssociationType_.label();

  PFCand_AssoMapAlgos::GetPFCandInputCollections(iEvent,iSetup);
    
  PFCand_AssoMapAlgos::SetMotherPointer(pfCandH);

  if ( ( asstype == "PFCandsToVertex" ) || ( asstype == "Both" ) ) {
    auto_ptr<PFCandToVertexAssMap> PFCand2Vertex = CreatePFCandToVertexMap(pfCandH, iSetup);
    iEvent.put( SortPFCandAssociationMap( &(*PFCand2Vertex) ) );
  }
 
  if ( ( asstype == "VertexToPFCands" ) || ( asstype == "Both" ) ) {
    auto_ptr<VertexToPFCandAssMap> Vertex2PFCand = CreateVertexToPFCandMap(pfCandH, iSetup);
    iEvent.put( Vertex2PFCand );
  }

}

// ------------ method fills 'descriptions' with the allowed parameters for the module  ------------
void
PFCand_AssoMap::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  //The following says we do not know what parameters are allowed so do no validation
  // Please change this to state exactly what you do use, even if it is no parameters
  edm::ParameterSetDescription desc;
  desc.setUnknown();
  descriptions.addDefault(desc);
}

//define this as a plug-in
DEFINE_FWK_MODULE(PFCand_AssoMap);