libswift: add CalleeAnalysis

Bridging to BasicCalleeAnalysis

Author: eeckstein

include/swift/SILOptimizer/OptimizerBridging.h

@@ -33,6 +33,16 @@ typedef struct {
   const void * _Nonnull aliasAnalysis;
 } BridgedAliasAnalysis;
 
+typedef struct {
+  void * _Nullable bca;
+} BridgedCalleeAnalysis;
+
+typedef struct {
+  void * _Nonnull opaquePtr;
+  unsigned char kind;
+  unsigned char incomplete;
+} BridgedCalleeList;
+
 typedef void (* _Nonnull BridgedFunctionPassRunFn)(BridgedFunctionPassCtxt);
 typedef void (* _Nonnull BridgedInstructionPassRunFn)(BridgedInstructionPassCtxt);
 
@@ -49,6 +59,14 @@ BridgedMemoryBehavior AliasAnalysis_getMemBehavior(BridgedAliasAnalysis aa,
                                                    BridgedInstruction inst,
                                                    BridgedValue addr);
 
+BridgedCalleeAnalysis PassContext_getCalleeAnalysis(BridgedPassContext context);
+
+BridgedCalleeList CalleeAnalysis_getCallees(BridgedCalleeAnalysis calleeAnalysis,
+                                            BridgedValue callee);
+SwiftInt BridgedFunctionArray_size(BridgedCalleeList callees);
+BridgedFunction BridgedFunctionArray_get(BridgedCalleeList callees,
+                                         SwiftInt index);
+
 #ifdef __cplusplus
 } // extern "C"
 #endif

lib/SILOptimizer/Analysis/BasicCalleeAnalysis.cpp

@@ -16,6 +16,8 @@
 #include "swift/AST/ProtocolConformance.h"
 #include "swift/Basic/Statistic.h"
 #include "swift/SIL/SILModule.h"
+#include "swift/SIL/SILBridgingUtils.h"
+#include "swift/SILOptimizer/OptimizerBridging.h"
 #include "swift/SILOptimizer/Utils/InstOptUtils.h"
 #include "llvm/Support/Compiler.h"
 
@@ -318,3 +320,29 @@ void BasicCalleeAnalysis::print(llvm::raw_ostream &os) const {
     }
   }
 }
+
+//===----------------------------------------------------------------------===//
+//                            Swift Bridging
+//===----------------------------------------------------------------------===//
+
+BridgedCalleeList CalleeAnalysis_getCallees(BridgedCalleeAnalysis calleeAnalysis,
+                                            BridgedValue callee) {
+  BasicCalleeAnalysis *bca = static_cast<BasicCalleeAnalysis *>(calleeAnalysis.bca);
+  CalleeList cl = bca->getCalleeListOfValue(castToSILValue(callee));
+  return {cl.getOpaquePtr(), cl.getOpaqueKind(), cl.isIncomplete()};
+}
+
+SwiftInt BridgedFunctionArray_size(BridgedCalleeList callees) {
+  CalleeList cl = CalleeList::fromOpaque(callees.opaquePtr, callees.kind,
+                                         callees.incomplete);
+  return cl.end() - cl.begin();
+}
+
+BridgedFunction BridgedFunctionArray_get(BridgedCalleeList callees,
+                                         SwiftInt index) {
+  CalleeList cl = CalleeList::fromOpaque(callees.opaquePtr, callees.kind,
+                                         callees.incomplete);
+  auto iter = cl.begin() + index;
+  assert(index >= 0 && iter < cl.end());
+  return {*iter};
+}

lib/SILOptimizer/PassManager/PassManager.cpp

@@ -1172,3 +1172,7 @@ BridgedAliasAnalysis PassContext_getAliasAnalysis(BridgedPassContext context,
   return {pm->getAnalysis<AliasAnalysis>(castToFunction(function))};
 }
 
+BridgedCalleeAnalysis PassContext_getCalleeAnalysis(BridgedPassContext context) {
+  SILPassManager *pm = castToPassInvocation(context)->getPassManager();
+  return {pm->getAnalysis<BasicCalleeAnalysis>()};
+}

libswift/Sources/Optimizer/Analysis/CMakeLists.txt

@@ -7,4 +7,5 @@
 # See http://swift.org/CONTRIBUTORS.txt for Swift project authors
 
 libswift_sources(Optimizer
-  AliasAnalysis.swift)
+  AliasAnalysis.swift
+  CalleeAnalysis.swift)

libswift/Sources/Optimizer/Analysis/CalleeAnalysis.swift

@@ -0,0 +1,40 @@
+//===--- CalleeAnalysis.swift - the callee analysis -----------------------===//
+//
+// This source file is part of the Swift.org open source project
+//
+// Copyright (c) 2014 - 2021 Apple Inc. and the Swift project authors
+// Licensed under Apache License v2.0 with Runtime Library Exception
+//
+// See https://swift.org/LICENSE.txt for license information
+// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
+//
+//===----------------------------------------------------------------------===//
+
+import OptimizerBridging
+import SIL
+
+public struct CalleeAnalysis {
+  let bridged: BridgedCalleeAnalysis
+
+  public func getCallees(callee: Value) -> FunctionArray {
+    return FunctionArray(bridged: CalleeAnalysis_getCallees(bridged, callee.bridged))
+  }
+}
+
+public struct FunctionArray : RandomAccessCollection, CustomReflectable {
+  fileprivate let bridged: BridgedCalleeList
+  
+  public var startIndex: Int { 0 }
+  public var endIndex: Int { BridgedFunctionArray_size(bridged) }
+
+  public subscript(_ index: Int) -> Function {
+    return BridgedFunctionArray_get(bridged, index).function
+  }
+
+  public var allCalleesKnown: Bool { bridged.incomplete == 0 }
+
+  public var customMirror: Mirror {
+    let c: [Mirror.Child] = map { (label: nil, value: $0.name) }
+    return Mirror(self, children: c)
+  }
+}

libswift/Sources/Optimizer/PassManager/PassUtils.swift

@@ -35,6 +35,11 @@ struct FunctionPassContext {
     let bridgedAA = PassContext_getAliasAnalysis(passContext, function.bridged)
     return AliasAnalysis(bridged: bridgedAA)
   }
+  
+  var calleeAnalysis: CalleeAnalysis {
+    let bridgeCA = PassContext_getCalleeAnalysis(passContext)
+    return CalleeAnalysis(bridged: bridgeCA)
+  }
 }
 
 struct FunctionPass {