ThinnedAssociationsHelper.h

Go to the documentation of this file.

#ifndef DataFormats_Provenance_ThinnedAssociationsHelper_h
#define DataFormats_Provenance_ThinnedAssociationsHelper_h
 
#include "DataFormats/Provenance/interface/BranchID.h"
 
#include <map>
#include <set>
#include <vector>
 
namespace edm {
 
  class BranchDescription;
 
  class ThinnedAssociationBranches {
  public:
    ThinnedAssociationBranches();
    ThinnedAssociationBranches(BranchID const&, BranchID const&, BranchID const&);
 
    BranchID const& parent() const { return parent_; }
    BranchID const& association() const { return association_; }
    BranchID const& thinned() const { return thinned_; }
 
    bool operator<(ThinnedAssociationBranches const& rhs) const { return parent_ < rhs.parent_; }
 
  private:
    BranchID parent_;
    BranchID association_;
    BranchID thinned_;
  };
 
  class ThinnedAssociationsHelper {
  public:
    ThinnedAssociationsHelper();
 
    std::vector<ThinnedAssociationBranches>::const_iterator begin() const;
    std::vector<ThinnedAssociationBranches>::const_iterator end() const;
 
    std::vector<ThinnedAssociationBranches>::const_iterator parentBegin(BranchID const&) const;
    std::vector<ThinnedAssociationBranches>::const_iterator parentEnd(BranchID const&) const;
 
    void addAssociation(BranchID const&, BranchID const&, BranchID const&);
    void addAssociation(ThinnedAssociationBranches const&);
 
    std::vector<std::pair<BranchID, ThinnedAssociationBranches const*> > associationToBranches() const;
 
    void sort();
    void clear() { vThinnedAssociationBranches_.clear(); }
 
    void selectAssociationProducts(std::vector<BranchDescription const*> const& associationDescriptions,
                                   std::set<BranchID> const& keptProductsInEvent,
                                   std::map<BranchID, bool>& keepAssociation) const;
 
    std::vector<ThinnedAssociationBranches> const& data() const { return vThinnedAssociationBranches_; }
 
    void requireMatch(ThinnedAssociationBranches const& input) const;
 
    void updateFromPrimaryInput(ThinnedAssociationsHelper const&);
 
    void updateFromSecondaryInput(ThinnedAssociationsHelper const&,
                                  std::vector<BranchID> const& associationsFromSecondary);
 
    void updateFromParentProcess(
        ThinnedAssociationsHelper const& parentThinnedAssociationsHelper,
        std::map<BranchID, bool> const& keepAssociation,
        std::map<BranchID::value_type, BranchID::value_type> const& droppedBranchIDToKeptBranchID);
 
    void initAssociationsFromSecondary(std::vector<BranchID> const&, ThinnedAssociationsHelper const&);
 
  private:
    bool shouldKeepAssociation(
        BranchID const& association,
        std::vector<std::pair<BranchID, ThinnedAssociationBranches const*> > const& associationToBranches,
        std::set<BranchID>& branchesInRecursion,
        std::set<BranchID> const& keptProductsInEvent,
        std::map<BranchID, bool>& keepAssociation) const;
 
    std::vector<ThinnedAssociationBranches> vThinnedAssociationBranches_;
  };
}  // namespace edm
#endif