blob: e07bd4ca6f78b8f11267caa36cc80e81de683a3c [file] [log] [blame]
# Copyright (c) 2012 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import itertools
import optparse
import re
VENDOR_PATTERN = re.compile("^(?P<id>[0-9a-fA-F]{4})\s+(?P<name>.+)$")
PRODUCT_PATTERN = re.compile("^\t(?P<id>[0-9a-fA-F]{4})\s+(?P<name>.+)$")
def EscapeName(name):
name = name.replace("\\", "\\\\")
name = name.replace("\"", "\\\"")
name = name.replace("?", "\?")
return name
def ParseTable(input_path):
input_file = open(input_path, "r")
input = input_file.read().split("\n")
input_file.close()
table = {}
vendor = None
for line in input:
vendor_match = VENDOR_PATTERN.match(line)
if vendor_match:
if vendor:
table[vendor["id"]] = vendor
vendor = {}
vendor["id"] = int(vendor_match.group("id"), 16)
vendor["name"] = vendor_match.group("name")
vendor["products"] = []
continue
product_match = PRODUCT_PATTERN.match(line)
if product_match:
if not vendor:
raise Exception("Product seems to appear before vendor.")
product = {}
product["id"] = int(product_match.group("id"), 16)
product["name"] = product_match.group("name")
vendor["products"].append(product)
return table
def GenerateDeviceDefinitions(table):
output = ""
for vendor_id in sorted(table.keys()):
vendor = table[vendor_id]
if len(vendor["products"]) == 0:
continue
output += "static const UsbProduct vendor_%.4x_products[] = {\n" % \
vendor["id"]
for product in vendor["products"]:
output += " {0x%.4x, \"%s\"},\n" % (product["id"],
EscapeName(product["name"]))
output += "};\n"
return output
def GenerateVendorDefinitions(table):
output = "const size_t UsbIds::vendor_size_ = %d;\n" % len(table.keys())
output += "const UsbVendor UsbIds::vendors_[] = {\n"
for vendor_id in sorted(table.keys()):
vendor = table[vendor_id]
product_table = "NULL"
if len(vendor["products"]) != 0:
product_table = "vendor_%.4x_products" % (vendor["id"])
output += " {0x%.4x, \"%s\", %d, %s},\n" % (vendor["id"],
EscapeName(vendor["name"]), len(vendor["products"]), product_table)
output += "};\n"
return output
if __name__ == "__main__":
parser = optparse.OptionParser(
description="Generates a C++ USB ID lookup table.")
parser.add_option("-i", "--input", help="Path to usb.ids")
parser.add_option("-o", "--output", help="Output file path")
(opts, args) = parser.parse_args()
table = ParseTable(opts.input)
output = """// Generated from %s
#ifndef GENERATED_USB_IDS_H_
#define GENERATED_USB_IDS_H_
#include "device/usb/usb_ids.h"
namespace device {
""" % (opts.input)
output += GenerateDeviceDefinitions(table)
output += GenerateVendorDefinitions(table)
output += """
} // namespace device
#endif // GENERATED_USB_IDS_H_
"""
output_file = open(opts.output, "w+")
output_file.write(output)
output_file.close()