]> git.tdb.fi Git - libs/datafile.git/blobdiff - source/collection.h
Move most of Collection::get implementation to collection.cpp
[libs/datafile.git] / source / collection.h
index 0595aaca7caddb801d094d9ed847362c669c8be7..be9ae88c491595f925db46e30aeff2daa67deb5f 100644 (file)
@@ -4,8 +4,12 @@
 #include <msp/core/maputils.h>
 #include <msp/core/meta.h>
 #include <msp/core/refptr.h>
+#include "collectionsource.h"
 #include "loader.h"
 
+/* XXX This file is a big mess with too many things in it.  However, the
+dependencies between those things make it difficult to split up. */
+
 namespace Msp {
 namespace DataFile {
 
@@ -39,9 +43,8 @@ While this class can be instantiated by itself and used for storing objects,
 loading requires that a subclass defines the supported types.  See the add_type
 method for details.
 
-Collections also support a notion of "future objects".  These are objects which
-are known to be possible to load, but loading them is deferred to the first
-time they are requested.
+Collections can have sources for loading objects on demand.  Automatic loading
+only works on a non-const Collection.  See class CollectionSource for details.
 */
 class Collection
 {
@@ -55,40 +58,35 @@ public:
                template<typename T> friend class CollectionItemType;
 
        private:
-               template<typename T, typename S, bool = NeedsCollection<typename T::Loader>::value>
-               struct Add;
-
                Collection &coll;
 
        public:
                Loader(Collection &);
                Collection &get_object() const { return coll; }
        private:
-               template<typename T, typename S, typename C>
-               void coll_item(const std::string &n)
-               {
-                       RefPtr<T> it = new T;
-                       load_sub(*it, dynamic_cast<C &>(coll));
-                       coll.add<S>(n, it.get());
-                       it.release();
-               }
-
                template<typename T, typename S>
                void item(const std::string &n)
                {
                        RefPtr<T> it = new T;
-                       load_sub(*it);
+                       ItemLoader<T> ldr(*it, coll);
+                       load_sub_with(ldr);
                        coll.add<S>(n, it.get());
                        it.release();
                }
        };
 
+protected:
+       template<typename T, bool = NeedsCollection<typename T::Loader>::value>
+       class ItemLoader;
+
 private:
        typedef std::map<std::string, Variant> ItemMap;
        typedef std::list<CollectionItemTypeBase *> TypeList;
+       typedef std::list<CollectionSource *> SourceList;
 
        TypeList types;
        ItemMap items;
+       SourceList sources;
 
        Collection(const Collection &);
        Collection &operator=(const Collection &);
@@ -104,82 +102,63 @@ public:
                if(!item)
                        throw std::invalid_argument("Collection::add(item)");
 
-               typedef RefPtr<typename RemoveConst<T>::Type> RPNCT;
-
-               ItemMap::iterator i = items.find(name);
-               if(i!=items.end())
-               {
-                       if(i->second.check_type<RPNCT>())
-                       {
-                               // Replace a future object placeholder
-                               RPNCT &ptr = i->second.value<RPNCT>();
-                               if(!ptr)
-                               {
-                                       ptr = item;
-                                       return;
-                               }
-                       }
-
-                       throw key_error(typeid(ItemMap));
-               }
-
-               items.insert(ItemMap::value_type(name, RPNCT(item)));
+               insert_unique(items, name, RefPtr<typename RemoveConst<T>::Type>(item));
        }
 
-protected:
-       /** Adds the name of a future object to the collection.  The object itself
-       will be loaded on first access.  The calling subclass should be prepared to
-       create the object on request. */
+       /// Gets a typed object from the collection.
        template<typename T>
-       void add_future(const std::string &name)
+       T &get(const std::string &name) const
        {
-               RefPtr<typename RemoveConst<T>::Type> ptr(0);
-               insert_unique(items, name, ptr);
+               return extract<T>(get_item(items, name));
        }
 
-public:
-       /// Gets a typed object from the collection.
+       /** Gets a typed object from the collection.  If the name is not found,
+       automatic creation with the type's creator function (if defined) or from
+       sources (if present) is attempted. */
        template<typename T>
-       T &get(const std::string &name) const
+       T &get(const std::string &name)
        {
-               typedef typename RemoveConst<T>::Type NCT;
-
-               T *ptr = get_item(items, name).value<RefPtr<NCT> >();
-               if(!ptr)
-                       throw key_error(typeid(ItemMap));
-               return *ptr;
+               return extract<T>(get_var(name, get_type<T>()));
        }
 
-       /** Gets a typed object from the collection.  If the name is not found in
-       and a creator for the item type is defined, it is invoked. */
+private:
+       const Variant &get_var(const std::string &, const CollectionItemTypeBase *);
+
        template<typename T>
-       T &get(const std::string &);
+       T &extract(const Variant &var) const
+       {
+               return *var.value<RefPtr<typename RemoveConst<T>::Type> >();
+       }
 
-private:
        template<typename T>
-       void collect_items(std::list<T *> *objects, std::list<std::string> *names, std::list<std::string> *future_names)
+       void collect_items(std::list<T *> *objects, std::list<std::string> *names, std::list<std::string> *future_names) const
        {
                typedef RefPtr<typename RemoveConst<T>::Type> RPNCT;
 
                for(ItemMap::const_iterator i=items.begin(); i!=items.end(); ++i)
                        if(i->second.check_type<RPNCT>())
                        {
-                               T *ptr = i->second.value<RPNCT>().get();
-                               if(ptr)
+                               if(objects)
+                                       objects->push_back(i->second.value<RPNCT>().get());
+                               if(names)
+                                       names->push_back(i->first);
+                       }
+
+               if(future_names)
+                       if(CollectionItemTypeBase *type = get_type<T>())
+                       {
+                               for(SourceList::const_iterator i=sources.begin(); i!=sources.end(); ++i)
                                {
-                                       if(objects)
-                                               objects->push_back(ptr);
-                                       if(names)
-                                               names->push_back(i->first);
+                                       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))
+                                                       future_names->push_back(*j);
                                }
-                               else if(future_names)
-                                       future_names->push_back(i->first);
                        }
        }
 
 public:
-       /** Returns a list of the names of loaded objects of one type in the
-       collection. */
+       /// Returns a list of the names of objects of one type in the collection.
        template<typename T>
        std::list<std::string> get_names() const
        {
@@ -188,8 +167,8 @@ public:
                return result;
        }
 
-       /** Returns a list of the names of objects of one type in the collection,
-       including any future objects. */
+       /** Returns a list of the names of objects of one type in the collection or
+       available from sources. */
        template<typename T>
        std::list<std::string> get_names()
        {
@@ -198,7 +177,7 @@ public:
                return result;
        }
 
-       /// Returns a list of loaded objects of one type in the collection.
+       /// Returns a list of objects of one type in the collection.
        template<typename T>
        std::list<T *> get_list() const
        {
@@ -207,8 +186,8 @@ public:
                return result;
        }
 
-       /** Returns a list of objects of one type in the collection.  Any future
-       objects of that type are loaded and returned in the list. */
+       /** Returns a list of objects of one type, loading them from sources if
+       necessary. */
        template<typename T>
        std::list<T *> get_list()
        {
@@ -226,24 +205,31 @@ private:
        {
                ItemMap::const_iterator i = items.find(name);
                if(i==items.end())
-                       return false;
+               {
+                       if(CollectionItemTypeBase *type = get_type<T>())
+                       {
+                               for(SourceList::const_iterator j=sources.begin(); j!=sources.end(); ++j)
+                                       if((*j)->is_loadable(*type, name))
+                                               return 2;
+                       }
+                       return 0;
+               }
 
                typedef RefPtr<typename RemoveConst<T>::Type> RPNCT;
                if(!i->second.check_type<RPNCT>())
-                       return false;
+                       return 0;
 
-               T *ptr = i->second.value<RPNCT>().get();
-               return ptr ? 1 : 2;
+               return 1;
        }
 
 public:
-       /// Checks whether a typed object exists and is loaded in the collection.
+       /// Checks whether a typed object exists in the collection.
        template<typename T>
        bool contains(const std::string &name) const
        { return get_status<T>(name)==1; }
 
-       /** Checks whether a typed object exists in the collection, as either a
-       loaded or future object. */
+       /** Checks whether a typed object exists in the collection or is loadable
+       from a source. */
        template<typename T>
        bool contains(const std::string &name)
        { return get_status<T>(name)>0; }
@@ -268,21 +254,30 @@ protected:
        can be used to define how objects of that type can be loaded. */
        template<typename T>
        CollectionItemType<T> &add_type();
-};
 
+       /** Returns the descriptor for a type, or null if one isn't defined. */
+       template<typename T>
+       CollectionItemTypeBase *get_type() const;
+
+       void add_source(CollectionSource &);
+};
 
-template<typename T, typename S>
-struct Collection::Loader::Add<T, S, false>
+template<typename T>
+class Collection::ItemLoader<T, false>: public T::Loader
 {
-       static void add(Loader &loader, const std::string &kwd)
-       { loader.add(kwd, &Loader::item<T, S>); }
+public:
+       ItemLoader(T &o, Collection &):
+               T::Loader(o)
+       { }
 };
 
-template<typename T, typename S>
-struct Collection::Loader::Add<T, S, true>
+template<typename T>
+class Collection::ItemLoader<T, true>: public T::Loader
 {
-       static void add(Loader &loader, const std::string &kwd)
-       { loader.add(kwd, &Loader::coll_item<T, S, typename T::Loader::Collection>); }
+public:
+       ItemLoader(T &o, Collection &c):
+               T::Loader(o, dynamic_cast<typename T::Loader::Collection &>(c))
+       { }
 };
 
 
@@ -302,15 +297,21 @@ protected:
        { };
 
        std::string kwd;
+       std::vector<std::string> suffixes;
        TagBase *tag;
 
        CollectionItemTypeBase();
 public:
        virtual ~CollectionItemTypeBase();
 
+       void set_keyword(const std::string &);
+       const std::string &get_keyword() const { return kwd; }
+       void add_suffix(const std::string &);
+       bool match_name(const std::string &) const;
        virtual void add_to_loader(Collection::Loader &) const = 0;
        virtual bool can_create() const = 0;
        virtual void create_item(Collection &, const std::string &) const = 0;
+       virtual void load_item(Collection &, Parser &, const std::string &) const = 0;
 
        template<typename T>
        bool check_type() const
@@ -372,7 +373,7 @@ private:
                { coll.add(name, static_cast<S *>(obj)); }
 
                virtual void add_to_loader(Collection::Loader &loader, const std::string &kwd)
-               { Collection::Loader::Add<T, S>::add(loader, kwd); }
+               { loader.add(kwd, &Collection::Loader::item<T, S>); }
        };
 
        CreatorBase *creat;
@@ -394,7 +395,17 @@ public:
        item's name. */
        CollectionItemType &keyword(const std::string &k)
        {
-               kwd = k;
+               set_keyword(k);
+               return *this;
+       }
+
+       /** Adds a suffix that is used to match names when looking for future
+       objects.  There is no implied separator; a name matches if it ends with the
+       suffix.  If a keyword is defined before any suffixes, then "."+keyword is
+       added as a suffix. */
+       CollectionItemType &suffix(const std::string &s)
+       {
+               add_suffix(s);
                return *this;
        }
 
@@ -438,28 +449,17 @@ public:
                if(obj)
                        store->store(coll, name, obj);
        }
-};
-
-
-template<typename T>
-T &Collection::get(const std::string &name)
-{
-       typedef typename RemoveConst<T>::Type NCT;
 
-       ItemMap::iterator i = items.find(name);
-       if(i!=items.end())
+       virtual void load_item(Collection &coll, Parser &parser, const std::string &name) const
        {
-               NCT *ptr = i->second.value<RefPtr<NCT> >().get();
-               if(ptr)
-                       return *ptr;
+               RefPtr<T> obj = new T;
+               Collection::ItemLoader<T> ldr(*obj, coll);
+               ldr.load(parser);
+               store->store(coll, name, obj.get());
+               obj.release();
        }
+};
 
-       for(TypeList::iterator j=types.begin(); j!=types.end(); ++j)
-               if((*j)->can_create() && (*j)->check_type<NCT>())
-                       (*j)->create_item(*this, name);
-
-       return *get_item(items, name).value<RefPtr<NCT> >();
-}
 
 template<typename T>
 CollectionItemType<T> &Collection::add_type()
@@ -469,6 +469,15 @@ CollectionItemType<T> &Collection::add_type()
        return *type;
 }
 
+template<typename T>
+CollectionItemTypeBase *Collection::get_type() const
+{
+       for(TypeList::const_iterator j=types.begin(); j!=types.end(); ++j)
+               if((*j)->check_type<typename RemoveConst<T>::Type>())
+                       return *j;
+       return 0;
+}
+
 } // namespace DataFile
 } // namespace Msp