StreamerInputModule.h

Go to the documentation of this file.

#ifndef IOPool_Streamer_StreamerInputModule_h
#define IOPool_Streamer_StreamerInputModule_h

#include "IOPool/Streamer/interface/StreamerInputSource.h"

#include "FWCore/Utilities/interface/DebugMacros.h"
#include "FWCore/Utilities/interface/propagate_const.h"

#include <memory>
#include <string>
#include <fstream>
#include <vector>
#include <utility>
#include <iostream>
#include <algorithm> 
#include <iterator>

namespace edm
{
  template <typename Producer>
  class StreamerInputModule : public StreamerInputSource {
  public:  
    explicit StreamerInputModule(ParameterSet const& pset,
                 InputSourceDescription const& desc);
    virtual ~StreamerInputModule();
  private:
    virtual void genuineCloseFile() override {
      if(pr_.get() != nullptr) pr_->closeFile();
    }

    virtual bool checkNextEvent();

    //ProductRegistry const* prod_reg_;
    edm::propagate_const<std::unique_ptr<Producer>> pr_;
  }; //end-of-class-def

  template <typename Producer>
  StreamerInputModule<Producer>::~StreamerInputModule() {}

  template <typename Producer>
  StreamerInputModule<Producer>::StreamerInputModule(
                    ParameterSet const& pset,
                    InputSourceDescription const& desc):
    StreamerInputSource(pset, desc),
    //prod_reg_(&productRegistry()), 
    pr_(new Producer(pset)) {
    //Get header/init from Producer
    InitMsgView const* header = pr_->getHeader();
    deserializeAndMergeWithRegistry(*header); 
  }

  template <typename Producer>
  bool StreamerInputModule<Producer>::checkNextEvent() {

    EventMsgView const* eview = pr_->getNextEvent();

    if (pr_->newHeader()) {   
        FDEBUG(6) << "A new file has been opened and we must compare Headers here !!" << std::endl;
        // A new file has been opened and we must compare Headers here !!
        //Get header/init from Producer
        InitMsgView const* header = pr_->getHeader();
        deserializeAndMergeWithRegistry(*header, true);
    } 
    if (eview == 0) {
        return  false;
    }
    deserializeEvent(*eview);
    return true;
  }

} // end of namespace-edm
  
#endif