1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
// Copyright (C) 2019 EDF
// All Rights Reserved
// This code is published under the GNU Lesser General Public License (GNU LGPL)
#ifndef TREEGENERS_H
#define TREEGENERS_H
#include <Eigen/Dense>
#include "geners/GenericIO.hh"
#include "geners/vectorIO.hh"
#include "geners/arrayIO.hh"
#include "StOpt/tree/Tree.h"
/** \file TreeGeners.h
* \file Define serialization with geners
* \author Xavier Warin
*/
/// specialize the ClassIdSpecialization template
/// so that a ClassId object can be associated with the class we want to
/// serialize. The second argument is the version number.
///@{
gs_specialize_class_id(StOpt::Tree, 1)
/// an external class
gs_declare_type_external(StOpt::Tree)
///@}
namespace gs
{
//
/// \brief This is how the specialization of GenericWriter should look like
//
template <class Stream, class State >
struct GenericWriter < Stream, State, StOpt::Tree,
Int2Type<IOTraits<int>::ISEXTERNAL> >
{
inline static bool process(const StOpt::Tree &p_tree, Stream &p_os,
State *, const bool p_processClassId)
{
// If necessary, serialize the class id
static const ClassId current(ClassId::makeId<StOpt::Tree >());
const bool status = p_processClassId ? ClassId::makeId<StOpt::Tree >().write(p_os) : true;
// Serialize object data if the class id was successfully
// written out
if (status)
{
gs::write_item(p_os, p_tree.getProba());
gs::write_item(p_os, p_tree.getConnected());
}
// Return "true" on success, "false" on failure
return status && !p_os.fail();
}
};
/// \brief And this is the specialization of GenericReader
//
template <class Stream, class State >
struct GenericReader < Stream, State, StOpt::Tree, Int2Type<IOTraits<int>::ISEXTERNAL> >
{
inline static bool readIntoPtr(StOpt::Tree *&ptr, Stream &p_is,
State *p_st, const bool p_processClassId)
{
if (p_processClassId)
{
static const ClassId current(ClassId::makeId<StOpt::Tree>());
ClassId id(p_is, 1);
current.ensureSameName(id);
}
// Deserialize object data.
std::unique_ptr<std::vector< double > > proba = gs::read_item< std::vector< double > >(p_is);
std::unique_ptr<std::vector< std::vector<std::array<int, 2> > > > connected = gs::read_item< std::vector< std::vector<std::array<int, 2> > > >(p_is);
//Build the object from the stored data
if (ptr)
{
*ptr = StOpt::Tree();
ptr->update(*proba, *connected);
}
else
ptr = new StOpt::Tree(*proba, *connected);
return true;
}
inline static bool process(StOpt::Tree &s, Stream &is,
State *st, const bool p_processClassId)
{
// Simply convert reading by reference into reading by pointer
StOpt::Tree *ps = &s;
return readIntoPtr(ps, is, st, p_processClassId);
}
};
}
#endif
|
