#include <boost/test/unit_test.hpp>
#include <boost/serialization/base_object.hpp>
#include <boost/serialization/export.hpp>
#include <boost/serialization/nvp.hpp>
#include <boost/serialization/shared_ptr.hpp>
#include <boost/archive/polymorphic_xml_oarchive.hpp>
#include <boost/archive/polymorphic_xml_iarchive.hpp>
#include <fstream>
#include <boost/make_shared.hpp>

struct A {
public:
  A() {}
  virtual ~A() {}
        
  template<class Archive>
  void serialize(Archive &ar, unsigned int version) {
    ar & BOOST_SERIALIZATION_NVP(i);
  }

  int i;
};
BOOST_CLASS_EXPORT(A);

struct B : A {
  template<class Archive>
  void serialize(Archive &ar, unsigned int version) {
    ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(A);
    ar & BOOST_SERIALIZATION_NVP(f);
  }

  float f;
};
BOOST_CLASS_EXPORT(B);

BOOST_AUTO_TEST_SUITE(example_boost_class_export)

BOOST_AUTO_TEST_CASE(plain_boost_class_export)  {
  boost::shared_ptr<B> b = boost::make_shared<B>();
  b->f = 1.0;
  b->i = 3;
  boost::shared_ptr<A const> a = b;

  std::ofstream ofs("demo.xml");
  {
    boost::archive::xml_oarchive oa(ofs);
    oa << BOOST_SERIALIZATION_NVP(a);
  }
  ofs.close();

  boost::shared_ptr<A> p;
  std::ifstream ifs("demo.xml");
  {
    boost::archive::xml_iarchive ia(ifs);
    ia >> BOOST_SERIALIZATION_NVP(p);
  }
  ifs.close();

  boost::shared_ptr<B> bp = boost::dynamic_pointer_cast<B>(p);
  BOOST_REQUIRE(bp);
}
BOOST_AUTO_TEST_SUITE_END()