|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
#ifndef RecoAlgos_SortCollectionSelector_h
#define RecoAlgos_SortCollectionSelector_h
/** \class SortCollectionSelector
*
* selects the first N elements based on a sorting algorithm
*
* \author Luca Lista, INFN
*
* \version $Revision: 1.1 $
*
* $Id: SortCollectionSelector.h,v 1.1 2009/03/03 13:07:28 llista Exp $
*
*/
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "CommonTools/UtilAlgos/interface/SelectionAdderTrait.h"
#include "CommonTools/UtilAlgos/interface/StoreContainerTrait.h"
#include "CommonTools/UtilAlgos/interface/ParameterAdapter.h"
#include "CommonTools/UtilAlgos/interface/SelectedOutputCollectionTrait.h"
#include <algorithm>
#include <utility>
namespace edm {
class Event;
class EventSetup;
} // namespace edm
template <typename InputCollection,
typename Comparator,
typename OutputCollection = typename helper::SelectedOutputCollectionTrait<InputCollection>::type,
typename StoreContainer = typename helper::StoreContainerTrait<OutputCollection>::type,
typename RefAdder = typename helper::SelectionAdderTrait<InputCollection, StoreContainer>::type>
class SortCollectionSelector {
public:
typedef InputCollection collection;
private:
typedef const typename InputCollection::value_type *reference;
typedef std::pair<reference, size_t> pair;
typedef StoreContainer container;
typedef typename container::const_iterator const_iterator;
public:
SortCollectionSelector(const edm::ParameterSet &cfg, edm::ConsumesCollector &&iC)
: compare_(Comparator()), maxNumber_(cfg.template getParameter<unsigned int>("maxNumber")) {}
const_iterator begin() const { return selected_.begin(); }
const_iterator end() const { return selected_.end(); }
void select(const edm::Handle<InputCollection> &c, const edm::Event &, const edm::EventSetup &) {
std::vector<pair> v;
for (size_t idx = 0; idx < c->size(); ++idx)
v.push_back(std::make_pair(&(*c)[idx], idx));
std::sort(v.begin(), v.end(), compare_);
selected_.clear();
for (size_t i = 0; i < maxNumber_ && i < v.size(); ++i)
addRef_(selected_, c, v[i].second);
}
static void fillPSetDescription(edm::ParameterSetDescription &desc) { desc.add<unsigned int>("maxNumber", 1); }
private:
struct PairComparator {
PairComparator(const Comparator &cmp) : cmp_(cmp) {}
bool operator()(const pair &t1, const pair &t2) const { return cmp_(*t1.first, *t2.first); }
Comparator cmp_;
};
PairComparator compare_;
unsigned int maxNumber_;
StoreContainer selected_;
RefAdder addRef_;
};
#endif
|