target.y = (screen.get_height() - target.y) / 2
if not paused or go:
+
for obj in list(active):
new = obj.step()
total += len(new)
clock.tick(60)
+ for event in pygame.event.get():
+ if event.type == pygame.QUIT:
+ raise SystemExit
+ elif event.type == pygame.KEYDOWN:
+ if event.key == pygame.K_SPACE:
+ paused ^= True
+ elif event.key == pygame.K_RIGHT:
+ go = True
+
print " Finished: %04d: %d bullets." % (frames, total)
if __name__ == "__main__":
def __init__(self, owner, parent, actions, params, rank):
self.actions = actions
self.parent = parent
- self.repeat = 0
+ self.repeat = 1
self.wait_frames = 0
self.speed = 0
self.speed_frames = 0
if parent:
self.copy_state(parent)
+ def __repr__(self):
+ return "%s(pc=%r, actions=%r)" % (
+ type(self).__name__, self.pc, self.actions)
+
+ @property
+ def finished(self):
+ return self.pc is None
+
def vanish(self):
if self.parent:
self.parent.vanish()
- self.repeat = 0
self.pc = None
def copy_state(self, other):
if self.speed_frames > 0:
self.speed_frames -= 1
self.owner.speed += self.speed
+
if self.direction_frames > 0:
self.direction_frames -= 1
if self.direction_frames <= 0:
self.owner.direction += self.owner.aim
else:
self.owner.direction += self.direction
+
if self.accel_frames > 0:
self.accel_frames -= 1
self.owner.mx += self.mx
while True:
self.pc += 1
+
if self.pc >= len(self.actions):
self.repeat -= 1
if self.repeat <= 0:
self.pc = None
if self.parent is not None:
self.parent.copy_state(self)
- self.owner.replace(self, self.parent)
+ self.owner.replace(self, self.parent)
break
else:
self.pc = 0
if rank is None:
rank = parent.rank if parent else 0.5
for action, params in actions:
- self.actions.append(
- Action(self, parent, action, params, rank))
+ # New bullets reset the parent hierarchy.
+ self.actions.append(Action(self, None, action, params, rank))
def __repr__(self):
return ("%s(%r, %r, accel=%r, direction=%r, speed=%r, "
@property
def finished(self):
- return self.vanished and not self.actions
+ for action in self.actions:
+ if not action.finished:
+ return False
+ return self.vanished
def vanish(self):
"""Vanish this bullet and stop all actions."""
def step(self):
created = []
- self.actions = filter(None, self.actions)
-
for action in self.actions:
created.extend(action.step())
direction = math.radians(self.direction)
self.x += self.mx + math.sin(direction) * self.speed
- self.y += self.my + math.cos(direction) * self.speed
+ self.y += self.my - math.cos(direction) * self.speed
return created