Permalink
Browse files

add scalesgui

  • Loading branch information...
1 parent 52b6a3d commit 27bb181c0330d1ec4f5e0445e5b23a9737460bbc @derf committed Aug 30, 2012
Showing with 209 additions and 0 deletions.
  1. +209 −0 scalesgui.py
View
@@ -0,0 +1,209 @@
+#!/usr/bin/python
+"""scalesgui.py
+
+<http://abstrakraft.org/cwiid/attachment/ticket/63/scalesgui.py>
+
+"""
+
+import sys
+
+try:
+ import pygame
+except:
+ print "Sorry, I can't seem to import pygame for some reason."
+ print "Please check that the python-pygame package is installed, or get the latest version of pygame from http://www.pygame.org/"
+ sys.exit(1)
+
+try:
+ import cwiid
+except:
+ print "Sorry, I can't seem to import cwiid for some reason."
+ print "Please check that it and it's python bindings are installed, and also the balance board patch from:"
+ print "http://abstrakraft.org/cwiid/ticket/63"
+ sys.exit(1)
+
+import os, math, random
+import time as ptime
+from pygame.locals import *
+from ConfigParser import ConfigParser
+from threading import Thread
+
+class WeightSprite(pygame.sprite.Sprite):
+ """This class describes a sprite containing the weight."""
+ def __init__(self):
+ pygame.sprite.Sprite.__init__(self)
+ self.weight = 0.0
+ self.update()
+
+ def update(self):
+ global screen_res, sys_font_weight_fgcolour, sys_font_weight, screen_res
+
+ if self.weight > 2:
+ self.text = "%.2f" % self.weight
+ else:
+ self.text = "_.__"
+ #print "LESS THAN 2"
+ #while len(self.text) < 4:
+ # self.text = "0" + self.text
+
+ self.image = sys_font_weight.render(self.text, True, sys_font_weight_fgcolour)
+
+ self.rect = self.image.get_rect()
+ self.rect.bottomright = screen_res
+
+def quit_app():
+ pygame.quit()
+ sys.exit(0)
+
+def calcweight( readings, calibrations ):
+ """
+ Determine the weight of the user on the board in hundredths of a kilogram
+ """
+ weight = 0
+ for sensor in ('right_top', 'right_bottom', 'left_top', 'left_bottom'):
+ reading = readings[sensor]
+ calibration = calibrations[sensor]
+ #if reading < calibration[0]:
+ # print "Warning, %s reading below lower calibration value" % sensor
+ if reading > calibration[2]:
+ print "Warning, %s reading above upper calibration value" % sensor
+ # 1700 appears to be the step the calibrations are against.
+ # 17kg per sensor is 68kg, 1/2 of the advertised Japanese weight limit.
+ if reading < calibration[1]:
+ weight += 1700 * (reading - calibration[0]) / (calibration[1] - calibration[0])
+ else:
+ weight += 1700 * (reading - calibration[1]) / (calibration[2] - calibration[1]) + 1700
+
+ return weight
+
+def gsc(readings, pos):
+ global named_calibration
+ reading = readings[pos]
+ calibration = named_calibration[pos]
+
+ if reading < calibration[1]:
+ return 1700 * (reading - calibration[0]) / (calibration[1] - calibration[0])
+ else:
+ return 1700 * (reading - calibration[1]) / (calibration[2] - calibration[1]) + 1700
+
+
+print "Please press the red 'connect' button on the balance board, inside the battery compartment."
+print "Do not step on the balance board."
+
+global wiimote
+if len(sys.argv) > 1:
+ wiimote = cwiid.Wiimote(sys.argv[1])
+else:
+ wiimote = cwiid.Wiimote()
+
+wiimote.rpt_mode = cwiid.RPT_BALANCE | cwiid.RPT_BTN
+wiimote.request_status()
+
+balance_calibration = wiimote.get_balance_cal()
+named_calibration = { 'right_top': balance_calibration[0],
+ 'right_bottom': balance_calibration[1],
+ 'left_top': balance_calibration[2],
+ 'left_bottom': balance_calibration[3],
+ }
+
+system_file = "system.ini"
+
+if not os.path.lexists(system_file):
+ print "Problem: System configuration file (system.ini) doesn't exist."
+ sys.exit(1)
+
+sconf = ConfigParser()
+sconf.read(system_file)
+
+
+xdisplay = sconf.get("display", "xdisplay")
+if len(xdisplay) > 1:
+ # using alternate display.
+ print "Attempting to use device", xdisplay, "instead of the default."
+ os.putenv("DISPLAY", xdisplay)
+
+pygame.init()
+
+sys_font_weight = pygame.font.SysFont(sconf.get("font_weight", "face"), int(sconf.get("font_weight", "size")))
+
+sys_font_weight.set_italic(False)
+sys_font_weight.set_underline(False)
+
+bgcolour = (0, 0, 0)
+sys_font_weight_fgcolour = (255, 255, 255)
+screen_res = (int(sconf.get("display", "width")), int(sconf.get("display", "height")))
+refresh_delay = int(sconf.get("display", "refresh_delay"))
+
+screen_options = 0
+if int(sconf.get("display", "fullscreen")) >= 1 and len(xdisplay) <= 1:
+ screen_options = screen_options | pygame.FULLSCREEN
+
+if int(sconf.get("display", "double_buffers")) >= 1:
+ screen_options = screen_options | pygame.DOUBLEBUF
+
+if int(sconf.get("display", "hardware_surface")) >= 1:
+ screen_options = screen_options | pygame.HWSURFACE
+
+if int(sconf.get("display", "opengl")) >= 1:
+ screen_options = screen_options | pygame.OPENGL
+
+screen = pygame.display.set_mode(screen_res, screen_options)
+pygame.display.set_caption("scales application")
+
+weight_sprite = WeightSprite()
+weight_sprite.weight = 40.33
+frame = 0
+while True:
+ for event in pygame.event.get():
+ if event.type == KEYDOWN:
+ if event.key == K_F12:
+ quit_app()
+
+ wiimote.request_status()
+ frame = frame + 1
+ if frame == 50:
+ frame = 0
+ weight = (calcweight(wiimote.state['balance'], named_calibration) / 100.0)
+ #print "%.2fkg" % weight
+ weight_sprite.weight = weight
+
+
+ readings = wiimote.state['balance']
+
+ try:
+ x_balance = (float(gsc(readings,'right_top')+gsc(readings,'right_bottom'))) / (float(gsc(readings,'left_top')+gsc(readings,'left_bottom')))
+ if x_balance > 1:
+ x_balance = (((float(gsc(readings,'left_top')+gsc(readings,'left_bottom'))) / (float(gsc(readings,'right_top')+gsc(readings,'right_bottom'))))*-1.)+1.
+ else:
+ x_balance = x_balance -1.
+ y_balance = (float(gsc(readings,'left_bottom')+gsc(readings,'right_bottom'))) / (float(gsc(readings,'left_top')+gsc(readings,'right_top')))
+ if y_balance > 1:
+ y_balance = (((float(gsc(readings,'left_top')+gsc(readings,'right_top'))) / (float(gsc(readings,'left_bottom')+gsc(readings,'right_bottom'))))*-1.)+1.
+ else:
+ y_balance = y_balance -1.
+ except:
+ x_balance = 1.
+ y_balance = 1.
+
+ #print "readings:",readings
+
+ screen.fill(bgcolour) # blank the screen.
+
+ # line up the lines
+ pygame.draw.line(screen, (0,0,255), (screen_res[0]/2,0), (screen_res[0]/2,screen_res[1]), 2)
+ pygame.draw.line(screen, (0,0,255), (0,screen_res[1]/2), (screen_res[0],screen_res[1]/2), 2)
+
+ weight_sprite.update()
+
+ screen.blit(weight_sprite.image, weight_sprite.rect)
+
+ xpos = (x_balance * (screen_res[0]/2)) + (screen_res[0]/2)
+ ypos = (y_balance * (screen_res[1]/2)) + (screen_res[1]/2)
+
+ #print "balance:", x_balance, y_balance
+ #print "position:", xpos,ypos
+ pygame.draw.circle(screen, (255,0,0), (int(xpos), int(ypos)), 5)
+ pygame.display.flip()
+ pygame.time.wait(refresh_delay)
+
+

0 comments on commit 27bb181

Please sign in to comment.