《设计模式》学习笔记——组合模式

组合模式（Composite Pattern）将对象组合成树形结构以表示"部分-整体"的层次结构，组合模式使得用户对单个对象和组合对象的使用具有一致性。掌握组合模式的重点是要理解清楚 "部分/整体" 还有 "单个对象" 与 "组合对象" 的含义。组合模式可以让客户端像修改配置文件一样简单的完成本来需要流程控制语句来完成的功能。

CompositePattern.h

#pragma once
#include<iostream>
#include<string>
#include<list>


// 抽象接口
class IFile
{
public:
	virtual void displayd() = 0;
	virtual int add(IFile* ifile) = 0;
	virtual int remove(IFile* ifile) = 0;
	virtual std::list<IFile*>* getChiled() = 0;
};

// 具体类,文件节点
class File:public IFile
{
public:
	File(std::string fileName): m_fileName(fileName){}
	virtual void displayd()
	{
		std::cout << m_fileName << std::endl;
	}
	virtual int add(IFile* ifile)
	{
		return -1;
	}
	virtual int remove(IFile* ifile)
	{
		return -1;
	}
	virtual std::list<IFile*>* getChiled()
	{
		return nullptr;
	}
private:
	std::string m_fileName;
};


// 具体实现，目录节点
class Dir:public IFile
{
public:
	Dir(std::string dirName) : m_dirName(dirName) { m_list_p = new std::list<IFile*>; m_list_p->clear(); }
	virtual void displayd()
	{
		std::cout << m_dirName << std::endl;
	}
	virtual int add(IFile* ifile)
	{
		m_list_p->push_back(ifile);
		return 0;
	}
	virtual int remove(IFile* ifile)
	{
		m_list_p->remove(ifile);
		return 0;
	}
	virtual std::list<IFile*>* getChiled()
	{
		return m_list_p;
	}
private:
	std::string m_dirName;
	std::list<IFile*>* m_list_p;
};

class CompositePattern
{
public:
	CompositePattern() {};
	~CompositePattern() {};
};

 

CompositePattern.cpp

#include "CompositePattern.h"

 

mainTest.cpp

#include<iostream>
#include"CompositePattern.h"


void showTree(IFile* root,int level);
int main(void)
{
	Dir* root = new Dir("C");
	//root->displayd();

	Dir* dir = new Dir("dir");
	File* file = new File("file.txt");

	// 添加
	root->add(dir);
	root->add(file);
	// 获取root节点的孩子集合
	std::list<IFile*>* list = root->getChiled();

	for (std::list<IFile*>::iterator it = list->begin(); it != list->end(); it++)
		(*it)->displayd();


	Dir* dir2 = new Dir("dir2");
	File* file2 = new File("file2.txt");
	dir->add(dir2);
	dir->add(file2);

	showTree(root,0);

	system("pause");
	return 0;
}

void showTree(IFile* root, int level)
{
	for (int i=0;i<level;i++)
		std::cout << "\t";

	// 如果root不空
	if (nullptr == root)	return;
		// 1、显示根节点
	root->displayd();

		// 2、如果根节点有孩子
	std::list<IFile*>* childList = root->getChiled();		// 获取子节点集合
	if (nullptr != childList)		// 说明是目录
	{
		for (std::list<IFile*>::iterator it = childList->begin(); it != childList->end(); it++)
		{
			if (nullptr == (*it)->getChiled())
			{
				for(int i=0; i<= level;i++)
					std::cout << "\t";
				(*it)->displayd();// 如果是文件，显示文件名
			}
			else
			{
				showTree((*it), level+1);// 如果是目录，递归调用showTree
			}
		}
			
	}
				
				
}