duck.h

来自「head first 设计模式中的策略模式的C++实现代码」· C头文件 代码 · 共 64 行

#include<iostream>
using namespace std;

#include "QuackBehavior.h"
#include "FlyBehavior.h"

class Duck
{
public:
	QuackBehavior *quackBehavior;
	FlyBehavior *flyBehavior;

	Duck(){}

	void setFlyBehavior( FlyBehavior *fb)
	{
		flyBehavior = fb;
	}

	void setQuackBehavior( QuackBehavior *qb)
	{
		quackBehavior = qb;
	}

	void performQuack()
	{
		quackBehavior->quack();
	}

	void performFly()
	{
		flyBehavior->fly();
	}

	void swim()
	{
		cout<<"All duck float, even decoys!"<<endl;
	}
};


class MallarDuck: public Duck
{
public:
	MallarDuck()
	{
		quackBehavior = new Quack();
		flyBehavior = new FlyWithWings();
	}
};

class ModelDuck:public Duck
{
public:
	ModelDuck()
	{
		flyBehavior = new FlyNoWay();
	}

	void display()
	{
		cout<<"I'm a model duck!"<<endl;
	}
};