def run(): global servo_move, speed_set, servo_command, functionMode, init_get, R_set, G_set, B_set, SR_mode servo.servo_init() move.setup() findline.setup() direction_command = 'no' turn_command = 'no' servo_command = 'no' speed_set = 100 rad = 0.5 info_threading = threading.Thread( target=info_send_client) #Define a thread for FPV and OpenCV info_threading.setDaemon( True ) #'True' means it is a front thread,it would close when the mainloop() closes info_threading.start() #Thread starts servo_move = Servo_ctrl() servo_move.start() servo_move.pause() findline.setup() while True: data = '' data = str(tcpCliSock.recv(BUFSIZ).decode()) if not data: continue elif 'forward' == data: direction_command = 'forward' move.move(speed_set, direction_command) elif 'backward' == data: direction_command = 'backward' move.move(speed_set, direction_command) elif 'DS' in data: direction_command = 'no' move.move(speed_set, direction_command) elif 'left' == data: # turn_command = 'left' servo.turnLeft() elif 'right' == data: # turn_command = 'right' servo.turnRight() elif 'TS' in data: # turn_command = 'no' servo.turnMiddle() elif 'Switch_1_on' in data: fpv.Sounds(1) tcpCliSock.send(('Switch_1_on').encode()) elif 'Switch_1_off' in data: fpv.Sounds(0) tcpCliSock.send(('Switch_1_off').encode()) elif 'Switch_2_on' in data: #switch.switch(2,1) fpv.Radar(1) tcpCliSock.send(('Switch_2_on').encode()) elif 'Switch_2_off' in data: #switch.switch(2,0) fpv.Radar(0) tcpCliSock.send(('Switch_2_off').encode()) elif 'Switch_3_on' in data: switch.switch(3, 1) tcpCliSock.send(('Switch_3_on').encode()) elif 'Switch_3_off' in data: switch.switch(3, 0) tcpCliSock.send(('Switch_3_off').encode()) elif 'Switch_A_on' in data: tcpCliSock.send(('Switch_A_on').encode()) elif 'Switch_A_off' in data: tcpCliSock.send(('Switch_A_off').encode()) elif 'Switch_B_on' in data: tcpCliSock.send(('Switch_B_on').encode()) elif 'Switch_B_off' in data: tcpCliSock.send(('Switch_B_off').encode()) elif 'Switch_C_on' in data: fpv.HaarJJ(1) tcpCliSock.send(('Switch_C_on').encode()) elif 'Switch_C_off' in data: fpv.HaarJJ(0) tcpCliSock.send(('Switch_C_off').encode()) elif 'Face_Track_on' in data: # functionMode = 3 fpv.FaceTrack(1) tcpCliSock.send(('Face_Track_on').encode()) elif 'Face_Track_off' in data: # functionMode = 0 fpv.FaceTrack(0) tcpCliSock.send(('Face_Track_off').encode()) elif 'function_1_on' in data: servo.ahead() time.sleep(0.2) tcpCliSock.send(('function_1_on').encode()) radar_send = servo.radar_scan() tcpCliSock.sendall(radar_send.encode()) print(radar_send) time.sleep(0.3) tcpCliSock.send(('function_1_off').encode()) elif 'function_2_on' in data: functionMode = 2 fpv.FindColor(1) tcpCliSock.send(('function_2_on').encode()) elif 'function_2_off' in data: functionMode = 0 fpv.FindColor(0) switch.switch(1, 0) switch.switch(2, 0) switch.switch(3, 0) tcpCliSock.send(('function_2_off').encode()) elif 'function_3_on' in data: functionMode = 3 fpv.WatchDog(1) tcpCliSock.send(('function_3_on').encode()) elif 'function_3_off' in data: functionMode = 0 fpv.WatchDog(0) tcpCliSock.send(('function_3_off').encode()) elif 'function_4_on' in data: functionMode = 4 servo_move.resume() tcpCliSock.send(('function_4_on').encode()) elif 'function_4_off' in data: functionMode = 0 servo_move.pause() move.motorStop() tcpCliSock.send(('function_4_off').encode()) elif 'function_5_on' in data: functionMode = 5 servo_move.resume() tcpCliSock.send(('function_5_on').encode()) elif 'function_5_off' in data: functionMode = 0 servo_move.pause() move.motorStop() tcpCliSock.send(('function_5_off').encode()) elif 'function_6_on' in data: if MPU_connection: functionMode = 6 servo_move.resume() tcpCliSock.send(('function_6_on').encode()) elif 'function_6_off' in data: functionMode = 0 servo_move.pause() move.motorStop() init_get = 0 tcpCliSock.send(('function_6_off').encode()) #elif 'function_1_off' in data: # tcpCliSock.send(('function_1_off').encode()) elif 'lookleft' == data: servo_command = 'lookleft' servo_move.resume() elif 'lookright' == data: servo_command = 'lookright' servo_move.resume() elif 'up' == data: servo_command = 'up' servo_move.resume() elif 'down' == data: servo_command = 'down' servo_move.resume() elif 'stop' == data: if not functionMode: servo_move.pause() servo_command = 'no' pass elif 'home' == data: servo.ahead() elif 'CVrun' == data: if not FPV.CVrun: FPV.CVrun = 1 tcpCliSock.send(('CVrun_on').encode()) else: FPV.CVrun = 0 tcpCliSock.send(('CVrun_off').encode()) elif 'wsR' in data: try: set_R = data.split() R_set = int(set_R[1]) led.colorWipe(R_set, G_set, B_set) except: pass elif 'wsG' in data: try: set_G = data.split() G_set = int(set_G[1]) led.colorWipe(R_set, G_set, B_set) except: pass elif 'wsB' in data: try: set_B = data.split() B_set = int(set_B[1]) led.colorWipe(R_set, G_set, B_set) except: pass elif 'pwm0' in data: try: set_pwm0 = data.split() pwm0_set = int(set_pwm0[1]) servo.setPWM(0, pwm0_set) except: pass elif 'pwm1' in data: try: set_pwm1 = data.split() pwm1_set = int(set_pwm1[1]) servo.setPWM(1, pwm1_set) except: pass elif 'pwm2' in data: try: set_pwm2 = data.split() pwm2_set = int(set_pwm2[1]) servo.setPWM(2, pwm2_set) except: pass elif 'Speed' in data: try: set_speed = data.split() speed_set = int(set_speed[1]) except: pass elif 'Save' in data: try: servo.saveConfig() except: pass elif 'CVFL' in data: if not FPV.FindLineMode: FPV.FindLineMode = 1 tcpCliSock.send(('CVFL_on').encode()) else: move.motorStop() FPV.FindLineMode = 0 tcpCliSock.send(('CVFL_off').encode()) elif 'Render' in data: if FPV.frameRender: FPV.frameRender = 0 else: FPV.frameRender = 1 elif 'WBswitch' in data: if FPV.lineColorSet == 255: FPV.lineColorSet = 0 else: FPV.lineColorSet = 255 elif 'lip1' in data: try: set_lip1 = data.split() lip1_set = int(set_lip1[1]) FPV.linePos_1 = lip1_set except: pass elif 'lip2' in data: try: set_lip2 = data.split() lip2_set = int(set_lip2[1]) FPV.linePos_2 = lip2_set except: pass elif 'err' in data: try: set_err = data.split() err_set = int(set_err[1]) FPV.findLineError = err_set except: pass elif 'FCSET' in data: FCSET = data.split() fpv.colorFindSet(int(FCSET[1]), int(FCSET[2]), int(FCSET[3])) elif 'setEC' in data: #Z ECset = data.split() try: fpv.setExpCom(int(ECset[1])) except: pass elif 'defEC' in data: #Z fpv.defaultExpCom() elif 'police' in data: if LED.ledfunc != 'police': tcpCliSock.send(('rainbow_off').encode()) LED.ledfunc = 'police' ledthread.resume() tcpCliSock.send(('police_on').encode()) elif LED.ledfunc == 'police': LED.ledfunc = '' ledthread.pause() tcpCliSock.send(('police_off').encode()) elif 'rainbow' in data: if LED.ledfunc != 'rainbow': tcpCliSock.send(('police_off').encode()) LED.ledfunc = 'rainbow' ledthread.resume() tcpCliSock.send(('rainbow_on').encode()) elif LED.ledfunc == 'rainbow': LED.ledfunc = '' ledthread.pause() tcpCliSock.send(('rainbow_off').encode()) elif 'sr' in data: if not SR_mode: if SR_dect: SR_mode = 1 tcpCliSock.send(('sr_on').encode()) sr.resume() elif SR_mode: SR_mode = 0 sr.pause() move.motorStop() tcpCliSock.send(('sr_off').encode()) else: pass print(data)
def run(): global servo_move, speed_set, servo_command, functionMode, init_get, R_set, G_set, B_set, SR_mode servo.servo_init() move.setup() findline.setup() direction_command = "no" turn_command = "no" servo_command = "no" speed_set = 100 rad = 0.5 info_threading = threading.Thread( target=info_send_client ) # Define a thread for FPV and OpenCV info_threading.setDaemon( True ) #'True' means it is a front thread,it would close when the mainloop() closes info_threading.start() # Thread starts servo_move = Servo_ctrl() servo_move.start() servo_move.pause() findline.setup() while True: data = "" data = str(tcpCliSock.recv(BUFSIZ).decode()) if not data: continue elif "forward" == data: direction_command = "forward" move.move(speed_set, direction_command) elif "backward" == data: direction_command = "backward" move.move(speed_set, direction_command) elif "DS" in data: direction_command = "no" move.move(speed_set, direction_command) elif "left" == data: # turn_command = 'left' servo.turnLeft() elif "right" == data: # turn_command = 'right' servo.turnRight() elif "TS" in data: # turn_command = 'no' servo.turnMiddle() elif "Switch_1_on" in data: switch.switch(1, 1) tcpCliSock.send(("Switch_1_on").encode()) elif "Switch_1_off" in data: switch.switch(1, 0) tcpCliSock.send(("Switch_1_off").encode()) elif "Switch_2_on" in data: switch.switch(2, 1) tcpCliSock.send(("Switch_2_on").encode()) elif "Switch_2_off" in data: switch.switch(2, 0) tcpCliSock.send(("Switch_2_off").encode()) elif "Switch_3_on" in data: switch.switch(3, 1) tcpCliSock.send(("Switch_3_on").encode()) elif "Switch_3_off" in data: switch.switch(3, 0) tcpCliSock.send(("Switch_3_off").encode()) elif "function_1_on" in data: servo.ahead() time.sleep(0.2) tcpCliSock.send(("function_1_on").encode()) radar_send = servo.radar_scan() tcpCliSock.sendall(radar_send.encode()) print(radar_send) time.sleep(0.3) tcpCliSock.send(("function_1_off").encode()) elif "function_2_on" in data: functionMode = 2 fpv.FindColor(1) tcpCliSock.send(("function_2_on").encode()) elif "function_3_on" in data: functionMode = 3 fpv.WatchDog(1) tcpCliSock.send(("function_3_on").encode()) elif "function_4_on" in data: functionMode = 4 servo_move.resume() tcpCliSock.send(("function_4_on").encode()) elif "function_5_on" in data: functionMode = 5 servo_move.resume() tcpCliSock.send(("function_5_on").encode()) elif "function_6_on" in data: if MPU_connection: functionMode = 6 servo_move.resume() tcpCliSock.send(("function_6_on").encode()) # elif 'function_1_off' in data: # tcpCliSock.send(('function_1_off').encode()) elif "function_2_off" in data: functionMode = 0 fpv.FindColor(0) switch.switch(1, 0) switch.switch(2, 0) switch.switch(3, 0) tcpCliSock.send(("function_2_off").encode()) elif "function_3_off" in data: functionMode = 0 fpv.WatchDog(0) tcpCliSock.send(("function_3_off").encode()) elif "function_4_off" in data: functionMode = 0 servo_move.pause() move.motorStop() tcpCliSock.send(("function_4_off").encode()) elif "function_5_off" in data: functionMode = 0 servo_move.pause() move.motorStop() tcpCliSock.send(("function_5_off").encode()) elif "function_6_off" in data: functionMode = 0 servo_move.pause() move.motorStop() init_get = 0 tcpCliSock.send(("function_6_off").encode()) elif "lookleft" == data: servo_command = "lookleft" servo_move.resume() elif "lookright" == data: servo_command = "lookright" servo_move.resume() elif "up" == data: servo_command = "up" servo_move.resume() elif "down" == data: servo_command = "down" servo_move.resume() elif "stop" == data: if not functionMode: servo_move.pause() servo_command = "no" pass elif "home" == data: servo.ahead() elif "CVrun" == data: if not FPV.CVrun: FPV.CVrun = 1 tcpCliSock.send(("CVrun_on").encode()) else: FPV.CVrun = 0 tcpCliSock.send(("CVrun_off").encode()) elif "wsR" in data: try: set_R = data.split() R_set = int(set_R[1]) led.colorWipe(R_set, G_set, B_set) except: pass elif "wsG" in data: try: set_G = data.split() G_set = int(set_G[1]) led.colorWipe(R_set, G_set, B_set) except: pass elif "wsB" in data: try: set_B = data.split() B_set = int(set_B[1]) led.colorWipe(R_set, G_set, B_set) except: pass elif "pwm0" in data: try: set_pwm0 = data.split() pwm0_set = int(set_pwm0[1]) servo.setPWM(0, pwm0_set) except: pass elif "pwm1" in data: try: set_pwm1 = data.split() pwm1_set = int(set_pwm1[1]) servo.setPWM(1, pwm1_set) except: pass elif "pwm2" in data: try: set_pwm2 = data.split() pwm2_set = int(set_pwm2[1]) servo.setPWM(2, pwm2_set) except: pass elif "Speed" in data: try: set_speed = data.split() speed_set = int(set_speed[1]) except: pass elif "Save" in data: try: servo.saveConfig() except: pass elif "CVFL" in data: if not FPV.FindLineMode: FPV.FindLineMode = 1 tcpCliSock.send(("CVFL_on").encode()) else: move.motorStop() FPV.FindLineMode = 0 tcpCliSock.send(("CVFL_off").encode()) elif "Render" in data: if FPV.frameRender: FPV.frameRender = 0 else: FPV.frameRender = 1 elif "WBswitch" in data: if FPV.lineColorSet == 255: FPV.lineColorSet = 0 else: FPV.lineColorSet = 255 elif "lip1" in data: try: set_lip1 = data.split() lip1_set = int(set_lip1[1]) FPV.linePos_1 = lip1_set except: pass elif "lip2" in data: try: set_lip2 = data.split() lip2_set = int(set_lip2[1]) FPV.linePos_2 = lip2_set except: pass elif "err" in data: try: set_err = data.split() err_set = int(set_err[1]) FPV.findLineError = err_set except: pass elif "FCSET" in data: FCSET = data.split() fpv.colorFindSet(int(FCSET[1]), int(FCSET[2]), int(FCSET[3])) elif "setEC" in data: # Z ECset = data.split() try: fpv.setExpCom(int(ECset[1])) except: pass elif "defEC" in data: # Z fpv.defaultExpCom() elif "police" in data: if LED.ledfunc != "police": tcpCliSock.send(("rainbow_off").encode()) LED.ledfunc = "police" ledthread.resume() tcpCliSock.send(("police_on").encode()) elif LED.ledfunc == "police": LED.ledfunc = "" ledthread.pause() tcpCliSock.send(("police_off").encode()) elif "rainbow" in data: if LED.ledfunc != "rainbow": tcpCliSock.send(("police_off").encode()) LED.ledfunc = "rainbow" ledthread.resume() tcpCliSock.send(("rainbow_on").encode()) elif LED.ledfunc == "rainbow": LED.ledfunc = "" ledthread.pause() tcpCliSock.send(("rainbow_off").encode()) elif "sr" in data: if not SR_mode: if SR_dect: SR_mode = 1 tcpCliSock.send(("sr_on").encode()) sr.resume() elif SR_mode: SR_mode = 0 sr.pause() move.motorStop() tcpCliSock.send(("sr_off").encode()) else: pass print(data)