11 def __init__(self, *args):
19 raise TypeError, "__init__() takes zero or two arguments (%d given)"%len(args)
22 return "%d.%d"%(self.major, self.minor)
25 return "Version(%d, %d)"%(self.major, self.minor)
28 return "VERSION_%d_%d"%(self.major, self.minor)
30 def __lt__(self, other):
34 if self.major!=other.major:
35 return self.major<other.major
36 return self.minor<other.minor
38 def __gt__(self, other):
42 if self.major!=other.major:
43 return self.major>other.major
44 return self.minor>other.minor
46 def __eq__(self, other):
50 return (self.major==other.major and self.minor==other.minor)
59 self.core_version = None
60 self.deprecated_version = None
64 def __init__(self, name, kind):
70 def get_or_create_api_support(self, api):
71 supp = self.api_support.get(api)
73 supp = Thing.ApiSupport()
74 self.api_support[api] = supp
78 class Function(Thing):
79 def __init__(self, name):
80 Thing.__init__(self, name, Thing.FUNCTION)
81 self.return_type = "void"
85 r_bitmask = re.compile("_BIT[0-9]*(_|$)")
88 def __init__(self, name):
89 Thing.__init__(self, name, Thing.ENUM)
91 self.bitmask = bool(r_bitmask.search(self.name))
95 def __init__(self, name, api):
97 underscore = name.find('_')
98 self.ext_type = name[0:underscore]
99 self.base_name = name[underscore+1:]
103 if self.ext_type=="EXT":
105 elif self.ext_type=="ARB" or self.ext_type=="OES":
107 self.backport = False
111 def __init__(self, name):
113 self.latest_version = None
114 self.core_things = {}
118 def get_nested_elements(elem, path):
119 childElements = [c for c in elem.childNodes if c.nodeType==xml.dom.Node.ELEMENT_NODE]
121 head, tail = path.split('/', 1)
123 for c in childElements:
125 result += get_nested_elements(c, tail)
128 return [c for c in childElements if c.tagName==path]
130 def get_first_child(elem, tag):
131 for c in elem.childNodes:
132 if c.nodeType==xml.dom.Node.ELEMENT_NODE and c.tagName==tag:
136 def get_text_contents(node):
138 for c in node.childNodes:
139 if c.nodeType==xml.dom.Node.TEXT_NODE or c.nodeType==xml.dom.Node.CDATA_SECTION_NODE:
142 result += get_text_contents(c)
145 def get_or_create(map, name, type, *args):
148 obj = type(name, *args)
158 def parse_command(self, cmd):
159 proto = get_first_child(cmd, "proto")
160 name = get_text_contents(get_first_child(proto, "name"))
161 func = get_or_create(self.things, name, Function)
163 aliases = get_nested_elements(cmd, "alias")
164 func.aliases = [a.getAttribute("name") for a in aliases]
166 ptype = get_first_child(proto, "ptype")
168 func.return_type = get_text_contents(ptype)
170 for c in proto.childNodes:
171 if c.nodeType==xml.dom.Node.TEXT_NODE and c.data.strip():
172 func.return_type = c.data.strip()
175 params = get_nested_elements(cmd, "param")
176 func.params = map(get_text_contents, params)
178 def parse_enum(self, en):
179 name = en.getAttribute("name")
180 enum = get_or_create(self.things, name, Enum)
182 enum.value = int(en.getAttribute("value"), 16)
184 alias = en.getAttribute("alias")
186 enum.aliases.append(alias)
188 def parse_feature(self, feat):
189 api_name = feat.getAttribute("api")
190 api = get_or_create(self.apis, api_name, Api)
192 version = feat.getAttribute("number")
194 version = Version(*map(int, version.split('.')))
198 requires = get_nested_elements(feat, "require")
200 commands = get_nested_elements(req, "command")
201 enums = get_nested_elements(req, "enum")
202 for t in itertools.chain(commands, enums):
203 name = t.getAttribute("name")
204 thing = self.things.get(name)
206 supp = thing.get_or_create_api_support(api.name)
207 if not supp.core_version or version<supp.core_version:
208 supp.core_version = version
209 api.core_things[thing.name] = thing
211 removes = get_nested_elements(feat, "remove")
213 commands = get_nested_elements(rem, "command")
214 enums = get_nested_elements(rem, "enum")
216 for t in itertools.chain(commands, enums):
217 name = t.getAttribute("name")
218 thing = self.things.get(name)
220 supp = thing.get_or_create_api_support(api.name)
221 supp.deprecated_version = version
223 def parse_extension(self, ext):
224 ext_things_by_api = {}
225 requires = get_nested_elements(ext, "require")
227 api = req.getAttribute("api")
228 ext_things = ext_things_by_api.setdefault(api, [])
230 commands = get_nested_elements(req, "command")
231 enums = get_nested_elements(req, "enum")
232 for t in itertools.chain(commands, enums):
233 name = t.getAttribute("name")
234 thing = self.things.get(name)
236 ext_things.append(thing)
238 ext_name = ext.getAttribute("name")
239 if ext_name.startswith("GL_"):
240 ext_name = ext_name[3:]
242 common_things = ext_things_by_api.get("", [])
243 supported = ext.getAttribute("supported").split('|')
245 api = self.apis.get(s)
249 ext = get_or_create(api.extensions, ext_name, Extension, api)
250 api_things = ext_things_by_api.get(s, [])
251 for t in itertools.chain(common_things, api_things):
252 ext.things[t.name] = t
253 t.get_or_create_api_support(api.name).extensions.append(ext)
255 def parse_file(self, fn):
256 doc = xml.dom.minidom.parse(fn)
257 root = doc.documentElement
259 commands = get_nested_elements(root, "commands/command")
261 self.parse_command(cmd)
263 enums = get_nested_elements(root, "enums/enum")
267 features = get_nested_elements(root, "feature")
268 for feat in features:
269 self.parse_feature(feat)
271 extensions = get_nested_elements(root, "extensions/extension")
272 for ext in extensions:
273 self.parse_extension(ext)
275 def check_backport_extensions(self, api):
276 for e in api.extensions.itervalues():
277 if e.ext_type!="ARB":
281 for t in e.things.itervalues():
282 if t.name.endswith(e.ext_type):
286 def resolve_enum_aliases(self, api):
287 for e in api.extensions.itervalues():
288 ext_enums = filter((lambda t: t.kind==Thing.ENUM), e.things.itervalues())
289 enum_suffix = "_"+e.ext_type
291 if n.api_support[api.name].core_version:
295 if name.endswith(enum_suffix):
296 name = name[:-len(enum_suffix)]
297 ce = api.core_things.get(name)
298 if ce and ce.value==n.value and ce.name not in n.aliases:
299 n.aliases.append(ce.name)
301 def resolve_sources(self, api):
302 for e in api.extensions.itervalues():
303 for t in e.things.itervalues():
305 # There are a few cases where a vendor function is aliased to
306 # an EXT or ARB function but those are rare and not relevant for
308 alias = api.core_things.get(a)
310 sources = alias.api_support[api.name].sources
314 def sort_extensions(self):
315 for t in self.things.itervalues():
316 for s in t.api_support.itervalues():
317 s.extensions.sort(key=(lambda e: e.preference), reverse=True)
320 for a in self.apis.itervalues():
321 self.check_backport_extensions(a)
322 self.resolve_enum_aliases(a)
323 self.resolve_sources(a)
324 self.sort_extensions()
327 def detect_core_version(host_api, things):
328 max_version = Version(1, 0)
333 supp = t.api_support.get(host_api.name)
334 if supp and supp.core_version:
335 if supp.core_version>max_version:
336 max_version = supp.core_version
337 lower_count += max_count
339 elif supp.core_version==max_version:
346 if lower_count>max_count or (missing and len(missing)*2<lower_count+max_count):
347 print "Warning: Inconsistent core version %s"%max_version
354 def detect_deprecated_version(host_api, things):
358 supp = t.api_support.get(host_api.name)
359 if supp and supp.deprecated_version:
360 if min_version is None:
361 min_version = supp.deprecated_version
363 min_version = min(min_version, supp.deprecated_version)
366 if min_version and len(deprecated)*2<len(things):
367 print "Warning: Inconsistent deprecation version %s"%min_version
371 def detect_backport_extension(host_api, things):
374 supp = t.api_support.get(host_api.name)
375 if supp and supp.core_version:
376 for e in supp.extensions:
377 if e.backport and e not in candidates:
380 total_count = len(things)
384 things_in_ext = filter((lambda t: t.name in e.things), things)
385 count = len(things_in_ext)
386 if count==total_count:
388 elif count>best_count:
392 if best_count*2>=total_count:
393 print "Warning: Inconsistent backport extension %s"%best_ext.name
395 def collect_extensions(thing, api, exts):
396 supp = thing.api_support.get(api)
400 for e in supp.extensions:
401 if not e.backport and e.ext_type!="MSP" and e not in exts:
404 for s in supp.sources:
405 collect_extensions(s, api, exts)
407 def detect_source_extension(host_api, things):
411 collect_extensions(t, host_api.name, exts)
413 things_by_ext.setdefault(e, []).append(t)
416 missing = set(things)
417 while missing and things_by_ext:
420 for e, t in things_by_ext.iteritems():
422 if count>largest_count:
424 largest_count = count
425 elif count==largest_count and e.preference>largest_ext.preference:
428 extensions.append(largest_ext)
429 for t in things_by_ext[largest_ext]:
434 del things_by_ext[largest_ext]
435 for e in things_by_ext.keys():
436 unseen = filter((lambda t: t in missing), things_by_ext[e])
438 things_by_ext[e] = unseen
448 class SourceGenerator:
449 def __init__(self, host_api, ext_name, things, optional_things):
450 self.host_api = host_api
451 self.api_prefix = "GL"
452 if self.host_api.name=="gles2":
453 self.api_prefix = "GL_ES"
454 self.ext_name = ext_name
455 all_things = things+optional_things
456 self.funcs = filter((lambda t: t.kind==Thing.FUNCTION), all_things)
457 self.funcs.sort(key=(lambda f: f.name))
458 self.func_typedefs = dict((f.name, "FPtr_"+f.name) for f in self.funcs)
459 self.enums = filter((lambda t: t.kind==Thing.ENUM), all_things)
460 self.enums.sort(key=(lambda e: e.value))
461 self.core_version = detect_core_version(host_api, things)
462 self.deprecated_version = detect_deprecated_version(host_api, things)
463 self.backport_ext = detect_backport_extension(host_api, things);
464 self.source_exts = detect_source_extension(host_api, things)
466 if not self.core_version and not self.backport_ext and not self.source_exts:
467 print "Warning: Not supportable on host API"
469 def write_header_intro(self, out):
470 out.write("#ifndef MSP_GL_%s_\n"%self.ext_name.upper())
471 out.write("#define MSP_GL_%s_\n"%self.ext_name.upper())
474 #include <msp/gl/extension.h>
475 #include <msp/gl/gl.h>
482 def write_enum_definitions(self, out):
483 enums_by_category = {}
486 supp = e.api_support.get(self.host_api.name)
488 if supp.core_version:
489 cat = "%s_%s"%(self.api_prefix, supp.core_version.as_define())
490 elif supp.extensions:
491 cat = "GL_"+supp.extensions[0].name
492 enums_by_category.setdefault(cat, []).append(e)
494 for cat in sorted(enums_by_category.keys()):
496 out.write("#ifndef %s\n"%cat)
497 for e in enums_by_category[cat]:
498 out.write("#define %s 0x%04X\n"%(e.name, e.value))
500 out.write("#endif\n")
503 def write_function_pointer_declarations(self, out):
505 typedef = self.func_typedefs[f.name]
506 out.write("typedef %s (APIENTRY *%s)(%s);\n"%(f.return_type, typedef, ", ".join(f.params)))
507 out.write("extern %s %s;\n"%(typedef, f.name))
510 def write_header_outro(self, out):
518 def write_source_intro(self, out):
519 out.write("#include \"%s.h\"\n"%self.ext_name.lower())
523 #define GET_PROC_ADDRESS(x) &::x
525 #define GET_PROC_ADDRESS(x) get_proc_address(#x)
529 #define GET_PROC_ADDRESS_1_1(x) &::x
531 #define GET_PROC_ADDRESS_1_1(x) GET_PROC_ADDRESS(x)
540 def write_function_pointer_definitions(self, out):
542 out.write("%s %s = 0;\n"%(self.func_typedefs[f.name], f.name))
544 def write_init_function(self, out):
545 out.write("\nExtension::SupportLevel init_%s()\n{\n"%self.ext_name.lower())
546 if self.core_version:
547 out.write("\tif(is_disabled(\"GL_%s\"))\n\t\treturn Extension::UNSUPPORTED;\n"%self.ext_name)
548 out.write("#if !defined(__APPLE__) || defined(%s_%s)\n"%(self.api_prefix, self.core_version.as_define()))
550 if self.backport_ext:
551 out.write("is_supported(\"GL_%s\") || "%self.backport_ext.name)
552 out.write("is_supported(%r"%self.core_version)
553 if self.deprecated_version:
554 out.write(", %r"%self.deprecated_version)
555 out.write("))\n\t{\n")
557 supp = f.api_support[self.host_api.name]
559 if supp.core_version is not None and supp.core_version<=Version(1, 1):
561 out.write("\t\t%s = reinterpret_cast<%s>(GET_PROC_ADDRESS%s(%s));\n"%(f.name, self.func_typedefs[f.name], gpa_suffix, f.name))
562 out.write("\t\treturn Extension::CORE;\n")
564 out.write("#endif\n")
566 out.write("#if !defined(__APPLE__) || defined(GL_%s)\n"%self.ext_name)
567 out.write("\tif(%s)\n\t{\n"%" && ".join("is_supported(\"GL_%s\")"%s.name for s in self.source_exts))
569 supp = f.api_support[self.host_api.name]
572 for e in self.source_exts:
573 for s in supp.sources:
574 if s.name in e.things:
583 out.write("\t\t%s = reinterpret_cast<%s>(GET_PROC_ADDRESS(%s));\n"%(f.name, self.func_typedefs[f.name], src.name))
584 out.write("\t\treturn Extension::EXTENSION;\n")
586 out.write("#endif\n")
587 out.write("\treturn Extension::UNSUPPORTED;\n")
590 def write_source_outro(self, out):
596 def write_header(self, fn):
598 self.write_header_intro(out)
599 self.write_enum_definitions(out)
600 self.write_function_pointer_declarations(out)
601 out.write("extern Extension %s;\n"%self.ext_name)
602 self.write_header_outro(out)
604 def write_source(self, fn):
606 self.write_source_intro(out)
607 self.write_function_pointer_definitions(out)
608 self.write_init_function(out)
609 out.write("\nExtension %s(\"GL_%s\", init_%s);\n"%(self.ext_name, self.ext_name, self.ext_name.lower()))
610 self.write_source_outro(out)
613 class ExtensionParser:
614 def __init__(self, host_api):
615 self.host_api = host_api
616 self.target_ext = None
617 self.core_version = None
618 self.deprecated_version = None
619 self.backport_ext = None
620 self.ignore_things = []
621 self.optional_things = []
624 for line in open(fn):
626 if line.startswith("#"):
633 api, keyword = keyword.split(":")
635 if api is not None and api!=self.host_api:
638 if keyword=="extension":
639 self.target_ext = parts[1]
640 elif keyword=="core_version":
641 self.core_version = Version(*map(int, parts[1].split('.')))
642 elif keyword=="deprecated":
643 self.deprecated_version = Version(*map(int, parts[1].split('.')))
644 elif keyword=="backport":
645 self.backport_ext = parts[1]
646 elif keyword=="ignore":
647 self.ignore_things.append(parts[1])
648 elif keyword=="optional":
649 self.optional_things.append(parts[1])
651 print "Unknown keyword "+keyword
657 def get_extension(api_map, ext_name):
659 ext_api_name, ext_name = ext_name.split(".")
663 return api_map[ext_api_name].extensions[ext_name]
665 def resolve_things(api, things):
668 ct = filter(None, map(api.core_things.get, t.aliases))
676 def collect_extension_things(host_api, target_ext, ignore):
677 ext_things = [t for n, t in target_ext.things.iteritems() if n not in ignore]
678 return resolve_things(target_ext.api, ext_things)
680 def collect_optional_things(target_ext, names):
683 if t in target_ext.things:
684 things.append(target_ext.things[t])
686 things.append(target_ext.api.core_things[t])
687 return resolve_things(target_ext.api, things)
692 extgen.py [api] <extfile> [<outfile>]
694 Reads gl.xml and generates C++ source files to use an OpenGL extension
695 described in <extfile>. If <outfile> is absent, the extension's lowercased
696 name is used. Anything after the last dot in <outfile> is removed and
697 replaced with cpp and h."""
703 if sys.argv[i].startswith("gl"):
704 host_api_name = sys.argv[i]
707 ext_parser = ExtensionParser(host_api_name)
708 if not ext_parser.parse(sys.argv[i]):
713 out_base = os.path.splitext(sys.argv[i])[0]
715 out_base = ext_parser.target_ext.lower()
717 xml_parser = GlXmlParser()
718 xml_parser.parse_file("gl.xml")
719 xml_parser.parse_file("gl.fixes.xml")
720 xml_parser.parse_file("gl.msp.xml")
721 xml_parser.finalize()
723 host_api = xml_parser.apis[host_api_name]
724 target_ext = get_extension(xml_parser.apis, ext_parser.target_ext)
725 things = collect_extension_things(host_api, target_ext, ext_parser.ignore_things+ext_parser.optional_things)
726 optional_things = collect_optional_things(target_ext, ext_parser.optional_things)
728 generator = SourceGenerator(host_api, target_ext.name, things, optional_things)
729 if ext_parser.core_version:
730 generator.core_version = ext_parser.core_version
731 if ext_parser.deprecated_version:
732 generator.deprecated_version = ext_parser.deprecated_version
733 if ext_parser.backport_ext:
734 if ext_parser.backport_ext=="none":
735 generator.backport_ext = None
737 generator.backport_ext = get_extension(xml_parser.apis, ext_parser.backport_ext)
738 generator.write_header(out_base+".h")
739 generator.write_source(out_base+".cpp")
741 if __name__=="__main__":