forked from dmangame/dmangame
/
cli.py
66 lines (52 loc) · 1.25 KB
/
cli.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
#! /usr/bin/env python
import ai
import cairo
import glob
import os
import worldmap
import mapobject
import world
import worldtalker
import itertools
import settings
import logging
log = logging.getLogger("CLI")
LIFESPAN = 800
import sys
import os
World = None
AI = []
def main(ai_classes=[]):
w = world.World()
wt = worldtalker.WorldTalker(w)
global World
World = w
for ai in ai_classes:
AI.append(ai(wt))
for ai in AI:
ai._init()
ai_cycler = itertools.cycle(AI)
if settings.SAVE_IMAGES:
surface = cairo.ImageSurface(cairo.FORMAT_ARGB32, 200, 200)
cairo_context = cairo.Context(surface)
for ai in AI:
b = mapobject.Building(wt)
w.buildings[b] = next(ai_cycler)
w.map.placeObject(b,
w.map.getRandomSquare())
for turn in xrange(LIFESPAN):
for ai in AI:
ai._spin()
# try:
# ai.spin()
# except Exception, e:
# log.info("AI raised exception %s, skipping this turn for it" % (e))
w.Turn()
if settings.SAVE_IMAGES:
worldmap.draw_map(cairo_context, 200, 200, AI, w)
log.info("Finished simulating the world")
def end_game():
for ai in AI:
log.info("%s:%s", ai.__class__, ai.score)
if __name__ == "__main__":
main()