6 from datetime import datetime
7 import sqlalchemy as sa
8 import sqlalchemy.sql.functions as func
9 from colorsys import rgb_to_hls, hls_to_rgb
11 from pyramid.paster import bootstrap
12 from xonstat.models import *
13 from xonstat.util import qfont_decode, _all_colors
16 # similar to html_colors() from util.py
17 _contrast_threshold = 0.5
19 _dec_colors = [ (0.5,0.5,0.5),
32 # maximal number of query results (for testing, set to 0 to get all)
35 # image dimensions (likely breaks layout if changed)
39 # output filename, parameter is player id
40 OUTPUT = "output/%d.png"
52 KILLDEATH_POS = (390,6)
55 PLAYTIME_POS = (452,64)
59 # parameters to affect the output, could be changed via URL
61 'bg': 1, # 0 - black, 1 - dark_wall, ...
62 'overlay': 0, # 0 - none, 1 - default overlay, ...
63 'font': 0, # 0 - xolonium, 1 - dejavu sans
67 # parse cmdline parameters (for testing)
68 for arg in sys.argv[1:]:
71 if params.has_key(k.lower()):
78 """Return player data as dict.
80 This function is similar to the function in player.py but more optimized
87 # duel/dm/tdm/ctf elo + rank
89 player_id = player.player_id
93 games_played = DBSession.query(
94 Game.game_type_cd, func.count(), func.sum(PlayerGameStat.alivetime)).\
95 filter(Game.game_id == PlayerGameStat.game_id).\
96 filter(PlayerGameStat.player_id == player_id).\
97 group_by(Game.game_type_cd).\
98 order_by(func.count().desc()).\
99 limit(3).all() # limit to 3 gametypes!
101 total_stats['games'] = 0
102 total_stats['games_breakdown'] = {} # this is a dictionary inside a dictionary .. dictception?
103 total_stats['games_alivetime'] = {}
104 total_stats['gametypes'] = []
105 for (game_type_cd, games, alivetime) in games_played:
106 total_stats['games'] += games
107 total_stats['gametypes'].append(game_type_cd)
108 total_stats['games_breakdown'][game_type_cd] = games
109 total_stats['games_alivetime'][game_type_cd] = alivetime
111 (total_stats['kills'], total_stats['deaths'], total_stats['alivetime'],) = DBSession.query(
112 func.sum(PlayerGameStat.kills),
113 func.sum(PlayerGameStat.deaths),
114 func.sum(PlayerGameStat.alivetime)).\
115 filter(PlayerGameStat.player_id == player_id).\
118 # (total_stats['wins'],) = DBSession.query(
120 # filter(Game.game_id == PlayerGameStat.game_id).\
121 # filter(PlayerGameStat.player_id == player_id).\
122 # filter(Game.winner == PlayerGameStat.team or PlayerGameStat.rank == 1).\
125 (total_stats['wins'],) = DBSession.\
126 query("total_wins").\
128 "select count(*) total_wins "
129 "from games g, player_game_stats pgs "
130 "where g.game_id = pgs.game_id "
131 "and player_id=:player_id "
132 "and (g.winner = pgs.team or pgs.rank = 1)"
133 ).params(player_id=player_id).one()
135 ranks = DBSession.query("game_type_cd", "rank", "max_rank").\
137 "select pr.game_type_cd, pr.rank, overall.max_rank "
138 "from player_ranks pr, "
139 "(select game_type_cd, max(rank) max_rank "
141 "group by game_type_cd) overall "
142 "where pr.game_type_cd = overall.game_type_cd "
143 "and player_id = :player_id "
145 params(player_id=player_id).all()
148 for gtc,rank,max_rank in ranks:
149 ranks_dict[gtc] = (rank, max_rank)
151 elos = DBSession.query(PlayerElo).\
152 filter_by(player_id=player_id).\
153 order_by(PlayerElo.elo.desc()).\
159 elos_dict[elo.game_type_cd] = elo.elo
163 'total_stats':total_stats,
172 def render_image(data):
173 """Render an image from the given data fields."""
175 if params['font'] == 1:
178 total_stats = data['total_stats']
179 total_games = total_stats['games']
181 ranks = data["ranks"]
186 surf = C.ImageSurface(C.FORMAT_RGB24, WIDTH, HEIGHT)
187 ctx = C.Context(surf)
188 ctx.set_antialias(C.ANTIALIAS_GRAY)
190 # draw background (just plain fillcolor)
191 if params['bg'] == 0:
192 ctx.rectangle(0, 0, WIDTH, HEIGHT)
193 ctx.set_source_rgb(0.04, 0.04, 0.04) # bgcolor of Xonotic forum
196 # draw background image (try to get correct tiling, too)
199 if params['bg'] == 1:
200 bg = C.ImageSurface.create_from_png("img/dark_wall.png")
201 elif params['bg'] == 2:
202 bg = C.ImageSurface.create_from_png("img/asfalt.png")
203 elif params['bg'] == 3:
204 bg = C.ImageSurface.create_from_png("img/broken_noise.png")
205 elif params['bg'] == 4:
206 bg = C.ImageSurface.create_from_png("img/burried.png")
207 elif params['bg'] == 5:
208 bg = C.ImageSurface.create_from_png("img/dark_leather.png")
209 elif params['bg'] == 6:
210 bg = C.ImageSurface.create_from_png("img/txture.png")
211 elif params['bg'] == 7:
212 bg = C.ImageSurface.create_from_png("img/black_linen_v2.png")
213 elif params['bg'] == 8:
214 bg = C.ImageSurface.create_from_png("img/background_archer-v1.png")
218 bg_w, bg_h = bg.get_width(), bg.get_height()
220 while bg_xoff < WIDTH:
222 while bg_yoff < HEIGHT:
223 ctx.set_source_surface(bg, bg_xoff, bg_yoff)
228 # draw overlay graphic
229 if params['overlay'] > 0:
231 if params['overlay'] == 1:
232 overlay = C.ImageSurface.create_from_png("img/overlay.png")
234 ctx.set_source_surface(overlay, 0, 0)
235 ctx.mask_surface(overlay)
239 ## draw player's nickname with fancy colors
241 # deocde nick, strip all weird-looking characters
242 qstr = qfont_decode(player.nick).replace('^^', '^').replace(u'\x00', '')
247 qstr = ''.join(chars)
248 stripped_nick = strip_colors(qstr)
250 # fontsize is reduced if width gets too large
251 ctx.select_font_face(font, C.FONT_SLANT_NORMAL, C.FONT_WEIGHT_NORMAL)
252 ctx.set_font_size(20)
253 xoff, yoff, tw, th = ctx.text_extents(stripped_nick)[:4]
254 if tw > NICK_MAXWIDTH:
255 ctx.set_font_size(18)
256 xoff, yoff, tw, th = ctx.text_extents(stripped_nick)[:4]
257 if tw > NICK_MAXWIDTH:
258 ctx.set_font_size(16)
259 xoff, yoff, tw, th = ctx.text_extents(stripped_nick)[:4]
260 if tw > NICK_MAXWIDTH:
261 ctx.set_font_size(14)
262 xoff, yoff, tw, th = ctx.text_extents(stripped_nick)[:4]
263 if tw > NICK_MAXWIDTH:
264 ctx.set_font_size(12)
265 xoff, yoff, tw, th = ctx.text_extents("_")[:4]
268 # split up nick into colored segments and draw each of them
270 # split nick into colored segments
272 _all_colors = re.compile(r'(\^\d|\^x[\dA-Fa-f]{3})')
273 #print qstr, _all_colors.findall(qstr), _all_colors.split(qstr)
275 parts = _all_colors.split(qstr)
276 while len(parts) > 0:
279 if _all_colors.match(txt):
280 tag = txt[1:] # strip leading '^'
287 if not txt or len(txt) == 0:
288 # only colorcode and no real text, skip this
291 r,g,b = _dec_colors[7]
293 if tag.startswith('x'):
294 r = int(tag[1] * 2, 16) / 255.0
295 g = int(tag[2] * 2, 16) / 255.0
296 b = int(tag[3] * 2, 16) / 255.0
297 hue, light, satur = rgb_to_hls(r, g, b)
298 if light < _contrast_threshold:
299 light = _contrast_threshold
300 r, g, b = hls_to_rgb(hue, light, satur)
302 r,g,b = _dec_colors[int(tag[0])]
304 r,g,b = _dec_colors[7]
306 ctx.set_source_rgb(r, g, b)
307 ctx.move_to(NICK_POS[0] + xoffset, NICK_POS[1])
310 xoff, yoff, tw, th = ctx.text_extents(txt)[:4]
311 tw += (len(txt)-len(txt.strip())) * space_w # account for lost whitespaces
315 ## print elos and ranks
317 # show up to three gametypes the player has participated in
319 for gt in total_stats['gametypes'][:3]:
320 ctx.select_font_face(font, C.FONT_SLANT_NORMAL, C.FONT_WEIGHT_BOLD)
321 ctx.set_font_size(10)
322 ctx.set_source_rgb(1.0, 1.0, 1.0)
323 txt = "[ %s ]" % gt.upper()
324 xoff, yoff, tw, th = ctx.text_extents(txt)[:4]
325 ctx.move_to(GAMES_POS[0]+xoffset-xoff-tw/2, GAMES_POS[1]-yoff-4)
329 old_aa = ctx.get_antialias()
330 ctx.set_antialias(C.ANTIALIAS_NONE)
331 ctx.set_source_rgb(0.8, 0.8, 0.8)
332 ctx.set_line_width(1)
333 ctx.move_to(GAMES_POS[0]+xoffset-GAMES_WIDTH/2+5, GAMES_POS[1]+8)
334 ctx.line_to(GAMES_POS[0]+xoffset+GAMES_WIDTH/2-5, GAMES_POS[1]+8)
336 ctx.move_to(GAMES_POS[0]+xoffset-GAMES_WIDTH/2+5, GAMES_POS[1]+32)
337 ctx.line_to(GAMES_POS[0]+xoffset+GAMES_WIDTH/2-5, GAMES_POS[1]+32)
339 ctx.set_antialias(old_aa)
341 if not elos.has_key(gt) or not ranks.has_key(gt):
342 ctx.select_font_face(font, C.FONT_SLANT_NORMAL, C.FONT_WEIGHT_BOLD)
343 ctx.set_font_size(12)
344 ctx.set_source_rgb(0.8, 0.2, 0.2)
345 txt = "no stats yet!"
346 xoff, yoff, tw, th = ctx.text_extents(txt)[:4]
347 ctx.move_to(GAMES_POS[0]+xoffset-xoff-tw/2, GAMES_POS[1]+28-yoff-4)
349 ctx.rotate(math.radians(-10))
353 ctx.select_font_face(font, C.FONT_SLANT_NORMAL, C.FONT_WEIGHT_NORMAL)
354 ctx.set_font_size(10)
355 ctx.set_source_rgb(1.0, 1.0, 0.5)
356 txt = "Elo: %.0f" % round(elos[gt], 0)
357 xoff, yoff, tw, th = ctx.text_extents(txt)[:4]
358 ctx.move_to(GAMES_POS[0]+xoffset-xoff-tw/2, GAMES_POS[1]+15-yoff-4)
361 ctx.select_font_face(font, C.FONT_SLANT_NORMAL, C.FONT_WEIGHT_NORMAL)
363 ctx.set_source_rgb(0.8, 0.8, 1.0)
364 txt = "Rank %d of %d" % ranks[gt]
365 xoff, yoff, tw, th = ctx.text_extents(txt)[:4]
366 ctx.move_to(GAMES_POS[0]+xoffset-xoff-tw/2, GAMES_POS[1]+25-yoff-3)
369 xoffset += GAMES_WIDTH
372 # print win percentage
374 if params['overlay'] == 0:
375 ctx.rectangle(WINLOSS_POS[0]-WINLOSS_WIDTH/2, WINLOSS_POS[1]-7, WINLOSS_WIDTH, 15)
376 ctx.set_source_rgba(0.8, 0.8, 0.8, 0.1)
379 ctx.select_font_face(font, C.FONT_SLANT_NORMAL, C.FONT_WEIGHT_NORMAL)
380 ctx.set_font_size(10)
381 ctx.set_source_rgb(0.8, 0.8, 0.8)
382 txt = "Win Percentage"
383 xoff, yoff, tw, th = ctx.text_extents(txt)[:4]
384 ctx.move_to(WINLOSS_POS[0]-xoff-tw/2, WINLOSS_POS[1]-yoff-3)
387 wins, losses = total_stats["wins"], total_games-total_stats["wins"]
390 ratio = float(wins)/total_games
391 txt = "%.2f%%" % round(ratio * 100, 2)
394 ctx.select_font_face(font, C.FONT_SLANT_NORMAL, C.FONT_WEIGHT_BOLD)
395 ctx.set_font_size(12)
397 ctx.set_source_rgb(0.2, 1.0, 1.0)
399 ctx.set_source_rgb(0.5, 1.0, 1.0)
401 ctx.set_source_rgb(0.5, 1.0, 0.8)
403 ctx.set_source_rgb(0.8, 1.0, 0.5)
405 ctx.set_source_rgb(1.0, 1.0, 0.5)
406 xoff, yoff, tw, th = ctx.text_extents(txt)[:4]
407 ctx.move_to(WINLOSS_POS[0]-xoff-tw/2, WINLOSS_POS[1]+18-yoff-4)
410 ctx.select_font_face(font, C.FONT_SLANT_NORMAL, C.FONT_WEIGHT_NORMAL)
412 ctx.set_source_rgb(0.6, 0.8, 0.8)
413 txt = "%d win" % wins
416 xoff, yoff, tw, th = ctx.text_extents(txt)[:4]
417 ctx.move_to(WINLOSS_POS[0]-xoff-tw/2, WINLOSS_POS[1]+30-yoff-3)
419 ctx.set_source_rgb(0.8, 0.8, 0.6)
420 txt = "%d loss" % losses
423 xoff, yoff, tw, th = ctx.text_extents(txt)[:4]
424 ctx.move_to(WINLOSS_POS[0]-xoff-tw/2, WINLOSS_POS[1]+40-yoff-3)
428 # print kill/death ratio
430 if params['overlay'] == 0:
431 ctx.rectangle(KILLDEATH_POS[0]-KILLDEATH_WIDTH/2, KILLDEATH_POS[1]-7, KILLDEATH_WIDTH, 15)
432 ctx.set_source_rgba(0.8, 0.8, 0.8, 0.1)
435 ctx.select_font_face(font, C.FONT_SLANT_NORMAL, C.FONT_WEIGHT_NORMAL)
436 ctx.set_font_size(10)
437 ctx.set_source_rgb(0.8, 0.8, 0.8)
439 xoff, yoff, tw, th = ctx.text_extents(txt)[:4]
440 ctx.move_to(KILLDEATH_POS[0]-xoff-tw/2, KILLDEATH_POS[1]-yoff-3)
443 kills, deaths = total_stats['kills'] , total_stats['deaths']
446 ratio = float(kills)/deaths
447 txt = "%.3f" % round(ratio, 3)
451 ctx.select_font_face(font, C.FONT_SLANT_NORMAL, C.FONT_WEIGHT_BOLD)
452 ctx.set_font_size(12)
454 ctx.set_source_rgb(0.0, 1.0, 0.0)
456 ctx.set_source_rgb(0.2, 1.0, 0.2)
458 ctx.set_source_rgb(0.5, 1.0, 0.5)
460 ctx.set_source_rgb(1.0, 0.5, 0.5)
462 ctx.set_source_rgb(1.0, 0.2, 0.2)
463 xoff, yoff, tw, th = ctx.text_extents(txt)[:4]
464 ctx.move_to(KILLDEATH_POS[0]-xoff-tw/2, KILLDEATH_POS[1]+18-yoff-4)
467 ctx.select_font_face(font, C.FONT_SLANT_NORMAL, C.FONT_WEIGHT_NORMAL)
469 ctx.set_source_rgb(0.6, 0.8, 0.6)
470 txt = "%d kill" % kills
473 xoff, yoff, tw, th = ctx.text_extents(txt)[:4]
474 ctx.move_to(KILLDEATH_POS[0]-xoff-tw/2, KILLDEATH_POS[1]+30-yoff-3)
476 ctx.set_source_rgb(0.8, 0.6, 0.6)
477 txt = "%d death" % deaths
480 xoff, yoff, tw, th = ctx.text_extents(txt)[:4]
481 ctx.move_to(KILLDEATH_POS[0]-xoff-tw/2, KILLDEATH_POS[1]+40-yoff-3)
487 if params['overlay'] == 0:
488 ctx.rectangle( PLAYTIME_POS[0]-PLAYTIME_WIDTH/2, PLAYTIME_POS[1]-7, PLAYTIME_WIDTH, 14)
489 ctx.set_source_rgba(0.8, 0.8, 0.8, 0.6)
492 ctx.select_font_face(font, C.FONT_SLANT_NORMAL, C.FONT_WEIGHT_NORMAL)
493 ctx.set_font_size(10)
494 ctx.set_source_rgb(0.1, 0.1, 0.1)
495 txt = "Playing time: %s" % str(total_stats['alivetime'])
496 xoff, yoff, tw, th = ctx.text_extents(txt)[:4]
497 ctx.move_to(PLAYTIME_POS[0]-xoff-tw/2, PLAYTIME_POS[1]-yoff-4)
502 surf.write_to_png(OUTPUT % data['player'].player_id)
506 env = bootstrap('../../../development.ini')
508 req.matchdict = {'id':3}
510 print "Requesting player data from db ..."
511 start = datetime.now()
512 players = DBSession.query(Player).\
513 filter(Player.player_id == PlayerElo.player_id).\
514 filter(Player.nick != None).\
515 filter(Player.player_id > 2).\
516 filter(Player.active_ind == True).\
519 stop = datetime.now()
521 total_seconds = (td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / 10**6
522 print "Query took %.2f seconds" % (total_seconds)
524 print "Creating badges for %d players ..." % len(players)
525 start = datetime.now()
526 data_time, render_time = 0,0
527 for player in players:
528 req.matchdict['id'] = player.player_id
530 sstart = datetime.now()
531 data = get_data(player)
532 sstop = datetime.now()
534 total_seconds = (td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / 10**6
535 data_time += total_seconds
537 sstart = datetime.now()
539 sstop = datetime.now()
541 total_seconds = (td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / 10**6
542 render_time += total_seconds
544 stop = datetime.now()
546 total_seconds = (td.microseconds + (td.seconds + td.days * 24 * 3600) * 10**6) / 10**6
547 print "Creating the badges took %.2f seconds (%.2f s per player)" % (total_seconds, total_seconds/float(len(players)))
548 print "Total time for redering images: %.2f s" % render_time
549 print "Total time for getting data: %.2f s" % data_time