RequirementMachine: Implement RewriteSystem::getLocalRules()

Author: slavapestov

lib/AST/RequirementMachine/RequirementMachine.cpp

@@ -335,6 +335,10 @@ std::string RequirementMachine::getRuleAsStringForDiagnostics(
   return out.str();
 }
 
+ArrayRef<Rule> RequirementMachine::getLocalRules() const {
+  return System.getLocalRules();
+}
+
 bool RequirementMachine::isComplete() const {
   return Complete;
 }

lib/AST/RequirementMachine/RequirementMachine.h

@@ -158,6 +158,8 @@ class RequirementMachine final {
   std::vector<Requirement>
   computeMinimalGenericSignatureRequirements(bool reconstituteSugar);
 
+  ArrayRef<Rule> getLocalRules() const;
+
   std::string getRuleAsStringForDiagnostics(unsigned ruleID) const;
 
   GenericSignatureErrors getErrors() const;

lib/AST/RequirementMachine/RewriteSystem.h

@@ -144,10 +144,19 @@ class RewriteSystem final {
     return (unsigned)(&rule - &*Rules.begin());
   }
 
+  /// Get an array of all rewrite rules.
   ArrayRef<Rule> getRules() const {
     return Rules;
   }
 
+  /// Get an array of rewrite rules, not including rewrite rules imported
+  /// from referenced protocols.
+  ArrayRef<Rule> getLocalRules() const {
+    return getRules().slice(FirstLocalRule);
+  }
+
+  /// Get the rewrite rule at the given index. Note that this is an index
+  /// into getRules(), *NOT* getLocalRules().
   Rule &getRule(unsigned ruleID) {
     return Rules[ruleID];
   }