-
Notifications
You must be signed in to change notification settings - Fork 0
/
socket_server.py
96 lines (84 loc) · 2.65 KB
/
socket_server.py
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
#!/usr/bin/env python3
import RPi.GPIO as GPIO;
from time import sleep;
GPIO.setmode(GPIO.BCM)
import socket
import numpy as np
import json
import time
import random
import pickle
import math
import os
import csv
import sys
HOST = '192.168.1.237' # Standard loopback interface address (localhost)
PORT = 65431 # Port to listen on (non-privileged ports are > 1023)
sample = 0
POUT1 = 18 #red
POUT2 = 23 #green
GPIO.setup(POUT1, GPIO.OUT);
GPIO.setup(POUT2, GPIO.OUT);
GPIO.output(POUT1, False);
GPIO.output(POUT2, False);
def get_graph_data():
global sample
sample += 1
graph_to_send = json.dumps({
'x':random.uniform(-1,1),
'y':random.uniform(-1,1),
'z':random.uniform(-1,1),
's':sample
})
return graph_to_send
try:
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
s.bind((HOST, PORT))
s.listen()
conn, addr = s.accept()
with conn:
print('Connected by', addr)
a = 0
#accel_arr = np.zeros((1000,3))
#abs_acc = np.zeros(1000)
while True:
data = conn.recv(1024)
dec = data.decode()
movement_list = ["predict","fall","run","walk","still"]
print(movement_list[int(dec)])
if int(dec)==0:
GPIO.output(POUT2, True);
GPIO.output(POUT1, True);
elif int(dec)==1:
GPIO.output(POUT1, True);
GPIO.output(POUT2, False);
else:
GPIO.output(POUT1, False);
GPIO.output(POUT2, True);
except KeyboardInterrupt:
GPIO.cleanup() # clean up GPIO on CTRL+C exit
GPIO.cleanup() # clean up GPIO on normal exit
"""
if (dec!=""):
x = dec.split(",")[:-1]
#print(x)
n_data = int(len(x)/3)
for b in range(n_data):
accel_arr[a] = np.array(x[3*b:3*(b+1)])
sq_sum = np.sum(accel_arr[a]**2)
print(math.sqrt(sq_sum))
abs_acc[a] = math.sqrt(sq_sum)
a = a+1
#print(x[3*(b-1):3*b])
#print(dec)
"""
"""
with open('accel_arr_'+sys.argv[1], 'wb') as f:
pickle.dump(accel_arr,f)
#print(type(data))
#conn.sendall(get_graph_data().encode('utf-8'))
with open('abs_acc_'+sys.argv[1]+'.csv', 'w', newline='', encoding='utf-8') as wf:
writer = csv.writer(wf)
for index in range (1000):
writer.writerow([abs_acc[index]])
"""