CMS 3D CMS Logo

List of all members | Public Member Functions | Static Public Member Functions | Private Member Functions | Private Attributes
edm::StreamerFileReader Class Reference

#include <StreamerFileReader.h>

Inheritance diagram for edm::StreamerFileReader:
edm::StreamerInputSource edm::RawInputSource edm::InputSource

Public Member Functions

 StreamerFileReader (ParameterSet const &pset, InputSourceDescription const &desc)
 
 ~StreamerFileReader () override
 
- Public Member Functions inherited from edm::StreamerInputSource
void deserializeAndMergeWithRegistry (InitMsgView const &initView, bool subsequent=false)
 
void deserializeEvent (EventMsgView const &eventView)
 
std::unique_ptr< SendJobHeaderdeserializeRegistry (InitMsgView const &initView)
 
bool isBufferLZMA (unsigned char const *inputBuffer, unsigned int inputSize)
 
bool isBufferZSTD (unsigned char const *inputBuffer, unsigned int inputSize)
 
 StreamerInputSource (ParameterSet const &pset, InputSourceDescription const &desc)
 
 ~StreamerInputSource () override
 
- Public Member Functions inherited from edm::RawInputSource
 RawInputSource (ParameterSet const &pset, InputSourceDescription const &desc)
 
 ~RawInputSource () override
 
- Public Member Functions inherited from edm::InputSource
std::shared_ptr< ActivityRegistryactReg () const
 Accessor for Activity Registry. More...
 
std::shared_ptr< BranchIDListHelper const > branchIDListHelper () const
 Accessors for branchIDListHelper. More...
 
std::shared_ptr< BranchIDListHelper > & branchIDListHelper ()
 
void closeFile (FileBlock *, bool cleaningUpAfterException)
 close current file More...
 
void doBeginJob ()
 Called by framework at beginning of job. More...
 
virtual void doBeginLumi (LuminosityBlockPrincipal &lbp, ProcessContext const *)
 Called by framework at beginning of lumi block. More...
 
virtual void doBeginRun (RunPrincipal &rp, ProcessContext const *)
 Called by framework at beginning of run. More...
 
void doEndJob ()
 Called by framework at end of job. More...
 
void fillProcessBlockHelper ()
 Fill the ProcessBlockHelper with info for the current file. More...
 
ProcessingController::ForwardState forwardState () const
 
bool goToEvent (EventID const &eventID)
 
 InputSource (ParameterSet const &, InputSourceDescription const &)
 Constructor. More...
 
 InputSource (InputSource const &)=delete
 
void issueReports (EventID const &eventID, StreamID streamID)
 issue an event report More...
 
LuminosityBlockNumber_t luminosityBlock () const
 Accessor for current luminosity block number. More...
 
std::shared_ptr< LuminosityBlockAuxiliaryluminosityBlockAuxiliary () const
 Called by the framework to merge or insert lumi in principal cache. More...
 
int maxEvents () const
 
int maxLuminosityBlocks () const
 
ModuleDescription const & moduleDescription () const
 Accessor for 'module' description. More...
 
ItemType nextItemType ()
 Advances the source to the next item. More...
 
bool nextProcessBlock (ProcessBlockPrincipal &)
 Next process block, return false if there is none, sets the processName in the principal. More...
 
InputSourceoperator= (InputSource const &)=delete
 
std::shared_ptr< ProcessBlockHelper const > processBlockHelper () const
 Accessors for processBlockHelper. More...
 
std::shared_ptr< ProcessBlockHelper > & processBlockHelper ()
 
ProcessConfiguration const & processConfiguration () const
 Accessor for Process Configuration. More...
 
std::string const & processGUID () const
 Accessor for global process identifier. More...
 
ProcessHistoryRegistry const & processHistoryRegistry () const
 Accessors for process history registry. More...
 
ProcessHistoryRegistryprocessHistoryRegistry ()
 
ProcessingMode processingMode () const
 RunsLumisAndEvents (default), RunsAndLumis, or Runs. More...
 
std::shared_ptr< ProductRegistry const > productRegistry () const
 Accessors for product registry. More...
 
bool randomAccess () const
 
void readAndMergeLumi (LuminosityBlockPrincipal &lbp)
 Read next luminosity block (same as a prior lumi) More...
 
void readAndMergeRun (RunPrincipal &rp)
 Read next run (same as a prior run) More...
 
void readEvent (EventPrincipal &ep, StreamContext &)
 Read next event. More...
 
bool readEvent (EventPrincipal &ep, EventID const &, StreamContext &)
 Read a specific event. More...
 
std::shared_ptr< FileBlockreadFile ()
 Read next file. More...
 
void readLuminosityBlock (LuminosityBlockPrincipal &lumiPrincipal, HistoryAppender &historyAppender)
 Read next luminosity block (new lumi) More...
 
std::shared_ptr< LuminosityBlockAuxiliaryreadLuminosityBlockAuxiliary ()
 Read next luminosity block Auxilary. More...
 
void readProcessBlock (ProcessBlockPrincipal &)
 Read next process block. More...
 
void readRun (RunPrincipal &runPrincipal, HistoryAppender &historyAppender)
 Read next run (new run) More...
 
std::shared_ptr< RunAuxiliaryreadRunAuxiliary ()
 Read next run Auxiliary. More...
 
ProcessHistoryID const & reducedProcessHistoryID () const
 
virtual void registerProducts ()
 Register any produced products. More...
 
int remainingEvents () const
 
int remainingLuminosityBlocks () const
 
void repeat ()
 Reset the remaining number of events/lumis to the maximum number. More...
 
std::pair< SharedResourcesAcquirer *, std::recursive_mutex * > resourceSharedWithDelayedReader ()
 Returns nullptr if no resource shared between the Source and a DelayedReader. More...
 
ProcessingController::ReverseState reverseState () const
 
void rewind ()
 Begin again at the first event. More...
 
RunNumber_t run () const
 Accessor for current run number. More...
 
std::shared_ptr< RunAuxiliaryrunAuxiliary () const
 Called by the framework to merge or insert run in principal cache. More...
 
void setLuminosityBlockNumber_t (LuminosityBlockNumber_t lb)
 Set the luminosity block ID. More...
 
void setRunNumber (RunNumber_t r)
 Set the run number. More...
 
void skipEvents (int offset)
 
void switchTo (std::shared_ptr< ProductRegistry > iOther)
 switch to a different ProductRegistry. More...
 
std::shared_ptr< ThinnedAssociationsHelper const > thinnedAssociationsHelper () const
 Accessors for thinnedAssociationsHelper. More...
 
std::shared_ptr< ThinnedAssociationsHelper > & thinnedAssociationsHelper ()
 
Timestamp const & timestamp () const
 Accessor for the current time, as seen by the input source. More...
 
virtual ~InputSource () noexcept(false)
 Destructor. More...
 

Static Public Member Functions

static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
- Static Public Member Functions inherited from edm::StreamerInputSource
static void fillDescription (ParameterSetDescription &description)
 
static void mergeIntoRegistry (SendJobHeader const &header, ProductRegistry &, BranchIDListHelper &, ThinnedAssociationsHelper &, bool subsequent)
 
static unsigned int uncompressBuffer (unsigned char *inputBuffer, unsigned int inputSize, std::vector< unsigned char > &outputBuffer, unsigned int expectedFullSize)
 
static unsigned int uncompressBufferLZMA (unsigned char *inputBuffer, unsigned int inputSize, std::vector< unsigned char > &outputBuffer, unsigned int expectedFullSize, bool hasHeader=true)
 
static unsigned int uncompressBufferZSTD (unsigned char *inputBuffer, unsigned int inputSize, std::vector< unsigned char > &outputBuffer, unsigned int expectedFullSize, bool hasHeader=true)
 
- Static Public Member Functions inherited from edm::RawInputSource
static void fillDescription (ParameterSetDescription &description)
 
- Static Public Member Functions inherited from edm::InputSource
static const std::string & baseType ()
 
static void fillDescription (ParameterSetDescription &desc)
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &)
 

Private Member Functions

Next checkNext () override
 
std::shared_ptr< EventSkipperByID const > eventSkipperByID () const
 
std::shared_ptr< EventSkipperByID > & eventSkipperByID ()
 
void genuineCloseFile () override
 
void genuineReadFile () override
 
InitMsgView const * getHeader ()
 
EventMsgView const * getNextEvent ()
 
bool newHeader ()
 
void reset_ () override
 
void skip (int toSkip) override
 

Private Attributes

edm::propagate_const< std::shared_ptr< EventSkipperByID > > eventSkipperByID_
 
int initialNumberOfEventsToSkip_
 
bool isFirstFile_ = true
 
int prefetchMBytes_
 
std::vector< FileCatalogItemstreamerNames_
 
edm::propagate_const< std::unique_ptr< StreamerInputFile > > streamReader_
 

Additional Inherited Members

- Public Types inherited from edm::RawInputSource
enum  Next { Next::kEvent, Next::kFile, Next::kStop }
 
- Public Types inherited from edm::InputSource
enum  ItemType {
  IsInvalid, IsStop, IsFile, IsRun,
  IsLumi, IsEvent, IsRepeat, IsSynchronize
}
 
enum  ProcessingMode { Runs, RunsAndLumis, RunsLumisAndEvents }
 
- Public Attributes inherited from edm::InputSource
signalslot::Signal< void(StreamContext const &, ModuleCallingContext const &)> postEventReadFromSourceSignal_
 
signalslot::Signal< void(StreamContext const &, ModuleCallingContext const &)> preEventReadFromSourceSignal_
 
- Protected Member Functions inherited from edm::StreamerInputSource
void resetAfterEndRun ()
 
- Protected Member Functions inherited from edm::RawInputSource
void makeEvent (EventPrincipal &eventPrincipal, EventAuxiliary const &eventAuxiliary)
 
void setInputFileTransitionsEachEvent ()
 
- Protected Member Functions inherited from edm::InputSource
virtual void beginJob ()
 Begin protected makes it easier to do template programming. More...
 
void decreaseRemainingEventsBy (int iSkipped)
 
bool eventCached () const
 
bool newLumi () const
 
bool newRun () const
 
ProcessHistoryRegistryprocessHistoryRegistryForUpdate ()
 
ProductRegistryproductRegistryUpdate ()
 
void reset () const
 
void resetEventCached ()
 
void resetLuminosityBlockAuxiliary (bool isNewLumi=true) const
 
void resetNewLumi ()
 
void resetNewRun ()
 
void resetRunAuxiliary (bool isNewRun=true) const
 
void setEventCached ()
 Called by the framework to merge or ached() const {return eventCached_;}. More...
 
void setLuminosityBlockAuxiliary (LuminosityBlockAuxiliary *lbp)
 
void setNewLumi ()
 
void setNewRun ()
 
void setRunAuxiliary (RunAuxiliary *rp)
 
void setTimestamp (Timestamp const &theTime)
 To set the current time, as seen by the input source. More...
 
ItemType state () const
 
- Static Protected Member Functions inherited from edm::StreamerInputSource
static void buildClassCache (SendDescs const &descs)
 
static void declareStreamers (SendDescs const &descs)
 

Detailed Description

Definition at line 22 of file StreamerFileReader.h.

Constructor & Destructor Documentation

◆ StreamerFileReader()

edm::StreamerFileReader::StreamerFileReader ( ParameterSet const &  pset,
InputSourceDescription const &  desc 
)

Definition at line 15 of file StreamerFileReader.cc.

References getEcalConditions_orcoffint2r_cff::catalog, muonDTDigis_cfi::pset, reset_(), streamerNames_, and AlCaHLTBitMon_QueryRunRegistry::string.

17  streamReader_(),
19  initialNumberOfEventsToSkip_(pset.getUntrackedParameter<unsigned int>("skipEvents")),
20  prefetchMBytes_(pset.getUntrackedParameter<unsigned int>("prefetchMBytes")) {
21  InputFileCatalog catalog(pset.getUntrackedParameter<std::vector<std::string> >("fileNames"),
22  pset.getUntrackedParameter<std::string>("overrideCatalog"));
23  streamerNames_ = catalog.fileCatalogItems();
24  reset_();
25  }
static std::unique_ptr< EventSkipperByID > create(ParameterSet const &pset)
StreamerInputSource(ParameterSet const &pset, InputSourceDescription const &desc)
std::vector< FileCatalogItem > streamerNames_
edm::propagate_const< std::unique_ptr< StreamerInputFile > > streamReader_
edm::propagate_const< std::shared_ptr< EventSkipperByID > > eventSkipperByID_

◆ ~StreamerFileReader()

edm::StreamerFileReader::~StreamerFileReader ( )
override

Definition at line 27 of file StreamerFileReader.cc.

27 {}

Member Function Documentation

◆ checkNext()

StreamerFileReader::Next edm::StreamerFileReader::checkNext ( )
overrideprivatevirtual

◆ eventSkipperByID() [1/2]

std::shared_ptr<EventSkipperByID const> edm::StreamerFileReader::eventSkipperByID ( ) const
inlineprivate

Definition at line 40 of file StreamerFileReader.h.

References eventSkipperByID_, and edm::get_underlying_safe().

Referenced by reset_().

constexpr std::shared_ptr< T > & get_underlying_safe(propagate_const< std::shared_ptr< T >> &iP)
edm::propagate_const< std::shared_ptr< EventSkipperByID > > eventSkipperByID_

◆ eventSkipperByID() [2/2]

std::shared_ptr<EventSkipperByID>& edm::StreamerFileReader::eventSkipperByID ( )
inlineprivate

Definition at line 41 of file StreamerFileReader.h.

References eventSkipperByID_, and edm::get_underlying_safe().

constexpr std::shared_ptr< T > & get_underlying_safe(propagate_const< std::shared_ptr< T >> &iP)
edm::propagate_const< std::shared_ptr< EventSkipperByID > > eventSkipperByID_

◆ fillDescriptions()

void edm::StreamerFileReader::fillDescriptions ( ConfigurationDescriptions descriptions)
static

Definition at line 113 of file StreamerFileReader.cc.

References edm::ConfigurationDescriptions::add(), submitPVResolutionJobs::desc, edm::EventSkipperByID::fillDescription(), edm::StreamerInputSource::fillDescription(), AlCaHLTBitMon_QueryRunRegistry::string, and mitigatedMETSequence_cff::U.

113  {
115  desc.setComment("Reads events from streamer files.");
116  desc.addUntracked<std::vector<std::string> >("fileNames")->setComment("Names of files to be processed.");
117  desc.addUntracked<unsigned int>("skipEvents", 0U)
118  ->setComment("Skip the first 'skipEvents' events that otherwise would have been processed.");
119  desc.addUntracked<std::string>("overrideCatalog", std::string());
120  //This next parameter is read in the base class, but its default value depends on the derived class, so it is set here.
121  desc.addUntracked<bool>("inputFileTransitionsEachEvent", false);
122  desc.addUntracked<unsigned int>("prefetchMBytes", 0);
125  descriptions.add("source", desc);
126  }
static void fillDescription(ParameterSetDescription &description)
static void fillDescription(ParameterSetDescription &desc)

◆ genuineCloseFile()

void edm::StreamerFileReader::genuineCloseFile ( )
overrideprivatevirtual

Reimplemented from edm::RawInputSource.

Definition at line 75 of file StreamerFileReader.cc.

References streamReader_.

75  {
76  if (streamReader_.get() != nullptr)
77  streamReader_->closeStreamerFile();
78  }
edm::propagate_const< std::unique_ptr< StreamerInputFile > > streamReader_

◆ genuineReadFile()

void edm::StreamerFileReader::genuineReadFile ( )
overrideprivatevirtual

Reimplemented from edm::RawInputSource.

Definition at line 80 of file StreamerFileReader.cc.

References edm::StreamerInputSource::deserializeAndMergeWithRegistry(), getHeader(), RecoTauValidation_cfi::header, isFirstFile_, and streamReader_.

80  {
81  if (isFirstFile_) {
82  //The file was already opened in the constructor
83  isFirstFile_ = false;
84  return;
85  }
86  streamReader_->openNextFile();
87  // FDEBUG(6) << "A new file has been opened and we must compare Headers here !!" << std::endl;
88  // A new file has been opened and we must compare Heraders here !!
89  //Get header/init from reader
90  InitMsgView const* header = getHeader();
92  }
edm::propagate_const< std::unique_ptr< StreamerInputFile > > streamReader_
InitMsgView const * getHeader()
void deserializeAndMergeWithRegistry(InitMsgView const &initView, bool subsequent=false)

◆ getHeader()

InitMsgView const * edm::StreamerFileReader::getHeader ( )
private

Definition at line 96 of file StreamerFileReader.cc.

References Exception, edm::errors::FileReadError, RecoTauValidation_cfi::header, Header::INIT, and streamReader_.

Referenced by genuineReadFile(), and reset_().

96  {
97  InitMsgView const* header = streamReader_->startMessage();
98 
99  if (header->code() != Header::INIT) { //INIT Msg
100  throw Exception(errors::FileReadError, "StreamerFileReader::readHeader")
101  << "received wrong message type: expected INIT, got " << header->code() << "\n";
102  }
103  return header;
104  }
edm::propagate_const< std::unique_ptr< StreamerInputFile > > streamReader_

◆ getNextEvent()

EventMsgView const * edm::StreamerFileReader::getNextEvent ( )
private

Definition at line 106 of file StreamerFileReader.cc.

References edm::StreamerInputFile::kEvent, and streamReader_.

Referenced by checkNext(), and skip().

106  {
108  return nullptr;
109  }
110  return streamReader_->currentRecord();
111  }
edm::propagate_const< std::unique_ptr< StreamerInputFile > > streamReader_

◆ newHeader()

bool edm::StreamerFileReader::newHeader ( )
private

Definition at line 94 of file StreamerFileReader.cc.

References streamReader_.

Referenced by checkNext().

94 { return streamReader_->newHeader(); }
edm::propagate_const< std::unique_ptr< StreamerInputFile > > streamReader_

◆ reset_()

void edm::StreamerFileReader::reset_ ( )
overrideprivatevirtual

Reimplemented from edm::RawInputSource.

Definition at line 29 of file StreamerFileReader.cc.

References edm::StreamerInputSource::deserializeAndMergeWithRegistry(), eventSkipperByID(), Exception, edm::errors::FileReadError, getHeader(), RecoTauValidation_cfi::header, initialNumberOfEventsToSkip_, isFirstFile_, prefetchMBytes_, skip(), streamerNames_, and streamReader_.

Referenced by StreamerFileReader().

29  {
30  if (streamerNames_.size() > 1) {
31  streamReader_ = std::make_unique<StreamerInputFile>(streamerNames_, eventSkipperByID(), prefetchMBytes_);
32  } else if (streamerNames_.size() == 1) {
33  streamReader_ = std::make_unique<StreamerInputFile>(streamerNames_.at(0).fileNames()[0],
34  streamerNames_.at(0).logicalFileName(),
37  } else {
38  throw Exception(errors::FileReadError, "StreamerFileReader::StreamerFileReader")
39  << "No fileNames were specified\n";
40  }
41  isFirstFile_ = true;
42  InitMsgView const* header = getHeader();
46  }
47  }
std::vector< FileCatalogItem > streamerNames_
edm::propagate_const< std::unique_ptr< StreamerInputFile > > streamReader_
void skip(int toSkip) override
InitMsgView const * getHeader()
std::shared_ptr< EventSkipperByID const > eventSkipperByID() const
void deserializeAndMergeWithRegistry(InitMsgView const &initView, bool subsequent=false)

◆ skip()

void edm::StreamerFileReader::skip ( int  toSkip)
overrideprivatevirtual

Reimplemented from edm::InputSource.

Definition at line 62 of file StreamerFileReader.cc.

References EventMsgView::event(), eventSkipperByID_, getNextEvent(), mps_fire::i, EventMsgView::lumi(), and EventMsgView::run().

Referenced by reset_().

62  {
63  for (int i = 0; i != toSkip; ++i) {
64  EventMsgView const* evMsg = getNextEvent();
65  if (evMsg == nullptr) {
66  return;
67  }
68  // If the event would have been skipped anyway, don't count it as a skipped event.
69  if (eventSkipperByID_ && eventSkipperByID_->skipIt(evMsg->run(), evMsg->lumi(), evMsg->event())) {
70  --i;
71  }
72  }
73  }
uint32 lumi() const
Definition: EventMessage.cc:81
uint64 event() const
Definition: EventMessage.cc:76
edm::propagate_const< std::shared_ptr< EventSkipperByID > > eventSkipperByID_
uint32 run() const
Definition: EventMessage.cc:71
EventMsgView const * getNextEvent()

Member Data Documentation

◆ eventSkipperByID_

edm::propagate_const<std::shared_ptr<EventSkipperByID> > edm::StreamerFileReader::eventSkipperByID_
private

Definition at line 45 of file StreamerFileReader.h.

Referenced by eventSkipperByID(), and skip().

◆ initialNumberOfEventsToSkip_

int edm::StreamerFileReader::initialNumberOfEventsToSkip_
private

Definition at line 46 of file StreamerFileReader.h.

Referenced by reset_().

◆ isFirstFile_

bool edm::StreamerFileReader::isFirstFile_ = true
private

Definition at line 48 of file StreamerFileReader.h.

Referenced by genuineReadFile(), and reset_().

◆ prefetchMBytes_

int edm::StreamerFileReader::prefetchMBytes_
private

Definition at line 47 of file StreamerFileReader.h.

Referenced by reset_().

◆ streamerNames_

std::vector<FileCatalogItem> edm::StreamerFileReader::streamerNames_
private

Definition at line 43 of file StreamerFileReader.h.

Referenced by reset_(), and StreamerFileReader().

◆ streamReader_

edm::propagate_const<std::unique_ptr<StreamerInputFile> > edm::StreamerFileReader::streamReader_
private