2019-04-23 17:05:21 +00:00
|
|
|
#!/usr/bin/python2
|
|
|
|
#
|
|
|
|
# Copyright 2019 Google Inc.
|
|
|
|
#
|
|
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
|
|
# found in the LICENSE file.
|
|
|
|
|
2019-07-26 16:55:40 +00:00
|
|
|
import StringIO
|
|
|
|
import argparse
|
2019-04-23 17:05:21 +00:00
|
|
|
import os
|
2019-07-26 16:55:40 +00:00
|
|
|
import sys
|
|
|
|
|
|
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument('-n', '--dry-run', action='store_true',
|
|
|
|
help='Just check there is nothing to rewrite.')
|
|
|
|
parser.add_argument('sources', nargs='*',
|
|
|
|
help='Source files to rewrite, or all if empty.')
|
|
|
|
args = parser.parse_args()
|
2019-04-23 17:05:21 +00:00
|
|
|
|
|
|
|
roots = [
|
|
|
|
'bench',
|
|
|
|
'dm',
|
|
|
|
'docs',
|
|
|
|
'example',
|
|
|
|
'experimental',
|
|
|
|
'fuzz',
|
|
|
|
'gm',
|
|
|
|
'include',
|
|
|
|
'modules',
|
|
|
|
'platform_tools/android/apps',
|
|
|
|
'samplecode',
|
|
|
|
'src',
|
|
|
|
'tests',
|
2019-04-24 17:46:06 +00:00
|
|
|
'third_party/etc1',
|
2019-04-23 17:05:21 +00:00
|
|
|
'third_party/gif',
|
|
|
|
'tools'
|
|
|
|
]
|
|
|
|
|
2019-07-25 16:18:06 +00:00
|
|
|
# Don't count our local Vulkan headers as Skia headers;
|
|
|
|
# we don't want #include <vulkan/vulkan_foo.h> rewritten to point to them.
|
|
|
|
blacklist = ['include/third_party/vulkan']
|
2019-07-25 14:00:52 +00:00
|
|
|
|
2019-04-23 17:05:21 +00:00
|
|
|
# Map short name -> absolute path for all Skia headers.
|
|
|
|
headers = {}
|
|
|
|
for root in roots:
|
|
|
|
for path, _, files in os.walk(root):
|
2019-07-25 16:18:06 +00:00
|
|
|
if not any(snippet in path for snippet in blacklist):
|
|
|
|
for file_name in files:
|
|
|
|
if file_name.endswith('.h'):
|
|
|
|
if file_name in headers:
|
|
|
|
print path, file_name, headers[file_name]
|
|
|
|
assert file_name not in headers
|
|
|
|
headers[file_name] = os.path.abspath(os.path.join(path, file_name))
|
2019-04-23 17:05:21 +00:00
|
|
|
|
2019-07-26 16:55:40 +00:00
|
|
|
def to_rewrite():
|
|
|
|
if args.sources:
|
|
|
|
for path in args.sources:
|
|
|
|
yield path
|
|
|
|
else:
|
|
|
|
for root in roots:
|
|
|
|
for path, _, files in os.walk(root):
|
|
|
|
for file_name in files:
|
|
|
|
yield os.path.join(path, file_name)
|
|
|
|
|
2019-04-23 17:05:21 +00:00
|
|
|
# Rewrite any #includes relative to Skia's top-level directory.
|
2019-07-26 16:55:40 +00:00
|
|
|
rc = 0
|
|
|
|
for file_path in to_rewrite():
|
|
|
|
if 'generated' in file_path:
|
|
|
|
continue
|
|
|
|
if (file_path.endswith('.h') or
|
|
|
|
file_path.endswith('.c') or
|
|
|
|
file_path.endswith('.m') or
|
|
|
|
file_path.endswith('.mm') or
|
|
|
|
file_path.endswith('.inc') or
|
|
|
|
file_path.endswith('.fp') or
|
|
|
|
file_path.endswith('.cc') or
|
|
|
|
file_path.endswith('.cpp')):
|
|
|
|
# Read the whole file into memory.
|
|
|
|
lines = open(file_path).readlines()
|
2019-04-23 17:05:21 +00:00
|
|
|
|
2019-07-26 16:55:40 +00:00
|
|
|
# Write it back out again line by line with substitutions for #includes.
|
|
|
|
output = StringIO.StringIO() if args.dry_run else open(file_path, 'w')
|
2019-04-23 17:05:21 +00:00
|
|
|
|
2019-07-26 16:55:40 +00:00
|
|
|
includes = []
|
|
|
|
for line in lines:
|
|
|
|
parts = line.replace('<', '"').replace('>', '"').split('"')
|
|
|
|
if (len(parts) == 3
|
|
|
|
and '#' in parts[0]
|
|
|
|
and 'include' in parts[0]
|
|
|
|
and os.path.basename(parts[1]) in headers):
|
|
|
|
header = headers[os.path.basename(parts[1])]
|
|
|
|
includes.append(parts[0] +
|
|
|
|
'"%s"' % os.path.relpath(header, '.') +
|
|
|
|
parts[2])
|
|
|
|
else:
|
|
|
|
for inc in sorted(includes):
|
|
|
|
print >>output, inc.strip('\n')
|
|
|
|
includes = []
|
|
|
|
print >>output, line.strip('\n')
|
2019-04-23 17:05:21 +00:00
|
|
|
|
2019-07-26 16:55:40 +00:00
|
|
|
if args.dry_run and output.getvalue() != open(file_path).read():
|
|
|
|
print file_path, 'has #includes that need rewriting.'
|
|
|
|
rc = 1
|
|
|
|
output.close()
|
|
|
|
sys.exit(rc)
|