# HG changeset patch
# User Jeremy Thurgood <firxen@gmail.com>
# Date 1260094297 0
# Node ID 23fbf4651f9974b1a5f3739108b057a62d19f876
# Parent  f9276a4f7bdfa033d74220ba262b6e40af223b15
Roosters.

diff -r f9276a4f7bdf -r 23fbf4651f99 gamelib/animal.py
--- a/gamelib/animal.py	Thu Dec 03 19:29:54 2009 +0000
+++ b/gamelib/animal.py	Sun Dec 06 10:11:37 2009 +0000
@@ -228,9 +228,14 @@
     def lay(self):
         """See if the chicken lays an egg"""
         if self.abode and self.abode.building.HENHOUSE:
+            # TODO: Find a cleaner way to do this
+            fertilised = False
+            for bird in self.abode.building.occupants():
+                if getattr(bird, 'ROOSTER', None):
+                    fertilised = True
             if not self.eggs:
                 for x in range(random.randint(1, 4)):
-                    self.eggs.append(Egg(self.pos, self.gameboard))
+                    self.eggs.append(Egg(self.pos, self.gameboard, fertilised=fertilised))
                 self.equip(equipment.NestEgg())
             self.gameboard.eggs += self.get_num_eggs()
 
@@ -295,25 +300,52 @@
         for weapon in self.weapons():
             weapon.refresh_ammo()
 
+
+class Rooster(Chicken):
+    """A rooster"""
+
+    IMAGE_FILE = 'sprites/rooster.png'
+    ROOSTER = True
+
+    AGGRESSION = 50
+
+    def lay(self):
+        # Roosters don't lay eggs.
+        pass
+
+    def start_night(self):
+        Chicken.start_night(self)
+        self._manly_fight()
+
+    def _manly_fight(self):
+        if self.abode:
+            for rival in [occ for occ in self.abode.building.occupants()
+                          if getattr(occ, 'ROOSTER', False)]:
+                if random.randint(1, 100) <= self.AGGRESSION:
+                    rival.damage()
+
+
 class Egg(Animal):
     """An egg"""
 
     IMAGE_FILE = 'sprites/equip_egg.png'
 
-    SIMPLIFY = Animal.SIMPLIFY + ['timer']
+    SIMPLIFY = Animal.SIMPLIFY + ['timer', 'fertilised']
 
-    def __init__(self, pos, gameboard):
+    def __init__(self, pos, gameboard, fertilised=False):
         Animal.__init__(self, pos, gameboard)
+        self.fertilised = fertilised
         self.timer = 2
 
     # Eggs don't move
 
     def hatch(self):
         self.timer -= 1
-        if self.timer == 0:
-            return Chicken(self.pos, self.gameboard)
+        if self.timer == 0 and self.fertilised:
+            return random.choice([Chicken, Rooster])(self.pos, self.gameboard)
         return None
 
+
 class Fox(Animal):
     """A fox"""
 
diff -r f9276a4f7bdf -r 23fbf4651f99 gamelib/gameboard.py
--- a/gamelib/gameboard.py	Thu Dec 03 19:29:54 2009 +0000
+++ b/gamelib/gameboard.py	Sun Dec 06 10:11:37 2009 +0000
@@ -1094,7 +1094,7 @@
 
     def add_start_chickens(self, _map, tile, value):
         """Add chickens as specified by the code layer"""
-        chick = animal.Chicken((tile.tx, tile.ty), self)
+        chick = random.choice([animal.Chicken, animal.Rooster])((tile.tx, tile.ty), self)
         for equip_cls in equipment.EQUIP_MAP[value]:
             item = equip_cls()
             chick.equip(item)