methods.py 40 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062
  1. import os
  2. import re
  3. import glob
  4. import subprocess
  5. from collections import OrderedDict
  6. from collections.abc import Mapping
  7. from typing import Iterator
  8. # We need to define our own `Action` method to control the verbosity of output
  9. # and whenever we need to run those commands in a subprocess on some platforms.
  10. from SCons import Node
  11. from SCons.Script import Action
  12. from SCons.Script import ARGUMENTS
  13. from SCons.Script import Glob
  14. from SCons.Variables.BoolVariable import _text2bool
  15. from platform_methods import run_in_subprocess
  16. def add_source_files(self, sources, files):
  17. # Convert string to list of absolute paths (including expanding wildcard)
  18. if isinstance(files, (str, bytes)):
  19. # Keep SCons project-absolute path as they are (no wildcard support)
  20. if files.startswith("#"):
  21. if "*" in files:
  22. print("ERROR: Wildcards can't be expanded in SCons project-absolute path: '{}'".format(files))
  23. return
  24. files = [files]
  25. else:
  26. # Exclude .gen.cpp files from globbing, to avoid including obsolete ones.
  27. # They should instead be added manually.
  28. skip_gen_cpp = "*" in files
  29. dir_path = self.Dir(".").abspath
  30. files = sorted(glob.glob(dir_path + "/" + files))
  31. if skip_gen_cpp:
  32. files = [f for f in files if not f.endswith(".gen.cpp")]
  33. # Add each path as compiled Object following environment (self) configuration
  34. for path in files:
  35. obj = self.Object(path)
  36. if obj in sources:
  37. print('WARNING: Object "{}" already included in environment sources.'.format(obj))
  38. continue
  39. sources.append(obj)
  40. def disable_warnings(self):
  41. # 'self' is the environment
  42. if self.msvc:
  43. # We have to remove existing warning level defines before appending /w,
  44. # otherwise we get: "warning D9025 : overriding '/W3' with '/w'"
  45. warn_flags = ["/Wall", "/W4", "/W3", "/W2", "/W1", "/WX"]
  46. self.Append(CCFLAGS=["/w"])
  47. self.Append(CFLAGS=["/w"])
  48. self.Append(CXXFLAGS=["/w"])
  49. self["CCFLAGS"] = [x for x in self["CCFLAGS"] if not x in warn_flags]
  50. self["CFLAGS"] = [x for x in self["CFLAGS"] if not x in warn_flags]
  51. self["CXXFLAGS"] = [x for x in self["CXXFLAGS"] if not x in warn_flags]
  52. else:
  53. self.Append(CCFLAGS=["-w"])
  54. self.Append(CFLAGS=["-w"])
  55. self.Append(CXXFLAGS=["-w"])
  56. def force_optimization_on_debug(self):
  57. # 'self' is the environment
  58. if self["target"] != "debug":
  59. return
  60. if self.msvc:
  61. self.Append(CCFLAGS=["/O2"])
  62. else:
  63. self.Append(CCFLAGS=["-O3"])
  64. def add_module_version_string(self, s):
  65. self.module_version_string += "." + s
  66. def update_version(module_version_string=""):
  67. build_name = "custom_build"
  68. if os.getenv("BUILD_NAME") != None:
  69. build_name = str(os.getenv("BUILD_NAME"))
  70. print("Using custom build name: " + build_name)
  71. import version
  72. # NOTE: It is safe to generate this file here, since this is still executed serially
  73. f = open("core/version_generated.gen.h", "w")
  74. f.write("/* THIS FILE IS GENERATED DO NOT EDIT */\n")
  75. f.write("#ifndef VERSION_GENERATED_GEN_H\n")
  76. f.write("#define VERSION_GENERATED_GEN_H\n")
  77. f.write('#define VERSION_SHORT_NAME "' + str(version.short_name) + '"\n')
  78. f.write('#define VERSION_NAME "' + str(version.name) + '"\n')
  79. f.write("#define VERSION_MAJOR " + str(version.major) + "\n")
  80. f.write("#define VERSION_MINOR " + str(version.minor) + "\n")
  81. f.write("#define VERSION_PATCH " + str(version.patch) + "\n")
  82. # For dev snapshots (alpha, beta, RC, etc.) we do not commit status change to Git,
  83. # so this define provides a way to override it without having to modify the source.
  84. godot_status = str(version.status)
  85. if os.getenv("GODOT_VERSION_STATUS") != None:
  86. godot_status = str(os.getenv("GODOT_VERSION_STATUS"))
  87. print("Using version status '{}', overriding the original '{}'.".format(godot_status, str(version.status)))
  88. f.write('#define VERSION_STATUS "' + godot_status + '"\n')
  89. f.write('#define VERSION_BUILD "' + str(build_name) + '"\n')
  90. f.write('#define VERSION_MODULE_CONFIG "' + str(version.module_config) + module_version_string + '"\n')
  91. f.write("#define VERSION_YEAR " + str(version.year) + "\n")
  92. f.write('#define VERSION_WEBSITE "' + str(version.website) + '"\n')
  93. f.write("#endif // VERSION_GENERATED_GEN_H\n")
  94. f.close()
  95. # NOTE: It is safe to generate this file here, since this is still executed serially
  96. fhash = open("core/version_hash.gen.h", "w")
  97. fhash.write("/* THIS FILE IS GENERATED DO NOT EDIT */\n")
  98. fhash.write("#ifndef VERSION_HASH_GEN_H\n")
  99. fhash.write("#define VERSION_HASH_GEN_H\n")
  100. githash = ""
  101. gitfolder = ".git"
  102. if os.path.isfile(".git"):
  103. module_folder = open(".git", "r").readline().strip()
  104. if module_folder.startswith("gitdir: "):
  105. gitfolder = module_folder[8:]
  106. if os.path.isfile(os.path.join(gitfolder, "HEAD")):
  107. head = open(os.path.join(gitfolder, "HEAD"), "r", encoding="utf8").readline().strip()
  108. if head.startswith("ref: "):
  109. head = os.path.join(gitfolder, head[5:])
  110. if os.path.isfile(head):
  111. githash = open(head, "r").readline().strip()
  112. else:
  113. githash = head
  114. fhash.write('#define VERSION_HASH "' + githash + '"\n')
  115. fhash.write("#endif // VERSION_HASH_GEN_H\n")
  116. fhash.close()
  117. def parse_cg_file(fname, uniforms, sizes, conditionals):
  118. fs = open(fname, "r")
  119. line = fs.readline()
  120. while line:
  121. if re.match(r"^\s*uniform", line):
  122. res = re.match(r"uniform ([\d\w]*) ([\d\w]*)")
  123. type = res.groups(1)
  124. name = res.groups(2)
  125. uniforms.append(name)
  126. if type.find("texobj") != -1:
  127. sizes.append(1)
  128. else:
  129. t = re.match(r"float(\d)x(\d)", type)
  130. if t:
  131. sizes.append(int(t.groups(1)) * int(t.groups(2)))
  132. else:
  133. t = re.match(r"float(\d)", type)
  134. sizes.append(int(t.groups(1)))
  135. if line.find("[branch]") != -1:
  136. conditionals.append(name)
  137. line = fs.readline()
  138. fs.close()
  139. def get_cmdline_bool(option, default):
  140. """We use `ARGUMENTS.get()` to check if options were manually overridden on the command line,
  141. and SCons' _text2bool helper to convert them to booleans, otherwise they're handled as strings.
  142. """
  143. cmdline_val = ARGUMENTS.get(option)
  144. if cmdline_val is not None:
  145. return _text2bool(cmdline_val)
  146. else:
  147. return default
  148. def detect_modules(search_path, recursive=False):
  149. """Detects and collects a list of C++ modules at specified path
  150. `search_path` - a directory path containing modules. The path may point to
  151. a single module, which may have other nested modules. A module must have
  152. "register_types.h", "SCsub", "config.py" files created to be detected.
  153. `recursive` - if `True`, then all subdirectories are searched for modules as
  154. specified by the `search_path`, otherwise collects all modules under the
  155. `search_path` directory. If the `search_path` is a module, it is collected
  156. in all cases.
  157. Returns an `OrderedDict` with module names as keys, and directory paths as
  158. values. If a path is relative, then it is a built-in module. If a path is
  159. absolute, then it is a custom module collected outside of the engine source.
  160. """
  161. modules = OrderedDict()
  162. def add_module(path):
  163. module_name = os.path.basename(path)
  164. module_path = path.replace("\\", "/") # win32
  165. modules[module_name] = module_path
  166. def is_engine(path):
  167. # Prevent recursively detecting modules in self and other
  168. # Godot sources when using `custom_modules` build option.
  169. version_path = os.path.join(path, "version.py")
  170. if os.path.exists(version_path):
  171. with open(version_path) as f:
  172. if 'short_name = "godot"' in f.read():
  173. return True
  174. return False
  175. def get_files(path):
  176. files = glob.glob(os.path.join(path, "*"))
  177. # Sort so that `register_module_types` does not change that often,
  178. # and plugins are registered in alphabetic order as well.
  179. files.sort()
  180. return files
  181. if not recursive:
  182. if is_module(search_path):
  183. add_module(search_path)
  184. for path in get_files(search_path):
  185. if is_engine(path):
  186. continue
  187. if is_module(path):
  188. add_module(path)
  189. else:
  190. to_search = [search_path]
  191. while to_search:
  192. path = to_search.pop()
  193. if is_module(path):
  194. add_module(path)
  195. for child in get_files(path):
  196. if not os.path.isdir(child):
  197. continue
  198. if is_engine(child):
  199. continue
  200. to_search.insert(0, child)
  201. return modules
  202. def is_module(path):
  203. if not os.path.isdir(path):
  204. return False
  205. must_exist = ["register_types.h", "SCsub", "config.py"]
  206. for f in must_exist:
  207. if not os.path.exists(os.path.join(path, f)):
  208. return False
  209. return True
  210. def write_disabled_classes(class_list):
  211. f = open("core/disabled_classes.gen.h", "w")
  212. f.write("/* THIS FILE IS GENERATED DO NOT EDIT */\n")
  213. f.write("#ifndef DISABLED_CLASSES_GEN_H\n")
  214. f.write("#define DISABLED_CLASSES_GEN_H\n\n")
  215. for c in class_list:
  216. cs = c.strip()
  217. if cs != "":
  218. f.write("#define ClassDB_Disable_" + cs + " 1\n")
  219. f.write("\n#endif\n")
  220. def write_modules(modules):
  221. includes_cpp = ""
  222. preregister_cpp = ""
  223. register_cpp = ""
  224. unregister_cpp = ""
  225. for name, path in modules.items():
  226. try:
  227. with open(os.path.join(path, "register_types.h")):
  228. includes_cpp += '#include "' + path + '/register_types.h"\n'
  229. preregister_cpp += "#ifdef MODULE_" + name.upper() + "_ENABLED\n"
  230. preregister_cpp += "#ifdef MODULE_" + name.upper() + "_HAS_PREREGISTER\n"
  231. preregister_cpp += "\tpreregister_" + name + "_types();\n"
  232. preregister_cpp += "#endif\n"
  233. preregister_cpp += "#endif\n"
  234. register_cpp += "#ifdef MODULE_" + name.upper() + "_ENABLED\n"
  235. register_cpp += "\tregister_" + name + "_types();\n"
  236. register_cpp += "#endif\n"
  237. unregister_cpp += "#ifdef MODULE_" + name.upper() + "_ENABLED\n"
  238. unregister_cpp += "\tunregister_" + name + "_types();\n"
  239. unregister_cpp += "#endif\n"
  240. except OSError:
  241. pass
  242. modules_cpp = """// register_module_types.gen.cpp
  243. /* THIS FILE IS GENERATED DO NOT EDIT */
  244. #include "register_module_types.h"
  245. #include "modules/modules_enabled.gen.h"
  246. %s
  247. void preregister_module_types() {
  248. %s
  249. }
  250. void register_module_types() {
  251. %s
  252. }
  253. void unregister_module_types() {
  254. %s
  255. }
  256. """ % (
  257. includes_cpp,
  258. preregister_cpp,
  259. register_cpp,
  260. unregister_cpp,
  261. )
  262. # NOTE: It is safe to generate this file here, since this is still executed serially
  263. with open("modules/register_module_types.gen.cpp", "w") as f:
  264. f.write(modules_cpp)
  265. def convert_custom_modules_path(path):
  266. if not path:
  267. return path
  268. path = os.path.realpath(os.path.expanduser(os.path.expandvars(path)))
  269. err_msg = "Build option 'custom_modules' must %s"
  270. if not os.path.isdir(path):
  271. raise ValueError(err_msg % "point to an existing directory.")
  272. if path == os.path.realpath("modules"):
  273. raise ValueError(err_msg % "be a directory other than built-in `modules` directory.")
  274. return path
  275. def disable_module(self):
  276. self.disabled_modules.append(self.current_module)
  277. def module_check_dependencies(self, module, dependencies, silent=False):
  278. """
  279. Checks if module dependencies are enabled for a given module,
  280. and prints a warning if they aren't.
  281. Meant to be used in module `can_build` methods.
  282. Returns a boolean (True if dependencies are satisfied).
  283. """
  284. missing_deps = []
  285. for dep in dependencies:
  286. opt = "module_{}_enabled".format(dep)
  287. if not opt in self or not self[opt]:
  288. missing_deps.append(dep)
  289. if missing_deps != []:
  290. if not silent:
  291. print(
  292. "Disabling '{}' module as the following dependencies are not satisfied: {}".format(
  293. module, ", ".join(missing_deps)
  294. )
  295. )
  296. return False
  297. else:
  298. return True
  299. def use_windows_spawn_fix(self, platform=None):
  300. if os.name != "nt":
  301. return # not needed, only for windows
  302. # On Windows, due to the limited command line length, when creating a static library
  303. # from a very high number of objects SCons will invoke "ar" once per object file;
  304. # that makes object files with same names to be overwritten so the last wins and
  305. # the library loses symbols defined by overwritten objects.
  306. # By enabling quick append instead of the default mode (replacing), libraries will
  307. # got built correctly regardless the invocation strategy.
  308. # Furthermore, since SCons will rebuild the library from scratch when an object file
  309. # changes, no multiple versions of the same object file will be present.
  310. self.Replace(ARFLAGS="q")
  311. def mySubProcess(cmdline, env):
  312. startupinfo = subprocess.STARTUPINFO()
  313. startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
  314. proc = subprocess.Popen(
  315. cmdline,
  316. stdin=subprocess.PIPE,
  317. stdout=subprocess.PIPE,
  318. stderr=subprocess.PIPE,
  319. startupinfo=startupinfo,
  320. shell=False,
  321. env=env,
  322. )
  323. _, err = proc.communicate()
  324. rv = proc.wait()
  325. if rv:
  326. print("=====")
  327. print(err)
  328. print("=====")
  329. return rv
  330. def mySpawn(sh, escape, cmd, args, env):
  331. newargs = " ".join(args[1:])
  332. cmdline = cmd + " " + newargs
  333. rv = 0
  334. env = {str(key): str(value) for key, value in iter(env.items())}
  335. if len(cmdline) > 32000 and cmd.endswith("ar"):
  336. cmdline = cmd + " " + args[1] + " " + args[2] + " "
  337. for i in range(3, len(args)):
  338. rv = mySubProcess(cmdline + args[i], env)
  339. if rv:
  340. break
  341. else:
  342. rv = mySubProcess(cmdline, env)
  343. return rv
  344. self["SPAWN"] = mySpawn
  345. def save_active_platforms(apnames, ap):
  346. for x in ap:
  347. names = ["logo"]
  348. if os.path.isfile(x + "/run_icon.png"):
  349. names.append("run_icon")
  350. for name in names:
  351. pngf = open(x + "/" + name + ".png", "rb")
  352. b = pngf.read(1)
  353. str = " /* AUTOGENERATED FILE, DO NOT EDIT */ \n"
  354. str += " static const unsigned char _" + x[9:] + "_" + name + "[]={"
  355. while len(b) == 1:
  356. str += hex(ord(b))
  357. b = pngf.read(1)
  358. if len(b) == 1:
  359. str += ","
  360. str += "};\n"
  361. pngf.close()
  362. # NOTE: It is safe to generate this file here, since this is still executed serially
  363. wf = x + "/" + name + ".gen.h"
  364. with open(wf, "w") as pngw:
  365. pngw.write(str)
  366. def no_verbose(sys, env):
  367. colors = {}
  368. # Colors are disabled in non-TTY environments such as pipes. This means
  369. # that if output is redirected to a file, it will not contain color codes
  370. if sys.stdout.isatty():
  371. colors["cyan"] = "\033[96m"
  372. colors["purple"] = "\033[95m"
  373. colors["blue"] = "\033[94m"
  374. colors["green"] = "\033[92m"
  375. colors["yellow"] = "\033[93m"
  376. colors["red"] = "\033[91m"
  377. colors["end"] = "\033[0m"
  378. else:
  379. colors["cyan"] = ""
  380. colors["purple"] = ""
  381. colors["blue"] = ""
  382. colors["green"] = ""
  383. colors["yellow"] = ""
  384. colors["red"] = ""
  385. colors["end"] = ""
  386. compile_source_message = "{}Compiling {}==> {}$SOURCE{}".format(
  387. colors["blue"], colors["purple"], colors["yellow"], colors["end"]
  388. )
  389. java_compile_source_message = "{}Compiling {}==> {}$SOURCE{}".format(
  390. colors["blue"], colors["purple"], colors["yellow"], colors["end"]
  391. )
  392. compile_shared_source_message = "{}Compiling shared {}==> {}$SOURCE{}".format(
  393. colors["blue"], colors["purple"], colors["yellow"], colors["end"]
  394. )
  395. link_program_message = "{}Linking Program {}==> {}$TARGET{}".format(
  396. colors["red"], colors["purple"], colors["yellow"], colors["end"]
  397. )
  398. link_library_message = "{}Linking Static Library {}==> {}$TARGET{}".format(
  399. colors["red"], colors["purple"], colors["yellow"], colors["end"]
  400. )
  401. ranlib_library_message = "{}Ranlib Library {}==> {}$TARGET{}".format(
  402. colors["red"], colors["purple"], colors["yellow"], colors["end"]
  403. )
  404. link_shared_library_message = "{}Linking Shared Library {}==> {}$TARGET{}".format(
  405. colors["red"], colors["purple"], colors["yellow"], colors["end"]
  406. )
  407. java_library_message = "{}Creating Java Archive {}==> {}$TARGET{}".format(
  408. colors["red"], colors["purple"], colors["yellow"], colors["end"]
  409. )
  410. env.Append(CXXCOMSTR=[compile_source_message])
  411. env.Append(CCCOMSTR=[compile_source_message])
  412. env.Append(SHCCCOMSTR=[compile_shared_source_message])
  413. env.Append(SHCXXCOMSTR=[compile_shared_source_message])
  414. env.Append(ARCOMSTR=[link_library_message])
  415. env.Append(RANLIBCOMSTR=[ranlib_library_message])
  416. env.Append(SHLINKCOMSTR=[link_shared_library_message])
  417. env.Append(LINKCOMSTR=[link_program_message])
  418. env.Append(JARCOMSTR=[java_library_message])
  419. env.Append(JAVACCOMSTR=[java_compile_source_message])
  420. def detect_visual_c_compiler_version(tools_env):
  421. # tools_env is the variable scons uses to call tools that execute tasks, SCons's env['ENV'] that executes tasks...
  422. # (see the SCons documentation for more information on what it does)...
  423. # in order for this function to be well encapsulated i choose to force it to receive SCons's TOOLS env (env['ENV']
  424. # and not scons setup environment (env)... so make sure you call the right environment on it or it will fail to detect
  425. # the proper vc version that will be called
  426. # There is no flag to give to visual c compilers to set the architecture, i.e. scons bits argument (32,64,ARM etc)
  427. # There are many different cl.exe files that are run, and each one compiles & links to a different architecture
  428. # As far as I know, the only way to figure out what compiler will be run when Scons calls cl.exe via Program()
  429. # is to check the PATH variable and figure out which one will be called first. Code below does that and returns:
  430. # the following string values:
  431. # "" Compiler not detected
  432. # "amd64" Native 64 bit compiler
  433. # "amd64_x86" 64 bit Cross Compiler for 32 bit
  434. # "x86" Native 32 bit compiler
  435. # "x86_amd64" 32 bit Cross Compiler for 64 bit
  436. # There are other architectures, but Godot does not support them currently, so this function does not detect arm/amd64_arm
  437. # and similar architectures/compilers
  438. # Set chosen compiler to "not detected"
  439. vc_chosen_compiler_index = -1
  440. vc_chosen_compiler_str = ""
  441. # Start with Pre VS 2017 checks which uses VCINSTALLDIR:
  442. if "VCINSTALLDIR" in tools_env:
  443. # print("Checking VCINSTALLDIR")
  444. # find() works with -1 so big ifs below are needed... the simplest solution, in fact
  445. # First test if amd64 and amd64_x86 compilers are present in the path
  446. vc_amd64_compiler_detection_index = tools_env["PATH"].find(tools_env["VCINSTALLDIR"] + "BIN\\amd64;")
  447. if vc_amd64_compiler_detection_index > -1:
  448. vc_chosen_compiler_index = vc_amd64_compiler_detection_index
  449. vc_chosen_compiler_str = "amd64"
  450. vc_amd64_x86_compiler_detection_index = tools_env["PATH"].find(tools_env["VCINSTALLDIR"] + "BIN\\amd64_x86;")
  451. if vc_amd64_x86_compiler_detection_index > -1 and (
  452. vc_chosen_compiler_index == -1 or vc_chosen_compiler_index > vc_amd64_x86_compiler_detection_index
  453. ):
  454. vc_chosen_compiler_index = vc_amd64_x86_compiler_detection_index
  455. vc_chosen_compiler_str = "amd64_x86"
  456. # Now check the 32 bit compilers
  457. vc_x86_compiler_detection_index = tools_env["PATH"].find(tools_env["VCINSTALLDIR"] + "BIN;")
  458. if vc_x86_compiler_detection_index > -1 and (
  459. vc_chosen_compiler_index == -1 or vc_chosen_compiler_index > vc_x86_compiler_detection_index
  460. ):
  461. vc_chosen_compiler_index = vc_x86_compiler_detection_index
  462. vc_chosen_compiler_str = "x86"
  463. vc_x86_amd64_compiler_detection_index = tools_env["PATH"].find(tools_env["VCINSTALLDIR"] + "BIN\\x86_amd64;")
  464. if vc_x86_amd64_compiler_detection_index > -1 and (
  465. vc_chosen_compiler_index == -1 or vc_chosen_compiler_index > vc_x86_amd64_compiler_detection_index
  466. ):
  467. vc_chosen_compiler_index = vc_x86_amd64_compiler_detection_index
  468. vc_chosen_compiler_str = "x86_amd64"
  469. # and for VS 2017 and newer we check VCTOOLSINSTALLDIR:
  470. if "VCTOOLSINSTALLDIR" in tools_env:
  471. # Newer versions have a different path available
  472. vc_amd64_compiler_detection_index = (
  473. tools_env["PATH"].upper().find(tools_env["VCTOOLSINSTALLDIR"].upper() + "BIN\\HOSTX64\\X64;")
  474. )
  475. if vc_amd64_compiler_detection_index > -1:
  476. vc_chosen_compiler_index = vc_amd64_compiler_detection_index
  477. vc_chosen_compiler_str = "amd64"
  478. vc_amd64_x86_compiler_detection_index = (
  479. tools_env["PATH"].upper().find(tools_env["VCTOOLSINSTALLDIR"].upper() + "BIN\\HOSTX64\\X86;")
  480. )
  481. if vc_amd64_x86_compiler_detection_index > -1 and (
  482. vc_chosen_compiler_index == -1 or vc_chosen_compiler_index > vc_amd64_x86_compiler_detection_index
  483. ):
  484. vc_chosen_compiler_index = vc_amd64_x86_compiler_detection_index
  485. vc_chosen_compiler_str = "amd64_x86"
  486. vc_x86_compiler_detection_index = (
  487. tools_env["PATH"].upper().find(tools_env["VCTOOLSINSTALLDIR"].upper() + "BIN\\HOSTX86\\X86;")
  488. )
  489. if vc_x86_compiler_detection_index > -1 and (
  490. vc_chosen_compiler_index == -1 or vc_chosen_compiler_index > vc_x86_compiler_detection_index
  491. ):
  492. vc_chosen_compiler_index = vc_x86_compiler_detection_index
  493. vc_chosen_compiler_str = "x86"
  494. vc_x86_amd64_compiler_detection_index = (
  495. tools_env["PATH"].upper().find(tools_env["VCTOOLSINSTALLDIR"].upper() + "BIN\\HOSTX86\\X64;")
  496. )
  497. if vc_x86_amd64_compiler_detection_index > -1 and (
  498. vc_chosen_compiler_index == -1 or vc_chosen_compiler_index > vc_x86_amd64_compiler_detection_index
  499. ):
  500. vc_chosen_compiler_index = vc_x86_amd64_compiler_detection_index
  501. vc_chosen_compiler_str = "x86_amd64"
  502. return vc_chosen_compiler_str
  503. def find_visual_c_batch_file(env):
  504. from SCons.Tool.MSCommon.vc import (
  505. get_default_version,
  506. get_host_target,
  507. find_batch_file,
  508. )
  509. version = get_default_version(env)
  510. (host_platform, target_platform, _) = get_host_target(env)
  511. return find_batch_file(env, version, host_platform, target_platform)[0]
  512. def generate_cpp_hint_file(filename):
  513. if os.path.isfile(filename):
  514. # Don't overwrite an existing hint file since the user may have customized it.
  515. pass
  516. else:
  517. try:
  518. with open(filename, "w") as fd:
  519. fd.write("#define GDCLASS(m_class, m_inherits)\n")
  520. except OSError:
  521. print("Could not write cpp.hint file.")
  522. def glob_recursive(pattern, node="."):
  523. results = []
  524. for f in Glob(str(node) + "/*", source=True):
  525. if type(f) is Node.FS.Dir:
  526. results += glob_recursive(pattern, f)
  527. results += Glob(str(node) + "/" + pattern, source=True)
  528. return results
  529. def add_to_vs_project(env, sources):
  530. for x in sources:
  531. if type(x) == type(""):
  532. fname = env.File(x).path
  533. else:
  534. fname = env.File(x)[0].path
  535. pieces = fname.split(".")
  536. if len(pieces) > 0:
  537. basename = pieces[0]
  538. basename = basename.replace("\\\\", "/")
  539. if os.path.isfile(basename + ".h"):
  540. env.vs_incs += [basename + ".h"]
  541. elif os.path.isfile(basename + ".hpp"):
  542. env.vs_incs += [basename + ".hpp"]
  543. if os.path.isfile(basename + ".c"):
  544. env.vs_srcs += [basename + ".c"]
  545. elif os.path.isfile(basename + ".cpp"):
  546. env.vs_srcs += [basename + ".cpp"]
  547. def generate_vs_project(env, num_jobs):
  548. batch_file = find_visual_c_batch_file(env)
  549. if batch_file:
  550. class ModuleConfigs(Mapping):
  551. # This version information (Win32, x64, Debug, Release, Release_Debug seems to be
  552. # required for Visual Studio to understand that it needs to generate an NMAKE
  553. # project. Do not modify without knowing what you are doing.
  554. PLATFORMS = ["Win32", "x64"]
  555. PLATFORM_IDS = ["32", "64"]
  556. CONFIGURATIONS = ["debug", "release", "release_debug"]
  557. CONFIGURATION_IDS = ["tools", "opt", "opt.tools"]
  558. @staticmethod
  559. def for_every_variant(value):
  560. return [value for _ in range(len(ModuleConfigs.CONFIGURATIONS) * len(ModuleConfigs.PLATFORMS))]
  561. def __init__(self):
  562. shared_targets_array = []
  563. self.names = []
  564. self.arg_dict = {
  565. "variant": [],
  566. "runfile": shared_targets_array,
  567. "buildtarget": shared_targets_array,
  568. "cpppaths": [],
  569. "cppdefines": [],
  570. "cmdargs": [],
  571. }
  572. self.add_mode() # default
  573. def add_mode(
  574. self,
  575. name: str = "",
  576. includes: str = "",
  577. cli_args: str = "",
  578. defines=None,
  579. ):
  580. if defines is None:
  581. defines = []
  582. self.names.append(name)
  583. self.arg_dict["variant"] += [
  584. f'{config}{f"_[{name}]" if name else ""}|{platform}'
  585. for config in ModuleConfigs.CONFIGURATIONS
  586. for platform in ModuleConfigs.PLATFORMS
  587. ]
  588. self.arg_dict["runfile"] += [
  589. f'bin\\godot.windows.{config_id}.{plat_id}{f".{name}" if name else ""}.exe'
  590. for config_id in ModuleConfigs.CONFIGURATION_IDS
  591. for plat_id in ModuleConfigs.PLATFORM_IDS
  592. ]
  593. self.arg_dict["cpppaths"] += ModuleConfigs.for_every_variant(env["CPPPATH"] + [includes])
  594. self.arg_dict["cppdefines"] += ModuleConfigs.for_every_variant(env["CPPDEFINES"] + defines)
  595. self.arg_dict["cmdargs"] += ModuleConfigs.for_every_variant(cli_args)
  596. def build_commandline(self, commands):
  597. configuration_getter = (
  598. "$(Configuration"
  599. + "".join([f'.Replace("{name}", "")' for name in self.names[1:]])
  600. + '.Replace("_[]", "")'
  601. + ")"
  602. )
  603. common_build_prefix = [
  604. 'cmd /V /C set "plat=$(PlatformTarget)"',
  605. '(if "$(PlatformTarget)"=="x64" (set "plat=x86_amd64"))',
  606. 'set "tools=%s"' % env["tools"],
  607. f'(if "{configuration_getter}"=="release" (set "tools=no"))',
  608. 'call "' + batch_file + '" !plat!',
  609. ]
  610. # Windows allows us to have spaces in paths, so we need
  611. # to double quote off the directory. However, the path ends
  612. # in a backslash, so we need to remove this, lest it escape the
  613. # last double quote off, confusing MSBuild
  614. common_build_postfix = [
  615. "--directory=\"$(ProjectDir.TrimEnd('\\'))\"",
  616. "platform=windows",
  617. f"target={configuration_getter}",
  618. "progress=no",
  619. "tools=!tools!",
  620. "-j%s" % num_jobs,
  621. ]
  622. if env["tests"]:
  623. common_build_postfix.append("tests=yes")
  624. if env["custom_modules"]:
  625. common_build_postfix.append("custom_modules=%s" % env["custom_modules"])
  626. result = " ^& ".join(common_build_prefix + [" ".join([commands] + common_build_postfix)])
  627. return result
  628. # Mappings interface definitions
  629. def __iter__(self) -> Iterator[str]:
  630. for x in self.arg_dict:
  631. yield x
  632. def __len__(self) -> int:
  633. return len(self.names)
  634. def __getitem__(self, k: str):
  635. return self.arg_dict[k]
  636. add_to_vs_project(env, env.core_sources)
  637. add_to_vs_project(env, env.drivers_sources)
  638. add_to_vs_project(env, env.main_sources)
  639. add_to_vs_project(env, env.modules_sources)
  640. add_to_vs_project(env, env.scene_sources)
  641. add_to_vs_project(env, env.servers_sources)
  642. if env["tests"]:
  643. add_to_vs_project(env, env.tests_sources)
  644. add_to_vs_project(env, env.editor_sources)
  645. for header in glob_recursive("**/*.h"):
  646. env.vs_incs.append(str(header))
  647. module_configs = ModuleConfigs()
  648. import modules.mono.build_scripts.mono_reg_utils as mono_reg
  649. if env.get("module_mono_enabled"):
  650. mono_root = env.get("mono_prefix") or mono_reg.find_mono_root_dir(env["bits"])
  651. if mono_root:
  652. module_configs.add_mode(
  653. "mono",
  654. includes=os.path.join(mono_root, "include", "mono-2.0"),
  655. cli_args="module_mono_enabled=yes mono_glue=yes",
  656. defines=[("MONO_GLUE_ENABLED",)],
  657. )
  658. else:
  659. print("Mono installation directory not found. Generated project will not have build variants for Mono.")
  660. env["MSVSBUILDCOM"] = module_configs.build_commandline("scons")
  661. env["MSVSREBUILDCOM"] = module_configs.build_commandline("scons vsproj=yes")
  662. env["MSVSCLEANCOM"] = module_configs.build_commandline("scons --clean")
  663. if not env.get("MSVS"):
  664. env["MSVS"]["PROJECTSUFFIX"] = ".vcxproj"
  665. env["MSVS"]["SOLUTIONSUFFIX"] = ".sln"
  666. env.MSVSProject(
  667. target=["#godot" + env["MSVSPROJECTSUFFIX"]],
  668. incs=env.vs_incs,
  669. srcs=env.vs_srcs,
  670. auto_build_solution=1,
  671. **module_configs,
  672. )
  673. else:
  674. print("Could not locate Visual Studio batch file to set up the build environment. Not generating VS project.")
  675. def precious_program(env, program, sources, **args):
  676. program = env.ProgramOriginal(program, sources, **args)
  677. env.Precious(program)
  678. return program
  679. def add_shared_library(env, name, sources, **args):
  680. library = env.SharedLibrary(name, sources, **args)
  681. env.NoCache(library)
  682. return library
  683. def add_library(env, name, sources, **args):
  684. library = env.Library(name, sources, **args)
  685. env.NoCache(library)
  686. return library
  687. def add_program(env, name, sources, **args):
  688. program = env.Program(name, sources, **args)
  689. env.NoCache(program)
  690. return program
  691. def CommandNoCache(env, target, sources, command, **args):
  692. result = env.Command(target, sources, command, **args)
  693. env.NoCache(result)
  694. return result
  695. def Run(env, function, short_message, subprocess=True):
  696. output_print = short_message if not env["verbose"] else ""
  697. if not subprocess:
  698. return Action(function, output_print)
  699. else:
  700. return Action(run_in_subprocess(function), output_print)
  701. def detect_darwin_sdk_path(platform, env):
  702. sdk_name = ""
  703. if platform == "osx":
  704. sdk_name = "macosx"
  705. var_name = "MACOS_SDK_PATH"
  706. elif platform == "iphone":
  707. sdk_name = "iphoneos"
  708. var_name = "IPHONESDK"
  709. elif platform == "iphonesimulator":
  710. sdk_name = "iphonesimulator"
  711. var_name = "IPHONESDK"
  712. else:
  713. raise Exception("Invalid platform argument passed to detect_darwin_sdk_path")
  714. if not env[var_name]:
  715. try:
  716. sdk_path = subprocess.check_output(["xcrun", "--sdk", sdk_name, "--show-sdk-path"]).strip().decode("utf-8")
  717. if sdk_path:
  718. env[var_name] = sdk_path
  719. except (subprocess.CalledProcessError, OSError):
  720. print("Failed to find SDK path while running xcrun --sdk {} --show-sdk-path.".format(sdk_name))
  721. raise
  722. def is_vanilla_clang(env):
  723. if not using_clang(env):
  724. return False
  725. try:
  726. version = subprocess.check_output([env.subst(env["CXX"]), "--version"]).strip().decode("utf-8")
  727. except (subprocess.CalledProcessError, OSError):
  728. print("Couldn't parse CXX environment variable to infer compiler version.")
  729. return False
  730. return not version.startswith("Apple")
  731. def get_compiler_version(env):
  732. """
  733. Returns an array of version numbers as ints: [major, minor, patch].
  734. The return array should have at least two values (major, minor).
  735. """
  736. if not env.msvc:
  737. # Not using -dumpversion as some GCC distros only return major, and
  738. # Clang used to return hardcoded 4.2.1: # https://reviews.llvm.org/D56803
  739. try:
  740. version = subprocess.check_output([env.subst(env["CXX"]), "--version"]).strip().decode("utf-8")
  741. except (subprocess.CalledProcessError, OSError):
  742. print("Couldn't parse CXX environment variable to infer compiler version.")
  743. return None
  744. else: # TODO: Implement for MSVC
  745. return None
  746. match = re.search(
  747. "(?:(?<=version )|(?<=\) )|(?<=^))"
  748. "(?P<major>\d+)"
  749. "(?:\.(?P<minor>\d*))?"
  750. "(?:\.(?P<patch>\d*))?"
  751. "(?:-(?P<metadata1>[0-9a-zA-Z-]*))?"
  752. "(?:\+(?P<metadata2>[0-9a-zA-Z-]*))?"
  753. "(?: (?P<date>[0-9]{8}|[0-9]{6})(?![0-9a-zA-Z]))?",
  754. version,
  755. )
  756. if match is not None:
  757. return match.groupdict()
  758. else:
  759. return None
  760. def using_gcc(env):
  761. return "gcc" in os.path.basename(env["CC"])
  762. def using_clang(env):
  763. return "clang" in os.path.basename(env["CC"])
  764. def using_emcc(env):
  765. return "emcc" in os.path.basename(env["CC"])
  766. def show_progress(env):
  767. import sys
  768. from SCons.Script import Progress, Command, AlwaysBuild
  769. screen = sys.stdout
  770. # Progress reporting is not available in non-TTY environments since it
  771. # messes with the output (for example, when writing to a file)
  772. show_progress = env["progress"] and sys.stdout.isatty()
  773. node_count = 0
  774. node_count_max = 0
  775. node_count_interval = 1
  776. node_count_fname = str(env.Dir("#")) + "/.scons_node_count"
  777. import time, math
  778. class cache_progress:
  779. # The default is 1 GB cache and 12 hours half life
  780. def __init__(self, path=None, limit=1073741824, half_life=43200):
  781. self.path = path
  782. self.limit = limit
  783. self.exponent_scale = math.log(2) / half_life
  784. if env["verbose"] and path != None:
  785. screen.write(
  786. "Current cache limit is {} (used: {})\n".format(
  787. self.convert_size(limit), self.convert_size(self.get_size(path))
  788. )
  789. )
  790. self.delete(self.file_list())
  791. def __call__(self, node, *args, **kw):
  792. nonlocal node_count, node_count_max, node_count_interval, node_count_fname, show_progress
  793. if show_progress:
  794. # Print the progress percentage
  795. node_count += node_count_interval
  796. if node_count_max > 0 and node_count <= node_count_max:
  797. screen.write("\r[%3d%%] " % (node_count * 100 / node_count_max))
  798. screen.flush()
  799. elif node_count_max > 0 and node_count > node_count_max:
  800. screen.write("\r[100%] ")
  801. screen.flush()
  802. else:
  803. screen.write("\r[Initial build] ")
  804. screen.flush()
  805. def delete(self, files):
  806. if len(files) == 0:
  807. return
  808. if env["verbose"]:
  809. # Utter something
  810. screen.write("\rPurging %d %s from cache...\n" % (len(files), len(files) > 1 and "files" or "file"))
  811. [os.remove(f) for f in files]
  812. def file_list(self):
  813. if self.path is None:
  814. # Nothing to do
  815. return []
  816. # Gather a list of (filename, (size, atime)) within the
  817. # cache directory
  818. file_stat = [(x, os.stat(x)[6:8]) for x in glob.glob(os.path.join(self.path, "*", "*"))]
  819. if file_stat == []:
  820. # Nothing to do
  821. return []
  822. # Weight the cache files by size (assumed to be roughly
  823. # proportional to the recompilation time) times an exponential
  824. # decay since the ctime, and return a list with the entries
  825. # (filename, size, weight).
  826. current_time = time.time()
  827. file_stat = [(x[0], x[1][0], (current_time - x[1][1])) for x in file_stat]
  828. # Sort by the most recently accessed files (most sensible to keep) first
  829. file_stat.sort(key=lambda x: x[2])
  830. # Search for the first entry where the storage limit is
  831. # reached
  832. sum, mark = 0, None
  833. for i, x in enumerate(file_stat):
  834. sum += x[1]
  835. if sum > self.limit:
  836. mark = i
  837. break
  838. if mark is None:
  839. return []
  840. else:
  841. return [x[0] for x in file_stat[mark:]]
  842. def convert_size(self, size_bytes):
  843. if size_bytes == 0:
  844. return "0 bytes"
  845. size_name = ("bytes", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB")
  846. i = int(math.floor(math.log(size_bytes, 1024)))
  847. p = math.pow(1024, i)
  848. s = round(size_bytes / p, 2)
  849. return "%s %s" % (int(s) if i == 0 else s, size_name[i])
  850. def get_size(self, start_path="."):
  851. total_size = 0
  852. for dirpath, dirnames, filenames in os.walk(start_path):
  853. for f in filenames:
  854. fp = os.path.join(dirpath, f)
  855. total_size += os.path.getsize(fp)
  856. return total_size
  857. def progress_finish(target, source, env):
  858. nonlocal node_count, progressor
  859. try:
  860. with open(node_count_fname, "w") as f:
  861. f.write("%d\n" % node_count)
  862. progressor.delete(progressor.file_list())
  863. except Exception:
  864. pass
  865. try:
  866. with open(node_count_fname) as f:
  867. node_count_max = int(f.readline())
  868. except Exception:
  869. pass
  870. cache_directory = os.environ.get("SCONS_CACHE")
  871. # Simple cache pruning, attached to SCons' progress callback. Trim the
  872. # cache directory to a size not larger than cache_limit.
  873. cache_limit = float(os.getenv("SCONS_CACHE_LIMIT", 1024)) * 1024 * 1024
  874. progressor = cache_progress(cache_directory, cache_limit)
  875. Progress(progressor, interval=node_count_interval)
  876. progress_finish_command = Command("progress_finish", [], progress_finish)
  877. AlwaysBuild(progress_finish_command)
  878. def dump(env):
  879. # Dumps latest build information for debugging purposes and external tools.
  880. from json import dump
  881. def non_serializable(obj):
  882. return "<<non-serializable: %s>>" % (type(obj).__qualname__)
  883. with open(".scons_env.json", "w") as f:
  884. dump(env.Dictionary(), f, indent=4, default=non_serializable)