| |
| # (C) Copyright IBM Corporation 2004, 2005 |
| # All Rights Reserved. |
| # |
| # Permission is hereby granted, free of charge, to any person obtaining a |
| # copy of this software and associated documentation files (the "Software"), |
| # to deal in the Software without restriction, including without limitation |
| # on the rights to use, copy, modify, merge, publish, distribute, sub |
| # license, and/or sell copies of the Software, and to permit persons to whom |
| # the Software is furnished to do so, subject to the following conditions: |
| # |
| # The above copyright notice and this permission notice (including the next |
| # paragraph) shall be included in all copies or substantial portions of the |
| # Software. |
| # |
| # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
| # FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL |
| # IBM AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
| # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING |
| # FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS |
| # IN THE SOFTWARE. |
| # |
| # Authors: |
| # Ian Romanick <idr@us.ibm.com> |
| |
| from collections import OrderedDict |
| from decimal import Decimal |
| import xml.etree.ElementTree as ET |
| import re, sys |
| import os.path |
| import typeexpr |
| import static_data |
| |
| |
| def parse_GL_API(file_name, factory=None, pointer_size=0): |
| |
| if not factory: |
| factory = gl_item_factory() |
| |
| api = factory.create_api(pointer_size) |
| api.parse_file(file_name) |
| return api |
| |
| |
| def is_attr_true( element, name, default = "false" ): |
| """Read a name value from an element's attributes. |
| |
| The value read from the attribute list must be either 'true' or |
| 'false'. If the value is 'false', zero will be returned. If the |
| value is 'true', non-zero will be returned. An exception will be |
| raised for any other value.""" |
| |
| value = element.get( name, default ) |
| if value == "true": |
| return 1 |
| elif value == "false": |
| return 0 |
| else: |
| raise RuntimeError('Invalid value "%s" for boolean "%s".' % (value, name)) |
| |
| |
| class gl_print_base(object): |
| """Base class of all API pretty-printers. |
| |
| In the model-view-controller pattern, this is the view. Any derived |
| class will want to over-ride the printBody, printRealHader, and |
| printRealFooter methods. Some derived classes may want to over-ride |
| printHeader and printFooter, or even Print (though this is unlikely). |
| """ |
| |
| def __init__(self): |
| # Name of the script that is generating the output file. |
| # Every derived class should set this to the name of its |
| # source file. |
| |
| self.name = "a" |
| |
| |
| # License on the *generated* source file. This may differ |
| # from the license on the script that is generating the file. |
| # Every derived class should set this to some reasonable |
| # value. |
| # |
| # See license.py for an example of a reasonable value. |
| |
| self.license = "The license for this file is unspecified." |
| |
| |
| # The header_tag is the name of the C preprocessor define |
| # used to prevent multiple inclusion. Typically only |
| # generated C header files need this to be set. Setting it |
| # causes code to be generated automatically in printHeader |
| # and printFooter. |
| |
| self.header_tag = None |
| |
| |
| # List of file-private defines that must be undefined at the |
| # end of the file. This can be used in header files to define |
| # names for use in the file, then undefine them at the end of |
| # the header file. |
| |
| self.undef_list = [] |
| return |
| |
| |
| def Print(self, api): |
| self.printHeader() |
| self.printBody(api) |
| self.printFooter() |
| return |
| |
| |
| def printHeader(self): |
| """Print the header associated with all files and call the printRealHeader method.""" |
| |
| print('/* DO NOT EDIT - This file generated automatically by %s script */' \ |
| % (self.name)) |
| print('') |
| print('/*') |
| print((' * ' + self.license.replace('\n', '\n * ')).replace(' \n', '\n')) |
| print(' */') |
| print('') |
| if self.header_tag: |
| print('#if !defined( %s )' % (self.header_tag)) |
| print('# define %s' % (self.header_tag)) |
| print('') |
| self.printRealHeader(); |
| return |
| |
| |
| def printFooter(self): |
| """Print the header associated with all files and call the printRealFooter method.""" |
| |
| self.printRealFooter() |
| |
| if self.undef_list: |
| print('') |
| for u in self.undef_list: |
| print("# undef %s" % (u)) |
| |
| if self.header_tag: |
| print('') |
| print('#endif /* !defined( %s ) */' % (self.header_tag)) |
| |
| |
| def printRealHeader(self): |
| """Print the "real" header for the created file. |
| |
| In the base class, this function is empty. All derived |
| classes should over-ride this function.""" |
| return |
| |
| |
| def printRealFooter(self): |
| """Print the "real" footer for the created file. |
| |
| In the base class, this function is empty. All derived |
| classes should over-ride this function.""" |
| return |
| |
| |
| def printNoinline(self): |
| """Conditionally define `NOINLINE' function attribute. |
| |
| Conditionally defines a preprocessor macro `NOINLINE' that |
| wraps GCC's `noinline' function attribute. The conditional |
| code can be easilly adapted to other compilers that support a |
| similar feature. |
| |
| The name is also added to the file's undef_list. |
| """ |
| |
| self.undef_list.append("NOINLINE") |
| print("""# if defined(__GNUC__) |
| # define NOINLINE __attribute__((noinline)) |
| # else |
| # define NOINLINE |
| # endif""") |
| return |
| |
| |
| def real_function_name(element): |
| name = element.get( "name" ) |
| alias = element.get( "alias" ) |
| |
| if alias: |
| return alias |
| else: |
| return name |
| |
| |
| def real_category_name(c): |
| if re.compile("[1-9][0-9]*[.][0-9]+").match(c): |
| return "GL_VERSION_" + c.replace(".", "_") |
| else: |
| return c |
| |
| |
| def classify_category(name, number): |
| """Based on the category name and number, select a numerical class for it. |
| |
| Categories are divided into four classes numbered 0 through 3. The |
| classes are: |
| |
| 0. Core GL versions, sorted by version number. |
| 1. ARB extensions, sorted by extension number. |
| 2. Non-ARB extensions, sorted by extension number. |
| 3. Un-numbered extensions, sorted by extension name. |
| """ |
| |
| try: |
| core_version = float(name) |
| except Exception: |
| core_version = 0.0 |
| |
| if core_version > 0.0: |
| cat_type = 0 |
| key = name |
| elif name.startswith("GL_ARB_") or name.startswith("GLX_ARB_") or name.startswith("WGL_ARB_"): |
| cat_type = 1 |
| key = int(number) |
| else: |
| if number != None: |
| cat_type = 2 |
| key = int(number) |
| else: |
| cat_type = 3 |
| key = name |
| |
| |
| return [cat_type, key] |
| |
| |
| def create_parameter_string(parameters, include_names): |
| """Create a parameter string from a list of gl_parameters.""" |
| |
| list = [] |
| for p in parameters: |
| if p.is_padding: |
| continue |
| |
| if include_names: |
| list.append( p.string() ) |
| else: |
| list.append( p.type_string() ) |
| |
| if len(list) == 0: list = ["void"] |
| |
| return ", ".join(list) |
| |
| |
| class gl_item(object): |
| def __init__(self, element, context, category): |
| self.context = context |
| self.name = element.get( "name" ) |
| self.category = real_category_name( category ) |
| |
| return |
| |
| |
| class gl_type( gl_item ): |
| def __init__(self, element, context, category): |
| gl_item.__init__(self, element, context, category) |
| self.size = int( element.get( "size" ), 0 ) |
| |
| te = typeexpr.type_expression( None ) |
| tn = typeexpr.type_node() |
| tn.size = int( element.get( "size" ), 0 ) |
| tn.integer = not is_attr_true( element, "float" ) |
| tn.unsigned = is_attr_true( element, "unsigned" ) |
| tn.pointer = is_attr_true( element, "pointer" ) |
| tn.name = "GL" + self.name |
| te.set_base_type_node( tn ) |
| |
| self.type_expr = te |
| return |
| |
| |
| class gl_enum( gl_item ): |
| def __init__(self, element, context, category): |
| gl_item.__init__(self, element, context, category) |
| self.value = int( element.get( "value" ), 0 ) |
| |
| temp = element.get( "count" ) |
| if not temp or temp == "?": |
| self.default_count = -1 |
| else: |
| try: |
| c = int(temp) |
| except Exception: |
| raise RuntimeError('Invalid count value "%s" for enum "%s" in function "%s" when an integer was expected.' % (temp, self.name, n)) |
| |
| self.default_count = c |
| |
| return |
| |
| |
| def priority(self): |
| """Calculate a 'priority' for this enum name. |
| |
| When an enum is looked up by number, there may be many |
| possible names, but only one is the 'preferred' name. The |
| priority is used to select which name is the 'best'. |
| |
| Highest precedence is given to core GL name. ARB extension |
| names have the next highest, followed by EXT extension names. |
| Vendor extension names are the lowest. |
| """ |
| |
| if self.name.endswith( "_BIT" ): |
| bias = 1 |
| else: |
| bias = 0 |
| |
| if self.category.startswith( "GL_VERSION_" ): |
| priority = 0 |
| elif self.category.startswith( "GL_ARB_" ): |
| priority = 2 |
| elif self.category.startswith( "GL_EXT_" ): |
| priority = 4 |
| else: |
| priority = 6 |
| |
| return priority + bias |
| |
| |
| |
| class gl_parameter(object): |
| def __init__(self, element, context): |
| self.name = element.get( "name" ) |
| |
| ts = element.get( "type" ) |
| self.type_expr = typeexpr.type_expression( ts, context ) |
| |
| temp = element.get( "variable_param" ) |
| if temp: |
| self.count_parameter_list = temp.split( ' ' ) |
| else: |
| self.count_parameter_list = [] |
| |
| # The count tag can be either a numeric string or the name of |
| # a variable. If it is the name of a variable, the int(c) |
| # statement will throw an exception, and the except block will |
| # take over. |
| |
| c = element.get( "count" ) |
| try: |
| count = int(c) |
| self.count = count |
| self.counter = None |
| except Exception: |
| count = 1 |
| self.count = 0 |
| self.counter = c |
| |
| self.marshal_count = element.get("marshal_count") |
| self.marshal_large_count = element.get("marshal_large_count") |
| self.count_scale = int(element.get( "count_scale", "1" )) |
| |
| elements = (count * self.count_scale) |
| if elements == 1: |
| elements = 0 |
| |
| #if ts == "GLdouble": |
| # print '/* stack size -> %s = %u (before)*/' % (self.name, self.type_expr.get_stack_size()) |
| # print '/* # elements = %u */' % (elements) |
| self.type_expr.set_elements( elements ) |
| #if ts == "GLdouble": |
| # print '/* stack size -> %s = %u (after) */' % (self.name, self.type_expr.get_stack_size()) |
| |
| self.is_client_only = is_attr_true( element, 'client_only' ) |
| self.is_output = is_attr_true( element, 'output' ) |
| |
| |
| # Pixel data has special parameters. |
| |
| self.width = element.get('img_width') |
| self.height = element.get('img_height') |
| self.depth = element.get('img_depth') |
| self.extent = element.get('img_extent') |
| |
| self.img_xoff = element.get('img_xoff') |
| self.img_yoff = element.get('img_yoff') |
| self.img_zoff = element.get('img_zoff') |
| self.img_woff = element.get('img_woff') |
| |
| self.img_format = element.get('img_format') |
| self.img_type = element.get('img_type') |
| self.img_target = element.get('img_target') |
| |
| self.img_pad_dimensions = is_attr_true( element, 'img_pad_dimensions' ) |
| self.img_null_flag = is_attr_true( element, 'img_null_flag' ) |
| self.img_send_null = is_attr_true( element, 'img_send_null' ) |
| |
| self.is_padding = is_attr_true( element, 'padding' ) |
| return |
| |
| |
| def compatible(self, other): |
| return 1 |
| |
| |
| def is_array(self): |
| return self.is_pointer() |
| |
| |
| def is_pointer(self): |
| return self.type_expr.is_pointer() |
| |
| |
| def is_image(self): |
| if self.width: |
| return 1 |
| else: |
| return 0 |
| |
| |
| def is_variable_length(self): |
| return (len(self.count_parameter_list) or self.counter or |
| self.marshal_count or self.marshal_large_count) |
| |
| |
| def is_64_bit(self): |
| count = self.type_expr.get_element_count() |
| if count: |
| if (self.size() / count) == 8: |
| return 1 |
| else: |
| if self.size() == 8: |
| return 1 |
| |
| return 0 |
| |
| |
| def string(self): |
| if self.type_expr.original_string[-1] == '*': |
| return self.type_expr.original_string + self.name |
| else: |
| return self.type_expr.original_string + " " + self.name |
| |
| |
| def type_string(self): |
| return self.type_expr.original_string |
| |
| |
| def get_base_type_string(self): |
| return self.type_expr.get_base_name() |
| |
| |
| def get_dimensions(self): |
| if not self.width: |
| return [ 0, "0", "0", "0", "0" ] |
| |
| dim = 1 |
| w = self.width |
| h = "1" |
| d = "1" |
| e = "1" |
| |
| if self.height: |
| dim = 2 |
| h = self.height |
| |
| if self.depth: |
| dim = 3 |
| d = self.depth |
| |
| if self.extent: |
| dim = 4 |
| e = self.extent |
| |
| return [ dim, w, h, d, e ] |
| |
| |
| def get_stack_size(self): |
| return self.type_expr.get_stack_size() |
| |
| |
| def size(self): |
| if self.is_image(): |
| return 0 |
| else: |
| return self.type_expr.get_element_size() |
| |
| |
| def get_element_count(self): |
| c = self.type_expr.get_element_count() |
| if c == 0: |
| return 1 |
| |
| return c |
| |
| |
| def size_string(self, use_parens = 1, marshal = 0): |
| base_size_str = "" |
| |
| count = self.get_element_count() |
| if count: |
| base_size_str = "%d * " % count |
| |
| base_size_str += "sizeof(%s)" % ( self.get_base_type_string() ) |
| |
| if (self.counter or self.count_parameter_list or |
| (marshal and (self.marshal_count or self.marshal_large_count))): |
| list = [ "compsize" ] |
| |
| if marshal and self.marshal_count: |
| list = [ self.marshal_count ] |
| elif marshal and self.marshal_large_count: |
| list = [ self.marshal_large_count ] |
| elif self.counter and self.count_parameter_list: |
| list.append( self.counter ) |
| elif self.counter: |
| list = [ self.counter ] |
| |
| if self.size() > 1: |
| list.append( base_size_str ) |
| |
| # Don't use safe_mul if marshal_count is used, which indicates |
| # a small size. |
| if len(list) > 1 and use_parens and not self.marshal_count: |
| return "safe_mul(%s)" % ", ".join(list) |
| else: |
| return " * ".join(list) |
| |
| elif self.is_image(): |
| return "compsize" |
| else: |
| return base_size_str |
| |
| |
| def format_string(self): |
| if self.type_expr.original_string == "GLenum": |
| return "0x%x" |
| else: |
| return self.type_expr.format_string() |
| |
| |
| class gl_function( gl_item ): |
| def __init__(self, element, context): |
| self.context = context |
| self.name = None |
| |
| self.entry_points = [] |
| self.return_type = "void" |
| self.parameters = [] |
| self.offset = -1 |
| self.initialized = 0 |
| self.images = [] |
| self.exec_flavor = 'mesa' |
| self.has_hw_select_variant = False |
| self.desktop = True |
| self.deprecated = None |
| self.has_no_error_variant = False |
| |
| # self.api_map[api] is a decimal value indicating the earliest |
| # version of the given API in which ANY alias for the function |
| # exists. The map only lists APIs which contain the function |
| # in at least one version. For example, for the ClipPlanex |
| # function, self.api_map == { 'es1': |
| # Decimal('1.1') }. |
| self.api_map = {} |
| |
| # Track the parameter string (for the function prototype) |
| # for each entry-point. This is done because some functions |
| # change their prototype slightly when promoted from extension |
| # to ARB extension to core. glTexImage3DEXT and glTexImage3D |
| # are good examples of this. Scripts that need to generate |
| # code for these differing aliases need to real prototype |
| # for each entry-point. Otherwise, they may generate code |
| # that won't compile. |
| |
| self.entry_point_parameters = {} |
| |
| self.process_element( element ) |
| |
| return |
| |
| |
| def process_element(self, element): |
| name = element.get( "name" ) |
| alias = element.get( "alias" ) |
| |
| # marshal isn't allowed with alias |
| assert not alias or not element.get('marshal') |
| assert not alias or not element.get('marshal_count') |
| assert not alias or not element.get('marshal_large_count') |
| assert not alias or not element.get('marshal_sync') |
| assert not alias or not element.get('marshal_call_before') |
| assert not alias or not element.get('marshal_call_after') |
| assert not alias or not element.get('deprecated') |
| |
| self.entry_points.append( name ) |
| |
| for api in ('es1', 'es2'): |
| version_str = element.get(api, 'none') |
| assert version_str is not None |
| if version_str != 'none': |
| version_decimal = Decimal(version_str) |
| if api not in self.api_map or \ |
| version_decimal < self.api_map[api]: |
| self.api_map[api] = version_decimal |
| |
| exec_flavor = element.get('exec') |
| if exec_flavor: |
| self.exec_flavor = exec_flavor |
| |
| deprecated = element.get('deprecated', 'none') |
| if deprecated != 'none': |
| self.deprecated = Decimal(deprecated) |
| |
| if not is_attr_true(element, 'desktop', 'true'): |
| self.desktop = False |
| |
| if self.has_no_error_variant or is_attr_true(element, 'no_error'): |
| self.has_no_error_variant = True |
| else: |
| self.has_no_error_variant = False |
| |
| if alias: |
| true_name = alias |
| if name in static_data.offsets: |
| raise RuntimeError("Aliased entry-point %s shouldn't be in static_data.py." % (name)) |
| else: |
| true_name = name |
| |
| self.has_hw_select_variant = exec_flavor == 'beginend' and name[0:6] == 'Vertex' |
| |
| # Only try to set the offset when a non-alias entry-point |
| # is being processed. |
| |
| if name in static_data.offsets: |
| self.offset = static_data.offsets[name] |
| elif self.exec_flavor != "skip": |
| raise RuntimeError("Entry-point %s is missing in static_data.py. Add one into all_functions." % (name)) |
| |
| if not self.name: |
| self.name = true_name |
| elif self.name != true_name: |
| raise RuntimeError("Function true name redefined. Was %s, now %s." % (self.name, true_name)) |
| |
| |
| # There are two possible cases. The first time an entry-point |
| # with data is seen, self.initialized will be 0. On that |
| # pass, we just fill in the data. The next time an |
| # entry-point with data is seen, self.initialized will be 1. |
| # On that pass we have to make that the new values match the |
| # valuse from the previous entry-point. |
| |
| parameters = [] |
| return_type = "void" |
| for child in element: |
| if child.tag == "return": |
| return_type = child.get( "type", "void" ) |
| elif child.tag == "param": |
| param = self.context.factory.create_parameter(child, self.context) |
| parameters.append( param ) |
| |
| |
| if self.initialized: |
| if self.return_type != return_type: |
| raise RuntimeError( "Return type changed in %s. Was %s, now %s." % (name, self.return_type, return_type)) |
| |
| if len(parameters) != len(self.parameters): |
| raise RuntimeError( "Parameter count mismatch in %s. Was %d, now %d." % (name, len(self.parameters), len(parameters))) |
| |
| for j in range(0, len(parameters)): |
| p1 = parameters[j] |
| p2 = self.parameters[j] |
| if not p1.compatible( p2 ): |
| raise RuntimeError( 'Parameter type mismatch in %s. "%s" was "%s", now "%s".' % (name, p2.name, p2.type_expr.original_string, p1.type_expr.original_string)) |
| |
| |
| if true_name == name or not self.initialized: |
| self.return_type = return_type |
| self.parameters = parameters |
| |
| for param in self.parameters: |
| if param.is_image(): |
| self.images.append( param ) |
| |
| if list(element): |
| self.initialized = 1 |
| self.entry_point_parameters[name] = parameters |
| else: |
| self.entry_point_parameters[name] = [] |
| |
| return |
| |
| def get_images(self): |
| """Return potentially empty list of input images.""" |
| return self.images |
| |
| |
| def parameterIterator(self, name = None): |
| if name is not None: |
| return iter(self.entry_point_parameters[name]); |
| else: |
| return iter(self.parameters); |
| |
| |
| def get_parameter_string(self, entrypoint = None): |
| if entrypoint: |
| params = self.entry_point_parameters[ entrypoint ] |
| else: |
| params = self.parameters |
| |
| return create_parameter_string( params, 1 ) |
| |
| def get_called_parameter_string(self): |
| p_string = "" |
| comma = "" |
| |
| for p in self.parameterIterator(): |
| if p.is_padding: |
| continue |
| p_string = p_string + comma + p.name |
| comma = ", " |
| |
| return p_string |
| |
| def static_name(self, name): |
| if name in static_data.libgl_public_functions: |
| return name |
| else: |
| return "_dispatch_stub_%u" % (self.offset) |
| |
| class gl_item_factory(object): |
| """Factory to create objects derived from gl_item.""" |
| |
| def create_function(self, element, context): |
| return gl_function(element, context) |
| |
| def create_type(self, element, context, category): |
| return gl_type(element, context, category) |
| |
| def create_enum(self, element, context, category): |
| return gl_enum(element, context, category) |
| |
| def create_parameter(self, element, context): |
| return gl_parameter(element, context) |
| |
| def create_api(self, pointer_size): |
| return gl_api(self, pointer_size) |
| |
| |
| class gl_api(object): |
| def __init__(self, factory, pointer_size): |
| self.functions_by_name = OrderedDict() |
| self.enums_by_name = {} |
| self.types_by_name = {} |
| |
| self.category_dict = {} |
| self.categories = [{}, {}, {}, {}] |
| |
| self.factory = factory |
| |
| self.next_offset = 0 |
| self.pointer_size = pointer_size |
| |
| typeexpr.create_initial_types() |
| return |
| |
| def parse_file(self, file_name): |
| doc = ET.parse( file_name ) |
| self.process_element(file_name, doc) |
| |
| |
| def process_element(self, file_name, doc): |
| element = doc.getroot() |
| if element.tag == "OpenGLAPI": |
| self.process_OpenGLAPI(file_name, element) |
| return |
| |
| |
| def process_OpenGLAPI(self, file_name, element): |
| for child in element: |
| if child.tag == "category": |
| self.process_category( child ) |
| elif child.tag == "OpenGLAPI": |
| self.process_OpenGLAPI( file_name, child ) |
| elif child.tag == '{http://www.w3.org/2001/XInclude}include': |
| href = child.get('href') |
| href = os.path.join(os.path.dirname(file_name), href) |
| self.parse_file(href) |
| |
| return |
| |
| |
| def process_category(self, cat): |
| cat_name = cat.get( "name" ) |
| cat_number = cat.get( "number" ) |
| |
| [cat_type, key] = classify_category(cat_name, cat_number) |
| self.categories[cat_type][key] = [cat_name, cat_number] |
| |
| for child in cat: |
| if child.tag == "function": |
| func_name = real_function_name( child ) |
| |
| temp_name = child.get( "name" ) |
| self.category_dict[ temp_name ] = [cat_name, cat_number] |
| |
| if func_name in self.functions_by_name: |
| func = self.functions_by_name[ func_name ] |
| func.process_element( child ) |
| else: |
| func = self.factory.create_function( child, self ) |
| self.functions_by_name[ func_name ] = func |
| |
| if func.offset >= self.next_offset: |
| self.next_offset = func.offset + 1 |
| |
| |
| elif child.tag == "enum": |
| enum = self.factory.create_enum( child, self, cat_name ) |
| self.enums_by_name[ enum.name ] = enum |
| elif child.tag == "type": |
| t = self.factory.create_type( child, self, cat_name ) |
| self.types_by_name[ "GL" + t.name ] = t |
| |
| return |
| |
| |
| def functionIterateByCategory(self, cat = None): |
| """Iterate over functions by category. |
| |
| If cat is None, all known functions are iterated in category |
| order. See classify_category for details of the ordering. |
| Within a category, functions are sorted by name. If cat is |
| not None, then only functions in that category are iterated. |
| """ |
| lists = [{}, {}, {}, {}] |
| |
| for func in self.functionIterateAll(): |
| [cat_name, cat_number] = self.category_dict[func.name] |
| |
| if (cat == None) or (cat == cat_name): |
| [func_cat_type, key] = classify_category(cat_name, cat_number) |
| |
| if key not in lists[func_cat_type]: |
| lists[func_cat_type][key] = {} |
| |
| lists[func_cat_type][key][func.name] = func |
| |
| |
| functions = [] |
| for func_cat_type in range(0,4): |
| keys = sorted(lists[func_cat_type].keys()) |
| |
| for key in keys: |
| names = sorted(lists[func_cat_type][key].keys()) |
| |
| for name in names: |
| functions.append(lists[func_cat_type][key][name]) |
| |
| return iter(functions) |
| |
| |
| def functionIterateByOffset(self): |
| max_offset = -1 |
| for func in self.functions_by_name.values(): |
| if func.offset > max_offset: |
| max_offset = func.offset |
| |
| |
| temp = [None for i in range(0, max_offset + 1)] |
| for func in self.functions_by_name.values(): |
| if func.offset != -1: |
| temp[ func.offset ] = func |
| |
| |
| list = [] |
| for i in range(0, max_offset + 1): |
| if temp[i]: |
| list.append(temp[i]) |
| |
| return iter(list); |
| |
| |
| def functionIterateAll(self): |
| return self.functions_by_name.values() |
| |
| |
| def categoryIterate(self): |
| """Iterate over categories. |
| |
| Iterate over all known categories in the order specified by |
| classify_category. Each iterated value is a tuple of the |
| name and number (which may be None) of the category. |
| """ |
| |
| list = [] |
| for cat_type in range(0,4): |
| keys = sorted(self.categories[cat_type].keys()) |
| |
| for key in keys: |
| list.append(self.categories[cat_type][key]) |
| |
| return iter(list) |
| |
| |
| def find_type( self, type_name ): |
| if type_name in self.types_by_name: |
| return self.types_by_name[ type_name ].type_expr |
| else: |
| print("Unable to find base type matching \"%s\"." % (type_name)) |
| return None |
