hint/serializedSTL.cpp at master · StetsonMathCS/hint · GitHub

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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
#include <iostream>
#include <fstream>
#include <string>
#include <map>

#include <boost/archive/text_iarchive.hpp>
#include <boost/archive/text_oarchive.hpp>
/*if saving to xml
#include <boost/archive/xml_iarchive.hpp>
#include <boost/archive/xml_oarchive.hpp>
*/

//for extended classes
#include <boost/serialization/base_object.hpp>
//each of the stl containers have a file such as list.hpp, vector.hpp, set.hpp
#include <boost/serialization/map.hpp>
#include <boost/serialization/string.hpp>
using namespace std;

class inventory
{
	//serialization is private to the inventory  class,
	//but accessible through its "friend"
	friend class boost::serialization::access;
	map<string, int> items;
	//saves or loads an archive depending on the case
	template<class Archive>
	void serialize(Archive & ar, const unsigned int version)
	{
        	//& stands in for << or >> and works with all primitives
		//stl objects also work if you import the respective hpp
							//^see header
		ar & items;
	}
	public:
    		inventory(){}
		virtual int add_item(const string i)
		{
			if(items.find(i)==items.end())
			{
				items[i]=1;
			}
			else
			{
				items[i]++;
			}
			return items[i];
		}
		virtual int remove_item(const string i)
                {
			if(items.find(i)==items.end()||items[i]==1)
			{
				items[i] = 0;
			}
			else
			{
				items[i]--;
			}
			return items[i];
		}
		void empty_items()
		{
			items.clear();
		}
		void print()
		{
			for(auto& i : items)
			{
				cout << i.first << ": " << i.second << endl;
			}
		}
};

class bag : inventory
{
	friend class boost::serialization::access;
	int capacity;
	int held;
	template<class Archive>
	void serialize(Archive & ar, const unsigned int version)
	{
		// serialize base class information
		ar & boost::serialization::base_object<inventory>(*this);
		// serialize new variables
		ar & capacity;
		ar & held;
	}
	public:
		bag(const int c){
			capacity = c;
		}
		int add_item(string i)
		{
			if(held<capacity)
			{
				held++;
				return inventory::add_item(i);
			}
			return -1;
		}
		int remove_item(string i)
                {
			if(held>0)
			{
				held--;
				return inventory::remove_item(i);
			}
			return -1;
		}
};

//saves serialized archive to a file
void save_inventory(const inventory &s, const char * filename)
{
	ofstream ofs(filename);
	//if xml, boost::archive::xml_oarchive oa(ofs);
	boost::archive::text_oarchive oa(ofs);
	//serialize(oa, 1)
	oa << s;
}


//load serialized archive to a file
void load_inventory(inventory &s, const char * filename)
{
        ifstream ifs(filename);
        //if xml, boost::archive::xml_iarchive ia(ifs);
	boost::archive::text_iarchive ia(ifs);
	//serialize(oa, 1)
        ia >> s;
}

/*alternative non member serialization method
namespace boost {
namespace serialization {

template<class Archive>
void serialize(Archive & ar, inventory & i, const unsigned int version)
{
	ar & i.items;
}*/

int main()
{
	inventory i;
	string file = "savefile.txt"; //or savefile.xml
	i.add_item("apple");
	i.add_item("apple");
	i.add_item("orange");
	//creates or changes savefile.txt to store the inventory
	//savefile.txt can be found in ls -a or viewed in vim
	save_inventory(i, file.c_str());
	i.add_item("knife");
	//loads the saved list
	//knife is lost as it was not saved
	load_inventory(i, file.c_str());
	i.print();
	cout << getenv("HOME") << endl;
}