#ifndef ASSETOARCHIVE_H
#define ASSETOARCHIVE_H

#include <ostream>
#include <boost/archive/archive_exception.hpp>
#include <boost/archive/text_oarchive.hpp>
#include <boost/archive/detail/archive_pointer_oserializer.hpp>
#include <boost/type_traits.hpp>

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

private:
    AssetManagerType* TheAssetManager;
    osBool RootAssetSaved;

public:
    asset_oarchive(AssetManagerType* theAssetManager, std::ostream& os, unsigned int flags = 0)
        :archive_base_t(os, flags)
        ,TheAssetManager(theAssetManager)
        ,RootAssetSaved(false)
    {}
    ~asset_oarchive()
    {
    }

public:
    template<class T>
    bool save_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 (!RootAssetSaved)
            {
                RootAssetSaved = true;
                return false;
            }
            else
            {
                AssetManagerType::ResourceID ResourceId = TheAssetManager->ResourceIdForAsset(t);
                operator&(ResourceId);
                return true;
            }
        }
        else
        {
            return false;
        }
    }
};

#include <boost/archive/impl/basic_text_oarchive.ipp>
#include <boost/archive/impl/text_oarchive_impl.ipp>
#include <boost/archive/impl/archive_pointer_oserializer.ipp>

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

namespace boost {
namespace archive {

template class basic_text_oarchive< asset_oarchive<AssetManager,Asset> > ;
template class text_oarchive_impl< asset_oarchive<AssetManager,Asset> > ;
template class detail::archive_pointer_oserializer< asset_oarchive<AssetManager,Asset> > ;

} // namespace archive
} // namespace boost

#define BOOST_ARCHIVE_CUSTOM_OARCHIVE_TYPES asset_oarchive<AssetManager, Asset>

#endif//ASSETOARCHIVE_H