Commit 034a53fd authored by Xavier Thompson's avatar Xavier Thompson

Compute the MRO for each cycplass

parent 631d465b
...@@ -2198,6 +2198,13 @@ if VALUE is not None: ...@@ -2198,6 +2198,13 @@ if VALUE is not None:
return property return property
class ComputeMROTransform(CythonTransform):
def visit_CppClassNode(self, node):
if node.cypclass:
node.entry.type.compute_mro()
return node
class CalculateQualifiedNamesTransform(EnvTransform): class CalculateQualifiedNamesTransform(EnvTransform):
""" """
Calculate and store the '__qualname__' and the global Calculate and store the '__qualname__' and the global
......
...@@ -143,6 +143,7 @@ def create_pipeline(context, mode, exclude_classes=()): ...@@ -143,6 +143,7 @@ def create_pipeline(context, mode, exclude_classes=()):
from .Visitor import PrintTree from .Visitor import PrintTree
from .ParseTreeTransforms import WithTransform, NormalizeTree, PostParse, PxdPostParse from .ParseTreeTransforms import WithTransform, NormalizeTree, PostParse, PxdPostParse
from .ParseTreeTransforms import ForwardDeclareTypes, InjectGilHandling, AnalyseDeclarationsTransform from .ParseTreeTransforms import ForwardDeclareTypes, InjectGilHandling, AnalyseDeclarationsTransform
from .ParseTreeTransforms import ComputeMROTransform
from .ParseTreeTransforms import AnalyseExpressionsTransform, FindInvalidUseOfFusedTypes from .ParseTreeTransforms import AnalyseExpressionsTransform, FindInvalidUseOfFusedTypes
from .ParseTreeTransforms import CreateClosureClasses, MarkClosureVisitor, DecoratorTransform from .ParseTreeTransforms import CreateClosureClasses, MarkClosureVisitor, DecoratorTransform
from .ParseTreeTransforms import TrackNumpyAttributes, InterpretCompilerDirectives, TransformBuiltinMethods from .ParseTreeTransforms import TrackNumpyAttributes, InterpretCompilerDirectives, TransformBuiltinMethods
...@@ -197,6 +198,7 @@ def create_pipeline(context, mode, exclude_classes=()): ...@@ -197,6 +198,7 @@ def create_pipeline(context, mode, exclude_classes=()):
ForwardDeclareTypes(context), ForwardDeclareTypes(context),
InjectGilHandling(), InjectGilHandling(),
AnalyseDeclarationsTransform(context), AnalyseDeclarationsTransform(context),
ComputeMROTransform(context),
AutoTestDictTransform(context), AutoTestDictTransform(context),
EmbedSignature(context), EmbedSignature(context),
EarlyReplaceBuiltinCalls(context), ## Necessary? EarlyReplaceBuiltinCalls(context), ## Necessary?
......
...@@ -4109,9 +4109,63 @@ class CppScopedEnumType(CType): ...@@ -4109,9 +4109,63 @@ class CppScopedEnumType(CType):
env.use_utility_code(rst) env.use_utility_code(rst)
# Merge procedure for the C3 linearisation used to produce the MRO
#
# sources:
# - https://www.python.org/download/releases/2.3/mro/
# - https://mail.python.org/pipermail/python-dev/2002-October/029176.html
# - https://github.com/mikeboers/C3Linearize/blob/master/c3linearize.py
#
def mro_C3_merge(sequences):
# make copies of the input lists to avoid side effect mutations
sequences = [list(s) for s in sequences]
result = []
while True:
# remove empty sequences from consideration
sequences = [s for s in sequences if s]
# the result is fully computed when there are no more sequences to examine
if not sequences:
return result
# find the first good head: a head of one of the sequences that is not in any tail
for sequence in sequences:
head = sequence[0]
if not any(head in s[1:] for s in sequences):
break
else:
error(None, "Inconsistent inheritance hierarchy for %s" % self.name)
# make the head that was found the next element in the linearisation
result.append(head)
# remove that head from consideration
for seq in sequences:
if seq[0] == head:
del seq[0]
# Compute MRO for generic classes with minimal assumptions
#
# One assumption only:
# - if a class has bases, they are held in 'base_classes' attribute
#
def compute_mro_generic(cls):
print("GENERIC C3 !!")
if not hasattr(cls, "base_classes") or cls.base_classes is None:
return [cls]
inputs = [[cls]]
for base in cls.base_classes:
inputs.append(compute_mro_generic(base))
inputs.append(cls.base_classes)
return mro_C3_merge(inputs)
class CypClassType(CppClassType): class CypClassType(CppClassType):
# lock_mode string (tri-state: "nolock"/"checklock"/"autolock") # lock_mode string (tri-state: "nolock"/"checklock"/"autolock")
# mro [CppClassType] or None The Method Resolution Order of this cypclass according to Python
is_cyp_class = 1 is_cyp_class = 1
...@@ -4119,6 +4173,27 @@ class CypClassType(CppClassType): ...@@ -4119,6 +4173,27 @@ class CypClassType(CppClassType):
CppClassType.__init__(self, name, scope, cname, base_classes, templates, template_type, nogil) CppClassType.__init__(self, name, scope, cname, base_classes, templates, template_type, nogil)
self.lock_mode = lock_mode if lock_mode else "autolock" self.lock_mode = lock_mode if lock_mode else "autolock"
self.activable = activable self.activable = activable
self.mro = None
# compute the MRO for this cypclass
# the mro is also computed for bases when needed
# based on https://mail.python.org/pipermail/python-dev/2002-October/029176.html
def compute_mro(self):
if self.mro is not None:
return self.mro
if self.base_classes is None:
self.mro = [self]
return self.mro
inputs = [[self]]
for base in self.base_classes:
if base.is_cyp_class:
base_mro = base.compute_mro()
else:
base_mro = compute_mro_generic(base)
inputs.append(base_mro)
inputs.append(self.base_classes)
self.mro = mro_C3_merge(inputs)
return self.mro
def empty_declaration_code(self): def empty_declaration_code(self):
if self._empty_declaration is None: if self._empty_declaration is None:
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment