PrincipalCache.h

Go to the documentation of this file.

#ifndef FWCore_Framework_PrincipalCache_h
#define FWCore_Framework_PrincipalCache_h

/*
Contains a shared pointer to the RunPrincipal,
LuminosityBlockPrincipal, and EventPrincipal.
Manages merging of run and luminosity block
principals when there is more than one principal
from the same run or luminosity block and having
the same reduced ProcessHistoryID.

The EventPrincipal is reused each event and is created
by the EventProcessor or SubProcess which contains
an object of this type as a data member.

The RunPrincipal and LuminosityBlockPrincipal is
created by the InputSource each time a different
run or luminosity block is encountered.

Performs checks that process history IDs or runs and
lumis, run numbers, and luminosity numbers are consistent.

Original Author: W. David Dagenhart
*/

#include "DataFormats/Provenance/interface/ProcessHistoryID.h"
#include "DataFormats/Provenance/interface/ProcessHistoryRegistry.h"
#include "DataFormats/Provenance/interface/RunID.h"
#include "DataFormats/Provenance/interface/LuminosityBlockID.h"

#include <memory>
#include <vector>
#include <cassert>

namespace edm {

  class RunPrincipal;
  class LuminosityBlockPrincipal;
  class EventPrincipal;
  class RunAuxiliary;
  class LuminosityBlockAuxiliary;
  class ProductRegistry;
  class PreallocationConfiguration;

  class PrincipalCache {
  public:

    PrincipalCache();
    ~PrincipalCache();

    RunPrincipal& runPrincipal(ProcessHistoryID const& phid, RunNumber_t run) const;
    std::shared_ptr<RunPrincipal> const& runPrincipalPtr(ProcessHistoryID const& phid, RunNumber_t run) const;
    RunPrincipal& runPrincipal() const;
    std::shared_ptr<RunPrincipal> const& runPrincipalPtr() const;
    bool hasRunPrincipal() const {return bool(runPrincipal_);}

    LuminosityBlockPrincipal& lumiPrincipal(ProcessHistoryID const& phid, RunNumber_t run, LuminosityBlockNumber_t lumi) const;
    std::shared_ptr<LuminosityBlockPrincipal> const& lumiPrincipalPtr(ProcessHistoryID const& phid, RunNumber_t run, LuminosityBlockNumber_t lumi) const;
    LuminosityBlockPrincipal& lumiPrincipal() const;
    std::shared_ptr<LuminosityBlockPrincipal> const& lumiPrincipalPtr() const;
    bool hasLumiPrincipal() const {return bool(lumiPrincipal_);}

    EventPrincipal& eventPrincipal(unsigned int iStreamIndex) const { return *(eventPrincipals_[iStreamIndex]); }

    void merge(std::shared_ptr<RunAuxiliary> aux, std::shared_ptr<ProductRegistry const> reg);
    void merge(std::shared_ptr<LuminosityBlockAuxiliary> aux, std::shared_ptr<ProductRegistry const> reg);

    void setNumberOfConcurrentPrincipals(PreallocationConfiguration const&);
    void insert(std::shared_ptr<RunPrincipal> rp);
    void insert(std::shared_ptr<LuminosityBlockPrincipal> lbp);
    void insert(std::shared_ptr<EventPrincipal> ep);

    void deleteRun(ProcessHistoryID const& phid, RunNumber_t run);
    void deleteLumi(ProcessHistoryID const& phid, RunNumber_t run, LuminosityBlockNumber_t lumi);

    void adjustEventsToNewProductRegistry(std::shared_ptr<ProductRegistry const> reg);

    void adjustIndexesAfterProductRegistryAddition();

    void setProcessHistoryRegistry(ProcessHistoryRegistry const& phr) {processHistoryRegistry_ = &phr;}

  private:

    void throwRunMissing() const;
    void throwLumiMissing() const;

    // These are explicitly cleared when finished with the run,
    // lumi, or event
    std::shared_ptr<RunPrincipal> runPrincipal_;
    std::shared_ptr<LuminosityBlockPrincipal> lumiPrincipal_;
    std::vector<std::shared_ptr<EventPrincipal>> eventPrincipals_;

    // This is just an accessor to the registry owned by the input source. 
    ProcessHistoryRegistry const* processHistoryRegistry_; // We don't own this

    // These are intentionally not cleared so that when inserting
    // the next principal the conversion from full ProcessHistoryID_
    // to reduced ProcessHistoryID_ is still in memory and does
    // not need to be recalculated if the ID does not change. I
    // expect that very often these ID's will not change from one
    // principal to the next and a good amount of CPU can be saved
    // by not recalculating.
    ProcessHistoryID inputProcessHistoryID_;
    ProcessHistoryID reducedInputProcessHistoryID_;
    RunNumber_t run_;
    LuminosityBlockNumber_t lumi_;
  };
}

#endif