1 #!/usr/bin/env python

     2 ################################################################################

     3 #                                                                              #

     4 #   This program is free software: you can redistribute it and/or modify       #

     5 #   it under the terms of the GNU General Public License as published by       #

     6 #   the Free Software Foundation, either version 3 of the License, or          #

     7 #   (at your option) any later version.                                        #

     8 #                                                                              #

     9 #   This program is distributed in the hope that it will be useful,            #

    10 #   but WITHOUT ANY WARRANTY; without even the implied warranty of             #

    11 #   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the              #

    12 #   GNU General Public License for more details.                               #

    13 #                                                                              #

    14 #   You should have received a copy of the GNU General Public License          #

    15 #   along with this program.  If not, see <http://www.gnu.org/licenses/>.      #

    16 #                                                                              #

    17 ################################################################################

    18 

    19 import cwiid

    20 import sys

    21 from threading import Thread

    22 ## Configuration

    23 wm = None

    24 wiimote_hwaddr = '' # Use your address to speed up the connection proccess

    25 #wiimote_hwaddr = '00:19:1D:5D:5D:DC'

    26 

    27 last_point = (0,0)

    28 btA = 0

    29 btB = 0

    30 

    31 def cback(messages):

    32     '''Wiimote callback managing method

    33     Recieves a message list, each element is different, see the libcwiid docs'''

    34     global btA, btB, last_point

    35     #print messages

    36     #print "wiimote callback"

    37     for msg in messages:

    38 #        if msg[0] == cwiid.MESG_IR:

    39 #            # msg is of the form (cwiid.MESG_IR, (((x, y), size) or None * 4))

    40 #            for p in msg[1]:

    41 #                if p:

    42 #                    pos = p['pos'][0], p['pos'][1] # point is mirrored

    43 #                    #s = max(p['size'], 1)

    44 #                        

    45 #                    last_point = tuple(pos)

    46 #                    #print "last_point",last_point

    47         if msg[0] == cwiid.MESG_BTN:

    48             # msg is of the form (cwiid.MESG_BTN, cwiid.BTN_*)

    49             if msg[1] & cwiid.BTN_A:

    50                 btA = 1

    51                 #print "btA = 1"

    52             else:

    53                 btA = 0

    54                 #print "btA = 0"

    55                 

    56             if msg[1] & cwiid.BTN_B:

    57                 btB = 1

    58                 #print "btB = 1"

    59             else:

    60                 btB = 0

    61         elif msg[0] == cwiid.MESG_NUNCHUK:

    62                #sbb = msg[1]['buttons']

    63                last_point = msg[1]['stick']

    64                #ssx = msg[1]['stick'][0]

    65                #ssy = msg[1]['stick'][1]

    66                #msg[1]['acc'][0]

    67                #msg[1]['acc'][1]

    68                #msg[1]['acc'][2]

    69                 #print "btB = 0"

    70         #elif msg[0] == cwiid.MESG_STATUS:

    71         #    # msg is of the form (cwiid.MESG_BTN, { 'status' : value, ... })

    72         #    print msg[1]

    73 

    74 def Connect_Wiimote(frameobj):

    75     global wm,wiimote_hwaddr

    76     try:

    77     #if False:

    78         print "Press 1+2 Now !!!!"

    79         sys.stdout.flush()

    80         wm = cwiid.Wiimote(wiimote_hwaddr)

    81         if wm is not None:

    82             # each message will contain info about ir and buttons

    83             wm.rpt_mode = cwiid.RPT_BTN | cwiid.RPT_EXT # | cwiid.RPT_STATUS | cwiid.RPT_IR | 

    84             # tell cwiid to use the callback interface and allways send button events

    85             wm.enable(cwiid.FLAG_MESG_IFC)

    86                       #| cwiid.FLAG_NONBLOCK

    87                       #| cwiid.FLAG_REPEAT_BTN)

    88 

    89             # specify wich function will manage messages AFTER the other settings

    90             wm.mesg_callback = cback

    91 

    92             # quick check on the wiimote

    93             print "Got Wiimote!"

    94             frameobj.label_1.SetLabel("Got Wiimote !")

    95             st = wm.state

    96             for e in st:

    97                 print str(e).ljust(8), ">", st[e]

    98     except:

    99     #else:

   100         print "Error with wiimote " + str(wiimote_hwaddr)

   101         frameobj.label_1.SetLabel("Wiimote NOK")

   102     sys.stdout.flush()

   103           

   104 def _runtime_cleanup():

   105     if wm is not None:

   106         wm.close()

   107 

   108 def start_manu(self,evt):

   109     self.label_1.SetLabel("press 1+2 now !!!")

   110     wx.CallAfter(Connect_Wiimote, self)

   111     evt.Skip()

   112 HMIFrame.start_manu = start_manu

   113 

   114 def _runtime_begin():

   115     pass

   116     #wx.CallAfter(Connect_Wiimote)

   117 #Thread(target=Connect_Wiimote).start()

   118 

   119 #create_frame()

   120 #wx.Yield()