Skip to content

Commit

Permalink
LP-156 Added update script in python.
Browse files Browse the repository at this point in the history
  • Loading branch information
m-thread committed Oct 15, 2015
1 parent aa209db commit d8f4945
Showing 1 changed file with 62 additions and 0 deletions.
62 changes: 62 additions & 0 deletions fix_vehicle_templates.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,62 @@
#!/usr/bin/python

###############################################################################
#
# The LibrePilot Project, http://www.librepilot.org Copyright (C) 2015.
# Script to update vehicle templates files to match current UAVO structure
# and data.
#
###############################################################################

import json
import re
import collections
import fnmatch
import os

class DecimalEncoder(json.JSONEncoder):
def default(self, obj):
if isinstance(obj, Decimal):
return float(obj)
return json.JSONEncoder.default(self, obj)

json.encoder.FLOAT_REPR = lambda o: format(o, '.17g')

files = []
for root, dirnames, filenames in os.walk('ground/gcs/src/share/vehicletemplates/'):
for filename in fnmatch.filter(filenames, '*.optmpl'):
files.append(os.path.join(root, filename))

for f in files:
data = json.load(open(f, 'r'), object_pairs_hook=collections.OrderedDict)

for item in data['objects']:
fieldsToRemove = []
for i in item['fields']:
name = i['name']
values = i['values']
if re.compile('ThrustPIDScaleCurve').match(name):
i['type'] = "int8"
for j in values:
j['value'] = int(j['value'] * 100)
elif re.compile('AcroInsanityFactor').match(name):
i['type'] = "int8"
value = 0
for j in values:
value = int(j['value'] * 100)
values.pop()
values.append({'name': 'roll', 'value': value})
values.append({'name': 'pitch', 'value': value})
values.append({'name': 'yaw', 'value': value})
elif re.compile('FeedForward').match(name):
fieldsToRemove.append(i)
elif re.compile('MaxAccel').match(name):
fieldsToRemove.append(i)
elif re.compile('AccelTime').match(name):
fieldsToRemove.append(i)
elif re.compile('DecelTime').match(name):
fieldsToRemove.append(i)
for field in fieldsToRemove:
item['fields'].remove(field)
with open(f, 'w') as outfile:
json.dump(data, outfile, indent=4, separators=(',', ': '), cls=DecimalEncoder)

0 comments on commit d8f4945

Please sign in to comment.