EvFOutputModule.h

Go to the documentation of this file.

#ifndef EventFilter_Utilities_EvFOutputModule_h
#define EventFilter_Utilities_EvFOutputModule_h

#include "IOPool/Streamer/interface/StreamerOutputFile.h"
#include "FWCore/Framework/interface/one/OutputModule.h"
#include "IOPool/Streamer/interface/StreamerOutputModuleCommon.h"
#include "FWCore/Utilities/interface/EDGetToken.h"

#include "EventFilter/Utilities/interface/JsonMonitorable.h"
#include "EventFilter/Utilities/interface/FastMonitor.h"
#
typedef edm::detail::TriggerResultsBasedEventSelector::handle_t Trig;

namespace evf {

  class FastMonitoringService;
//  class EvFOutputEventWriter;

  class EvFOutputEventWriter
  {
  public:

    explicit EvFOutputEventWriter(std::string const& filePath):
      filePath_(filePath),
      accepted_(0),
      stream_writer_events_(new StreamerOutputFile(filePath))
    {
    }

    ~EvFOutputEventWriter() {
    }

    void close() {
        stream_writer_events_->close();
    }

    void doOutputEvent(EventMsgBuilder const& msg) {
      EventMsgView eview(msg.startAddress());
      stream_writer_events_->write(eview);
    }

    uint32 get_adler32() const {
      return stream_writer_events_->adler32();
    }

    std::string const& getFilePath() const {return filePath_;}

    unsigned long getAccepted() const {return accepted_;}
    void incAccepted() {accepted_++;}

  private:
    std::string filePath_;
    unsigned long accepted_;
    edm::propagate_const<std::unique_ptr<StreamerOutputFile>> stream_writer_events_;

  };


  class EvFOutputJSONWriter {
    public:
    EvFOutputJSONWriter(edm::ParameterSet const& ps, edm::SelectedProducts const* selections, std::string const& streamLabel);
 
    edm::StreamerOutputModuleCommon streamerCommon_;

    jsoncollector::IntJ processed_;
    jsoncollector::IntJ accepted_;
    jsoncollector::IntJ errorEvents_; 
    jsoncollector::IntJ retCodeMask_; 
    jsoncollector::StringJ filelist_;
    jsoncollector::IntJ filesize_; 
    jsoncollector::StringJ inputFiles_;
    jsoncollector::IntJ fileAdler32_; 
    jsoncollector::StringJ transferDestination_; 
    jsoncollector::StringJ mergeType_;
    jsoncollector::IntJ hltErrorEvents_; 
    std::shared_ptr<jsoncollector::FastMonitor> jsonMonitor_;
    jsoncollector::DataPointDefinition outJsonDef_;
  };


  typedef edm::one::OutputModule<edm::one::WatchRuns,edm::LuminosityBlockCache<evf::EvFOutputEventWriter>> EvFOutputModuleType;

  class EvFOutputModule : 
    public EvFOutputModuleType
  {
  public:
    explicit EvFOutputModule(edm::ParameterSet const& ps);
    ~EvFOutputModule() override;
    static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
    
  private:
    void beginRun(edm::RunForOutput const& run) override;
    void write(edm::EventForOutput const& e) override;

    //pure in parent class but unused here
    void writeLuminosityBlock(edm::LuminosityBlockForOutput const&) override {}
    void writeRun(edm::RunForOutput const&) override {}
    void endRun(edm::RunForOutput const&) override {}

    std::shared_ptr<EvFOutputEventWriter> globalBeginLuminosityBlock(edm::LuminosityBlockForOutput const& iLB) const override;
    void globalEndLuminosityBlock(edm::LuminosityBlockForOutput const& iLB) override;

    Trig getTriggerResults(edm::EDGetTokenT<edm::TriggerResults> const& token, edm::EventForOutput const& e) const;

    edm::ParameterSet const& ps_;
    std::string streamLabel_;
    edm::EDGetTokenT<edm::TriggerResults> trToken_;

    evf::FastMonitoringService *fms_;

    std::unique_ptr<evf::EvFOutputJSONWriter> jsonWriter_;


  }; //end-of-class-def

} // end of namespace-evf

#endif