mirror of
https://github.com/facebook/rocksdb.git
synced 2024-11-27 11:43:49 +00:00
7a23e4d8ca
This commit adds two new targets to the Makefile: rocksdb.cc and rocksdb.h These files, when combined with the c.h header, are a self-contained RocksDB source distribution called an amalgamation. (The name comes from SQLite's, which is similar in concept.) The main benefit of an amalgamation is that it's very easy to drop into a new project. It also compiles faster compared to compiling individual source files and potentially gives the compiler more opportunity to make optimizations since it can see all functions at once. rocksdb.cc and rocksdb.h are generated by a new script, amalgamate.py. A detailed description of how amalgamate.py works is in a comment at the top of the file. There are also some small changes to existing files to enable the amalgamation: * Use quotes for includes in unity build * Fix an old header inclusion in util/xfunc.cc * Move some includes outside ifdef in util/env_hdfs.cc * Separate out tool sources in Makefile so they won't be included in unity.cc * Unity build now produces a static library Closes #733
111 lines
4.6 KiB
Python
Executable file
111 lines
4.6 KiB
Python
Executable file
#!/usr/bin/python
|
|
|
|
# amalgamate.py creates an amalgamation from a unity build.
|
|
# It can be run with either Python 2 or 3.
|
|
# An amalgamation consists of a header that includes the contents of all public
|
|
# headers and a source file that includes the contents of all source files and
|
|
# private headers.
|
|
#
|
|
# This script works by starting with the unity build file and recursively expanding
|
|
# #include directives. If the #include is found in a public include directory,
|
|
# that header is expanded into the amalgamation header.
|
|
#
|
|
# A particular header is only expanded once, so this script will
|
|
# break if there are multiple inclusions of the same header that are expected to
|
|
# expand differently. Similarly, this type of code causes issues:
|
|
#
|
|
# #ifdef FOO
|
|
# #include "bar.h"
|
|
# // code here
|
|
# #else
|
|
# #include "bar.h" // oops, doesn't get expanded
|
|
# // different code here
|
|
# #endif
|
|
#
|
|
# The solution is to move the include out of the #ifdef.
|
|
|
|
from __future__ import print_function
|
|
|
|
import argparse
|
|
from os import path
|
|
import re
|
|
import sys
|
|
|
|
include_re = re.compile('^[ \t]*#include[ \t]+"(.*)"[ \t]*$')
|
|
included = set()
|
|
excluded = set()
|
|
|
|
def find_header(name, abs_path, include_paths):
|
|
samedir = path.join(path.dirname(abs_path), name)
|
|
if path.exists(samedir):
|
|
return samedir
|
|
for include_path in include_paths:
|
|
include_path = path.join(include_path, name)
|
|
if path.exists(include_path):
|
|
return include_path
|
|
return None
|
|
|
|
def expand_include(include_path, f, abs_path, source_out, header_out, include_paths, public_include_paths):
|
|
if include_path in included:
|
|
return False
|
|
|
|
included.add(include_path)
|
|
with open(include_path) as f:
|
|
print('#line 1 "{}"'.format(include_path), file=source_out)
|
|
process_file(f, include_path, source_out, header_out, include_paths, public_include_paths)
|
|
return True
|
|
|
|
def process_file(f, abs_path, source_out, header_out, include_paths, public_include_paths):
|
|
for (line, text) in enumerate(f):
|
|
m = include_re.match(text)
|
|
if m:
|
|
filename = m.groups()[0]
|
|
# first check private headers
|
|
include_path = find_header(filename, abs_path, include_paths)
|
|
if include_path:
|
|
if include_path in excluded:
|
|
source_out.write(text)
|
|
expanded = False
|
|
else:
|
|
expanded = expand_include(include_path, f, abs_path, source_out, header_out, include_paths, public_include_paths)
|
|
else:
|
|
# now try public headers
|
|
include_path = find_header(filename, abs_path, public_include_paths)
|
|
if include_path:
|
|
# found public header
|
|
expanded = False
|
|
if include_path in excluded:
|
|
source_out.write(text)
|
|
else:
|
|
expand_include(include_path, f, abs_path, header_out, None, public_include_paths, [])
|
|
else:
|
|
sys.exit("unable to find {}, included in {} on line {}".format(filename, abs_path, line))
|
|
|
|
if expanded:
|
|
print('#line {} "{}"'.format(line+1, abs_path), file=source_out)
|
|
elif text != "#pragma once\n":
|
|
source_out.write(text)
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="Transform a unity build into an amalgamation")
|
|
parser.add_argument("source", help="source file")
|
|
parser.add_argument("-I", action="append", dest="include_paths", help="include paths for private headers")
|
|
parser.add_argument("-i", action="append", dest="public_include_paths", help="include paths for public headers")
|
|
parser.add_argument("-x", action="append", dest="excluded", help="excluded header files")
|
|
parser.add_argument("-o", dest="source_out", help="output C++ file", required=True)
|
|
parser.add_argument("-H", dest="header_out", help="output C++ header file", required=True)
|
|
args = parser.parse_args()
|
|
|
|
include_paths = list(map(path.abspath, args.include_paths or []))
|
|
public_include_paths = list(map(path.abspath, args.public_include_paths or []))
|
|
excluded.update(map(path.abspath, args.excluded or []))
|
|
filename = args.source
|
|
abs_path = path.abspath(filename)
|
|
with open(filename) as f, open(args.source_out, 'w') as source_out, open(args.header_out, 'w') as header_out:
|
|
print('#line 1 "{}"'.format(filename), file=source_out)
|
|
print('#include "{}"'.format(header_out.name), file=source_out)
|
|
process_file(f, abs_path, source_out, header_out, include_paths, public_include_paths)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|