[AST] Tail allocate ellipsis location and index in TupleTypeRepr

Author: rintaro

include/swift/AST/TypeRepr.h

@@ -545,20 +545,20 @@ class ImplicitlyUnwrappedOptionalTypeRepr : public TypeRepr {
 /// \endcode
 class TupleTypeRepr final : public TypeRepr,
     private llvm::TrailingObjects<TupleTypeRepr, TypeRepr*, Identifier,
-                                  SourceLoc> {
+                                  SourceLoc, std::pair<SourceLoc, unsigned>> {
   friend TrailingObjects;
+  typedef std::pair<SourceLoc, unsigned> SourceLocAndIdx;
 
   unsigned NumElements;
   SourceRange Parens;
-  SourceLoc Ellipsis;
-  unsigned EllipsisIdx;
 
   enum {
     NotNamed = 0,
     HasNames = 1,
     HasLabels = 2
-  } NameStatus;
-
+  } NameStatus : 2;
+  bool HasEllipsis : 1;
+  
   size_t numTrailingObjects(OverloadToken<TypeRepr *>) const {
     return NumElements;
   }
@@ -622,13 +622,21 @@ class TupleTypeRepr final : public TypeRepr,
   }
 
   SourceRange getParens() const { return Parens; }
-  SourceLoc getEllipsisLoc() const { return Ellipsis; }
-  unsigned getEllipsisIndex() const { return EllipsisIdx; }
-  bool hasEllipsis() const { return Ellipsis.isValid(); }
+  SourceLoc getEllipsisLoc() const {
+    return HasEllipsis ?
+      getTrailingObjects<SourceLocAndIdx>()[0].first : SourceLoc();
+  }
+  unsigned getEllipsisIndex() const {
+    return HasEllipsis ?
+      getTrailingObjects<SourceLocAndIdx>()[0].second : NumElements;
+  }
+  bool hasEllipsis() const { return HasEllipsis; }
 
   void removeEllipsis() {
-    Ellipsis = SourceLoc();
-    EllipsisIdx = NumElements;
+    if (HasEllipsis) {
+      HasEllipsis = false;
+      getTrailingObjects<SourceLocAndIdx>()[0] = {SourceLoc(), NumElements};
+    }
   }
 
   bool isParenType() const {

lib/AST/TypeRepr.cpp

@@ -394,7 +394,7 @@ TupleTypeRepr::TupleTypeRepr(ArrayRef<TypeRepr *> Elements, SourceRange Parens,
                              ArrayRef<SourceLoc> underscoreLocs,
                              SourceLoc Ellipsis, unsigned EllipsisIdx)
     : TypeRepr(TypeReprKind::Tuple), NumElements(Elements.size()),
-      Parens(Parens), Ellipsis(Ellipsis), EllipsisIdx(EllipsisIdx) {
+      Parens(Parens), HasEllipsis(Ellipsis.isValid()) {
 
   NameStatus = ElementNames.empty() ? NotNamed
              : underscoreLocs.empty() ? HasNames : HasLabels;
@@ -415,6 +415,10 @@ TupleTypeRepr::TupleTypeRepr(ArrayRef<TypeRepr *> Elements, SourceRange Parens,
   std::uninitialized_copy(underscoreLocs.begin(), underscoreLocs.end(),
                           getTrailingObjects<SourceLoc>() +
                               ElementNameLocs.size());
+
+  // Set ellipsis location and index.
+  if (Ellipsis.isValid())
+    getTrailingObjects<SourceLocAndIdx>()[0] = {Ellipsis, EllipsisIdx};
 }
 
 TupleTypeRepr *TupleTypeRepr::create(const ASTContext &C,
@@ -432,10 +436,12 @@ TupleTypeRepr *TupleTypeRepr::create(const ASTContext &C,
   assert(Ellipsis.isValid() ? EllipsisIdx < Elements.size()
                             : EllipsisIdx == Elements.size());
 
-  size_t size = totalSizeToAlloc<TypeRepr *, Identifier, SourceLoc>(
+  size_t size =
+    totalSizeToAlloc<TypeRepr *, Identifier, SourceLoc, SourceLocAndIdx>(
       Elements.size(),
       ElementNames.size(),
-      ElementNameLocs.size() + underscoreLocs.size());
+      ElementNameLocs.size() + underscoreLocs.size(),
+      Ellipsis.isValid() ? 1 : 0);
   void *mem = C.Allocate(size, alignof(TupleTypeRepr));
   return new (mem) TupleTypeRepr(Elements, Parens, ElementNames,
                                  ElementNameLocs, underscoreLocs,