+#include <set>
#include "collection.h"
+using namespace std;
+
namespace Msp {
namespace DataFile {
+Collection::Collection():
+ fallback(0)
+{ }
+
Collection::~Collection()
{
for(TypeList::iterator i = types.begin(); i!=types.end(); ++i)
delete *i;
}
-bool Collection::contains(const std::string &n) const
+const Variant &Collection::get_var(const string &name, const CollectionItemTypeBase *type)
+{
+ ItemMap::iterator i = items.find(name);
+ if(i!=items.end())
+ return i->second;
+
+ if(type)
+ {
+ bool loaded = false;
+ if(type->can_create())
+ {
+ type->create_item(*this, name);
+ loaded = items.count(name);
+ }
+ for(SourceList::iterator j=sources.begin(); (!loaded && j!=sources.end()); ++j)
+ {
+ (*j)->load(*this, *type, name);
+ loaded = items.count(name);
+ }
+ if(!loaded && fallback)
+ if(CollectionItemTypeBase *fb_type = fallback->get_type(*type))
+ if(fallback->get_status(name, *fb_type))
+ return fallback->get_var(name, fb_type);
+ }
+
+ 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;
+ if(fallback)
+ if(CollectionItemTypeBase *fb_type = fallback->get_type(type))
+ return fallback->get_status(name, *fb_type);
+ return 0;
+ }
+
+ return type.check_item_type(i->second);
+}
+
+CollectionItemTypeBase *Collection::get_type(const CollectionItemTypeBase &type) const
+{
+ for(TypeList::const_iterator j=types.begin(); j!=types.end(); ++j)
+ if((*j)->is_same_type(type))
+ return *j;
+ return 0;
+}
+
+CollectionItemTypeBase *Collection::get_type_for_item(const Variant &var) const
+{
+ for(TypeList::const_iterator i=types.begin(); i!=types.end(); ++i)
+ if((*i)->check_item_type(var))
+ return *i;
+ return 0;
+}
+
+void Collection::add_source(const CollectionSource &s)
+{
+ sources.push_back(&s);
+}
+
+IO::Seekable *Collection::open_raw(const string &name) const
{
- return items.count(n);
+ for(SourceList::const_iterator i=sources.begin(); i!=sources.end(); ++i)
+ if(IO::Seekable *io = (*i)->open(name))
+ return io;
+
+ return 0;
+}
+
+void Collection::gather_names_from_sources(list<string> &names, const CollectionItemTypeBase &type) const
+{
+ set<string> new_names;
+ 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))
+ new_names.insert(*j);
+ }
+ names.insert(names.end(), new_names.begin(), new_names.end());
+}
+
+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))
+ {
+ bool loaded = false;
+ if(type.can_create())
+ {
+ type.create_item(*this, *j);
+ loaded = items.count(*j);
+ }
+ if(!loaded)
+ (*i)->load(*this, type, *j);
+ }
+ }
+}
+
+void Collection::set_fallback(Collection *f)
+{
+ fallback = f;
}
(*i)->add_to_loader(*this);
}
+
+CollectionItemTypeBase::~CollectionItemTypeBase()
+{
+ for(vector<ExtractorBase *>::iterator i=extractors.begin(); i!=extractors.end(); ++i)
+ delete *i;
+}
+
+void CollectionItemTypeBase::set_keyword(const string &k)
+{
+ kwd = k;
+ if(suffixes.empty())
+ add_suffix("."+kwd);
+}
+
+void CollectionItemTypeBase::add_suffix(const string &s)
+{
+ suffixes.push_back(s);
+}
+
+bool CollectionItemTypeBase::match_name(const string &name) const
+{
+ for(vector<string>::const_iterator i=suffixes.begin(); i!=suffixes.end(); ++i)
+ if(name.size()>i->size() && !name.compare(name.size()-i->size(), string::npos, *i))
+ return true;
+ return false;
+}
+
} // namespace DataFile
} // namespace Msp