[find-all-symbols] Implement operator== for SymbolAndSignals and SymbolInfo::Signals.

llvm-svn: 296461

2 files changed, 10 insertions, 0 deletions

diff --git a/clang-tools-extra/include-fixer/find-all-symbols/SymbolInfo.cpp b/clang-tools-extra/include-fixer/find-all-symbols/SymbolInfo.cpp
index db9327a6211..7194f2e24fd 100644
--- a/clang-tools-extra/include-fixer/find-all-symbols/SymbolInfo.cpp
+++ b/clang-tools-extra/include-fixer/find-all-symbols/SymbolInfo.cpp
@@ -113,6 +113,14 @@ SymbolInfo::Signals SymbolInfo::Signals::operator+(const Signals &RHS) const {
   return Result;
 }
 
+bool SymbolInfo::Signals::operator==(const Signals &RHS) const {
+  return std::tie(Seen, Used) == std::tie(RHS.Seen, RHS.Used);
+}
+
+bool SymbolAndSignals::operator==(const SymbolAndSignals& RHS) const {
+  return std::tie(Symbol, Signals) == std::tie(RHS.Symbol, RHS.Signals);
+}
+
 bool WriteSymbolInfosToStream(llvm::raw_ostream &OS,
                               const SymbolInfo::SignalMap &Symbols) {
   llvm::yaml::Output yout(OS);
diff --git a/clang-tools-extra/include-fixer/find-all-symbols/SymbolInfo.h b/clang-tools-extra/include-fixer/find-all-symbols/SymbolInfo.h
index 698e722521e..e2960bfe20a 100644
--- a/clang-tools-extra/include-fixer/find-all-symbols/SymbolInfo.h
+++ b/clang-tools-extra/include-fixer/find-all-symbols/SymbolInfo.h
@@ -59,6 +59,7 @@ public:
 
     Signals &operator+=(const Signals &RHS);
     Signals operator+(const Signals &RHS) const;
+    bool operator==(const Signals &RHS) const;
   };
 
   using SignalMap = std::map<SymbolInfo, Signals>;
@@ -128,6 +129,7 @@ private:
 struct SymbolAndSignals {
   SymbolInfo Symbol;
   SymbolInfo::Signals Signals;
+  bool operator==(const SymbolAndSignals& RHS) const;
 };
 
 /// \brief Write SymbolInfos to a stream (YAML format).