+#!/usr/bin/python
+# $Id$
+
+import sys
+import os
+
+class InputFile:
+ def __init__(self, fn):
+ self.file = open(fn)
+
+ def __iter__(self):
+ for l in self.file:
+ h = l.find("#")
+ if h==0 or (h>0 and l[h-1].isspace()):
+ l = l[:h]
+
+ l = l.rstrip()
+ if not l:
+ continue
+
+ yield l
+
+
+def strip_name(name):
+ """Strips any vendor suffix and GL prefix from a name (but not GLX prefix)"""
+
+ suffix = ""
+ if name.endswith(" *"):
+ suffix = " *"
+ name = name[:-2]
+ elif name.endswith("Pointer"):
+ suffix = "Pointer"
+ name = name[:-7]
+
+ prefix = ""
+ if name.startswith("const "):
+ prefix = "const "
+ name = name[6:]
+
+ if name.startswith("GL") and not name.startswith("GLX"):
+ name = name[2:]
+ if name.endswith(("EXT", "ARB", "SGI", "IBM", "ATI")):
+ return prefix+name[:-3]+suffix
+ elif name.endswith("SGIX"):
+ return prefix+name[:-4]+suffix
+ elif name.endswith(("NV", "HP")):
+ return prefix+name[:-2]+suffix
+ else:
+ return prefix+name+suffix
+
+
+class Typemap:
+ def __init__(self, fn):
+ self.map = {}
+ for line in InputFile(fn):
+ parts = [p.strip() for p in line.split(',')]
+ if parts[3]=="*":
+ parts[3] = parts[0]
+ elif parts[3][-1]=='*' and parts[3][-2]!=' ':
+ parts[3] = parts[3][:-1]+" *"
+ self.map[tuple(parts[0:3])] = tuple(parts[3:6])
+
+ def wildcard_match(self, a, b):
+ if a=="*" or b=="*":
+ return True
+ return a==b
+
+ def __getitem__(self, key):
+ try:
+ return self.map[(key[0], "*", "*")]
+ except KeyError:
+ for k, v in self.map.iteritems():
+ if strip_name(k[0])==strip_name(key[0]) and self.wildcard_match(k[1], key[1]) and self.wildcard_match(k[2], key[2]):
+ return v
+ raise KeyError, key
+
+ def update(self, other):
+ self.map.update(other.map)
+
+
+class IOmap:
+ def __init__(self, fn):
+ self.map = {}
+ self.map["void"] = None
+ for line in InputFile(fn):
+ parts = [p.strip() for p in line.split(',')]
+ self.map[parts[0]] = tuple(parts[1:])
+
+ def __getitem__(self, key):
+ return self.map[strip_name(key)]
+
+
+class Function:
+ class Parameter:
+ def __init__(self, func, name):
+ self.func = func
+ self.name = name
+ self.type = None
+ self.direction = "in"
+ self.kind = "value"
+ self.size = None
+ self.ctype = None
+ self.csize = None
+ self.io = None
+
+ def set_type(self, type, dir, kind):
+ self.type = type
+ self.direction = dir
+ self.kind = kind
+
+ def set_size(self, size):
+ if type(size)==str and size.isdigit():
+ self.size = int(size)
+ else:
+ self.size = size
+
+ def derive_ctype(self, typemap, iomap):
+ m = typemap[(self.type, self.direction, self.kind)]
+ self.ctype = m[0]
+ if m[1]!="*":
+ self.direction = m[1]
+ if m[2]!="*":
+ self.kind = m[2]
+ self.base_ctype = self.ctype
+ if self.kind=="value":
+ if self.base_ctype.startswith("const "):
+ self.base_ctype = self.base_ctype[6:]
+ else:
+ if self.direction=="in":
+ self.ctype = "const "+self.ctype
+ self.ctype = self.ctype+" *"
+ self.io = iomap[self.base_ctype]
+
+ def derive_csize(self):
+ if self.kind=="array" and self.size is not None:
+ self.csize = None
+ if type(self.size)==int:
+ self.csize = "%d*sizeof(%s)"%(self.size, self.base_ctype)
+ elif self.size.startswith("COMPSIZE("):
+ self.csize = self.func.compsize(self.size[9:-1], self.base_ctype)
+ elif self.size=="pname":
+ self.csize = "paramsize(pname)*sizeof(%s)"%self.base_ctype
+ else:
+ s = self.func.get_param(self.size.split('*')[0])
+ if (s.type=="SizeI" or s.type.endswith("Int32") or s.type.startswith("BufferSize")):
+ if s.kind=="value":
+ self.csize = "%s"%self.size
+ if self.func.name.startswith("glUniformMatrix") and self.func.name[16]!='x':
+ self.csize += "*%d"%(int(self.func.name[15])**2)
+ elif self.func.name.startswith("glUniform") and self.func.name[9].isdigit():
+ self.csize += "*%s"%self.func.name[9]
+ if strip_name(self.base_ctype)!="void":
+ self.csize += "*sizeof(%s)"%self.base_ctype
+ elif s.kind=="array" and s.size==1:
+ self.csize = "*%s"%s.name
+ if not self.csize:
+ sys.stderr.write("Could not determine size for array parameter '%s[%s]' of function '%s'\n"%(self.name, self.size, self.func.name))
+ elif self.kind=="reference":
+ self.csize = "sizeof(%s)"%self.base_ctype
+
+ def __init__(self, name, pnames):
+ self.name = name
+ self.ret = Function.Parameter(self, "ret")
+ self.params = [Function.Parameter(self, n) for n in pnames]
+ self.category = None
+
+ def get_param(self, pname):
+ for p in self.params:
+ if p.name==pname:
+ return p
+ raise KeyError, pname
+
+ def set_category(self, cat):
+ self.category = cat
+
+ def compsize(self, size, btype):
+ if not size:
+ return
+
+ res = ""
+ have_type = False
+ for c in size.replace(',', '/').split('/'):
+ param = self.get_param(c)
+ if not param:
+ sys.stderr.write("Compsize '%s' for function '%s' failed: No parameter '%s'\n"%(size, self.name, c))
+ return
+
+ if res:
+ res += "*"
+
+ cn = strip_name(param.type)
+ if cn.endswith("Type"):
+ res += "typesize(%s)"%param.name
+ have_type = True
+ elif cn.endswith("Format"):
+ res += "formatsize(%s)"%param.name
+ elif param.name=="pname" or cn.endswith("Parameter") or (param.name=="target" and cn=="enum"):
+ res += "paramsize(%s)"%param.name
+ elif param.name=="buffer" and cn=="enum":
+ res += "buffersize(%s)"%param.name
+ elif (cn=="SizeI" or cn.endswith("Int32")) and not param.size:
+ res += param.name
+ else:
+ sys.stderr.write("Compsize '%s' for function '%s' failed: Parameter '%s' has unknown type '%s'\n"%(size, self.name, param.name, param.type))
+ return
+ if not have_type:
+ res += "*sizeof(%s)"%btype
+ return res
+
+ def finalize(self, typemap, iomap):
+ self.ret.derive_ctype(typemap, iomap)
+ for p in self.params:
+ p.derive_ctype(typemap, iomap)
+ for p in self.params:
+ p.derive_csize()
+
+
+class Enum:
+ def __init__(self, name, category, value):
+ self.name = name
+ self.category = category
+ self.value = value
+
+
+class Template:
+ def __init__(self, fn):
+ self.mode = "functions"
+ self.sections = []
+ self.handcode = []
+
+ literal = True
+ text = ""
+ for line in InputFile(fn):
+ if line[0]==':':
+ if not literal and text:
+ self.add_section(text, literal)
+ text = ""
+ text += line[1:]+"\n"
+ literal = True
+ elif line[0]=='!':
+ parts = line[1:].split()
+ if parts[0]=="handcode":
+ self.handcode.append(parts[1])
+ elif parts[0]=="mode":
+ self.mode = parts[1]
+ else:
+ sys.stderr.write("Unknown keyword '%s'\n"%parts[0])
+ else:
+ if literal and text:
+ self.add_section(text, literal)
+ text = ""
+ text += line+"\n"
+ literal = False
+ if text:
+ self.add_section(text, literal)
+
+ def add_section(self, text, literal):
+ if literal:
+ self.sections.append(text)
+ else:
+ self.sections.append(compile(text, "-", "exec"))
+
+ def write(self, str, *args):
+ sys.stdout.write(str%args)
+
+ def writeln(self, str, *args):
+ sys.stdout.write(str%args+"\n")
+
+ def process_function(self, code, func):
+ if func.name in self.handcode:
+ return
+
+ globals = {
+ "w": self.write,
+ "wl": self.writeln,
+ "func": func,
+ "ret": func.ret,
+ "params": func.params
+ }
+ eval(code, globals)
+
+ def process_enum(self, code, enum):
+ if enum.value is None:
+ return
+
+ globals = {
+ "w": self.write,
+ "wl": self.writeln,
+ "enum": enum
+ }
+ eval(code, globals)
+
+ def process_sections(self, objects, handler):
+ for sect in self.sections:
+ if type(sect)==str:
+ print sect
+ else:
+ for obj in objects:
+ handler(sect, obj)
+
+ def process(self, apis):
+ if self.mode=="functions":
+ functions = []
+ for api in apis:
+ typemap = Typemap(api.typemap)
+ iomap = IOmap(api.iomap)
+ for spec in api.specs:
+ funcs = read_spec(spec, api.prefix)
+ funcs = [f for f in funcs if f.name not in api.ignore_funcs and f.category not in api.ignore_categs]
+ for func in funcs:
+ func.finalize(typemap, iomap)
+ names = [f.name for f in funcs]
+ functions = [f for f in functions if f.name not in names]+funcs
+ self.process_sections(functions, self.process_function)
+ elif self.mode=="enums":
+ enums = []
+ for api in apis:
+ for spec in api.enumspecs:
+ ens = read_enums(spec, api.enumprefix)
+ enums += ens
+ enums.sort(lambda x, y: cmp(x.value, y.value)*2+cmp(x.category, y.category))
+ self.process_sections(enums, self.process_enum)
+
+
+class Api:
+ def __init__(self, fn):
+ self.typemap = None
+ self.iomap = None
+ self.specs = []
+ self.prefix = None
+ self.enumspecs = []
+ self.enumprefix = None
+ self.ignore_categs = []
+ self.ignore_funcs = []
+
+ for line in InputFile(fn):
+ parts = line.split()
+ if parts[0]=="typemap":
+ self.typemap = parts[1]
+ elif parts[0]=="iomap":
+ self.iomap = parts[1]
+ elif parts[0]=="spec":
+ self.specs.append(parts[1])
+ elif parts[0]=="prefix":
+ self.prefix = parts[1]
+ elif parts[0]=="enumspec":
+ self.enumspecs.append(parts[1])
+ elif parts[0]=="enumprefix":
+ self.enumprefix = parts[1]
+ elif parts[0]=="ignore":
+ if parts[1]=="category":
+ self.ignore_categs.append(parts[2])
+ elif parts[1]=="function":
+ self.ignore_funcs.append(parts[2])
+ else:
+ sys.stderr.write("Unknown keyword '%s'\n", parts[0])
+
+
+def read_spec(fn, prefix):
+ funcs = []
+ cur_func = None
+
+ for line in InputFile(fn):
+ if line.find(':')>=0:
+ continue
+ elif line[0]=='\t' and cur_func:
+ parts = line.split()
+ if parts[0]=="return":
+ cur_func.ret.set_type(parts[1], "out", "value")
+ elif parts[0]=="param":
+ bracket = parts[4].find('[')
+ if bracket>=0:
+ parts.insert(5, parts[4][bracket:])
+ parts[4] = parts[4][:bracket]
+
+ param = cur_func.get_param(parts[1])
+ param.set_type(parts[2], parts[3], parts[4])
+ if len(parts)==6 or (len(parts)>6 and parts[6]!="retained"):
+ param.set_size(parts[5][1:-1])
+ elif parts[0]=="category":
+ cur_func.set_category(parts[1])
+ elif parts[0]=="glxvendorglx" and cur_func.category=="glx":
+ cur_func.set_category("glxext")
+ else:
+ paren = line.find('(')
+ if paren>0:
+ cparen = line.rfind(')')
+ if cparen>paren+1:
+ pnames = [n.strip() for n in line[paren+1:cparen].split(",")]
+ else:
+ pnames = []
+ cur_func = Function(prefix+line[:paren], pnames)
+ funcs.append(cur_func)
+
+ return funcs
+
+def read_enums(fn, prefix):
+ enums = []
+ cur_categ = []
+
+ for line in InputFile(fn):
+ if ':' in line:
+ parts = line[:line.find(':')].split()
+ if len(parts)==2 and parts[1]=="enum":
+ cur_categ = parts[0]
+ elif cur_categ:
+ parts = line.split()
+ if parts[0]=="use":
+ enums.append(Enum(prefix+parts[2], cur_categ, None))
+ elif parts[1]=="=":
+ try:
+ enums.append(Enum(prefix+parts[0], cur_categ, int(parts[2], 0)))
+ except ValueError, e:
+ sys.stderr.write("Syntax error in %s: %s\n"%(fn, e))
+
+ for e in enums:
+ if e.value is None:
+ for n in enums:
+ if n.name==e.name and n.value is not None:
+ e.value = n.value
+ if e.value is None:
+ sys.stderr.write("Could not find value for enum reference %s in category %s\n"%(e.name, e.category))
+
+ return enums
+
+template = Template(sys.argv[1])
+apis = []
+for i in sys.argv[2:]:
+ apis.append(Api(i))
+
+template.process(apis)