|
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
/** \file
* Implementation of class RawDataMapperByLabel
*
*/
#include "DataFormats/Provenance/interface/ProcessHistory.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/FEDRawData/interface/FEDRawDataCollection.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include <iostream>
using namespace edm;
class RawDataMapperByLabel : public edm::stream::EDProducer<> {
public:
///Constructor
RawDataMapperByLabel(const edm::ParameterSet& pset);
///Destructor
~RawDataMapperByLabel() override;
void produce(edm::Event& e, const edm::EventSetup& c) override;
static void fillDescriptions(edm::ConfigurationDescriptions&);
private:
typedef std::vector<edm::InputTag>::const_iterator tag_iterator_t;
typedef std::vector<edm::EDGetTokenT<FEDRawDataCollection>>::const_iterator tok_iterator_t;
std::vector<edm::InputTag> inputTags_;
std::vector<edm::EDGetTokenT<FEDRawDataCollection>> inputTokens_;
edm::InputTag mainCollectionTag_;
edm::InputTag filledCollectionName_;
bool firstEvent_;
};
RawDataMapperByLabel::RawDataMapperByLabel(const edm::ParameterSet& pset)
: inputTags_(pset.getParameter<std::vector<edm::InputTag>>("rawCollectionList")),
mainCollectionTag_(pset.getParameter<edm::InputTag>("mainCollection")),
filledCollectionName_(edm::InputTag("")),
firstEvent_(true) {
inputTokens_.reserve(inputTags_.size());
for (auto const& inputTag : inputTags_) {
inputTokens_.push_back(consumes<FEDRawDataCollection>(inputTag));
}
produces<FEDRawDataCollection>();
}
RawDataMapperByLabel::~RawDataMapperByLabel() {}
void RawDataMapperByLabel::produce(Event& e, const EventSetup& c) {
bool alreadyACollectionFilled = false;
tag_iterator_t inputTag = inputTags_.begin();
for (tok_iterator_t inputTok = inputTokens_.begin(); inputTok != inputTokens_.end(); ++inputTok, ++inputTag) {
Handle<FEDRawDataCollection> input;
if (e.getByToken(*inputTok, input)) {
if (input.isValid()) {
if (alreadyACollectionFilled)
throw cms::Exception("BadInput")
<< "Two input collections are present." << std::endl
<< "Please make sure that the input dataset has only one FEDRawDataCollector collection filled";
if (firstEvent_) {
filledCollectionName_ = *inputTag;
alreadyACollectionFilled = true;
firstEvent_ = false;
}
if (!(filledCollectionName_ == *inputTag))
throw cms::Exception("BadInput") << "The filled collection has changed!";
if (!(mainCollectionTag_ == filledCollectionName_))
e.put(std::make_unique<FEDRawDataCollection>(*input.product()));
}
}
}
}
void RawDataMapperByLabel::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
edm::ParameterSetDescription desc;
desc.add<std::vector<edm::InputTag>>("rawCollectionList",
{{"rawDataCollector", "", "@skipCurrentProcess"},
{"rawDataRepacker"},
{"rawPrimeDataRepacker"},
{"rawDataReducedFormat"}});
desc.add<edm::InputTag>("mainCollection", edm::InputTag("rawDataCollector"));
descriptions.add("rawDataMapperByLabel", desc);
}
DEFINE_FWK_MODULE(RawDataMapperByLabel);
|