3 import string, time, os
4 from subprocess import Popen, PIPE
5 from random import random
6 from MessageKeeper import MessageKeeper
13 def __init__(self, v):
17 """Displays next stage of the idler. Returns time to the next step"""
21 """Resets the idler to a known intial state"""
25 """Returns True if the idler is considered finished"""
29 """How much we want this idler to be the next one chosen"""
32 class GreetingIdler(Idler):
33 def __init__(self, v, secs_to_greeting = None):
35 self.secs_to_greeting = secs_to_greeting
36 self.message_displayed = False
39 if not self.secs_to_greeting is None:
40 x = self.secs_to_greeting
41 self.secs_to_greeting = None
44 self.v.display('UCC SNACKS')
45 self.message_displayed = True
49 self.message_displayed = False
50 self.secs_to_greeting = None
53 return self.message_displayed
58 class TrainIdler(Idler):
59 def __init__(self, v):
63 def put_shark(self, s, l):
66 elif self.s[l] == 'X':
72 # does the next stage of a dance
74 shark1 = self.idle_state % 18
76 self.put_shark('^', shark1)
78 self.put_shark('^', 18-shark1)
80 shark2 = ((self.idle_state+4) % 36)/2
82 self.put_shark('<', shark2)
84 self.put_shark('<', 18-shark2)
86 shark3 = ((self.idle_state+7) % 54)/3
88 self.put_shark('>', 9-shark3)
90 self.put_shark('>', 9-(18-shark3))
92 train1 = ((self.idle_state%(18*36)))
94 if train1 > train1_start and train1 < train1_start+(10*2):
96 ptr = i+train1-train1_start-5
97 if ptr >= 0 and ptr < 10: self.s[ptr] = '#'
99 train2 = ((self.idle_state%(18*36)))
101 if train2 > train2_start and train2 < train2_start+(10*2):
103 ptr = i+train2-train2_start-5
104 if ptr >= 0 and ptr < 10: self.s[9-ptr] = '#'
106 train3 = ((self.idle_state%(18*36)))
108 if train3 > train3_start and train3 < train3_start+(10*2):
110 ptr = i+train3-train3_start-10
111 if ptr >= 0 and ptr < 10: self.s[ptr] = '-'
113 self.v.display(string.join(self.s, ''))
115 self.idle_state %= 18*36*54
121 def __init__(self, n=8):
123 self.make_factorials(n)
125 def make_factorials(self, n):
128 for i in range(1,n+1):
129 self.factorial.append(a)
132 def order(self, index):
134 for i in range(0,self.n):
140 a = index/self.factorial[i]
141 index %= self.factorial[i]
145 for i in range(0,self.n):
147 for j in range(res[i],self.n):
152 def __getitem__(self, i):
155 class GrayIdler(Idler):
156 def __init__(self, v, one=None, zero=None, reorder=0):
158 self.size = 1 << self.bits
164 self.reorder = reorder
167 orderings = OrderMaker()
170 output = self.do_next_state()
171 # does the next stage of a dance
173 output = string.replace(output, "0", self.zero)
175 output = string.replace(output, "1", self.one)
180 newoutput += output[orderings[self.reorder][i]]
182 self.v.display(" %8.8s " % (output))
183 self.i = (self.i + 1) % self.size
185 def do_next_state(self):
186 self.grayCode = self.i ^ (self.i >> 1)
187 output = self.dec2bin(self.grayCode)
189 return "0"*(self.bits-len(output))+output
192 def dec2bin(self,num):
193 """Convert long/integer number to binary string.
195 E.g. dec2bin(12) ==> '1100'.
197 from http://starship.python.net/~gherman/playground/decbingray/decbingray.py"""
199 assert num >= 0, "Decimal number must be >= 0!"
201 # Gracefully handle degenerate case.
202 # (Not really needed, but anyway.)
206 # Find highest value bit.
209 val, j = val*2L, j+1L
230 self.reorder = int(random()*40319)+1
233 class StringIdler(Idler):
234 def __init__(self, v, text="Hello Cruel World! ",repeat=True):
236 self.mk = MessageKeeper(v)
237 self.text = self.clean_text(text) + " "
239 msg = [("",False, None),(self.text, repeat, IDLER_TEXT_SPEED)]
240 self.mk.set_messages(msg)
242 def clean_text(self, text):
243 # nothing like a bit of good clean text :)
244 valid = string.digits \
246 + string.punctuation \
249 text = string.upper(text)
259 self.mk.update_display()
262 return self.mk.done()
264 class ClockIdler(Idler):
265 def __init__(self, v):
271 if int(time.time()*2) & 1: colonchar = ' '
272 output = time.strftime("%%H%c%%M%c%%S"%(colonchar,colonchar))
273 if output != self.last:
274 self.v.display(" %8.8s " % (output))
280 class FortuneIdler(StringIdler):
281 def __init__(self, v):
282 fortune = "/usr/games/fortune"
283 text = "I broke my wookie...."
284 if os.access(fortune,os.F_OK|os.X_OK):
285 (lines, unused) = Popen((fortune,), close_fds=True, stdout=PIPE).communicate()
286 text = string.join(lines)
287 StringIdler.__init__(self, v, text,repeat=False)
292 class PipeIdler(StringIdler):
293 def __init__(self, v, command, args):
294 text = "I ate my cookie...."
295 if os.access(command,os.F_OK|os.X_OK):
296 (lines, unused) = Popen([command,] + args.split(), close_fds=True, stdout=PIPE).communicate()
297 text = string.join(lines)
298 StringIdler.__init__(self, v, text,repeat=False)
303 class FileIdler(StringIdler):
304 def __init__(self, v, thefile=None, repeat=False, affinity=8):
305 text = "I broke my wookie...."
306 self._affinity = affinity
308 if file and os.access(thefile,os.F_OK|os.R_OK):
309 f = file(thefile,'r')
310 text = string.join(f.readlines())
312 StringIdler.__init__(self, v, text,repeat=repeat)
315 return self._affinity