godotengine
/
godot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Support for #includes in built-in GL shaders.

This commit is contained in:
Saracen 2016-05-01 13:02:57 +01:00
parent 8dc9e55707
commit 6c0071be96
1 changed files with 85 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

138
methods.py
View File

@ -664,11 +664,7 @@ def build_hlsl_dx9_headers( target, source, env ):
return 0 return 0
def build_legacygl_header( filename, include, class_suffix, output_attribs ): class LegacyGLHeaderStruct:
fs = open(filename,"r")
line=fs.readline()
vertex_lines=[] vertex_lines=[]
fragment_lines=[] fragment_lines=[]
uniforms=[] uniforms=[]
@ -676,32 +672,57 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):
fbos=[] fbos=[]
conditionals=[] conditionals=[]
enums={} enums={}
enum_constants=[]
texunits=[] texunits=[]
texunit_names=[] texunit_names=[]
ubos=[] ubos=[]
ubo_names=[] ubo_names=[]
reading="" vertex_included_files=[]
fragment_included_files=[]
line_offset=0 line_offset=0
vertex_offset=0 vertex_offset=0
fragment_offset=0 fragment_offset=0
def include_file_in_legacygl_header( filename, header_data, depth ):
fs = open(filename,"r")
line=fs.readline()
reading=""
while(line): while(line):
if (line.find("[vertex]")!=-1): if (line.find("[vertex]")!=-1):
reading="vertex" reading="vertex"
line=fs.readline() line=fs.readline()
line_offset+=1 header_data.line_offset+=1
vertex_offset=line_offset header_data.vertex_offset=header_data.line_offset
continue continue
if (line.find("[fragment]")!=-1): if (line.find("[fragment]")!=-1):
reading="fragment" reading="fragment"
line=fs.readline() line=fs.readline()
line_offset+=1 header_data.line_offset+=1
fragment_offset=line_offset header_data.fragment_offset=header_data.line_offset
continue continue
while(line.find("#include ")!=-1):
includeline = line.replace("#include ","").strip()[1:-1]
import os.path
included_file = os.path.relpath(os.path.dirname(filename) + "/" + includeline)
if (not included_file in header_data.vertex_included_files and reading=="vertex"):
header_data.vertex_included_files+=[included_file]
if(include_file_in_legacygl_header( included_file, header_data, depth + 1 ) == None):
print "Error in file '" + filename + "': #include " + includeline + "could not be found!"
elif (not included_file in header_data.fragment_included_files and reading=="fragment"):
header_data.fragment_included_files+=[included_file]
if(include_file_in_legacygl_header( included_file, header_data, depth + 1 ) == None):
print "Error in file '" + filename + "': #include " + includeline + "could not be found!"
line=fs.readline()
if (line.find("#ifdef ")!=-1 or line.find("#elif defined(")!=-1): if (line.find("#ifdef ")!=-1 or line.find("#elif defined(")!=-1):
if (line.find("#ifdef ")!=-1): if (line.find("#ifdef ")!=-1):
@ -715,13 +736,13 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):
ifdefline = ifdefline.replace("_EN_","_") ifdefline = ifdefline.replace("_EN_","_")
line = line.replace("_EN_","_") line = line.replace("_EN_","_")
# print(enumbase+":"+ifdefline); # print(enumbase+":"+ifdefline);
if (enumbase not in enums): if (enumbase not in header_data.enums):
enums[enumbase]=[] header_data.enums[enumbase]=[]
if (ifdefline not in enums[enumbase]): if (ifdefline not in header_data.enums[enumbase]):
enums[enumbase].append(ifdefline); header_data.enums[enumbase].append(ifdefline);
elif (not ifdefline in conditionals): elif (not ifdefline in header_data.conditionals):
conditionals+=[ifdefline] header_data.conditionals+=[ifdefline]
if (line.find("uniform")!=-1 and line.lower().find("texunit:")!=-1): if (line.find("uniform")!=-1 and line.lower().find("texunit:")!=-1):
#texture unit #texture unit
@ -743,9 +764,9 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):
#unfiorm array #unfiorm array
x = x[ :x.find("[") ] x = x[ :x.find("[") ]
if (not x in texunit_names): if (not x in header_data.texunit_names):
texunits+=[(x,texunit)] header_data.texunits+=[(x,texunit)]
texunit_names+=[x] header_data.texunit_names+=[x]
@ -761,8 +782,8 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):
#unfiorm array #unfiorm array
x = x[ :x.find("[") ] x = x[ :x.find("[") ]
if (not x in uniforms): if (not x in header_data.uniforms):
uniforms+=[x] header_data.uniforms+=[x]
if ((line.strip().find("in ")==0 or line.strip().find("attribute ")==0) and line.find("attrib:")!=-1): if ((line.strip().find("in ")==0 or line.strip().find("attribute ")==0) and line.find("attrib:")!=-1):
@ -778,7 +799,7 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):
if (bind.find("attrib:")!=-1): if (bind.find("attrib:")!=-1):
name=name.strip() name=name.strip()
bind=bind.replace("attrib:","").strip() bind=bind.replace("attrib:","").strip()
attributes+=[(name,bind)] header_data.attributes+=[(name,bind)]
line=line.replace("\r","") line=line.replace("\r","")
@ -788,17 +809,28 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):
#line=line+"\\n\\" #line=line+"\\n\\"
if (reading=="vertex"): if (reading=="vertex"):
vertex_lines+=[line] header_data.vertex_lines+=[line]
if (reading=="fragment"): if (reading=="fragment"):
fragment_lines+=[line] header_data.fragment_lines+=[line]
line=fs.readline() line=fs.readline()
line_offset+=1 header_data.line_offset+=1
fs.close(); fs.close();
return header_data
def build_legacygl_header( filename, include, class_suffix, output_attribs ):
header_data = LegacyGLHeaderStruct()
include_file_in_legacygl_header( filename, header_data, 0 )
out_file = filename+".h" out_file = filename+".h"
fd = open(out_file,"w") fd = open(out_file,"w")
enum_constants=[]
fd.write("/* WARNING, THIS FILE WAS GENERATED, DO NOT EDIT */\n"); fd.write("/* WARNING, THIS FILE WAS GENERATED, DO NOT EDIT */\n");
@ -819,21 +851,21 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):
fd.write("public:\n\n"); fd.write("public:\n\n");
if (len(conditionals)): if (len(header_data.conditionals)):
fd.write("\tenum Conditionals {\n"); fd.write("\tenum Conditionals {\n");
for x in conditionals: for x in header_data.conditionals:
fd.write("\t\t"+x.upper()+",\n"); fd.write("\t\t"+x.upper()+",\n");
fd.write("\t};\n\n"); fd.write("\t};\n\n");
if (len(uniforms)): if (len(header_data.uniforms)):
fd.write("\tenum Uniforms {\n"); fd.write("\tenum Uniforms {\n");
for x in uniforms: for x in header_data.uniforms:
fd.write("\t\t"+x.upper()+",\n"); fd.write("\t\t"+x.upper()+",\n");
fd.write("\t};\n\n"); fd.write("\t};\n\n");
fd.write("\t_FORCE_INLINE_ int get_uniform(Uniforms p_uniform) const { return _get_uniform(p_uniform); }\n\n"); fd.write("\t_FORCE_INLINE_ int get_uniform(Uniforms p_uniform) const { return _get_uniform(p_uniform); }\n\n");
if (len(conditionals)): if (len(header_data.conditionals)):
fd.write("\t_FORCE_INLINE_ void set_conditional(Conditionals p_conditional,bool p_enable) { _set_conditional(p_conditional,p_enable); }\n\n"); fd.write("\t_FORCE_INLINE_ void set_conditional(Conditionals p_conditional,bool p_enable) { _set_conditional(p_conditional,p_enable); }\n\n");
fd.write("\t#define _FU if (get_uniform(p_uniform)<0) return; ERR_FAIL_COND( get_active()!=this );\n\n "); fd.write("\t#define _FU if (get_uniform(p_uniform)<0) return; ERR_FAIL_COND( get_active()!=this );\n\n ");
@ -940,16 +972,16 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):
enum_value_count=0; enum_value_count=0;
if (len(enums)): if (len(header_data.enums)):
fd.write("\t\t//Written using math, given nonstandarity of 64 bits integer constants..\n"); fd.write("\t\t//Written using math, given nonstandarity of 64 bits integer constants..\n");
fd.write("\t\tstatic const Enum _enums[]={\n") fd.write("\t\tstatic const Enum _enums[]={\n")
bitofs=len(conditionals) bitofs=len(header_data.conditionals)
enum_vals=[] enum_vals=[]
for xv in enums: for xv in header_data.enums:
x=enums[xv] x=header_data.enums[xv]
bits=1 bits=1
amt = len(x); amt = len(x);
# print(x) # print(x)
@ -985,70 +1017,70 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):
fd.write("\t\tstatic const Enum *_enums=NULL;\n") fd.write("\t\tstatic const Enum *_enums=NULL;\n")
fd.write("\t\tstatic const EnumValue *_enum_values=NULL;\n") fd.write("\t\tstatic const EnumValue *_enum_values=NULL;\n")
if (len(conditionals)): if (len(header_data.conditionals)):
fd.write("\t\tstatic const char* _conditional_strings[]={\n") fd.write("\t\tstatic const char* _conditional_strings[]={\n")
if (len(conditionals)): if (len(header_data.conditionals)):
for x in conditionals: for x in header_data.conditionals:
fd.write("\t\t\t\"#define "+x+"\\n\",\n"); fd.write("\t\t\t\"#define "+x+"\\n\",\n");
fd.write("\t\t};\n\n"); fd.write("\t\t};\n\n");
else: else:
fd.write("\t\tstatic const char **_conditional_strings=NULL;\n") fd.write("\t\tstatic const char **_conditional_strings=NULL;\n")
if (len(uniforms)): if (len(header_data.uniforms)):
fd.write("\t\tstatic const char* _uniform_strings[]={\n") fd.write("\t\tstatic const char* _uniform_strings[]={\n")
if (len(uniforms)): if (len(header_data.uniforms)):
for x in uniforms: for x in header_data.uniforms:
fd.write("\t\t\t\""+x+"\",\n"); fd.write("\t\t\t\""+x+"\",\n");
fd.write("\t\t};\n\n"); fd.write("\t\t};\n\n");
else: else:
fd.write("\t\tstatic const char **_uniform_strings=NULL;\n") fd.write("\t\tstatic const char **_uniform_strings=NULL;\n")
if output_attribs: if output_attribs:
if (len(attributes)): if (len(header_data.attributes)):
fd.write("\t\tstatic AttributePair _attribute_pairs[]={\n") fd.write("\t\tstatic AttributePair _attribute_pairs[]={\n")
for x in attributes: for x in header_data.attributes:
fd.write("\t\t\t{\""+x[0]+"\","+x[1]+"},\n"); fd.write("\t\t\t{\""+x[0]+"\","+x[1]+"},\n");
fd.write("\t\t};\n\n"); fd.write("\t\t};\n\n");
else: else:
fd.write("\t\tstatic AttributePair *_attribute_pairs=NULL;\n") fd.write("\t\tstatic AttributePair *_attribute_pairs=NULL;\n")
if (len(texunits)): if (len(header_data.texunits)):
fd.write("\t\tstatic TexUnitPair _texunit_pairs[]={\n") fd.write("\t\tstatic TexUnitPair _texunit_pairs[]={\n")
for x in texunits: for x in header_data.texunits:
fd.write("\t\t\t{\""+x[0]+"\","+x[1]+"},\n"); fd.write("\t\t\t{\""+x[0]+"\","+x[1]+"},\n");
fd.write("\t\t};\n\n"); fd.write("\t\t};\n\n");
else: else:
fd.write("\t\tstatic TexUnitPair *_texunit_pairs=NULL;\n") fd.write("\t\tstatic TexUnitPair *_texunit_pairs=NULL;\n")
fd.write("\t\tstatic const char _vertex_code[]={\n") fd.write("\t\tstatic const char _vertex_code[]={\n")
for x in vertex_lines: for x in header_data.vertex_lines:
for i in range(len(x)): for i in range(len(x)):
fd.write(str(ord(x[i]))+","); fd.write(str(ord(x[i]))+",");
fd.write(str(ord('\n'))+","); fd.write(str(ord('\n'))+",");
fd.write("\t\t0};\n\n"); fd.write("\t\t0};\n\n");
fd.write("\t\tstatic const int _vertex_code_start="+str(vertex_offset)+";\n") fd.write("\t\tstatic const int _vertex_code_start="+str(header_data.vertex_offset)+";\n")
fd.write("\t\tstatic const char _fragment_code[]={\n") fd.write("\t\tstatic const char _fragment_code[]={\n")
for x in fragment_lines: for x in header_data.fragment_lines:
for i in range(len(x)): for i in range(len(x)):
fd.write(str(ord(x[i]))+","); fd.write(str(ord(x[i]))+",");
fd.write(str(ord('\n'))+","); fd.write(str(ord('\n'))+",");
fd.write("\t\t0};\n\n"); fd.write("\t\t0};\n\n");
fd.write("\t\tstatic const int _fragment_code_start="+str(fragment_offset)+";\n") fd.write("\t\tstatic const int _fragment_code_start="+str(header_data.fragment_offset)+";\n")
if output_attribs: if output_attribs:
fd.write("\t\tsetup(_conditional_strings,"+str(len(conditionals))+",_uniform_strings,"+str(len(uniforms))+",_attribute_pairs,"+str(len(attributes))+", _texunit_pairs,"+str(len(texunits))+",_vertex_code,_fragment_code,_vertex_code_start,_fragment_code_start);\n") 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))+",_vertex_code,_fragment_code,_vertex_code_start,_fragment_code_start);\n")
else: else:
fd.write("\t\tsetup(_conditional_strings,"+str(len(conditionals))+",_uniform_strings,"+str(len(uniforms))+",_texunit_pairs,"+str(len(texunits))+",_enums,"+str(len(enums))+",_enum_values,"+str(enum_value_count)+",_vertex_code,_fragment_code,_vertex_code_start,_fragment_code_start);\n") 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)+",_vertex_code,_fragment_code,_vertex_code_start,_fragment_code_start);\n")
fd.write("\t};\n\n") fd.write("\t};\n\n")