ValueMapTraslator.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    ValueMapTraslator
// Class:      ValueMapTraslator
//
//
// Original Author:  Shervin Nourbakhsh,32 4-C03,+41227672087,
//         Created:  Sat Jul 13 15:40:56 CEST 2013
// $Id$
//
//

// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

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

#include "DataFormats/Common/interface/ValueMap.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectronFwd.h"

//#define DEBUG
//
// class declaration
//

class ValueMapTraslator : public edm::stream::EDProducer<> {
  typedef double value_t;
  typedef edm::ValueMap<value_t> Map_t;

public:
  explicit ValueMapTraslator(const edm::ParameterSet&);
  ~ValueMapTraslator() override;

  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

private:
  void produce(edm::Event&, const edm::EventSetup&) override;

  // ----------member data ---------------------------
  edm::InputTag referenceCollectionTAG, oldreferenceCollectionTAG;
  edm::InputTag inputCollectionTAG;
  std::string outputCollectionName;

  edm::EDGetTokenT<reco::GsfElectronCollection> referenceToken_;
  edm::EDGetTokenT<reco::GsfElectronCollection> oldreferenceToken_;
  edm::EDGetTokenT<Map_t> inputToken_;
};

//
// constants, enums and typedefs
//

//
// static data member definitions
//

//
// constructors and destructor
//
ValueMapTraslator::ValueMapTraslator(const edm::ParameterSet& iConfig)
    : referenceCollectionTAG(iConfig.getParameter<edm::InputTag>("referenceCollection")),
      oldreferenceCollectionTAG(iConfig.getParameter<edm::InputTag>("oldreferenceCollection")),
      inputCollectionTAG(iConfig.getParameter<edm::InputTag>("inputCollection")),
      outputCollectionName(iConfig.getParameter<std::string>("outputCollection")) {
  //now do what ever other initialization is needed
  referenceToken_ = consumes<reco::GsfElectronCollection>(referenceCollectionTAG);
  oldreferenceToken_ = consumes<reco::GsfElectronCollection>(oldreferenceCollectionTAG);
  inputToken_ = consumes<Map_t>(inputCollectionTAG);
  produces<Map_t>(outputCollectionName);
}

ValueMapTraslator::~ValueMapTraslator() {
  // do anything here that needs to be done at desctruction time
  // (e.g. close files, deallocate resources etc.)
}

//
// member functions
//

// ------------ method called to produce the data  ------------
void ValueMapTraslator::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  using namespace edm;
  std::vector<value_t> valueVector;
  auto valueVectorPtr = std::make_unique<Map_t>();

  //------------------------------
  Handle<reco::GsfElectronCollection> referenceHandle;
  Handle<reco::GsfElectronCollection> oldreferenceHandle;
  Handle<Map_t> inputHandle;

  iEvent.getByToken(referenceToken_, referenceHandle);
  iEvent.getByToken(oldreferenceToken_, oldreferenceHandle);
  iEvent.getByToken(inputToken_, inputHandle);

  for (Map_t::const_iterator valueMap_itr = inputHandle->begin(); valueMap_itr != inputHandle->end(); valueMap_itr++) {
    for (unsigned int i = 0; i < valueMap_itr.size(); i++) {
#ifdef DEBUG
      std::cout << valueMap_itr[i] << std::endl;
#endif
      //       valueVector.push_back((valueMap_itr[i])); //valueMap_itr-inputHandle->begin()]));
    }
    break;
  }

#ifdef DEBUG
  std::cout << "Size: " << referenceHandle->size() << "\t" << oldreferenceHandle->size() << "\t" << inputHandle->size()
            << "\t" << valueVector.size() << std::endl;
#endif
  for (reco::GsfElectronCollection::const_iterator electronNew = referenceHandle->begin();
       electronNew != referenceHandle->end();
       electronNew++) {
    for (reco::GsfElectronCollection::const_iterator electron = oldreferenceHandle->begin();
         electron != oldreferenceHandle->end();
         electron++) {
      //if(electronNew->GsfTrackF
      if (electron->gsfTrack() != electronNew->gsfTrack())
        continue;  
      //if(fabs(electronNew->eta() - electron->eta())>0.0001) continue;
      //if(fabs(electronNew->phi() - electron->phi())>0.0001) continue;

      reco::GsfElectronRef eleRef(oldreferenceHandle, electron - oldreferenceHandle->begin());
      reco::GsfElectronRef eleRef2(referenceHandle, electronNew - referenceHandle->begin());

#ifdef DEBUG
      std::cout << eleRef->eta() << "\t" << eleRef2->eta() << "\t" << eleRef->phi() << "\t" << eleRef2->phi() << "\t"
                << eleRef->energy() << "\t" << eleRef2->energy() << "\t" << (eleRef->gsfTrack() == eleRef2->gsfTrack())
                << "\t" << (eleRef == eleRef2) << "\t" << (*inputHandle)[eleRef] << std::endl;
#endif
      valueVector.push_back((*inputHandle)[eleRef]);  //valueMap_itr-inputHandle->begin()]));
    }
  }
  //#endif
  Map_t::Filler filler(*valueVectorPtr);
  filler.insert(referenceHandle, valueVector.begin(), valueVector.end());
  filler.fill();

  iEvent.put(std::move(valueVectorPtr));
}

// ------------ method fills 'descriptions' with the allowed parameters for the module  ------------
void ValueMapTraslator::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(ValueMapTraslator);