src/extras/ball.py
changeset 0 ca70ae20a155
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/extras/ball.py	Tue Feb 16 10:07:05 2010 +0530
@@ -0,0 +1,121 @@
+# Copyright (c) 2005 Nokia Corporation
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import appuifw
+from graphics import *
+import e32
+from key_codes import *
+
+class Keyboard(object):
+    def __init__(self,onevent=lambda:None):
+        self._keyboard_state={}
+        self._downs={}
+        self._onevent=onevent
+    def handle_event(self,event):
+        if event['type'] == appuifw.EEventKeyDown:
+            code=event['scancode']
+            if not self.is_down(code):
+                self._downs[code]=self._downs.get(code,0)+1
+            self._keyboard_state[code]=1
+        elif event['type'] == appuifw.EEventKeyUp:
+            self._keyboard_state[event['scancode']]=0
+        self._onevent()
+    def is_down(self,scancode):
+        return self._keyboard_state.get(scancode,0)
+    def pressed(self,scancode):
+        if self._downs.get(scancode,0):
+            self._downs[scancode]-=1
+            return True
+        return False
+keyboard=Keyboard()
+
+appuifw.app.screen='full'
+img=None
+def handle_redraw(rect):
+    if img:
+        canvas.blit(img)
+appuifw.app.body=canvas=appuifw.Canvas(
+    event_callback=keyboard.handle_event,
+    redraw_callback=handle_redraw)
+img=Image.new(canvas.size)
+
+running=1
+def quit():
+    global running
+    running=0
+appuifw.app.exit_key_handler=quit
+
+location=[img.size[0]/2,img.size[1]/2]
+speed=[0.,0.]
+blobsize=16
+xs,ys=img.size[0]-blobsize,img.size[1]-blobsize
+gravity=0.03
+acceleration=0.05
+
+import time
+start_time=time.clock()
+n_frames=0
+# To speed things up, we prerender the text.
+labeltext=u'Use arrows to move ball'
+textrect=img.measure_text(labeltext, font='normal')[0]
+text_img=Image.new((textrect[2]-textrect[0],textrect[3]-textrect[1]))
+text_img.clear(0)
+text_img.text((-textrect[0],-textrect[1]),labeltext,fill=0xffffff,font='normal')
+
+while running:
+    img.clear(0)
+    img.blit(text_img, (0,0))
+    img.point((location[0]+blobsize/2,location[1]+blobsize/2),
+              0x00ff00,width=blobsize)
+    handle_redraw(())
+    e32.ao_yield()
+    speed[0]*=0.999
+    speed[1]*=0.999
+    speed[1]+=gravity
+    location[0]+=speed[0]
+    location[1]+=speed[1]
+    if location[0]>xs:
+        location[0]=xs-(location[0]-xs)
+        speed[0]=-0.80*speed[0]
+        speed[1]=0.90*speed[1]
+    if location[0]<0:
+        location[0]=-location[0]
+        speed[0]=-0.80*speed[0]
+        speed[1]=0.90*speed[1]
+    if location[1]>ys:
+        location[1]=ys-(location[1]-ys)
+        speed[0]=0.90*speed[0]
+        speed[1]=-0.80*speed[1]
+    if location[1]<0:
+        location[1]=-location[1]
+        speed[0]=0.90*speed[0]
+        speed[1]=-0.80*speed[1]
+        
+    if keyboard.is_down(EScancodeLeftArrow):  speed[0] -= acceleration
+    if keyboard.is_down(EScancodeRightArrow): speed[0] += acceleration
+    if keyboard.is_down(EScancodeDownArrow):  speed[1] += acceleration
+    if keyboard.is_down(EScancodeUpArrow):    speed[1] -= acceleration
+    if keyboard.pressed(EScancodeHash):
+        filename=u'e:\\screenshot.png'
+        canvas.text((0,32),u'Saving screenshot to:',fill=0xffff00)
+        canvas.text((0,48),filename,fill=0xffff00)
+        img.save(filename)
+
+    n_frames+=1
+end_time=time.clock()
+total=end_time-start_time
+
+print "%d frames, %f seconds, %f FPS, %f ms/frame."%(n_frames,total,
+                                                     n_frames/total,
+                                                     total/n_frames*1000.)