设计模式学习之----组合模式

　　单个对象和组合对象的使用具有一致性。将对象组合成树形结构以表示"部分--整体"

　　例如：文件和目录的显示，单个对象是文件，组合对象是目录，一致性是显示。

#include <iostream>
#include <string>
#include <list>

using namespace std;
class IFile
{
public:
    virtual void display() = 0;
    virtual int add(IFile *iFile) = 0;
    virtual int remove(IFile *iFile) = 0;
    virtual list<IFile*>* getChild() = 0;
};

//文件节点
class File : public IFile
{
public:
    File(string name)
    {
        m_name = name;
    }
    virtual void display()
    {
        cout << m_name << endl;
    }

    virtual int add(IFile *iFile)
    {
        return -1;
    }
    virtual int remove(IFile *iFile)
    {
        return -1;
    }
    virtual list<IFile*>* getChild()
    {
        return NULL;
    }

private:
    string m_name;
};

//目录节点
class Dir : public IFile
{
public:
    Dir(string name)
    {
        m_name = name;
        m_list = new list<IFile *>;
    }

    virtual void display()
    {
        cout << m_name << endl;
    }

    virtual int add(IFile *iFile)
    {
        m_list->push_back(iFile);
        return 1;
    }

    virtual int remove(IFile *iFile)
    {
        m_list->remove(iFile);
        return 1;
    }

    virtual list<IFile*>* getChild()
    {
        return m_list;
    }

private:
    string m_name;
    list<IFile*> *m_list;
};

int main()
{
    Dir *root = new Dir("C.....");
    root->display();
    Dir *dir1   = new Dir("1111dir1");
    File *file1 = new File("file1");
    root->add(dir1);
    root->add(file1);

    using iter = list<IFile*>::iterator;
    list<IFile*> *list = root->getChild();
    for (iter it = list->begin(); it != list->end(); it++)
    {
        (*it)->display();
    }

    return system("pause");
}