ProducingModuleAdaptorBase.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:     FWCore/Framework
// Class  :     edm::stream::ProducingModuleAdaptorBase
// 
// Implementation:
//     [Notes on implementation]
//
// Original Author:  Chris Jones
//         Created:  Fri, 02 Aug 2013 21:43:44 GMT
//

// system include files
#include <cassert>

// user include files
#include "FWCore/Framework/interface/stream/ProducingModuleAdaptorBase.h"
#include "FWCore/Framework/interface/LuminosityBlock.h"
#include "FWCore/Framework/interface/Run.h"
#include "FWCore/Framework/interface/LuminosityBlockPrincipal.h"
#include "FWCore/Framework/interface/RunPrincipal.h"
#include "FWCore/Framework/src/PreallocationConfiguration.h"

//
// constants, enums and typedefs
//

//
// static data member definitions
//

//
// constructors and destructor
//
namespace edm {
  namespace stream {
    template< typename T>
    ProducingModuleAdaptorBase<T>::ProducingModuleAdaptorBase()
    {
    }
    
    template< typename T>
    ProducingModuleAdaptorBase<T>::~ProducingModuleAdaptorBase()
    {
      for(auto m: m_streamModules) {
        delete m;
      }
    }
    
    //
    // member functions
    //
    
    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::doPreallocate(PreallocationConfiguration const& iPrealloc) {
      m_streamModules.resize(iPrealloc.numberOfStreams(),
                             static_cast<T*>(nullptr));
      setupStreamModules();
      preallocLumis(iPrealloc.numberOfLuminosityBlocks());
    }

    
    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::registerProductsAndCallbacks(ProducingModuleAdaptorBase const*, ProductRegistry* reg) {
      auto firstMod = m_streamModules[0];
      if(firstMod->registrationCallback() and m_streamModules.size()>1) {
        //we have a callback so we will collect all callbacks and create a new callback which calls them all.
        
        std::vector<std::function<void(BranchDescription const&)>> callbacks;
        callbacks.reserve(m_streamModules.size());
        
        for(auto mod: m_streamModules) {
          callbacks.push_back(mod->registrationCallback());
        }
        //Since only the first module will actually do the registration
        // we will change its callback to call all the callbacks
        firstMod->callWhenNewProductsRegistered([callbacks](BranchDescription const& iBD) {
          for(auto c: callbacks) {
            c(iBD);
          }
        });
      }
      firstMod->registerProducts(firstMod,reg,moduleDescription_);
    }
    
    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::itemsToGet(BranchType iType, std::vector<ProductResolverIndexAndSkipBit>& iIndices) const {
      assert(not m_streamModules.empty());
      m_streamModules[0]->itemsToGet(iType,iIndices);
    }
    
    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::itemsMayGet(BranchType iType, std::vector<ProductResolverIndexAndSkipBit>& iIndices) const {
      assert(not m_streamModules.empty());
      m_streamModules[0]->itemsMayGet(iType,iIndices);
    }

    template<typename T>
    std::vector<edm::ProductResolverIndexAndSkipBit> const&
    ProducingModuleAdaptorBase<T>::itemsToGetFrom(BranchType iType) const {
      assert(not m_streamModules.empty());
      return m_streamModules[0]->itemsToGetFrom(iType);
    }

    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::modulesWhoseProductsAreConsumed(std::vector<ModuleDescription const*>& modules,
                                                                   ProductRegistry const& preg,
                                                                   std::map<std::string, ModuleDescription const*> const& labelsToDesc,
                                                                   std::string const& processName) const {
      assert(not m_streamModules.empty());
      return m_streamModules[0]->modulesWhoseProductsAreConsumed(modules, preg, labelsToDesc, processName);
    }

    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::convertCurrentProcessAlias(std::string const& processName) {
      for(auto mod: m_streamModules) {
        mod->convertCurrentProcessAlias(processName);
      }
    }

    template< typename T>
    std::vector<edm::ConsumesInfo>
    ProducingModuleAdaptorBase<T>::consumesInfo() const {
      assert(not m_streamModules.empty());
      return m_streamModules[0]->consumesInfo();
    }

    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::updateLookup(BranchType iType,
                                                ProductResolverIndexHelper const& iHelper,
                                                bool iPrefetchMayGet) {
      for(auto mod: m_streamModules) {
        mod->updateLookup(iType,iHelper,iPrefetchMayGet);
      }
    }

    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::resolvePutIndicies(BranchType iBranchType,
                            ModuleToResolverIndicies const& iIndicies,
                            std::string const& moduleLabel) {
      for(auto mod: m_streamModules) {
        mod->resolvePutIndicies(iBranchType,iIndicies,moduleLabel);
      }
    }


    template< typename T>
    std::vector<edm::ProductResolverIndex> const&
    ProducingModuleAdaptorBase<T>::indiciesForPutProducts(BranchType iBranchType) const {
      return m_streamModules[0]->indiciesForPutProducts(iBranchType);
    }

    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::doBeginJob() {
      
    }
    
    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::doBeginStream(StreamID id) {
      m_streamModules[id]->beginStream(id);
    }
    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::doEndStream(StreamID id) {
      m_streamModules[id]->endStream();
    }
    
    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::doStreamBeginRun(StreamID id,
                                                    RunPrincipal const& rp,
                                                    EventSetup const& c,
                                                    ModuleCallingContext const* mcc)
    {
      auto mod = m_streamModules[id];
      setupRun(mod, rp.index());
      
      Run r(rp, moduleDescription_, mcc, false);
      r.setConsumer(mod);
      mod->beginRun(r, c);
      
    }    
    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::doStreamEndRun(StreamID id,
                                                  RunPrincipal const& rp,
                                                  EventSetup const& c,
                                                  ModuleCallingContext const* mcc)
    {
      auto mod = m_streamModules[id];
      Run r(rp, moduleDescription_, mcc, true);
      r.setConsumer(mod);
      mod->endRun(r, c);
      streamEndRunSummary(mod,r,c);
    }
    
    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::doStreamBeginLuminosityBlock(StreamID id,
                                                                LuminosityBlockPrincipal const& lbp,
                                                                EventSetup const& c,
                                                                ModuleCallingContext const* mcc) {
      auto mod = m_streamModules[id];
      setupLuminosityBlock(mod,lbp.index());
      
      LuminosityBlock lb(lbp, moduleDescription_, mcc, false);
      lb.setConsumer(mod);
      mod->beginLuminosityBlock(lb, c);
    }
    
    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::doStreamEndLuminosityBlock(StreamID id,
                                                              LuminosityBlockPrincipal const& lbp,
                                                              EventSetup const& c,
                                                              ModuleCallingContext const* mcc)
    {
      auto mod = m_streamModules[id];
      LuminosityBlock lb(lbp, moduleDescription_, mcc, true);
      lb.setConsumer(mod);
      mod->endLuminosityBlock(lb, c);
      streamEndLuminosityBlockSummary(mod,lb, c);
    }
    
    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::doRespondToOpenInputFile(FileBlock const&){}
    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::doRespondToCloseInputFile(FileBlock const&){}

    template< typename T>
    void
    ProducingModuleAdaptorBase<T>::doRegisterThinnedAssociations(ProductRegistry const& registry,
                                                                 ThinnedAssociationsHelper& helper) {
      assert(not m_streamModules.empty());
      auto mod = m_streamModules[0];
      mod->registerThinnedAssociations(registry, helper);
    }
  }
}