Add an API to open files from a collection's sources

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

diff --git a/source/collection.cpp b/source/collection.cpp
index e60b2009a4fc2c2cebf723ba700e5ab59967a00f..984fb74da8e7a6007645e6f85aa002b8ae6f894a 100644 (file)
--- a/source/collection.cpp
+++ b/source/collection.cpp
@@ -1,17 +1,130 @@
 #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)
+               {
+                       for(TypeList::const_iterator j=fallback->types.begin(); j!=fallback->types.end(); ++j)
+                               if((*j)->is_same_type(*type))
+                               {
+                                       if(fallback->get_status(name, **j))
+                                               return fallback->get_var(name, *j);
+                                       break;
+                               }
+               }
+       }
+
+       return get_item(items, name);
+}
+
+void Collection::gather_items(list<const Variant *> *vars, list<string> *names, const CollectionItemTypeBase &type, bool include_sources) const
 {
-       return items.count(n);
+       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)
+               if((*i)->check_item_type(var))
+                       return *i;
+       return 0;
+}
+
+void Collection::add_source(CollectionSource &s)
+{
+       sources.push_back(&s);
+}
+
+IO::Seekable *Collection::open_from_sources(const string &name)
+{
+       for(SourceList::const_iterator i=sources.begin(); i!=sources.end(); ++i)
+               if(IO::Seekable *io = (*i)->open(name))
+                       return io;
+
+       throw IO::file_not_found(name);
+}
+
+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);
+       }
+}
+
+void Collection::set_fallback(Collection *f)
+{
+       fallback = f;
 }
 
 
@@ -22,5 +135,32 @@ Collection::Loader::Loader(Collection &c):
                (*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