demo_tank.py

# Copyright (C) 2014 Marvin Poul <ponder@creshal.de>
import math

from copanzers.scripts import get_logger

def destroy_target (w, target):
    if w.till_reloaded <= 0:
        w.shoot ()
    return target.destroyed

def drive_circle (t, game, length):

    start = game.time
    while start + length > game.time:
        t.rotation = (game.time - start) / length * math.pi * 2
        yield (lambda: True)

def main (tank, game):

    x, y = tank.position

    tank.throttle = 1
    tank.rotation = - math.pi / 2
    yield (lambda: tank.position.y <= 50)

    tank.rotation = 0
    yield (lambda: tank.position.x >= 500)

    tank.rotation = math.pi / 2
    yield (lambda: tank.position.y >= 250)

    tank.rotation = - 5/6 * math.pi
    tank.throttle = 1
    yield (lambda: tank.position.x <= x)

    diff = y - tank.position.y
    tank.rotation = math.pi / 2 * diff / abs (diff)

    log = get_logger ("DemoTank")
    log.info ("Imma driving a circle!")
    tank.throttle = .5
    yield from drive_circle (tank, game, 4)

    tank.throttle = 0
    tank.rotation = 0

    log.info ("I'll be shooting everything in a radius of 200px.")
    cannon = tank.mounts [0]
    radar  = tank.mounts [1]

    for o in radar.visible:
        d = o.position - tank.position
        if abs (d) <= 200 and o != tank and o not in tank.mounts:
            log.info ("Aiming at %s", o.e)
            cannon.rotation = d.angle
            yield (lambda: destroy_target(cannon, o))