File: embed_networks_in_cpp.py

package info (click to toggle)
chromium 139.0.7258.127-1
  • links: PTS, VCS
  • area: main
  • in suites:
  • size: 6,122,068 kB
  • sloc: cpp: 35,100,771; ansic: 7,163,530; javascript: 4,103,002; python: 1,436,920; asm: 946,517; xml: 746,709; pascal: 187,653; perl: 88,691; sh: 88,436; objc: 79,953; sql: 51,488; cs: 44,583; fortran: 24,137; makefile: 22,147; tcl: 15,277; php: 13,980; yacc: 8,984; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (88 lines) | stat: -rwxr-xr-x 2,768 bytes parent folder | download | duplicates (11)
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
#!/usr/bin/env python
# Copyright 2015 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Embeds standalone JavaScript snippets in C++ code.

The script requires the devtools/front_end/toolbox/OverridesUI.js file from
WebKit that lists the preset network conditions to be passed in as the only
argument. The list of network conditions will be written to a C-style string to
be parsed with JSONReader.
"""

import optparse
import re
import subprocess
import sys

import cpp_source

UNLIMITED_THROUGHPUT = ('WebInspector.OverridesSupport'
                        '.NetworkThroughputUnlimitedValue')


def quotizeKeys(s, keys):
  """Returns the string s with each instance of each key wrapped in quotes.

  Args:
    s: a string containing keys that need to be wrapped in quotes.
    keys: an iterable of keys to be wrapped in quotes in the string.
  """
  for key in keys:
    s = re.sub('%s: ' % key, '"%s": ' % key, s)
  return s


def evaluateMultiplications(s):
  """Returns the string s with each bare multiplication evaluated.

  Since the source is JavaScript, which includes bare arithmetic, and the
  output must be JSON format, we must evaluate all expressions.

  Args:
    s: a string containing bare multiplications that need to be evaluated.
  """
  def evaluateBinaryMultiplication(match):
    return str(float(match.group(1)) * float(match.group(2)))

  return re.sub('([0-9\.]+) \* ([0-9\.]+)', evaluateBinaryMultiplication, s)


def main():
  parser = optparse.OptionParser()
  parser.add_option(
      '', '--directory', type='string', default='.',
      help='Path to directory where the cc/h files should be created')
  options, args = parser.parse_args()

  networks = '['
  file_name = args[0]
  inside_list = False
  with open(file_name, 'r', encoding='utf-8') as f:
    for line in f:
      if not inside_list:
        if 'WebInspector.OverridesUI._networkConditionsPresets = [' in line:
          inside_list = True
      else:
        if line.strip() == '];':
          inside_list = False
          continue
        line = line.replace(UNLIMITED_THROUGHPUT, "-1")
        networks += line.strip()

  output_dir = 'chrome/test/chromedriver/chrome'
  networks += ']'
  networks = quotizeKeys(networks, ['id', 'title', 'throughput', 'latency'])
  networks = evaluateMultiplications(networks)
  cpp_source.WriteSource('network_list',
                         output_dir,
                         options.directory, {'kNetworks': networks})

  clang_format = ['clang-format', '-i']
  subprocess.Popen(clang_format + ['%s/network_list.cc' % output_dir])
  subprocess.Popen(clang_format + ['%s/network_list.h' % output_dir])


if __name__ == '__main__':
  sys.exit(main())