Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2023-12-05 03:11:55

0001 // -*- C++ -*-
0002 //
0003 // Package:     FWCore/Framework
0004 // Class  :     global::EDFilterBase
0005 //
0006 // Implementation:
0007 //     [Notes on implementation]
0008 //
0009 // Original Author:  Chris Jones
0010 //         Created:  Thu, 02 May 2013 21:56:04 GMT
0011 //
0012 
0013 // system include files
0014 
0015 // user include files
0016 #include <memory>
0017 
0018 #include "FWCore/Framework/interface/Event.h"
0019 #include "FWCore/Framework/interface/EventSetup.h"
0020 #include "FWCore/Framework/interface/LuminosityBlock.h"
0021 #include "FWCore/Framework/interface/ProcessBlock.h"
0022 #include "FWCore/Framework/interface/Run.h"
0023 #include "FWCore/Framework/interface/global/EDFilterBase.h"
0024 #include "FWCore/Framework/src/EventAcquireSignalsSentry.h"
0025 #include "FWCore/Framework/src/EventSignalsSentry.h"
0026 #include "FWCore/Framework/interface/PreallocationConfiguration.h"
0027 #include "FWCore/Framework/src/edmodule_mightGet_config.h"
0028 #include "FWCore/Framework/interface/TransitionInfoTypes.h"
0029 #include "FWCore/Framework/interface/EventForTransformer.h"
0030 #include "FWCore/ServiceRegistry/interface/ESParentContext.h"
0031 
0032 #include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
0033 #include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
0034 
0035 //
0036 // constants, enums and typedefs
0037 //
0038 namespace edm {
0039 
0040   class WaitingTaskWithArenaHolder;
0041 
0042   namespace global {
0043     //
0044     // static data member definitions
0045     //
0046 
0047     //
0048     // constructors and destructor
0049     //
0050     EDFilterBase::EDFilterBase()
0051         : ProducerBase(), moduleDescription_(), previousParentages_(), previousParentageIds_() {}
0052 
0053     EDFilterBase::~EDFilterBase() {}
0054 
0055     bool EDFilterBase::doEvent(EventTransitionInfo const& info,
0056                                ActivityRegistry* act,
0057                                ModuleCallingContext const* mcc) {
0058       Event e(info, moduleDescription_, mcc);
0059       e.setConsumer(this);
0060       const auto streamIndex = e.streamID().value();
0061       e.setProducer(
0062           this, &previousParentages_[streamIndex], hasAcquire() ? &gotBranchIDsFromAcquire_[streamIndex] : nullptr);
0063       EventSignalsSentry sentry(act, mcc);
0064       ESParentContext parentC(mcc);
0065       const EventSetup c{
0066           info, static_cast<unsigned int>(Transition::Event), esGetTokenIndices(Transition::Event), parentC};
0067       bool returnValue = this->filter(e.streamID(), e, c);
0068       commit_(e, &previousParentageIds_[streamIndex]);
0069       return returnValue;
0070     }
0071 
0072     void EDFilterBase::doAcquire(EventTransitionInfo const& info,
0073                                  ActivityRegistry* act,
0074                                  ModuleCallingContext const* mcc,
0075                                  WaitingTaskWithArenaHolder& holder) {
0076       Event e(info, moduleDescription_, mcc);
0077       e.setConsumer(this);
0078       const auto streamIndex = e.streamID().value();
0079       e.setProducerForAcquire(this, nullptr, gotBranchIDsFromAcquire_[streamIndex]);
0080       EventAcquireSignalsSentry sentry(act, mcc);
0081       ESParentContext parentC(mcc);
0082       const EventSetup c{
0083           info, static_cast<unsigned int>(Transition::Event), esGetTokenIndices(Transition::Event), parentC};
0084       this->doAcquire_(e.streamID(), e, c, holder);
0085     }
0086 
0087     void EDFilterBase::doTransformAsync(WaitingTaskHolder iTask,
0088                                         size_t iTransformIndex,
0089                                         EventPrincipal const& iEvent,
0090                                         ActivityRegistry* iAct,
0091                                         ModuleCallingContext iMCC,
0092                                         ServiceWeakToken const& iToken) {
0093       EventForTransformer ev(iEvent, iMCC);
0094       transformAsync_(iTask, iTransformIndex, ev, iAct, iToken);
0095     }
0096 
0097     size_t EDFilterBase::transformIndex_(edm::BranchDescription const& iBranch) const { return -1; }
0098     ProductResolverIndex EDFilterBase::transformPrefetch_(std::size_t iIndex) const { return 0; }
0099     void EDFilterBase::transformAsync_(WaitingTaskHolder iTask,
0100                                        std::size_t iIndex,
0101                                        edm::EventForTransformer& iEvent,
0102                                        edm::ActivityRegistry* iAct,
0103                                        ServiceWeakToken const& iToken) const {}
0104 
0105     void EDFilterBase::doPreallocate(PreallocationConfiguration const& iPrealloc) {
0106       const auto nStreams = iPrealloc.numberOfStreams();
0107       previousParentages_ = std::make_unique<std::vector<BranchID>[]>(nStreams);
0108       if (hasAcquire()) {
0109         gotBranchIDsFromAcquire_ = std::make_unique<std::vector<BranchID>[]>(nStreams);
0110       }
0111       previousParentageIds_ = std::make_unique<ParentageID[]>(nStreams);
0112       preallocStreams(nStreams);
0113       preallocRuns(iPrealloc.numberOfRuns());
0114       preallocRunsSummary(iPrealloc.numberOfRuns());
0115       preallocLumis(iPrealloc.numberOfLuminosityBlocks());
0116       preallocLumisSummary(iPrealloc.numberOfLuminosityBlocks());
0117       preallocate(iPrealloc);
0118     }
0119 
0120     void EDFilterBase::doBeginJob() { this->beginJob(); }
0121 
0122     void EDFilterBase::doEndJob() { this->endJob(); }
0123 
0124     void EDFilterBase::doBeginProcessBlock(ProcessBlockPrincipal const& pbp, ModuleCallingContext const* mcc) {
0125       ProcessBlock processBlock(pbp, moduleDescription_, mcc, false);
0126       processBlock.setConsumer(this);
0127       ProcessBlock const& constProcessBlock = processBlock;
0128       this->doBeginProcessBlock_(constProcessBlock);
0129       processBlock.setProducer(this);
0130       this->doBeginProcessBlockProduce_(processBlock);
0131       commit_(processBlock);
0132     }
0133 
0134     void EDFilterBase::doAccessInputProcessBlock(ProcessBlockPrincipal const& pbp, ModuleCallingContext const* mcc) {
0135       ProcessBlock processBlock(pbp, moduleDescription_, mcc, false);
0136       processBlock.setConsumer(this);
0137       ProcessBlock const& constProcessBlock = processBlock;
0138       this->doAccessInputProcessBlock_(constProcessBlock);
0139     }
0140 
0141     void EDFilterBase::doEndProcessBlock(ProcessBlockPrincipal const& pbp, ModuleCallingContext const* mcc) {
0142       ProcessBlock processBlock(pbp, moduleDescription_, mcc, true);
0143       processBlock.setConsumer(this);
0144       ProcessBlock const& constProcessBlock = processBlock;
0145       this->doEndProcessBlock_(constProcessBlock);
0146       processBlock.setProducer(this);
0147       this->doEndProcessBlockProduce_(processBlock);
0148       commit_(processBlock);
0149     }
0150 
0151     void EDFilterBase::doBeginRun(RunTransitionInfo const& info, ModuleCallingContext const* mcc) {
0152       Run r(info, moduleDescription_, mcc, false);
0153       r.setConsumer(this);
0154       Run const& cnstR = r;
0155       ESParentContext parentC(mcc);
0156       const EventSetup c{
0157           info, static_cast<unsigned int>(Transition::BeginRun), esGetTokenIndices(Transition::BeginRun), parentC};
0158       this->doBeginRun_(cnstR, c);
0159       this->doBeginRunSummary_(cnstR, c);
0160       r.setProducer(this);
0161       this->doBeginRunProduce_(r, c);
0162       commit_(r);
0163     }
0164 
0165     void EDFilterBase::doEndRun(RunTransitionInfo const& info, ModuleCallingContext const* mcc) {
0166       Run r(info, moduleDescription_, mcc, true);
0167       r.setConsumer(this);
0168       r.setProducer(this);
0169       Run const& cnstR = r;
0170       ESParentContext parentC(mcc);
0171       const EventSetup c{
0172           info, static_cast<unsigned int>(Transition::EndRun), esGetTokenIndices(Transition::EndRun), parentC};
0173       this->doEndRunSummary_(r, c);
0174       this->doEndRunProduce_(r, c);
0175       this->doEndRun_(cnstR, c);
0176       commit_(r);
0177     }
0178 
0179     void EDFilterBase::doBeginLuminosityBlock(LumiTransitionInfo const& info, ModuleCallingContext const* mcc) {
0180       LuminosityBlock lb(info, moduleDescription_, mcc, false);
0181       lb.setConsumer(this);
0182       LuminosityBlock const& cnstLb = lb;
0183       ESParentContext parentC(mcc);
0184       const EventSetup c{info,
0185                          static_cast<unsigned int>(Transition::BeginLuminosityBlock),
0186                          esGetTokenIndices(Transition::BeginLuminosityBlock),
0187                          parentC};
0188       this->doBeginLuminosityBlock_(cnstLb, c);
0189       this->doBeginLuminosityBlockSummary_(cnstLb, c);
0190       lb.setProducer(this);
0191       this->doBeginLuminosityBlockProduce_(lb, c);
0192       commit_(lb);
0193     }
0194 
0195     void EDFilterBase::doEndLuminosityBlock(LumiTransitionInfo const& info, ModuleCallingContext const* mcc) {
0196       LuminosityBlock lb(info, moduleDescription_, mcc, true);
0197       lb.setConsumer(this);
0198       lb.setProducer(this);
0199       LuminosityBlock const& cnstLb = lb;
0200       ESParentContext parentC(mcc);
0201       const EventSetup c{info,
0202                          static_cast<unsigned int>(Transition::EndLuminosityBlock),
0203                          esGetTokenIndices(Transition::EndLuminosityBlock),
0204                          parentC};
0205       this->doEndLuminosityBlockSummary_(cnstLb, c);
0206       this->doEndLuminosityBlockProduce_(lb, c);
0207       this->doEndLuminosityBlock_(cnstLb, c);
0208       commit_(lb);
0209     }
0210 
0211     void EDFilterBase::doBeginStream(StreamID id) { doBeginStream_(id); }
0212     void EDFilterBase::doEndStream(StreamID id) { doEndStream_(id); }
0213     void EDFilterBase::doStreamBeginRun(StreamID id, RunTransitionInfo const& info, ModuleCallingContext const* mcc) {
0214       Run r(info, moduleDescription_, mcc, false);
0215       r.setConsumer(this);
0216       ESParentContext parentC(mcc);
0217       const EventSetup c{
0218           info, static_cast<unsigned int>(Transition::BeginRun), esGetTokenIndices(Transition::BeginRun), parentC};
0219       this->doStreamBeginRun_(id, r, c);
0220     }
0221     void EDFilterBase::doStreamEndRun(StreamID id, RunTransitionInfo const& info, ModuleCallingContext const* mcc) {
0222       Run r(info, moduleDescription_, mcc, true);
0223       r.setConsumer(this);
0224       ESParentContext parentC(mcc);
0225       const EventSetup c{
0226           info, static_cast<unsigned int>(Transition::EndRun), esGetTokenIndices(Transition::EndRun), parentC};
0227       this->doStreamEndRun_(id, r, c);
0228       this->doStreamEndRunSummary_(id, r, c);
0229     }
0230     void EDFilterBase::doStreamBeginLuminosityBlock(StreamID id,
0231                                                     LumiTransitionInfo const& info,
0232                                                     ModuleCallingContext const* mcc) {
0233       LuminosityBlock lb(info, moduleDescription_, mcc, false);
0234       lb.setConsumer(this);
0235       ESParentContext parentC(mcc);
0236       const EventSetup c{info,
0237                          static_cast<unsigned int>(Transition::BeginLuminosityBlock),
0238                          esGetTokenIndices(Transition::BeginLuminosityBlock),
0239                          parentC};
0240       this->doStreamBeginLuminosityBlock_(id, lb, c);
0241     }
0242 
0243     void EDFilterBase::doStreamEndLuminosityBlock(StreamID id,
0244                                                   LumiTransitionInfo const& info,
0245                                                   ModuleCallingContext const* mcc) {
0246       LuminosityBlock lb(info, moduleDescription_, mcc, true);
0247       lb.setConsumer(this);
0248       ESParentContext parentC(mcc);
0249       const EventSetup c{info,
0250                          static_cast<unsigned int>(Transition::EndLuminosityBlock),
0251                          esGetTokenIndices(Transition::EndLuminosityBlock),
0252                          parentC};
0253       this->doStreamEndLuminosityBlock_(id, lb, c);
0254       this->doStreamEndLuminosityBlockSummary_(id, lb, c);
0255     }
0256 
0257     void EDFilterBase::preallocStreams(unsigned int) {}
0258     void EDFilterBase::preallocRuns(unsigned int) {}
0259     void EDFilterBase::preallocRunsSummary(unsigned int) {}
0260     void EDFilterBase::preallocLumis(unsigned int) {}
0261     void EDFilterBase::preallocLumisSummary(unsigned int) {}
0262     void EDFilterBase::preallocate(PreallocationConfiguration const&) {}
0263     void EDFilterBase::doBeginStream_(StreamID id) {}
0264     void EDFilterBase::doEndStream_(StreamID id) {}
0265     void EDFilterBase::doStreamBeginRun_(StreamID id, Run const& rp, EventSetup const& c) {}
0266     void EDFilterBase::doStreamEndRun_(StreamID id, Run const& rp, EventSetup const& c) {}
0267     void EDFilterBase::doStreamEndRunSummary_(StreamID id, Run const& rp, EventSetup const& c) {}
0268     void EDFilterBase::doStreamBeginLuminosityBlock_(StreamID id, LuminosityBlock const& lbp, EventSetup const& c) {}
0269     void EDFilterBase::doStreamEndLuminosityBlock_(StreamID id, LuminosityBlock const& lbp, EventSetup const& c) {}
0270     void EDFilterBase::doStreamEndLuminosityBlockSummary_(StreamID id,
0271                                                           LuminosityBlock const& lbp,
0272                                                           EventSetup const& c) {}
0273 
0274     void EDFilterBase::doBeginProcessBlock_(ProcessBlock const&) {}
0275     void EDFilterBase::doAccessInputProcessBlock_(ProcessBlock const&) {}
0276     void EDFilterBase::doEndProcessBlock_(ProcessBlock const&) {}
0277     void EDFilterBase::doBeginRun_(Run const& rp, EventSetup const& c) {}
0278     void EDFilterBase::doEndRun_(Run const& rp, EventSetup const& c) {}
0279     void EDFilterBase::doBeginRunSummary_(Run const& rp, EventSetup const& c) {}
0280     void EDFilterBase::doEndRunSummary_(Run const& rp, EventSetup const& c) {}
0281 
0282     void EDFilterBase::doBeginLuminosityBlock_(LuminosityBlock const& lbp, EventSetup const& c) {}
0283     void EDFilterBase::doEndLuminosityBlock_(LuminosityBlock const& lbp, EventSetup const& c) {}
0284     void EDFilterBase::doBeginLuminosityBlockSummary_(LuminosityBlock const& rp, EventSetup const& c) {}
0285     void EDFilterBase::doEndLuminosityBlockSummary_(LuminosityBlock const& lb, EventSetup const& c) {}
0286 
0287     void EDFilterBase::doBeginProcessBlockProduce_(ProcessBlock&) {}
0288     void EDFilterBase::doEndProcessBlockProduce_(ProcessBlock&) {}
0289     void EDFilterBase::doBeginRunProduce_(Run& rp, EventSetup const& c) {}
0290     void EDFilterBase::doEndRunProduce_(Run& rp, EventSetup const& c) {}
0291     void EDFilterBase::doBeginLuminosityBlockProduce_(LuminosityBlock& lbp, EventSetup const& c) {}
0292     void EDFilterBase::doEndLuminosityBlockProduce_(LuminosityBlock& lbp, EventSetup const& c) {}
0293 
0294     void EDFilterBase::clearInputProcessBlockCaches() {}
0295 
0296     void EDFilterBase::doAcquire_(StreamID, Event const&, EventSetup const&, WaitingTaskWithArenaHolder&) {}
0297 
0298     void EDFilterBase::fillDescriptions(ConfigurationDescriptions& descriptions) {
0299       ParameterSetDescription desc;
0300       desc.setUnknown();
0301       descriptions.addDefault(desc);
0302     }
0303 
0304     void EDFilterBase::prevalidate(ConfigurationDescriptions& iConfig) { edmodule_mightGet_config(iConfig); }
0305 
0306     static const std::string kBaseType("EDFilter");
0307 
0308     const std::string& EDFilterBase::baseType() { return kBaseType; }
0309 
0310   }  // namespace global
0311 }  // namespace edm