#!/usr/bin/env python
-"""Go timer"""
+"""Go timer.
+
+This module defines a UI that combines game clock with automatic game recording.
+When player presses the clock, an image of the board is taken. The sequence of
+images can later be analysed by Imago to produce a game record in abstract
+notation.
+"""
import os
import sys
import pygame
class Timer:
+ """Keeps track of time remaining one player."""
def __init__(self, main_time, byop, byot):
self._last = 0
self._elapsed = 0.
self._byo = False
def get_time(self):
+ """Return string representation of remaining time."""
if self._last > 0: # when running:
r_time = self._main_t - (time.time() - self._last + self._elapsed)
if (r_time - int(r_time)) < 0.75:
return "{0:0>2}".format(int(r_time / 60)) + sep + "{0:0>2}".format(int(r_time % 60))
def start(self):
+ """Start the clock."""
self._last = time.time()
def stop(self):
+ """Stop the clock."""
self._elapsed += time.time() - self._last
self._last = 0
def is_running(self):
+ """Return True if the clock is running."""
if self._last > 0:
return True
else:
return False
-parser = argparse.ArgumentParser(description=__doc__)
-parser.add_argument('-m', type=int, default=10,
- help="main time in minutes (default is 10)")
-parser.add_argument('-b', type=int, nargs=2, default=[5, 20],
- help="japanese byoyomi: number of periods, period length in"
- " seconds (default is 5 periods, 20 seconds)")
-parser.add_argument('-c', '--camera', dest='cam', action='store_true',
- help="camera on")
-parser.add_argument('-d', type=int, default=0,
- help="video device id")
-parser.add_argument('-r', type=int, nargs=2, default=[640, 480],
- help="set camera resolution")
-args = parser.parse_args()
-
-
-pygame.init()
-pygame.display.set_mode((600, 130))
-pygame.display.set_caption("Go timer")
-screen = pygame.display.get_surface()
-
-clock = pygame.time.Clock()
-
-font = pygame.font.Font(pygame.font.match_font('monospace'), 80)
-font2 = pygame.font.Font(None, 25)
-
-done = False
-first = True
-
-main_time = args.m * 60
-
-timers = (Timer(main_time, args.b[0], args.b[1]), Timer(main_time, args.b[0], args.b[1]))
-
-if args.cam:
- capture = subprocess.Popen(['python', 'capture.py', '-c', '-d', str(args.d), '-r',
- str(args.r[0]),
- str(args.r[1])], stdin=subprocess.PIPE)
-last = 0
-
-while not done:
- for event in pygame.event.get():
- if event.type == pygame.QUIT:
- done = True
- elif event.type == pygame.KEYDOWN:
- if time.time() - last < 0.7:
- continue
- last = time.time()
- if first:
- timers[0].start()
- first = False
+def main():
+ """Parse the arguments and run the clock."""
+ # TODO refactor into smaller functions
+ parser = argparse.ArgumentParser(description=__doc__)
+ parser.add_argument('-m', type=int, default=10,
+ help="main time in minutes (default is 10)")
+ parser.add_argument('-b', type=int, nargs=2, default=[5, 20],
+ help="japanese byoyomi: number of periods, period length in"
+ " seconds (default is 5 periods, 20 seconds)")
+ parser.add_argument('-c', '--camera', dest='cam', action='store_true',
+ help="camera on")
+ parser.add_argument('-d', type=int, default=0,
+ help="video device id")
+ parser.add_argument('-r', type=int, nargs=2, default=[640, 480],
+ help="set camera resolution")
+ args = parser.parse_args()
+
+
+ pygame.init()
+ pygame.display.set_mode((600, 130))
+ pygame.display.set_caption("Go timer")
+ screen = pygame.display.get_surface()
+
+ clock = pygame.time.Clock()
+
+ font = pygame.font.Font(pygame.font.match_font('monospace'), 80)
+ font2 = pygame.font.Font(None, 25)
+
+ done = False
+ first = True
+
+ main_time = args.m * 60
+
+ timers = (Timer(main_time, args.b[0], args.b[1]), Timer(main_time, args.b[0], args.b[1]))
+
+ if args.cam:
+ capture = subprocess.Popen(['python', 'capture.py', '-c', '-d', str(args.d), '-r',
+ str(args.r[0]),
+ str(args.r[1])], stdin=subprocess.PIPE)
+ last = 0
+
+ while not done:
+ for event in pygame.event.get():
+ if event.type == pygame.QUIT:
+ done = True
+ elif event.type == pygame.KEYDOWN:
+ if time.time() - last < 0.7:
+ continue
+ last = time.time()
+ if first:
+ timers[0].start()
+ first = False
+ if args.cam:
+ print >> capture.stdin, "stop"
+ print >> capture.stdin, "take"
+ continue
if args.cam:
- print >> capture.stdin, "stop"
print >> capture.stdin, "take"
- continue
- if args.cam:
- print >> capture.stdin, "take"
- for timer in timers:
- if timer.is_running():
- timer.stop()
- else:
- timer.start()
-
- screen.fill([0, 0, 0])
- text1 = font.render(timers[0].get_time(), True, [128, 255, 128])
- screen.blit(text1, [10, 10])
- text2 = font.render(timers[1].get_time(), True, [128, 255, 128])
- screen.blit(text2, [300, 10])
- text3 = font2.render(timers[0].byost, True, [128, 255, 128])
- screen.blit(text3, [10, 90])
- text4 = font2.render(timers[1].byost, True, [128, 255, 128])
- screen.blit(text4, [300, 90])
- pygame.display.flip()
- clock.tick(15)
-
-if args.cam:
- print >> capture.stdin, "exit"
+ for timer in timers:
+ if timer.is_running():
+ timer.stop()
+ else:
+ timer.start()
+
+ screen.fill([0, 0, 0])
+ text1 = font.render(timers[0].get_time(), True, [128, 255, 128])
+ screen.blit(text1, [10, 10])
+ text2 = font.render(timers[1].get_time(), True, [128, 255, 128])
+ screen.blit(text2, [300, 10])
+ text3 = font2.render(timers[0].byost, True, [128, 255, 128])
+ screen.blit(text3, [10, 90])
+ text4 = font2.render(timers[1].byost, True, [128, 255, 128])
+ screen.blit(text4, [300, 90])
+ pygame.display.flip()
+ clock.tick(15)
+
+ if args.cam:
+ print >> capture.stdin, "exit"
+
+if __name__ == '__main__':
+ try:
+ main()
+ finally:
+ pygame.quit()
projekty / imago.git / blobdiff