#!/usr/bin/env python
import serial
import time
import math
import string
import rospy
import threespace as tsa
import find_dng
from threespace import *
# CALIBRATES ALL CONNECTED IMU
rospy.init_node("calibration")
dng_list = find_dng.returnDev("dng")
for d in dng_list:
tsa.TSDongle.broadcastSynchronizationPulse(d)
for device in d:
if device == None:
rospy.logerr("No Sensor Found")
else:
att = 0
res = False
rospy.logwarn("Calibrating %s", str(device))
while res == False:
att += 1
if att > 10:
break
try:
res = tsa.TSWLSensor.beginGyroscopeAutoCalibration(device)
rospy.logwarn("Calibration Successful")
except:
rospy.logerr("Calibration attempt #%d failed, restarting",
#!/usr/bin/env python
import rospy
import find_dng
import threespace as tsa
# PRINTS CHARGE LEVEL FOR ALL CONNECTED IMUS
rospy.init_node("charge_monitor")
r = rospy.Rate(100)
devlist = find_dng.returnDev("dev")
if len(devlist) == 0:
rospy.logwarn("No devices found, exiting")
exit()
while not rospy.is_shutdown():
for device in devlist:
id_ = str(device)
id_ = id_[id_.find('W'):-1]
rospy.logwarn("Device : %s is at %d", id_, tsa.TSWLSensor.getBatteryPercentRemaining(device))
rospy.sleep(rospy.Duration(5))
rospy.logerr("------------------------")
for d in devlist:
tsa.TSWLSensor.close(d)
#!/usr/bin/env python
import roslib
import rospy
import string
import find_dng
import threespace_api as tsa
from threespace_api import *
# PAIRS ANY CONNECTED IMU WITH
# THE CONNECTED DONGLE
rospy.init_node("pairer")
dng_list = find_dng.returnDev("dng")
dev_list = find_dng.returnDev("dev")
if len(dng_list) == 0:
rospy.logerr("No dongles found, exiting")
exit()
if len(dev_list) == 0:
rospy.logerr("No device found, exiting")
exit()
rospy.logwarn(dng_list)
rospy.logwarn(dev_list)
d = dng_list[0]
wc = None
att = 0
while wc is None:
if att > 10:
exit()
att += 1
#!/usr/bin/env python
import roslib
import rospy
import string
import find_dng
import threespace_api as tsa
from threespace_api import *
# PAIRS ANY CONNECTED IMU WITH
# THE CONNECTED DONGLE
rospy.init_node("pairer")
dng_list = find_dng.returnDev("dng")
dev_list = find_dng.returnDev("dev")
if len(dng_list) == 0:
rospy.logerr("No dongles found, exiting")
exit()
if len(dev_list) == 0:
rospy.logerr("No device found, exiting")
exit()
rospy.logwarn(dng_list)
rospy.logwarn(dev_list)
d = dng_list[0]
wc = None
att = 0
while wc is None:
if att > 10:
exit()
att += 1
#!/usr/bin/env python
import serial
import time
import math
import string
import rospy
import threespace_api as tsa
import find_dng
from threespace_api import *
# CALIBRATES ALL CONNECTED IMU
rospy.init_node("calibration")
dng_list = find_dng.returnDev("dng")
for d in dng_list:
tsa.TSDongle.broadcastSynchronizationPulse(d)
for device in d:
if device == None:
rospy.logerr("No Sensor Found")
else:
att = 0
res = False
rospy.logwarn("Calibrating %s", str(device))
while res == False:
att += 1
if att > 10:
break
try:
res = tsa.TSWLSensor.beginGyroscopeAutoCalibration(device)
rospy.logwarn("Calibration Successful")
except:
import time
import math
import string
import glob
import sensor_table
import find_dng
import threespace_api as tsa
from threespace_api import *
from socket import *
from threespace_ros.msg import dataVec
from gait_hmm_ros.msg import imu_vector
rospy.init_node("broadcasterV2")
r = rospy.Rate(100)
suffix = "_data_vec"
dongle_list = find_dng.returnDev("dng")
publishers = {}
dv_publishers = {}
broadcasters = {}
frame_list = []
publishers_list = []
if len(dongle_list) == 0:
rospy.logerr("No Dongles Found, Exiting")
exit()
for d in dongle_list:
tsa.TSDongle.broadcastSynchronizationPulse(d)
for device in d:
if device is None:
def __init__(self):
rospy.init_node("publisher")
r = rospy.Rate(100)
suffix = "_data_vec"
dongle_list = find_dng.returnDev("dng")
publishers = {}
dv_publishers = {}
broadcasters = {}
if len(dongle_list) == 0:
rospy.logerr("No dongles found, exiting")
exit()
for d in dongle_list:
tsa.TSDongle.broadcastSynchronizationPulse(d)
for device in d:
if device is None:
rospy.logerr("No Sensor Found")
else:
id_ = str(device)
id_ = id_[id_.find('W'):-1]
rospy.logerr(id_)
frame = sensor_table.sensor_table.get(id_)
rospy.logwarn("Adding publisher for %s : %s", id_, frame)
rospy.logwarn(
"Battery at %s Percent ",
tsa.TSWLSensor.getBatteryPercentRemaining(device))
br = tf2_ros.TransformBroadcaster()
dv_pub = rospy.Publisher(frame + suffix,
dataVec,
queue_size=100)
broadcasters[frame] = br
dv_publishers[frame] = dv_pub
tsa.TSWLSensor.setStreamingSlots(
device,
slot0='getTaredOrientationAsQuaternion',
slot1='getAllCorrectedComponentSensorData')
t = geometry_msgs.msg.TransformStamped()
dv = dataVec()
dev_list = []
for d in dongle_list:
for dev in d:
if dev is not None:
dev_list.append(dev)
while not rospy.is_shutdown():
for device in dev_list:
if device is not None:
batch = tsa.TSWLSensor.getStreamingBatch(device)
if batch is not None:
quat = batch[0:4]
full = batch[4:]
id_ = str(device)
id_ = id_[id_.find('W'):-1]
frame = sensor_table.sensor_table.get(id_)
dp = dv_publishers.get(frame)
b = broadcasters.get(frame)
t.header.stamp = rospy.Time.now()
t.header.frame_id = "world"
t.child_frame_id = frame
t.transform.translation.x = 0.0
t.transform.translation.y = 0.0
t.transform.translation.z = 0.0
dv.header.stamp = rospy.get_rostime()
dv.quat.quaternion.x = -quat[2]
dv.quat.quaternion.y = quat[0]
dv.quat.quaternion.z = -quat[1]
dv.quat.quaternion.w = quat[3]
dv.gyroX = full[0]
dv.gyroY = full[1]
dv.gyroZ = full[2]
dv.accX = full[3]
dv.accY = full[4]
dv.accZ = full[5]
dv.comX = full[6]
dv.comY = full[7]
dv.comZ = full[8]
t.transform.rotation = dv.quat.quaternion
b.sendTransform(t)
dp.publish(dv)
else:
# rospy.logerr("None")
pass
r.sleep()
for d in dongle_list:
tsa.TSDongle.close(d)
print(publishers)
print(dv_publishers)
print(broadcasters)
def __init__(self):
rospy.init_node("publisher")
r = rospy.Rate(100)
suffix = "_data_vec"
dongle_list = find_dng.returnDev("dng")
publishers = {}
dv_publishers = {}
broadcasters = {}
joints = {}
upper_topic = rospy.get_param('upper', 'none')
rospy.logerr(upper_topic)
if upper_topic == 'none':
rospy.logerr('No topic for upper joint found, exiting')
else:
joints[upper_topic] = ThreeJoint('upper', upper_topic, 'world',
0.30)
lower_topic = rospy.get_param('lower', 'none')
rospy.logerr(lower_topic)
if lower_topic == 'none':
rospy.logerr('No topic for lower joint found')
else:
joints[lower_topic] = ThreeJoint(
'lower', lower_topic,
joints.get(upper_topic).name + '2', 0.30)
hand_topic = rospy.get_param('hand', 'none')
rospy.logerr(hand_topic)
if hand_topic == 'none':
rospy.logerr('No topic for hand joint found')
else:
joints[hand_topic] = ThreeJoint(
'hand', hand_topic,
joints.get(lower_topic).name + '2', 0.05)
if len(dongle_list) == 0:
rospy.logerr("No dongles found, exiting")
exit()
for d in dongle_list:
tsa.TSDongle.broadcastSynchronizationPulse(d)
for device in d:
if device is None:
rospy.logerr("No Sensor Found")
else:
id_ = str(device)
id_ = id_[id_.find('W'):-1]
rospy.logerr(id_)
frame = sensor_table.sensor_table.get(id_)
rospy.logwarn("Adding publisher for %s : %s", id_, frame)
rospy.logwarn(
"Battery at %s Percent ",
tsa.TSWLSensor.getBatteryPercentRemaining(device))
br = tf2_ros.TransformBroadcaster()
dv_pub = rospy.Publisher(frame + suffix,
dataVec,
queue_size=100)
broadcasters[frame] = br
dv_publishers[frame] = dv_pub
tsa.TSWLSensor.setStreamingSlots(
device,
slot0='getTaredOrientationAsQuaternion',
slot1='getAllCorrectedComponentSensorData')
t = geometry_msgs.msg.TransformStamped()
t2 = geometry_msgs.msg.TransformStamped()
dv = dataVec()
dev_list = []
for d in dongle_list:
for dev in d:
if dev is not None:
dev_list.append(dev)
while not rospy.is_shutdown():
for device in dev_list:
if device is not None:
batch = tsa.TSWLSensor.getStreamingBatch(device)
if batch is not None:
frame = sensor_table.sensor_table.get(id_)
quat = batch[0:4]
full = batch[4:]
id_ = str(device)
id_ = id_[id_.find('W'):-1]
dp = dv_publishers.get(frame)
dv.header.stamp = rospy.get_rostime()
dv.quat.quaternion.x = -quat[2]
dv.quat.quaternion.y = quat[0]
dv.quat.quaternion.z = -quat[1]
dv.quat.quaternion.w = quat[3]
t.transform.rotation = dv.quat.quaternion
dv.gyroX = full[0]
dv.gyroY = full[1]
dv.gyroZ = full[2]
dv.accX = full[3]
dv.accY = full[4]
dv.accZ = full[5]
dv.comX = full[6]
dv.comY = full[7]
dv.comZ = full[8]
t.header.stamp = t2.header.stamp = t2.header.stamprospy.Time.now(
)
t.header.frame_id = t2.header.frame_id = joints.get(
frame).parent
t.child_frame_id = joints.get(frame).name
t2.child_frame_id = joints.get(frame).name + '2'
(roll, pitch,
yaw) = tf.transformations.euler_from_quaternion([
t.transform.rotation.x, t.transform.rotation.y,
t.transform.rotation.z, t.transform.rotation.w
])
if yaw == 0.0 or pitch == 0.0 or roll == 0.0:
continue
if joints.get(frame).set is False:
joints.get(frame).initYaw = yaw
joints.get(frame).intPitch = pitch
joints.get(frame).initRoll = roll
rospy.logerr([
joints.get(frame).initYaw,
joints.get(frame).initPitch,
joints.get(frame).initRoll
])
joints.get(frame).set = True
if abs(
joints.get(frame).roll - roll -
joints.get(frame).initRoll) > 0.01:
joints.get(
frame).roll = roll - joints.get(frame).initRoll
if abs(
joints.get(frame).pitch - pitch -
joints.get(frame).initPitch) > 0.01:
joints.get(frame).pitch = pitch - joints.get(
frame).initPitch
if abs(
joints.get(frame).yaw - yaw -
joints.get(frame).initYaw) > 0.01:
joints.get(
frame).yaw = yaw - joints.get(frame).initYaw
q = tf.transformations.quaternion_from_euler(
joints.get(frame).roll,
joints.get(frame).pitch,
joints.get(frame).yaw)
msg = geometry_msgs.msg.Quaternion(
q[0], q[1], q[2], q[3])
eq = tf.transformations.quaternion_from_euler(0, 0, 0)
empty = geometry_msgs.msg.Quaternion(
eq[0], eq[1], eq[2], eq[3])
t.transform.rotation = msg
t2.transform.rotation = empty
joints.get(frame).x = joints.get(frame).radius * math.cos(joints.get(frame).yaw) * \
math.cos(joints.get(frame).pitch)
joints.get(frame).y = joints.get(frame).radius * math.cos(joints.get(frame).pitch) * \
math.sin(joints.get(frame).yaw)
joints.get(
frame).z = -joints.get(frame).radius * math.sin(
joints.get(frame).pitch)
t.transform.translation.x = joints.get(frame).x
t.transform.translation.y = joints.get(frame).y
t.transform.translation.z = joints.get(frame).z
t2.transform.translation.x = joints.get(frame).x * 2
t2.transform.translation.y = joints.get(frame).y * 2
t2.transform.translation.z = joints.get(frame).z * 2
br = tf2_ros.TransformBroadcaster()
br2 = tf2_ros.TransformBroadcaster()
br.sendTransform(t)
br2.sendTransform(t2)
dp.publish(dv)
else:
pass
r.sleep()
for d in dongle_list:
tsa.TSDongle.close(d)
print(publishers)
print(dv_publishers)
print(broadcasters)
exit()
import serial
import time
import math
import string
import glob
import sensor_table
import find_dng
import threespace as tsa
from threespace import *
from socket import *
from threespace_ros.msg import dataVec
rospy.init_node("publisher")
r = rospy.Rate(100)
suffix = "_data_vec"
dongle_list = find_dng.returnDev("dng")
publishers = {}
dv_publishers = {}
broadcasters = {}
frame_list = []
publishers_list = []
if len(dongle_list) == 0:
rospy.logerr("No Dongles Found, Exiting")
exit()
for d in dongle_list:
tsa.TSDongle.broadcastSynchronizationPulse(d)
for device in d:
if device is None:
def __init__(self):
rospy.init_node("publisher")
r = rospy.Rate(100)
suffix = "_data_vec"
dongle_list = find_dng.returnDev("dng")
publishers = {}
dv_publishers = {}
broadcasters = {}
if len(dongle_list) == 0:
rospy.logerr("No dongles found, exiting")
exit()
for d in dongle_list:
tsa.TSDongle.broadcastSynchronizationPulse(d)
for device in d:
if device is None:
rospy.logerr("No Sensor Found")
else:
id_ = str(device)
id_ = id_[id_.find('W'):-1]
rospy.logerr(id_)
frame = sensor_table.sensor_table.get(id_)
rospy.logwarn("Adding publisher for %s : %s", id_, frame)
rospy.logwarn("Battery at %s Percent ", tsa.TSWLSensor.getBatteryPercentRemaining(device))
br = tf2_ros.TransformBroadcaster()
dv_pub = rospy.Publisher(frame + suffix, dataVec, queue_size=100)
broadcasters[frame] = br
dv_publishers[frame] = dv_pub
tsa.TSWLSensor.setStreamingSlots(device, slot0='getTaredOrientationAsQuaternion',
slot1='getAllCorrectedComponentSensorData')
t = geometry_msgs.msg.TransformStamped()
dv = dataVec()
dev_list = []
for d in dongle_list:
for dev in d:
if dev is not None:
dev_list.append(dev)
while not rospy.is_shutdown():
for device in dev_list:
if device is not None:
batch = tsa.TSWLSensor.getStreamingBatch(device)
if batch is not None:
quat = batch[0:4]
full = batch[4:]
id_ = str(device)
id_ = id_[id_.find('W'):-1]
frame = sensor_table.sensor_table.get(id_)
dp = dv_publishers.get(frame)
b = broadcasters.get(frame)
t.header.stamp = rospy.Time.now()
t.header.frame_id = "world"
t.child_frame_id = frame
t.transform.translation.x = 0.0
t.transform.translation.y = 0.0
t.transform.translation.z = 0.0
dv.header.stamp = rospy.get_rostime()
dv.quat.quaternion.x = -quat[2]
dv.quat.quaternion.y = quat[0]
dv.quat.quaternion.z = -quat[1]
dv.quat.quaternion.w = quat[3]
dv.gyroX = full[0]
dv.gyroY = full[1]
dv.gyroZ = full[2]
dv.accX = full[3]
dv.accY = full[4]
dv.accZ = full[5]
dv.comX = full[6]
dv.comY = full[7]
dv.comZ = full[8]
t.transform.rotation = dv.quat.quaternion
b.sendTransform(t)
dp.publish(dv)
else:
# rospy.logerr("None")
pass
r.sleep()
for d in dongle_list:
tsa.TSDongle.close(d)
print (publishers)
print (dv_publishers)
print (broadcasters)