def bge_get_location_from_placename(name):
"""Use the coil's place string to get its coordinates as displayed onscreen."""
# get the index of the coil
import scripts.ema_blender.coil_info as ci
ind = ci.find_sensor_index(name)
# get the location of the appropriately-named cube
import scripts.ema_blender.ema_bge.bge_standard_gamefns as sgf
scene, objs, *_ = sgf.get_scene_info()
cubeobj = objs.get('CoilCube{}'.format(str(ind).zfill(2)), False)
if cubeobj:
return cubeobj.worldPosition
else:
return None
def bge_get_location_from_placename(name):
"""Use the coil's place string to get its coordinates as displayed onscreen."""
# get the index of the coil
import scripts.ema_blender.coil_info as ci
ind = ci.find_sensor_index(name)
# get the location of the appropriately-named cube
import scripts.ema_blender.ema_bge.bge_standard_gamefns as sgf
scene, objs, *_ = sgf.get_scene_info()
cubeobj = objs.get('CoilCube{}'.format(str(ind).zfill(2)), False)
if cubeobj:
return cubeobj.worldPosition
else:
return None
def get_obj_from_placename(name):
"""Get the bge object from the placename"""
# get the location of the appropriately-named cube
import scripts.ema_blender.ema_bge.bge_standard_gamefns as sgf
scene, objs, *_ = sgf.get_scene_info()
literal_obj = objs.get(name, False)
# scene has an object called name
if literal_obj:
return literal_obj
import scripts.ema_blender.coil_info as ci
ind = ci.find_sensor_index(name)
cubeobj = objs.get(bsh.ema_mesh_name_rule(ind), False)
if cubeobj:
return cubeobj
else:
return None
def get_obj_from_placename(name):
"""Get the bge object from the placename"""
# get the location of the appropriately-named cube
import scripts.ema_blender.ema_bge.bge_standard_gamefns as sgf
scene, objs, *_ = sgf.get_scene_info()
literal_obj = objs.get(name, False)
# scene has an object called name
if literal_obj:
return literal_obj
import scripts.ema_blender.coil_info as ci
ind = ci.find_sensor_index(name)
cubeobj = objs.get(bsh.ema_mesh_name_rule(ind), False)
if cubeobj:
return cubeobj
else:
return None
def get_obj_from_placename(name):
"""Get the bge object from the placename"""
scene = bpy.context.scene
literal_obj = scene.objects.get(name, False)
if literal_obj:
return literal_obj
# no object with that name, must look in sensor_index file
import scripts.ema_blender.coil_info as ci
ind = ci.find_sensor_index(name)
# get the location of the appropriately-named cube
cubeobj = scene.objects.get(bsh.ema_mesh_name_rule(ind), False)
if cubeobj:
return cubeobj
else:
return None
def gs_record_biteplate_now(self):
print("Performing a biteplate recording, current values:",
self.bp_in_rs, self.rp_in_gs)
"""read the biteplate into BitePlane class - create mapping from global space to biteplate space"""
response = self.inform_recording_start()
# if streaming ready, stream biteplate recording for 5 seconds
if response == 1:
print('NOW DOING HEAD-CORRECTION')
# get information about which sensors are head-correction and which are biteplate
active, biteplate, reference = ci.get_sensor_roles_no_blender()
print('these are sensor roles', active, biteplate, reference)
server.repl._stop_b = True # stop the normal behaviour of the data queue
print('replies queue lives', server.repl._b.is_alive())
# start streaming, get either 1 or (in properties) defined seconds of streaming data
server.gs_start_streaming()
print('started streaming')
time.sleep(2 if pps.head_correction_time is None else pps.
head_correction_time)
# stop streaming
server.gs_stop_streaming()
print('stopped streaming')
time.sleep(1)
# access all the streamed data, empty the queue, saved elements in replies
print('qsize is ', server.repl._q34.qsize())
all_streamed = []
while not server.repl._q34.empty():
streamed_df = DataFrame(rawdf=server.repl._q34.get()[2])
print('streamed df was', streamed_df)
all_streamed.append(streamed_df)
server.repl.latest_df = None
server.repl.last_x_dfs.clear()
print('\n\nstreamed data looks like', all_streamed[:1])
server.repl._stop_b = False # restart the normal behaviour of streamed data
last_frames = dm.remove_first_ms_of_list(
all_streamed,
ms=1 if pps.head_correction_exclude_first_ms is None else
pps.head_correction_exclude_first_ms)
no_outliers = dm.remove_outliers(last_frames)
average_bp = DataFrame(fromlist=no_outliers)
print(average_bp)
# prepare the transformations
# isolate the coil objects
refcoils = [
average_bp.give_coils()[reference[x][0]]
for x in range(len(reference))
]
bpcoils = [
average_bp.give_coils()[biteplate[x][0]]
for x in range(len(biteplate))
]
lind, rind, find = [
ci.find_sensor_index(n) for n in ['BP1', 'BP2', 'BP3']
] #todo: check order
# create a coordinate-system based on reference sensors (fully-defined here)
rp_in_gs = ReferencePlane(*[x.abs_loc for x in refcoils[:3]])
for c in bpcoils:
c.ref_loc = rp_in_gs.project_to_lcs(c.abs_loc)
# create the bp_in_rs (origin still to be determined)
print('biteplate coils are:', lind, rind, find,
*[average_bp.give_coils()[x] for x in [lind, rind, find]])
bp_in_rs = BitePlane(*[
average_bp.give_coils()[x].ref_loc for x in [lind, rind, find]
])
pickle.dump(
bp_in_rs,
open(
os.path.normpath(os.getcwd() + os.path.sep +
pps.biteplate_cs_storage), 'wb'))
pickle.dump(
rp_in_gs,
open(
os.path.normpath(os.getcwd() + os.path.sep +
pps.refspace_cs_storage), 'wb'))
# now rp_in_gs and bp_in_gs can be used to add attributes to the dataframes
server.repl.print_tsv = cl_args.print
print('correcting matrices are:',
rp_in_gs.give_local_to_global_mat(),
bp_in_rs.give_global_to_local_mat())
if os.name == 'nt':
import ctypes
notification = ctypes.windll.user32.MessageBoxA(
0,
b"Set up your wave/server for streaming of active sensor data",
b"Biteplate recording finished", 0)
else: # no popup yet for linux/mac
print('NOW PREPARE FOR NORMAL STREAMING')
return bp_in_rs, rp_in_gs
else: # presses cancel recording biteplate do nothing.
return None, None
def gs_record_biteplate_now(self):
print("Performing a biteplate recording, current values:", self.bp_in_rs, self.rp_in_gs)
"""read the biteplate into BitePlane class - create mapping from global space to biteplate space"""
response = self.inform_recording_start()
# if streaming ready, stream biteplate recording for 5 seconds
if response == 1:
print('NOW DOING HEAD-CORRECTION')
# get information about which sensors are head-correction and which are biteplate
active, biteplate, reference = ci.get_sensor_roles_no_blender()
print('these are sensor roles', active, biteplate, reference)
server.repl._stop_b = True # stop the normal behaviour of the data queue
print('replies queue lives', server.repl._b.is_alive())
# start streaming, get either 1 or (in properties) defined seconds of streaming data
server.gs_start_streaming()
print('started streaming')
time.sleep(2 if pps.head_correction_time is None else pps.head_correction_time)
# stop streaming
server.gs_stop_streaming()
print('stopped streaming')
time.sleep(1)
# access all the streamed data, empty the queue, saved elements in replies
print('qsize is ',server.repl._q34.qsize())
all_streamed = []
while not server.repl._q34.empty():
streamed_df = DataFrame(rawdf=server.repl._q34.get()[2])
print('streamed df was', streamed_df)
all_streamed.append(streamed_df)
server.repl.latest_df = None
server.repl.last_x_dfs.clear()
print('\n\nstreamed data looks like', all_streamed[:1])
server.repl._stop_b = False # restart the normal behaviour of streamed data
last_frames = dm.remove_first_ms_of_list(all_streamed,
ms=1 if pps.head_correction_exclude_first_ms is None
else pps.head_correction_exclude_first_ms)
no_outliers= dm.remove_outliers(last_frames)
average_bp = DataFrame(fromlist=no_outliers)
print(average_bp)
# prepare the transformations
# isolate the coil objects
refcoils = [average_bp.give_coils()[reference[x][0]] for x in range(len(reference))]
bpcoils = [average_bp.give_coils()[biteplate[x][0]] for x in range(len(biteplate))]
lind, rind, find = [ci.find_sensor_index(n) for n in ['BP1', 'BP2', 'BP3']] #todo: check order
# create a coordinate-system based on reference sensors (fully-defined here)
rp_in_gs = ReferencePlane(*[x.abs_loc for x in refcoils[:3]])
for c in bpcoils:
c.ref_loc = rp_in_gs.project_to_lcs(c.abs_loc)
# create the bp_in_rs (origin still to be determined)
print('biteplate coils are:', lind, rind, find,*[average_bp.give_coils()[x] for x in [lind, rind, find]])
bp_in_rs = BitePlane(*[average_bp.give_coils()[x].ref_loc for x in [lind, rind, find]])
pickle.dump(bp_in_rs, open(os.path.normpath(os.getcwd() + os.path.sep + pps.biteplate_cs_storage), 'wb'))
pickle.dump(rp_in_gs, open(os.path.normpath(os.getcwd() + os.path.sep + pps.refspace_cs_storage), 'wb'))
# now rp_in_gs and bp_in_gs can be used to add attributes to the dataframes
server.repl.print_tsv = cl_args.print
print('correcting matrices are:', rp_in_gs.give_local_to_global_mat(), bp_in_rs.give_global_to_local_mat())
if os.name == 'nt':
import ctypes
notification = ctypes.windll.user32.MessageBoxA(0,
b"Set up your wave/server for streaming of active sensor data",
b"Biteplate recording finished",
0)
else: # no popup yet for linux/mac
print('NOW PREPARE FOR NORMAL STREAMING')
return bp_in_rs, rp_in_gs
else: # presses cancel recording biteplate do nothing.
return None, None