odroid_app/ws_server.py at master · pshulga-dev/odroid_app · 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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
import websockets
import asyncio
import time
import json
from hk_usb_io import *

usb = init()

ior = {'ANSELA': 0x5b, 'ANSELB' : 0x5c ,'ANSELC' : 0x5d, 'ANSELD' : 0x5e, 'ANSELE' : 0x5f,
	   'PORTA' : 0x80, 'PORTB' :  0x81, 'PORTC'  : 0x82, 'PORTD'  : 0x83, 'PORTE'  : 0x84,
	  'TRISA' : 0x92, 'TRISB' :  0x93, 'TRISC'  : 0x94, 'TRISD'  : 0x95, 'TRISE'  : 0x96,
	  'LATA'  : 0x89, 'LATB'  :  0x8a, 'LATC'   : 0x8b, 'LATD'   : 0x8c, 'LATE'   : 0x8d }

sfr_set_regbit(usb, ior['ANSELA'], 5, 0)	# RA5 is digital
sfr_set_regbit(usb, ior['ANSELA'], 4, 0)	# RA4 is digital
sfr_set_regbit(usb, ior['ANSELA'], 1, 0)	# RA1 is digital
sfr_set_regbit(usb, ior['ANSELB'], 4, 0)	# RB4 is digital

sfr_set_regbit(usb, ior['TRISA'],  5, 0)	# RA5 pin 0 - output, 1 - input
sfr_set_regbit(usb, ior['TRISA'],  4, 1)	# RA4 pin 0 - output, 1 - input
sfr_set_regbit(usb, ior['TRISA'],  1, 1)	# RA1 pin 0 - output, 1 - input
sfr_set_regbit(usb, ior['TRISB'],  4, 0)	# RA4 pin 0 - output, 1 - input

RA5 = sfr_get_regbit(usb, ior['PORTA'], 5)
RA1 = sfr_get_regbit(usb, ior['PORTA'], 1)
RA4 = sfr_get_regbit(usb, ior['PORTA'], 4)
RB4 = sfr_get_regbit(usb, ior['PORTB'], 4)


class Handler(object):
    def __init__(self):
        self.ramp = '0'
        self.velocity1 = '0'
        self.velocity2 = '0'
        self.dwell = '0'
        self.limit_upper = '0'
        self.limit_lower = '0'

    def check_port(self):
        try:

            if RA5 == 0:
                ser_puts(usb, "ZA"+self.ramp)
            time.sleep(0.01)

            if RA5 == 0:
                ser_puts(usb, "ZD"+self.velocity1)
            time.sleep(0.01)

            if RA5 == 0:
                ser_puts(usb, "ZF"+self.velocity2)
            time.sleep(0.01)

            if RA5 == 0:
                ser_puts(usb, "ZB"+self.limit_upper)
            time.sleep(0.01)

            if RA5 == 0:
                ser_puts(usb, "ZE"+self.limit_lower)
            time.sleep(0.01)

            if RA5 == 0 and self.dwell != '0':
                ser_puts(usb, "ZJ"+self.dwell)
            time.sleep(0.01)

            if RA5 == 0:
                ser_puts(usb, "H")
            sfr_set_regbit(usb, ior['TRISA'],  5, 0)
        except Exception as e:
            print(e)

    async def set_graph_data(self, x):
        try:
            # y = 0.0
            # if RA1 == 1:
            #     y = ser_getc(usb)
            # if RA1 == 0:
            #     y = ser_getc(usb)
            y = ser_getc(usb)
            print(y)
            return json.dumps((x, float(y)))
        except Exception as e:
            print(e)

    async def handler(self, websocket, path):
        x = 0
        while True:
            check_first_click = await websocket.recv()
            check_first_click = json.loads(check_first_click)
            if check_first_click['start']:
                self.velocity1 = check_first_click['velocity1']
                self.velocity2 = check_first_click['velocity2']
                self.ramp = check_first_click['ramp']
                self.dwell = check_first_click['dwell']
                self.limit_lower = check_first_click['lower']
                self.limit_upper = check_first_click['upper']
                sfr_set_regbit(usb, ior['TRISA'],  5, 1)
                self.check_port()
                while True:
                    tasks = [asyncio.ensure_future(websocket.recv()), asyncio.ensure_future(self.set_graph_data(x))]
                    done, panding = await asyncio.wait(tasks, return_when=asyncio.FIRST_COMPLETED)
                    if tasks[0] in done:
                        if json.loads(tasks[0].result())['start'] is False:
                            sfr_set_regbit(usb, ior['TRISA'],  5, 0)
                            break
                    else:
                        tasks[0].cancel()
                    if tasks[1] in done:
                        mess = tasks[1].result()
                        await websocket.send(str(mess))
                        await asyncio.sleep(0.1)
                    else:
                        tasks[1].cancel()
                    x += 1

start_server = websockets.serve(Handler().handler, 'localhost', 8765)

loop = asyncio.get_event_loop()

loop.run_until_complete(start_server)

loop.run_forever()