Memeto_Pattern - 8BitsCoding/RobotMentor GitHub Wiki

void main() {
    Caretaker mCaretaker;
    Originator* originator = new Originator();

    originator->setState(_T("state1"));
    mCaretaker.pushMemento(originator->createMemento());

    originator->setState(_T("state2"));
    mCaretaker.pushMemento(originator->createMemento());

    originator->setMemeto(mCaretaker.popMemento());
    originator->setMemeto(mCaretaker.popMemento());

    delete originator;
}

//------------------------------------------------------------------
// Caretaker 클래스 (Memento 관리)
class Caretaker
{
public:
    void pushMemento(Memento* m) { mStack.push(m); }
    Memento* popMemento() { Memento* m = mStack.top(); mStack.pop(); return m; }
 
private:
    stack<Memento*> mStack;
};

// Originator 클래스
class Originator
{
public:
    void setState(tstring _state) 
    { 
        state = _state;
        wcout << _T("Originator: Setting state to ") << state << endl;
    }
 
public:
    void setMemento(Memento *m) 
    {
        if (m) 
        {
            state = m->getState(); 
            delete m; 
            wcout << _T("Originator: State after restoring from Memento ") << state << endl; 
        } 
    }
 
    Memento* createMemento() 
    {
        wcout << _T("Originator: Create to Memento ") << state << endl;
        return new Memento(state); 
    }
 
private:
    tstring state;
};

// Memento 클래스
class Memento
{
public:
    Memento(tstring _state) { state = _state; }
 
public:
    tstring getState() const { return state; }
 
private:
    tstring state;
};