Memento 模式

Memento 模式的关键就是要在不破坏封装行的前提下，捕获并保存一个类的内部状态，这样就可以利用该保存的状态实施恢复操作。

/////////Originator.h//////////////////////////////////////////
#pragma  once
#include <string>
using namespace std;
typedef string State ;

class Memento ;
class Originator
{
public:
    Originator();
    Originator(const State& std);
    ~Originator();
    Memento* GreatMemento();
    void SetMemento();
    void RestoreToMemento(); 
    State GetState(); 
    void SetState(const State& sdt); 
    void PrintState(); 

protected:
private:
    State _sdt ;
    Memento* _mt; 
};

/////////Memento.h//////////////////////////////////
#pragma  once
#include <string>
using namespace std;

typedef string State ;


class Memento
{
public:
protected:
private:
    friend class Originator ;
    Memento();
    Memento(const State& sdt);
    ~Memento();
    void SetState(const State& sdt);
    State GetState();
    State _sdt ;

};

////////////Originator.cpp/////////////////////////////////////////////////
#include "Originator.h"
#include "Memento.h"
#include <iostream>
using namespace std;
Originator::Originator(const State& std)
{
    _sdt = std ;
    _mt = 0;
}
Originator::Originator()
{
    _sdt = "" ;
    _mt = 0;
}
void Originator::SetMemento()
{
    if (_mt == 0)
    {
        this->_mt = GreatMemento();
    }
    else
    {
        _mt->SetState(_sdt);
    }
    
}
Memento* Originator::GreatMemento()
{
    return new Memento(_sdt);
}
void Originator::RestoreToMemento()
{
    if (this->_mt != 0)
    {
        _sdt = _mt->GetState();
    }
    
}
void Originator::SetState(const State& sdt)
{
    _sdt = sdt ;
}

State Originator::GetState()
{
    return _sdt;
}

void Originator::PrintState() 
{ 
    cout<<this->_sdt<<"....."<<endl; 
} 
Originator::~Originator()
{
    delete _mt;
}

///////////Memento.cpp/////////////////////////
#include "Memento.h"
#include <string>
using namespace std;

typedef string State ;

Memento::Memento(const State& sdt)
{
    _sdt = sdt ;
}
Memento::Memento()
{

}
void Memento::SetState(const State& sdt)
{
    _sdt = sdt ;
}

State Memento::GetState()
{
    return _sdt ;
}
Memento::~Memento()
{

}

///////////////main.cpp///////////////////////////////////////////////////////////
#include "Memento.h"
#include "Originator.h"
#include <string>
#include <iostream>
using namespace std;

typedef string State ;

int main()
{
    Originator* O = new Originator();
    O->SetState("备忘前状态 Old");
    O->PrintState();

    O->SetMemento();//备忘

    O->SetState("备忘后状态 New");
    O->PrintState();

    O->RestoreToMemento();//恢复

    O->PrintState();

    getchar();
    return 0 ;
}