#ifndef ASSETIARCHIVE_H
#define ASSETIARCHIVE_H

#include <istream>
#include <boost/archive/archive_exception.hpp>
#include <boost/archive/text_iarchive.hpp>
#include <boost/archive/detail/archive_pointer_iserializer.hpp>
#include <boost/type_traits.hpp>

template< class AssetManagerType, class AssetType >
class asset_iarchive : public boost::archive::text_iarchive_impl< asset_iarchive<AssetManagerType, AssetType> >
{
    friend class CubesDemo;
    typedef boost::archive::text_iarchive_impl< asset_iarchive > archive_base_t;

private:
    AssetManagerType* TheAssetManager;
    osBool RootAssetLoaded;

public:
    asset_iarchive(AssetManagerType* theAssetManager, std::istream& is, unsigned int flags = 0)
        :archive_base_t(is, flags)
        ,TheAssetManager(theAssetManager)
        ,RootAssetLoaded(false)
    {}
    ~asset_iarchive()
    {
    }

public:
    template<class T>
    bool load_object_ptr_custom(T& t)
    {
        BOOST_STATIC_ASSERT(boost::is_pointer<T>::value);
        if (boost::is_base_of<AssetType, boost::remove_pointer<T>::type >::value)
        {
            if (!RootAssetLoaded)
            {
                RootAssetLoaded = true;
                return false;
            }
            else
            {
                AssetManagerType::ResourceID ResourceId;
                operator&(ResourceId);
                AssetType* a = TheAssetManager->FindAsset(ResourceId);
                if (!a)
                {
                    a = TheAssetManager->LoadAsset(ResourceId);
                }
                t = static_cast<T>(a);
                return true;
            }
        }
        else
        {
            return false;
        }
    }
};

#include <boost/archive/impl/basic_text_iarchive.ipp>
#include <boost/archive/impl/text_iarchive_impl.ipp>
#include <boost/archive/impl/archive_pointer_iserializer.ipp>

#include "AssetManager.hpp"
#include "Asset.hpp"

namespace boost {
namespace archive {

template class basic_text_iarchive< asset_iarchive<AssetManager,Asset> > ;
template class text_iarchive_impl< asset_iarchive<AssetManager,Asset> > ;
template class detail::archive_pointer_iserializer< asset_iarchive<AssetManager,Asset> > ;

} // namespace archive
} // namespace boost

#define BOOST_ARCHIVE_CUSTOM_IARCHIVE_TYPES asset_iarchive<AssetManager, Asset>

#endif//ASSETIARCHIVE_H