| 1 |
#ifndef RecoAlgos_GsfElectronSelector_h
|
| 2 |
#define RecoAlgos_GsfElectronSelector_h
|
| 3 |
/** \class GsfElectronSelector
|
| 4 |
*
|
| 5 |
* selects a subset of an electron collection. Also clones
|
| 6 |
* all referenced objects
|
| 7 |
*
|
| 8 |
* \author Luca Lista, INFN
|
| 9 |
*
|
| 10 |
* \version $Revision: 1.1 $
|
| 11 |
*
|
| 12 |
* $Id: GsfElectronSelector.h,v 1.1 2009/10/20 17:15:14 dgele Exp $
|
| 13 |
*
|
| 14 |
*/
|
| 15 |
|
| 16 |
#include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
|
| 17 |
#include "DataFormats/EgammaReco/interface/SuperCluster.h"
|
| 18 |
#include "DataFormats/GsfTrackReco/interface/GsfTrack.h"
|
| 19 |
#include "DataFormats/GsfTrackReco/interface/GsfTrackExtra.h"
|
| 20 |
#include "DataFormats/TrackingRecHit/interface/TrackingRecHit.h"
|
| 21 |
#include "PhysicsTools/UtilAlgos/interface/ObjectSelector.h"
|
| 22 |
|
| 23 |
namespace helper {
|
| 24 |
struct GsfElectronCollectionStoreManager {
|
| 25 |
typedef reco::GsfElectronCollection collection;
|
| 26 |
GsfElectronCollectionStoreManager(const edm::Handle<reco::GsfElectronCollection>&) :
|
| 27 |
selElectrons_( new reco::GsfElectronCollection ),
|
| 28 |
selSuperClusters_( new reco::SuperClusterCollection ),
|
| 29 |
selTracks_( new reco::GsfTrackCollection ),
|
| 30 |
selTrackExtras_( new reco::TrackExtraCollection ),
|
| 31 |
selGsfTrackExtras_( new reco::GsfTrackExtraCollection ),
|
| 32 |
selHits_( new TrackingRecHitCollection ) {
|
| 33 |
}
|
| 34 |
template<typename I>
|
| 35 |
void cloneAndStore( const I & begin, const I & end, edm::Event & evt ) {
|
| 36 |
using namespace reco;
|
| 37 |
TrackingRecHitRefProd rHits = evt.template getRefBeforePut<TrackingRecHitCollection>();
|
| 38 |
TrackExtraRefProd rTrackExtras = evt.template getRefBeforePut<TrackExtraCollection>();
|
| 39 |
GsfTrackExtraRefProd rGsfTrackExtras = evt.template getRefBeforePut<GsfTrackExtraCollection>();
|
| 40 |
GsfTrackRefProd rTracks = evt.template getRefBeforePut<GsfTrackCollection>();
|
| 41 |
GsfElectronRefProd rElectrons = evt.template getRefBeforePut<GsfElectronCollection>();
|
| 42 |
SuperClusterRefProd rSuperClusters = evt.template getRefBeforePut<SuperClusterCollection>();
|
| 43 |
size_t idx = 0, tidx = 0, hidx = 0;
|
| 44 |
for( I i = begin; i != end; ++ i ) {
|
| 45 |
const GsfElectron & ele = * * i;
|
| 46 |
selElectrons_->push_back( GsfElectron( ele ) );
|
| 47 |
selElectrons_->back().setGsfTrack( GsfTrackRef( rTracks, idx ) );
|
| 48 |
selElectrons_->back().setSuperCluster( SuperClusterRef( rSuperClusters, idx ++ ) );
|
| 49 |
selSuperClusters_->push_back( SuperCluster( * ( ele.superCluster() ) ) );
|
| 50 |
GsfTrackRef trkRef = ele.gsfTrack();
|
| 51 |
if ( trkRef.isNonnull() ) {
|
| 52 |
selTracks_->push_back( GsfTrack( * trkRef ) );
|
| 53 |
GsfTrack & trk = selTracks_->back();
|
| 54 |
selTrackExtras_->push_back( TrackExtra( trk.outerPosition(), trk.outerMomentum(), trk.outerOk(),
|
| 55 |
trk.innerPosition(), trk.innerMomentum(), trk.innerOk(),
|
| 56 |
trk.outerStateCovariance(), trk.outerDetId(),
|
| 57 |
trk.innerStateCovariance(), trk.innerDetId(),
|
| 58 |
trk.seedDirection() ) );
|
| 59 |
selGsfTrackExtras_->push_back( GsfTrackExtra( *(trk.gsfExtra()) ) );
|
| 60 |
TrackExtra & tx = selTrackExtras_->back();
|
| 61 |
for( trackingRecHit_iterator hit = trk.recHitsBegin(); hit != trk.recHitsEnd(); ++ hit ) {
|
| 62 |
selHits_->push_back( (*hit)->clone() );
|
| 63 |
tx.add( TrackingRecHitRef( rHits, hidx ++ ) );
|
| 64 |
}
|
| 65 |
trk.setGsfExtra( GsfTrackExtraRef( rGsfTrackExtras, tidx ) );
|
| 66 |
trk.setExtra( TrackExtraRef( rTrackExtras, tidx ++ ) );
|
| 67 |
}
|
| 68 |
}
|
| 69 |
}
|
| 70 |
|
| 71 |
edm::OrphanHandle<reco::GsfElectronCollection> put( edm::Event & evt ) {
|
| 72 |
edm::OrphanHandle<reco::GsfElectronCollection> h = evt.put( selElectrons_ );
|
| 73 |
evt.put( selSuperClusters_ );
|
| 74 |
evt.put( selTracks_ );
|
| 75 |
evt.put( selTrackExtras_ );
|
| 76 |
evt.put( selGsfTrackExtras_ );
|
| 77 |
evt.put( selHits_ );
|
| 78 |
return h;
|
| 79 |
}
|
| 80 |
|
| 81 |
size_t size() const { return selElectrons_->size(); }
|
| 82 |
private:
|
| 83 |
std::auto_ptr<reco::GsfElectronCollection> selElectrons_;
|
| 84 |
std::auto_ptr<reco::SuperClusterCollection> selSuperClusters_;
|
| 85 |
std::auto_ptr<reco::GsfTrackCollection> selTracks_;
|
| 86 |
std::auto_ptr<reco::TrackExtraCollection> selTrackExtras_;
|
| 87 |
std::auto_ptr<reco::GsfTrackExtraCollection> selGsfTrackExtras_;
|
| 88 |
std::auto_ptr<TrackingRecHitCollection> selHits_;
|
| 89 |
};
|
| 90 |
|
| 91 |
class GsfElectronSelectorBase : public edm::EDFilter {
|
| 92 |
public:
|
| 93 |
GsfElectronSelectorBase( const edm::ParameterSet & cfg ) {
|
| 94 |
std::string alias( cfg.getParameter<std::string>( "@module_label" ) );
|
| 95 |
produces<reco::GsfElectronCollection>().setBranchAlias( alias + "GsfElectrons" );
|
| 96 |
produces<reco::SuperClusterCollection>().setBranchAlias( alias + "SuperClusters" );
|
| 97 |
produces<reco::GsfTrackCollection>().setBranchAlias( alias + "GsfTracks" );
|
| 98 |
produces<reco::GsfTrackExtraCollection>().setBranchAlias( alias + "GsfTrackExtras" );
|
| 99 |
produces<reco::TrackExtraCollection>().setBranchAlias( alias + "TrackExtras" );
|
| 100 |
produces<TrackingRecHitCollection>().setBranchAlias( alias + "RecHits" );
|
| 101 |
}
|
| 102 |
};
|
| 103 |
|
| 104 |
template<>
|
| 105 |
struct StoreManagerTrait<reco::GsfElectronCollection> {
|
| 106 |
typedef GsfElectronCollectionStoreManager type;
|
| 107 |
typedef GsfElectronSelectorBase base;
|
| 108 |
};
|
| 109 |
}
|
| 110 |
|
| 111 |
#endif
|
