浅谈设计模式之七——Composite模式

同样地，先上uml图：

组合模式的意图：将对象组合合成树形结构以表示“部分-整体”的层次结构。Composite使得用户对单个对象和组合对象的使用具有一致性。
uml解析：leaf是叶节点，Composite可以理解为是一个多个叶节点的组合，它是非叶节点，就相当于一个元件以及一个元件容器，里面有很多元件一样。这里可以用list或者vector实现。
本来是想用模板来实现，后来发现实在是多此一举，而且反而更不好了。
这里需要说的一点就是Leaf是叶节点，因此肯定没有添加、删除等功能，因为这些是没有意义的，这一点也是组合模式需要考虑的问题吧，这样会使得操作Leaf跟Composite会有少许的不同。
下面给出可执行代码：
Composition.h

#pragma once
#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
using namespace std;

/*template<typename T>*/
class Component
{
public:
    Component(void);
    virtual ~Component(void);

public:
    virtual void Display() = 0;
    virtual void Add(Component &leaf){}
    virtual bool Remove(Component &leaf){return false;}
    virtual Component * GetChild(int Index)=0;
protected:
    std::string m_string;

};


class Composite:public Component
{
public:
    Composite(){}
    ~Composite(){}
    void Display();
    void Add(Component &leaf);
    bool Remove(Component &leaf);
    Component* GetChild(int Index);
private:
    vector<Component *> m_Com;
};

// template<typename T>
class Leaf:public Component
{
public:
    Leaf(std::string & str);
    Leaf(){}
    void Display();
//  bool Add(Leaf &leaf){}
//  bool Remove(){}
    Component* GetChild(int Index){ return NULL;}
private:
    std::string mstring;
};

Composition.cpp

#include "Component.h"

Component::Component()
{
}

Component::~Component()
{
}

void Composite::Add(Component &leaf) 
{
    m_Com.push_back(&leaf);
}

/*template<typename T>*/
bool Composite::Remove(Component &leaf)
{

    vector<Component *>::iterator itor=find(m_Com.begin(),m_Com.end(),&leaf);
    if (itor == m_Com.end())
    {
        return false;
    }
    else
    {
        m_Com.erase(itor);
        return true;
    }   
}

Component * Composite::GetChild(int Index)
{
    return m_Com[Index];
};

void Composite::Display()
{
    vector<Component*>::iterator itor = m_Com.begin();
    for (;itor!=m_Com.end();itor++)
    {
        (*itor)->Display();
    }
}

Leaf::Leaf(std::string &str):mstring(str)
{
}

// template<typename T>
void Leaf::Display()
{
    cout<<mstring<<endl;
}

main.cpp

#include "Component.h"

int main(int argc,char **argv)
{
    std::string str1="Leaf1";
    std::string str2="Leaf2";
    std::string str3="Leaf3";
    Leaf leaf1(str1);
    Leaf leaf2(str2);
    Leaf leaf3(str3);

    Component *f = new Composite;
    f->Add(leaf1);
    f->Add(leaf2);
    f->Add(leaf3);

    f->Display();

    f->Remove(leaf2);
    f->Display();

    return 0;
}

执行之后的结果：

一开始显示Leaf1、Leaf2、Leaf3,删除之后显示Leaf1、Leaf3。
这就是组合模式，祝读者生活快乐。