SwiftCompilerSources API for SourceDestAddrInstructions.

Author: atrick

SwiftCompilerSources/Sources/SIL/Instruction.swift

@@ -254,6 +254,7 @@ extension UnaryInstruction {
 final public class UnimplementedInstruction : Instruction {
 }
 
+/// Only one of the operands may have an address type.
 public protocol StoringInstruction : Instruction {
   var operands: OperandArray { get }
 }
@@ -303,14 +304,16 @@ final public class AssignInst : Instruction, StoringInstruction {
   }
 }
 
-final public class AssignByWrapperInst : Instruction {}
+final public class AssignByWrapperInst : Instruction, StoringInstruction {}
 
-final public class AssignOrInitInst : Instruction {}
+final public class AssignOrInitInst : Instruction, StoringInstruction {}
 
 /// Instruction that copy or move from a source to destination address.
 public protocol SourceDestAddrInstruction : Instruction {
   var sourceOperand: Operand { get }
   var destinationOperand: Operand { get }
+  var isTakeOfSrc: Bool { get }
+  var isInitializationOfDest: Bool { get }
 }
 
 extension SourceDestAddrInstruction {
@@ -333,6 +336,13 @@ final public class CopyAddrInst : Instruction, SourceDestAddrInstruction {
 final public class ExplicitCopyAddrInst : Instruction, SourceDestAddrInstruction {
   public var source: Value { return sourceOperand.value }
   public var destination: Value { return destinationOperand.value }
+
+  public var isTakeOfSrc: Bool {
+    bridged.ExplicitCopyAddrInst_isTakeOfSrc()
+  }
+  public var isInitializationOfDest: Bool {
+    bridged.ExplicitCopyAddrInst_isInitializationOfDest()
+  }
 }
 
 final public class EndAccessInst : Instruction, UnaryInstruction {
@@ -590,7 +600,10 @@ class UncheckedRefCastInst : SingleValueInstruction, ConversionInstruction {
 }
 
 final public
-class UncheckedRefCastAddrInst : Instruction, SourceDestAddrInstruction {}
+class UncheckedRefCastAddrInst : Instruction, SourceDestAddrInstruction {
+  public var isTakeOfSrc: Bool { true }
+  public var isInitializationOfDest: Bool { true }
+}
 
 final public class UncheckedAddrCastInst : SingleValueInstruction, UnaryInstruction {
   public var fromAddress: Value { operand.value }
@@ -777,7 +790,9 @@ class TupleElementAddrInst : SingleValueInstruction, UnaryInstruction {
   public var fieldIndex: Int { bridged.TupleElementAddrInst_fieldIndex() }
 }
 
-final public class TupleAddrConstructorInst : Instruction {}
+final public class TupleAddrConstructorInst : Instruction {
+  public var destinationOperand: Operand { operands[0] }
+}
 
 final public class StructInst : SingleValueInstruction, ForwardingInstruction {
 }
@@ -1027,7 +1042,10 @@ final public class IsEscapingClosureInst : SingleValueInstruction, UnaryInstruct
 final public
 class MarkUnresolvedNonCopyableValueInst : SingleValueInstruction, UnaryInstruction {}
 
-final public class MarkUnresolvedMoveAddrInst : Instruction, SourceDestAddrInstruction {}
+final public class MarkUnresolvedMoveAddrInst : Instruction, SourceDestAddrInstruction {
+  public var isTakeOfSrc: Bool { true }
+  public var isInitializationOfDest: Bool { true }
+}
 
 final public
 class CopyableToMoveOnlyWrapperAddrInst : SingleValueInstruction, UnaryInstruction {}

SwiftCompilerSources/Sources/SIL/Operand.swift

@@ -163,6 +163,25 @@ extension Sequence where Element == Operand {
   }
 }
 
+extension Operand {
+  /// Return true if this operation will store a full value into this
+  /// operand's address.
+  public var isAddressInitialization: Bool {
+    if !value.type.isAddress {
+      return false
+    }
+    switch instruction {
+    case is StoringInstruction:
+      return true
+    case let srcDestInst as SourceDestAddrInstruction
+           where srcDestInst.destinationOperand == self:
+      return true
+    default:
+      return false
+    }
+  }
+}
+
 extension OptionalBridgedOperand {
   init(bridged: BridgedOperand?) {
     self = OptionalBridgedOperand(op: bridged?.op)

include/swift/SIL/SILBridging.h

@@ -826,6 +826,8 @@ struct BridgedInstruction {
   BRIDGED_INLINE bool BeginAccessInst_isStatic() const;
   BRIDGED_INLINE bool CopyAddrInst_isTakeOfSrc() const;
   BRIDGED_INLINE bool CopyAddrInst_isInitializationOfDest() const;
+  BRIDGED_INLINE bool ExplicitCopyAddrInst_isTakeOfSrc() const;
+  BRIDGED_INLINE bool ExplicitCopyAddrInst_isInitializationOfDest() const;
   BRIDGED_INLINE SwiftInt MarkUninitializedInst_getKind() const;
   BRIDGED_INLINE void RefCountingInst_setIsAtomic(bool isAtomic) const;
   BRIDGED_INLINE bool RefCountingInst_getIsAtomic() const;

include/swift/SIL/SILBridgingImpl.h

@@ -1092,6 +1092,14 @@ bool BridgedInstruction::CopyAddrInst_isInitializationOfDest() const {
   return getAs<swift::CopyAddrInst>()->isInitializationOfDest();
 }
 
+bool BridgedInstruction::ExplicitCopyAddrInst_isTakeOfSrc() const {
+  return getAs<swift::ExplicitCopyAddrInst>()->isTakeOfSrc();
+}
+
+bool BridgedInstruction::ExplicitCopyAddrInst_isInitializationOfDest() const {
+  return getAs<swift::ExplicitCopyAddrInst>()->isInitializationOfDest();
+}
+
 SwiftInt BridgedInstruction::MarkUninitializedInst_getKind() const {
   return (SwiftInt)getAs<swift::MarkUninitializedInst>()->getMarkUninitializedKind();
 }