TheWarehouse Class Reference

TheWarehouse is a container for MooseObjects that allows querying/filtering over various customizeable attributes. More...

#include <TheWarehouse.h>

Classes
class	QueryCache
	QueryCache is a convenient way to construct and pass around (possible partially constructed) warehouse queries. More...

Public Types
template<typename T >
using	KeyType = typename T::Key
template<typename T >
using	AttribType = T *
using	Query = QueryCache<>

Public Member Functions
	TheWarehouse ()
	~TheWarehouse ()
template<typename T , typename... Args>
unsigned int	registerAttribute (const std::string &name, Args... dummy_args)
	registers a new "tracked" attribute of type T for the warehouse.
unsigned int	attribID (const std::string &name)
	Returns a unique ID associated with the given attribute name - i.e.
void	add (std::shared_ptr< MooseObject > obj)
	add adds a new object to the warehouse and stores attributes/metadata about it for running queries/filtering.
void	update (MooseObject *obj)
	update updates the metadata/attribute-info stored for the given object obj that must already exists in the warehouse.
void	update (MooseObject *obj, const Attribute &extra)
	update updates the metadata/attribute-info stored for the given object obj that must already exists in the warehouse.
Query	query ()
	query creates and returns an initialized a query object for querying objects from the warehouse.
std::size_t	count (const std::vector< std::unique_ptr< Attribute > > &conds)
	count returns the number of objects that match the provided query conditions.
template<typename T >
std::vector< T * > &	queryInto (const std::vector< std::unique_ptr< Attribute > > &conds, std::vector< T * > &results)
	queryInto takes the given conditions (i.e.
std::size_t	queryID (const std::vector< std::unique_ptr< Attribute > > &conds)
template<typename T >
std::vector< T * > &	queryInto (int query_id, std::vector< T * > &results, bool show_all=false)

Private Member Functions
int	prepare (std::vector< std::unique_ptr< Attribute > > conds)
	prepares a query and returns an associated query_id (i.e. for use with the query function).
const std::vector< MooseObject * > &	query (int query_id)
	callers of this function must lock _obj_cache_mutex as long as a reference to the returned vector is being used.
void	readAttribs (const MooseObject *obj, std::vector< std::unique_ptr< Attribute > > &attribs)

Private Attributes
std::unique_ptr< WarehouseStorage >	_store
std::vector< std::shared_ptr< MooseObject > >	_objects
std::unordered_map< MooseObject *, std::size_t >	_obj_ids
std::vector< std::vector< MooseObject * > >	_obj_cache
std::unordered_map< std::vector< std::unique_ptr< Attribute > >, int >	_query_cache
std::unordered_map< std::string, unsigned int >	_attrib_ids
std::vector< std::unique_ptr< Attribute > >	_attrib_list
std::mutex	_obj_mutex
std::mutex	_query_cache_mutex
std::mutex	_obj_cache_mutex

Detailed Description

TheWarehouse is a container for MooseObjects that allows querying/filtering over various customizeable attributes.

The meta-data about the objects is read/stored when the objects are added to the warehouse - updates to objects' state will not be reflected in query results unless the object is explicitly updated through the warehouse interface. The warehouse object can safely be queried concurrently from multiple threads.

Once Query and Attribute objects have been constructed, they are tied to the specific warehouse they were created with. They must not be used for different warehouses or the attribute ID they store internally will be wrong and that is bad.

Definition at line 186 of file TheWarehouse.h.

Member Typedef Documentation

AttribType

template<typename T >

using TheWarehouse::AttribType = T *

Definition at line 192 of file TheWarehouse.h.

KeyType

template<typename T >

using TheWarehouse::KeyType = typename T::Key

Definition at line 190 of file TheWarehouse.h.

Query

using TheWarehouse::Query = QueryCache<>

Definition at line 410 of file TheWarehouse.h.

Constructor & Destructor Documentation

TheWarehouse()

TheWarehouse::TheWarehouse

(

)

Definition at line 116 of file TheWarehouse.C.

: _store(std::make_unique<VecStore>()) {}

~TheWarehouse()

TheWarehouse::~TheWarehouse

(

)

Definition at line 117 of file TheWarehouse.C.

{}

Member Function Documentation

add()

void TheWarehouse::add

(

std::shared_ptr< MooseObject >

obj

)

add adds a new object to the warehouse and stores attributes/metadata about it for running queries/filtering.

The warehouse will maintain a pointer to the object indefinitely.

Definition at line 122 of file TheWarehouse.C.

{
  isValid(obj.get());
 
  std::size_t obj_id = 0;
  {
    std::lock_guard<std::mutex> lock(_obj_mutex);
 
    mooseAssert(!_obj_ids.count(obj.get()), obj->typeAndName() + " has already been added");
 
    _objects.push_back(obj);
    obj_id = _objects.size() - 1;
    _obj_ids[obj.get()] = obj_id;
 
    // reset/invalidate the query cache since query results may have been affected by this warehouse
    // insertion.
    _obj_cache.clear();
    _query_cache.clear();
  }
 
  std::vector<std::unique_ptr<Attribute>> attribs;
  readAttribs(obj.get(), attribs);
  _store->add(obj_id, std::move(attribs));
}

Referenced by NonlinearSystemBase::addBoundaryCondition(), NonlinearSystemBase::addDGKernel(), NonlinearSystemBase::addDiracKernel(), FEProblemBase::addFVGradientMethod(), FEProblemBase::addFVInterpolationMethod(), NonlinearSystemBase::addHDGKernel(), NonlinearSystemBase::addInterfaceKernel(), NonlinearSystemBase::addKernel(), NonlinearSystemBase::addNodalKernel(), FEProblemBase::addObject(), NonlinearSystemBase::addScalarKernel(), NonlinearSystemBase::addSplit(), and FEProblemBase::addUserObject().

attribID()

unsigned int TheWarehouse::attribID ( const std::string &  name )

inline

Returns a unique ID associated with the given attribute name - i.e.

an attribute and name that were previously registered via calls to registerAttribute. Users should generally not need to use this function.

Definition at line 444 of file TheWarehouse.h.

  {
    auto it = _attrib_ids.find(name);
    if (it != _attrib_ids.end())
      return it->second;
    mooseError("no ID exists for unregistered attribute '", name, "'");
  }

count()

std::size_t TheWarehouse::count

(

const std::vector< std::unique_ptr< Attribute > > &

conds

)

count returns the number of objects that match the provided query conditions.

This requires executing a full query operation (i.e. as if calling queryInto). A Query object should generally be used via the query() member function instead.

Definition at line 276 of file TheWarehouse.C.

{
  auto query_id = queryID(conds);
  std::lock_guard<std::mutex> lock(_obj_cache_mutex);
  auto & objs = query(query_id);
  std::size_t count = 0;
  for (auto obj : objs)
    if (obj->enabled())
      count++;
  return count;
}

Referenced by TheWarehouse::QueryCache< Attribs >::count().

prepare()

int TheWarehouse::prepare ( std::vector< std::unique_ptr< Attribute > >  conds )

private

prepares a query and returns an associated query_id (i.e. for use with the query function).

Definition at line 172 of file TheWarehouse.C.

{
  bool sort = false;
  std::unique_ptr<Attribute> sorted_attrib;
  if (!conds.empty() && dynamic_cast<AttribSorted *>(conds.back().get()))
  {
    sorted_attrib = std::move(conds.back());
    static const AttribSorted sorted_attrib_true(*this, true);
    sort = sorted_attrib->isMatch(sorted_attrib_true);
    // Remove the sorted condition temporarily
    conds.pop_back();
  }
 
#ifdef DEBUG
  for (auto & cond : conds)
    mooseAssert(!dynamic_cast<AttribSorted *>(cond.get()),
                "There should be no sorted attributes in this container.");
#endif
 
  auto obj_ids = _store->query(conds);
  if (sorted_attrib)
    conds.push_back(std::move(sorted_attrib));
 
  std::lock_guard<std::mutex> lock(_obj_cache_mutex);
  auto & vec = _obj_cache.emplace_back(obj_ids.size());
  const auto query_id = _obj_cache.size() - 1;
  {
    std::lock_guard<std::mutex> lock(_query_cache_mutex);
    _query_cache[std::move(conds)] = query_id;
  }
 
  std::lock_guard<std::mutex> o_lock(_obj_mutex);
  for (const auto i : index_range(obj_ids))
  {
    auto obj = _objects[obj_ids[i]].get();
    mooseAssert(std::find(vec.begin(), vec.end(), obj) == vec.end(), "Duplicate object");
    vec[i] = obj;
  }
 
  if (sort && !vec.empty() && dynamic_cast<DependencyResolverInterface *>(vec[0]))
  {
    std::vector<DependencyResolverInterface *> dependers;
    for (auto obj : vec)
    {
      auto d = dynamic_cast<DependencyResolverInterface *>(obj);
      if (!d)
      {
        dependers.clear();
        break;
      }
      dependers.push_back(d);
    }
 
    try
    {
      DependencyResolverInterface::sort(dependers);
    }
    catch (CyclicDependencyException<DependencyResolverInterface *> & e)
    {
      DependencyResolverInterface::cyclicDependencyError<MooseObject *>(
          e,
          "Cyclic dependency detected in object ordering",
          [](DependencyResolverInterface * obj)
          {
            auto * moose_obj = dynamic_cast<MooseObject *>(obj);
            mooseAssert(moose_obj, "Failed to cast dependency object to MooseObject");
            return moose_obj->name();
          });
    }
 
    mooseAssert(dependers.size() == vec.size(), "Dependency resolution size mismatch");
    for (unsigned int i = 0; i < dependers.size(); i++)
      vec[i] = dynamic_cast<MooseObject *>(dependers[i]);
  }
 
  return query_id;
}

Referenced by queryID().

query() [1/2]

Query TheWarehouse::query ( )

inline

query creates and returns an initialized a query object for querying objects from the warehouse.

Definition at line 467 of file TheWarehouse.h.

{ return Query(*this); }

Referenced by CheckFVBCAction::act(), EXTERN_C_END::appendFVSetupObjects(), NonlinearSystemBase::checkKernelCoverage(), FEProblemBase::checkUserObjectJacobianRequirement(), FEProblemBase::checkUserObjects(), ComputeResidualAndJacobianThread::compute(), NonlinearSystemBase::computeJacobianInternal(), NonlinearSystemBase::computeResidualAndJacobianInternal(), NonlinearSystemBase::computeResidualInternal(), LinearSystem::containsTimeKernel(), count(), FEProblemBase::customSetup(), ComputeResidualAndJacobianThread::determineObjectWarehouses(), ComputeResidualThread::determineObjectWarehouses(), MFEMProblem::executeMFEMObjects(), FEProblemBase::executeSamplers(), ComputeLinearFVElementalThread::fetchBlockSystemContributionObjects(), ComputeLinearFVFaceThread::fetchBlockSystemContributionObjects(), FEProblemBase::getDistribution(), FEProblemBase::getFVGradientMethod(), FEProblemBase::getFVInterpolationMethod(), FEProblemBase::getKokkosUserObject(), MFEMProblem::getMFEMObject(), FEProblemBase::getMortarUserObjects(), FEProblemBase::getPositionsObject(), FEProblemBase::getPostprocessorObjectByName(), FEProblemBase::getSampler(), CompositionDT::getTimeSteppers(), FEProblemBase::getUOQuery(), FEProblemBase::getUserObject(), FEProblemBase::getUserObjectBase(), FEProblemBase::getVectorPostprocessorObjectByName(), FEProblemBase::hasDistribution(), FEProblemBase::hasFVGradientMethod(), FEProblemBase::hasFVInterpolationMethod(), MFEMProblem::hasMFEMObject(), FEProblemBase::hasUserObject(), VectorPostprocessorInterface::hasVectorPostprocessorByName(), MFEMProblem::initialSetup(), BoundaryLinearFVFluxIntegral::initialSetup(), SideFVFluxBCIntegral::initialSetup(), LinearSystem::initialSetup(), ExplicitTimeIntegrator::initialSetup(), AdvancedOutput::initPostprocessorOrVectorPostprocessorLists(), MooseVariableDataFV< OutputType >::MooseVariableDataFV(), FEProblemBase::needBoundaryMaterialOnSide(), FEProblemBase::needInterfaceMaterialOnSide(), FEProblemBase::needInternalNeighborSideMaterial(), JSONOutput::outputReporters(), BlockRestrictionDebugOutput::printBlockRestrictionGroups(), BlockRestrictionDebugOutput::printBlockRestrictionMap(), BlockRestrictionDebugOutput::printBoundaryRestrictionGroups(), queryInto(), ComputeLinearFVElementalThread::setupSystemContributionObjects(), ComputeLinearFVFaceThread::setupSystemContributionObjects(), and FEProblemBase::timestepSetup().

query() [2/2]

const std::vector< MooseObject * > & TheWarehouse::query ( int  query_id )

private

callers of this function must lock _obj_cache_mutex as long as a reference to the returned vector is being used.

Definition at line 251 of file TheWarehouse.C.

{
  if (static_cast<std::size_t>(query_id) >= _obj_cache.size())
    throw std::runtime_error("unknown query id");
  return _obj_cache[query_id];
}

queryID()

std::size_t TheWarehouse::queryID

(

const std::vector< std::unique_ptr< Attribute > > &

conds

)

Definition at line 259 of file TheWarehouse.C.

{
  {
    std::lock_guard<std::mutex> lock(_query_cache_mutex);
    auto it = _query_cache.find(conds);
    if (it != _query_cache.end())
      return it->second;
  }
 
  std::vector<std::unique_ptr<Attribute>> conds_clone;
  conds_clone.resize(conds.size());
  for (std::size_t i = 0; i < conds.size(); i++)
    conds_clone[i] = conds[i]->clone();
  return prepare(std::move(conds_clone));
}

Referenced by count(), JSONOutput::outputReporters(), queryInto(), and TheWarehouse::QueryCache< Attribs >::queryIntoHelper().

queryInto() [1/2]

template<typename T >

std::vector< T * > & TheWarehouse::queryInto ( const std::vector< std::unique_ptr< Attribute > > &  conds, std::vector< T * > &  results  )

inline

queryInto takes the given conditions (i.e.

Attributes holding the values to filter/match over) and filters all objects in the warehouse that match all conditions (i.e. "and"ing the conditions together) and stores them in the results vector. All result objects must be castable to the templated type T. This function filters out disabled objects on the fly - only returning enabled ones.

Definition at line 478 of file TheWarehouse.h.

  {
    return queryInto(queryID(conds), results);
  }

Referenced by JSONOutput::outputReporters(), queryInto(), and TheWarehouse::QueryCache< Attribs >::queryIntoHelper().

queryInto() [2/2]

template<typename T >

std::vector< T * > & TheWarehouse::queryInto ( int  query_id, std::vector< T * > &  results, bool  show_all = false  )

inline

Definition at line 487 of file TheWarehouse.h.

  {
    std::lock_guard<std::mutex> lock(_obj_cache_mutex);
    auto & objs = query(query_id);
    results.clear();
    results.reserve(objs.size());
    for (auto & obj : objs)
    {
      mooseAssert(obj, "Null object");
      T * cast_obj;
      // We've been using dynamic_cast plus an assert that it
      // succeeds, but if we know the correct type T then often we can
      // use a much-cheaper static_cast. cast_ptr will still
      // use a dynamic_cast + assertion in devel/dbg modes.
      if constexpr (std::is_convertible_v<MooseObject *, std::remove_cv_t<T> *> ||
                    std::is_convertible_v<std::remove_cv_t<T> *, MooseObject *>)
      {
        cast_obj = cast_ptr<T *>(obj);
      }
      // Side-casts or virtual inheritence are always expensive.
      // Someone should figure out how to get rid of this code path.
      else
      {
        cast_obj = dynamic_cast<T *>(obj);
        mooseAssert(cast_obj,
                    "Queried object " + obj->typeAndName() + " has incompatible c++ type with " +
                        MooseUtils::prettyCppType<T>());
      }
 
      mooseAssert(std::find(results.begin(), results.end(), cast_obj) == results.end(),
                  "Duplicate object");
      if (show_all || obj->enabled())
        results.push_back(cast_obj);
    }
    return results;
  }

readAttribs()

void TheWarehouse::readAttribs ( const MooseObject *  obj, std::vector< std::unique_ptr< Attribute > > &  attribs  )

private

Definition at line 289 of file TheWarehouse.C.

{
  for (auto & ref : _attrib_list)
  {
    attribs.emplace_back(ref->clone());
    attribs.back()->initFrom(obj);
  }
}

Referenced by add(), and update().

registerAttribute()

template<typename T , typename... Args>

unsigned int TheWarehouse::registerAttribute ( const std::string &  name, Args...  dummy_args  )

inline

registers a new "tracked" attribute of type T for the warehouse.

args are all arguments necessary to create an instance of the T class excluding the warehouse reference/pointer which is assumed to be first and automatically inserted. An instance of every registered attribute will be created for and initialized to each object added to the warehouse allowing queries to be executed over specific values the attribute may take on. Attributes must be registered before objects are added to the warehouse. A unique ID associated with the registered attribute is returned - which is generally not needed used by users.

As an example, to register a class with the constructor "YourAttribute(TheWarehouse& w, int foo)", you would call "registerAttribute<YourAttribute>("your_attrib_name", constructor_arg1, ...)". Custom attribute classes are required to pass an attribute name (i.e. "your_attrib_name") to the Attribute base class. The dummy args are forwarded to the attrib class' constructor. The name passed here into registerAttribute must be the same string as the name passed to the Attribute base class's constructor.

Definition at line 430 of file TheWarehouse.h.

  {
    auto it = _attrib_ids.find(name);
    if (it != _attrib_ids.end())
      return it->second;
 
    _attrib_ids[name] = _attrib_list.size();
    _attrib_list.push_back(std::unique_ptr<Attribute>(new T(*this, dummy_args...)));
    return _attrib_list.size() - 1;
  }

update() [1/2]

void TheWarehouse::update

(

MooseObject *

obj

)

update updates the metadata/attribute-info stored for the given object obj that must already exists in the warehouse.

Call this if an object's state has changed in such a way that its warehouse attributes have become stale/incorrect.

Definition at line 160 of file TheWarehouse.C.

{
  std::vector<std::unique_ptr<Attribute>> attribs;
  readAttribs(obj, attribs);
  _store->set(_obj_ids[obj], std::move(attribs));
  // reset/invalidate the query cache since query results may have been affected by this object
  // attribute modification.
  _obj_cache.clear();
  _query_cache.clear();
}

Referenced by groupUserObjects().

update() [2/2]

void TheWarehouse::update	(	MooseObject *	obj,
		const Attribute &	extra
	)

update updates the metadata/attribute-info stored for the given object obj that must already exists in the warehouse.

Call this if an object's state has changed in such a way that its warehouse attributes have become stale/incorrect. Any attribute specified in extra overwrites/trumps one read from the object's current state.

Definition at line 148 of file TheWarehouse.C.

{
  std::vector<std::unique_ptr<Attribute>> attribs;
  attribs.push_back(extra.clone());
  _store->set(_obj_ids[obj], std::move(attribs));
  // reset/invalidate the query cache since query results may have been affected by this object
  // attribute modification.
  _obj_cache.clear();
  _query_cache.clear();
}

Member Data Documentation

_attrib_ids

std::unordered_map<std::string, unsigned int> TheWarehouse::_attrib_ids

private

Definition at line 546 of file TheWarehouse.h.

Referenced by attribID(), and registerAttribute().

_attrib_list

std::vector<std::unique_ptr<Attribute> > TheWarehouse::_attrib_list

private

Definition at line 547 of file TheWarehouse.h.

Referenced by readAttribs(), and registerAttribute().

_obj_cache

std::vector<std::vector<MooseObject *> > TheWarehouse::_obj_cache

private

Definition at line 540 of file TheWarehouse.h.

Referenced by add(), prepare(), query(), update(), and update().

_obj_cache_mutex

std::mutex TheWarehouse::_obj_cache_mutex

private

Definition at line 551 of file TheWarehouse.h.

Referenced by count(), prepare(), and queryInto().

_obj_ids

std::unordered_map<MooseObject *, std::size_t> TheWarehouse::_obj_ids

private

Definition at line 536 of file TheWarehouse.h.

Referenced by add(), update(), and update().

_obj_mutex

std::mutex TheWarehouse::_obj_mutex

private

Definition at line 549 of file TheWarehouse.h.

Referenced by add(), and prepare().

_objects

std::vector<std::shared_ptr<MooseObject> > TheWarehouse::_objects

private

Definition at line 535 of file TheWarehouse.h.

Referenced by add(), and prepare().

_query_cache

std::unordered_map<std::vector<std::unique_ptr<Attribute> >, int> TheWarehouse::_query_cache

private

Definition at line 544 of file TheWarehouse.h.

Referenced by add(), prepare(), queryID(), update(), and update().

_query_cache_mutex

std::mutex TheWarehouse::_query_cache_mutex

private

Definition at line 550 of file TheWarehouse.h.

Referenced by prepare(), and queryID().

_store

std::unique_ptr<WarehouseStorage> TheWarehouse::_store

private

Definition at line 534 of file TheWarehouse.h.

Referenced by add(), prepare(), update(), and update().

---

The documentation for this class was generated from the following files:

• include/base/TheWarehouse.h
• src/base/TheWarehouse.C