blob: d02bd906b700f2256c18e52c4a950ec3fdde407c [file] [log] [blame]
Mike Klein89eda8a2016-11-02 09:14:55 -04001#!/usr/bin/env python
2#
3# Copyright 2016 Google Inc.
4#
5# Use of this source code is governed by a BSD-style license that can be
6# found in the LICENSE file.
7
Ben Wagner9494f3e2018-05-25 15:48:31 -04008import collections
9import json
Mike Klein89eda8a2016-11-02 09:14:55 -040010import os
Ben Wagner9494f3e2018-05-25 15:48:31 -040011import subprocess
Mike Klein89eda8a2016-11-02 09:14:55 -040012import sys
13
Ben Wagner9494f3e2018-05-25 15:48:31 -040014# Finds all public sources in include directories then write them to skia.h.
Mike Klein89eda8a2016-11-02 09:14:55 -040015
Ben Wagner9494f3e2018-05-25 15:48:31 -040016# Also write skia.h.deps, which Ninja uses to track dependencies. It's the
Mike Klein89eda8a2016-11-02 09:14:55 -040017# very same mechanism Ninja uses to know which .h files affect which .cpp files.
18
Ben Wagner9494f3e2018-05-25 15:48:31 -040019gn = sys.argv[1]
20absolute_source = sys.argv[2]
21skia_h = sys.argv[3]
22include_dirs = sys.argv[4:]
Mike Klein89eda8a2016-11-02 09:14:55 -040023
Ben Wagner9494f3e2018-05-25 15:48:31 -040024absolute_source = os.path.normpath(absolute_source)
Mike Klein89eda8a2016-11-02 09:14:55 -040025
Ben Wagner9494f3e2018-05-25 15:48:31 -040026include_dirs = [os.path.join(os.path.normpath(include_dir), '')
27 for include_dir in include_dirs]
28include_dirs.sort(key=len, reverse=True)
Mike Klein89eda8a2016-11-02 09:14:55 -040029
Florin Malita8f480d92018-05-30 13:17:49 -040030gn_desc_cmd = [gn, 'desc', '.', '--root=%s' % absolute_source, '--format=json',
31 '*']
Ben Wagner9494f3e2018-05-25 15:48:31 -040032
Florin Malita8f480d92018-05-30 13:17:49 -040033desc_json_txt = ''
Ben Wagner9494f3e2018-05-25 15:48:31 -040034try:
Florin Malita8f480d92018-05-30 13:17:49 -040035 desc_json_txt = subprocess.check_output(gn_desc_cmd)
Ben Wagner9494f3e2018-05-25 15:48:31 -040036except subprocess.CalledProcessError as e:
37 print e.output
38 raise
39
Florin Malita8f480d92018-05-30 13:17:49 -040040desc_json = {}
Ben Wagner9494f3e2018-05-25 15:48:31 -040041try:
Florin Malita8f480d92018-05-30 13:17:49 -040042 desc_json = json.loads(desc_json_txt)
Ben Wagner9494f3e2018-05-25 15:48:31 -040043except ValueError:
Florin Malita8f480d92018-05-30 13:17:49 -040044 print desc_json_txt
Ben Wagner9494f3e2018-05-25 15:48:31 -040045 raise
46
Florin Malita8f480d92018-05-30 13:17:49 -040047sources = set()
48
49for target in desc_json.itervalues():
50 # We'll use `public` headers if they're listed, or pull them from `sources`
51 # if not. GN sneaks in a default "public": "*" into the JSON if you don't
52 # set one explicitly.
53 search_list = target.get('public')
54 if search_list == '*':
55 search_list = target.get('sources', [])
56
57 for name in search_list:
58 sources.add(os.path.join(absolute_source, os.path.normpath(name[2:])))
Ben Wagner9494f3e2018-05-25 15:48:31 -040059
60Header = collections.namedtuple('Header', ['absolute', 'include'])
61headers = {}
62for source in sources:
63 source_as_include = [source[len(include_dir):]
64 for include_dir in include_dirs
65 if source.startswith(include_dir)]
66 if not source_as_include:
67 continue
68 statinfo = os.stat(source)
69 key = str(statinfo.st_ino) + ':' + str(statinfo.st_dev)
70 # On Windows os.stat st_ino is 0 until 3.3.4 and st_dev is 0 until 3.4.0.
71 if key == '0:0':
72 key = source
73 include_path = source_as_include[0]
74 if key not in headers or len(include_path) < len(headers[key].include):
75 headers[key] = Header(source, include_path)
76
77headers = headers.values()
78headers.sort(key=lambda x: x.include)
79
80with open(skia_h, 'w') as f:
Mike Klein89eda8a2016-11-02 09:14:55 -040081 f.write('// skia.h generated by GN.\n')
82 f.write('#ifndef skia_h_DEFINED\n')
83 f.write('#define skia_h_DEFINED\n')
Ben Wagner9494f3e2018-05-25 15:48:31 -040084 for header in headers:
85 f.write('#include "' + header.include + '"\n')
Mike Klein89eda8a2016-11-02 09:14:55 -040086 f.write('#endif//skia_h_DEFINED\n')
87
Ben Wagner9494f3e2018-05-25 15:48:31 -040088with open(skia_h + '.deps', 'w') as f:
Mike Klein89eda8a2016-11-02 09:14:55 -040089 f.write(skia_h + ':')
Ben Wagner9494f3e2018-05-25 15:48:31 -040090 for header in headers:
91 f.write(' ' + header.absolute)
92 f.write(' build.ninja.d')
Mike Klein89eda8a2016-11-02 09:14:55 -040093 f.write('\n')
94
95# Temporary: during development this file wrote skia.h.d, not skia.h.deps,
96# and I think we have some bad versions of those files laying around.
97if os.path.exists(skia_h + '.d'):
98 os.remove(skia_h + '.d')