From 6680ed3956f751a8f6a1ffb48b8d34d40a54bfdc Mon Sep 17 00:00:00 2001
From: Dmitri Hrybenko <dgribenko@apple.com>
Date: Thu, 17 Apr 2014 16:13:47 +0000
Subject: [PATCH] Comment parsing: for now, treat classes, protocols and
 structs as "class-like" things

Swift SVN r16453
---
 lib/IDE/CommentConversion.cpp                      | 3 +++
 test/PrintAsObjC/Inputs/comments-expected-output.h | 5 +++++
 2 files changed, 8 insertions(+)

diff --git a/lib/IDE/CommentConversion.cpp b/lib/IDE/CommentConversion.cpp
index 0b4cd055d7b..c38171f7427 100644
--- a/lib/IDE/CommentConversion.cpp
+++ b/lib/IDE/CommentConversion.cpp
@@ -198,6 +198,9 @@ void CommentToXMLConverter::visitFullComment(const FullComment *FC) {
   if (isa<AbstractFunctionDecl>(D)) {
     OS << "<Function";
     RootEndTag = "</Function>";
+  } else if (isa<StructDecl>(D) || isa<ClassDecl>(D) || isa<ProtocolDecl>(D)) {
+    OS << "<Class";
+    RootEndTag = "</Class>";
   } else {
     OS << "<Other";
     RootEndTag = "</Other>";
diff --git a/test/PrintAsObjC/Inputs/comments-expected-output.h b/test/PrintAsObjC/Inputs/comments-expected-output.h
index eddbaa8208d..af01853fb38 100644
--- a/test/PrintAsObjC/Inputs/comments-expected-output.h
+++ b/test/PrintAsObjC/Inputs/comments-expected-output.h
@@ -19,6 +19,11 @@ SWIFT_CLASS("_TtC8comments21A010_AttachToEntities")
 @end
 
 
+SWIFT_PROTOCOL("_TtP8comments21A013_AttachToEntities_")
+@protocol A013_AttachToEntities
+@end
+
+
 SWIFT_CLASS("_TtC8comments18A100_EmptyComments")
 @interface A100_EmptyComments