mirror of https://github.com/mono/CppSharp.git
2 changed files with 132 additions and 122 deletions
@ -0,0 +1,132 @@
@@ -0,0 +1,132 @@
|
||||
using System; |
||||
using System.Collections.Generic; |
||||
using System.Linq; |
||||
using CppSharp.AST.Extensions; |
||||
|
||||
namespace CppSharp.AST |
||||
{ |
||||
public static class ClassExtensions |
||||
{ |
||||
public static IEnumerable<Method> FindOperator(this Class c, CXXOperatorKind kind) |
||||
{ |
||||
return c.Operators.Where(method => method.OperatorKind == kind); |
||||
} |
||||
|
||||
public static IEnumerable<Method> FindMethodByOriginalName(this Class c, string name) |
||||
{ |
||||
return c.Methods.Where(method => method.OriginalName == name); |
||||
} |
||||
|
||||
public static IEnumerable<Variable> FindVariableByOriginalName(this Class c, string originalName) |
||||
{ |
||||
return c.Variables.Where(v => v.OriginalName == originalName); |
||||
} |
||||
|
||||
public static IEnumerable<Function> GetFunctionOverloads(this Class c, Function function) |
||||
{ |
||||
if (function.IsOperator) |
||||
return c.FindOperator(function.OperatorKind); |
||||
return c.Methods.Where(method => method.Name == function.Name); |
||||
} |
||||
|
||||
public static IEnumerable<T> FindHierarchy<T>(this Class c, Func<Class, IEnumerable<T>> func) |
||||
where T : Declaration |
||||
{ |
||||
foreach (var elem in func(c)) |
||||
yield return elem; |
||||
|
||||
foreach (var @base in c.Bases) |
||||
{ |
||||
if (!@base.IsClass) continue; |
||||
foreach(var elem in @base.Class.FindHierarchy<T>(func)) |
||||
yield return elem; |
||||
} |
||||
} |
||||
|
||||
public static Method GetRootBaseMethod(this Class c, Method @override, bool onlyFirstBase = false) |
||||
{ |
||||
return (from @base in c.Bases |
||||
where @base.IsClass && (!onlyFirstBase || !@base.Class.IsInterface) |
||||
let baseMethod = ( |
||||
from method in @base.Class.Methods |
||||
where |
||||
method.Name == @override.Name && |
||||
method.ReturnType == @override.ReturnType && |
||||
method.Parameters.Count == @override.Parameters.Count && |
||||
method.Parameters.SequenceEqual(@override.Parameters, |
||||
new ParameterTypeComparer()) |
||||
select method).FirstOrDefault() |
||||
let rootBaseMethod = @base.Class.GetRootBaseMethod(@override) ?? baseMethod |
||||
where rootBaseMethod != null || onlyFirstBase |
||||
select rootBaseMethod).FirstOrDefault(); |
||||
} |
||||
|
||||
public static Property GetRootBaseProperty(this Class c, Property @override, bool onlyFirstBase = false) |
||||
{ |
||||
return (from @base in c.Bases |
||||
where (!onlyFirstBase || !@base.Class.IsInterface) && @base.IsClass |
||||
let baseProperty = ( |
||||
from property in @base.Class.Properties |
||||
where |
||||
property.Name == @override.Name && |
||||
property.Parameters.Count == @override.Parameters.Count && |
||||
property.Parameters.SequenceEqual(@override.Parameters, |
||||
new ParameterTypeComparer()) |
||||
select property).FirstOrDefault() |
||||
let rootBaseProperty = @base.Class.GetRootBaseProperty(@override) ?? baseProperty |
||||
where rootBaseProperty != null || onlyFirstBase |
||||
select rootBaseProperty).FirstOrDefault(); |
||||
} |
||||
|
||||
public static Property GetPropertyByName(this Class c, string propertyName) |
||||
{ |
||||
Property property = c.Properties.FirstOrDefault(m => m.Name == propertyName); |
||||
if (property != null) |
||||
return property; |
||||
Declaration decl; |
||||
foreach (var baseClassSpecifier in c.Bases.Where( |
||||
b => b.Type.IsTagDecl(out decl) && !b.Class.Ignore)) |
||||
{ |
||||
property = baseClassSpecifier.Class.GetPropertyByName(propertyName); |
||||
if (property != null) |
||||
return property; |
||||
} |
||||
return null; |
||||
} |
||||
|
||||
public static Property GetPropertyByConstituentMethod(this Class c, Method method) |
||||
{ |
||||
var property = c.Properties.FirstOrDefault(p => p.GetMethod == method); |
||||
if (property != null) |
||||
return property; |
||||
property = c.Properties.FirstOrDefault(p => p.SetMethod == method); |
||||
if (property != null) |
||||
return property; |
||||
Declaration decl; |
||||
foreach (BaseClassSpecifier @base in c.Bases.Where(b => b.Type.IsTagDecl(out decl))) |
||||
{ |
||||
property = @base.Class.GetPropertyByConstituentMethod(method); |
||||
if (property != null) |
||||
return property; |
||||
} |
||||
return null; |
||||
} |
||||
|
||||
public static Method GetMethodByName(this Class c, string methodName) |
||||
{ |
||||
var method = c.Methods.FirstOrDefault( |
||||
// HACK: because of the non-shared v-table entries bug one copy may have been renamed and the other not
|
||||
m => string.Compare(m.Name, methodName, StringComparison.OrdinalIgnoreCase) == 0); |
||||
if (method != null) |
||||
return method; |
||||
Declaration decl; |
||||
foreach (BaseClassSpecifier @base in c.Bases.Where(b => b.Type.IsTagDecl(out decl))) |
||||
{ |
||||
method = @base.Class.GetMethodByName(methodName); |
||||
if (method != null) |
||||
return method; |
||||
} |
||||
return null; |
||||
} |
||||
} |
||||
} |
Loading…
Reference in new issue