DQMStreamerReader.h

Go to the documentation of this file.

#ifndef DQMServices_StreamerIO_DQMStreamerReader_h
#define DQMServices_StreamerIO_DQMStreamerReader_h
 
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "IOPool/Streamer/interface/StreamerInputSource.h"
#include "IOPool/Streamer/interface/StreamerInputFile.h"
#include "IOPool/Streamer/interface/MsgTools.h"
 
#include "DQMFileIterator.h"
#include "DQMMonitoringService.h"
#include "TriggerSelector.h"
 
#include <filesystem>
 
#include <memory>
#include <string>
#include <vector>
#include <iterator>
#include <boost/property_tree/json_parser.hpp>
#include <boost/property_tree/ptree.hpp>
 
namespace dqmservices {
 
  class DQMStreamerReader : public edm::StreamerInputSource {
  public:
    DQMStreamerReader(edm::ParameterSet const& pset, edm::InputSourceDescription const& desc);
    ~DQMStreamerReader() override;
 
    bool newHeader();
    static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
 
    typedef std::vector<std::string> Strings;
 
  protected:
    Next checkNext() override;      /* from raw input source */
    void skip(int toSkip) override; /* from raw input source */
    void genuineReadFile() override;
    void genuineCloseFile() override;
 
  private:
    // our own, but we do inherit reset(),
    // which will break things if called
    void reset_() override;
 
    void openFileImp_(const DQMFileIterator::LumiEntry& entry);
    void closeFileImp_(const std::string& reason);
 
    bool openNextFileImp_();
 
    InitMsgView const* getHeaderMsg();
    EventMsgView const* getEventMsg();
 
    EventMsgView const* prepareNextEvent();
    bool prepareNextFile();
    bool acceptEvent(const EventMsgView*);
 
    bool triggerSel();
    bool matchTriggerSel(Strings const& tnames);
    bool acceptAllEvt_;
    bool matchTriggerSel_;
    bool isFirstFile_ = true;
 
    unsigned int runNumber_;
    std::string runInputDir_;
    std::string streamLabel_;
    Strings hltSel_;
 
    unsigned int processedEventPerLs_;
    unsigned int minEventsPerLs_;
 
    bool flagSkipFirstLumis_;
    bool flagEndOfRunKills_;
    bool flagDeleteDatFiles_;
 
    DQMFileIterator fiterator_;
 
    struct OpenFile {
      std::unique_ptr<edm::StreamerInputFile> streamFile_;
      DQMFileIterator::LumiEntry lumi_;
 
      bool open() { return (streamFile_.get() != nullptr); }
 
    } file_;
 
    std::shared_ptr<edm::EventSkipperByID> eventSkipperByID_;
    std::shared_ptr<TriggerSelector> eventSelector_;
 
    /* this is for monitoring */
    edm::Service<DQMMonitoringService> mon_;
  };
 
}  // namespace dqmservices
 
#endif