Rework the list and containment queries

[libs/datafile.git] / source / collection.cpp

diff --git a/source/collection.cpp b/source/collection.cpp
index 2333aaf2726082a6214d2b950aa4570ebc1ea64b..fbe894a8bb4b073c0baa227655bd2d3ecf3781e3 100644 (file)
--- a/source/collection.cpp
+++ b/source/collection.cpp
@@ -35,6 +35,35 @@ const Variant &Collection::get_var(const string &name, const CollectionItemTypeB
        return get_item(items, name);
 }
 
+void Collection::gather_items(list<const Variant *> *vars, list<string> *names, const CollectionItemTypeBase &type, bool include_sources) const
+{
+       for(ItemMap::const_iterator i=items.begin(); i!=items.end(); ++i)
+               if(type.check_item_type(i->second))
+               {
+                       if(vars)
+                               vars->push_back(&i->second);
+                       if(names)
+                               names->push_back(i->first);
+               }
+
+       if(include_sources && names)
+               gather_names_from_sources(*names, type);
+}
+
+unsigned Collection::get_status(const string &name, const CollectionItemTypeBase &type) const
+{
+       ItemMap::const_iterator i = items.find(name);
+       if(i==items.end())
+       {
+               for(SourceList::const_iterator j=sources.begin(); j!=sources.end(); ++j)
+                       if((*j)->is_loadable(type, name))
+                               return 2;
+               return 0;
+       }
+
+       return type.check_item_type(i->second);
+}
+
 CollectionItemTypeBase *Collection::get_type_for_item(const Variant &var) const
 {
        for(TypeList::const_iterator i=types.begin(); i!=types.end(); ++i)
@@ -48,6 +77,28 @@ void Collection::add_source(CollectionSource &s)
        sources.push_back(&s);
 }
 
+void Collection::gather_names_from_sources(list<string> &names, const CollectionItemTypeBase &type) const
+{
+       for(SourceList::const_iterator i=sources.begin(); i!=sources.end(); ++i)
+       {
+               std::list<std::string> available_names = (*i)->get_names(type);
+               for(std::list<std::string>::iterator j=available_names.begin(); j!=available_names.end(); ++j)
+                       if(!items.count(*j))
+                               names.push_back(*j);
+       }
+}
+
+void Collection::load_items_from_sources(const CollectionItemTypeBase &type)
+{
+       for(SourceList::const_iterator i=sources.begin(); i!=sources.end(); ++i)
+       {
+               std::list<std::string> available_names = (*i)->get_names(type);
+               for(std::list<std::string>::iterator j=available_names.begin(); j!=available_names.end(); ++j)
+                       if(!items.count(*j))
+                               (*i)->load(*this, type, *j);
+       }
+}
+
 
 Collection::Loader::Loader(Collection &c):
        coll(c)