extra/usb_power/convert_power_log_board.py

#!/usr/bin/env python2
# Copyright 2018 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""
Program to convert sweetberry config to servod config template.
"""

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)