BranchIDListHelper.h

Go to the documentation of this file.

#ifndef FWCore_Framework_BranchIDListHelper_h
#define FWCore_Framework_BranchIDListHelper_h

#include "DataFormats/Provenance/interface/BranchListIndex.h"
#include "DataFormats/Provenance/interface/BranchIDList.h"
#include "DataFormats/Provenance/interface/ProductID.h"
#include "DataFormats/Provenance/interface/ProvenanceFwd.h"

#include <map>
#include <vector>
#include <limits>

namespace edm {
    
  class BranchIDListHelper {
  public:
    typedef std::pair<BranchListIndex, ProductIndex> IndexPair;
    typedef std::multimap<BranchID, IndexPair> BranchIDToIndexMap;

    BranchIDListHelper();

    //CMS-THREADING called when a new file is opened
    bool updateFromInput(BranchIDLists const& bidlists);
    void fixBranchListIndexes(BranchListIndexes& indexes) const;

    void updateFromRegistry(ProductRegistry const& reg);

    //CMS-THREADING this is called in SubJob::beginJob
    BranchIDLists& mutableBranchIDLists() {return branchIDLists_;}

    //Used by the EventPrincipal
    BranchIDLists const& branchIDLists() const {return branchIDLists_;}
    BranchIDToIndexMap const& branchIDToIndexMap() const {return branchIDToIndexMap_;}
    BranchListIndex producedBranchListIndex() const {return producedBranchListIndex_;}
    bool hasProducedProducts() const {
      return producedBranchListIndex_ != std::numeric_limits<BranchListIndex>::max();
    }

  private:
    BranchIDLists branchIDLists_;
    BranchIDToIndexMap branchIDToIndexMap_;
    std::vector<BranchListIndex> inputIndexToJobIndex_;
    BranchListIndex producedBranchListIndex_;
  };
}

#endif