Source code for sensors.xsens
'''
Import data saved with XSens-sensors
'''
'''
Author: Thomas Haslwanter
Version: 0.1
Date: May-2016
'''
import numpy as np
import pandas as pd
[docs]def get_data(in_file):
'''Get the sampling rate, as well as the recorded data.
Parameters
----------
in_file : string
Filename of the data-file
Returns
-------
out_list: list
Contains the following parameters:
- rate
- acceleration
- angular_velocity
- mag_field_direction
'''
# Get the sampling rate from the second line in the file
try:
fh = open(in_file)
fh.readline()
line = fh.readline()
rate = np.float(line.split(':')[1].split('H')[0])
fh.close()
returnValues = [rate]
except FileNotFoundError:
print('{0} does not exist!'.format(in_file))
return -1
# Read the data
data = pd.read_csv(in_file,
sep='\t',
skiprows=4,
index_col=False)
# Extract the columns that you want, by name
paramList=['Acc', 'Gyr', 'Mag']
for param in paramList:
Expression = param + '*'
returnValues.append(data.filter(regex=Expression).values)
return returnValues