methods.py 40 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057
  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('#define VERSION_DOCS_BRANCH "' + str(version.docs) + '"\n')
  94. f.write('#define VERSION_DOCS_URL "https://docs.godotengine.org/en/" VERSION_DOCS_BRANCH\n')
  95. f.write("#endif // VERSION_GENERATED_GEN_H\n")
  96. f.close()
  97. # NOTE: It is safe to generate this file here, since this is still executed serially
  98. fhash = open("core/version_hash.gen.cpp", "w")
  99. fhash.write("/* THIS FILE IS GENERATED DO NOT EDIT */\n")
  100. fhash.write('#include "core/version.h"\n')
  101. githash = ""
  102. gitfolder = ".git"
  103. if os.path.isfile(".git"):
  104. module_folder = open(".git", "r").readline().strip()
  105. if module_folder.startswith("gitdir: "):
  106. gitfolder = module_folder[8:]
  107. if os.path.isfile(os.path.join(gitfolder, "HEAD")):
  108. head = open(os.path.join(gitfolder, "HEAD"), "r", encoding="utf8").readline().strip()
  109. if head.startswith("ref: "):
  110. head = os.path.join(gitfolder, head[5:])
  111. if os.path.isfile(head):
  112. githash = open(head, "r").readline().strip()
  113. else:
  114. githash = head
  115. fhash.write('const char *const VERSION_HASH = "' + githash + '";\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["blue"] = "\033[0;94m"
  372. colors["bold_blue"] = "\033[1;94m"
  373. colors["reset"] = "\033[0m"
  374. else:
  375. colors["blue"] = ""
  376. colors["bold_blue"] = ""
  377. colors["reset"] = ""
  378. # There is a space before "..." to ensure that source file names can be
  379. # Ctrl + clicked in the VS Code terminal.
  380. compile_source_message = "{}Compiling {}$SOURCE{} ...{}".format(
  381. colors["blue"], colors["bold_blue"], colors["blue"], colors["reset"]
  382. )
  383. java_compile_source_message = "{}Compiling {}$SOURCE{} ...{}".format(
  384. colors["blue"], colors["bold_blue"], colors["blue"], colors["reset"]
  385. )
  386. compile_shared_source_message = "{}Compiling shared {}$SOURCE{} ...{}".format(
  387. colors["blue"], colors["bold_blue"], colors["blue"], colors["reset"]
  388. )
  389. link_program_message = "{}Linking Program {}$TARGET{} ...{}".format(
  390. colors["blue"], colors["bold_blue"], colors["blue"], colors["reset"]
  391. )
  392. link_library_message = "{}Linking Static Library {}$TARGET{} ...{}".format(
  393. colors["blue"], colors["bold_blue"], colors["blue"], colors["reset"]
  394. )
  395. ranlib_library_message = "{}Ranlib Library {}$TARGET{} ...{}".format(
  396. colors["blue"], colors["bold_blue"], colors["blue"], colors["reset"]
  397. )
  398. link_shared_library_message = "{}Linking Shared Library {}$TARGET{} ...{}".format(
  399. colors["blue"], colors["bold_blue"], colors["blue"], colors["reset"]
  400. )
  401. java_library_message = "{}Creating Java Archive {}$TARGET{} ...{}".format(
  402. colors["blue"], colors["bold_blue"], colors["blue"], colors["reset"]
  403. )
  404. env.Append(CXXCOMSTR=[compile_source_message])
  405. env.Append(CCCOMSTR=[compile_source_message])
  406. env.Append(SHCCCOMSTR=[compile_shared_source_message])
  407. env.Append(SHCXXCOMSTR=[compile_shared_source_message])
  408. env.Append(ARCOMSTR=[link_library_message])
  409. env.Append(RANLIBCOMSTR=[ranlib_library_message])
  410. env.Append(SHLINKCOMSTR=[link_shared_library_message])
  411. env.Append(LINKCOMSTR=[link_program_message])
  412. env.Append(JARCOMSTR=[java_library_message])
  413. env.Append(JAVACCOMSTR=[java_compile_source_message])
  414. def detect_visual_c_compiler_version(tools_env):
  415. # tools_env is the variable scons uses to call tools that execute tasks, SCons's env['ENV'] that executes tasks...
  416. # (see the SCons documentation for more information on what it does)...
  417. # in order for this function to be well encapsulated i choose to force it to receive SCons's TOOLS env (env['ENV']
  418. # and not scons setup environment (env)... so make sure you call the right environment on it or it will fail to detect
  419. # the proper vc version that will be called
  420. # There is no flag to give to visual c compilers to set the architecture, i.e. scons bits argument (32,64,ARM etc)
  421. # There are many different cl.exe files that are run, and each one compiles & links to a different architecture
  422. # As far as I know, the only way to figure out what compiler will be run when Scons calls cl.exe via Program()
  423. # is to check the PATH variable and figure out which one will be called first. Code below does that and returns:
  424. # the following string values:
  425. # "" Compiler not detected
  426. # "amd64" Native 64 bit compiler
  427. # "amd64_x86" 64 bit Cross Compiler for 32 bit
  428. # "x86" Native 32 bit compiler
  429. # "x86_amd64" 32 bit Cross Compiler for 64 bit
  430. # There are other architectures, but Godot does not support them currently, so this function does not detect arm/amd64_arm
  431. # and similar architectures/compilers
  432. # Set chosen compiler to "not detected"
  433. vc_chosen_compiler_index = -1
  434. vc_chosen_compiler_str = ""
  435. # Start with Pre VS 2017 checks which uses VCINSTALLDIR:
  436. if "VCINSTALLDIR" in tools_env:
  437. # print("Checking VCINSTALLDIR")
  438. # find() works with -1 so big ifs below are needed... the simplest solution, in fact
  439. # First test if amd64 and amd64_x86 compilers are present in the path
  440. vc_amd64_compiler_detection_index = tools_env["PATH"].find(tools_env["VCINSTALLDIR"] + "BIN\\amd64;")
  441. if vc_amd64_compiler_detection_index > -1:
  442. vc_chosen_compiler_index = vc_amd64_compiler_detection_index
  443. vc_chosen_compiler_str = "amd64"
  444. vc_amd64_x86_compiler_detection_index = tools_env["PATH"].find(tools_env["VCINSTALLDIR"] + "BIN\\amd64_x86;")
  445. if vc_amd64_x86_compiler_detection_index > -1 and (
  446. vc_chosen_compiler_index == -1 or vc_chosen_compiler_index > vc_amd64_x86_compiler_detection_index
  447. ):
  448. vc_chosen_compiler_index = vc_amd64_x86_compiler_detection_index
  449. vc_chosen_compiler_str = "amd64_x86"
  450. # Now check the 32 bit compilers
  451. vc_x86_compiler_detection_index = tools_env["PATH"].find(tools_env["VCINSTALLDIR"] + "BIN;")
  452. if vc_x86_compiler_detection_index > -1 and (
  453. vc_chosen_compiler_index == -1 or vc_chosen_compiler_index > vc_x86_compiler_detection_index
  454. ):
  455. vc_chosen_compiler_index = vc_x86_compiler_detection_index
  456. vc_chosen_compiler_str = "x86"
  457. vc_x86_amd64_compiler_detection_index = tools_env["PATH"].find(tools_env["VCINSTALLDIR"] + "BIN\\x86_amd64;")
  458. if vc_x86_amd64_compiler_detection_index > -1 and (
  459. vc_chosen_compiler_index == -1 or vc_chosen_compiler_index > vc_x86_amd64_compiler_detection_index
  460. ):
  461. vc_chosen_compiler_index = vc_x86_amd64_compiler_detection_index
  462. vc_chosen_compiler_str = "x86_amd64"
  463. # and for VS 2017 and newer we check VCTOOLSINSTALLDIR:
  464. if "VCTOOLSINSTALLDIR" in tools_env:
  465. # Newer versions have a different path available
  466. vc_amd64_compiler_detection_index = (
  467. tools_env["PATH"].upper().find(tools_env["VCTOOLSINSTALLDIR"].upper() + "BIN\\HOSTX64\\X64;")
  468. )
  469. if vc_amd64_compiler_detection_index > -1:
  470. vc_chosen_compiler_index = vc_amd64_compiler_detection_index
  471. vc_chosen_compiler_str = "amd64"
  472. vc_amd64_x86_compiler_detection_index = (
  473. tools_env["PATH"].upper().find(tools_env["VCTOOLSINSTALLDIR"].upper() + "BIN\\HOSTX64\\X86;")
  474. )
  475. if vc_amd64_x86_compiler_detection_index > -1 and (
  476. vc_chosen_compiler_index == -1 or vc_chosen_compiler_index > vc_amd64_x86_compiler_detection_index
  477. ):
  478. vc_chosen_compiler_index = vc_amd64_x86_compiler_detection_index
  479. vc_chosen_compiler_str = "amd64_x86"
  480. vc_x86_compiler_detection_index = (
  481. tools_env["PATH"].upper().find(tools_env["VCTOOLSINSTALLDIR"].upper() + "BIN\\HOSTX86\\X86;")
  482. )
  483. if vc_x86_compiler_detection_index > -1 and (
  484. vc_chosen_compiler_index == -1 or vc_chosen_compiler_index > vc_x86_compiler_detection_index
  485. ):
  486. vc_chosen_compiler_index = vc_x86_compiler_detection_index
  487. vc_chosen_compiler_str = "x86"
  488. vc_x86_amd64_compiler_detection_index = (
  489. tools_env["PATH"].upper().find(tools_env["VCTOOLSINSTALLDIR"].upper() + "BIN\\HOSTX86\\X64;")
  490. )
  491. if vc_x86_amd64_compiler_detection_index > -1 and (
  492. vc_chosen_compiler_index == -1 or vc_chosen_compiler_index > vc_x86_amd64_compiler_detection_index
  493. ):
  494. vc_chosen_compiler_index = vc_x86_amd64_compiler_detection_index
  495. vc_chosen_compiler_str = "x86_amd64"
  496. return vc_chosen_compiler_str
  497. def find_visual_c_batch_file(env):
  498. from SCons.Tool.MSCommon.vc import (
  499. get_default_version,
  500. get_host_target,
  501. find_batch_file,
  502. )
  503. version = get_default_version(env)
  504. (host_platform, target_platform, _) = get_host_target(env)
  505. return find_batch_file(env, version, host_platform, target_platform)[0]
  506. def generate_cpp_hint_file(filename):
  507. if os.path.isfile(filename):
  508. # Don't overwrite an existing hint file since the user may have customized it.
  509. pass
  510. else:
  511. try:
  512. with open(filename, "w") as fd:
  513. fd.write("#define GDCLASS(m_class, m_inherits)\n")
  514. except OSError:
  515. print("Could not write cpp.hint file.")
  516. def glob_recursive(pattern, node="."):
  517. results = []
  518. for f in Glob(str(node) + "/*", source=True):
  519. if type(f) is Node.FS.Dir:
  520. results += glob_recursive(pattern, f)
  521. results += Glob(str(node) + "/" + pattern, source=True)
  522. return results
  523. def add_to_vs_project(env, sources):
  524. for x in sources:
  525. if type(x) == type(""):
  526. fname = env.File(x).path
  527. else:
  528. fname = env.File(x)[0].path
  529. pieces = fname.split(".")
  530. if len(pieces) > 0:
  531. basename = pieces[0]
  532. basename = basename.replace("\\\\", "/")
  533. if os.path.isfile(basename + ".h"):
  534. env.vs_incs += [basename + ".h"]
  535. elif os.path.isfile(basename + ".hpp"):
  536. env.vs_incs += [basename + ".hpp"]
  537. if os.path.isfile(basename + ".c"):
  538. env.vs_srcs += [basename + ".c"]
  539. elif os.path.isfile(basename + ".cpp"):
  540. env.vs_srcs += [basename + ".cpp"]
  541. def generate_vs_project(env, num_jobs):
  542. batch_file = find_visual_c_batch_file(env)
  543. if batch_file:
  544. class ModuleConfigs(Mapping):
  545. # This version information (Win32, x64, Debug, Release, Release_Debug seems to be
  546. # required for Visual Studio to understand that it needs to generate an NMAKE
  547. # project. Do not modify without knowing what you are doing.
  548. PLATFORMS = ["Win32", "x64"]
  549. PLATFORM_IDS = ["32", "64"]
  550. CONFIGURATIONS = ["debug", "release", "release_debug"]
  551. CONFIGURATION_IDS = ["tools", "opt", "opt.tools"]
  552. @staticmethod
  553. def for_every_variant(value):
  554. return [value for _ in range(len(ModuleConfigs.CONFIGURATIONS) * len(ModuleConfigs.PLATFORMS))]
  555. def __init__(self):
  556. shared_targets_array = []
  557. self.names = []
  558. self.arg_dict = {
  559. "variant": [],
  560. "runfile": shared_targets_array,
  561. "buildtarget": shared_targets_array,
  562. "cpppaths": [],
  563. "cppdefines": [],
  564. "cmdargs": [],
  565. }
  566. self.add_mode() # default
  567. def add_mode(
  568. self,
  569. name: str = "",
  570. includes: str = "",
  571. cli_args: str = "",
  572. defines=None,
  573. ):
  574. if defines is None:
  575. defines = []
  576. self.names.append(name)
  577. self.arg_dict["variant"] += [
  578. f'{config}{f"_[{name}]" if name else ""}|{platform}'
  579. for config in ModuleConfigs.CONFIGURATIONS
  580. for platform in ModuleConfigs.PLATFORMS
  581. ]
  582. self.arg_dict["runfile"] += [
  583. f'bin\\godot.windows.{config_id}.{plat_id}{f".{name}" if name else ""}.exe'
  584. for config_id in ModuleConfigs.CONFIGURATION_IDS
  585. for plat_id in ModuleConfigs.PLATFORM_IDS
  586. ]
  587. self.arg_dict["cpppaths"] += ModuleConfigs.for_every_variant(env["CPPPATH"] + [includes])
  588. self.arg_dict["cppdefines"] += ModuleConfigs.for_every_variant(env["CPPDEFINES"] + defines)
  589. self.arg_dict["cmdargs"] += ModuleConfigs.for_every_variant(cli_args)
  590. def build_commandline(self, commands):
  591. configuration_getter = (
  592. "$(Configuration"
  593. + "".join([f'.Replace("{name}", "")' for name in self.names[1:]])
  594. + '.Replace("_[]", "")'
  595. + ")"
  596. )
  597. common_build_prefix = [
  598. 'cmd /V /C set "plat=$(PlatformTarget)"',
  599. '(if "$(PlatformTarget)"=="x64" (set "plat=x86_amd64"))',
  600. 'set "tools=%s"' % env["tools"],
  601. f'(if "{configuration_getter}"=="release" (set "tools=no"))',
  602. 'call "' + batch_file + '" !plat!',
  603. ]
  604. # Windows allows us to have spaces in paths, so we need
  605. # to double quote off the directory. However, the path ends
  606. # in a backslash, so we need to remove this, lest it escape the
  607. # last double quote off, confusing MSBuild
  608. common_build_postfix = [
  609. "--directory=\"$(ProjectDir.TrimEnd('\\'))\"",
  610. "platform=windows",
  611. f"target={configuration_getter}",
  612. "progress=no",
  613. "tools=!tools!",
  614. "-j%s" % num_jobs,
  615. ]
  616. if env["tests"]:
  617. common_build_postfix.append("tests=yes")
  618. if env["custom_modules"]:
  619. common_build_postfix.append("custom_modules=%s" % env["custom_modules"])
  620. result = " ^& ".join(common_build_prefix + [" ".join([commands] + common_build_postfix)])
  621. return result
  622. # Mappings interface definitions
  623. def __iter__(self) -> Iterator[str]:
  624. for x in self.arg_dict:
  625. yield x
  626. def __len__(self) -> int:
  627. return len(self.names)
  628. def __getitem__(self, k: str):
  629. return self.arg_dict[k]
  630. add_to_vs_project(env, env.core_sources)
  631. add_to_vs_project(env, env.drivers_sources)
  632. add_to_vs_project(env, env.main_sources)
  633. add_to_vs_project(env, env.modules_sources)
  634. add_to_vs_project(env, env.scene_sources)
  635. add_to_vs_project(env, env.servers_sources)
  636. if env["tests"]:
  637. add_to_vs_project(env, env.tests_sources)
  638. add_to_vs_project(env, env.editor_sources)
  639. for header in glob_recursive("**/*.h"):
  640. env.vs_incs.append(str(header))
  641. module_configs = ModuleConfigs()
  642. if env.get("module_mono_enabled"):
  643. import modules.mono.build_scripts.mono_reg_utils as mono_reg
  644. mono_root = env.get("mono_prefix") or mono_reg.find_mono_root_dir(env["bits"])
  645. if mono_root:
  646. module_configs.add_mode(
  647. "mono",
  648. includes=os.path.join(mono_root, "include", "mono-2.0"),
  649. cli_args="module_mono_enabled=yes mono_glue=yes",
  650. defines=[("MONO_GLUE_ENABLED",)],
  651. )
  652. else:
  653. print("Mono installation directory not found. Generated project will not have build variants for Mono.")
  654. env["MSVSBUILDCOM"] = module_configs.build_commandline("scons")
  655. env["MSVSREBUILDCOM"] = module_configs.build_commandline("scons vsproj=yes")
  656. env["MSVSCLEANCOM"] = module_configs.build_commandline("scons --clean")
  657. if not env.get("MSVS"):
  658. env["MSVS"]["PROJECTSUFFIX"] = ".vcxproj"
  659. env["MSVS"]["SOLUTIONSUFFIX"] = ".sln"
  660. env.MSVSProject(
  661. target=["#godot" + env["MSVSPROJECTSUFFIX"]],
  662. incs=env.vs_incs,
  663. srcs=env.vs_srcs,
  664. auto_build_solution=1,
  665. **module_configs,
  666. )
  667. else:
  668. print("Could not locate Visual Studio batch file to set up the build environment. Not generating VS project.")
  669. def precious_program(env, program, sources, **args):
  670. program = env.ProgramOriginal(program, sources, **args)
  671. env.Precious(program)
  672. return program
  673. def add_shared_library(env, name, sources, **args):
  674. library = env.SharedLibrary(name, sources, **args)
  675. env.NoCache(library)
  676. return library
  677. def add_library(env, name, sources, **args):
  678. library = env.Library(name, sources, **args)
  679. env.NoCache(library)
  680. return library
  681. def add_program(env, name, sources, **args):
  682. program = env.Program(name, sources, **args)
  683. env.NoCache(program)
  684. return program
  685. def CommandNoCache(env, target, sources, command, **args):
  686. result = env.Command(target, sources, command, **args)
  687. env.NoCache(result)
  688. return result
  689. def Run(env, function, short_message, subprocess=True):
  690. output_print = short_message if not env["verbose"] else ""
  691. if not subprocess:
  692. return Action(function, output_print)
  693. else:
  694. return Action(run_in_subprocess(function), output_print)
  695. def detect_darwin_sdk_path(platform, env):
  696. sdk_name = ""
  697. if platform == "osx":
  698. sdk_name = "macosx"
  699. var_name = "MACOS_SDK_PATH"
  700. elif platform == "iphone":
  701. sdk_name = "iphoneos"
  702. var_name = "IPHONESDK"
  703. elif platform == "iphonesimulator":
  704. sdk_name = "iphonesimulator"
  705. var_name = "IPHONESDK"
  706. else:
  707. raise Exception("Invalid platform argument passed to detect_darwin_sdk_path")
  708. if not env[var_name]:
  709. try:
  710. sdk_path = subprocess.check_output(["xcrun", "--sdk", sdk_name, "--show-sdk-path"]).strip().decode("utf-8")
  711. if sdk_path:
  712. env[var_name] = sdk_path
  713. except (subprocess.CalledProcessError, OSError):
  714. print("Failed to find SDK path while running xcrun --sdk {} --show-sdk-path.".format(sdk_name))
  715. raise
  716. def is_vanilla_clang(env):
  717. if not using_clang(env):
  718. return False
  719. try:
  720. version = subprocess.check_output([env.subst(env["CXX"]), "--version"]).strip().decode("utf-8")
  721. except (subprocess.CalledProcessError, OSError):
  722. print("Couldn't parse CXX environment variable to infer compiler version.")
  723. return False
  724. return not version.startswith("Apple")
  725. def get_compiler_version(env):
  726. """
  727. Returns an array of version numbers as ints: [major, minor, patch].
  728. The return array should have at least two values (major, minor).
  729. """
  730. if not env.msvc:
  731. # Not using -dumpversion as some GCC distros only return major, and
  732. # Clang used to return hardcoded 4.2.1: # https://reviews.llvm.org/D56803
  733. try:
  734. version = subprocess.check_output([env.subst(env["CXX"]), "--version"]).strip().decode("utf-8")
  735. except (subprocess.CalledProcessError, OSError):
  736. print("Couldn't parse CXX environment variable to infer compiler version.")
  737. return None
  738. else: # TODO: Implement for MSVC
  739. return None
  740. match = re.search(
  741. "(?:(?<=version )|(?<=\) )|(?<=^))"
  742. "(?P<major>\d+)"
  743. "(?:\.(?P<minor>\d*))?"
  744. "(?:\.(?P<patch>\d*))?"
  745. "(?:-(?P<metadata1>[0-9a-zA-Z-]*))?"
  746. "(?:\+(?P<metadata2>[0-9a-zA-Z-]*))?"
  747. "(?: (?P<date>[0-9]{8}|[0-9]{6})(?![0-9a-zA-Z]))?",
  748. version,
  749. )
  750. if match is not None:
  751. return match.groupdict()
  752. else:
  753. return None
  754. def using_gcc(env):
  755. return "gcc" in os.path.basename(env["CC"])
  756. def using_clang(env):
  757. return "clang" in os.path.basename(env["CC"])
  758. def using_emcc(env):
  759. return "emcc" in os.path.basename(env["CC"])
  760. def show_progress(env):
  761. import sys
  762. from SCons.Script import Progress, Command, AlwaysBuild
  763. screen = sys.stdout
  764. # Progress reporting is not available in non-TTY environments since it
  765. # messes with the output (for example, when writing to a file)
  766. show_progress = env["progress"] and sys.stdout.isatty()
  767. node_count = 0
  768. node_count_max = 0
  769. node_count_interval = 1
  770. node_count_fname = str(env.Dir("#")) + "/.scons_node_count"
  771. import time, math
  772. class cache_progress:
  773. # The default is 1 GB cache and 12 hours half life
  774. def __init__(self, path=None, limit=1073741824, half_life=43200):
  775. self.path = path
  776. self.limit = limit
  777. self.exponent_scale = math.log(2) / half_life
  778. if env["verbose"] and path != None:
  779. screen.write(
  780. "Current cache limit is {} (used: {})\n".format(
  781. self.convert_size(limit), self.convert_size(self.get_size(path))
  782. )
  783. )
  784. self.delete(self.file_list())
  785. def __call__(self, node, *args, **kw):
  786. nonlocal node_count, node_count_max, node_count_interval, node_count_fname, show_progress
  787. if show_progress:
  788. # Print the progress percentage
  789. node_count += node_count_interval
  790. if node_count_max > 0 and node_count <= node_count_max:
  791. screen.write("\r[%3d%%] " % (node_count * 100 / node_count_max))
  792. screen.flush()
  793. elif node_count_max > 0 and node_count > node_count_max:
  794. screen.write("\r[100%] ")
  795. screen.flush()
  796. else:
  797. screen.write("\r[Initial build] ")
  798. screen.flush()
  799. def delete(self, files):
  800. if len(files) == 0:
  801. return
  802. if env["verbose"]:
  803. # Utter something
  804. screen.write("\rPurging %d %s from cache...\n" % (len(files), len(files) > 1 and "files" or "file"))
  805. [os.remove(f) for f in files]
  806. def file_list(self):
  807. if self.path is None:
  808. # Nothing to do
  809. return []
  810. # Gather a list of (filename, (size, atime)) within the
  811. # cache directory
  812. file_stat = [(x, os.stat(x)[6:8]) for x in glob.glob(os.path.join(self.path, "*", "*"))]
  813. if file_stat == []:
  814. # Nothing to do
  815. return []
  816. # Weight the cache files by size (assumed to be roughly
  817. # proportional to the recompilation time) times an exponential
  818. # decay since the ctime, and return a list with the entries
  819. # (filename, size, weight).
  820. current_time = time.time()
  821. file_stat = [(x[0], x[1][0], (current_time - x[1][1])) for x in file_stat]
  822. # Sort by the most recently accessed files (most sensible to keep) first
  823. file_stat.sort(key=lambda x: x[2])
  824. # Search for the first entry where the storage limit is
  825. # reached
  826. sum, mark = 0, None
  827. for i, x in enumerate(file_stat):
  828. sum += x[1]
  829. if sum > self.limit:
  830. mark = i
  831. break
  832. if mark is None:
  833. return []
  834. else:
  835. return [x[0] for x in file_stat[mark:]]
  836. def convert_size(self, size_bytes):
  837. if size_bytes == 0:
  838. return "0 bytes"
  839. size_name = ("bytes", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB")
  840. i = int(math.floor(math.log(size_bytes, 1024)))
  841. p = math.pow(1024, i)
  842. s = round(size_bytes / p, 2)
  843. return "%s %s" % (int(s) if i == 0 else s, size_name[i])
  844. def get_size(self, start_path="."):
  845. total_size = 0
  846. for dirpath, dirnames, filenames in os.walk(start_path):
  847. for f in filenames:
  848. fp = os.path.join(dirpath, f)
  849. total_size += os.path.getsize(fp)
  850. return total_size
  851. def progress_finish(target, source, env):
  852. nonlocal node_count, progressor
  853. try:
  854. with open(node_count_fname, "w") as f:
  855. f.write("%d\n" % node_count)
  856. progressor.delete(progressor.file_list())
  857. except Exception:
  858. pass
  859. try:
  860. with open(node_count_fname) as f:
  861. node_count_max = int(f.readline())
  862. except Exception:
  863. pass
  864. cache_directory = os.environ.get("SCONS_CACHE")
  865. # Simple cache pruning, attached to SCons' progress callback. Trim the
  866. # cache directory to a size not larger than cache_limit.
  867. cache_limit = float(os.getenv("SCONS_CACHE_LIMIT", 1024)) * 1024 * 1024
  868. progressor = cache_progress(cache_directory, cache_limit)
  869. Progress(progressor, interval=node_count_interval)
  870. progress_finish_command = Command("progress_finish", [], progress_finish)
  871. AlwaysBuild(progress_finish_command)
  872. def dump(env):
  873. # Dumps latest build information for debugging purposes and external tools.
  874. from json import dump
  875. def non_serializable(obj):
  876. return "<<non-serializable: %s>>" % (type(obj).__qualname__)
  877. with open(".scons_env.json", "w") as f:
  878. dump(env.Dictionary(), f, indent=4, default=non_serializable)