Wiimote-Cursor/CursorFunctions.py at main · ced4rtree/Wiimote-Cursor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
# Coded by SpicyMileHigh, script based off of that of The Raspberry Pi Guy's.
import cwiid, time, turtle, matplotlib
matplotlib.use('Agg')
button_delay = 0.1
color = 1
time.sleep(1)
bClearing = True
# This code attempts to connect to your Wiimote and if it fails the program quits
try:
  print("Please Press the 1 and 2 Buttons on Your Wiimote to Connect")
  wii=cwiid.Wiimote()
except:
  print("Connection failed. Please try again.")
  quit()
circle_width = 5
print("Connection established. Feel free to draw.")
#I honestly have no idea what this does and am too scared to remove it. Came from The Raspberry Pi Guys code.
wii.rpt_mode = cwiid.RPT_BTN

#All turtle preamble
bg = turtle.Screen()
bg.bgcolor("black")
bg.title("Good Morning!")
turtle.speed(0)
turtle.begin_fill()
turtle.goto(200,200)
turtle.hideturtle()
turtle.pendown()
turtle.begin_fill()
turtle.pencolor('#FFFFFF')
turtle.fillcolor('#FFFFFF')
turtle.circle(circle_width)

#Cursor Movement function
def movement(button):
  turtle.pendown()
  turtle.begin_fill()
  turtle.fillcolor('#FFFFFF')
  turtle.pencolor('#FFFFFF')
  turtle.circle(circle_width)
  turtle.end_fill()
  wii.rpt_mode = cwiid.RPT_BTN | cwiid.RPT_ACC
  roll=(wii.state['acc'][0]-120)
  pitch=(wii.state['acc'][1]-121)
  check = 0
  while check == 0:
    if (roll != 0):
      turtle.setheading(0)
      turtle.penup()
      turtle.forward(roll*2)
      turtle.pendown()
    if (pitch != 0):
      turtle.setheading(90)
      turtle.penup()
      turtle.backward(pitch*2)
      turtle.pendown()
    time.sleep(0.01)
    check = (buttons & button)
  time.sleep(button_delay)

#Stuff to do with buttons
while True:
  buttons = wii.state['buttons']

  if (buttons - cwiid.BTN_PLUS - cwiid.BTN_MINUS == 0):
    print '\nClosing connection ...'
    # NOTE: This is how you RUMBLE the Wiimote
    wii.rumble = 1
    time.sleep(1)
    wii.rumble = 0
    exit(wii)
  if (buttons & cwiid.BTN_MINUS):
    bClearing = False
  if (buttons & cwiid.BTN_PLUS):
    bClearing = True
#Draws
  if (buttons & cwiid.BTN_A):
    movement(cwiid.BTN_A)
#Moves the cursor
  if (buttons & cwiid.BTN_B):
    if (bClearing == True):
      turtle.clear()
      movement(cwiid.BTN_B)
    elif (bClearing == False):
      movement(cwiid.BTN_B)
#sets the cursor to center of screen
  if (buttons & cwiid.BTN_HOME):
    turtle.penup()
    turtle.goto(200,200)
    turtle.pendown()
    turtle.begin_fill()
    turtle.pencolor('#FFFFFF')
    turtle.fillcolor('#FFFFFF')
    turtle.circle(circle_width)
    turtle.end_fill()
    turtle.clear()
    time.sleep(button_delay)

turtle.endfill()