]> git.tdb.fi Git - libs/datafile.git/blobdiff - source/collection.h
Use const sources in Collection
[libs/datafile.git] / source / collection.h
index 2243715b1d121d8fd64093f1a4b59b94e53e60e0..612a0af01879deaac213f1883db29df76d3cd57d 100644 (file)
@@ -17,17 +17,14 @@ namespace DataFile {
 Helper struct to determine whether a Loader has a Collection typedef.
 */
 template<typename T>
-struct NeedsCollection
+struct NeedsCollection: public Sfinae
 {
-       struct Yes { char c[2]; };
-       struct No { char c; };
-
        template<typename U>
        static Yes f(typename U::Collection *);
        template<typename U>
        static No f(...);
 
-       enum { value = (sizeof(f<T>(0))==sizeof(Yes)) };
+       enum { value = Evaluate<sizeof(f<T>(0))>::value };
 };
 
 class CollectionItemTypeBase;
@@ -45,6 +42,11 @@ method for details.
 
 Collections can have sources for loading objects on demand.  Automatic loading
 only works on a non-const Collection.  See class CollectionSource for details.
+
+As a last resort, a fallback collection can be designated for loading items
+that are not present.  Items retrieted from the fallback collection are shared
+between the collections, and are only deleted when all collections in the chain
+have been destroyed.
 */
 class Collection
 {
@@ -75,23 +77,23 @@ public:
                }
        };
 
-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;
+       typedef std::list<const CollectionSource *> SourceList;
 
        TypeList types;
        ItemMap items;
        SourceList sources;
+       Collection *fallback;
 
        Collection(const Collection &);
        Collection &operator=(const Collection &);
 public:
-       Collection() { }
+       Collection();
        virtual ~Collection();
 
        /** Adds an object into the collection.  The name must not pre-exist.  The
@@ -102,14 +104,24 @@ public:
                if(!item)
                        throw std::invalid_argument("Collection::add(item)");
 
-               insert_unique(items, name, RefPtr<typename RemoveConst<T>::Type>(item));
+               RefPtr<typename RemoveConst<T>::Type> ptr(item);
+               try
+               {
+                       insert_unique(items, name, ptr);
+               }
+               catch(...)
+               {
+                       // Avoid deleting the object
+                       ptr.release();
+                       throw;
+               }
        }
 
        /// Gets a typed object from the collection.
        template<typename T>
        T &get(const std::string &name) const
        {
-               return extract<T>(get_item(items, name));
+               return extract<typename RemoveConst<T>::Type>(get_item(items, name));
        }
 
        /** Gets a typed object from the collection.  If the name is not found,
@@ -118,7 +130,8 @@ public:
        template<typename T>
        T &get(const std::string &name)
        {
-               return extract<T>(get_var(name, get_type<T>()));
+               typedef typename RemoveConst<T>::Type NCT;
+               return extract<NCT>(get_var(name, get_type<NCT>()));
        }
 
 private:
@@ -128,30 +141,23 @@ private:
        T &extract(const Variant &var) const;
 
        template<typename T>
-       void collect_items(std::list<T *> *objects, std::list<std::string> *names, std::list<std::string> *future_names) const
+       std::list<T *> extract_list(const std::list<const Variant *> &vars) const
        {
-               typedef RefPtr<typename RemoveConst<T>::Type> RPNCT;
+               std::list<T *> result;
+               for(std::list<const Variant *>::const_iterator i=vars.begin(); i!=vars.end(); ++i)
+                       result.push_back(&extract<T>(**i));
+               return result;
+       }
 
-               for(ItemMap::const_iterator i=items.begin(); i!=items.end(); ++i)
-                       if(i->second.check_type<RPNCT>())
-                       {
-                               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)
-                               {
-                                       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);
-                               }
-                       }
+       void gather_items(std::list<const Variant *> *, std::list<std::string> *, const CollectionItemTypeBase &, bool) const;
+
+       template<typename T>
+       void gather_items(std::list<const Variant *> *vars, std::list<std::string> *names, const CollectionItemTypeBase *type, bool include_sources) const
+       {
+               if(type || (type = get_type<T>()))
+                       gather_items(vars, names, *type, include_sources);
+               else
+                       gather_items(vars, names, CollectionItemType<T>(), false);
        }
 
 public:
@@ -159,9 +165,9 @@ public:
        template<typename T>
        std::list<std::string> get_names() const
        {
-               std::list<std::string> result;
-               collect_items<T>(0, &result, 0);
-               return result;
+               std::list<std::string> names;
+               gather_items<typename RemoveConst<T>::Type>(0, &names, 0, false);
+               return names;
        }
 
        /** Returns a list of the names of objects of one type in the collection or
@@ -169,18 +175,18 @@ public:
        template<typename T>
        std::list<std::string> get_names()
        {
-               std::list<std::string> result;
-               collect_items<T>(0, &result, &result);
-               return result;
+               std::list<std::string> names;
+               gather_items<typename RemoveConst<T>::Type>(0, &names, 0, true);
+               return names;
        }
 
        /// Returns a list of objects of one type in the collection.
        template<typename T>
        std::list<T *> get_list() const
        {
-               std::list<T *> result;
-               collect_items<T>(&result, 0, 0);
-               return result;
+               std::list<const Variant *> vars;
+               gather_items<typename RemoveConst<T>::Type>(&vars, 0, 0, false);
+               return extract_list<T>(vars);
        }
 
        /** Returns a list of objects of one type, loading them from sources if
@@ -188,48 +194,40 @@ public:
        template<typename T>
        std::list<T *> get_list()
        {
-               std::list<T *> result;
-               std::list<std::string> future;
-               collect_items<T>(&result, 0, &future);
-               for(std::list<std::string>::iterator i=future.begin(); i!=future.end(); ++i)
-                       result.push_back(&get<T>(*i));
-               return result;
+               CollectionItemTypeBase *type = get_type<typename RemoveConst<T>::Type>();
+               if(type)
+                       load_items_from_sources(*type);
+
+               std::list<const Variant *> vars;
+               gather_items<typename RemoveConst<T>::Type>(&vars, 0, type, true);
+               return extract_list<T>(vars);
        }
 
 private:
+       unsigned get_status(const std::string &, const CollectionItemTypeBase &) const;
+
        template<typename T>
        unsigned get_status(const std::string &name) const
        {
-               ItemMap::const_iterator i = items.find(name);
-               if(i==items.end())
-               {
-                       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;
-               }
+               // XXX Should go through all applicable types
+               if(CollectionItemTypeBase *type = get_type<T>())
+                       return get_status(name, *type);
 
-               typedef RefPtr<typename RemoveConst<T>::Type> RPNCT;
-               if(!i->second.check_type<RPNCT>())
-                       return 0;
-
-               return 1;
+               ItemMap::const_iterator i = items.find(name);
+               return (i!=items.end() && i->second.check_type<RefPtr<T> >());
        }
 
 public:
        /// 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; }
+       { return get_status<typename RemoveConst<T>::Type>(name)==1; }
 
        /** 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; }
+       { return get_status<typename RemoveConst<T>::Type>(name)>0; }
 
        /// Returns the name of an item in the collection.
        template<typename T>
@@ -252,14 +250,40 @@ protected:
        template<typename T>
        CollectionItemType<T> &add_type();
 
+private:
        /// Returns the descriptor for a type, or null if one isn't defined.
        template<typename T>
        CollectionItemTypeBase *get_type() const;
 
+       /// Gets a descriptor with the same type as another descriptor.
+       CollectionItemTypeBase *get_type(const CollectionItemTypeBase &) const;
+
        /// Returns the descriptor for an item, or null if it's of an unknown type.
        CollectionItemTypeBase *get_type_for_item(const Variant &) const;
 
-       void add_source(CollectionSource &);
+protected:
+       /** Adds a source for automatically loading items.  Sources are consulted
+       in the order they are added. */
+       void add_source(const CollectionSource &);
+
+public:
+       /** Opens a raw resource, without interpreting it as object data.  Null is
+       returned if no such file is found.  The caller must dispose of the returned
+       object when done with it. */
+       IO::Seekable *open_raw(const std::string &) const;
+
+protected:
+       IO::Seekable *open_from_sources(const std::string &n) { return open_raw(n); }
+
+private:
+       void gather_names_from_sources(std::list<std::string> &, const CollectionItemTypeBase &) const;
+
+       void load_items_from_sources(const CollectionItemTypeBase &);
+
+protected:
+       /** Sets a fallback collection, which will be consulted if an item is not
+       found. */
+       void set_fallback(Collection *);
 };
 
 template<typename T>
@@ -307,6 +331,7 @@ public:
        const std::string &get_keyword() const { return kwd; }
        void add_suffix(const std::string &);
        bool match_name(const std::string &) const;
+       virtual bool is_same_type(const CollectionItemTypeBase &) const = 0;
        virtual bool check_item_type(const Variant &) const = 0;
        virtual void add_to_loader(Collection::Loader &) const = 0;
        virtual bool can_create() const = 0;
@@ -358,7 +383,7 @@ private:
                Creator(FuncPtr f): func(f) { }
 
                virtual T *create(Collection &coll, const std::string &name) const
-               { return (static_cast<C &>(coll).*func)(name); }
+               { return (dynamic_cast<C &>(coll).*func)(name); }
        };
 
        template<typename B>
@@ -421,6 +446,9 @@ public:
                return *this;
        }
 
+       virtual bool is_same_type(const CollectionItemTypeBase &other) const
+       { return dynamic_cast<const CollectionItemType<T> *>(&other); }
+
        virtual bool check_item_type(const Variant &var) const
        { return var.check_type<RefPtr<T> >(); }
 
@@ -444,8 +472,8 @@ public:
                RefPtr<T> obj = new T;
                Collection::ItemLoader<T> ldr(*obj, coll);
                ldr.load(parser);
-               // Collection::add will delete the object if it fails
-               coll.add(name, obj.release());
+               coll.add(name, obj.get());
+               obj.release();
        }
 };
 
@@ -453,14 +481,12 @@ public:
 template<typename T>
 T &Collection::extract(const Variant &var) const
 {
-       typedef RefPtr<typename RemoveConst<T>::Type> RPNCT;
-
-       if(!var.check_type<RPNCT>())
+       if(!var.check_type<RefPtr<T> >())
                if(CollectionItemTypeBase *type = get_type_for_item(var))
                        if(T *item = type->extract<T>(var))
                                return *item;
 
-       return *var.value<RPNCT>();
+       return *var.value<RefPtr<T> >();
 }
 
 template<typename T>
@@ -475,7 +501,7 @@ template<typename T>
 CollectionItemTypeBase *Collection::get_type() const
 {
        for(TypeList::const_iterator j=types.begin(); j!=types.end(); ++j)
-               if(dynamic_cast<CollectionItemType<typename RemoveConst<T>::Type> *>(*j))
+               if(dynamic_cast<CollectionItemType<T> *>(*j))
                        return *j;
        for(TypeList::const_iterator j=types.begin(); j!=types.end(); ++j)
                if((*j)->can_extract<T>())