#
# ISC License
#
# Copyright (c) 2016, Autonomous Vehicle Systems Lab, University of Colorado at Boulder
#
# Permission to use, copy, modify, and/or distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
#
from xml.etree import ElementTree
import numpy as np
[docs]
class SRPLookupTableHandler:
"""Class to handle an SRP Lookup table"""
def __init__(self):
self.sHatBLookup = np.zeros([1, 3])
self.forceBLookup = np.zeros([1, 3])
self.torqueBLookup = np.zeros([1, 3])
def parseAndLoadXML(self, filePath):
document = ElementTree.parse(filePath)
sHatBTree = document.find('sHatBValues')
forceBTree = document.find('forceBValues')
torqueBTree = document.find('torqueBValues')
self.sHatBLookup.resize([len(list(sHatBTree)), 3])
self.forceBLookup.resize([len(list(forceBTree)), 3])
self.torqueBLookup.resize([len(list(torqueBTree)), 3])
for node in list(sHatBTree):
idx = int(node.attrib['index'])
for value in list(node):
if value.tag == 'value_1':
self.sHatBLookup[idx, 0] = value.text
if value.tag == 'value_2':
self.sHatBLookup[idx, 1] = value.text
if value.tag == 'value_3':
self.sHatBLookup[idx, 2] = value.text
for node in list(forceBTree):
idx = int(node.attrib['index'])
for value in list(node):
if value.tag == 'value_1':
self.forceBLookup[idx, 0] = value.text
if value.tag == 'value_2':
self.forceBLookup[idx, 1] = value.text
if value.tag == 'value_3':
self.forceBLookup[idx, 2] = value.text
for node in list(torqueBTree):
idx = int(node.attrib['index'])
for value in list(node):
if value.tag == 'value_1':
self.torqueBLookup[idx, 0] = value.text
if value.tag == 'value_2':
self.torqueBLookup[idx, 1] = value.text
if value.tag == 'value_3':
self.torqueBLookup[idx, 2] = value.text