Skip to content

Commit

Permalink
Merge pull request #4505 from SaracenOne/glsl_includes
Browse files Browse the repository at this point in the history 
Support for #includes in built-in GL shaders.
  • Loading branch information
@akien-mga
akien-mga committed May 2, 2016
2 parents 5bb7cef + 6c0071b commit 85c2dbe
Showing 1 changed file with 85 additions and 53 deletions.
138 changes: 85 additions & 53 deletions methods.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -664,44 +664,65 @@ def build_hlsl_dx9_headers( target, source, env ):
return 0


def build_legacygl_header( filename, include, class_suffix, output_attribs ):

fs = open(filename,"r")
line=fs.readline()

class LegacyGLHeaderStruct:
vertex_lines=[]
fragment_lines=[]
uniforms=[]
attributes=[]
fbos=[]
conditionals=[]
enums={}
enum_constants=[]
texunits=[]
texunit_names=[]
ubos=[]
ubo_names=[]

reading=""

vertex_included_files=[]
fragment_included_files=[]

line_offset=0
vertex_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):

if (line.find("[vertex]")!=-1):
reading="vertex"
line=fs.readline()
line_offset+=1
vertex_offset=line_offset
header_data.line_offset+=1
header_data.vertex_offset=header_data.line_offset
continue

if (line.find("[fragment]")!=-1):
reading="fragment"
line=fs.readline()
line_offset+=1
fragment_offset=line_offset
header_data.line_offset+=1
header_data.fragment_offset=header_data.line_offset
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):
Expand All @@ -715,13 +736,13 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):
ifdefline = ifdefline.replace("_EN_","_")
line = line.replace("_EN_","_")
# print(enumbase+":"+ifdefline);
if (enumbase not in enums):
enums[enumbase]=[]
if (ifdefline not in enums[enumbase]):
enums[enumbase].append(ifdefline);
if (enumbase not in header_data.enums):
header_data.enums[enumbase]=[]
if (ifdefline not in header_data.enums[enumbase]):
header_data.enums[enumbase].append(ifdefline);

elif (not ifdefline in conditionals):
conditionals+=[ifdefline]
elif (not ifdefline in header_data.conditionals):
header_data.conditionals+=[ifdefline]

if (line.find("uniform")!=-1 and line.lower().find("texunit:")!=-1):
#texture unit
Expand All @@ -743,9 +764,9 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):
#unfiorm array
x = x[ :x.find("[") ]

if (not x in texunit_names):
texunits+=[(x,texunit)]
texunit_names+=[x]
if (not x in header_data.texunit_names):
header_data.texunits+=[(x,texunit)]
header_data.texunit_names+=[x]



Expand All @@ -761,8 +782,8 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):
#unfiorm array
x = x[ :x.find("[") ]

if (not x in uniforms):
uniforms+=[x]
if (not x in header_data.uniforms):
header_data.uniforms+=[x]


if ((line.strip().find("in ")==0 or line.strip().find("attribute ")==0) and line.find("attrib:")!=-1):
Expand All @@ -778,7 +799,7 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):
if (bind.find("attrib:")!=-1):
name=name.strip()
bind=bind.replace("attrib:","").strip()
attributes+=[(name,bind)]
header_data.attributes+=[(name,bind)]


line=line.replace("\r","")
Expand All @@ -788,17 +809,28 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):
#line=line+"\\n\\"

if (reading=="vertex"):
vertex_lines+=[line]
header_data.vertex_lines+=[line]
if (reading=="fragment"):
fragment_lines+=[line]
header_data.fragment_lines+=[line]

line=fs.readline()
line_offset+=1

header_data.line_offset+=1
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"
fd = open(out_file,"w")

enum_constants=[]

fd.write("/* WARNING, THIS FILE WAS GENERATED, DO NOT EDIT */\n");

Expand All @@ -819,21 +851,21 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):

fd.write("public:\n\n");

if (len(conditionals)):
if (len(header_data.conditionals)):
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};\n\n");


if (len(uniforms)):
if (len(header_data.uniforms)):
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};\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#define _FU if (get_uniform(p_uniform)<0) return; ERR_FAIL_COND( get_active()!=this );\n\n ");
Expand Down Expand Up @@ -940,16 +972,16 @@ def build_legacygl_header( filename, include, class_suffix, output_attribs ):

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\tstatic const Enum _enums[]={\n")

bitofs=len(conditionals)
bitofs=len(header_data.conditionals)
enum_vals=[]

for xv in enums:
x=enums[xv]
for xv in header_data.enums:
x=header_data.enums[xv]
bits=1
amt = len(x);
# print(x)
Expand Down Expand Up @@ -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 EnumValue *_enum_values=NULL;\n")

if (len(conditionals)):
if (len(header_data.conditionals)):

fd.write("\t\tstatic const char* _conditional_strings[]={\n")
if (len(conditionals)):
for x in conditionals:
if (len(header_data.conditionals)):
for x in header_data.conditionals:
fd.write("\t\t\t\"#define "+x+"\\n\",\n");
fd.write("\t\t};\n\n");
else:
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")
if (len(uniforms)):
for x in uniforms:
if (len(header_data.uniforms)):
for x in header_data.uniforms:
fd.write("\t\t\t\""+x+"\",\n");
fd.write("\t\t};\n\n");
else:
fd.write("\t\tstatic const char **_uniform_strings=NULL;\n")

if output_attribs:
if (len(attributes)):
if (len(header_data.attributes)):

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};\n\n");
else:
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")
for x in texunits:
for x in header_data.texunits:
fd.write("\t\t\t{\""+x[0]+"\","+x[1]+"},\n");
fd.write("\t\t};\n\n");
else:
fd.write("\t\tstatic TexUnitPair *_texunit_pairs=NULL;\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)):
fd.write(str(ord(x[i]))+",");

fd.write(str(ord('\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")
for x in fragment_lines:
for x in header_data.fragment_lines:
for i in range(len(x)):
fd.write(str(ord(x[i]))+",");

fd.write(str(ord('\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:
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:
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")

Expand Down

0 comments on commit 85c2dbe

Please sign in to comment.