Ejemplo n.º 1
0
            Listener.temp = quat.x
        else:
            print("Lower your hand/n")
            sleep(1)
            Listener.temp = quat.x

        #     newx = 1
        #     Listener.temp = quat.x
        #     sleep(1)
        # else:
        #     newx = 15
        #     Listener.temp = quat.x
        #     sleep(1)
        # print newx
        # response = urllib2.urlopen('http://rpi.michaelbailey.co/control?dc='+str(newx))
        # html = response.read()




init()
hub = Hub()
hub.run(1000, Listener())
try:
    while True:
        sleep(0.5)
except KeyboardInterrupt:
    print('\nQuit')
finally:
    hub.shutdown()  # !! crucial
Ejemplo n.º 2
0
def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('-p', '--person', type=str)
    parser.add_argument('-t', '--type', type=str)
    parser.add_argument('-s', '--session', type=str)
    parser.add_argument('-i', '--iteration', type=str)
    args = parser.parse_args()

    data_sub_folder = os.path.join(DATA_FOLDER, args.person)

    if not os.path.exists(data_sub_folder):
        os.mkdir(data_sub_folder)

    start_time = datetime.now()

    init(
        'C:\\Users\\Ivan\\PycharmProjects\\myo_arm_band\\lib\myo-sdk-win-0.9.0\\bin'
    )
    arm_band_listener = ArmBandListener(start_time)
    hub = Hub()

    controller = Leap.Controller()
    leap_listener = LeapListener(start_time)
    controller.add_listener(leap_listener)

    sleep(10)

    try:
        hub.run(1000, arm_band_listener)
        print('start recording')

        if args.type == '0':
            record_simple_movements()
        elif args.type == '1':
            record_sophisticated_movements()
        else:
            print('unknown type of recording')
    except KeyboardInterrupt:
        pass
    finally:
        emg_data = arm_band_listener.get_emg_data()
        orientation_data = arm_band_listener.get_orientation_data()
        angles = leap_listener.get_angles_data()

        print('EMG data length: {}\n'
              'Orientation data length: {}'
              '\nAngles data length: {}'.format(len(emg_data),
                                                len(orientation_data),
                                                len(angles)))

        with open(
                os.path.join(
                    data_sub_folder,
                    'raw_{}_{}_{}_{}.pickle'.format(args.person, args.type,
                                                    args.session,
                                                    args.iteration)),
                'wb') as handle:
            pickle.dump((emg_data, orientation_data, angles), handle)

        controller.remove_listener(leap_listener)
        hub.shutdown()
Ejemplo n.º 3
0
		self.data_EMG['x'] = range(self.sEMG_length)
		
		if len(self.select_angle_names)!= 0:
			if self.angle_posit == self.Angle_length - self.window_size:
				self.angle_posit = 0
			for _name1 in self.select_angle_names:
				_index1 = self.Angle_names.index(_name1)
				self.Angle_StreamWin[0] = Myos_angle.angle_init(myo_device,[0,0],[_index1])[0]
				self.stream_Angle[_index1][self.angle_posit : self.angle_posit + self.window_size] = self.Angle_StreamWin
				self.data_Angle[_name1] = self.stream_Angle[_index1]
			self.angle_posit += 1
		self.data_Angle['x'] = range(self.Angle_length)		
				
if __name__ == "__main__":
	init()
	feed = Feed()
	hub = Hub()
	times = 0
	hub.run(1000, feed)	
	try:
		myo_device = feed.get_devices()
		print(myo_device)
		time.sleep(1)
		myo_device[0].set_stream_emg(StreamEmg.enabled)
		myo_device[1].set_stream_emg(StreamEmg.enabled)
		MYO_Grapher = MYOGrapher()
		MYO_Grapher.configure_traits() 
	finally:
		hub.shutdown()
		
Ejemplo n.º 4
0
            x_f_h.append(wl(X1[:, b]))
            x_f_h.append(aac(X1[:, b]))
        # y_i = model.predict(np.column_stack(np.asarray(x_f_h)), verbose=0)
        # y_i_class = y_i.argmax(axis=-1)
        '''''
        a_i = 0
        y_i = y_i[0]
        max_var = max(y_i)
        for a in range(len(y_i)):
            if (y_i[a] == max_var):
                a_i = a + 1
        ''' ''
        p2 = clf2.predict([x_f_h])
        p3 = clf3.predict([x_f_h])
        if p2 == p3:
            if p2[0] == 1:
                print('Pred --- ', ges[0])
            if p2[0] == 2:
                print('Pred --- ', ges[1])
            if p2[0] == 3:
                print('Pred --- ', ges[2])
            if p2[0] == 4:
                print('Pred --- ', ges[3])
            if p2[0] == 5:
                print('Pred --- ', ges[4])

        X = []

sleep(1)
hub.shutdown()
Ejemplo n.º 5
0
    def on_pair(self, myo, timestamp, firmware_version):
        print("Hello, Myo!")

    def on_unpair(self, myo, timestamp):
        print("Goodbye, Myo!")

    def on_orientation_data(self, myo, timestamp, quat):
        print("Orientation:", quat.x, quat.y, quat.z, quat.w)

    def on_battery_level_received(myo, timestamp, level):
    	print("Battery Level:", level)

    def on_accelerometor_data(myo, timestamp, acceleration):
    	print("Accelerometor Data:", acceleration)


init()
listener = DeviceListener()
hub = Hub()
hub.run(1000, listener)

Listener.on_pair()


try:
  while True: sleep(0.5)
finally:
  hub.shutdown()  # !! crucial