【C++设计模式】装饰者模式

#ifndef __DECORATOR_H__
#define __DECORATOR_H__

#include <string>

//装饰者模式（Decorator）：动态地给一个对象添加一些额外的职责。就增加功能来说，Decorator模式比生成子类更为灵活。

//它的工作原理是：创建一个始于Decorator对象（负责新功能的对象）终止于原对象的一个对象“链”。

//不需要对被装饰的类做任何修改即可增加新的功能。

//装饰者模式包括以下组成部分：
//1、Component（抽象构件）角色：给出一个抽象接口，以规范要接收附加责任的对象。 
//2、Concrete Component（具体构件）角色：定义一个将要接收附加责任的类。 
//3、Decorator（装饰）角色：持有一个Component对象的实例，并定义一个与抽象构件接口一致的接口。 
//4、Concrete Decorator（具体装饰）角色：负责给构件对象“贴上”附加的责任。

class Display
{
public:
	Display();
	virtual ~Display();      //虚析构函数保证通过基类指针释放派生类对象

public:
	virtual int GetColumns() = 0;

	virtual int GetRows() = 0;

	virtual std::string GetRowText(int row) = 0;

	void Show();
};

//具体的被装饰物，当增加新功能时，我们不对它做任何修改
class StringDisplay : public Display
{
public:
	StringDisplay(const std::string &str);
	virtual ~StringDisplay();

public:

	virtual int GetColumns();

	virtual int GetRows();

	virtual std::string GetRowText(int row);

private:
	std::string m_str;
};

//装饰者：通过继承和委托，装饰边框与被装饰物具有相同的接口
class Border : public Display
{
public:
	Border();
	Border(Display * display);
	virtual ~Border();

protected:
	Display * m_display;
};

//具体装饰者A
class SideBorder : public Border
{
public:
	SideBorder();
	SideBorder(Display * display, const std::string & ch);
	virtual ~SideBorder();

	virtual int GetColumns();

	virtual int GetRows();

	virtual std::string GetRowText(int row);

private:
	std::string m_ch;
};

//具体装饰者B
class FullBorder : public Border
{
public:
	FullBorder();
	FullBorder(Display * display);
	virtual ~FullBorder();

	virtual int GetColumns();

	virtual int GetRows();

	virtual std::string GetRowText(int row);
};

void TestDecorator();



#endif


#include "Decorator.h"


Display::Display()
{

}

Display::~Display()
{

}

void Display::Show()
{
	for (int i=0; i<GetRows(); i++)
	{
		printf("%s \n", GetRowText(i).c_str());
	}
}

StringDisplay::StringDisplay(const std::string &str)
{
	m_str = str;
}

StringDisplay::~StringDisplay()
{

}

int StringDisplay::GetColumns()
{
	return m_str.size();
}

int StringDisplay::GetRows()
{
	return 1;
}

std::string StringDisplay::GetRowText(int row)
{
	return m_str.c_str();
}

Border::Border()
{

}

Border::Border(Display * display)
{
	m_display = display;
}

Border::~Border()
{
	if (m_display)
	{
		delete m_display;
	}
}

SideBorder::SideBorder()
{

}

SideBorder::SideBorder(Display * display, const std::string & ch) : Border(display), m_ch(ch)
{

}

SideBorder::~SideBorder()
{

}

int SideBorder::GetColumns()
{
	return 1 + m_display->GetColumns() + 1;
}

int SideBorder::GetRows()
{
	return m_display->GetRows();
}

std::string SideBorder::GetRowText(int row)
{
	return m_ch + m_display->GetRowText(row) + m_ch;
}

FullBorder::FullBorder()
{

}

FullBorder::FullBorder(Display * display) : Border(display)
{

}

FullBorder::~FullBorder()
{

}

int FullBorder::GetColumns()
{
	return 1 + m_display->GetColumns() + 1;
}

int FullBorder::GetRows()
{
	return 1 + m_display->GetRows() + 1;
}

std::string FullBorder::GetRowText(int row)
{
	if (row == 0)
	{
		return "+-------------------------------+";
	}
	else if (row == m_display->GetRows() + 1)
	{
		return "+-------------------------------+";
	}
	else
	{
		return "|" + m_display->GetRowText(row) + "|";
	}
}

void TestDecorator()
{
	//被装饰物
	Display * b1 = new StringDisplay("Hello world");

	b1->Show();

	//装饰模式支持递归，但要注意对象的内存释放
	Display * b2 = new SideBorder(new FullBorder(new SideBorder(b1, "#")),"*");

	b2->Show();

	delete b2;	
}

【C++设计模式】装饰者模式

猜你喜欢