Address review comments from Matt on the sort_includes.py script.

1) Teach it to handle files with #include on the first line -- these do
   actually exist in LLVM.
2) Support llvm-c and clang-c include projects.
3) Nuke some stail imports.
4) Switch to using os.path to split the file extension off.
5) Remove debugging leftovers.
6) Add docstring (a really puny one) for the sort function.

I'm continuing te avoid stripping the whitespace on the RHS to preserve
whatever newline characters happen to be in the original file.

llvm-svn: 169222
This commit is contained in:
Chandler Carruth 2012-12-04 07:04:58 +00:00
parent dd7ca93abc
commit 43342d5c88
1 changed files with 9 additions and 10 deletions

View File

@ -10,13 +10,12 @@ welcome for more functionality, and sorting other header groups.
import argparse
import os
import re
import sys
import tempfile
def sort_includes(f):
"""Sort the #include lines of a specific file."""
lines = f.readlines()
look_for_api_header = f.name[-4:] == '.cpp'
look_for_api_header = os.path.splitext(f.name)[1] == '.cpp'
found_headers = False
headers_begin = 0
headers_end = 0
api_headers = []
@ -27,8 +26,9 @@ def sort_includes(f):
if l.strip() == '':
continue
if l.startswith('#include'):
if headers_begin == 0:
if not found_headers:
headers_begin = i
found_headers = True
headers_end = i
header = l[len('#include'):].lstrip()
if look_for_api_header and header.startswith('"'):
@ -38,7 +38,8 @@ def sort_includes(f):
if header.startswith('<'):
system_headers.append(header)
continue
if header.startswith('"llvm/') or header.startswith('"clang/'):
if (header.startswith('"llvm/') or header.startswith('"llvm-c/') or
header.startswith('"clang/') or header.startswith('"clang-c/')):
project_headers.append(header)
continue
local_headers.append(header)
@ -46,12 +47,12 @@ def sort_includes(f):
# Only allow comments and #defines prior to any includes. If either are
# mixed with includes, the order might be sensitive.
if headers_begin != 0:
if found_headers:
break
if l.startswith('//') or l.startswith('#define') or l.startswith('#ifndef'):
continue
break
if headers_begin == 0:
if not found_headers:
return
local_headers.sort()
@ -61,8 +62,6 @@ def sort_includes(f):
header_lines = ['#include ' + h for h in headers]
lines = lines[:headers_begin] + header_lines + lines[headers_end + 1:]
#for l in lines[headers_begin:headers_end]:
# print l.rstrip()
f.seek(0)
f.truncate()
f.writelines(lines)