svn merge ^/trunk/blender -r41227:41230 . --accept=theirs-full
[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 import os
29 from os.path import join, dirname, normpath, abspath
30
31 SOURCE_DIR = join(dirname(__file__), "..", "..")
32 SOURCE_DIR = normpath(SOURCE_DIR)
33 SOURCE_DIR = abspath(SOURCE_DIR)
34
35
36 def is_c_header(filename):
37     ext = os.path.splitext(filename)[1]
38     return (ext in (".h", ".hpp", ".hxx"))
39
40
41 def is_c(filename):
42     ext = os.path.splitext(filename)[1]
43     return (ext in (".c", ".cpp", ".cxx", ".m", ".mm", ".rc"))
44
45
46 def is_c_any(filename):
47     return os.path.s_c(filename) or is_c_header(filename)
48
49
50 # copied from project_info.py
51 CMAKE_DIR = "."
52
53
54 def cmake_cache_var(var):
55     cache_file = open(join(CMAKE_DIR, "CMakeCache.txt"))
56     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("#")]
57     cache_file.close()
58
59     for l in lines:
60         if l.split(":")[0] == var:
61             return l.split("=", 1)[-1]
62     return None
63
64
65 def do_ignore(filepath, ignore_prefix_list):
66     if ignore_prefix_list is None:
67         return False
68
69     relpath = os.path.relpath(filepath, SOURCE_DIR)
70     return any([relpath.startswith(prefix) for prefix in ignore_prefix_list])
71
72
73 def makefile_log():
74     import subprocess
75     import time
76     # Check blender is not 2.5x until it supports playback again
77     print("running make with --dry-run ...")
78     process = subprocess.Popen(["make", "--always-make", "--dry-run", "--keep-going", "VERBOSE=1"],
79                                 stdout=subprocess.PIPE,
80                                 )
81
82     while process.poll():
83         time.sleep(1)
84
85     out = process.stdout.read()
86     process.stdout.close()
87     print("done!", len(out), "bytes")
88     return out.decode("ascii").split("\n")
89
90
91 def build_info(use_c=True, use_cxx=True, ignore_prefix_list=None):
92
93     makelog = makefile_log()
94
95     source = []
96
97     compilers = []
98     if use_c:
99         compilers.append(cmake_cache_var("CMAKE_C_COMPILER"))
100     if use_cxx:
101         compilers.append(cmake_cache_var("CMAKE_CXX_COMPILER"))
102
103     print("compilers:", " ".join(compilers))
104
105     fake_compiler = "%COMPILER%"
106
107     print("parsing make log ...")
108
109     for line in makelog:
110
111         args = line.split()
112
113         if not any([(c in args) for c in compilers]):
114             continue
115
116         # join args incase they are not.
117         args = ' '.join(args)
118         args = args.replace(" -isystem", " -I")
119         args = args.replace(" -D ", " -D")
120         args = args.replace(" -I ", " -I")
121
122         for c in compilers:
123             args = args.replace(c, fake_compiler)
124         args = args.split()
125         # end
126
127         # remove compiler
128         args[:args.index(fake_compiler) + 1] = []
129
130         c_files = [f for f in args if is_c(f)]
131         inc_dirs = [f[2:].strip() for f in args if f.startswith('-I')]
132         defs = [f[2:].strip() for f in args if f.startswith('-D')]
133         for c in sorted(c_files):
134
135             if do_ignore(c, ignore_prefix_list):
136                 continue
137
138             source.append((c, inc_dirs, defs))
139
140         # safety check that our includes are ok
141         for f in inc_dirs:
142             if not os.path.exists(f):
143                 raise Exception("%s missing" % f)
144
145     print("done!")
146
147     return source
148
149
150 # could be moved elsewhere!, this just happens to be used by scripts that also
151 # use this module.
152 def queue_processes(process_funcs, job_total=-1):
153     """ Takes a list of function arg pairs, each function must return a process
154     """
155     import sys
156
157     if job_total == -1:
158         import multiprocessing
159         job_total = multiprocessing.cpu_count()
160         del multiprocessing
161
162     if job_total == 1:
163         for func, args in process_funcs:
164             sys.stdout.flush()
165             sys.stderr.flush()
166
167             process = func(*args)
168             process.wait()
169     else:
170         import time
171
172         processes = []
173         for func, args in process_funcs:
174             # wait until a thread is free
175             while 1:
176                 processes[:] = [p for p in processes if p.poll() is None]
177
178                 if len(processes) <= job_total:
179                     break
180                 else:
181                     time.sleep(0.1)
182
183             sys.stdout.flush()
184             sys.stderr.flush()
185
186             processes.append(func(*args))
187
188
189 def main():
190     if not os.path.exists(join(CMAKE_DIR, "CMakeCache.txt")):
191         print("This script must run from the cmake build dir")
192         return
193
194     for s in build_info():
195         print(s)
196
197 if __name__ == "__main__":
198     main()