RequirementMachine: Add ProtocolGraph::isKnownProtocol()

Author: slavapestov

lib/AST/RequirementMachine/ProtocolGraph.cpp

@@ -28,6 +28,11 @@ void ProtocolGraph::visitRequirements(ArrayRef<Requirement> reqs) {
   }
 }
 
+/// Return true if we know about this protocol.
+bool ProtocolGraph::isKnownProtocol(const ProtocolDecl *proto) const {
+  return Info.count(proto) > 0;
+}
+
 /// Look up information about a known protocol.
 const ProtocolInfo &ProtocolGraph::getProtocolInfo(
     const ProtocolDecl *proto) const {

lib/AST/RequirementMachine/ProtocolGraph.h

@@ -83,6 +83,8 @@ struct ProtocolGraph {
 
   void visitRequirements(ArrayRef<Requirement> reqs);
 
+  bool isKnownProtocol(const ProtocolDecl *proto) const;
+
   const ProtocolInfo &getProtocolInfo(
       const ProtocolDecl *proto) const;