diff options
Diffstat (limited to 'Spirolaterals.py')
| -rwxr-xr-x | Spirolaterals.py | 345 |
1 files changed, 345 insertions, 0 deletions
diff --git a/Spirolaterals.py b/Spirolaterals.py new file mode 100755 index 0000000..54020b9 --- /dev/null +++ b/Spirolaterals.py @@ -0,0 +1,345 @@ +#!/usr/bin/env python
+# Spirolaterals.py
+"""
+ Copyright (C) 2010 Peter Hewitt
+
+ This program is free software: you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation, either version 3 of the License, or
+ (at your option) any later version.
+
+"""
+import g,utils,pygame,buttons,my_turtle,random,os,sys,load_save,slider
+try:
+ import gtk
+except:
+ pass
+
+class Spirolaterals:
+
+ def __init__(self):
+ self.journal=True # set to False if we come in via main()
+ self.canvas=None # set to the pygame canvas if we come in via activity.py
+
+ def display(self): # called each loop
+ if g.big:
+ g.screen.fill((0,0,0))
+ utils.centre_blit(g.screen,g.big_surface,(g.sx(16),g.sy(11.5)))
+ else:
+ g.screen.blit(g.bgd,(g.sx(0),0))
+ g.screen.blit(g.box,(g.x0,g.y0))
+ g.screen.blit(g.box,(g.x1,g.y0))
+ utils.centre_blit(g.screen,g.magician,g.magician_c)
+ self.draw_goal()
+ utils.centre_blit(g.screen,g.turtle,(g.x0+4*g.dd,g.y0+6*g.dd))
+ self.tu.draw()
+ if self.tu.win:
+ utils.centre_blit(g.screen,g.smiley,(g.sx(16.6),g.sy(2.2)))
+ buttons.on('cyan')
+ self.draw_nos()
+ buttons.draw()
+ self.slider.draw()
+ if g.score>0: utils.display_score()
+ utils.display_number1(g.pattern,(g.sx(2.4),g.sy(2)),\
+ g.font1,utils.BLUE)
+
+ def do_button(self,bu):
+ if bu=='cyan':
+ g.pattern+=1
+ if g.pattern==123: g.pattern=1
+ g.help1=0; g.help2=0; self.get_goal()
+ self.tu.win=False; g.finale=False; g.show_help=False
+ self.tu.changed=True
+ buttons.off('cyan')
+ self.mouse_1st_no() # to 1st number
+ elif bu=='black':
+ self.tu.current=utils.copy_list(g.numbers)
+ self.tu.setup(); g.show_help=False
+ elif bu=='green':
+ g.show_help=False
+ if self.tu.changed:
+ self.tu.current=utils.copy_list(g.numbers)
+ self.tu.setup(); self.tu.changed=False
+ self.tu.running=True
+ elif self.tu.win or self.tu.crashed:
+ self.tu.setup(); self.tu.running=True
+ else:
+ if self.tu.step_count==0: self.tu.ms=pygame.time.get_ticks()
+ self.tu.running=True
+ elif bu=='red':
+ self.tu.running=False
+
+ def do_key(self,key):
+ if key in g.CROSS:
+ if utils.mouse_on_img1(g.magician,g.magician_c):
+ self.help2(); return
+ bu=buttons.check()
+ if bu!='': self.do_button(bu); return
+ g.show_help=False; self.check_nos(1); return
+ if key in g.CIRCLE: self.check_nos(3); return
+ if key in g.RIGHT: self.mouse_right(); return
+ if key in g.LEFT: self.mouse_left(); return
+ if key in g.SQUARE:
+ if buttons.active('cyan'): self.do_button('cyan')
+ return
+ if key in g.TICK: self.change_level(); return
+ if key==pygame.K_v: g.version_display=not g.version_display; return
+
+ def mouse_1st_no(self):
+ c=g.n_cx0+g.sy(.2),g.n_cy0+g.sy(1.2); pygame.mouse.set_pos(c); g.pos=c
+
+ def mouse_magician(self):
+ x,y=g.magician_c; x-=g.sy(.15); y-=g.sy(.52); c=x,y
+ pygame.mouse.set_pos(c); g.pos=c
+
+ def mouse_left(self):
+ bu=''; cx=g.n_cx0; cy=g.n_cy0; c=None
+ if utils.mouse_on_img1(g.magician,g.magician_c):
+ c=(cx+4*g.n_dx,cy)
+ elif buttons.mouse_on('cyan'): self.mouse_magician(); return
+ elif buttons.mouse_on('green'):
+ if buttons.active('cyan'): bu='cyan'
+ else: self.mouse_magician(); return
+ elif buttons.mouse_on('red'): bu='green'
+ elif buttons.mouse_on('black'): bu='red'
+ if bu!='': buttons.set_mouse(bu); return
+ if c==None:
+ c=(cx,cy) # default to 1st no.
+ for i in range(5):
+ n=g.numbers[i]
+ if utils.mouse_on_img_rect(g.n[n-1],(cx,cy)):
+ if i==0: buttons.set_mouse('black'); return
+ c=(cx-g.n_dx,cy); break
+ cx+=g.n_dx
+ cx,cy=c; cx+=g.sy(.2); cy+=g.sy(1.2); c=cx,cy
+ pygame.mouse.set_pos(c); g.pos=c; return
+
+ def mouse_right(self):
+ bu=''
+ if utils.mouse_on_img1(g.magician,g.magician_c):
+ bu='green'
+ if buttons.active('cyan'): bu='cyan'
+ elif buttons.mouse_on('cyan'): bu='green'
+ elif buttons.mouse_on('green'): bu='red'
+ elif buttons.mouse_on('red'): bu='black'
+ if bu!='': buttons.set_mouse(bu); return
+ cx=g.n_cx0; cy=g.n_cy0; c=(cx,cy) # default to 1st no.
+ if buttons.mouse_on('black'): pass
+ else:
+ for i in range(5):
+ n=g.numbers[i]
+ if utils.mouse_on_img_rect(g.n[n-1],(cx,cy)):
+ if i==4: self.mouse_magician(); return
+ c=(cx+g.n_dx,cy); break
+ cx+=g.n_dx
+ cx,cy=c; cx+=g.sy(.2); cy+=g.sy(1.2); c=cx,cy
+ pygame.mouse.set_pos(c); g.pos=c; return
+
+ def change_level(self):
+ g.level+=1
+ if g.level>self.slider.steps: g.level=1
+ g.delay=(3-g.level)*400
+
+ def draw_goal(self): # draws the left hand pattern
+ x1=g.x0+4*g.dd; y1=g.y0+6*g.dd; dx=0; dy=-g.dd
+ for i in range(4):
+ for j in g.goal:
+ for k in range(j):
+ x2=x1+dx; y2=y1+dy
+ pygame.draw.line(g.screen,utils.CYAN,(x1,y1),(x2,y2),4)
+ x1=x2; y1=y2
+ if dy==-g.dd: dx=g.dd;dy=0
+ elif dx==g.dd: dx=0;dy=g.dd
+ elif dy==g.dd: dx=-g.dd;dy=0
+ else: dx=0;dy=-g.dd
+
+ def calc_steps(self,l): # calculates total # of steps for a given pattern
+ # eg [1,2,3,4,5] = (1+2+3+4+5)*4=60
+ return sum(l)*4
+
+ def get_goal(self):
+ fname=os.path.join('data','patterns.dat')
+ try:
+ f=open(fname, 'r')
+ for n in range(0,g.pattern): s=f.readline()
+ s=s[0:5]
+ except:
+ s=11132; g.pattern=1
+ f.close
+ l=[int(c) for c in str(s)]
+ g.goal=l; g.steps=self.calc_steps(l)
+
+ def draw_nos(self): # draw the numbers with glow in correct position
+ pos=self.calc_pos(self.tu.step_count)
+ x=g.n_cx0
+ for i in range(5):
+ if i==pos: x_glow=x
+ n=g.numbers[i];utils.centre_blit(g.screen,g.n[n-1],(x,g.n_cy0))
+ x+=g.n_dx
+ if self.tu.changed==False or g.show_help:
+ if self.tu.step_count<self.tu.steps: # no glow if finished
+ n=self.tu.current[pos]
+ utils.centre_blit(g.screen,g.n_glow[n-1],(x_glow,g.n_cy0))
+
+ def check_nos(self,mouse_button):
+ w=g.n[3].get_width(); h=g.n[3].get_height() # "4" is widest
+ x1=g.n_cx0-w/2; y1=g.n_cy0-h/2
+ x2=g.n_cx0+w/2; y2=g.n_cy0+h/2
+ for pos in range(5):
+ if utils.mouse_in(x1,y1,x2,y2):
+ self.tu.changed=True; self.tu.running=False
+ if mouse_button==1:
+ self.inc_numbers(pos)
+ elif mouse_button==3:
+ self.dec_numbers(pos)
+ return True
+ x1+=g.n_dx; x2+=g.n_dx
+ return False
+
+ def calc_pos(self,step_count): # calculate which number we are currently on
+ steps=1
+ if self.tu.crashed:step_count-=1
+ for i in range(4):
+ pos=0
+ for j in self.tu.current:
+ for k in range(j):
+ if steps>=step_count: return pos #****
+ steps+=1
+ pos+=1
+
+ def inc_numbers(self,pos): # pos 0 to 4 - called with numberclicked
+ v=g.numbers[pos]+1
+ if v==6: v=1
+ g.numbers[pos]=v
+
+ def dec_numbers(self,pos): # pos 0 to 4 - called with numberclicked
+ v=g.numbers[pos]-1
+ if v==0: v=5
+ g.numbers[pos]=v
+
+ def solution(self):
+ s=''
+ for i in range(5):s+=str(g.goal[i])+' '
+ s=s[:9]
+ return s
+
+ def big_pic(self):
+ if not self.tu.running:
+ d=g.sy(1); s=g.bw-2*d
+ self.tu.draw()
+ g.player_surface.blit(g.screen,(0,0),(g.x1+d,g.y0+d,s,s))
+ g.big=True
+ g.big_surface=pygame.transform.scale2x(g.player_surface)
+
+ def help2(self):
+ self.tu.current=utils.copy_list(g.numbers)
+ self.tu.crashed=False
+ g.help1=0
+ looking=True
+ while looking:
+ g.help1+=1;ind=g.help1-1
+ if g.numbers[ind]<>g.goal[ind]:
+ g.numbers[ind]=g.goal[ind]; self.tu.current[ind]=g.goal[ind]
+ g.show_help=True; self.tu.changed=True; g.help2+=1
+ looking=False
+ if g.help1>4: g.show_help=True; looking=False
+
+ def flush_queue(self):
+ flushing=True
+ while flushing:
+ flushing=False
+ if self.journal:
+ while gtk.events_pending(): gtk.main_iteration()
+ for event in pygame.event.get(): flushing=True
+
+ def run(self):
+ g.init()
+ if not self.journal: utils.load()
+ load_save.retrieve()
+ g.delay=(3-g.level)*400
+ self.tu=my_turtle.TurtleClass()
+ self.tu.current=[1,1,1,3,2]
+ self.get_goal()
+ if g.pattern==1: self.tu.current=utils.copy_list(g.goal)
+ self.tu.setup()
+ g.numbers=utils.copy_list(self.tu.current)
+ #buttons
+ x=g.sx(7.3); y=g.sy(16.5); dx=g.sy(2.6);
+ buttons.Button("cyan",(x,y),True); x+=dx
+ buttons.off('cyan')
+ buttons.Button("green",(x,y),True); x+=dx
+ buttons.Button("red",(x,y),True); x+=dx
+ buttons.Button("black",(x,y),True); x+=dx
+ self.slider=slider.Slider(g.sx(23.5),g.sy(21),3,utils.YELLOW)
+ self.mouse_1st_no() # to 1st number
+ if self.canvas<>None: self.canvas.grab_focus()
+ ctrl=False
+ pygame.key.set_repeat(600,120); key_ms=pygame.time.get_ticks()
+ going=True
+ while going:
+ if self.journal:
+ # Pump GTK messages.
+ while gtk.events_pending(): gtk.main_iteration()
+
+ # Pump PyGame messages.
+ for event in pygame.event.get():
+ if event.type==pygame.QUIT:
+ if not self.journal: utils.save()
+ going=False
+ elif event.type == pygame.MOUSEMOTION:
+ g.pos=event.pos
+ g.redraw=True
+ if self.canvas<>None: self.canvas.grab_focus()
+ elif event.type == pygame.MOUSEBUTTONDOWN:
+ g.redraw=True
+ if g.big:
+ g.big=False
+ else:
+ bu=buttons.check()
+ if bu<>'':
+ self.do_button(bu); self.flush_queue()
+ elif utils.mouse_on_img1(g.magician,g.magician_c):
+ self.help2()
+ elif utils.mouse_in(g.x1,g.y0,g.x1+g.bw,g.y0+g.bw):
+ self.big_pic()
+ elif self.slider.mouse():
+ g.delay=(3-g.level)*400
+ else:
+ g.show_help=False
+ self.check_nos(event.button)
+ elif event.type == pygame.KEYDOWN:
+ # throttle keyboard repeat
+ if pygame.time.get_ticks()-key_ms>110:
+ key_ms=pygame.time.get_ticks()
+ if ctrl:
+ if event.key==pygame.K_q:
+ if not self.journal: utils.save()
+ going=False; break
+ else:
+ ctrl=False
+ if event.key in (pygame.K_LCTRL,pygame.K_RCTRL):
+ ctrl=True; break
+ self.do_key(event.key); g.redraw=True
+ self.flush_queue()
+ elif event.type == pygame.KEYUP:
+ ctrl=False
+ if not going: break
+ if self.tu.running: self.tu.move()
+ if not g.crash_drawn: g.crash_drawn=True; g.redraw=True
+ if g.redraw:
+ self.display()
+ if g.version_display: utils.version_display()
+ g.screen.blit(g.pointer,g.pos)
+ pygame.display.flip()
+ g.redraw=False
+ g.clock.tick(40)
+
+if __name__=="__main__":
+ pygame.init()
+ pygame.display.set_mode((1024,768),pygame.FULLSCREEN)
+ game=Spirolaterals()
+ game.journal=False
+ game.run()
+ pygame.display.quit()
+ pygame.quit()
+ sys.exit(0)
|