//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 //~_MEMENTO_H_