RootEmbeddedFileSequence.h

Go to the documentation of this file.

#ifndef IOPool_Input_RootEmbeddedFileSequence_h
#define IOPool_Input_RootEmbeddedFileSequence_h
 
/*----------------------------------------------------------------------
 
RootEmbeddedFileSequence: This is an InputSource
 
----------------------------------------------------------------------*/
 
#include "RootInputFileSequence.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Sources/interface/VectorInputSource.h"
#include "DataFormats/Provenance/interface/ProcessHistoryID.h"
 
#include <memory>
#include <string>
#include <vector>
 
namespace CLHEP {
  class HepRandomEngine;
}
 
namespace edm {
 
  class BranchID;
  class FileCatalogItem;
  class InputFileCatalog;
  class ParameterSetDescription;
  class EmbeddedRootSource;
  class RootFile;
 
  class RootEmbeddedFileSequence : public RootInputFileSequence {
  public:
    explicit RootEmbeddedFileSequence(ParameterSet const& pset,
                                      EmbeddedRootSource& input,
                                      InputFileCatalog const& catalog);
    ~RootEmbeddedFileSequence() override;
 
    RootEmbeddedFileSequence(RootEmbeddedFileSequence const&) = delete;             // Disallow copying and moving
    RootEmbeddedFileSequence& operator=(RootEmbeddedFileSequence const&) = delete;  // Disallow copying and moving
 
    void closeFile_() override;
    void endJob();
    void skipEntries(unsigned int offset);
    bool readOneEvent(
        EventPrincipal& cache, size_t& fileNameHash, CLHEP::HepRandomEngine*, EventID const* id, bool recycleFiles);
    bool readOneRandom(EventPrincipal& cache, size_t& fileNameHash, CLHEP::HepRandomEngine*, EventID const*, bool);
    bool readOneRandomWithID(
        EventPrincipal& cache, size_t& fileNameHash, CLHEP::HepRandomEngine*, EventID const* id, bool);
    bool readOneSequential(
        EventPrincipal& cache, size_t& fileNameHash, CLHEP::HepRandomEngine*, EventID const*, bool recycleFiles);
    bool readOneSequentialWithID(
        EventPrincipal& cache, size_t& fileNameHash, CLHEP::HepRandomEngine*, EventID const* id, bool);
    void readOneSpecified(EventPrincipal& cache, size_t& fileNameHash, SecondaryEventIDAndFileInfo const& id);
 
    static void fillDescription(ParameterSetDescription& desc);
 
  private:
    void initFile_(bool skipBadFiles) override;
    RootFileSharedPtr makeRootFile(std::shared_ptr<InputFile> filePtr) override;
 
    EmbeddedRootSource& input_;
 
    std::vector<ProcessHistoryID> orderedProcessHistoryIDs_;
 
    bool sequential_;
    bool sameLumiBlock_;
    bool (RootEmbeddedFileSequence::*fptr_)(EventPrincipal&, size_t&, CLHEP::HepRandomEngine*, EventID const*, bool);
    int eventsRemainingInFile_;
    int initialNumberOfEventsToSkip_;
    unsigned int treeCacheSize_;
    bool enablePrefetching_;
    bool enforceGUIDInFileName_;
    unsigned int maxFileSkips_;
  };  // class RootEmbeddedFileSequence
}  // namespace edm
#endif