使用面向对象概念优化条件判断语句的一个小应用

Question

转：飘啊飘  

使用面向对象概念优化条件判断语句的一个小应用

以前就知道可以通过面向对象来优化条件语句，甚至看到过说面向对象概念可以完全替代条件语句，新想那得弄多少类啊，觉得太累了就没细研究，而且那时是在用C语言，没有面向对象的语法支持。

今天写代码时候，写出了这样的代码：

2.                 if isinstance(b,Iron):#铁
3.                     if self._size == Bullet.BIG:
4.                         b.health -= 1
5.                     elif self._size == Bullet.LARGE:
6.                         b.health -= 3
7.                     elif self._size == Bullet.HUGE:
8.                         b.health -= 5
10.                 if isinstance(b,Soil):#土
11.                     if self._size == Bullet.SMALL:
12.                         b.health -= 1
13.                     elif self._size == Bullet.BIG:
14.                         b.health -= 5

复制代码又臭又长又丑，看了下，也没什么好办法改，条件总是要判断的呀，铁和转块受到攻击时效果不一样，受到不同大小子弹的攻击，效果不一样。
突然灵光一闪，想起了还有继承这么回事。三下五除二，写下了这样的代码：

2. class Floor(pygame.sprite.Sprite):
3.     """
4.     Stage的组成部分。Stage向Game对象传递进来的pygame.sprite.Group对象引用进行填充。
5.     这是个基类，不同种类的格子通过这个来继承。
6.     """
7.     image = None
8.    
9.     def _on_small_bullet(self):
10.         """收到小子弹攻击"""
11.         raise TypeError,"not implement"
12.    
13.     def _on_big_bullet(self):
14.         """收到大子弹攻击"""
15.         raise TypeError,"not implement"
16.    
17.     def _on_large_bullet(self):
18.         """收到巨大子弹攻击"""
19.         raise TypeError,"not implement"
20.     def _on_huge_bullet(self):
21.         """收到炮弹攻击"""
22.         raise TypeError,"not implement"
25. class Soil(Floor):
26.     """土"""
27.     def __init__(self,pos):
28.         Floor.__init__(self)
29.         self.pos = pos
30.         self.health = 5
31.         self.image = Soil.image
32.         self.rect = self.image.get_rect(topleft = pos)
33. #        self.rect = Rect(self.rect)
34.         pass
35.         
36.     def _on_small_bullet(self):
37.         """收到小子弹攻击"""
38.         self.health -= 1
39.    
40.     def _on_big_bullet(self):
41.         """收到大子弹攻击"""
42.         self.health -= 2
43.    
44.     def _on_large_bullet(self):
45.         """收到巨大子弹攻击"""
46.         self.health -= 3
47.     def _on_huge_bullet(self):
48.         """收到炮弹攻击"""
49.         self.health -= 5

复制代码啊哈，搞定了，而且代码好看很多。
所有砖块元素都继承自Floor，这样就不需要判断类型了。

1. if self._size == Bullet.SMALL:
2.                     b._on_small_bullet()
3.                 elif self._size == Bullet.BIG:
4.                     b._on_big_bullet()
5.                 elif self._size == Bullet.LARGE:
6.                     b._on_large_bullet()
7.                 elif self._size == Bullet.HUGE:
8.                     b._on_huge_bullet()

复制代码代码好看了很多。
还是得多写代码啊。。