1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
#!/usr/bin/env python
# Copyright 2018 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
#
# Ignore indention messages, since legacy scripts use 2 spaces instead of 4.
# pylint: disable=bad-indentation,docstring-section-indent
# pylint: disable=docstring-trailing-quotes
"""
Program to convert sweetberry config to servod config template.
"""
# Note: This is a py2/3 compatible file.
from __future__ import print_function
import json
import os
import sys
from powerlog import Spower
def fetch_records(board_file):
"""Import records from servo_ina file.
board files are json files, and have a list of tuples with
the INA data.
(name, rs, swetberry_num, net_name, channel)
Args:
board_file: board file
Returns:
list of tuples as described above.
"""
data = None
with open(board_file) as f:
data = json.load(f)
return data
def write_to_file(file, sweetberry, inas):
"""Writes records of |sweetberry| to |file|
Args:
file: file to write to.
sweetberry: sweetberry type. A or B.
inas: list of inas read from board file.
"""
with open(file, 'w') as pyfile:
pyfile.write('inas = [\n')
for rec in inas:
if rec['sweetberry'] != sweetberry:
continue
# EX : ('sweetberry', 0x40, 'SB_FW_CAM_2P8', 5.0, 1.000, 3, False),
channel, i2c_addr = Spower.CHMAP[rec['channel']]
record = (" ('sweetberry', 0x%02x, '%s', 5.0, %f, %d, 'True')"
",\n" % (i2c_addr, rec['name'], rec['rs'], channel))
pyfile.write(record)
pyfile.write(']\n')
def main(argv):
if len(argv) != 2:
print("usage:")
print(" %s input.board" % argv[0])
return
inputf = argv[1]
basename = os.path.splitext(inputf)[0]
inas = fetch_records(inputf)
sweetberry = set(rec['sweetberry'] for rec in inas)
if len(sweetberry) == 2:
print("Converting %s to %s and %s" % (inputf, basename + '_a.py',
basename + '_b.py'))
write_to_file(basename + '_a.py', 'A', inas)
write_to_file(basename + '_b.py', 'B', inas)
else:
print("Converting %s to %s" % (inputf, basename + '.py'))
write_to_file(basename + '.py', sweetberry.pop(), inas)
if __name__ == "__main__":
main(sys.argv)
|
