C++设计模式-Composite组合模式

ref:http://www.jellythink.com/archives/149

 

直接上源码：

#include <iostream>
#include <string>
#include <sstream>
#include <memory>
#include <vector>

using namespace std;

// 抽象的部件类描述将来所有部件共有的行为
class Component
{
public:
    Component(string name) : m_strCompname(name){}
    virtual ~Component(){}
    virtual void Operation() = 0;
    virtual void Add(Component *) = 0;
    virtual void Remove(Component *) = 0;
    virtual Component *GetChild(int) = 0;
    virtual string GetName()
    {
        return m_strCompname;
    }
    virtual void Print() = 0;
protected:
    string m_strCompname;
};
class Leaf : public Component
{
public:
    Leaf(string name) : Component(name)
    {}
    void Operation()
    {
        cout << "I'm " << m_strCompname << endl;
    }
    void Add(Component *pComponent){}
    void Remove(Component *pComponent){}
    Component *GetChild(int index)
    {
        return NULL;
    }
    void Print(){}
};
class Composite : public Component
{
public:
    Composite(string name) : Component(name)
    {}
    ~Composite()
    {
        vector<Component *>::iterator it = m_vecComp.begin();
        while (it != m_vecComp.end())
        {
            if (*it != NULL) {
                cout << "----delete " << (*it)->GetName() << "----" << endl;
                delete *it;
                *it = NULL;
            }
            it = m_vecComp.erase(it);
        }
    }
    void Operation()
    {
        cout << "I'm " << m_strCompname << endl;
    }
    void Add(Component *pComponent)
    {
        m_vecComp.push_back(pComponent);
    }
    void Remove(Component *pComponent)
    {
        for (vector<Component *>::iterator it = m_vecComp.begin(); it != m_vecComp.end(); ++it)
        {
            if ((*it)->GetName() == pComponent->GetName())
            {
                if (*it != NULL)
                {
                    delete *it;
                    *it = NULL;
                }
                m_vecComp.erase(it);
                break;
            }
        }
    }
    Component *GetChild(int index)
    {
        if (index > m_vecComp.size())
        {
            return NULL;
        }
        return m_vecComp[index - 1];
    }
    void Print()
    {
        for (vector<Component *>::iterator it = m_vecComp.begin(); it != m_vecComp.end(); ++it)
        {
            (*it)->Print();
            cout << (*it)->GetName() << endl;
        }
    }
private:
    vector<Component *> m_vecComp;
};
int main(int argc, char *argv[])
{
    Component *pNode = new Composite("Beijing Head Office");
    Component *pNodeHr = new Leaf("Beijing Human Resources Department");
    Component *pSubNodeSh = new Composite("Shanghai Branch");
    Component *pSubNodeCd = new Composite("Chengdu Branch");
    Component *pSubNodeBt = new Composite("Baotou Branch");
    pNode->Add(pNodeHr);
    pNode->Add(pSubNodeSh);
    pNode->Add(pSubNodeCd);
    pNode->Add(pSubNodeBt);
    pNode->Print();
    Component *pSubNodeShHr = new Leaf("Shanghai Human Resources Department");
    Component *pSubNodeShCg = new Leaf("Shanghai Purchasing Department");
    Component *pSubNodeShXs = new Leaf("Shanghai Sales department");
    Component *pSubNodeShZb = new Leaf("Shanghai Quality supervision Department");
    pSubNodeSh->Add(pSubNodeShHr);
    pSubNodeSh->Add(pSubNodeShCg);
    pSubNodeSh->Add(pSubNodeShXs);
    pSubNodeSh->Add(pSubNodeShZb);
    pNode->Print();
    // 公司不景气，需要关闭上海质量监督部门
    pSubNodeSh->Remove(pSubNodeShZb);
    if (pNode != NULL)
    {
        delete pNode;
        pNode = NULL;
    }
    getchar();
    return 0;
}

 

转载于:https://www.cnblogs.com/kaishan1990/p/4979822.html