patch [#29853] UTF-8 copy&paste for X11 GHOST
[blender.git] / build_files / cmake / project_source_info.py
1 # ***** BEGIN GPL LICENSE BLOCK *****
2 #
3 # This program is free software; you can redistribute it and/or
4 # modify it under the terms of the GNU General Public License
5 # as published by the Free Software Foundation; either version 2
6 # of the License, or (at your option) any later version.
7 #
8 # This program is distributed in the hope that it will be useful,
9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11 # GNU General Public License for more details.
12 #
13 # You should have received a copy of the GNU General Public License
14 # along with this program; if not, write to the Free Software Foundation,
15 # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16 #
17 # Contributor(s): Campbell Barton
18 #
19 # ***** END GPL LICENSE BLOCK *****
20
21 # <pep8 compliant>
22
23 __all__ = (
24     "build_info",
25     "SOURCE_DIR",
26     )
27
28
29 import sys
30 if not sys.version.startswith("3"):
31     print("\nPython3.x needed, found %s.\nAborting!\n" %
32           sys.version.partition(" ")[0])
33     sys.exit(1)
34
35
36 import os
37 from os.path import join, dirname, normpath, abspath
38
39 SOURCE_DIR = join(dirname(__file__), "..", "..")
40 SOURCE_DIR = normpath(SOURCE_DIR)
41 SOURCE_DIR = abspath(SOURCE_DIR)
42
43
44 def is_c_header(filename):
45     ext = os.path.splitext(filename)[1]
46     return (ext in (".h", ".hpp", ".hxx"))
47
48
49 def is_c(filename):
50     ext = os.path.splitext(filename)[1]
51     return (ext in (".c", ".cpp", ".cxx", ".m", ".mm", ".rc", ".cc", ".inl"))
52
53
54 def is_c_any(filename):
55     return os.path.s_c(filename) or is_c_header(filename)
56
57
58 # copied from project_info.py
59 CMAKE_DIR = "."
60
61
62 def cmake_cache_var(var):
63     cache_file = open(join(CMAKE_DIR, "CMakeCache.txt"))
64     lines = [l_strip for l in cache_file for l_strip in (l.strip(),) if l_strip if not l_strip.startswith("//") if not l_strip.startswith("#")]
65     cache_file.close()
66
67     for l in lines:
68         if l.split(":")[0] == var:
69             return l.split("=", 1)[-1]
70     return None
71
72
73 def do_ignore(filepath, ignore_prefix_list):
74     if ignore_prefix_list is None:
75         return False
76
77     relpath = os.path.relpath(filepath, SOURCE_DIR)
78     return any([relpath.startswith(prefix) for prefix in ignore_prefix_list])
79
80
81 def makefile_log():
82     import subprocess
83     import time
84
85     print("running make with --dry-run ...")
86     process = subprocess.Popen(["make", "--always-make", "--dry-run", "--keep-going", "VERBOSE=1"],
87                                 stdout=subprocess.PIPE,
88                                 )
89
90     while process.poll():
91         time.sleep(1)
92
93     out = process.stdout.read()
94     process.stdout.close()
95     print("done!", len(out), "bytes")
96     return out.decode("utf-8", errors="ignore").split("\n")
97
98
99 def build_info(use_c=True, use_cxx=True, ignore_prefix_list=None):
100
101     makelog = makefile_log()
102
103     source = []
104
105     compilers = []
106     if use_c:
107         compilers.append(cmake_cache_var("CMAKE_C_COMPILER"))
108     if use_cxx:
109         compilers.append(cmake_cache_var("CMAKE_CXX_COMPILER"))
110
111     print("compilers:", " ".join(compilers))
112
113     fake_compiler = "%COMPILER%"
114
115     print("parsing make log ...")
116
117     for line in makelog:
118
119         args = line.split()
120
121         if not any([(c in args) for c in compilers]):
122             continue
123
124         # join args incase they are not.
125         args = ' '.join(args)
126         args = args.replace(" -isystem", " -I")
127         args = args.replace(" -D ", " -D")
128         args = args.replace(" -I ", " -I")
129
130         for c in compilers:
131             args = args.replace(c, fake_compiler)
132         args = args.split()
133         # end
134
135         # remove compiler
136         args[:args.index(fake_compiler) + 1] = []
137
138         c_files = [f for f in args if is_c(f)]
139         inc_dirs = [f[2:].strip() for f in args if f.startswith('-I')]
140         defs = [f[2:].strip() for f in args if f.startswith('-D')]
141         for c in sorted(c_files):
142
143             if do_ignore(c, ignore_prefix_list):
144                 continue
145
146             source.append((c, inc_dirs, defs))
147
148         # safety check that our includes are ok
149         for f in inc_dirs:
150             if not os.path.exists(f):
151                 raise Exception("%s missing" % f)
152
153     print("done!")
154
155     return source
156
157
158 # could be moved elsewhere!, this just happens to be used by scripts that also
159 # use this module.
160 def queue_processes(process_funcs, job_total=-1):
161     """ Takes a list of function arg pairs, each function must return a process
162     """
163     import sys
164
165     if job_total == -1:
166         import multiprocessing
167         job_total = multiprocessing.cpu_count()
168         del multiprocessing
169
170     if job_total == 1:
171         for func, args in process_funcs:
172             sys.stdout.flush()
173             sys.stderr.flush()
174
175             process = func(*args)
176             process.wait()
177     else:
178         import time
179
180         processes = []
181         for func, args in process_funcs:
182             # wait until a thread is free
183             while 1:
184                 processes[:] = [p for p in processes if p.poll() is None]
185
186                 if len(processes) <= job_total:
187                     break
188                 else:
189                     time.sleep(0.1)
190
191             sys.stdout.flush()
192             sys.stderr.flush()
193
194             processes.append(func(*args))
195
196
197 def main():
198     if not os.path.exists(join(CMAKE_DIR, "CMakeCache.txt")):
199         print("This script must run from the cmake build dir")
200         return
201
202     for s in build_info():
203         print(s)
204
205 if __name__ == "__main__":
206     main()