-
Notifications
You must be signed in to change notification settings - Fork 27
/
test_point.py
59 lines (45 loc) · 1.59 KB
/
test_point.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
import rospy,sys
from utils.geometry import Vector2D
from utils.functions import *
from krssg_ssl_msgs.msg import point_2d
from krssg_ssl_msgs.msg import BeliefState
from krssg_ssl_msgs.msg import gr_Commands
from krssg_ssl_msgs.msg import gr_Robot_Command
from krssg_ssl_msgs.msg import BeliefState
from role import GoToBall, GoToPoint, pass_receive
from multiprocessing import Process
from kubs import kubs
from krssg_ssl_msgs.srv import bsServer
from math import atan2,pi
from utils.functions import *
pub = rospy.Publisher('/grsim_data',gr_Commands,queue_size=1000)
def function(id_,state):
kub = kubs.kubs(id_,state,pub)
kub.update_state(state)
print(kub.kubs_id)
g_fsm = GoToPoint.GoToPoint()
# g_fsm = GoToPoint.GoToPoint()
g_fsm.add_kub(kub)
g_fsm.add_point(point=Vector2D(state.homePos[id_].x,state.homePos[id_].y),orient=normalize_angle(pi+atan2(state.ballPos.y-state.homePos[id_].y,state.ballPos.x-state.homePos[id_].x)))
#g_fsm.add_theta(theta=normalize_angle(pi+atan2(state.ballPos.y,state.ballPos.x+3000)))
g_fsm.as_graphviz()
g_fsm.write_diagram_png()
g_fsm.spin()
#
#print str(kub.kubs_id) + str('***********')
rospy.init_node('node',anonymous=False)
start_time = rospy.Time.now()
start_time = 1.0*start_time.secs + 1.0*start_time.nsecs/pow(10,9)
# rospy.Subscriber('/belief_state', BeliefState, BS_callback, queue_size=1000)
while True:
state=None
rospy.wait_for_service('bsServer',)
getState = rospy.ServiceProxy('bsServer',bsServer)
try:
state = getState(state)
except rospy.ServiceException, e:
print e
if state:
function(1,state.stateB)
# break
rospy.spin()