mirror of
https://github.com/cp2k/cp2k.git
synced 2026-07-25 20:45:19 -04:00
64 lines
1.6 KiB
Python
Executable file
64 lines
1.6 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import re
|
|
import sys
|
|
from os import path
|
|
|
|
# pre-compiled regular expressions
|
|
re_program = re.compile(r"\n\s*end\s*program")
|
|
re_main = re.compile(r"\sint\s+main\s*\(")
|
|
|
|
|
|
# ============================================================================
|
|
def main():
|
|
if len(sys.argv) != 2:
|
|
print("Usage: discover_programs.py <src-dir>")
|
|
sys.exit(1)
|
|
|
|
srcdir = sys.argv[1]
|
|
|
|
sys.stderr.write("Discovering programs ...\n")
|
|
|
|
programs = []
|
|
for root, _, files in os.walk(srcdir):
|
|
if root.endswith("/preprettify"):
|
|
continue
|
|
if "/.git" in root:
|
|
continue
|
|
|
|
for fn in files:
|
|
abs_fn = path.join(root, fn)
|
|
if fn[-2:] == ".F":
|
|
if is_fortran_program(abs_fn):
|
|
programs.append(abs_fn)
|
|
|
|
elif fn[-2:] == ".c" or fn[-3:] == ".cu":
|
|
if has_main_function(abs_fn):
|
|
programs.append(abs_fn)
|
|
|
|
print(" ".join(programs))
|
|
|
|
|
|
# ============================================================================
|
|
def is_fortran_program(fn):
|
|
f = open(fn, encoding="utf8")
|
|
s = path.getsize(fn)
|
|
f.seek(max(0, s - 100))
|
|
tail = f.read()
|
|
f.close()
|
|
m = re_program.search(tail.lower())
|
|
return m is not None
|
|
|
|
|
|
# ============================================================================
|
|
def has_main_function(fn):
|
|
f = open(fn, encoding="utf8")
|
|
content = f.read()
|
|
f.close()
|
|
m = re_main.search(content)
|
|
return m is not None
|
|
|
|
|
|
# ===============================================================================
|
|
main()
|