//Memento.h

#ifndef _MEMENTO_H_ 
#define _MEMENTO_H_

#include <string> 
using namespace std;

class Memento;

class Originator 
{ 
public: 
	typedef string State;

	Originator();
	Originator(const State& sdt);
	~Originator();

	Memento* CreateMemento();
	void SetMemento(Memento* men);
	void RestoreToMemento(Memento* mt);

	State GetState();
	void SetState(const State& sdt);
	void PrintState();
protected:
private: 
	State _sdt;
	Memento* _mt; 
};

class Memento 
{ 
public:
protected:
private: 
	//這是最關鍵的地方，將Originator為friend類，可以訪問內部信息，
	//但是其他類不能訪問
	friend class Originator; 
	typedef string State;
	Memento();
	Memento(const State& sdt);
	~Memento();
	void SetState(const State& sdt);
	State GetState();
private: 
	State _sdt;
};

#endif