OutputModuleCommunicatorT.h

Go to the documentation of this file.

#ifndef FWCore_Framework_OutputModuleCommunicatorT_h
#define FWCore_Framework_OutputModuleCommunicatorT_h
/*----------------------------------------------------------------------
----------------------------------------------------------------------*/

#include "FWCore/Framework/src/OutputModuleCommunicator.h"

namespace edm {
  class ActivityRegistry;
  class OutputModule;
  class ThinnedAssociationsHelper;

  namespace one {
    class OutputModuleBase;
  }
  namespace global {
    class OutputModuleBase;
  }
  namespace limited {
    class OutputModuleBase;
  }
  namespace impl {
    std::unique_ptr<edm::OutputModuleCommunicator> createCommunicatorIfNeeded(void *);
    std::unique_ptr<edm::OutputModuleCommunicator> createCommunicatorIfNeeded(::edm::OutputModule *);
    std::unique_ptr<edm::OutputModuleCommunicator> createCommunicatorIfNeeded(::edm::one::OutputModuleBase *);
    std::unique_ptr<edm::OutputModuleCommunicator> createCommunicatorIfNeeded(::edm::global::OutputModuleBase *);
    std::unique_ptr<edm::OutputModuleCommunicator> createCommunicatorIfNeeded(::edm::limited::OutputModuleBase *);
  }
  
  template <typename T>
  
  class OutputModuleCommunicatorT : public edm::OutputModuleCommunicator {
  public:
    OutputModuleCommunicatorT(T* iModule):
    module_(iModule){}
    void closeFile() override;
    
    bool shouldWeCloseFile() const override;
    
    bool wantAllEvents() const override;
    
    void openFile(edm::FileBlock const& fb) override;
    
    void writeRunAsync(WaitingTaskHolder iTask,
                       edm::RunPrincipal const& rp,
                       ProcessContext const*,
                       ActivityRegistry*) override;
    
    void writeLumiAsync(WaitingTaskHolder iTask,
                        edm::LuminosityBlockPrincipal const& lbp,
                        ProcessContext const*,
                        ActivityRegistry*) override;
    
    bool limitReached() const override;
    
    void configure(edm::OutputModuleDescription const& desc) override;
    
    edm::SelectedProductsForBranchType const& keptProducts() const override;
    
    void selectProducts(edm::ProductRegistry const& preg, ThinnedAssociationsHelper const&) override;
    
    void setEventSelectionInfo(std::map<std::string, std::vector<std::pair<std::string, int> > > const& outputModulePathPositions,
                                       bool anyProductProduced) override;
    
    ModuleDescription const& description() const override;

    static std::unique_ptr<edm::OutputModuleCommunicator> createIfNeeded(T* iMod) {
      return std::move(impl::createCommunicatorIfNeeded(iMod));
    }

  private:
    inline T& module() const { return *module_;}
    T* module_;
  };
}
#endif