1257 lines
41 KiB
C
1257 lines
41 KiB
C
|
// This file is part of libigl, a simple c++ geometry processing library.
|
||
|
//
|
||
|
// Copyright (C) 2014 Christian Schüller <schuellchr@gmail.com>
|
||
|
//
|
||
|
// This Source Code Form is subject to the terms of the Mozilla Public License
|
||
|
// v. 2.0. If a copy of the MPL was not distributed with this file, You can
|
||
|
// obtain one at http://mozilla.org/MPL/2.0/.
|
||
|
#ifndef IGL_SERIALIZE_H
|
||
|
#define IGL_SERIALIZE_H
|
||
|
|
||
|
// -----------------------------------------------------------------------------
|
||
|
// Functions to save and load a serialization of fundamental c++ data types to
|
||
|
// and from a binary file. STL containers, Eigen matrix types and nested data
|
||
|
// structures are also supported. To serialize a user defined class implement
|
||
|
// the interface Serializable or SerializableBase.
|
||
|
//
|
||
|
// See also: xml/serialize_xml.h
|
||
|
// -----------------------------------------------------------------------------
|
||
|
// TODOs:
|
||
|
// * arbitrary pointer graph structures
|
||
|
// -----------------------------------------------------------------------------
|
||
|
|
||
|
// Known issues: This is not written in libigl-style so it isn't (easily)
|
||
|
// "dualized" into the static library.
|
||
|
//
|
||
|
|
||
|
#include <type_traits>
|
||
|
#include <iostream>
|
||
|
#include <fstream>
|
||
|
#include <cstdint>
|
||
|
#include <numeric>
|
||
|
#include <vector>
|
||
|
#include <set>
|
||
|
#include <map>
|
||
|
#include <memory>
|
||
|
#include <cstdint>
|
||
|
#include <list>
|
||
|
|
||
|
#include <Eigen/Dense>
|
||
|
#include <Eigen/Sparse>
|
||
|
|
||
|
#include "igl_inline.h"
|
||
|
|
||
|
// non-intrusive serialization helper macros
|
||
|
|
||
|
#define SERIALIZE_TYPE(Type,Params) \
|
||
|
namespace igl { namespace serialization { \
|
||
|
void _serialization(bool s,Type& obj,std::vector<char>& buffer) {Params} \
|
||
|
template<> inline void serialize(const Type& obj,std::vector<char>& buffer) { \
|
||
|
_serialization(true,const_cast<Type&>(obj),buffer); \
|
||
|
} \
|
||
|
template<> inline void deserialize(Type& obj,const std::vector<char>& buffer) { \
|
||
|
_serialization(false,obj,const_cast<std::vector<char>&>(buffer)); \
|
||
|
} \
|
||
|
}}
|
||
|
|
||
|
#define SERIALIZE_TYPE_SOURCE(Type,Params) \
|
||
|
namespace igl { namespace serialization { \
|
||
|
void _serialization(bool s,Type& obj,std::vector<char>& buffer) {Params} \
|
||
|
void _serialize(const Type& obj,std::vector<char>& buffer) { \
|
||
|
_serialization(true,const_cast<Type&>(obj),buffer); \
|
||
|
} \
|
||
|
void _deserialize(Type& obj,const std::vector<char>& buffer) { \
|
||
|
_serialization(false,obj,const_cast<std::vector<char>&>(buffer)); \
|
||
|
} \
|
||
|
}}
|
||
|
|
||
|
#define SERIALIZE_MEMBER(Object) igl::serializer(s,obj.Object,std::string(#Object),buffer);
|
||
|
#define SERIALIZE_MEMBER_NAME(Object,Name) igl::serializer(s,obj.Object,std::string(Name),buffer);
|
||
|
|
||
|
|
||
|
namespace igl
|
||
|
{
|
||
|
struct IndexedPointerBase;
|
||
|
|
||
|
// Serializes the given object either to a file or to a provided buffer
|
||
|
// Templates:
|
||
|
// T type of the object to serialize
|
||
|
// Inputs:
|
||
|
// obj object to serialize
|
||
|
// objectName unique object name,used for the identification
|
||
|
// overwrite set to true to overwrite an existing file
|
||
|
// filename name of the file containing the serialization
|
||
|
// Outputs:
|
||
|
// buffer binary serialization
|
||
|
//
|
||
|
template <typename T>
|
||
|
inline bool serialize(const T& obj,const std::string& filename);
|
||
|
template <typename T>
|
||
|
inline bool serialize(const T& obj,const std::string& objectName,const std::string& filename,bool overwrite = false);
|
||
|
template <typename T>
|
||
|
inline bool serialize(const T& obj,const std::string& objectName,std::vector<char>& buffer);
|
||
|
template <typename T>
|
||
|
inline bool serialize(const T& obj,const std::string& objectName,std::vector<char>& buffer);
|
||
|
|
||
|
// Deserializes the given data from a file or buffer back to the provided object
|
||
|
//
|
||
|
// Templates:
|
||
|
// T type of the object to serialize
|
||
|
// Inputs:
|
||
|
// buffer binary serialization
|
||
|
// objectName unique object name, used for the identification
|
||
|
// filename name of the file containing the serialization
|
||
|
// Outputs:
|
||
|
// obj object to load back serialization to
|
||
|
//
|
||
|
template <typename T>
|
||
|
inline bool deserialize(T& obj,const std::string& filename);
|
||
|
template <typename T>
|
||
|
inline bool deserialize(T& obj,const std::string& objectName,const std::string& filename);
|
||
|
template <typename T>
|
||
|
inline bool deserialize(T& obj,const std::string& objectName,const std::vector<char>& buffer);
|
||
|
|
||
|
// Wrapper to expose both, the de- and serialization as one function
|
||
|
//
|
||
|
template <typename T>
|
||
|
inline bool serializer(bool serialize,T& obj,const std::string& filename);
|
||
|
template <typename T>
|
||
|
inline bool serializer(bool serialize,T& obj,const std::string& objectName,const std::string& filename,bool overwrite = false);
|
||
|
template <typename T>
|
||
|
inline bool serializer(bool serialize,T& obj,const std::string& objectName,std::vector<char>& buffer);
|
||
|
|
||
|
// User defined types have to either overload the function igl::serialization::serialize()
|
||
|
// and igl::serialization::deserialize() for their type (non-intrusive serialization):
|
||
|
//
|
||
|
// namespace igl { namespace serialization
|
||
|
// {
|
||
|
// template<>
|
||
|
// inline void serialize(const UserType& obj,std::vector<char>& buffer) {
|
||
|
// ::igl::serialize(obj.var,"var",buffer);
|
||
|
// }
|
||
|
//
|
||
|
// template<>
|
||
|
// inline void deserialize(UserType& obj,const std::vector<char>& buffer) {
|
||
|
// ::igl::deserialize(obj.var,"var",buffer);
|
||
|
// }
|
||
|
// }}
|
||
|
//
|
||
|
// or use this macro for convenience:
|
||
|
//
|
||
|
// SERIALIZE_TYPE(UserType,
|
||
|
// SERIALIZE_MEMBER(var)
|
||
|
// )
|
||
|
//
|
||
|
// or to derive from the class Serializable and add their the members
|
||
|
// in InitSerialization like the following:
|
||
|
//
|
||
|
// class UserType : public igl::Serializable {
|
||
|
//
|
||
|
// int var;
|
||
|
//
|
||
|
// void InitSerialization() {
|
||
|
// this->Add(var,"var");
|
||
|
// }
|
||
|
// };
|
||
|
|
||
|
// Base interface for user defined types
|
||
|
struct SerializableBase
|
||
|
{
|
||
|
virtual void Serialize(std::vector<char>& buffer) const = 0;
|
||
|
virtual void Deserialize(const std::vector<char>& buffer) = 0;
|
||
|
};
|
||
|
|
||
|
// Convenient interface for user defined types
|
||
|
class Serializable: public SerializableBase
|
||
|
{
|
||
|
private:
|
||
|
|
||
|
template <typename T>
|
||
|
struct SerializationObject : public SerializableBase
|
||
|
{
|
||
|
bool Binary;
|
||
|
std::string Name;
|
||
|
std::unique_ptr<T> Object;
|
||
|
|
||
|
void Serialize(std::vector<char>& buffer) const override {
|
||
|
igl::serialize(*Object,Name,buffer);
|
||
|
}
|
||
|
|
||
|
void Deserialize(const std::vector<char>& buffer) override {
|
||
|
igl::deserialize(*Object,Name,buffer);
|
||
|
}
|
||
|
};
|
||
|
|
||
|
mutable bool initialized;
|
||
|
mutable std::vector<SerializableBase*> objects;
|
||
|
|
||
|
public:
|
||
|
|
||
|
// You **MUST** Override this function to add your member variables which
|
||
|
// should be serialized
|
||
|
//
|
||
|
// http://stackoverflow.com/a/6634382/148668
|
||
|
virtual void InitSerialization() = 0;
|
||
|
|
||
|
// Following functions can be overridden to handle the specific events.
|
||
|
// Return false to prevent the de-/serialization of an object.
|
||
|
inline virtual bool PreSerialization() const;
|
||
|
inline virtual void PostSerialization() const;
|
||
|
inline virtual bool PreDeserialization();
|
||
|
inline virtual void PostDeserialization();
|
||
|
|
||
|
// Default implementation of SerializableBase interface
|
||
|
inline void Serialize(std::vector<char>& buffer) const override final;
|
||
|
inline void Deserialize(const std::vector<char>& buffer) override final;
|
||
|
|
||
|
// Default constructor, destructor, assignment and copy constructor
|
||
|
inline Serializable();
|
||
|
inline Serializable(const Serializable& obj);
|
||
|
inline ~Serializable();
|
||
|
inline Serializable& operator=(const Serializable& obj);
|
||
|
|
||
|
// Use this function to add your variables which should be serialized
|
||
|
template <typename T>
|
||
|
inline void Add(T& obj,std::string name,bool binary = false);
|
||
|
};
|
||
|
|
||
|
// structure for pointer handling
|
||
|
struct IndexedPointerBase
|
||
|
{
|
||
|
enum { BEGIN,END } Type;
|
||
|
size_t Index;
|
||
|
};
|
||
|
template<typename T>
|
||
|
struct IndexedPointer: public IndexedPointerBase
|
||
|
{
|
||
|
const T* Object;
|
||
|
};
|
||
|
|
||
|
// internal functions
|
||
|
namespace serialization
|
||
|
{
|
||
|
// compile time type checks
|
||
|
template <typename T>
|
||
|
struct is_stl_container { static const bool value = false; };
|
||
|
template <typename T1,typename T2>
|
||
|
struct is_stl_container<std::pair<T1,T2> > { static const bool value = true; };
|
||
|
template <typename T1,typename T2>
|
||
|
struct is_stl_container<std::vector<T1,T2> > { static const bool value = true; };
|
||
|
template <typename T>
|
||
|
struct is_stl_container<std::set<T> > { static const bool value = true; };
|
||
|
template <typename T1,typename T2>
|
||
|
struct is_stl_container<std::map<T1,T2> > { static const bool value = true; };
|
||
|
template <typename T>
|
||
|
struct is_stl_container<std::list<T> > { static const bool value = true; };
|
||
|
|
||
|
template <typename T>
|
||
|
struct is_eigen_type { static const bool value = false; };
|
||
|
template <typename T,int R,int C,int P,int MR,int MC>
|
||
|
struct is_eigen_type<Eigen::Matrix<T,R,C,P,MR,MC> > { static const bool value = true; };
|
||
|
template <typename T,int P,typename I>
|
||
|
struct is_eigen_type<Eigen::SparseMatrix<T,P,I> > { static const bool value = true; };
|
||
|
|
||
|
template <typename T>
|
||
|
struct is_smart_ptr { static const bool value = false; };
|
||
|
template <typename T>
|
||
|
struct is_smart_ptr<std::shared_ptr<T> > { static const bool value = true; };
|
||
|
template <typename T>
|
||
|
struct is_smart_ptr<std::unique_ptr<T> > { static const bool value = true; };
|
||
|
template <typename T>
|
||
|
struct is_smart_ptr<std::weak_ptr<T> > { static const bool value = true; };
|
||
|
|
||
|
template <typename T>
|
||
|
struct is_serializable {
|
||
|
static const bool value = std::is_fundamental<T>::value || std::is_same<std::string,T>::value || std::is_enum<T>::value || std::is_base_of<SerializableBase,T>::value
|
||
|
|| is_stl_container<T>::value || is_eigen_type<T>::value || std::is_pointer<T>::value || serialization::is_smart_ptr<T>::value;
|
||
|
};
|
||
|
|
||
|
// non serializable types
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<!is_serializable<T>::value,size_t>::type getByteSize(const T& obj);
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<!is_serializable<T>::value>::type serialize(const T& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter);
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<!is_serializable<T>::value>::type deserialize(T& obj,std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
// fundamental types
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_fundamental<T>::value,size_t>::type getByteSize(const T& obj);
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_fundamental<T>::value>::type serialize(const T& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter);
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_fundamental<T>::value>::type deserialize(T& obj,std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
// std::string
|
||
|
inline size_t getByteSize(const std::string& obj);
|
||
|
inline void serialize(const std::string& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter);
|
||
|
inline void deserialize(std::string& obj,std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
// enum types
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_enum<T>::value,size_t>::type getByteSize(const T& obj);
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_enum<T>::value>::type serialize(const T& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter);
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_enum<T>::value>::type deserialize(T& obj,std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
// SerializableBase
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_base_of<SerializableBase,T>::value,size_t>::type getByteSize(const T& obj);
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_base_of<SerializableBase,T>::value>::type serialize(const T& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter);
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_base_of<SerializableBase,T>::value>::type deserialize(T& obj,std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
// stl containers
|
||
|
// std::pair
|
||
|
template <typename T1,typename T2>
|
||
|
inline size_t getByteSize(const std::pair<T1,T2>& obj);
|
||
|
template <typename T1,typename T2>
|
||
|
inline void serialize(const std::pair<T1,T2>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter);
|
||
|
template <typename T1,typename T2>
|
||
|
inline void deserialize(std::pair<T1,T2>& obj,std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
// std::vector
|
||
|
template <typename T1,typename T2>
|
||
|
inline size_t getByteSize(const std::vector<T1,T2>& obj);
|
||
|
template <typename T1,typename T2>
|
||
|
inline void serialize(const std::vector<T1,T2>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter);
|
||
|
template <typename T1,typename T2>
|
||
|
inline void deserialize(std::vector<T1,T2>& obj,std::vector<char>::const_iterator& iter);
|
||
|
template <typename T2>
|
||
|
inline void deserialize(std::vector<bool,T2>& obj,std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
// std::set
|
||
|
template <typename T>
|
||
|
inline size_t getByteSize(const std::set<T>& obj);
|
||
|
template <typename T>
|
||
|
inline void serialize(const std::set<T>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter);
|
||
|
template <typename T>
|
||
|
inline void deserialize(std::set<T>& obj,std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
// std::map
|
||
|
template <typename T1,typename T2>
|
||
|
inline size_t getByteSize(const std::map<T1,T2>& obj);
|
||
|
template <typename T1,typename T2>
|
||
|
inline void serialize(const std::map<T1,T2>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter);
|
||
|
template <typename T1,typename T2>
|
||
|
inline void deserialize(std::map<T1,T2>& obj,std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
// std::list
|
||
|
template <typename T>
|
||
|
inline size_t getByteSize(const std::list<T>& obj);
|
||
|
template <typename T>
|
||
|
inline void serialize(const std::list<T>& obj, std::vector<char>& buffer, std::vector<char>::iterator& iter);
|
||
|
template <typename T>
|
||
|
inline void deserialize(std::list<T>& obj, std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
// Eigen types
|
||
|
template<typename T,int R,int C,int P,int MR,int MC>
|
||
|
inline size_t getByteSize(const Eigen::Matrix<T,R,C,P,MR,MC>& obj);
|
||
|
template<typename T,int R,int C,int P,int MR,int MC>
|
||
|
inline void serialize(const Eigen::Matrix<T,R,C,P,MR,MC>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter);
|
||
|
template<typename T,int R,int C,int P,int MR,int MC>
|
||
|
inline void deserialize(Eigen::Matrix<T,R,C,P,MR,MC>& obj,std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
template<typename T,int P,typename I>
|
||
|
inline size_t getByteSize(const Eigen::SparseMatrix<T,P,I>& obj);
|
||
|
template<typename T,int P,typename I>
|
||
|
inline void serialize(const Eigen::SparseMatrix<T,P,I>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter);
|
||
|
template<typename T,int P,typename I>
|
||
|
inline void deserialize(Eigen::SparseMatrix<T,P,I>& obj,std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
template<typename T,int P>
|
||
|
inline size_t getByteSize(const Eigen::Quaternion<T,P>& obj);
|
||
|
template<typename T,int P>
|
||
|
inline void serialize(const Eigen::Quaternion<T,P>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter);
|
||
|
template<typename T,int P>
|
||
|
inline void deserialize(Eigen::Quaternion<T,P>& obj,std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
// raw pointers
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_pointer<T>::value,size_t>::type getByteSize(const T& obj);
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_pointer<T>::value>::type serialize(const T& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter);
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_pointer<T>::value>::type deserialize(T& obj,std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
// std::shared_ptr and std::unique_ptr
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<serialization::is_smart_ptr<T>::value,size_t>::type getByteSize(const T& obj);
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<serialization::is_smart_ptr<T>::value>::type serialize(const T& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter);
|
||
|
template <template<typename> class T0, typename T1>
|
||
|
inline typename std::enable_if<serialization::is_smart_ptr<T0<T1> >::value>::type deserialize(T0<T1>& obj,std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
// std::weak_ptr
|
||
|
template <typename T>
|
||
|
inline size_t getByteSize(const std::weak_ptr<T>& obj);
|
||
|
template <typename T>
|
||
|
inline void serialize(const std::weak_ptr<T>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter);
|
||
|
template <typename T>
|
||
|
inline void deserialize(std::weak_ptr<T>& obj,std::vector<char>::const_iterator& iter);
|
||
|
|
||
|
// functions to overload for non-intrusive serialization
|
||
|
template <typename T>
|
||
|
inline void serialize(const T& obj,std::vector<char>& buffer);
|
||
|
template <typename T>
|
||
|
inline void deserialize(T& obj,const std::vector<char>& buffer);
|
||
|
|
||
|
// helper functions
|
||
|
template <typename T>
|
||
|
inline void updateMemoryMap(T& obj,size_t size);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Always include inlines for these functions
|
||
|
|
||
|
// IMPLEMENTATION
|
||
|
|
||
|
namespace igl
|
||
|
{
|
||
|
template <typename T>
|
||
|
inline bool serialize(const T& obj,const std::string& filename)
|
||
|
{
|
||
|
return serialize(obj,"obj",filename,true);
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline bool serialize(const T& obj,const std::string& objectName,const std::string& filename,bool overwrite)
|
||
|
{
|
||
|
bool success = false;
|
||
|
|
||
|
std::vector<char> buffer;
|
||
|
|
||
|
std::ios_base::openmode mode = std::ios::out | std::ios::binary;
|
||
|
|
||
|
if(overwrite)
|
||
|
mode |= std::ios::trunc;
|
||
|
else
|
||
|
mode |= std::ios::app;
|
||
|
|
||
|
std::ofstream file(filename.c_str(),mode);
|
||
|
|
||
|
if(file.is_open())
|
||
|
{
|
||
|
serialize(obj,objectName,buffer);
|
||
|
|
||
|
file.write(&buffer[0],buffer.size());
|
||
|
|
||
|
file.close();
|
||
|
|
||
|
success = true;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
std::cerr << "serialization: file " << filename << " not found!" << std::endl;
|
||
|
}
|
||
|
|
||
|
return success;
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline bool serialize(const T& obj,const std::string& objectName,std::vector<char>& buffer)
|
||
|
{
|
||
|
// serialize object data
|
||
|
size_t size = serialization::getByteSize(obj);
|
||
|
std::vector<char> tmp(size);
|
||
|
auto it = tmp.begin();
|
||
|
serialization::serialize(obj,tmp,it);
|
||
|
|
||
|
std::string objectType(typeid(obj).name());
|
||
|
size_t newObjectSize = tmp.size();
|
||
|
size_t newHeaderSize = serialization::getByteSize(objectName) + serialization::getByteSize(objectType) + sizeof(size_t);
|
||
|
size_t curSize = buffer.size();
|
||
|
size_t newSize = curSize + newHeaderSize + newObjectSize;
|
||
|
|
||
|
buffer.resize(newSize);
|
||
|
|
||
|
std::vector<char>::iterator iter = buffer.begin()+curSize;
|
||
|
|
||
|
// serialize object header (name/type/size)
|
||
|
serialization::serialize(objectName,buffer,iter);
|
||
|
serialization::serialize(objectType,buffer,iter);
|
||
|
serialization::serialize(newObjectSize,buffer,iter);
|
||
|
|
||
|
// copy serialized data to buffer
|
||
|
iter = std::copy(tmp.begin(),tmp.end(),iter);
|
||
|
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline bool deserialize(T& obj,const std::string& filename)
|
||
|
{
|
||
|
return deserialize(obj,"obj",filename);
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline bool deserialize(T& obj,const std::string& objectName,const std::string& filename)
|
||
|
{
|
||
|
bool success = false;
|
||
|
|
||
|
std::ifstream file(filename.c_str(),std::ios::binary);
|
||
|
|
||
|
if(file.is_open())
|
||
|
{
|
||
|
file.seekg(0,std::ios::end);
|
||
|
std::streamoff size = file.tellg();
|
||
|
file.seekg(0,std::ios::beg);
|
||
|
|
||
|
std::vector<char> buffer(size);
|
||
|
file.read(&buffer[0],size);
|
||
|
|
||
|
success = deserialize(obj, objectName, buffer);
|
||
|
file.close();
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
std::cerr << "serialization: file " << filename << " not found!" << std::endl;
|
||
|
}
|
||
|
|
||
|
return success;
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline bool deserialize(T& obj,const std::string& objectName,const std::vector<char>& buffer)
|
||
|
{
|
||
|
bool success = false;
|
||
|
|
||
|
// find suitable object header
|
||
|
auto objectIter = buffer.cend();
|
||
|
auto iter = buffer.cbegin();
|
||
|
while(iter != buffer.end())
|
||
|
{
|
||
|
std::string name;
|
||
|
std::string type;
|
||
|
size_t size;
|
||
|
serialization::deserialize(name,iter);
|
||
|
serialization::deserialize(type,iter);
|
||
|
serialization::deserialize(size,iter);
|
||
|
|
||
|
if(name == objectName && type == typeid(obj).name())
|
||
|
{
|
||
|
objectIter = iter;
|
||
|
//break; // find first suitable object header
|
||
|
}
|
||
|
|
||
|
iter+=size;
|
||
|
}
|
||
|
|
||
|
if(objectIter != buffer.end())
|
||
|
{
|
||
|
serialization::deserialize(obj,objectIter);
|
||
|
success = true;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
obj = T();
|
||
|
}
|
||
|
|
||
|
return success;
|
||
|
}
|
||
|
|
||
|
// Wrapper function which combines both, de- and serialization
|
||
|
template <typename T>
|
||
|
inline bool serializer(bool s,T& obj,const std::string& filename)
|
||
|
{
|
||
|
return s ? serialize(obj,filename) : deserialize(obj,filename);
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline bool serializer(bool s,T& obj,const std::string& objectName,const std::string& filename,bool overwrite)
|
||
|
{
|
||
|
return s ? serialize(obj,objectName,filename,overwrite) : deserialize(obj,objectName,filename);
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline bool serializer(bool s,T& obj,const std::string& objectName,std::vector<char>& buffer)
|
||
|
{
|
||
|
return s ? serialize(obj,objectName,buffer) : deserialize(obj,objectName,buffer);
|
||
|
}
|
||
|
|
||
|
inline bool Serializable::PreSerialization() const
|
||
|
{
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
inline void Serializable::PostSerialization() const
|
||
|
{
|
||
|
}
|
||
|
|
||
|
inline bool Serializable::PreDeserialization()
|
||
|
{
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
inline void Serializable::PostDeserialization()
|
||
|
{
|
||
|
}
|
||
|
|
||
|
inline void Serializable::Serialize(std::vector<char>& buffer) const
|
||
|
{
|
||
|
if(this->PreSerialization())
|
||
|
{
|
||
|
if(initialized == false)
|
||
|
{
|
||
|
objects.clear();
|
||
|
(const_cast<Serializable*>(this))->InitSerialization();
|
||
|
initialized = true;
|
||
|
}
|
||
|
|
||
|
for(const auto& v : objects)
|
||
|
{
|
||
|
v->Serialize(buffer);
|
||
|
}
|
||
|
|
||
|
this->PostSerialization();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
inline void Serializable::Deserialize(const std::vector<char>& buffer)
|
||
|
{
|
||
|
if(this->PreDeserialization())
|
||
|
{
|
||
|
if(initialized == false)
|
||
|
{
|
||
|
objects.clear();
|
||
|
(const_cast<Serializable*>(this))->InitSerialization();
|
||
|
initialized = true;
|
||
|
}
|
||
|
|
||
|
for(auto& v : objects)
|
||
|
{
|
||
|
v->Deserialize(buffer);
|
||
|
}
|
||
|
|
||
|
this->PostDeserialization();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
inline Serializable::Serializable()
|
||
|
{
|
||
|
initialized = false;
|
||
|
}
|
||
|
|
||
|
inline Serializable::Serializable(const Serializable& obj)
|
||
|
{
|
||
|
initialized = false;
|
||
|
objects.clear();
|
||
|
}
|
||
|
|
||
|
inline Serializable::~Serializable()
|
||
|
{
|
||
|
initialized = false;
|
||
|
objects.clear();
|
||
|
}
|
||
|
|
||
|
inline Serializable& Serializable::operator=(const Serializable& obj)
|
||
|
{
|
||
|
if(this != &obj)
|
||
|
{
|
||
|
if(initialized)
|
||
|
{
|
||
|
initialized = false;
|
||
|
objects.clear();
|
||
|
}
|
||
|
}
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline void Serializable::Add(T& obj,const std::string name,bool binary)
|
||
|
{
|
||
|
auto object = new SerializationObject<T>();
|
||
|
object->Binary = binary;
|
||
|
object->Name = name;
|
||
|
object->Object = std::unique_ptr<T>(&obj);
|
||
|
|
||
|
objects.push_back(object);
|
||
|
}
|
||
|
|
||
|
namespace serialization
|
||
|
{
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<!is_serializable<T>::value,size_t>::type getByteSize(const T& obj)
|
||
|
{
|
||
|
return sizeof(std::vector<char>::size_type);
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<!is_serializable<T>::value>::type serialize(const T& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
// data
|
||
|
std::vector<char> tmp;
|
||
|
serialize<>(obj,tmp);
|
||
|
|
||
|
// size
|
||
|
size_t size = buffer.size();
|
||
|
serialization::serialize(tmp.size(),buffer,iter);
|
||
|
size_t cur = iter - buffer.begin();
|
||
|
|
||
|
buffer.resize(size+tmp.size());
|
||
|
iter = buffer.begin()+cur;
|
||
|
iter = std::copy(tmp.begin(),tmp.end(),iter);
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<!is_serializable<T>::value>::type deserialize(T& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
std::vector<char>::size_type size;
|
||
|
serialization::deserialize<>(size,iter);
|
||
|
|
||
|
std::vector<char> tmp;
|
||
|
tmp.resize(size);
|
||
|
std::copy(iter,iter+size,tmp.begin());
|
||
|
|
||
|
deserialize<>(obj,tmp);
|
||
|
iter += size;
|
||
|
}
|
||
|
|
||
|
// fundamental types
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_fundamental<T>::value,size_t>::type getByteSize(const T& obj)
|
||
|
{
|
||
|
return sizeof(T);
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_fundamental<T>::value>::type serialize(const T& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
//serialization::updateMemoryMap(obj,sizeof(T));
|
||
|
const uint8_t* ptr = reinterpret_cast<const uint8_t*>(&obj);
|
||
|
iter = std::copy(ptr,ptr+sizeof(T),iter);
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_fundamental<T>::value>::type deserialize(T& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
uint8_t* ptr = reinterpret_cast<uint8_t*>(&obj);
|
||
|
std::copy(iter,iter+sizeof(T),ptr);
|
||
|
iter += sizeof(T);
|
||
|
}
|
||
|
|
||
|
// std::string
|
||
|
|
||
|
inline size_t getByteSize(const std::string& obj)
|
||
|
{
|
||
|
return getByteSize(obj.length())+obj.length()*sizeof(uint8_t);
|
||
|
}
|
||
|
|
||
|
inline void serialize(const std::string& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
serialization::serialize(obj.length(),buffer,iter);
|
||
|
for(const auto& cur : obj)
|
||
|
{
|
||
|
serialization::serialize(cur,buffer,iter);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
inline void deserialize(std::string& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
size_t size;
|
||
|
serialization::deserialize(size,iter);
|
||
|
|
||
|
std::string str(size,'\0');
|
||
|
for(size_t i=0; i<size; ++i)
|
||
|
{
|
||
|
serialization::deserialize(str.at(i),iter);
|
||
|
}
|
||
|
|
||
|
obj = str;
|
||
|
}
|
||
|
|
||
|
// enum types
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_enum<T>::value,size_t>::type getByteSize(const T& obj)
|
||
|
{
|
||
|
return sizeof(T);
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_enum<T>::value>::type serialize(const T& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
const uint8_t* ptr = reinterpret_cast<const uint8_t*>(&obj);
|
||
|
iter = std::copy(ptr,ptr+sizeof(T),iter);
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_enum<T>::value>::type deserialize(T& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
uint8_t* ptr = reinterpret_cast<uint8_t*>(&obj);
|
||
|
std::copy(iter,iter+sizeof(T),ptr);
|
||
|
iter += sizeof(T);
|
||
|
}
|
||
|
|
||
|
// SerializableBase
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_base_of<SerializableBase,T>::value,size_t>::type getByteSize(const T& obj)
|
||
|
{
|
||
|
return sizeof(std::vector<char>::size_type);
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_base_of<SerializableBase,T>::value>::type serialize(const T& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
// data
|
||
|
std::vector<char> tmp;
|
||
|
obj.Serialize(tmp);
|
||
|
|
||
|
// size
|
||
|
size_t size = buffer.size();
|
||
|
serialization::serialize(tmp.size(),buffer,iter);
|
||
|
size_t cur = iter - buffer.begin();
|
||
|
|
||
|
buffer.resize(size+tmp.size());
|
||
|
iter = buffer.begin()+cur;
|
||
|
iter = std::copy(tmp.begin(),tmp.end(),iter);
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_base_of<SerializableBase,T>::value>::type deserialize(T& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
std::vector<char>::size_type size;
|
||
|
serialization::deserialize(size,iter);
|
||
|
|
||
|
std::vector<char> tmp;
|
||
|
tmp.resize(size);
|
||
|
std::copy(iter,iter+size,tmp.begin());
|
||
|
|
||
|
obj.Deserialize(tmp);
|
||
|
iter += size;
|
||
|
}
|
||
|
|
||
|
// STL containers
|
||
|
|
||
|
// std::pair
|
||
|
|
||
|
template <typename T1,typename T2>
|
||
|
inline size_t getByteSize(const std::pair<T1,T2>& obj)
|
||
|
{
|
||
|
return getByteSize(obj.first)+getByteSize(obj.second);
|
||
|
}
|
||
|
|
||
|
template <typename T1,typename T2>
|
||
|
inline void serialize(const std::pair<T1,T2>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
serialization::serialize(obj.first,buffer,iter);
|
||
|
serialization::serialize(obj.second,buffer,iter);
|
||
|
}
|
||
|
|
||
|
template <typename T1,typename T2>
|
||
|
inline void deserialize(std::pair<T1,T2>& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
serialization::deserialize(obj.first,iter);
|
||
|
serialization::deserialize(obj.second,iter);
|
||
|
}
|
||
|
|
||
|
// std::vector
|
||
|
|
||
|
template <typename T1,typename T2>
|
||
|
inline size_t getByteSize(const std::vector<T1,T2>& obj)
|
||
|
{
|
||
|
return std::accumulate(obj.begin(),obj.end(),sizeof(size_t),[](const size_t& acc,const T1& cur) { return acc+getByteSize(cur); });
|
||
|
}
|
||
|
|
||
|
template <typename T1,typename T2>
|
||
|
inline void serialize(const std::vector<T1,T2>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
size_t size = obj.size();
|
||
|
serialization::serialize(size,buffer,iter);
|
||
|
for(const T1& cur : obj)
|
||
|
{
|
||
|
serialization::serialize(cur,buffer,iter);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
template <typename T1,typename T2>
|
||
|
inline void deserialize(std::vector<T1,T2>& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
size_t size;
|
||
|
serialization::deserialize(size,iter);
|
||
|
|
||
|
obj.resize(size);
|
||
|
for(T1& v : obj)
|
||
|
{
|
||
|
serialization::deserialize(v,iter);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
template <typename T2>
|
||
|
inline void deserialize(std::vector<bool,T2>& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
size_t size;
|
||
|
serialization::deserialize(size,iter);
|
||
|
|
||
|
obj.resize(size);
|
||
|
for(int i=0;i<obj.size();i++)
|
||
|
{
|
||
|
bool val;
|
||
|
serialization::deserialize(val,iter);
|
||
|
obj[i] = val;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//std::set
|
||
|
|
||
|
template <typename T>
|
||
|
inline size_t getByteSize(const std::set<T>& obj)
|
||
|
{
|
||
|
return std::accumulate(obj.begin(),obj.end(),getByteSize(obj.size()),[](const size_t& acc,const T& cur) { return acc+getByteSize(cur); });
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline void serialize(const std::set<T>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
serialization::serialize(obj.size(),buffer,iter);
|
||
|
for(const T& cur : obj)
|
||
|
{
|
||
|
serialization::serialize(cur,buffer,iter);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline void deserialize(std::set<T>& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
size_t size;
|
||
|
serialization::deserialize(size,iter);
|
||
|
|
||
|
obj.clear();
|
||
|
for(size_t i=0; i<size; ++i)
|
||
|
{
|
||
|
T val;
|
||
|
serialization::deserialize(val,iter);
|
||
|
obj.insert(val);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// std::map
|
||
|
|
||
|
template <typename T1,typename T2>
|
||
|
inline size_t getByteSize(const std::map<T1,T2>& obj)
|
||
|
{
|
||
|
return std::accumulate(obj.begin(),obj.end(),sizeof(size_t),[](const size_t& acc,const std::pair<T1,T2>& cur) { return acc+getByteSize(cur); });
|
||
|
}
|
||
|
|
||
|
template <typename T1,typename T2>
|
||
|
inline void serialize(const std::map<T1,T2>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
serialization::serialize(obj.size(),buffer,iter);
|
||
|
for(const auto& cur : obj)
|
||
|
{
|
||
|
serialization::serialize(cur,buffer,iter);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
template <typename T1,typename T2>
|
||
|
inline void deserialize(std::map<T1,T2>& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
size_t size;
|
||
|
serialization::deserialize(size,iter);
|
||
|
|
||
|
obj.clear();
|
||
|
for(size_t i=0; i<size; ++i)
|
||
|
{
|
||
|
std::pair<T1,T2> pair;
|
||
|
serialization::deserialize(pair,iter);
|
||
|
obj.insert(pair);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//std::list
|
||
|
|
||
|
template <typename T>
|
||
|
inline size_t getByteSize(const std::list<T>& obj)
|
||
|
{
|
||
|
return std::accumulate(obj.begin(), obj.end(), getByteSize(obj.size()), [](const size_t& acc, const T& cur) { return acc + getByteSize(cur); });
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline void serialize(const std::list<T>& obj, std::vector<char>& buffer, std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
serialization::serialize(obj.size(), buffer, iter);
|
||
|
for (const T& cur : obj)
|
||
|
{
|
||
|
serialization::serialize(cur, buffer, iter);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline void deserialize(std::list<T>& obj, std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
size_t size;
|
||
|
serialization::deserialize(size, iter);
|
||
|
|
||
|
obj.clear();
|
||
|
for (size_t i = 0; i < size; ++i)
|
||
|
{
|
||
|
T val;
|
||
|
serialization::deserialize(val, iter);
|
||
|
obj.emplace_back(val);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
// Eigen types
|
||
|
template<typename T,int R,int C,int P,int MR,int MC>
|
||
|
inline size_t getByteSize(const Eigen::Matrix<T,R,C,P,MR,MC>& obj)
|
||
|
{
|
||
|
// space for numbers of rows,cols and data
|
||
|
return 2*sizeof(typename Eigen::Matrix<T,R,C,P,MR,MC>::Index)+sizeof(T)*obj.rows()*obj.cols();
|
||
|
}
|
||
|
|
||
|
template<typename T,int R,int C,int P,int MR,int MC>
|
||
|
inline void serialize(const Eigen::Matrix<T,R,C,P,MR,MC>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
serialization::serialize(obj.rows(),buffer,iter);
|
||
|
serialization::serialize(obj.cols(),buffer,iter);
|
||
|
size_t size = sizeof(T)*obj.rows()*obj.cols();
|
||
|
auto ptr = reinterpret_cast<const uint8_t*>(obj.data());
|
||
|
iter = std::copy(ptr,ptr+size,iter);
|
||
|
}
|
||
|
|
||
|
template<typename T,int R,int C,int P,int MR,int MC>
|
||
|
inline void deserialize(Eigen::Matrix<T,R,C,P,MR,MC>& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
typename Eigen::Matrix<T,R,C,P,MR,MC>::Index rows,cols;
|
||
|
serialization::deserialize(rows,iter);
|
||
|
serialization::deserialize(cols,iter);
|
||
|
size_t size = sizeof(T)*rows*cols;
|
||
|
obj.resize(rows,cols);
|
||
|
auto ptr = reinterpret_cast<uint8_t*>(obj.data());
|
||
|
std::copy(iter,iter+size,ptr);
|
||
|
iter+=size;
|
||
|
}
|
||
|
|
||
|
template<typename T,int P,typename I>
|
||
|
inline size_t getByteSize(const Eigen::SparseMatrix<T,P,I>& obj)
|
||
|
{
|
||
|
// space for numbers of rows,cols,nonZeros and tripplets with data (rowIdx,colIdx,value)
|
||
|
size_t size = sizeof(typename Eigen::SparseMatrix<T,P,I>::Index);
|
||
|
return 3*size+(sizeof(T)+2*size)*obj.nonZeros();
|
||
|
}
|
||
|
|
||
|
template<typename T,int P,typename I>
|
||
|
inline void serialize(const Eigen::SparseMatrix<T,P,I>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
serialization::serialize(obj.rows(),buffer,iter);
|
||
|
serialization::serialize(obj.cols(),buffer,iter);
|
||
|
serialization::serialize(obj.nonZeros(),buffer,iter);
|
||
|
|
||
|
for(int k=0;k<obj.outerSize();++k)
|
||
|
{
|
||
|
for(typename Eigen::SparseMatrix<T,P,I>::InnerIterator it(obj,k);it;++it)
|
||
|
{
|
||
|
serialization::serialize(it.row(),buffer,iter);
|
||
|
serialization::serialize(it.col(),buffer,iter);
|
||
|
serialization::serialize(it.value(),buffer,iter);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
template<typename T,int P,typename I>
|
||
|
inline void deserialize(Eigen::SparseMatrix<T,P,I>& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
typename Eigen::SparseMatrix<T,P,I>::Index rows,cols,nonZeros;
|
||
|
serialization::deserialize(rows,iter);
|
||
|
serialization::deserialize(cols,iter);
|
||
|
serialization::deserialize(nonZeros,iter);
|
||
|
|
||
|
obj.resize(rows,cols);
|
||
|
obj.setZero();
|
||
|
|
||
|
std::vector<Eigen::Triplet<T,I> > triplets;
|
||
|
for(int i=0;i<nonZeros;i++)
|
||
|
{
|
||
|
typename Eigen::SparseMatrix<T,P,I>::Index rowId,colId;
|
||
|
serialization::deserialize(rowId,iter);
|
||
|
serialization::deserialize(colId,iter);
|
||
|
T value;
|
||
|
serialization::deserialize(value,iter);
|
||
|
triplets.push_back(Eigen::Triplet<T,I>(rowId,colId,value));
|
||
|
}
|
||
|
obj.setFromTriplets(triplets.begin(),triplets.end());
|
||
|
}
|
||
|
|
||
|
template<typename T,int P>
|
||
|
inline size_t getByteSize(const Eigen::Quaternion<T,P>& obj)
|
||
|
{
|
||
|
return sizeof(T)*4;
|
||
|
}
|
||
|
|
||
|
template<typename T,int P>
|
||
|
inline void serialize(const Eigen::Quaternion<T,P>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
serialization::serialize(obj.w(),buffer,iter);
|
||
|
serialization::serialize(obj.x(),buffer,iter);
|
||
|
serialization::serialize(obj.y(),buffer,iter);
|
||
|
serialization::serialize(obj.z(),buffer,iter);
|
||
|
}
|
||
|
|
||
|
template<typename T,int P>
|
||
|
inline void deserialize(Eigen::Quaternion<T,P>& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
serialization::deserialize(obj.w(),iter);
|
||
|
serialization::deserialize(obj.x(),iter);
|
||
|
serialization::deserialize(obj.y(),iter);
|
||
|
serialization::deserialize(obj.z(),iter);
|
||
|
}
|
||
|
|
||
|
// pointers
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_pointer<T>::value,size_t>::type getByteSize(const T& obj)
|
||
|
{
|
||
|
size_t size = sizeof(bool);
|
||
|
|
||
|
if(obj)
|
||
|
size += getByteSize(*obj);
|
||
|
|
||
|
return size;
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_pointer<T>::value>::type serialize(const T& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
serialization::serialize(obj == nullptr,buffer,iter);
|
||
|
|
||
|
if(obj)
|
||
|
serialization::serialize(*obj,buffer,iter);
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<std::is_pointer<T>::value>::type deserialize(T& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
bool isNullPtr;
|
||
|
serialization::deserialize(isNullPtr,iter);
|
||
|
|
||
|
if(isNullPtr)
|
||
|
{
|
||
|
if(obj)
|
||
|
{
|
||
|
std::cout << "serialization: possible memory leak in serialization for '" << typeid(obj).name() << "'" << std::endl;
|
||
|
obj = nullptr;
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
if(obj)
|
||
|
{
|
||
|
std::cout << "serialization: possible memory corruption in deserialization for '" << typeid(obj).name() << "'" << std::endl;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
obj = new typename std::remove_pointer<T>::type();
|
||
|
}
|
||
|
serialization::deserialize(*obj,iter);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// std::shared_ptr and std::unique_ptr
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<serialization::is_smart_ptr<T>::value,size_t>::type getByteSize(const T& obj)
|
||
|
{
|
||
|
return getByteSize(obj.get());
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline typename std::enable_if<serialization::is_smart_ptr<T>::value>::type serialize(const T& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
serialize(obj.get(),buffer,iter);
|
||
|
}
|
||
|
|
||
|
template <template<typename> class T0,typename T1>
|
||
|
inline typename std::enable_if<serialization::is_smart_ptr<T0<T1> >::value>::type deserialize(T0<T1>& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
bool isNullPtr;
|
||
|
serialization::deserialize(isNullPtr,iter);
|
||
|
|
||
|
if(isNullPtr)
|
||
|
{
|
||
|
obj.reset();
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
obj = T0<T1>(new T1());
|
||
|
serialization::deserialize(*obj,iter);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// std::weak_ptr
|
||
|
|
||
|
template <typename T>
|
||
|
inline size_t getByteSize(const std::weak_ptr<T>& obj)
|
||
|
{
|
||
|
return sizeof(size_t);
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline void serialize(const std::weak_ptr<T>& obj,std::vector<char>& buffer,std::vector<char>::iterator& iter)
|
||
|
{
|
||
|
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline void deserialize(std::weak_ptr<T>& obj,std::vector<char>::const_iterator& iter)
|
||
|
{
|
||
|
|
||
|
}
|
||
|
|
||
|
// functions to overload for non-intrusive serialization
|
||
|
template <typename T>
|
||
|
inline void serialize(const T& obj,std::vector<char>& buffer)
|
||
|
{
|
||
|
std::cerr << typeid(obj).name() << " is not serializable: derive from igl::Serializable or spezialize the template function igl::serialization::serialize(const T& obj,std::vector<char>& buffer)" << std::endl;
|
||
|
}
|
||
|
|
||
|
template <typename T>
|
||
|
inline void deserialize(T& obj,const std::vector<char>& buffer)
|
||
|
{
|
||
|
std::cerr << typeid(obj).name() << " is not deserializable: derive from igl::Serializable or spezialize the template function igl::serialization::deserialize(T& obj, const std::vector<char>& buffer)" << std::endl;
|
||
|
}
|
||
|
|
||
|
// helper functions
|
||
|
|
||
|
template <typename T>
|
||
|
inline void updateMemoryMap(T& obj,size_t size,std::map<std::uintptr_t,IndexedPointerBase*>& memoryMap)
|
||
|
{
|
||
|
// check if object is already serialized
|
||
|
auto startPtr = new IndexedPointer<T>();
|
||
|
startPtr->Object = &obj;
|
||
|
auto startBasePtr = static_cast<IndexedPointerBase*>(startPtr);
|
||
|
startBasePtr->Type = IndexedPointerBase::BEGIN;
|
||
|
auto startAddress = reinterpret_cast<std::uintptr_t>(&obj);
|
||
|
auto p = std::pair<std::uintptr_t,IndexedPointerBase*>(startAddress,startBasePtr);
|
||
|
|
||
|
auto el = memoryMap.insert(p);
|
||
|
auto iter = ++el.first; // next elememt
|
||
|
if(el.second && (iter == memoryMap.end() || iter->second->Type != IndexedPointerBase::END))
|
||
|
{
|
||
|
// not yet serialized
|
||
|
auto endPtr = new IndexedPointer<T>();
|
||
|
auto endBasePtr = static_cast<IndexedPointerBase*>(endPtr);
|
||
|
endBasePtr->Type = IndexedPointerBase::END;
|
||
|
auto endAddress = reinterpret_cast<std::uintptr_t>(&obj) + size - 1;
|
||
|
auto p = std::pair<std::uintptr_t,IndexedPointerBase*>(endAddress,endBasePtr);
|
||
|
|
||
|
// insert end address
|
||
|
memoryMap.insert(el.first,p);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
// already serialized
|
||
|
|
||
|
// remove inserted address
|
||
|
memoryMap.erase(el.first);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#endif
|