methods.py 51 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565
  1. import os
  2. def add_source_files(self, sources, filetype, lib_env = None, shared = False):
  3. import glob;
  4. import string;
  5. #if not lib_objects:
  6. if not lib_env:
  7. lib_env = self
  8. if type(filetype) == type(""):
  9. dir = self.Dir('.').abspath
  10. list = glob.glob(dir + "/"+filetype)
  11. for f in list:
  12. sources.append( self.Object(f) )
  13. else:
  14. for f in filetype:
  15. sources.append(self.Object(f))
  16. def build_shader_header( target, source, env ):
  17. for x in source:
  18. print x
  19. name = str(x)
  20. name = name[ name.rfind("/")+1: ]
  21. name = name[ name.rfind("\\")+1: ]
  22. name = name.replace(".","_")
  23. fs = open(str(x),"r")
  24. fd = open(str(x)+".h","w")
  25. fd.write("/* this file has been generated by SCons, do not edit! */\n")
  26. fd.write("static const char *"+name+"=\n")
  27. line=fs.readline()
  28. while(line):
  29. line=line.replace("\r","")
  30. line=line.replace("\n","")
  31. line=line.replace("\\","\\\\")
  32. line=line.replace("\"","\\\"")
  33. fd.write("\""+line+"\\n\"\n")
  34. line=fs.readline()
  35. fd.write(";\n")
  36. return 0
  37. def build_glsl_header( filename ):
  38. fs = open(filename,"r")
  39. line=fs.readline()
  40. vertex_lines=[]
  41. fragment_lines=[]
  42. uniforms=[]
  43. attributes=[]
  44. fbos=[]
  45. conditionals=[]
  46. texunits=[]
  47. texunit_names=[]
  48. ubos=[]
  49. ubo_names=[]
  50. reading=""
  51. line_offset=0
  52. vertex_offset=0
  53. fragment_offset=0
  54. while(line):
  55. if (line.find("[vertex]")!=-1):
  56. reading="vertex"
  57. line=fs.readline()
  58. line_offset+=1
  59. vertex_offset=line_offset
  60. continue
  61. if (line.find("[fragment]")!=-1):
  62. reading="fragment"
  63. line=fs.readline()
  64. line_offset+=1
  65. fragment_offset=line_offset
  66. continue
  67. if (line.find("#ifdef ")!=-1):
  68. ifdefline = line.replace("#ifdef ","").strip()
  69. if (not ifdefline in conditionals):
  70. conditionals+=[ifdefline]
  71. if (line.find("#elif defined(")!=-1):
  72. ifdefline = line.replace("#elif defined(","").strip()
  73. ifdefline = ifdefline.replace(")","").strip()
  74. if (not ifdefline in conditionals):
  75. conditionals+=[ifdefline]
  76. import re
  77. if re.search(r"^\s*uniform", line):
  78. if (line.lower().find("texunit:")!=-1):
  79. #texture unit
  80. texunit = str(int( line[line.find(":")+1:].strip() ))
  81. uline=line[:line.lower().find("//")]
  82. uline = uline.replace("uniform","");
  83. uline = uline.replace(";","");
  84. lines = uline.split(",")
  85. for x in lines:
  86. x = x.strip()
  87. x = x[ x.rfind(" ")+1: ]
  88. if (x.find("[")!=-1):
  89. #unfiorm array
  90. x = x[ :x.find("[") ]
  91. if (not x in texunit_names):
  92. texunits+=[(x,texunit)]
  93. texunit_names+=[x]
  94. elif (line.lower().find("ubo:")!=-1):
  95. #ubo
  96. uboidx = str(int( line[line.find(":")+1:].strip() ))
  97. uline=line[:line.lower().find("//")]
  98. uline = uline[uline.find("uniform")+len("uniform"):];
  99. uline = uline.replace(";","");
  100. uline = uline.replace("{","").strip();
  101. lines = uline.split(",")
  102. for x in lines:
  103. x = x.strip()
  104. x = x[ x.rfind(" ")+1: ]
  105. if (x.find("[")!=-1):
  106. #unfiorm array
  107. x = x[ :x.find("[") ]
  108. if (not x in ubo_names):
  109. ubos+=[(x,uboidx)]
  110. ubo_names+=[x]
  111. else:
  112. uline = line.replace("uniform","");
  113. uline = uline.replace(";","");
  114. lines = uline.split(",")
  115. for x in lines:
  116. x = x.strip()
  117. x = x[ x.rfind(" ")+1: ]
  118. if (x.find("[")!=-1):
  119. #unfiorm array
  120. x = x[ :x.find("[") ]
  121. if (not x in uniforms):
  122. uniforms+=[x]
  123. if ((line.strip().find("in ")==0 or line.strip().find("attribute ")==0) and line.find("attrib:")!=-1):
  124. uline = line.replace("in ","");
  125. uline = uline.replace("attribute ","");
  126. uline = uline.replace(";","");
  127. uline = uline[ uline.find(" "): ].strip()
  128. if (uline.find("//")!=-1):
  129. name,bind = uline.split("//")
  130. if (bind.find("attrib:")!=-1):
  131. name=name.strip()
  132. bind=bind.replace("attrib:","").strip()
  133. attributes+=[(name,bind)]
  134. if (line.strip().find("out ")==0):
  135. uline = line.replace("out","").strip();
  136. uline = uline.replace(";","");
  137. uline = uline[ uline.find(" "): ].strip()
  138. if (uline.find("//")!=-1):
  139. name,bind = uline.split("//")
  140. if (bind.find("drawbuffer:")!=-1):
  141. name=name.strip()
  142. bind=bind.replace("drawbuffer:","").strip()
  143. fbos+=[(name,bind)]
  144. line=line.replace("\r","")
  145. line=line.replace("\n","")
  146. line=line.replace("\\","\\\\")
  147. line=line.replace("\"","\\\"")
  148. #line=line+"\\n\\" no need to anymore
  149. if (reading=="vertex"):
  150. vertex_lines+=[line]
  151. if (reading=="fragment"):
  152. fragment_lines+=[line]
  153. line=fs.readline()
  154. line_offset+=1
  155. fs.close();
  156. out_file = filename+".h"
  157. fd = open(out_file,"w")
  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. # print("out file "+out_file+" base " +out_file_base)
  163. out_file_ifdef = out_file_base.replace(".","_").upper()
  164. fd.write("#ifndef "+out_file_ifdef+"\n")
  165. fd.write("#define "+out_file_ifdef+"\n")
  166. out_file_class = out_file_base.replace(".glsl.h","").title().replace("_","").replace(".","")+"ShaderGL";
  167. fd.write("\n\n");
  168. fd.write("#include \"drivers/opengl/shader_gl.h\"\n\n\n");
  169. fd.write("class "+out_file_class+" : public ShaderGL {\n\n");
  170. fd.write("\t virtual String get_shader_name() const { return \""+out_file_class+"\"; }\n");
  171. fd.write("public:\n\n");
  172. if (len(conditionals)):
  173. fd.write("\tenum Conditionals {\n");
  174. for x in conditionals:
  175. fd.write("\t\t"+x+",\n");
  176. fd.write("\t};\n\n");
  177. if (len(uniforms)):
  178. fd.write("\tenum Uniforms {\n");
  179. for x in uniforms:
  180. fd.write("\t\t"+x.upper()+",\n");
  181. fd.write("\t};\n\n");
  182. fd.write("\t_FORCE_INLINE_ int get_uniform(Uniforms p_uniform) const { return _get_uniform(p_uniform); }\n\n");
  183. if (len(conditionals)):
  184. fd.write("\t_FORCE_INLINE_ void set_conditional(Conditionals p_conditional,bool p_enable) { _set_conditional(p_conditional,p_enable); }\n\n");
  185. fd.write("\t#define _FU if (get_uniform(p_uniform)<0) return; ERR_FAIL_COND( get_active()!=this );\n\n ");
  186. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, bool p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value?1:0); }\n\n");
  187. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, float p_value) { _FU glUniform1f(get_uniform(p_uniform),p_value); }\n\n");
  188. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, double p_value) { _FU glUniform1f(get_uniform(p_uniform),p_value); }\n\n");
  189. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, uint8_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  190. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, int8_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  191. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, uint16_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  192. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, int16_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  193. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, uint32_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  194. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, int32_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  195. #fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, uint64_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  196. #fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, int64_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  197. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, unsigned long p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  198. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, long p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  199. fd.write("\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");
  200. fd.write("\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");
  201. fd.write("\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");
  202. fd.write("\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");
  203. fd.write("\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");
  204. fd.write("\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");
  205. fd.write("""\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const Transform& p_transform) { _FU
  206. const Transform &tr = p_transform;
  207. GLfloat matrix[16]={ /* build a 16x16 matrix */
  208. tr.basis.elements[0][0],
  209. tr.basis.elements[1][0],
  210. tr.basis.elements[2][0],
  211. 0,
  212. tr.basis.elements[0][1],
  213. tr.basis.elements[1][1],
  214. tr.basis.elements[2][1],
  215. 0,
  216. tr.basis.elements[0][2],
  217. tr.basis.elements[1][2],
  218. tr.basis.elements[2][2],
  219. 0,
  220. tr.origin.x,
  221. tr.origin.y,
  222. tr.origin.z,
  223. 1
  224. };
  225. glUniformMatrix4fv(get_uniform(p_uniform),1,false,matrix);
  226. }
  227. """);
  228. fd.write("""\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const Matrix32& p_transform) { _FU
  229. const Matrix32 &tr = p_transform;
  230. GLfloat matrix[16]={ /* build a 16x16 matrix */
  231. tr.elements[0][0],
  232. tr.elements[0][1],
  233. 0,
  234. 0,
  235. tr.elements[1][0],
  236. tr.elements[1][1],
  237. 0,
  238. 0,
  239. 0,
  240. 0,
  241. 1,
  242. 0,
  243. tr.elements[2][0],
  244. tr.elements[2][1],
  245. 0,
  246. 1
  247. };
  248. glUniformMatrix4fv(get_uniform(p_uniform),1,false,matrix);
  249. }
  250. """);
  251. fd.write("""\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const CameraMatrix& p_matrix) { _FU
  252. GLfloat matrix[16];
  253. for (int i=0;i<4;i++) {
  254. for (int j=0;j<4;j++) {
  255. matrix[i*4+j]=p_matrix.matrix[i][j];
  256. }
  257. }
  258. glUniformMatrix4fv(get_uniform(p_uniform),1,false,matrix);
  259. }; """);
  260. fd.write("\n\n#undef _FU\n\n\n");
  261. fd.write("\tvirtual void init() {\n\n");
  262. if (len(conditionals)):
  263. fd.write("\t\tstatic const char* _conditional_strings[]={\n")
  264. if (len(conditionals)):
  265. for x in conditionals:
  266. fd.write("\t\t\t\"#define "+x+"\\n\",\n");
  267. fd.write("\t\t};\n\n");
  268. else:
  269. fd.write("\t\tstatic const char **_conditional_strings=NULL;\n")
  270. if (len(uniforms)):
  271. fd.write("\t\tstatic const char* _uniform_strings[]={\n")
  272. if (len(uniforms)):
  273. for x in uniforms:
  274. fd.write("\t\t\t\""+x+"\",\n");
  275. fd.write("\t\t};\n\n");
  276. else:
  277. fd.write("\t\tstatic const char **_uniform_strings=NULL;\n")
  278. if (len(attributes)):
  279. fd.write("\t\tstatic AttributePair _attribute_pairs[]={\n")
  280. for x in attributes:
  281. fd.write("\t\t\t{\""+x[0]+"\","+x[1]+"},\n");
  282. fd.write("\t\t};\n\n");
  283. else:
  284. fd.write("\t\tstatic AttributePair *_attribute_pairs=NULL;\n")
  285. if (len(fbos)):
  286. fd.write("\t\tstatic FBOPair _fbo_pairs[]={\n")
  287. for x in fbos:
  288. fd.write("\t\t\t{\""+x[0]+"\","+x[1]+"},\n");
  289. fd.write("\t\t};\n\n");
  290. else:
  291. fd.write("\t\tstatic FBOPair *_fbo_pairs=NULL;\n")
  292. if (len(ubos)):
  293. fd.write("\t\tstatic UBOPair _ubo_pairs[]={\n")
  294. for x in ubos:
  295. fd.write("\t\t\t{\""+x[0]+"\","+x[1]+"},\n");
  296. fd.write("\t\t};\n\n");
  297. else:
  298. fd.write("\t\tstatic UBOPair *_ubo_pairs=NULL;\n")
  299. if (len(texunits)):
  300. fd.write("\t\tstatic TexUnitPair _texunit_pairs[]={\n")
  301. for x in texunits:
  302. fd.write("\t\t\t{\""+x[0]+"\","+x[1]+"},\n");
  303. fd.write("\t\t};\n\n");
  304. else:
  305. fd.write("\t\tstatic TexUnitPair *_texunit_pairs=NULL;\n")
  306. fd.write("\t\tstatic const char* _vertex_code=\"\\\n")
  307. for x in vertex_lines:
  308. fd.write("\t\t\t"+x+"\n");
  309. fd.write("\t\t\";\n\n");
  310. fd.write("\t\tstatic const int _vertex_code_start="+str(vertex_offset)+";\n")
  311. fd.write("\t\tstatic const char* _fragment_code=\"\\\n")
  312. for x in fragment_lines:
  313. fd.write("\t\t\t"+x+"\n");
  314. fd.write("\t\t\";\n\n");
  315. fd.write("\t\tstatic const int _fragment_code_start="+str(fragment_offset)+";\n")
  316. fd.write("\t\tsetup(_conditional_strings,"+str(len(conditionals))+",_uniform_strings,"+str(len(uniforms))+",_attribute_pairs,"+str(len(attributes))+",_fbo_pairs,"+str(len(fbos))+",_ubo_pairs,"+str(len(ubos))+",_texunit_pairs,"+str(len(texunits))+",_vertex_code,_fragment_code,_vertex_code_start,_fragment_code_start);\n")
  317. fd.write("\t};\n\n")
  318. fd.write("};\n\n");
  319. fd.write("#endif\n\n");
  320. fd.close();
  321. def build_glsl_headers( target, source, env ):
  322. for x in source:
  323. build_glsl_header(str(x));
  324. return 0
  325. def build_hlsl_dx9_header( filename ):
  326. fs = open(filename,"r")
  327. line=fs.readline()
  328. vertex_lines=[]
  329. fragment_lines=[]
  330. uniforms=[]
  331. fragment_uniforms=[]
  332. attributes=[]
  333. fbos=[]
  334. conditionals=[]
  335. reading=""
  336. line_offset=0
  337. vertex_offset=0
  338. fragment_offset=0
  339. while(line):
  340. if (line.find("[vertex]")!=-1):
  341. reading="vertex"
  342. line=fs.readline()
  343. line_offset+=1
  344. vertex_offset=line_offset
  345. continue
  346. if (line.find("[fragment]")!=-1):
  347. reading="fragment"
  348. line=fs.readline()
  349. line_offset+=1
  350. fragment_offset=line_offset
  351. continue
  352. if (line.find("#ifdef ")!=-1):
  353. ifdefline = line.replace("#ifdef ","").strip()
  354. if (not ifdefline in conditionals):
  355. conditionals+=[ifdefline]
  356. if (line.find("#elif defined(")!=-1):
  357. ifdefline = line.replace("#elif defined(","").strip()
  358. ifdefline = ifdefline.replace(")","").strip()
  359. if (not ifdefline in conditionals):
  360. conditionals+=[ifdefline]
  361. if (line.find("uniform")!=-1):
  362. uline = line.replace("uniform","");
  363. uline = uline.replace(";","");
  364. lines = uline.split(",")
  365. for x in lines:
  366. x = x.strip()
  367. x = x[ x.rfind(" ")+1: ]
  368. if (x.find("[")!=-1):
  369. #unfiorm array
  370. x = x[ :x.find("[") ]
  371. if (not x in uniforms):
  372. uniforms+=[x]
  373. fragment_uniforms+=[reading=="fragment"]
  374. line=line.replace("\r","")
  375. line=line.replace("\n","")
  376. line=line.replace("\\","\\\\")
  377. line=line.replace("\"","\\\"")
  378. line=line+"\\n\\"
  379. if (reading=="vertex"):
  380. vertex_lines+=[line]
  381. if (reading=="fragment"):
  382. fragment_lines+=[line]
  383. line=fs.readline()
  384. line_offset+=1
  385. fs.close();
  386. out_file = filename+".h"
  387. fd = open(out_file,"w")
  388. fd.write("/* WARNING, THIS FILE WAS GENERATED, DO NOT EDIT */\n");
  389. out_file_base = out_file
  390. out_file_base = out_file_base[ out_file_base.rfind("/")+1: ]
  391. out_file_base = out_file_base[ out_file_base.rfind("\\")+1: ]
  392. # print("out file "+out_file+" base " +out_file_base)
  393. out_file_ifdef = out_file_base.replace(".","_").upper()
  394. fd.write("#ifndef "+out_file_ifdef+"\n")
  395. fd.write("#define "+out_file_ifdef+"\n")
  396. out_file_class = out_file_base.replace(".hlsl.h","").title().replace("_","").replace(".","")+"ShaderDX9";
  397. fd.write("\n\n");
  398. fd.write("#include \"drivers/directx9/shader_dx9.h\"\n\n\n");
  399. fd.write("class "+out_file_class+" : public ShaderDX9 {\n\n");
  400. fd.write("\t virtual String get_shader_name() const { return \""+out_file_class+"\"; }\n");
  401. fd.write("public:\n\n");
  402. if (len(conditionals)):
  403. fd.write("\tenum Conditionals {\n");
  404. for x in conditionals:
  405. fd.write("\t\t"+x+",\n");
  406. fd.write("\t};\n\n");
  407. if (len(uniforms)):
  408. fd.write("\tenum Uniforms {\n");
  409. for x in uniforms:
  410. fd.write("\t\t"+x.upper()+",\n");
  411. fd.write("\t};\n\n");
  412. if (len(conditionals)):
  413. fd.write("\t_FORCE_INLINE_ void set_conditional(Conditionals p_conditional,bool p_enable) { _set_conditional(p_conditional,p_enable); }\n\n");
  414. fd.write("\t#define _FU if (!_uniform_valid(p_uniform)) return; ERR_FAIL_COND( get_active()!=this );\n\n ");
  415. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, bool p_value) { _FU set_uniformb(p_uniform,p_value); }\n\n");
  416. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, float p_value) { _FU set_uniformf(p_uniform,p_value); }\n\n");
  417. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, double p_value) { _FU set_uniformf(p_uniform,p_value); }\n\n");
  418. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, uint8_t p_value) { _FU set_uniformi(p_uniform,p_value); }\n\n");
  419. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, int8_t p_value) { _FU set_uniformi(p_uniform,p_value); }\n\n");
  420. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, uint16_t p_value) { _FU set_uniformi(p_uniform,p_value); }\n\n");
  421. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, int16_t p_value) { _FU set_uniformi(p_uniform,p_value); }\n\n");
  422. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, uint32_t p_value) { _FU set_uniformi(p_uniform,p_value); }\n\n");
  423. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, int32_t p_value) { _FU set_uniformi(p_uniform,p_value); }\n\n");
  424. #fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, uint64_t p_value) { _FU set_uniformi(p_uniform,p_value); }\n\n");
  425. #fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, int64_t p_value) { _FU set_uniformi(p_uniform,p_value); }\n\n");
  426. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, unsigned long p_value) { _FU set_uniformi(p_uniform,p_value); }\n\n");
  427. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, long p_value) { _FU set_uniformi(p_uniform,p_value); }\n\n");
  428. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const Color& p_color) { _FU float col[4]={p_color.r,p_color.g,p_color.b,p_color.a}; set_uniformfv(p_uniform,col); }\n\n");
  429. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const Vector2& p_vec2) { _FU float vec2[4]={p_vec2.x,p_vec2.y,0,0}; set_uniformfv(p_uniform,vec2); }\n\n");
  430. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const Vector3& p_vec3) { _FU float vec3[4]={p_vec3.x,p_vec3.y,p_vec3.z,0}; set_uniformfv(p_uniform,vec3); }\n\n");
  431. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, float p_a, float p_b) { _FU float vec2[4]={p_a,p_b,0,0}; set_uniformfv(p_uniform,vec2); }\n\n");
  432. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, float p_a, float p_b, float p_c) { _FU float vec3[4]={p_a,p_b,p_c,0}; set_uniformfv(p_uniform,vec3); }\n\n");
  433. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, float p_a, float p_b, float p_c, float p_d) { _FU float vec4[4]={p_a,p_b,p_c,p_d}; set_uniformfv(p_uniform,vec4); }\n\n");
  434. fd.write("""\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const Transform& p_transform) { _FU
  435. const Transform &tr = p_transform;
  436. float matrix[16]={ /* build a 16x16 matrix */
  437. tr.basis.elements[0][0],
  438. tr.basis.elements[0][1],
  439. tr.basis.elements[0][2],
  440. tr.origin.x,
  441. tr.basis.elements[1][0],
  442. tr.basis.elements[1][1],
  443. tr.basis.elements[1][2],
  444. tr.origin.y,
  445. tr.basis.elements[2][0],
  446. tr.basis.elements[2][1],
  447. tr.basis.elements[2][2],
  448. tr.origin.z,
  449. 0,
  450. 0,
  451. 0,
  452. 1
  453. };
  454. set_uniformfv(p_uniform,&matrix[0],4);
  455. }
  456. """);
  457. fd.write("""\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const CameraMatrix& p_matrix) { _FU
  458. float matrix[16];
  459. for (int i=0;i<4;i++) {
  460. for (int j=0;j<4;j++) {
  461. matrix[i*4+j]=p_matrix.matrix[j][i];
  462. }
  463. }
  464. set_uniformfv(p_uniform,&matrix[0],4);
  465. }; """);
  466. fd.write("\n\n#undef _FU\n\n\n");
  467. fd.write("\tvirtual void init(IDirect3DDevice9 *p_device,ShaderSupport p_version) {\n\n");
  468. if (len(conditionals)):
  469. fd.write("\t\tstatic const char* _conditional_strings[]={\n")
  470. if (len(conditionals)):
  471. for x in conditionals:
  472. fd.write("\t\t\t\""+x+"\",\n");
  473. fd.write("\t\t};\n\n");
  474. else:
  475. fd.write("\t\tstatic const char **_conditional_strings=NULL;\n")
  476. if (len(uniforms)):
  477. fd.write("\t\tstatic const char* _uniform_strings[]={\n")
  478. if (len(uniforms)):
  479. for x in uniforms:
  480. fd.write("\t\t\t\""+x+"\",\n");
  481. fd.write("\t\t};\n\n");
  482. fd.write("\t\tstatic const bool _fragment_uniforms[]={\n")
  483. if (len(uniforms)):
  484. for x in fragment_uniforms:
  485. if (x):
  486. fd.write("\t\t\ttrue,\n");
  487. else:
  488. fd.write("\t\t\tfalse,\n");
  489. fd.write("\t\t};\n\n");
  490. else:
  491. fd.write("\t\tstatic const char **_uniform_strings=NULL;\n")
  492. fd.write("\t\tstatic const bool *_fragment_uniforms=NULL;\n")
  493. fd.write("\t\tstatic const char* _vertex_code=\"\\\n")
  494. for x in vertex_lines:
  495. fd.write("\t\t\t"+x+"\n");
  496. fd.write("\t\t\";\n\n");
  497. fd.write("\t\tstatic const int _vertex_code_start="+str(vertex_offset)+";\n")
  498. fd.write("\t\tstatic const char* _fragment_code=\"\\\n")
  499. for x in fragment_lines:
  500. fd.write("\t\t\t"+x+"\n");
  501. fd.write("\t\t\";\n\n");
  502. fd.write("\t\tstatic const int _fragment_code_start="+str(fragment_offset)+";\n")
  503. fd.write("\t\tsetup(p_device,p_version,_conditional_strings,"+str(len(conditionals))+",_uniform_strings,"+str(len(uniforms))+",_fragment_uniforms,_vertex_code,_fragment_code,_vertex_code_start,_fragment_code_start);\n")
  504. fd.write("\t};\n\n")
  505. fd.write("};\n\n");
  506. fd.write("#endif\n\n");
  507. fd.close();
  508. def build_hlsl_dx9_headers( target, source, env ):
  509. for x in source:
  510. build_hlsl_dx9_header(str(x));
  511. return 0
  512. class LegacyGLHeaderStruct:
  513. def __init__(self):
  514. self.vertex_lines=[]
  515. self.fragment_lines=[]
  516. self.uniforms=[]
  517. self.attributes=[]
  518. self.fbos=[]
  519. self.conditionals=[]
  520. self.enums={}
  521. self.texunits=[]
  522. self.texunit_names=[]
  523. self.ubos=[]
  524. self.ubo_names=[]
  525. self.vertex_included_files=[]
  526. self.fragment_included_files=[]
  527. self.reading=""
  528. self.line_offset=0
  529. self.vertex_offset=0
  530. self.fragment_offset=0
  531. def include_file_in_legacygl_header( filename, header_data, depth ):
  532. fs = open(filename,"r")
  533. line=fs.readline()
  534. while(line):
  535. if (line.find("[vertex]")!=-1):
  536. header_data.reading="vertex"
  537. line=fs.readline()
  538. header_data.line_offset+=1
  539. header_data.vertex_offset=header_data.line_offset
  540. continue
  541. if (line.find("[fragment]")!=-1):
  542. header_data.reading="fragment"
  543. line=fs.readline()
  544. header_data.line_offset+=1
  545. header_data.fragment_offset=header_data.line_offset
  546. continue
  547. while(line.find("#include ")!=-1):
  548. includeline = line.replace("#include ","").strip()[1:-1]
  549. import os.path
  550. included_file = os.path.relpath(os.path.dirname(filename) + "/" + includeline)
  551. if (not included_file in header_data.vertex_included_files and header_data.reading=="vertex"):
  552. header_data.vertex_included_files+=[included_file]
  553. if(include_file_in_legacygl_header( included_file, header_data, depth + 1 ) == None):
  554. print "Error in file '" + filename + "': #include " + includeline + "could not be found!"
  555. elif (not included_file in header_data.fragment_included_files and header_data.reading=="fragment"):
  556. header_data.fragment_included_files+=[included_file]
  557. if(include_file_in_legacygl_header( included_file, header_data, depth + 1 ) == None):
  558. print "Error in file '" + filename + "': #include " + includeline + "could not be found!"
  559. line=fs.readline()
  560. if (line.find("#ifdef ")!=-1 or line.find("#elif defined(")!=-1):
  561. if (line.find("#ifdef ")!=-1):
  562. ifdefline = line.replace("#ifdef ","").strip()
  563. else:
  564. ifdefline = line.replace("#elif defined(","").strip()
  565. ifdefline = ifdefline.replace(")","").strip()
  566. if (line.find("_EN_")!=-1):
  567. enumbase = ifdefline[:ifdefline.find("_EN_")];
  568. ifdefline = ifdefline.replace("_EN_","_")
  569. line = line.replace("_EN_","_")
  570. # print(enumbase+":"+ifdefline);
  571. if (enumbase not in header_data.enums):
  572. header_data.enums[enumbase]=[]
  573. if (ifdefline not in header_data.enums[enumbase]):
  574. header_data.enums[enumbase].append(ifdefline);
  575. elif (not ifdefline in header_data.conditionals):
  576. header_data.conditionals+=[ifdefline]
  577. if (line.find("uniform")!=-1 and line.lower().find("texunit:")!=-1):
  578. #texture unit
  579. texunitstr = line[line.find(":")+1:].strip()
  580. if (texunitstr=="auto"):
  581. texunit="-1"
  582. else:
  583. texunit = str(int(texunitstr ))
  584. uline=line[:line.lower().find("//")]
  585. uline = uline.replace("uniform","");
  586. uline = uline.replace("highp","");
  587. uline = uline.replace(";","");
  588. lines = uline.split(",")
  589. for x in lines:
  590. x = x.strip()
  591. x = x[ x.rfind(" ")+1: ]
  592. if (x.find("[")!=-1):
  593. #unfiorm array
  594. x = x[ :x.find("[") ]
  595. if (not x in header_data.texunit_names):
  596. header_data.texunits+=[(x,texunit)]
  597. header_data.texunit_names+=[x]
  598. elif (line.find("uniform")!=-1 and line.lower().find("ubo:")!=-1):
  599. #uniform buffer object
  600. ubostr = line[line.find(":")+1:].strip()
  601. ubo = str(int(ubostr ))
  602. uline=line[:line.lower().find("//")]
  603. uline = uline[uline.find("uniform")+len("uniform"):];
  604. uline = uline.replace("highp","");
  605. uline = uline.replace(";","");
  606. uline = uline.replace("{","").strip();
  607. lines = uline.split(",")
  608. for x in lines:
  609. x = x.strip()
  610. x = x[ x.rfind(" ")+1: ]
  611. if (x.find("[")!=-1):
  612. #unfiorm array
  613. x = x[ :x.find("[") ]
  614. if (not x in header_data.ubo_names):
  615. header_data.ubos+=[(x,ubo)]
  616. header_data.ubo_names+=[x]
  617. elif (line.find("uniform")!=-1 and line.find("{")==-1 and line.find(";")!=-1):
  618. uline = line.replace("uniform","");
  619. uline = uline.replace(";","");
  620. lines = uline.split(",")
  621. for x in lines:
  622. x = x.strip()
  623. x = x[ x.rfind(" ")+1: ]
  624. if (x.find("[")!=-1):
  625. #unfiorm array
  626. x = x[ :x.find("[") ]
  627. if (not x in header_data.uniforms):
  628. header_data.uniforms+=[x]
  629. if ( line.strip().find("attribute ")==0 and line.find("attrib:")!=-1):
  630. uline = line.replace("in ","");
  631. uline = uline.replace("attribute ","");
  632. uline = uline.replace("highp ","");
  633. uline = uline.replace(";","");
  634. uline = uline[ uline.find(" "): ].strip()
  635. if (uline.find("//")!=-1):
  636. name,bind = uline.split("//")
  637. if (bind.find("attrib:")!=-1):
  638. name=name.strip()
  639. bind=bind.replace("attrib:","").strip()
  640. header_data.attributes+=[(name,bind)]
  641. line=line.replace("\r","")
  642. line=line.replace("\n","")
  643. #line=line.replace("\\","\\\\")
  644. #line=line.replace("\"","\\\"")
  645. #line=line+"\\n\\"
  646. if (header_data.reading=="vertex"):
  647. header_data.vertex_lines+=[line]
  648. if (header_data.reading=="fragment"):
  649. header_data.fragment_lines+=[line]
  650. line=fs.readline()
  651. header_data.line_offset+=1
  652. fs.close();
  653. return header_data
  654. def build_legacygl_header( filename, include, class_suffix, output_attribs ):
  655. header_data = LegacyGLHeaderStruct()
  656. include_file_in_legacygl_header( filename, header_data, 0 )
  657. out_file = filename+".h"
  658. fd = open(out_file,"w")
  659. enum_constants=[]
  660. fd.write("/* WARNING, THIS FILE WAS GENERATED, DO NOT EDIT */\n");
  661. out_file_base = out_file
  662. out_file_base = out_file_base[ out_file_base.rfind("/")+1: ]
  663. out_file_base = out_file_base[ out_file_base.rfind("\\")+1: ]
  664. # print("out file "+out_file+" base " +out_file_base)
  665. out_file_ifdef = out_file_base.replace(".","_").upper()
  666. fd.write("#ifndef "+out_file_ifdef+class_suffix+"_120\n")
  667. fd.write("#define "+out_file_ifdef+class_suffix+"_120\n")
  668. out_file_class = out_file_base.replace(".glsl.h","").title().replace("_","").replace(".","")+"Shader"+class_suffix;
  669. fd.write("\n\n");
  670. fd.write("#include \"" + include + "\"\n\n\n");
  671. fd.write("class "+out_file_class+" : public Shader"+class_suffix+" {\n\n");
  672. fd.write("\t virtual String get_shader_name() const { return \""+out_file_class+"\"; }\n");
  673. fd.write("public:\n\n");
  674. if (len(header_data.conditionals)):
  675. fd.write("\tenum Conditionals {\n");
  676. for x in header_data.conditionals:
  677. fd.write("\t\t"+x.upper()+",\n");
  678. fd.write("\t};\n\n");
  679. if (len(header_data.uniforms)):
  680. fd.write("\tenum Uniforms {\n");
  681. for x in header_data.uniforms:
  682. fd.write("\t\t"+x.upper()+",\n");
  683. fd.write("\t};\n\n");
  684. fd.write("\t_FORCE_INLINE_ int get_uniform(Uniforms p_uniform) const { return _get_uniform(p_uniform); }\n\n");
  685. if (len(header_data.conditionals)):
  686. fd.write("\t_FORCE_INLINE_ void set_conditional(Conditionals p_conditional,bool p_enable) { _set_conditional(p_conditional,p_enable); }\n\n");
  687. fd.write("\t#define _FU if (get_uniform(p_uniform)<0) return; ERR_FAIL_COND( get_active()!=this );\n\n ");
  688. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, float p_value) { _FU glUniform1f(get_uniform(p_uniform),p_value); }\n\n");
  689. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, double p_value) { _FU glUniform1f(get_uniform(p_uniform),p_value); }\n\n");
  690. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, uint8_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  691. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, int8_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  692. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, uint16_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  693. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, int16_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  694. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, uint32_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  695. fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, int32_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  696. #fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, uint64_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  697. #fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, int64_t p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  698. #fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, unsigned long p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  699. #fd.write("\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, long p_value) { _FU glUniform1i(get_uniform(p_uniform),p_value); }\n\n");
  700. fd.write("\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");
  701. fd.write("\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");
  702. fd.write("\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");
  703. fd.write("\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");
  704. fd.write("\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");
  705. fd.write("\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");
  706. fd.write("""\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const Transform& p_transform) { _FU
  707. const Transform &tr = p_transform;
  708. GLfloat matrix[16]={ /* build a 16x16 matrix */
  709. tr.basis.elements[0][0],
  710. tr.basis.elements[1][0],
  711. tr.basis.elements[2][0],
  712. 0,
  713. tr.basis.elements[0][1],
  714. tr.basis.elements[1][1],
  715. tr.basis.elements[2][1],
  716. 0,
  717. tr.basis.elements[0][2],
  718. tr.basis.elements[1][2],
  719. tr.basis.elements[2][2],
  720. 0,
  721. tr.origin.x,
  722. tr.origin.y,
  723. tr.origin.z,
  724. 1
  725. };
  726. glUniformMatrix4fv(get_uniform(p_uniform),1,false,matrix);
  727. }
  728. """);
  729. fd.write("""\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const Matrix32& p_transform) { _FU
  730. const Matrix32 &tr = p_transform;
  731. GLfloat matrix[16]={ /* build a 16x16 matrix */
  732. tr.elements[0][0],
  733. tr.elements[0][1],
  734. 0,
  735. 0,
  736. tr.elements[1][0],
  737. tr.elements[1][1],
  738. 0,
  739. 0,
  740. 0,
  741. 0,
  742. 1,
  743. 0,
  744. tr.elements[2][0],
  745. tr.elements[2][1],
  746. 0,
  747. 1
  748. };
  749. glUniformMatrix4fv(get_uniform(p_uniform),1,false,matrix);
  750. }
  751. """);
  752. fd.write("""\t_FORCE_INLINE_ void set_uniform(Uniforms p_uniform, const CameraMatrix& p_matrix) { _FU
  753. GLfloat matrix[16];
  754. for (int i=0;i<4;i++) {
  755. for (int j=0;j<4;j++) {
  756. matrix[i*4+j]=p_matrix.matrix[i][j];
  757. }
  758. }
  759. glUniformMatrix4fv(get_uniform(p_uniform),1,false,matrix);
  760. }; """);
  761. fd.write("\n\n#undef _FU\n\n\n");
  762. fd.write("\tvirtual void init() {\n\n");
  763. enum_value_count=0;
  764. if (len(header_data.enums)):
  765. fd.write("\t\t//Written using math, given nonstandarity of 64 bits integer constants..\n");
  766. fd.write("\t\tstatic const Enum _enums[]={\n")
  767. bitofs=len(header_data.conditionals)
  768. enum_vals=[]
  769. for xv in header_data.enums:
  770. x=header_data.enums[xv]
  771. bits=1
  772. amt = len(x);
  773. # print(x)
  774. while(2**bits < amt):
  775. bits+=1
  776. # print("amount: "+str(amt)+" bits "+str(bits));
  777. strs="{"
  778. for i in range(amt):
  779. strs+="\"#define "+x[i]+"\\n\","
  780. v={}
  781. v["set_mask"]="uint64_t("+str(i)+")<<"+str(bitofs)
  782. v["clear_mask"]="((uint64_t(1)<<40)-1) ^ (((uint64_t(1)<<"+str(bits)+") - 1)<<"+str(bitofs)+")"
  783. enum_vals.append(v)
  784. enum_constants.append(x[i])
  785. strs+="NULL}"
  786. fd.write("\t\t\t{(uint64_t(1<<"+str(bits)+")-1)<<"+str(bitofs)+","+str(bitofs)+","+strs+"},\n");
  787. bitofs+=bits
  788. fd.write("\t\t};\n\n");
  789. fd.write("\t\tstatic const EnumValue _enum_values[]={\n")
  790. enum_value_count=len(enum_vals);
  791. for x in enum_vals:
  792. fd.write("\t\t\t{"+x["set_mask"]+","+x["clear_mask"]+"},\n");
  793. fd.write("\t\t};\n\n");
  794. else:
  795. fd.write("\t\tstatic const Enum *_enums=NULL;\n")
  796. fd.write("\t\tstatic const EnumValue *_enum_values=NULL;\n")
  797. if (len(header_data.conditionals)):
  798. fd.write("\t\tstatic const char* _conditional_strings[]={\n")
  799. if (len(header_data.conditionals)):
  800. for x in header_data.conditionals:
  801. fd.write("\t\t\t\"#define "+x+"\\n\",\n");
  802. fd.write("\t\t};\n\n");
  803. else:
  804. fd.write("\t\tstatic const char **_conditional_strings=NULL;\n")
  805. if (len(header_data.uniforms)):
  806. fd.write("\t\tstatic const char* _uniform_strings[]={\n")
  807. if (len(header_data.uniforms)):
  808. for x in header_data.uniforms:
  809. fd.write("\t\t\t\""+x+"\",\n");
  810. fd.write("\t\t};\n\n");
  811. else:
  812. fd.write("\t\tstatic const char **_uniform_strings=NULL;\n")
  813. if output_attribs:
  814. if (len(header_data.attributes)):
  815. fd.write("\t\tstatic AttributePair _attribute_pairs[]={\n")
  816. for x in header_data.attributes:
  817. fd.write("\t\t\t{\""+x[0]+"\","+x[1]+"},\n");
  818. fd.write("\t\t};\n\n");
  819. else:
  820. fd.write("\t\tstatic AttributePair *_attribute_pairs=NULL;\n")
  821. if (len(header_data.texunits)):
  822. fd.write("\t\tstatic TexUnitPair _texunit_pairs[]={\n")
  823. for x in header_data.texunits:
  824. fd.write("\t\t\t{\""+x[0]+"\","+x[1]+"},\n");
  825. fd.write("\t\t};\n\n");
  826. else:
  827. fd.write("\t\tstatic TexUnitPair *_texunit_pairs=NULL;\n")
  828. if (len(header_data.ubos)):
  829. fd.write("\t\tstatic UBOPair _ubo_pairs[]={\n")
  830. for x in header_data.ubos:
  831. fd.write("\t\t\t{\""+x[0]+"\","+x[1]+"},\n");
  832. fd.write("\t\t};\n\n");
  833. else:
  834. fd.write("\t\tstatic UBOPair *_ubo_pairs=NULL;\n")
  835. fd.write("\t\tstatic const char _vertex_code[]={\n")
  836. for x in header_data.vertex_lines:
  837. for i in range(len(x)):
  838. fd.write(str(ord(x[i]))+",");
  839. fd.write(str(ord('\n'))+",");
  840. fd.write("\t\t0};\n\n");
  841. fd.write("\t\tstatic const int _vertex_code_start="+str(header_data.vertex_offset)+";\n")
  842. fd.write("\t\tstatic const char _fragment_code[]={\n")
  843. for x in header_data.fragment_lines:
  844. for i in range(len(x)):
  845. fd.write(str(ord(x[i]))+",");
  846. fd.write(str(ord('\n'))+",");
  847. fd.write("\t\t0};\n\n");
  848. fd.write("\t\tstatic const int _fragment_code_start="+str(header_data.fragment_offset)+";\n")
  849. if output_attribs:
  850. fd.write("\t\tsetup(_conditional_strings,"+str(len(header_data.conditionals))+",_uniform_strings,"+str(len(header_data.uniforms))+",_attribute_pairs,"+str(len(header_data.attributes))+", _texunit_pairs,"+str(len(header_data.texunits))+",_ubo_pairs,"+str(len(header_data.ubos))+",_vertex_code,_fragment_code,_vertex_code_start,_fragment_code_start);\n")
  851. else:
  852. fd.write("\t\tsetup(_conditional_strings,"+str(len(header_data.conditionals))+",_uniform_strings,"+str(len(header_data.uniforms))+",_texunit_pairs,"+str(len(header_data.texunits))+",_enums,"+str(len(header_data.enums))+",_enum_values,"+str(enum_value_count)+",_ubo_pairs,"+str(len(header_data.ubos))+",_vertex_code,_fragment_code,_vertex_code_start,_fragment_code_start);\n")
  853. fd.write("\t};\n\n")
  854. if (len(enum_constants)):
  855. fd.write("\tenum EnumConditionals {\n")
  856. for x in enum_constants:
  857. fd.write("\t\t"+x.upper()+",\n");
  858. fd.write("\t};\n\n");
  859. fd.write("\tvoid set_enum_conditional(EnumConditionals p_cond) { _set_enum_conditional(p_cond); }\n")
  860. fd.write("};\n\n");
  861. fd.write("#endif\n\n");
  862. fd.close();
  863. def build_legacygl_headers( target, source, env ):
  864. for x in source:
  865. build_legacygl_header(str(x), include = "drivers/legacygl/shader_lgl.h", class_suffix = "LGL", output_attribs = False);
  866. return 0
  867. def build_gles2_headers( target, source, env ):
  868. for x in source:
  869. build_legacygl_header(str(x), include="drivers/gles2/shader_gles2.h", class_suffix = "GLES2", output_attribs = True)
  870. def build_gles3_headers( target, source, env ):
  871. for x in source:
  872. build_legacygl_header(str(x), include="drivers/gles3/shader_gles3.h", class_suffix = "GLES3", output_attribs = True)
  873. def update_version():
  874. rev = "custom_build"
  875. if (os.getenv("BUILD_REVISION")!=None):
  876. rev=os.getenv("BUILD_REVISION")
  877. print("Using custom revision: "+rev)
  878. import version
  879. f=open("core/version.h","wb")
  880. f.write("#define VERSION_SHORT_NAME "+str(version.short_name)+"\n")
  881. f.write("#define VERSION_NAME "+str(version.name)+"\n")
  882. f.write("#define VERSION_MAJOR "+str(version.major)+"\n")
  883. f.write("#define VERSION_MINOR "+str(version.minor)+"\n")
  884. if (hasattr(version, 'patch')):
  885. f.write("#define VERSION_PATCH "+str(version.patch)+"\n")
  886. f.write("#define VERSION_REVISION "+str(rev)+"\n")
  887. f.write("#define VERSION_STATUS "+str(version.status)+"\n")
  888. import datetime
  889. f.write("#define VERSION_YEAR "+str(datetime.datetime.now().year)+"\n")
  890. def parse_cg_file(fname, uniforms, sizes, conditionals):
  891. import re
  892. fs = open(fname, "r")
  893. line=fs.readline()
  894. while line:
  895. if re.match(r"^\s*uniform", line):
  896. res = re.match(r"uniform ([\d\w]*) ([\d\w]*)")
  897. type = res.groups(1)
  898. name = res.groups(2)
  899. uniforms.append(name);
  900. if (type.find("texobj") != -1):
  901. sizes.append(1);
  902. else:
  903. t = re.match(r"float(\d)x(\d)", type);
  904. if t:
  905. sizes.append(int(t.groups(1)) * int(t.groups(2)))
  906. else:
  907. t = re.match(r"float(\d)", type);
  908. sizes.append(int(t.groups(1)))
  909. if line.find("[branch]") != -1:
  910. conditionals.append(name);
  911. line = fs.readline();
  912. def build_cg_shader(sname):
  913. vp_uniforms = []
  914. vp_uniform_sizes = []
  915. vp_conditionals = []
  916. parse_cg_file("vp_"+sname+".cg", vp_uniforms, vp_uniform_sizes, vp_conditionals);
  917. fp_uniforms = []
  918. fp_uniform_sizes = []
  919. fp_conditionals = []
  920. parse_cg_file("fp_"+sname+".cg", fp_uniforms, fp_uniform_sizes, fp_conditionals);
  921. fd = open("shader_"+sname+".cg.h", "w");
  922. fd.write('\n#include "shader_cell.h"\n');
  923. fd.write("\nclass Shader_" + sname + " : public ShaderCell {\n");
  924. fd.write("\n\tstatic struct VertexUniforms[] = {\n");
  925. offset = 0;
  926. for i in range(0, len(vp_uniforms)):
  927. fd.write('\t\t{ "%s", %d, %d },\n' % (vp_uniforms[i], offset, vp_uniform_sizes[i]))
  928. offset = offset + vp_uniform_sizes[i];
  929. fd.write("\t};\n\n");
  930. fd.write("public:\n\n");
  931. fd.write("\tenum {\n");
  932. for i in range(0, len(vp_uniforms)):
  933. fd.write('\t\tVP_%s,\n' % vp_uniforms[i].upper())
  934. fd.write("\t};\n");
  935. import glob
  936. def detect_modules():
  937. module_list=[]
  938. includes_cpp=""
  939. register_cpp=""
  940. unregister_cpp=""
  941. files = glob.glob("modules/*")
  942. files.sort() #so register_module_types does not change that often, and also plugins are registered in alphabetic order
  943. for x in files:
  944. if (not os.path.isdir(x)):
  945. continue
  946. x=x.replace("modules/","") # rest of world
  947. x=x.replace("modules\\","") # win32
  948. module_list.append(x)
  949. try:
  950. with open("modules/"+x+"/register_types.h"):
  951. includes_cpp+='#include "modules/'+x+'/register_types.h"\n'
  952. register_cpp+='#ifdef MODULE_'+x.upper()+'_ENABLED\n'
  953. register_cpp+='\tregister_'+x+'_types();\n'
  954. register_cpp+='#endif\n'
  955. unregister_cpp+='#ifdef MODULE_'+x.upper()+'_ENABLED\n'
  956. unregister_cpp+='\tunregister_'+x+'_types();\n'
  957. unregister_cpp+='#endif\n'
  958. except IOError:
  959. pass
  960. modules_cpp="""
  961. // modules.cpp - THIS FILE IS GENERATED, DO NOT EDIT!!!!!!!
  962. #include "register_module_types.h"
  963. """+includes_cpp+"""
  964. void register_module_types() {
  965. """+register_cpp+"""
  966. }
  967. void unregister_module_types() {
  968. """+unregister_cpp+"""
  969. }
  970. """
  971. f=open("modules/register_module_types.cpp","wb")
  972. f.write(modules_cpp)
  973. return module_list
  974. def win32_spawn(sh, escape, cmd, args, env):
  975. import subprocess
  976. newargs = ' '.join(args[1:])
  977. cmdline = cmd + " " + newargs
  978. startupinfo = subprocess.STARTUPINFO()
  979. #startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
  980. for e in env:
  981. if type(env[e]) != type(""):
  982. env[e] = str(env[e])
  983. proc = subprocess.Popen(cmdline, stdin=subprocess.PIPE, stdout=subprocess.PIPE,
  984. stderr=subprocess.PIPE, startupinfo=startupinfo, shell = False, env = env)
  985. data, err = proc.communicate()
  986. rv = proc.wait()
  987. if rv:
  988. print "====="
  989. print err
  990. print "====="
  991. return rv
  992. """
  993. def win32_spawn(sh, escape, cmd, args, spawnenv):
  994. import win32file
  995. import win32event
  996. import win32process
  997. import win32security
  998. for var in spawnenv:
  999. spawnenv[var] = spawnenv[var].encode('ascii', 'replace')
  1000. sAttrs = win32security.SECURITY_ATTRIBUTES()
  1001. StartupInfo = win32process.STARTUPINFO()
  1002. newargs = ' '.join(map(escape, args[1:]))
  1003. cmdline = cmd + " " + newargs
  1004. # check for any special operating system commands
  1005. if cmd == 'del':
  1006. for arg in args[1:]:
  1007. win32file.DeleteFile(arg)
  1008. exit_code = 0
  1009. else:
  1010. # otherwise execute the command.
  1011. hProcess, hThread, dwPid, dwTid = win32process.CreateProcess(None, cmdline, None, None, 1, 0, spawnenv, None, StartupInfo)
  1012. win32event.WaitForSingleObject(hProcess, win32event.INFINITE)
  1013. exit_code = win32process.GetExitCodeProcess(hProcess)
  1014. win32file.CloseHandle(hProcess);
  1015. win32file.CloseHandle(hThread);
  1016. return exit_code
  1017. """
  1018. def android_add_maven_repository(self,url):
  1019. self.android_maven_repos.append(url)
  1020. def android_add_dependency(self,depline):
  1021. self.android_dependencies.append(depline)
  1022. def android_add_java_dir(self,subpath):
  1023. base_path = self.Dir(".").abspath+"/modules/"+self.current_module+"/"+subpath
  1024. self.android_java_dirs.append(base_path)
  1025. def android_add_res_dir(self,subpath):
  1026. base_path = self.Dir(".").abspath+"/modules/"+self.current_module+"/"+subpath
  1027. self.android_res_dirs.append(base_path)
  1028. def android_add_aidl_dir(self,subpath):
  1029. base_path = self.Dir(".").abspath+"/modules/"+self.current_module+"/"+subpath
  1030. self.android_aidl_dirs.append(base_path)
  1031. def android_add_jni_dir(self,subpath):
  1032. base_path = self.Dir(".").abspath+"/modules/"+self.current_module+"/"+subpath
  1033. self.android_jni_dirs.append(base_path)
  1034. def android_add_default_config(self,config):
  1035. self.android_default_config.append(config)
  1036. def android_add_to_manifest(self,file):
  1037. base_path = self.Dir(".").abspath+"/modules/"+self.current_module+"/"+file
  1038. f = open(base_path,"rb")
  1039. self.android_manifest_chunk+=f.read()
  1040. def android_add_to_permissions(self,file):
  1041. base_path = self.Dir(".").abspath+"/modules/"+self.current_module+"/"+file
  1042. f = open(base_path,"rb")
  1043. self.android_permission_chunk+=f.read()
  1044. def android_add_to_attributes(self,file):
  1045. base_path = self.Dir(".").abspath+"/modules/"+self.current_module+"/"+file
  1046. f = open(base_path,"rb")
  1047. self.android_appattributes_chunk+=f.read()
  1048. def disable_module(self):
  1049. self.disabled_modules.append(self.current_module)
  1050. def use_windows_spawn_fix(self, platform=None):
  1051. if (os.name!="nt"):
  1052. return #not needed, only for windows
  1053. self.split_drivers=True
  1054. import subprocess
  1055. def mySubProcess(cmdline,env):
  1056. prefix = ""
  1057. if(platform == 'javascript'):
  1058. prefix = "python.exe "
  1059. startupinfo = subprocess.STARTUPINFO()
  1060. startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
  1061. proc = subprocess.Popen(prefix + cmdline, stdin=subprocess.PIPE, stdout=subprocess.PIPE,
  1062. stderr=subprocess.PIPE, startupinfo=startupinfo, shell = False, env = env)
  1063. data, err = proc.communicate()
  1064. rv = proc.wait()
  1065. if rv:
  1066. print "====="
  1067. print err
  1068. print "====="
  1069. return rv
  1070. def mySpawn(sh, escape, cmd, args, env):
  1071. newargs = ' '.join(args[1:])
  1072. cmdline = cmd + " " + newargs
  1073. rv=0
  1074. env = {str(key): str(value) for key, value in env.iteritems()}
  1075. if len(cmdline) > 32000 and cmd.endswith("ar") :
  1076. cmdline = cmd + " " + args[1] + " " + args[2] + " "
  1077. for i in range(3,len(args)) :
  1078. rv = mySubProcess( cmdline + args[i], env )
  1079. if rv :
  1080. break
  1081. else:
  1082. rv = mySubProcess( cmdline, env )
  1083. return rv
  1084. self['SPAWN'] = mySpawn
  1085. def save_active_platforms(apnames,ap):
  1086. for x in ap:
  1087. pth = x+"/logo.png"
  1088. # print("open path: "+pth)
  1089. pngf=open(pth,"rb");
  1090. b=pngf.read(1);
  1091. str=" /* AUTOGENERATED FILE, DO NOT EDIT */ \n"
  1092. str+=" static const unsigned char _"+x[9:]+"_logo[]={"
  1093. while(len(b)==1):
  1094. str+=hex(ord(b))
  1095. b=pngf.read(1);
  1096. if (len(b)==1):
  1097. str+=","
  1098. str+="};\n"
  1099. wf = x+"/logo.h"
  1100. logow = open(wf,"wb")
  1101. logow.write(str)
  1102. def colored(sys,env):
  1103. #If the output is not a terminal, do nothing
  1104. if not sys.stdout.isatty():
  1105. return
  1106. colors = {}
  1107. colors['cyan'] = '\033[96m'
  1108. colors['purple'] = '\033[95m'
  1109. colors['blue'] = '\033[94m'
  1110. colors['green'] = '\033[92m'
  1111. colors['yellow'] = '\033[93m'
  1112. colors['red'] = '\033[91m'
  1113. colors['end'] = '\033[0m'
  1114. compile_source_message = '%sCompiling %s==> %s$SOURCE%s' % (colors['blue'], colors['purple'], colors['yellow'], colors['end'])
  1115. java_compile_source_message = '%sCompiling %s==> %s$SOURCE%s' % (colors['blue'], colors['purple'], colors['yellow'], colors['end'])
  1116. compile_shared_source_message = '%sCompiling shared %s==> %s$SOURCE%s' % (colors['blue'], colors['purple'], colors['yellow'], colors['end'])
  1117. link_program_message = '%sLinking Program %s==> %s$TARGET%s' % (colors['red'], colors['purple'], colors['yellow'], colors['end'])
  1118. link_library_message = '%sLinking Static Library %s==> %s$TARGET%s' % (colors['red'], colors['purple'], colors['yellow'], colors['end'])
  1119. ranlib_library_message = '%sRanlib Library %s==> %s$TARGET%s' % (colors['red'], colors['purple'], colors['yellow'], colors['end'])
  1120. link_shared_library_message = '%sLinking Shared Library %s==> %s$TARGET%s' % (colors['red'], colors['purple'], colors['yellow'], colors['end'])
  1121. java_library_message = '%sCreating Java Archive %s==> %s$TARGET%s' % (colors['red'], colors['purple'], colors['yellow'], colors['end'])
  1122. env.Append( CXXCOMSTR=[compile_source_message] )
  1123. env.Append( CCCOMSTR=[compile_source_message] )
  1124. env.Append( SHCCCOMSTR=[compile_shared_source_message] )
  1125. env.Append( SHCXXCOMSTR=[compile_shared_source_message] )
  1126. env.Append( ARCOMSTR=[link_library_message] )
  1127. env.Append( RANLIBCOMSTR=[ranlib_library_message] )
  1128. env.Append( SHLINKCOMSTR=[link_shared_library_message] )
  1129. env.Append( LINKCOMSTR=[link_program_message] )
  1130. env.Append( JARCOMSTR=[java_library_message] )
  1131. env.Append( JAVACCOMSTR=[java_compile_source_message] )
  1132. def detect_visual_c_compiler_version(tools_env):
  1133. # tools_env is the variable scons uses to call tools that execute tasks, SCons's env['ENV'] that executes tasks...
  1134. # (see the SCons documentation for more information on what it does)...
  1135. # in order for this function to be well encapsulated i choose to force it to recieve SCons's TOOLS env (env['ENV']
  1136. # and not scons setup environment (env)... so make sure you call the right environment on it or it will fail to detect
  1137. # the propper vc version that will be called
  1138. # These is no flag to give to visual c compilers to set the architecture, ie scons bits argument (32,64,ARM etc)
  1139. # There are many different cl.exe files that are run, and each one compiles & links to a different architecture
  1140. # As far as I know, the only way to figure out what compiler will be run when Scons calls cl.exe via Program()
  1141. # is to check the PATH varaible and figure out which one will be called first. Code bellow does that and returns:
  1142. # the following string values:
  1143. # "" Compiler not detected
  1144. # "amd64" Native 64 bit compiler
  1145. # "amd64_x86" 64 bit Cross Compiler for 32 bit
  1146. # "x86" Native 32 bit compiler
  1147. # "x86_amd64" 32 bit Cross Compiler for 64 bit
  1148. # There are other architectures, but Godot does not support them currently, so this function does not detect arm/amd64_arm
  1149. # and similar architectures/compilers
  1150. # Set chosen compiler to "not detected"
  1151. vc_chosen_compiler_index = -1
  1152. vc_chosen_compiler_str = ""
  1153. # find() works with -1 so big ifs bellow are needed... the simplest solution, in fact
  1154. # First test if amd64 and amd64_x86 compilers are present in the path
  1155. vc_amd64_compiler_detection_index = tools_env["PATH"].find(tools_env["VCINSTALLDIR"]+"BIN\\amd64;")
  1156. if(vc_amd64_compiler_detection_index > -1):
  1157. vc_chosen_compiler_index = vc_amd64_compiler_detection_index
  1158. vc_chosen_compiler_str = "amd64"
  1159. vc_amd64_x86_compiler_detection_index = tools_env["PATH"].find(tools_env["VCINSTALLDIR"]+"BIN\\amd64_x86;")
  1160. if(vc_amd64_x86_compiler_detection_index > -1
  1161. and (vc_chosen_compiler_index == -1
  1162. or vc_chosen_compiler_index > vc_amd64_x86_compiler_detection_index)):
  1163. vc_chosen_compiler_index = vc_amd64_x86_compiler_detection_index
  1164. vc_chosen_compiler_str = "amd64_x86"
  1165. # Now check the 32 bit compilers
  1166. vc_x86_compiler_detection_index = tools_env["PATH"].find(tools_env["VCINSTALLDIR"]+"BIN;")
  1167. if(vc_x86_compiler_detection_index > -1
  1168. and (vc_chosen_compiler_index == -1
  1169. or vc_chosen_compiler_index > vc_x86_compiler_detection_index)):
  1170. vc_chosen_compiler_index = vc_x86_compiler_detection_index
  1171. vc_chosen_compiler_str = "x86"
  1172. vc_x86_amd64_compiler_detection_index = tools_env["PATH"].find(tools_env['VCINSTALLDIR']+"BIN\\x86_amd64;")
  1173. if(vc_x86_amd64_compiler_detection_index > -1
  1174. and (vc_chosen_compiler_index == -1
  1175. or vc_chosen_compiler_index > vc_x86_amd64_compiler_detection_index)):
  1176. vc_chosen_compiler_index = vc_x86_amd64_compiler_detection_index
  1177. vc_chosen_compiler_str = "x86_amd64"
  1178. # debug help
  1179. #print vc_amd64_compiler_detection_index
  1180. #print vc_amd64_x86_compiler_detection_index
  1181. #print vc_x86_compiler_detection_index
  1182. #print vc_x86_amd64_compiler_detection_index
  1183. #print "chosen "+str(vc_chosen_compiler_index)+ " | "+str(vc_chosen_compiler_str)
  1184. return vc_chosen_compiler_str
  1185. def msvc_is_detected() :
  1186. # looks for VisualStudio env variable
  1187. # or for Visual C++ Build Tools (which is a standalone MSVC)
  1188. return os.getenv("VSINSTALLDIR") or os.getenv("VS100COMNTOOLS") or os.getenv("VS110COMNTOOLS") or os.getenv("VS120COMNTOOLS") or os.getenv("VS140COMNTOOLS");
  1189. def precious_program(env, program, sources, **args):
  1190. program = env.ProgramOriginal(program, sources, **args)
  1191. env.Precious(program)
  1192. return program