gles_builders.py 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744
  1. """Functions used to generate source files during build time
  2. All such functions are invoked in a subprocess on Windows to prevent build flakiness.
  3. """
  4. from platform_methods import subprocess_main
  5. class LegacyGLHeaderStruct:
  6. def __init__(self):
  7. self.vertex_lines = []
  8. self.fragment_lines = []
  9. self.uniforms = []
  10. self.attributes = []
  11. self.feedbacks = []
  12. self.fbos = []
  13. self.conditionals = []
  14. self.enums = {}
  15. self.texunits = []
  16. self.texunit_names = []
  17. self.ubos = []
  18. self.ubo_names = []
  19. self.vertex_included_files = []
  20. self.fragment_included_files = []
  21. self.reading = ""
  22. self.line_offset = 0
  23. self.vertex_offset = 0
  24. self.fragment_offset = 0
  25. def include_file_in_legacygl_header(filename, header_data, depth):
  26. fs = open(filename, "r")
  27. line = fs.readline()
  28. while line:
  29. if line.find("[vertex]") != -1:
  30. header_data.reading = "vertex"
  31. line = fs.readline()
  32. header_data.line_offset += 1
  33. header_data.vertex_offset = header_data.line_offset
  34. continue
  35. if line.find("[fragment]") != -1:
  36. header_data.reading = "fragment"
  37. line = fs.readline()
  38. header_data.line_offset += 1
  39. header_data.fragment_offset = header_data.line_offset
  40. continue
  41. while line.find("#include ") != -1:
  42. includeline = line.replace("#include ", "").strip()[1:-1]
  43. import os.path
  44. included_file = os.path.relpath(os.path.dirname(filename) + "/" + includeline)
  45. if not included_file in header_data.vertex_included_files and header_data.reading == "vertex":
  46. header_data.vertex_included_files += [included_file]
  47. if include_file_in_legacygl_header(included_file, header_data, depth + 1) is None:
  48. print("Error in file '" + filename + "': #include " + includeline + "could not be found!")
  49. elif not included_file in header_data.fragment_included_files and header_data.reading == "fragment":
  50. header_data.fragment_included_files += [included_file]
  51. if include_file_in_legacygl_header(included_file, header_data, depth + 1) is None:
  52. print("Error in file '" + filename + "': #include " + includeline + "could not be found!")
  53. line = fs.readline()
  54. if line.find("#ifdef ") != -1:
  55. if line.find("#ifdef ") != -1:
  56. ifdefline = line.replace("#ifdef ", "").strip()
  57. if line.find("_EN_") != -1:
  58. enumbase = ifdefline[: ifdefline.find("_EN_")]
  59. ifdefline = ifdefline.replace("_EN_", "_")
  60. line = line.replace("_EN_", "_")
  61. if enumbase not in header_data.enums:
  62. header_data.enums[enumbase] = []
  63. if ifdefline not in header_data.enums[enumbase]:
  64. header_data.enums[enumbase].append(ifdefline)
  65. elif not ifdefline in header_data.conditionals:
  66. header_data.conditionals += [ifdefline]
  67. if line.find("uniform") != -1 and line.lower().find("texunit:") != -1:
  68. # texture unit
  69. texunitstr = line[line.find(":") + 1 :].strip()
  70. if texunitstr == "auto":
  71. texunit = "-1"
  72. else:
  73. texunit = str(int(texunitstr))
  74. uline = line[: line.lower().find("//")]
  75. uline = uline.replace("uniform", "")
  76. uline = uline.replace("highp", "")
  77. uline = uline.replace(";", "")
  78. lines = uline.split(",")
  79. for x in lines:
  80. x = x.strip()
  81. x = x[x.rfind(" ") + 1 :]
  82. if x.find("[") != -1:
  83. # unfiorm array
  84. x = x[: x.find("[")]
  85. if not x in header_data.texunit_names:
  86. header_data.texunits += [(x, texunit)]
  87. header_data.texunit_names += [x]
  88. elif line.find("uniform") != -1 and line.lower().find("ubo:") != -1:
  89. # uniform buffer object
  90. ubostr = line[line.find(":") + 1 :].strip()
  91. ubo = str(int(ubostr))
  92. uline = line[: line.lower().find("//")]
  93. uline = uline[uline.find("uniform") + len("uniform") :]
  94. uline = uline.replace("highp", "")
  95. uline = uline.replace(";", "")
  96. uline = uline.replace("{", "").strip()
  97. lines = uline.split(",")
  98. for x in lines:
  99. x = x.strip()
  100. x = x[x.rfind(" ") + 1 :]
  101. if x.find("[") != -1:
  102. # unfiorm array
  103. x = x[: x.find("[")]
  104. if not x in header_data.ubo_names:
  105. header_data.ubos += [(x, ubo)]
  106. header_data.ubo_names += [x]
  107. elif line.find("uniform") != -1 and line.find("{") == -1 and line.find(";") != -1:
  108. uline = line.replace("uniform", "")
  109. uline = uline.replace(";", "")
  110. lines = uline.split(",")
  111. for x in lines:
  112. x = x.strip()
  113. x = x[x.rfind(" ") + 1 :]
  114. if x.find("[") != -1:
  115. # unfiorm array
  116. x = x[: x.find("[")]
  117. if not x in header_data.uniforms:
  118. header_data.uniforms += [x]
  119. if line.strip().find("attribute ") == 0 and line.find("attrib:") != -1:
  120. uline = line.replace("in ", "")
  121. uline = uline.replace("attribute ", "")
  122. uline = uline.replace("highp ", "")
  123. uline = uline.replace(";", "")
  124. uline = uline[uline.find(" ") :].strip()
  125. if uline.find("//") != -1:
  126. name, bind = uline.split("//")
  127. if bind.find("attrib:") != -1:
  128. name = name.strip()
  129. bind = bind.replace("attrib:", "").strip()
  130. header_data.attributes += [(name, bind)]
  131. if line.strip().find("out ") == 0 and line.find("tfb:") != -1:
  132. uline = line.replace("out ", "")
  133. uline = uline.replace("highp ", "")
  134. uline = uline.replace(";", "")
  135. uline = uline[uline.find(" ") :].strip()
  136. if uline.find("//") != -1:
  137. name, bind = uline.split("//")
  138. if bind.find("tfb:") != -1:
  139. name = name.strip()
  140. bind = bind.replace("tfb:", "").strip()
  141. header_data.feedbacks += [(name, bind)]
  142. line = line.replace("\r", "")
  143. line = line.replace("\n", "")
  144. if header_data.reading == "vertex":
  145. header_data.vertex_lines += [line]
  146. if header_data.reading == "fragment":
  147. header_data.fragment_lines += [line]
  148. line = fs.readline()
  149. header_data.line_offset += 1
  150. fs.close()
  151. return header_data
  152. def build_legacygl_header(filename, include, class_suffix, output_attribs, gles2=False):
  153. header_data = LegacyGLHeaderStruct()
  154. include_file_in_legacygl_header(filename, header_data, 0)
  155. out_file = filename + ".gen.h"
  156. fd = open(out_file, "w")
  157. enum_constants = []
  158. fd.write("/* WARNING, THIS FILE WAS GENERATED, DO NOT EDIT */\n")
  159. out_file_base = out_file
  160. out_file_base = out_file_base[out_file_base.rfind("/") + 1 :]
  161. out_file_base = out_file_base[out_file_base.rfind("\\") + 1 :]
  162. out_file_ifdef = out_file_base.replace(".", "_").upper()
  163. fd.write("#ifndef " + out_file_ifdef + class_suffix + "_120\n")
  164. fd.write("#define " + out_file_ifdef + class_suffix + "_120\n")
  165. out_file_class = (
  166. out_file_base.replace(".glsl.gen.h", "").title().replace("_", "").replace(".", "") + "Shader" + class_suffix
  167. )
  168. fd.write("\n\n")
  169. fd.write('#include "' + include + '"\n\n\n')
  170. fd.write("class " + out_file_class + " : public Shader" + class_suffix + " {\n\n")
  171. fd.write('\t virtual String get_shader_name() const { return "' + out_file_class + '"; }\n')
  172. fd.write("public:\n\n")
  173. if header_data.conditionals:
  174. fd.write("\tenum Conditionals {\n")
  175. for x in header_data.conditionals:
  176. fd.write("\t\t" + x.upper() + ",\n")
  177. fd.write("\t};\n\n")
  178. if header_data.uniforms:
  179. fd.write("\tenum Uniforms {\n")
  180. for x in header_data.uniforms:
  181. fd.write("\t\t" + x.upper() + ",\n")
  182. fd.write("\t};\n\n")
  183. fd.write("\t_FORCE_INLINE_ int get_uniform(Uniforms p_uniform) const { return _get_uniform(p_uniform); }\n\n")
  184. if header_data.conditionals:
  185. fd.write(
  186. "\t_FORCE_INLINE_ void set_conditional(Conditionals p_conditional,bool p_enable) { _set_conditional(p_conditional,p_enable); }\n\n"
  187. )
  188. fd.write("\t#ifdef DEBUG_ENABLED\n ")
  189. fd.write(
  190. "\t#define _FU if (get_uniform(p_uniform)<0) return; if (!is_version_valid()) return; ERR_FAIL_COND( get_active()!=this ); \n\n "
  191. )
  192. fd.write("\t#else\n ")
  193. fd.write("\t#define _FU if (get_uniform(p_uniform)<0) return; \n\n ")
  194. fd.write("\t#endif\n")
  195. fd.write(
  196. "\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, float p_value) { _FU glUniform1f(get_uniform(p_uniform),p_value); }\n\n"
  197. )
  198. fd.write(
  199. "\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, double p_value) { _FU glUniform1f(get_uniform(p_uniform),p_value); }\n\n"
  200. )
  201. fd.write(
  202. "\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, uint8_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n"
  203. )
  204. fd.write(
  205. "\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, int8_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n"
  206. )
  207. fd.write(
  208. "\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, uint16_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n"
  209. )
  210. fd.write(
  211. "\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, int16_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n"
  212. )
  213. fd.write(
  214. "\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, uint32_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n"
  215. )
  216. fd.write(
  217. "\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, int32_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n"
  218. )
  219. fd.write(
  220. "\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const Color& p_color) { _FU GLfloat col[4]={p_color.r,p_color.g,p_color.b,p_color.a}; glUniform4fv(get_uniform(p_uniform),1,col); }\n\n"
  221. )
  222. fd.write(
  223. "\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const Vector2& p_vec2) { _FU GLfloat vec2[2]={p_vec2.x,p_vec2.y}; glUniform2fv(get_uniform(p_uniform),1,vec2); }\n\n"
  224. )
  225. fd.write(
  226. "\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const Size2i& p_vec2) { _FU GLint vec2[2]={p_vec2.x,p_vec2.y}; glUniform2iv(get_uniform(p_uniform),1,vec2); }\n\n"
  227. )
  228. fd.write(
  229. "\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const Vector3& p_vec3) { _FU GLfloat vec3[3]={p_vec3.x,p_vec3.y,p_vec3.z}; glUniform3fv(get_uniform(p_uniform),1,vec3); }\n\n"
  230. )
  231. fd.write(
  232. "\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, float p_a, float p_b) { _FU glUniform2f(get_uniform(p_uniform),p_a,p_b); }\n\n"
  233. )
  234. fd.write(
  235. "\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, float p_a, float p_b, float p_c) { _FU glUniform3f(get_uniform(p_uniform),p_a,p_b,p_c); }\n\n"
  236. )
  237. fd.write(
  238. "\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, float p_a, float p_b, float p_c, float p_d) { _FU glUniform4f(get_uniform(p_uniform),p_a,p_b,p_c,p_d); }\n\n"
  239. )
  240. fd.write(
  241. """\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const Transform& p_transform) { _FU
  242. const Transform &tr = p_transform;
  243. GLfloat matrix[16]={ /* build a 16x16 matrix */
  244. tr.basis.elements[0][0],
  245. tr.basis.elements[1][0],
  246. tr.basis.elements[2][0],
  247. 0,
  248. tr.basis.elements[0][1],
  249. tr.basis.elements[1][1],
  250. tr.basis.elements[2][1],
  251. 0,
  252. tr.basis.elements[0][2],
  253. tr.basis.elements[1][2],
  254. tr.basis.elements[2][2],
  255. 0,
  256. tr.origin.x,
  257. tr.origin.y,
  258. tr.origin.z,
  259. 1
  260. };
  261. glUniformMatrix4fv(get_uniform(p_uniform),1,false,matrix);
  262. }
  263. """
  264. )
  265. fd.write(
  266. """_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const Transform2D& p_transform) { _FU
  267. const Transform2D &tr = p_transform;
  268. GLfloat matrix[16]={ /* build a 16x16 matrix */
  269. tr.elements[0][0],
  270. tr.elements[0][1],
  271. 0,
  272. 0,
  273. tr.elements[1][0],
  274. tr.elements[1][1],
  275. 0,
  276. 0,
  277. 0,
  278. 0,
  279. 1,
  280. 0,
  281. tr.elements[2][0],
  282. tr.elements[2][1],
  283. 0,
  284. 1
  285. };
  286. glUniformMatrix4fv(get_uniform(p_uniform),1,false,matrix);
  287. }
  288. """
  289. )
  290. fd.write(
  291. """_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const CameraMatrix& p_matrix) { _FU
  292. GLfloat matrix[16];
  293. for (int i=0;i<4;i++) {
  294. for (int j=0;j<4;j++) {
  295. matrix[i*4+j]=p_matrix.matrix[i][j];
  296. }
  297. }
  298. glUniformMatrix4fv(get_uniform(p_uniform),1,false,matrix);
  299. }
  300. """
  301. )
  302. fd.write("\n\n#undef _FU\n\n\n")
  303. fd.write("\tvirtual void init() {\n\n")
  304. enum_value_count = 0
  305. if header_data.enums:
  306. fd.write("\t\t//Written using math, given nonstandarity of 64 bits integer constants..\n")
  307. fd.write("\t\tstatic const Enum _enums[]={\n")
  308. bitofs = len(header_data.conditionals)
  309. enum_vals = []
  310. for xv in header_data.enums:
  311. x = header_data.enums[xv]
  312. bits = 1
  313. amt = len(x)
  314. while 2 ** bits < amt:
  315. bits += 1
  316. strs = "{"
  317. for i in range(amt):
  318. strs += '"#define ' + x[i] + '\\n",'
  319. c = {}
  320. c["set_mask"] = "uint64_t(" + str(i) + ")<<" + str(bitofs)
  321. c["clear_mask"] = (
  322. "((uint64_t(1)<<40)-1) ^ (((uint64_t(1)<<" + str(bits) + ") - 1)<<" + str(bitofs) + ")"
  323. )
  324. enum_vals.append(c)
  325. enum_constants.append(x[i])
  326. strs += "NULL}"
  327. fd.write(
  328. "\t\t\t{(uint64_t(1<<" + str(bits) + ")-1)<<" + str(bitofs) + "," + str(bitofs) + "," + strs + "},\n"
  329. )
  330. bitofs += bits
  331. fd.write("\t\t};\n\n")
  332. fd.write("\t\tstatic const EnumValue _enum_values[]={\n")
  333. enum_value_count = len(enum_vals)
  334. for x in enum_vals:
  335. fd.write("\t\t\t{" + x["set_mask"] + "," + x["clear_mask"] + "},\n")
  336. fd.write("\t\t};\n\n")
  337. conditionals_found = []
  338. if header_data.conditionals:
  339. fd.write("\t\tstatic const char* _conditional_strings[]={\n")
  340. if header_data.conditionals:
  341. for x in header_data.conditionals:
  342. fd.write('\t\t\t"#define ' + x + '\\n",\n')
  343. conditionals_found.append(x)
  344. fd.write("\t\t};\n\n")
  345. else:
  346. fd.write("\t\tstatic const char **_conditional_strings=NULL;\n")
  347. if header_data.uniforms:
  348. fd.write("\t\tstatic const char* _uniform_strings[]={\n")
  349. if header_data.uniforms:
  350. for x in header_data.uniforms:
  351. fd.write('\t\t\t"' + x + '",\n')
  352. fd.write("\t\t};\n\n")
  353. else:
  354. fd.write("\t\tstatic const char **_uniform_strings=NULL;\n")
  355. if output_attribs:
  356. if header_data.attributes:
  357. fd.write("\t\tstatic AttributePair _attribute_pairs[]={\n")
  358. for x in header_data.attributes:
  359. fd.write('\t\t\t{"' + x[0] + '",' + x[1] + "},\n")
  360. fd.write("\t\t};\n\n")
  361. else:
  362. fd.write("\t\tstatic AttributePair *_attribute_pairs=NULL;\n")
  363. feedback_count = 0
  364. if not gles2 and len(header_data.feedbacks):
  365. fd.write("\t\tstatic const Feedback _feedbacks[]={\n")
  366. for x in header_data.feedbacks:
  367. name = x[0]
  368. cond = x[1]
  369. if cond in conditionals_found:
  370. fd.write('\t\t\t{"' + name + '",' + str(conditionals_found.index(cond)) + "},\n")
  371. else:
  372. fd.write('\t\t\t{"' + name + '",-1},\n')
  373. feedback_count += 1
  374. fd.write("\t\t};\n\n")
  375. else:
  376. if gles2:
  377. pass
  378. else:
  379. fd.write("\t\tstatic const Feedback* _feedbacks=NULL;\n")
  380. if header_data.texunits:
  381. fd.write("\t\tstatic TexUnitPair _texunit_pairs[]={\n")
  382. for x in header_data.texunits:
  383. fd.write('\t\t\t{"' + x[0] + '",' + x[1] + "},\n")
  384. fd.write("\t\t};\n\n")
  385. else:
  386. fd.write("\t\tstatic TexUnitPair *_texunit_pairs=NULL;\n")
  387. if not gles2 and header_data.ubos:
  388. fd.write("\t\tstatic UBOPair _ubo_pairs[]={\n")
  389. for x in header_data.ubos:
  390. fd.write('\t\t\t{"' + x[0] + '",' + x[1] + "},\n")
  391. fd.write("\t\t};\n\n")
  392. else:
  393. if gles2:
  394. pass
  395. else:
  396. fd.write("\t\tstatic UBOPair *_ubo_pairs=NULL;\n")
  397. fd.write("\t\tstatic const char _vertex_code[]={\n")
  398. for x in header_data.vertex_lines:
  399. for c in x:
  400. fd.write(str(ord(c)) + ",")
  401. fd.write(str(ord("\n")) + ",")
  402. fd.write("\t\t0};\n\n")
  403. fd.write("\t\tstatic const int _vertex_code_start=" + str(header_data.vertex_offset) + ";\n")
  404. fd.write("\t\tstatic const char _fragment_code[]={\n")
  405. for x in header_data.fragment_lines:
  406. for c in x:
  407. fd.write(str(ord(c)) + ",")
  408. fd.write(str(ord("\n")) + ",")
  409. fd.write("\t\t0};\n\n")
  410. fd.write("\t\tstatic const int _fragment_code_start=" + str(header_data.fragment_offset) + ";\n")
  411. if output_attribs:
  412. if gles2:
  413. fd.write(
  414. "\t\tsetup(_conditional_strings,"
  415. + str(len(header_data.conditionals))
  416. + ",_uniform_strings,"
  417. + str(len(header_data.uniforms))
  418. + ",_attribute_pairs,"
  419. + str(len(header_data.attributes))
  420. + ", _texunit_pairs,"
  421. + str(len(header_data.texunits))
  422. + ",_vertex_code,_fragment_code,_vertex_code_start,_fragment_code_start);\n"
  423. )
  424. else:
  425. fd.write(
  426. "\t\tsetup(_conditional_strings,"
  427. + str(len(header_data.conditionals))
  428. + ",_uniform_strings,"
  429. + str(len(header_data.uniforms))
  430. + ",_attribute_pairs,"
  431. + str(len(header_data.attributes))
  432. + ", _texunit_pairs,"
  433. + str(len(header_data.texunits))
  434. + ",_ubo_pairs,"
  435. + str(len(header_data.ubos))
  436. + ",_feedbacks,"
  437. + str(feedback_count)
  438. + ",_vertex_code,_fragment_code,_vertex_code_start,_fragment_code_start);\n"
  439. )
  440. else:
  441. if gles2:
  442. fd.write(
  443. "\t\tsetup(_conditional_strings,"
  444. + str(len(header_data.conditionals))
  445. + ",_uniform_strings,"
  446. + str(len(header_data.uniforms))
  447. + ",_texunit_pairs,"
  448. + str(len(header_data.texunits))
  449. + ",_enums,"
  450. + str(len(header_data.enums))
  451. + ",_enum_values,"
  452. + str(enum_value_count)
  453. + ",_vertex_code,_fragment_code,_vertex_code_start,_fragment_code_start);\n"
  454. )
  455. else:
  456. fd.write(
  457. "\t\tsetup(_conditional_strings,"
  458. + str(len(header_data.conditionals))
  459. + ",_uniform_strings,"
  460. + str(len(header_data.uniforms))
  461. + ",_texunit_pairs,"
  462. + str(len(header_data.texunits))
  463. + ",_enums,"
  464. + str(len(header_data.enums))
  465. + ",_enum_values,"
  466. + str(enum_value_count)
  467. + ",_ubo_pairs,"
  468. + str(len(header_data.ubos))
  469. + ",_feedbacks,"
  470. + str(feedback_count)
  471. + ",_vertex_code,_fragment_code,_vertex_code_start,_fragment_code_start);\n"
  472. )
  473. fd.write("\t}\n\n")
  474. if enum_constants:
  475. fd.write("\tenum EnumConditionals {\n")
  476. for x in enum_constants:
  477. fd.write("\t\t" + x.upper() + ",\n")
  478. fd.write("\t};\n\n")
  479. fd.write("\tvoid set_enum_conditional(EnumConditionals p_cond) { _set_enum_conditional(p_cond); }\n")
  480. fd.write("};\n\n")
  481. fd.write("#endif\n\n")
  482. fd.close()
  483. def build_gles2_headers(target, source, env):
  484. for x in source:
  485. build_legacygl_header(
  486. str(x), include="drivers/gles2/shader_gles2.h", class_suffix="GLES2", output_attribs=True, gles2=True
  487. )
  488. class RDHeaderStruct:
  489. def __init__(self):
  490. self.vertex_lines = []
  491. self.fragment_lines = []
  492. self.compute_lines = []
  493. self.vertex_included_files = []
  494. self.fragment_included_files = []
  495. self.reading = ""
  496. self.line_offset = 0
  497. self.vertex_offset = 0
  498. self.fragment_offset = 0
  499. self.compute_offset = 0
  500. def include_file_in_rd_header(filename, header_data, depth):
  501. fs = open(filename, "r")
  502. line = fs.readline()
  503. while line:
  504. if line.find("[vertex]") != -1:
  505. header_data.reading = "vertex"
  506. line = fs.readline()
  507. header_data.line_offset += 1
  508. header_data.vertex_offset = header_data.line_offset
  509. continue
  510. if line.find("[fragment]") != -1:
  511. header_data.reading = "fragment"
  512. line = fs.readline()
  513. header_data.line_offset += 1
  514. header_data.fragment_offset = header_data.line_offset
  515. continue
  516. if line.find("[compute]") != -1:
  517. header_data.reading = "compute"
  518. line = fs.readline()
  519. header_data.line_offset += 1
  520. header_data.compute_offset = header_data.line_offset
  521. continue
  522. while line.find("#include ") != -1:
  523. includeline = line.replace("#include ", "").strip()[1:-1]
  524. import os.path
  525. included_file = os.path.relpath(os.path.dirname(filename) + "/" + includeline)
  526. if not included_file in header_data.vertex_included_files and header_data.reading == "vertex":
  527. header_data.vertex_included_files += [included_file]
  528. if include_file_in_rd_header(included_file, header_data, depth + 1) is None:
  529. print("Error in file '" + filename + "': #include " + includeline + "could not be found!")
  530. elif not included_file in header_data.fragment_included_files and header_data.reading == "fragment":
  531. header_data.fragment_included_files += [included_file]
  532. if include_file_in_rd_header(included_file, header_data, depth + 1) is None:
  533. print("Error in file '" + filename + "': #include " + includeline + "could not be found!")
  534. elif not included_file in header_data.compute_included_files and header_data.reading == "compute":
  535. header_data.compute_included_files += [included_file]
  536. if include_file_in_rd_header(included_file, header_data, depth + 1) is None:
  537. print("Error in file '" + filename + "': #include " + includeline + "could not be found!")
  538. line = fs.readline()
  539. line = line.replace("\r", "")
  540. line = line.replace("\n", "")
  541. if header_data.reading == "vertex":
  542. header_data.vertex_lines += [line]
  543. if header_data.reading == "fragment":
  544. header_data.fragment_lines += [line]
  545. if header_data.reading == "compute":
  546. header_data.compute_lines += [line]
  547. line = fs.readline()
  548. header_data.line_offset += 1
  549. fs.close()
  550. return header_data
  551. def build_rd_header(filename):
  552. header_data = RDHeaderStruct()
  553. include_file_in_rd_header(filename, header_data, 0)
  554. out_file = filename + ".gen.h"
  555. fd = open(out_file, "w")
  556. enum_constants = []
  557. fd.write("/* WARNING, THIS FILE WAS GENERATED, DO NOT EDIT */\n")
  558. out_file_base = out_file
  559. out_file_base = out_file_base[out_file_base.rfind("/") + 1 :]
  560. out_file_base = out_file_base[out_file_base.rfind("\\") + 1 :]
  561. out_file_ifdef = out_file_base.replace(".", "_").upper()
  562. fd.write("#ifndef " + out_file_ifdef + "_RD\n")
  563. fd.write("#define " + out_file_ifdef + "_RD\n")
  564. out_file_class = out_file_base.replace(".glsl.gen.h", "").title().replace("_", "").replace(".", "") + "ShaderRD"
  565. fd.write("\n")
  566. fd.write('#include "servers/rendering/rasterizer_rd/shader_rd.h"\n\n')
  567. fd.write("class " + out_file_class + " : public ShaderRD {\n\n")
  568. fd.write("public:\n\n")
  569. fd.write("\t" + out_file_class + "() {\n\n")
  570. if len(header_data.compute_lines):
  571. fd.write("\t\tstatic const char _compute_code[] = {\n")
  572. for x in header_data.compute_lines:
  573. for c in x:
  574. fd.write(str(ord(c)) + ",")
  575. fd.write(str(ord("\n")) + ",")
  576. fd.write("\t\t0};\n\n")
  577. fd.write('\t\tsetup(nullptr, nullptr, _compute_code, "' + out_file_class + '");\n')
  578. fd.write("\t}\n")
  579. else:
  580. fd.write("\t\tstatic const char _vertex_code[] = {\n")
  581. for x in header_data.vertex_lines:
  582. for c in x:
  583. fd.write(str(ord(c)) + ",")
  584. fd.write(str(ord("\n")) + ",")
  585. fd.write("\t\t0};\n\n")
  586. fd.write("\t\tstatic const char _fragment_code[]={\n")
  587. for x in header_data.fragment_lines:
  588. for c in x:
  589. fd.write(str(ord(c)) + ",")
  590. fd.write(str(ord("\n")) + ",")
  591. fd.write("\t\t0};\n\n")
  592. fd.write('\t\tsetup(_vertex_code, _fragment_code, nullptr, "' + out_file_class + '");\n')
  593. fd.write("\t}\n")
  594. fd.write("};\n\n")
  595. fd.write("#endif\n")
  596. fd.close()
  597. def build_rd_headers(target, source, env):
  598. for x in source:
  599. build_rd_header(str(x))
  600. if __name__ == "__main__":
  601. subprocess_main(globals())