|
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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
|
#ifndef FWCore_Framework_stream_ProducingModuleAdaptorBase_h
#define FWCore_Framework_stream_ProducingModuleAdaptorBase_h
// -*- C++ -*-
//
// Package: FWCore/Framework
// Class : ProducingModuleAdaptorBase
//
/**\class edm::stream::ProducingModuleAdaptorBase ProducingModuleAdaptorBase.h "FWCore/Framework/interface/stream/ProducingModuleAdaptorBase.h"
Description: [one line class summary]
Usage:
<usage>
*/
//
// Original Author: Chris Jones
// Created: Fri, 02 Aug 2013 18:09:15 GMT
//
// system include files
#include <array>
#include <map>
#include <string>
#include <vector>
#include <unordered_map>
// user include files
#include "FWCore/Utilities/interface/ProductResolverIndex.h"
#include "FWCore/Common/interface/FWCoreCommonFwd.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "DataFormats/Provenance/interface/ModuleDescription.h"
#include "FWCore/ParameterSet/interface/ParameterSetfwd.h"
#include "FWCore/Utilities/interface/BranchType.h"
#include "FWCore/Utilities/interface/StreamID.h"
#include "FWCore/Utilities/interface/RunIndex.h"
#include "FWCore/Utilities/interface/LuminosityBlockIndex.h"
#include "FWCore/Utilities/interface/ESIndices.h"
#include "FWCore/Utilities/interface/Transition.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ProcessBlock.h"
#include "FWCore/Framework/interface/Run.h"
#include "FWCore/Framework/interface/LuminosityBlock.h"
#include "FWCore/ServiceRegistry/interface/ServiceRegistryfwd.h"
// forward declarations
namespace edm {
class Event;
class ModuleCallingContext;
class ProductResolverIndexHelper;
class EDConsumerBase;
class PreallocationConfiguration;
class ProductResolverIndexAndSkipBit;
class ThinnedAssociationsHelper;
class ActivityRegistry;
class WaitingTaskHolder;
class ServiceWeakToken;
class SignallingProductRegistryFiller;
namespace maker {
template <typename T>
class ModuleHolderT;
}
namespace eventsetup {
struct ComponentDescription;
class ESRecordsToProductResolverIndices;
} // namespace eventsetup
namespace stream {
template <typename T>
class ProducingModuleAdaptorBase {
public:
template <typename U>
friend class edm::WorkerT;
template <typename U>
friend class edm::maker::ModuleHolderT;
ProducingModuleAdaptorBase();
ProducingModuleAdaptorBase(const ProducingModuleAdaptorBase&) = delete; // stop default
const ProducingModuleAdaptorBase& operator=(const ProducingModuleAdaptorBase&) = delete; // stop default
virtual ~ProducingModuleAdaptorBase();
// ---------- const member functions ---------------------
// ---------- static member functions --------------------
// ---------- member functions ---------------------------
const ModuleDescription& moduleDescription() const noexcept { return moduleDescription_; }
virtual bool wantsProcessBlocks() const noexcept = 0;
virtual bool wantsInputProcessBlocks() const noexcept = 0;
virtual bool wantsGlobalRuns() const noexcept = 0;
virtual bool wantsGlobalLuminosityBlocks() const noexcept = 0;
virtual bool hasAcquire() const noexcept = 0;
virtual bool hasAccumulator() const noexcept = 0;
virtual bool wantsStreamRuns() const noexcept = 0;
virtual bool wantsStreamLuminosityBlocks() const noexcept = 0;
void registerProductsAndCallbacks(ProducingModuleAdaptorBase const*, SignallingProductRegistryFiller* reg);
void itemsToGet(BranchType, std::vector<ProductResolverIndexAndSkipBit>&) const;
void itemsMayGet(BranchType, std::vector<ProductResolverIndexAndSkipBit>&) const;
std::vector<ProductResolverIndexAndSkipBit> const& itemsToGetFrom(BranchType) const;
std::vector<ESResolverIndex> const& esGetTokenIndicesVector(edm::Transition iTrans) const;
std::vector<ESRecordIndex> const& esGetTokenRecordIndicesVector(edm::Transition iTrans) const;
void updateLookup(BranchType iBranchType, ProductResolverIndexHelper const&, bool iPrefetchMayGet);
void updateLookup(eventsetup::ESRecordsToProductResolverIndices const&);
void releaseMemoryPostLookupSignal();
virtual void selectInputProcessBlocks(ProductRegistry const&, ProcessBlockHelperBase const&) = 0;
void modulesWhoseProductsAreConsumed(std::array<std::vector<ModuleDescription const*>*, NumBranchTypes>& modules,
ProductRegistry const& preg,
std::map<std::string, ModuleDescription const*> const& labelsToDesc,
std::string const& processName) const;
void esModulesWhoseProductsAreConsumed(
std::array<std::vector<eventsetup::ComponentDescription const*>*, kNumberOfEventSetupTransitions>& esModules,
eventsetup::ESRecordsToProductResolverIndices const&) const;
void convertCurrentProcessAlias(std::string const& processName);
std::vector<ModuleConsumesInfo> moduleConsumesInfos() const;
std::vector<ModuleConsumesESInfo> moduleConsumesESInfos(
eventsetup::ESRecordsToProductResolverIndices const&) const;
using ModuleToResolverIndicies =
std::unordered_multimap<std::string, std::tuple<edm::TypeID const*, const char*, edm::ProductResolverIndex>>;
void resolvePutIndicies(BranchType iBranchType,
ModuleToResolverIndicies const& iIndicies,
std::string const& moduleLabel);
std::vector<edm::ProductResolverIndex> const& indiciesForPutProducts(BranchType iBranchType) const;
ProductResolverIndex transformPrefetch_(size_t iTransformIndex) const noexcept;
size_t transformIndex_(edm::ProductDescription const& iBranch) const noexcept;
void doTransformAsync(WaitingTaskHolder iTask,
size_t iTransformIndex,
EventPrincipal const& iEvent,
ActivityRegistry*,
ModuleCallingContext,
ServiceWeakToken const&) noexcept;
protected:
template <typename F>
void createStreamModules(F iFunc) {
unsigned int iStreamModule = 0;
for (auto& m : m_streamModules) {
m = iFunc(iStreamModule);
m->setModuleDescriptionPtr(&moduleDescription_);
++iStreamModule;
}
}
void commit(ProcessBlock& iProcessBlock) {
iProcessBlock.commit_(m_streamModules[0]->indiciesForPutProducts(InProcess));
}
void commit(Run& iRun) { iRun.commit_(m_streamModules[0]->indiciesForPutProducts(InRun)); }
void commit(LuminosityBlock& iLumi) { iLumi.commit_(m_streamModules[0]->indiciesForPutProducts(InLumi)); }
template <typename I>
void commit(Event& iEvent, I* iID) {
iEvent.commit_(m_streamModules[0]->indiciesForPutProducts(InEvent), iID);
}
const EDConsumerBase* consumer() { return m_streamModules[0]; }
const ProducerBase* producer() { return m_streamModules[0]; }
void deleteModulesEarly();
private:
void doPreallocate(PreallocationConfiguration const&);
virtual void preallocRuns(unsigned int) {}
virtual void preallocLumis(unsigned int) {}
virtual void setupStreamModules() = 0;
virtual void doBeginJob() = 0;
virtual void doEndJob() = 0;
void doBeginStream(StreamID);
void doEndStream(StreamID);
void doStreamBeginRun(StreamID, RunTransitionInfo const&, ModuleCallingContext const*);
virtual void setupRun(T*, RunIndex) = 0;
void doStreamEndRun(StreamID, RunTransitionInfo const&, ModuleCallingContext const*);
virtual void streamEndRunSummary(T*, edm::Run const&, edm::EventSetup const&) = 0;
void doStreamBeginLuminosityBlock(StreamID, LumiTransitionInfo const&, ModuleCallingContext const*);
virtual void setupLuminosityBlock(T*, LuminosityBlockIndex) = 0;
void doStreamEndLuminosityBlock(StreamID, LumiTransitionInfo const&, ModuleCallingContext const*);
virtual void streamEndLuminosityBlockSummary(T*, edm::LuminosityBlock const&, edm::EventSetup const&) = 0;
virtual void doBeginProcessBlock(ProcessBlockPrincipal const&, ModuleCallingContext const*) = 0;
virtual void doAccessInputProcessBlock(ProcessBlockPrincipal const&, ModuleCallingContext const*) = 0;
virtual void doEndProcessBlock(ProcessBlockPrincipal const&, ModuleCallingContext const*) = 0;
virtual void doBeginRun(RunTransitionInfo const&, ModuleCallingContext const*) = 0;
virtual void doEndRun(RunTransitionInfo const&, ModuleCallingContext const*) = 0;
virtual void doBeginLuminosityBlock(LumiTransitionInfo const&, ModuleCallingContext const*) = 0;
virtual void doEndLuminosityBlock(LumiTransitionInfo const&, ModuleCallingContext const*) = 0;
void doRespondToOpenInputFile(FileBlock const&) {}
void doRespondToCloseInputFile(FileBlock const&) {}
virtual void doRespondToCloseOutputFile() = 0;
void doRegisterThinnedAssociations(ProductRegistry const&, ThinnedAssociationsHelper&);
// ---------- member data --------------------------------
void setModuleDescription(ModuleDescription const& md) { moduleDescription_ = md; }
ModuleDescription moduleDescription_;
protected:
std::vector<T*> m_streamModules;
};
} // namespace stream
} // namespace edm
#endif
|