You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.7 KiB
91 lines
2.7 KiB
#!/usr/bin/env python3
|
|
import glob
|
|
import os.path
|
|
import sys
|
|
|
|
mit_copyright = open("scripts/copyright_mit.txt",'r').read()
|
|
|
|
def add_cpp_copyright( f, content):
|
|
global mit_copyright
|
|
out = open(f,'w')
|
|
out.write("/*\n")
|
|
for line in mit_copyright.split('\n')[:-1]:
|
|
out.write(" *");
|
|
if line.strip() != "":
|
|
out.write(" %s" %line)
|
|
out.write("\n")
|
|
out.write(" */\n")
|
|
out.write(content.strip())
|
|
out.write("\n")
|
|
out.close()
|
|
|
|
def add_python_copyright( f, content):
|
|
global mit_copyright
|
|
out = open(f,'w')
|
|
for line in mit_copyright.split('\n')[:-1]:
|
|
out.write("#");
|
|
if line.strip() != "":
|
|
out.write(" %s" %line)
|
|
out.write("\n")
|
|
out.write(content.strip())
|
|
out.write("\n")
|
|
out.close()
|
|
|
|
def remove_comment( content ):
|
|
comment=True
|
|
out=""
|
|
for line in content.split('\n'):
|
|
if comment:
|
|
if line.startswith(' */'):
|
|
comment=False
|
|
elif line.startswith('/*') or line.startswith(' *'):
|
|
#print(line)
|
|
continue
|
|
else:
|
|
raise Exception("ERROR: not a comment ? '%s'"% line)
|
|
else:
|
|
out += line + "\n"
|
|
return out
|
|
def remove_comment_python( content ):
|
|
comment=True
|
|
out=""
|
|
for line in content.split('\n'):
|
|
if comment and line.startswith('#'):
|
|
continue
|
|
else:
|
|
comment = False
|
|
out += line + "\n"
|
|
return out
|
|
|
|
def check_file( path ):
|
|
root, f = os.path.split(path)
|
|
if f in ['.clang-tidy', '.clang-format']:
|
|
print("Skipping file: {}".format(path))
|
|
return
|
|
|
|
with open(path, 'r', encoding='utf-8') as fd:
|
|
content = fd.read()
|
|
_, extension = os.path.splitext(f)
|
|
|
|
if extension in ['.cpp', '.h', '.hpp', '.inl', '.cl', '.in', '.cs']:
|
|
if not content.startswith('/*'):
|
|
add_cpp_copyright(path, content)
|
|
elif extension == '.py' or f in ['SConstruct', 'SConscript']:
|
|
if not content.startswith('# Copyright'):
|
|
add_python_copyright(path, content)
|
|
elif f == 'CMakeLists.txt':
|
|
if not content.startswith('# Copyright'):
|
|
add_python_copyright(path, content)
|
|
else:
|
|
raise Exception("Unhandled file: {}".format(path))
|
|
|
|
if len(sys.argv) > 1:
|
|
for path in sys.argv[1:]:
|
|
check_file(path)
|
|
else:
|
|
for top in ['./arm_compute', './tests','./src','./examples','./utils/','./opencl-1.2-stubs/','./opengles-3.1-stubs/','./support']:
|
|
for root, _, files in os.walk(top):
|
|
for f in files:
|
|
path = os.path.join(root, f)
|
|
check_file(path)
|