[SILOptimizer] Implement lowering of type wrapper assignment instructions

Type wrapper originated assignment instructions are lowered similar
to property wrapper ones except they don't require initializer call
because assignments are done to tuple fields.

Author: xedin

lib/SILOptimizer/Mandatory/RawSILInstLowering.cpp

@@ -165,9 +165,10 @@ static void getAssignByWrapperArgs(SmallVectorImpl<SILValue> &args,
          "initializer or setter has too many arguments");
 }
 
-static void lowerAssignByWrapperInstruction(SILBuilderWithScope &b,
-                                             AssignByWrapperInst *inst,
-                                        SmallSetVector<SILValue, 8> &toDelete) {
+static void
+lowerAssignByWrapperInstruction(SILBuilderWithScope &b,
+                                AssignByWrapperInst *inst,
+                                SmallSetVector<SILValue, 8> &toDelete) {
   LLVM_DEBUG(llvm::dbgs() << "  *** Lowering " << *inst << "\n");
 
   ++numAssignRewritten;
@@ -186,29 +187,46 @@ static void lowerAssignByWrapperInstruction(SILBuilderWithScope &b,
       LLVM_FALLTHROUGH;
     case AssignByWrapperInst::Initialization:
     case AssignByWrapperInst::Assign: {
-      SILValue initFn = inst->getInitializer();
-      CanSILFunctionType fTy = initFn->getType().castTo<SILFunctionType>();
-      SILFunctionConventions convention(fTy, inst->getModule());
-      SmallVector<SILValue, 4> args;
-      if (convention.hasIndirectSILResults()) {
-        if (inst->getMode() == AssignByWrapperInst::Assign)
-          b.createDestroyAddr(loc, dest);
-
-        args.push_back(dest);
-        getAssignByWrapperArgs(args, src, convention, b, forCleanup);
-        b.createApply(loc, initFn, SubstitutionMap(), args);
-      } else {
-        getAssignByWrapperArgs(args, src, convention, b, forCleanup);
-        SILValue wrappedSrc = b.createApply(loc, initFn, SubstitutionMap(),
-                                            args);
+      switch (inst->getOriginator()) {
+      case AssignByWrapperInst::Originator::TypeWrapper: {
         if (inst->getMode() == AssignByWrapperInst::Initialization ||
             inst->getDest()->getType().isTrivial(*inst->getFunction())) {
-          b.createTrivialStoreOr(loc, wrappedSrc, dest,
-                                 StoreOwnershipQualifier::Init);
+          b.createTrivialStoreOr(loc, src, dest, StoreOwnershipQualifier::Init);
+        } else {
+          b.createStore(loc, src, dest, StoreOwnershipQualifier::Assign);
+        }
+        break;
+      }
+
+      case AssignByWrapperInst::Originator::PropertyWrapper: {
+        SILValue initFn = inst->getInitializer();
+        CanSILFunctionType fTy = initFn->getType().castTo<SILFunctionType>();
+        SILFunctionConventions convention(fTy, inst->getModule());
+        SmallVector<SILValue, 4> args;
+        if (convention.hasIndirectSILResults()) {
+          if (inst->getMode() == AssignByWrapperInst::Assign)
+            b.createDestroyAddr(loc, dest);
+
+          args.push_back(dest);
+          getAssignByWrapperArgs(args, src, convention, b, forCleanup);
+          b.createApply(loc, initFn, SubstitutionMap(), args);
         } else {
-          b.createStore(loc, wrappedSrc, dest, StoreOwnershipQualifier::Assign);
+          getAssignByWrapperArgs(args, src, convention, b, forCleanup);
+          SILValue wrappedSrc =
+              b.createApply(loc, initFn, SubstitutionMap(), args);
+          if (inst->getMode() == AssignByWrapperInst::Initialization ||
+              inst->getDest()->getType().isTrivial(*inst->getFunction())) {
+            b.createTrivialStoreOr(loc, wrappedSrc, dest,
+                                   StoreOwnershipQualifier::Init);
+          } else {
+            b.createStore(loc, wrappedSrc, dest,
+                          StoreOwnershipQualifier::Assign);
+          }
         }
+        break;
       }
+      }
+
       // The unused partial_apply violates memory lifetime rules in case "self"
       // is an inout. Therefore we cannot keep it as a dead closure to be
       // cleaned up later. We have to delete it in this pass.