edm::ThinnedAssociationsHelper Class Reference

#include <ThinnedAssociationsHelper.h>

Public Member Functions
void	addAssociation (BranchID const &, BranchID const &, BranchID const &)
void	addAssociation (ThinnedAssociationBranches const &)
std::vector< std::pair< BranchID, ThinnedAssociationBranches const * > >	associationToBranches () const
std::vector< ThinnedAssociationBranches >::const_iterator	begin () const
void	clear ()
std::vector< ThinnedAssociationBranches > const &	data () const
std::vector< ThinnedAssociationBranches >::const_iterator	end () const
void	initAssociationsFromSecondary (std::vector< BranchID > const &, ThinnedAssociationsHelper const &)
std::vector< ThinnedAssociationBranches >::const_iterator	parentBegin (BranchID const &) const
std::vector< ThinnedAssociationBranches >::const_iterator	parentEnd (BranchID const &) const
void	requireMatch (ThinnedAssociationBranches const &input) const
void	selectAssociationProducts (std::vector< BranchDescription const * > const &associationDescriptions, std::set< BranchID > const &keptProductsInEvent, std::map< BranchID, bool > &keepAssociation) const
void	sort ()
	ThinnedAssociationsHelper ()
void	updateFromParentProcess (ThinnedAssociationsHelper const &parentThinnedAssociationsHelper, std::map< BranchID, bool > const &keepAssociation, std::map< BranchID::value_type, BranchID::value_type > const &droppedBranchIDToKeptBranchID)
void	updateFromPrimaryInput (ThinnedAssociationsHelper const &)
void	updateFromSecondaryInput (ThinnedAssociationsHelper const &, std::vector< BranchID > const &associationsFromSecondary)

Private Member Functions
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

Private Attributes
std::vector< ThinnedAssociationBranches >	vThinnedAssociationBranches_

Detailed Description

Author

W. David Dagenhart, created 11 June 2014

Definition at line 35 of file ThinnedAssociationsHelper.h.

Constructor & Destructor Documentation

edm::ThinnedAssociationsHelper::ThinnedAssociationsHelper

(

)

Definition at line 18 of file ThinnedAssociationsHelper.cc.

                                                       {
  }

Member Function Documentation

void edm::ThinnedAssociationsHelper::addAssociation	(	BranchID const &	parent,
		BranchID const &	association,
		BranchID const &	thinned
	)

Definition at line 56 of file ThinnedAssociationsHelper.cc.

References edm::ThinnedAssociationBranches::ThinnedAssociationBranches().

Referenced by edm::ThinningProducer< Collection, Selector >::registerThinnedAssociations().

                                                                          {
    vThinnedAssociationBranches_.push_back(ThinnedAssociationBranches(parent, association, thinned));
  }

void edm::ThinnedAssociationsHelper::addAssociation

(

ThinnedAssociationBranches const &

branches

)

Definition at line 62 of file ThinnedAssociationsHelper.cc.

                                                                                           {
    vThinnedAssociationBranches_.push_back(branches);
  }

std::vector< std::pair< BranchID, ThinnedAssociationBranches const * > > edm::ThinnedAssociationsHelper::associationToBranches

(

)

const

Definition at line 67 of file ThinnedAssociationsHelper.cc.

References selectAssociationProducts(), and groupFilesInBlocks::temp.

Referenced by initAssociationsFromSecondary(), and updateFromSecondaryInput().

                                                         {
    std::vector<std::pair<BranchID, ThinnedAssociationBranches const*> > temp;
    for(auto const& item : vThinnedAssociationBranches_) {
      temp.push_back(std::make_pair(item.association(), &item));
    }
    std::sort(temp.begin(), temp.end(), [](std::pair<BranchID, ThinnedAssociationBranches const*> const& x,
                                           std::pair<BranchID, ThinnedAssociationBranches const*> const& y)
                                          { return x.first < y.first; });
    return temp;
  }

std::vector< ThinnedAssociationBranches >::const_iterator edm::ThinnedAssociationsHelper::begin

(

void

)

const

Definition at line 22 of file ThinnedAssociationsHelper.cc.

                                         {
    return vThinnedAssociationBranches_.begin();
  }

void edm::ThinnedAssociationsHelper::clear ( void  )

inline

Definition at line 52 of file ThinnedAssociationsHelper.h.

Referenced by Vispa.Views.WidgetView.WidgetView::closeEvent(), Vispa.Views.BoxDecayView.BoxDecayView::closeEvent(), Vispa.Share.FindAlgorithm.FindAlgorithm::findUsingFindDialog(), Vispa.Views.LineDecayView.LineDecayView::setDataObjects(), Vispa.Views.WidgetView.WidgetView::setDataObjects(), BeautifulSoup.Tag::setString(), Vispa.Views.TreeView.TreeView::updateContent(), Vispa.Views.TableView.TableView::updateContent(), Vispa.Views.BoxDecayView.BoxDecayView::updateContent(), and Vispa.Views.PropertyView.PropertyView::updateContent().

{ vThinnedAssociationBranches_.clear(); }

std::vector<ThinnedAssociationBranches> const& edm::ThinnedAssociationsHelper::data ( ) const

inline

Definition at line 58 of file ThinnedAssociationsHelper.h.

References edm::ThinnedAssociationBranches::association(), and input.

Referenced by updateFromParentProcess(), and updateFromPrimaryInput().

{ return vThinnedAssociationBranches_; }

std::vector< ThinnedAssociationBranches >::const_iterator edm::ThinnedAssociationsHelper::end

(

void

)

const

Definition at line 27 of file ThinnedAssociationsHelper.cc.

Referenced by Types.LuminosityBlockRange::cppID(), and Types.EventRange::cppID().

                                       {
    return vThinnedAssociationBranches_.end();
  }

void edm::ThinnedAssociationsHelper::initAssociationsFromSecondary	(	std::vector< BranchID > const &	associationsFromSecondary,
		ThinnedAssociationsHelper const &	fileAssociationsHelper
	)

Definition at line 231 of file ThinnedAssociationsHelper.cc.

References edm::ThinnedAssociationBranches::association(), associationToBranches(), Exception, and edm::errors::LogicError.

                                                                                                                    {
    if(associationsFromSecondary.empty()) return;

    std::vector<std::pair<BranchID, ThinnedAssociationBranches const*> > assocToBranches =
      fileAssociationsHelper.associationToBranches();

    for(BranchID const& association : associationsFromSecondary) {

      auto branches = std::lower_bound(assocToBranches.begin(), assocToBranches.end(),
                                       std::make_pair(association, static_cast<ThinnedAssociationBranches const*>(nullptr)),
                                        [](std::pair<BranchID, ThinnedAssociationBranches const*> const& x,
                                           std::pair<BranchID, ThinnedAssociationBranches const*> const& y)
                                          { return x.first < y.first; });
      // This should never happen
      if(branches == assocToBranches.end() || branches->first != association) {
        throw edm::Exception(errors::LogicError,
                             "ThinnedAssociationHelper::initAssociationsFromSecondary could not find branches information, contact Framework developers");
      }
      addAssociation(*(branches->second));
    }
    sort();
  }

std::vector< ThinnedAssociationBranches >::const_iterator edm::ThinnedAssociationsHelper::parentBegin

(

BranchID const &

parent

)

const

Definition at line 32 of file ThinnedAssociationsHelper.cc.

References edmPickEvents::target.

Referenced by BareRootProductGetter::getThinnedProduct(), fwlite::DataGetterHelper::getThinnedProduct(), BareRootProductGetter::getThinnedProducts(), and fwlite::DataGetterHelper::getThinnedProducts().

                                                                     {
    ThinnedAssociationBranches target(parent, BranchID(), BranchID());
    return std::lower_bound(vThinnedAssociationBranches_.begin(), vThinnedAssociationBranches_.end(), target,
                            [](ThinnedAssociationBranches const& x,
                               ThinnedAssociationBranches const& y)
                            { return x.parent() < y.parent(); });
  }

std::vector< ThinnedAssociationBranches >::const_iterator edm::ThinnedAssociationsHelper::parentEnd

(

BranchID const &

parent

)

const

Definition at line 41 of file ThinnedAssociationsHelper.cc.

References edmPickEvents::target.

Referenced by BareRootProductGetter::getThinnedProduct(), fwlite::DataGetterHelper::getThinnedProduct(), BareRootProductGetter::getThinnedProducts(), and fwlite::DataGetterHelper::getThinnedProducts().

                                                                    {
    ThinnedAssociationBranches target(parent, BranchID(), BranchID());
    return std::upper_bound(vThinnedAssociationBranches_.begin(), vThinnedAssociationBranches_.end(), target,
                            [](ThinnedAssociationBranches const& x,
                               ThinnedAssociationBranches const& y)
                            { return x.parent() < y.parent(); });
  }

void edm::ThinnedAssociationsHelper::requireMatch

(

ThinnedAssociationBranches const &

input

)

const

Definition at line 158 of file ThinnedAssociationsHelper.cc.

References edm::ThinnedAssociationBranches::association(), mps_splice::entry, Exception, edm::errors::MismatchedInputFiles, edm::ThinnedAssociationBranches::parent(), and edm::ThinnedAssociationBranches::thinned().

                                                                                            {
    bool foundMatch = false;
    for(auto entry = parentBegin(input.parent()), iEnd = parentEnd(input.parent());
        entry != iEnd; ++entry) {
      if(entry->association() == input.association() &&
         entry->thinned() == input.thinned()) {
        foundMatch = true;
        break;
      }
    }
    if(!foundMatch) {
      throw edm::Exception(errors::MismatchedInputFiles, "ThinnedAssociationHelper::requireMatch, Illegal attempt to merge files with different ThinnedAssociations");
    }
  }

void edm::ThinnedAssociationsHelper::selectAssociationProducts	(	std::vector< BranchDescription const * > const &	associationDescriptions,
		std::set< BranchID > const &	keptProductsInEvent,
		std::map< BranchID, bool > &	keepAssociation
	)		const

Definition at line 80 of file ThinnedAssociationsHelper.cc.

References edm::ThinnedAssociationBranches::association().

Referenced by associationToBranches(), edm::SubProcess::selectProducts(), edm::OutputModule::selectProducts(), edm::global::OutputModuleBase::selectProducts(), edm::limited::OutputModuleBase::selectProducts(), and edm::one::OutputModuleBase::selectProducts().

                                                                           {

    keepAssociation.clear();
    // Copy the elements in vThinnedAssociationBranches_ into a vector sorted on
    // the association BranchID so we can do searches on that BranchID faster.
    std::vector<std::pair<BranchID, ThinnedAssociationBranches const*> > assocToBranches =
      associationToBranches();

    for(auto association : associationDescriptions) {
      if(association->isAlias()) { // There is no reason to configure an association product with an EDAlias (ignore and drop them if they exist)
        keepAssociation.insert(std::make_pair(association->branchID(), false));
      } else {
        std::set<BranchID> branchesInRecursion;
        shouldKeepAssociation(association->branchID(),
                              assocToBranches,
                              branchesInRecursion,
                              keptProductsInEvent,
                              keepAssociation);
      }
    }
  }

bool edm::ThinnedAssociationsHelper::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

private

Definition at line 104 of file ThinnedAssociationsHelper.cc.

References edm::ThinnedAssociationBranches::association(), Exception, edm::errors::LogicError, and edm::match().

                                                   {

    // If we already decided to keep or drop this one, then
    // return the same decision.
    auto decision = keepAssociation.find(association);
    if(decision != keepAssociation.end()) {
      return decision->second;
    }

    // Be careful not to fall into an infinite loop because
    // of a circular recursion.
    if(!branchesInRecursion.insert(association).second) {
      return false;
    }

    // If the thinned collection is being kept then keep the association
    auto branches = std::lower_bound(associationToBranches.begin(), associationToBranches.end(),
                                     std::make_pair(association, static_cast<ThinnedAssociationBranches const*>(nullptr)),
                                      [](std::pair<BranchID, ThinnedAssociationBranches const*> const& x,
                                         std::pair<BranchID, ThinnedAssociationBranches const*> const& y)
                                        { return x.first < y.first; });
    // This should never happen
    if(branches == associationToBranches.end() || branches->first != association) {
      throw edm::Exception(errors::LogicError, "ThinnedAssociationHelper::shouldKeepAssociation could not find branches information, contact Framework developers");
    }
    BranchID const& thinnedCollection = branches->second->thinned();
    if(keptProductsInEvent.find(thinnedCollection) != keptProductsInEvent.end()) {
      keepAssociation.insert(std::make_pair(association , true));
      return true;
    }
    // otherwise loop over any associations where the thinned collection
    // is also a parent collection and recursively examine those to see
    // if their thinned collections are being kept.
    auto iterEnd = parentEnd(thinnedCollection);
    for(auto match = parentBegin(thinnedCollection); match != iterEnd; ++match) {
      if(shouldKeepAssociation(match->association(),
                               associationToBranches,
                               branchesInRecursion,
                               keptProductsInEvent,
                               keepAssociation)) {
        keepAssociation.insert(std::make_pair(association , true));
        return true;
      }
    }
    // drop the association
    keepAssociation.insert(std::make_pair(association , false));
    return false;
  }

void edm::ThinnedAssociationsHelper::sort

(

)

Definition at line 49 of file ThinnedAssociationsHelper.cc.

References edm::ThinnedAssociationBranches::association(), and edm::ThinnedAssociationBranches::parent().

                                       {
    std::sort(vThinnedAssociationBranches_.begin(), vThinnedAssociationBranches_.end(),
              [](ThinnedAssociationBranches const& x,
                 ThinnedAssociationBranches const& y)
              { return x.parent() < y.parent() ? true : y.parent() < x.parent() ? false : x.association() < y.association(); });
  }

void edm::ThinnedAssociationsHelper::updateFromParentProcess	(	ThinnedAssociationsHelper const &	parentThinnedAssociationsHelper,
		std::map< BranchID, bool > const &	keepAssociation,
		std::map< BranchID::value_type, BranchID::value_type > const &	droppedBranchIDToKeptBranchID
	)

Definition at line 207 of file ThinnedAssociationsHelper.cc.

References hitfit::clear(), data(), edm::BranchID::id(), keep, edm::ThinnedAssociationBranches::parent(), and edm::ThinnedAssociationBranches::thinned().

                                                                                                                                          {
    clear();
    for(auto const& associationBranches : parentThinnedAssociationsHelper.data()) {
      auto keep = keepAssociation.find(associationBranches.association());
      if(keep != keepAssociation.end() && keep->second) {
        BranchID parent = associationBranches.parent();
        auto iter = droppedBranchIDToKeptBranchID.find(parent.id());
        if(iter != droppedBranchIDToKeptBranchID.end()) {
          parent = BranchID(iter->second);
        }
        BranchID thinned = associationBranches.thinned();
        iter = droppedBranchIDToKeptBranchID.find(thinned.id());
        if(iter != droppedBranchIDToKeptBranchID.end()) {
          thinned = BranchID(iter->second);
        }
        addAssociation(parent, associationBranches.association(), thinned);
      }
    }
    sort();
  }

void edm::ThinnedAssociationsHelper::updateFromPrimaryInput

(

ThinnedAssociationsHelper const &

helper

)

Definition at line 173 of file ThinnedAssociationsHelper.cc.

References data(), and idDealer::inputData.

Referenced by edm::StreamerInputSource::mergeIntoRegistry().

                                                                                                {
    if(vThinnedAssociationBranches_.empty()) {
      vThinnedAssociationBranches_ = helper.data();
      return;
    }
    std::vector<ThinnedAssociationBranches> const& inputData = helper.data();
    for (auto const& inputEntry : inputData) {
      requireMatch(inputEntry);
    }
  }

void edm::ThinnedAssociationsHelper::updateFromSecondaryInput	(	ThinnedAssociationsHelper const &	helper,
		std::vector< BranchID > const &	associationsFromSecondary
	)

Definition at line 184 of file ThinnedAssociationsHelper.cc.

References edm::ThinnedAssociationBranches::association(), associationToBranches(), Exception, and edm::errors::LogicError.

                                                                                                                 {
    if(associationsFromSecondary.empty()) return;

    std::vector<std::pair<BranchID, ThinnedAssociationBranches const*> > assocToBranches = helper.associationToBranches();

    for(BranchID const& association : associationsFromSecondary) {

      auto branches = std::lower_bound(assocToBranches.begin(), assocToBranches.end(),
                                       std::make_pair(association, static_cast<ThinnedAssociationBranches const*>(nullptr)),
                                        [](std::pair<BranchID, ThinnedAssociationBranches const*> const& x,
                                           std::pair<BranchID, ThinnedAssociationBranches const*> const& y)
                                          { return x.first < y.first; });
      // This should never happen
      if(branches == assocToBranches.end() || branches->first != association) {
        throw edm::Exception(errors::LogicError,
                           "ThinnedAssociationHelper::initAssociationsFromSecondary could not find branches information, contact Framework developers");
      }
      requireMatch(*(branches->second));
    }
  }

Member Data Documentation

std::vector<ThinnedAssociationBranches> edm::ThinnedAssociationsHelper::vThinnedAssociationBranches_

private

Definition at line 83 of file ThinnedAssociationsHelper.h.