make class

This commit is contained in:
Kseninia Mikhaylova 2024-09-18 13:57:46 +03:00
parent 32439d2428
commit 6d85eae4fd
5 changed files with 208 additions and 13274 deletions

View File

@ -1,142 +1,195 @@
import socket
import json
from pprint import pprint
from tkinter import ttk
from func import *
s = socket.socket()
host = MODBUS_SERVER_HOST
port = 9760
s.connect((host, port))
# print(s)
def send_data(data):
response = None
if data["reqType"] == 'AddRCC':
# return
pass
s.send(str.encode(json.dumps(data)))
response_data = s.recv(1024)
response = json.loads(response_data)
if data["reqType"] == 'query':
# pprint({"request":data["queryAddr"], "response":response["queryData"]})
return response["queryData"]
elif data["reqType"] == 'command':
# pprint({"request":data["cmdData"], "response":response["cmdReply"]})
return response["cmdReply"]
elif data["reqType"] == 'AddRCC' and "cmdReply" in response.keys():
return response["cmdReply"]
else:
pprint(response)
class SocketRobotArm:
line_speed = 100.0
line_smooth = 9
line_tool = 1
def make_addrcc_data(q, emptyList=0):
# print(q)
return {"dsID": "www.hc-system.com.HCRemoteCommand","reqType":"AddRCC","emptyList":str(emptyList),"packID":"0","instructions":q}
def make_command_data(q):
return {"dsID": "www.hc-system.com.RemoteMonitor","reqType":"command","packID":"0","cmdData":q}
def make_query_data(q):
return {"dsID": "www.hc-system.com.RemoteMonitor","reqType":"query","packID":"0","queryAddr":q}
global_speed = 100
physical_speed = 50
# laser_id = 15
laser_id = 14
# filename = 'half-sphere-no-angle'
filename = "half-sphere"
axis_coord_raw = send_data(make_query_data(['axis-0','axis-1','axis-2','axis-3','axis-4','axis-5']))
axis_coord = []
for i in axis_coord_raw:
axis_coord.append(float(i))
pass_size = 4
world_coord_raw = send_data(make_query_data(['world-0','world-1','world-2','world-3','world-4','world-5']))
world_coord = []
for i in world_coord_raw:
world_coord.append(float(i))
x,y,z,u,v,w = world_coord
def __init__(self, *args, **kwargs):
self.socket = socket.socket()
self.host = MODBUS_SERVER_HOST
self.port = 9760
print('axis', axis_coord)
print('world', world_coord)
print('remote command count', send_data(make_query_data(['RemoteCmdLen'])))
self.host = "127.0.0.1"
self.port = 65432
line_speed = 100.0
line_smooth = 9
line_tool = 1
self.connect()
global_speed = 100
physical_speed = 50
# laser_id = 15
laser_id = 14
# filename = 'half-sphere-no-angle'
filename = 'half-sphere'
def __exit__(self, exc_type, exc_value, traceback):
self.close()
pass_size = 4
def connect(self):
self.socket.connect((self.host, self.port))
def make_world_step(type, p):
step = {
"oneshot": "0",
"delay":"0.0",
"speed":str(line_speed),"smooth":str(line_smooth),
"coord":"0","tool":str(line_tool),
"ckStatus":"0x3F",
}
if type == 'line':
m0,m1,m2,m3,m4,m5=p
step.update({
"action": "10",
"m0": round(x + m0,3), "m1": round(y + m1,3), "m2": round(z + m2,3),
"m3": round(u + m3,3), "m4": round(v + m4,3), "m5": round(w + m5,3),
"m6": 0, "m7": 0,
})
elif type == 'curve':
m0,m1,m2,m3,m4,m5, m0_p, m1_p,m2_p,m3_p,m4_p,m5_p=p
step.update({
"action": "17",
"m0": round(x + m0,3), "m1": round(y + m1,3), "m2": round(z + m2,3),
"m3": round(u + m3,3), "m4": round(v + m4,3), "m5": round(w + m5,3),
"m6": 0, "m7": 0,
"m0_p": round(x + m0_p,3), "m1_p": round(y + m1_p,3), "m2_p": round(z + m2_p,3),
"m3_p": round(u + m3_p,3), "m4_p": round(v + m4_p,3), "m5_p": round(w + m3_p,3),
"m6_p": 0, "m7_p": 0,
})
self.get_axis()
self.get_world()
self.get_command_count()
for s in step:
step[s] = str(step[s])
return step
self.send_data(self.set_global_speed())
result = []
with open(f'data/{filename}.nc.result', 'r') as fp:
for line in fp:
data = line.strip().split(' ')
prep = {}
for item in data:
prep[item[:1]] = float(item[1:])
result.append(
make_world_step(
'line',
(
prep.get('X', 0), prep.get('Y', 0), prep.get('Z', 0),
prep.get('U', 0), prep.get('V', 0), prep.get('W', 0)
)
self.add_rcc_list = (
[self.set_physical_speed(True), self.set_output_laser(True)]
+ self.steps_from_file()
+ [self.set_physical_speed(False), self.set_output_laser(False)]
)
step = 2
for i in range(0, len(self.add_rcc_list), step):
print(self.add_rcc_list[i : i + step])
self.send_data(
make_addrcc_data(self.add_rcc_list[i : i + step], int(not i))
)
self.send_data(self.start_cycle())
def send_data(self, data):
response = None
# if data["reqType"] == "AddRCC":
# return
self.socket.send(str.encode(json.dumps(data)))
response_data = self.socket.recv(1024)
response = json.loads(response_data)
if data["reqType"] == "query":
return response["queryData"]
elif data["reqType"] == "command":
return response["cmdReply"]
elif data["reqType"] == "AddRCC" and "cmdReply" in response.keys():
return response["cmdReply"]
else:
pprint(response)
def get_axis(self):
axis_coord_raw = self.send_data(
make_query_data(
["axis-0", "axis-1", "axis-2", "axis-3", "axis-4", "axis-5"]
)
)
self.start_axis_coordinates = [float(i) for i in axis_coord_raw]
print("start_axis_coordinates", self.start_axis_coordinates)
# Изменили глобальную скорость на global_speed%
send_data(make_command_data(['modifyGSPD', str(global_speed * 10)]))
send_data(make_addrcc_data([
# Включили физическую скорость
{"oneshot":"0", "action":"51","isUse":"1","speed":str(physical_speed*1000)},
# Поставили Y026 (laser id + 11) в TRUE -- ЛАЗЕР!!!!
{"oneshot":"0", "action":"200","type":"0","io_status":"1", "point":str(laser_id)},
], 1))
def get_world(self):
world_coord_raw = self.send_data(
make_query_data(
["world-0", "world-1", "world-2", "world-3", "world-4", "world-5"]
)
)
self.start_world_coordinates = [float(i) for i in world_coord_raw]
print("start_world_coordinates", self.start_world_coordinates)
res_list = len(result) + 1
if res_list > 300:
res_list = 300
def get_command_count(self):
res = self.send_data(make_query_data(["RemoteCmdLen"]))
self.remote_command_count = res
print(res)
for i in range(pass_size, res_list, pass_size):
print(f"Отправили на устройство команды {i} ... {i + 1}")
send_data(make_addrcc_data(result[i:i+1]))
def set_global_speed(self):
# Изменили глобальную скорость на global_speed%
return make_command_data(["modifyGSPD", str(self.global_speed * 10)])
send_data(make_addrcc_data([
{"oneshot":"0", "action":"51","isUse":"0"},
{"oneshot":"0", "action":"200","type":"0","io_status":"0", "point":str(laser_id)},
]))
def start_cycle(self):
return make_command_data(["actionSingleCycle"])
print('remote command count', send_data(make_query_data(['RemoteCmdLen'])))
send_data(make_command_data(['actionSingleCycle']))
# send_data(make_command_data(['actionStop']))
def round(v):
return round(v, 3)
def make_world_step(self, type, point):
step = {
"oneshot": "0",
"delay": "0.0",
"speed": str(self.line_speed),
"smooth": str(self.line_smooth),
"coord": "0",
"tool": str(self.line_tool),
"ckStatus": "0x3F",
}
if type == "line":
pairs = zip(self.start_world_coordinates, point)
m0, m1, m2, m3, m4, m5 = [round(sum(i), 3) for i in pairs]
step.update({"action": "10"})
step.update({"m0": m0, "m1": m1, "m2": m2, "m3": m3, "m4": m4, "m5": m5})
step.update({"m6": 0, "m7": 0})
elif type == "curve":
pairs = zip(self.start_world_coordinates, point[:5])
m0, m1, m2, m3, m4, m5 = [round(sum(i), 3) for i in pairs]
pairs_p = zip(self.start_world_coordinates, point[6:])
m0_p, m1_p, m2_p, m3_p, m4_p, m5_p = [round(sum(i), 3) for i in pairs_p]
step.update({"action": "17"})
step.update({"m0": m0, "m1": m1, "m2": m2, "m3": m3, "m4": m4, "m5": m5})
step.update({"m6": 0, "m7": 0})
step.update(
{
"m0_p": m0_p,
"m1_p": m1_p,
"m2_p": m2_p,
"m3_p": m3_p,
"m4_p": m4_p,
"m5_p": m5_p,
}
)
step.update({"m6_p": 0, "m7_p": 0})
for s in step:
step[s] = str(step[s])
return step
def set_physical_speed(self, status: bool = False):
return (
{
"oneshot": "0",
"action": "51",
"isUse": str(int(status)),
"speed": str(self.physical_speed * 1000),
},
)
def set_output_laser(self, status: bool = False):
return (
{
"oneshot": "0",
"action": "200",
"type": "0",
"io_status": str(int(status)),
"point": str(self.laser_id),
},
)
def steps_from_file(self):
result = []
with open(f"data/{self.filename}.nc.result", "r") as fp:
for line in fp:
data = line.strip().split(" ")
prep = {}
for item in data:
prep[item[:1]] = float(item[1:])
result.append(
self.make_world_step(
"line",
(
prep.get("X", 0),
prep.get("Y", 0),
prep.get("Z", 0),
prep.get("U", 0),
prep.get("V", 0),
prep.get("W", 0),
),
)
)
return result
SocketRobotArm()

File diff suppressed because it is too large Load Diff

File diff suppressed because it is too large Load Diff

10
func.py
View File

@ -112,4 +112,12 @@ def stop_immediately(client):
client.write_register(20004, 1, MODBUS_SLAVE_ID)
def multiply_1000(n):
return int(n * 1000)
return int(n * 1000)
def make_addrcc_data(q, emptyList=0):
# print(q)
return {"dsID": "www.hc-system.com.HCRemoteCommand","reqType":"AddRCC","emptyList":str(emptyList),"packID":"0","instructions":q}
def make_command_data(q):
return {"dsID": "www.hc-system.com.RemoteMonitor","reqType":"command","packID":"0","cmdData":q}
def make_query_data(q):
return {"dsID": "www.hc-system.com.RemoteMonitor","reqType":"query","packID":"0","queryAddr":q}

27
test_socket_server.py Normal file
View File

@ -0,0 +1,27 @@
import socket
import json
HOST = "127.0.0.1" # Standard loopback interface address (localhost)
PORT = 65432 # Port to listen on (non-privileged ports are > 1023)
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
s.bind((HOST, PORT))
s.listen()
conn, addr = s.accept()
with conn:
print(f"Connected by {addr}")
while True:
data = conn.recv(1024)
if data:
req = json.loads(data)
res = {"queryData": ["ok"]}
if "queryAddr" in req.keys() and "axis-0" in req["queryAddr"]:
res["queryData"] = [10, 11, 12, 13, 14, 15]
if "queryAddr" in req and "world-0" in req["queryAddr"]:
res["queryData"] = [100, 101, 102, 103, 104, 105]
if req["reqType"] == "command":
res["cmdReply"] = ['ok']
conn.sendall(json.dumps(res).encode())