Add tests for raw layout vector

Author: Azoy

lib/IRGen/GenRecord.h

@@ -203,9 +203,10 @@ class RecordTypeInfoImpl : public Base,
     }
 
     if (auto rawLayout = T.getRawLayout()) {
-      return takeRawLayout(IGF, dest, src, T, isOutlined,
-                           /* zeroizeIfSensitive */ false, rawLayout,
-                           /* isInit */ false);
+      return handleRawLayout(IGF, dest, src, T, isOutlined, rawLayout,
+            [&](const TypeInfo &ti, SILType type, Address dest, Address src) {
+        ti.assignWithTake(IGF, dest, src, type, isOutlined);
+      });
     }
 
     if (isOutlined || T.hasParameterizedExistential()) {
@@ -268,8 +269,10 @@ class RecordTypeInfoImpl : public Base,
       // If the fields are not ABI-accessible, use the value witness table.
       return emitInitializeWithTakeCall(IGF, T, dest, src);
     } else if (auto rawLayout = T.getRawLayout()) {
-      return takeRawLayout(IGF, dest, src, T, isOutlined, zeroizeIfSensitive,
-                           rawLayout, /* isInit */ true);
+      return handleRawLayout(IGF, dest, src, T, isOutlined, rawLayout,
+            [&](const TypeInfo &ti, SILType type, Address dest, Address src) {
+        ti.initializeWithTake(IGF, dest, src, type, isOutlined, zeroizeIfSensitive);
+      });
     } else if (isOutlined || T.hasParameterizedExistential()) {
       auto offsets = asImpl().getNonFixedOffsets(IGF, T);
       for (auto &field : getFields()) {
@@ -289,31 +292,21 @@ class RecordTypeInfoImpl : public Base,
       fillWithZerosIfSensitive(IGF, src, T);
   }
 
-  void takeRawLayout(IRGenFunction &IGF, Address dest, Address src, SILType T,
-                     bool isOutlined, bool zeroizeIfSensitive,
-                     RawLayoutAttr *rawLayout, bool isInit) const {
+  void handleRawLayout(IRGenFunction &IGF, Address dest, Address src, SILType T,
+                       bool isOutlined, RawLayoutAttr *rawLayout,
+                       std::function<void
+                          (const TypeInfo &, SILType, Address, Address)> body) const {
     if (rawLayout->shouldMoveAsLikeType()) {
       // Because we have a rawlayout attribute, we know this has to be a struct.
       auto structDecl = T.getStructOrBoundGenericStruct();
 
-      auto take = [&](const TypeInfo &likeTypeInfo, SILType loweredLikeType,
-                      Address dest, Address src) {
-        if (isInit) {
-          likeTypeInfo.initializeWithTake(IGF, dest, src, loweredLikeType,
-                                        isOutlined, zeroizeIfSensitive);
-        } else {
-          likeTypeInfo.assignWithTake(IGF, dest, src, loweredLikeType,
-                                      isOutlined);
-        }
-      };
-
       if (auto likeType = rawLayout->getResolvedScalarLikeType(structDecl)) {
         auto astT = T.getASTType();
         auto subs = astT->getContextSubstitutionMap();
         auto loweredLikeType = IGF.IGM.getLoweredType(likeType->subst(subs));
         auto &likeTypeInfo = IGF.IGM.getTypeInfo(loweredLikeType);
 
-        take(likeTypeInfo, loweredLikeType, dest, src);
+        body(likeTypeInfo, loweredLikeType, dest, src);
       }
 
       if (auto likeArray = rawLayout->getResolvedArrayLikeTypeAndCount(structDecl)) {
@@ -329,7 +322,7 @@ class RecordTypeInfoImpl : public Base,
         IGF.emitLoopOverElements(likeTypeInfo, loweredLikeType,
                                  countType->getCanonicalType(), dest, src,
             [&](Address dest, Address src) {
-          take(likeTypeInfo, loweredLikeType, dest, src);
+          body(likeTypeInfo, loweredLikeType, dest, src);
         });
       }
     }
@@ -342,6 +335,13 @@ class RecordTypeInfoImpl : public Base,
       return emitDestroyCall(IGF, T, addr);
     }
 
+    if (auto rawLayout = T.getRawLayout()) {
+      return handleRawLayout(IGF, Address(), addr, T, isOutlined, rawLayout,
+            [&](const TypeInfo &ti, SILType type, Address dest, Address src) {
+        ti.destroy(IGF, src, type, isOutlined);
+      });
+    }
+
     if (isOutlined || T.hasParameterizedExistential()) {
       auto offsets = asImpl().getNonFixedOffsets(IGF, T);
       for (auto &field : getFields()) {