git.roojs.org Git - gnome.gobject-introspection/blob - giscanner/glibtransformer.py

   1 # -*- Mode: Python -*-
   2 # GObject-Introspection - a framework for introspecting GObject libraries
   3 # Copyright (C) 2008  Johan Dahlin
   4 #
   5 # This program is free software; you can redistribute it and/or
   6 # modify it under the terms of the GNU General Public License
   7 # as published by the Free Software Foundation; either version 2
   8 # of the License, or (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program; if not, write to the Free Software
  17 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  18 # 02110-1301, USA.
  19 #
  20
  21 import ctypes
  22 from ctypes.util import find_library
  23
  24 from . import cgobject
  25 from .ast import (Callback, Enum, Function, Member, Namespace, Parameter,
  26                   Sequence, Property, Return, Struct, Type, Alias,
  27                   Union, type_name_from_ctype)
  28 from .transformer import Names
  29 from .glibast import (GLibBoxed, GLibEnum, GLibEnumMember, GLibFlags,
  30                       GLibInterface, GLibObject, GLibSignal, type_names)
  31 from .utils import extract_libtool, to_underscores
  32
  33
  34 class Unresolved(object):
  35
  36     def __init__(self, target):
  37         self.target = target
  38
  39
  40 class UnknownTypeError(Exception):
  41     pass
  42
  43
  44 class GLibTransformer(object):
  45
  46     def __init__(self, transformer, noclosure=False):
  47         self._transformer = transformer
  48         self._namespace_name = None
  49         self._names = Names()
  50         self._uscore_type_names = {}
  51         self._libraries = []
  52         self._failed_types = {}
  53         self._noclosure = noclosure
  54         self._validating = False
  55
  56     # Public API
  57
  58     def add_library(self, libname):
  59         # For testing mainly.
  60         if libname.endswith('.la'):
  61             libname = extract_libtool(libname)
  62         else:
  63             libname = find_library(libname)
  64         if not libname:
  65             raise ValueError("Failed to find library: %r" % (libname, ))
  66         self._libraries.append(ctypes.cdll.LoadLibrary(libname))
  67
  68     def parse(self):
  69         namespace = self._transformer.parse()
  70         self._namespace_name = namespace.name
  71
  72         # First pass, parsing
  73         for node in namespace.nodes:
  74             self._parse_node(node)
  75
  76         # Introspection is done from within parsing
  77
  78         # Second pass, delete class structures, resolve
  79         # all types we now know about
  80         nodes = list(self._names.names.itervalues())
  81         for (ns, node) in nodes:
  82             self._resolve_node(node)
  83             # associate GtkButtonClass with GtkButton
  84             if isinstance(node, Struct):
  85                 self._pair_class_struct(node)
  86         for (ns, alias) in self._names.aliases.itervalues():
  87             self._resolve_alias(alias)
  88
  89         # Third pass; ensure all types are known
  90         if not self._noclosure:
  91             self._validate(nodes)
  92
  93         # Create a new namespace with what we found
  94         namespace = Namespace(namespace.name)
  95         namespace.nodes = map(lambda x: x[1], self._names.aliases.itervalues())
  96         for (ns, x) in self._names.names.itervalues():
  97             namespace.nodes.append(x)
  98         return namespace
  99
 100     # Private
 101
 102     def _add_attribute(self, node, replace=False):
 103         node_name = node.name
 104         if (not replace) and node_name in self._names.names:
 105             return
 106         self._names.names[node_name] = (None, node)
 107
 108     def _remove_attribute(self, name):
 109         del self._names.names[name]
 110
 111     def _get_attribute(self, name):
 112         node = self._names.names.get(name)
 113         if node:
 114             return node[1]
 115         return None
 116
 117     def _register_internal_type(self, type_name, node):
 118         self._names.type_names[type_name] = (None, node)
 119         self._uscore_type_names[to_underscores(type_name).lower()] = node
 120         # Besides the straight underscore conversion, we also try
 121         # removing the underscores from the namespace as a possible C
 122         # mapping; e.g. it's webkit_web_view, not web_kit_web_view
 123         suffix = self._transformer.strip_namespace_object(type_name)
 124         prefix = type_name[:-len(suffix)]
 125         no_uscore_prefixed = (prefix + '_' + to_underscores(suffix)).lower()
 126         self._uscore_type_names[no_uscore_prefixed] = node
 127
 128     # Helper functions
 129
 130     def _create_type(self, type_id):
 131         ctype = cgobject.type_name(type_id)
 132         type_name = type_name_from_ctype(ctype)
 133         type_name = type_name.replace('*', '')
 134         type_name = self._resolve_type_name(type_name)
 135         return Type(type_name, ctype)
 136
 137     def _resolve_gtypename(self, gtype_name):
 138         try:
 139             return self._transformer.gtypename_to_giname(gtype_name,
 140                                                          self._names)
 141         except KeyError, e:
 142             return Unresolved(gtype_name)
 143
 144     def _create_gobject(self, node):
 145         type_name = 'G' + node.name
 146         if type_name == 'GObject':
 147             parent_gitype = None
 148             symbol = 'intern'
 149         else:
 150             type_id = cgobject.type_from_name(type_name)
 151             parent_type_name = cgobject.type_name(
 152                 cgobject.type_parent(type_id))
 153             parent_gitype = self._resolve_gtypename(parent_type_name)
 154             symbol = to_underscores(type_name).lower() + '_get_type'
 155         node = GLibObject(node.name, parent_gitype, type_name, symbol)
 156         type_id = cgobject.TYPE_OBJECT
 157         self._introspect_properties(node, type_id)
 158         self._introspect_signals(node, type_id)
 159         self._add_attribute(node)
 160         self._register_internal_type(type_name, node)
 161
 162     # Parser
 163
 164     def _parse_node(self, node):
 165         if isinstance(node, Enum):
 166             self._parse_enum(node)
 167         elif isinstance(node, Function):
 168             self._parse_function(node)
 169         elif isinstance(node, Struct):
 170             self._parse_struct(node)
 171         elif isinstance(node, Callback):
 172             self._parse_callback(node)
 173         elif isinstance(node, Alias):
 174             self._parse_alias(node)
 175         elif isinstance(node, Member):
 176             # FIXME: atk_misc_instance singletons
 177             pass
 178         elif isinstance(node, Union):
 179             self._parse_union(node)
 180         else:
 181             print 'GLIB Transformer: Unhandled node:', node
 182
 183     def _parse_alias(self, alias):
 184         self._names.aliases[alias.name] = (None, alias)
 185
 186     def _parse_enum(self, enum):
 187         self._add_attribute(enum)
 188
 189     def _parse_function(self, func):
 190         if self._parse_get_type_function(func):
 191             return
 192
 193         self._add_attribute(func)
 194
 195     def _parse_get_type_function(self, func):
 196         symbol = func.symbol
 197         if not symbol.endswith('_get_type'):
 198             return False
 199         # GType *_get_type(void)
 200         # This is a bit fishy, why do we need all these aliases?
 201         if func.retval.type.name not in ['Type',
 202                                          'GType',
 203                                          'Object.Type',
 204                                          'GObject.Type',
 205                                          'GObject.GType']:
 206             print ("Warning: *_get_type function returns '%r'"
 207                    ", not GObject.Type") % (func.retval.type.name, )
 208             return False
 209         if func.parameters:
 210             return False
 211
 212         if not self._libraries:
 213             print "Warning: No libraries loaded, cannot call %s" % (symbol, )
 214             return False
 215
 216         for library in self._libraries:
 217             try:
 218                 func = getattr(library, symbol)
 219                 break
 220             except AttributeError:
 221                 continue
 222         else:
 223             print 'Warning: could not find symbol: %s' % symbol
 224             name = symbol.replace('_get_type', '')
 225             self._failed_types[name] = True
 226             return False
 227
 228         func.restype = cgobject.GType
 229         func.argtypes = []
 230         type_id = func()
 231         self._introspect_type(type_id, symbol)
 232         return True
 233
 234     def _name_is_internal_gtype(self, giname):
 235         try:
 236             node = self._get_attribute(giname)
 237             return isinstance(node, (GLibObject, GLibInterface, GLibBoxed,
 238                                      GLibEnum, GLibFlags))
 239         except KeyError, e:
 240             return False
 241
 242     def _parse_method(self, func):
 243         if not func.parameters:
 244             return False
 245         return self._parse_method_common(func, True)
 246
 247     def _parse_constructor(self, func):
 248         return self._parse_method_common(func, False)
 249
 250     def _parse_method_common(self, func, is_method):
 251         if not is_method:
 252             target_arg = func.retval
 253         else:
 254             target_arg = func.parameters[0]
 255         target_arg.type = self._resolve_param_type(target_arg.type)
 256
 257         if is_method:
 258             # Methods require their first arg to be a known class
 259             # Look at the original C type (before namespace stripping), without
 260             # pointers: GtkButton -> gtk_button_, so we can figure out the
 261             # method name
 262             argtype = target_arg.type.ctype.replace('*', '')
 263             name = self._transformer.strip_namespace_object(argtype)
 264             name_uscore = to_underscores(name).lower()
 265             name_offset = func.symbol.find(name_uscore)
 266             if name_offset < 0:
 267                 return None
 268             prefix = func.symbol[:name_offset+len(name_uscore)]
 269         else:
 270             # Constructors must have _new
 271             # Take everything before that as class name
 272             new_idx = func.symbol.find('_new')
 273             if new_idx < 0:
 274                 return None
 275             prefix = func.symbol[:new_idx]
 276
 277         klass = None
 278         for key in self._uscore_type_names:
 279             klass = None
 280             if key.startswith(prefix):
 281                 klass = self._uscore_type_names.get(key)
 282                 if (klass is not None and
 283                     isinstance(klass, (GLibObject, GLibBoxed,
 284                                        GLibInterface))):
 285                     break
 286
 287         if klass is None:
 288             return
 289
 290         if not is_method:
 291             # Interfaces can't have constructors, punt to global scope
 292             if isinstance(klass, GLibInterface):
 293                 return None
 294             # TODO - check that the return type is a subclass of the
 295             # class from the prefix
 296
 297         self._remove_attribute(func.name)
 298         # Strip namespace and object prefix: gtk_window_new -> new
 299         func.name = func.symbol[len(prefix)+1:]
 300         if is_method:
 301             klass.methods.append(func)
 302         else:
 303             klass.constructors.append(func)
 304         return func
 305
 306     def _parse_struct(self, struct):
 307         # This is a hack, but GObject is a rather fundamental piece so.
 308         internal_names = ["Object", 'InitiallyUnowned']
 309         g_internal_names = ["G" + x for x in internal_names]
 310         if (self._namespace_name == 'GObject' and
 311             struct.name in internal_names):
 312             self._create_gobject(struct)
 313             return
 314         elif struct.name in g_internal_names:
 315             # Avoid duplicates
 316             return
 317         node = self._names.names.get(struct.name)
 318         if node is None:
 319             self._add_attribute(struct, replace=True)
 320             return
 321         (ns, node) = node
 322         node.fields = struct.fields[:]
 323
 324     def _parse_union(self, union):
 325         node = self._names.names.get(union.name)
 326         if node is None:
 327             self._add_attribute(union, replace=True)
 328             return
 329         (ns, node) = node
 330         node.fields = union.fields[:]
 331
 332     def _parse_callback(self, callback):
 333         self._add_attribute(callback)
 334
 335     def _strip_class_suffix(self, name):
 336         if (name.endswith('Class') or
 337             name.endswith('Iface')):
 338             return name[:-5]
 339         elif name.endswith('Interface'):
 340             return name[:-9]
 341         else:
 342             return name
 343
 344     def _arg_is_failed(self, param):
 345         ctype = self._transformer.ctype_of(param).replace('*', '')
 346         uscored = to_underscores(self._strip_class_suffix(ctype)).lower()
 347         if uscored in self._failed_types:
 348             print "Warning: failed type: %r" % (param, )
 349             return True
 350         return False
 351
 352     def _pair_class_struct(self, maybe_class):
 353         name = self._strip_class_suffix(maybe_class.name)
 354         if name == maybe_class.name:
 355             return
 356
 357         if self._arg_is_failed(maybe_class):
 358             print "WARNING: deleting no-type %r" % (maybe_class.name, )
 359             del self._names.names[maybe_class.name]
 360             return
 361
 362         name = self._resolve_type_name(name)
 363         resolved = self._transformer.strip_namespace_object(name)
 364         pair_class = self._get_attribute(resolved)
 365         if pair_class and isinstance(pair_class,
 366                                      (GLibObject, GLibBoxed, GLibInterface)):
 367             for field in maybe_class.fields[1:]:
 368                 pair_class.fields.append(field)
 369             return
 370         name = self._transformer.strip_namespace_object(maybe_class.name)
 371         pair_class = self._get_attribute(name)
 372         if pair_class and isinstance(pair_class,
 373                                      (GLibObject, GLibBoxed, GLibInterface)):
 374
 375             del self._names.names[maybe_class.name]
 376
 377     # Introspection
 378
 379     def _introspect_type(self, type_id, symbol):
 380         fundamental_type_id = cgobject.type_fundamental(type_id)
 381         if (fundamental_type_id == cgobject.TYPE_ENUM or
 382             fundamental_type_id == cgobject.TYPE_FLAGS):
 383             self._introspect_enum(fundamental_type_id, type_id, symbol)
 384         elif fundamental_type_id == cgobject.TYPE_OBJECT:
 385             self._introspect_object(type_id, symbol)
 386         elif fundamental_type_id == cgobject.TYPE_INTERFACE:
 387             self._introspect_interface(type_id, symbol)
 388         elif fundamental_type_id == cgobject.TYPE_BOXED:
 389             self._introspect_boxed(type_id, symbol)
 390         elif fundamental_type_id == cgobject.TYPE_BOXED:
 391             self._introspect_boxed(type_id, symbol)
 392         elif fundamental_type_id == cgobject.TYPE_POINTER:
 393             # FIXME: Should we do something about these?
 394             #        GHashTable, GValue and a few other fundamentals are
 395             #        covered here
 396             return
 397         else:
 398             print 'unhandled GType: %s(%d)' % (cgobject.type_name(type_id),
 399                                                type_id)
 400
 401     def _introspect_enum(self, ftype_id, type_id, symbol):
 402         type_class = cgobject.type_class_ref(type_id)
 403         if type_class is None:
 404             return
 405
 406         members = []
 407         for enum_value in type_class.get_values():
 408             members.append(GLibEnumMember(enum_value.value_nick,
 409                                           enum_value.value,
 410                                           enum_value.value_name,
 411                                           enum_value.value_nick))
 412
 413         klass = (GLibFlags if ftype_id == cgobject.TYPE_FLAGS else GLibEnum)
 414         type_name = cgobject.type_name(type_id)
 415         enum_name = self._transformer.strip_namespace_object(type_name)
 416         node = klass(enum_name, type_name, members, symbol)
 417         self._add_attribute(node, replace=True)
 418         self._register_internal_type(type_name, node)
 419
 420     def _introspect_object(self, type_id, symbol):
 421         type_name = cgobject.type_name(type_id)
 422         parent_type_name = cgobject.type_name(cgobject.type_parent(type_id))
 423         parent_gitype = self._resolve_gtypename(parent_type_name)
 424         node = GLibObject(
 425             self._transformer.strip_namespace_object(type_name),
 426             parent_gitype,
 427             type_name, symbol)
 428         self._introspect_properties(node, type_id)
 429         self._introspect_signals(node, type_id)
 430         self._add_attribute(node, replace=True)
 431         self._register_internal_type(type_name, node)
 432
 433     def _introspect_interface(self, type_id, symbol):
 434         type_name = cgobject.type_name(type_id)
 435         parent_type_name = cgobject.type_name(cgobject.type_parent(type_id))
 436         if parent_type_name == 'GInterface':
 437             parent_gitype = None
 438         else:
 439             parent_gitype = self._resolve_gtypename(parent_type_name)
 440         node = GLibInterface(
 441             self._transformer.strip_namespace_object(type_name),
 442             parent_gitype,
 443             type_name, symbol)
 444         self._introspect_properties(node, type_id)
 445         self._introspect_signals(node, type_id)
 446         self._add_attribute(node, replace=True)
 447         self._register_internal_type(type_name, node)
 448
 449     def _introspect_boxed(self, type_id, symbol):
 450         type_name = cgobject.type_name(type_id)
 451         node = GLibBoxed(self._transformer.strip_namespace_object(type_name),
 452                          type_name, symbol)
 453         self._add_attribute(node, replace=True)
 454         self._register_internal_type(type_name, node)
 455
 456     def _introspect_properties(self, node, type_id):
 457         fundamental_type_id = cgobject.type_fundamental(type_id)
 458         if fundamental_type_id == cgobject.TYPE_OBJECT:
 459             pspecs = cgobject.object_class_list_properties(type_id)
 460         elif fundamental_type_id == cgobject.TYPE_INTERFACE:
 461             pspecs = cgobject.object_interface_list_properties(type_id)
 462         else:
 463             raise AssertionError
 464
 465         for pspec in pspecs:
 466             if pspec.owner_type != type_id:
 467                 continue
 468             ctype = cgobject.type_name(pspec.value_type)
 469             node.properties.append(Property(
 470                 pspec.name,
 471                 type_name_from_ctype(ctype),
 472                 ctype,
 473                 ))
 474
 475     def _introspect_signals(self, node, type_id):
 476         for signal_info in cgobject.signal_list(type_id):
 477             rtype = self._create_type(signal_info.return_type)
 478             return_ = Return(rtype)
 479             signal = GLibSignal(signal_info.signal_name, return_)
 480             for i, parameter in enumerate(signal_info.get_params()):
 481                 if i == 0:
 482                     name = 'object'
 483                 else:
 484                     name = 'p%s' % (i-1, )
 485                 ptype = self._create_type(parameter)
 486                 param = Parameter(name, ptype)
 487                 signal.parameters.append(param)
 488             node.signals.append(signal)
 489
 490     # Resolver
 491
 492     def _resolve_type_name(self, type_name, ctype=None):
 493         # Workaround glib bug #548689, to be included in 2.18.0
 494         if type_name == "GParam":
 495             type_name = "GObject.ParamSpec"
 496
 497         res = self._transformer.resolve_type_name_full
 498         try:
 499             return res(type_name, ctype, self._names)
 500         except KeyError, e:
 501             return self._transformer.resolve_type_name(type_name, ctype)
 502
 503     def _validate_type_name(self, name):
 504         if name in type_names:
 505             return True
 506         if name.find('.') >= 0:
 507             return True
 508         if name in self._names.aliases:
 509             return True
 510         if name in self._names.names:
 511             return True
 512         return False
 513
 514     def _validate_type(self, ptype):
 515         if isinstance(ptype, Sequence):
 516             etype = ptype.element_type
 517             if isinstance(etype, Sequence):
 518                 return self._validate_type(etype)
 519             return self._validate_type_name(etype)
 520         return self._validate_type_name(ptype.name)
 521
 522     def _resolve_param_type_validate(self, ptype):
 523         ptype = self._resolve_param_type(ptype)
 524         if self._validating and not self._validate_type(ptype):
 525             raise UnknownTypeError("Unknown type %r" % (ptype, ))
 526         return ptype
 527
 528     def _resolve_param_type(self, ptype):
 529         try:
 530             return self._transformer.resolve_param_type_full(ptype,
 531                                                              self._names)
 532         except KeyError, e:
 533             return self._transformer.resolve_param_type(ptype)
 534         return ptype
 535
 536     def _resolve_node(self, node):
 537         if isinstance(node, Function):
 538             self._resolve_function_toplevel(node)
 539
 540         elif isinstance(node, Callback):
 541             self._resolve_function(node)
 542         elif isinstance(node, GLibObject):
 543             self._resolve_glib_object(node)
 544         elif isinstance(node, GLibInterface):
 545             self._resolve_glib_interface(node)
 546         elif isinstance(node, GLibBoxed):
 547             self._resolve_glib_boxed(node)
 548         elif isinstance(node, Struct):
 549             self._resolve_struct(node)
 550         elif isinstance(node, Union):
 551             self._resolve_union(node)
 552         elif isinstance(node, Alias):
 553             self._resolve_alias(node)
 554
 555     def _resolve_function_toplevel(self, func):
 556         newfunc = self._parse_constructor(func)
 557         if not newfunc:
 558             newfunc = self._parse_method(func)
 559             if not newfunc:
 560                 self._resolve_function(func)
 561                 return
 562         self._resolve_function(newfunc)
 563
 564     def _resolve_struct(self, node):
 565         for field in node.fields:
 566             self._resolve_field(field)
 567
 568     def _resolve_union(self, node):
 569         for field in node.fields:
 570             self._resolve_field(field)
 571
 572     def _resolve_parent(self, node):
 573         if not isinstance(node, (GLibInterface, GLibObject)):
 574             raise AssertionError
 575         if isinstance(node.parent, Unresolved):
 576             node.parent = \
 577                 self._transformer.gtypename_to_giname(node.parent.target,
 578                                                       self._names)
 579
 580     def _resolve_glib_interface(self, node):
 581         self._resolve_parent(node)
 582         self._resolve_methods(node.methods)
 583         self._resolve_properties(node.properties)
 584         self._resolve_signals(node.signals)
 585
 586     def _resolve_glib_object(self, node):
 587         self._resolve_parent(node)
 588         self._resolve_constructors(node.constructors)
 589         self._resolve_methods(node.methods)
 590         self._resolve_properties(node.properties)
 591         self._resolve_signals(node.signals)
 592
 593     def _resolve_glib_boxed(self, node):
 594         self._resolve_constructors(node.constructors)
 595         self._resolve_methods(node.methods)
 596
 597     def _resolve_constructors(self, constructors):
 598         for ctor in constructors:
 599             self._resolve_function(ctor)
 600
 601     def _resolve_methods(self, methods):
 602         for method in methods:
 603             self._resolve_function(method)
 604
 605     def _resolve_signals(self, signals):
 606         for signal in signals:
 607             self._resolve_function(signal)
 608
 609     def _resolve_properties(self, properties):
 610         for prop in properties:
 611             self._resolve_property(prop)
 612
 613     def _resolve_property(self, prop):
 614         prop.type = self._resolve_param_type(prop.type)
 615
 616     def _resolve_function(self, func):
 617         self._resolve_parameters(func.parameters)
 618         func.retval.type = self._resolve_param_type(func.retval.type)
 619
 620     def _resolve_parameters(self, parameters):
 621         for parameter in parameters:
 622             parameter.type = self._resolve_param_type(parameter.type)
 623
 624     def _resolve_field(self, field):
 625         if isinstance(field, Callback):
 626             self._resolve_function(field)
 627             return
 628         field.type = self._resolve_param_type(field.type)
 629
 630     def _resolve_alias(self, alias):
 631         alias.target = self._resolve_type_name(alias.target, alias.target)
 632
 633     # Validation
 634
 635     def _validate(self, nodes):
 636         nodes = list(self._names.names.itervalues())
 637         i = 0
 638         self._validating = True
 639         while True:
 640             print "Type resolution; pass=%d" % (i, )
 641             initlen = len(nodes)
 642             nodes = list(self._names.names.itervalues())
 643             for node in nodes:
 644                 try:
 645                     self._resolve_node(node)
 646                 except UnknownTypeError, e:
 647                     print "WARNING: %s: Deleting %r" % (e, node)
 648                     self._remove_attribute(node.name)
 649             if len(nodes) == initlen:
 650                 break
 651             i += 1
 652         self._validating = False