Merge pull request #18417 from mhong/denseinfo_source_range

Author: swift-ci

include/swift/Basic/SourceLoc.h

@@ -18,6 +18,7 @@
 #define SWIFT_BASIC_SOURCELOC_H
 
 #include "swift/Basic/LLVM.h"
+#include "llvm/ADT/DenseMapInfo.h"
 #include "llvm/ADT/StringRef.h"
 #include "llvm/Support/SMLoc.h"
 #include <functional>
@@ -214,4 +215,58 @@ class CharSourceRange {
 
 } // end namespace swift
 
+namespace llvm {
+template <typename T> struct DenseMapInfo;
+
+template <> struct DenseMapInfo<swift::SourceLoc> {
+  static swift::SourceLoc getEmptyKey() {
+    return swift::SourceLoc(
+        SMLoc::getFromPointer(DenseMapInfo<const char *>::getEmptyKey()));
+  }
+
+  static swift::SourceLoc getTombstoneKey() {
+    // Make this different from empty key. See for context:
+    // http://lists.llvm.org/pipermail/llvm-dev/2015-July/088744.html
+    return swift::SourceLoc(
+        SMLoc::getFromPointer(DenseMapInfo<const char *>::getTombstoneKey()));
+  }
+
+  static unsigned getHashValue(const swift::SourceLoc &Val) {
+    return DenseMapInfo<const void *>::getHashValue(
+        Val.getOpaquePointerValue());
+  }
+
+  static bool isEqual(const swift::SourceLoc &LHS,
+                      const swift::SourceLoc &RHS) {
+    return LHS == RHS;
+  }
+};
+
+template <> struct DenseMapInfo<swift::SourceRange> {
+  static swift::SourceRange getEmptyKey() {
+    return swift::SourceRange(swift::SourceLoc(
+        SMLoc::getFromPointer(DenseMapInfo<const char *>::getEmptyKey())));
+  }
+
+  static swift::SourceRange getTombstoneKey() {
+    // Make this different from empty key. See for context:
+    // http://lists.llvm.org/pipermail/llvm-dev/2015-July/088744.html
+    return swift::SourceRange(swift::SourceLoc(
+        SMLoc::getFromPointer(DenseMapInfo<const char *>::getTombstoneKey())));
+  }
+
+  static unsigned getHashValue(const swift::SourceRange &Val) {
+    return hash_combine(DenseMapInfo<const void *>::getHashValue(
+                            Val.Start.getOpaquePointerValue()),
+                        DenseMapInfo<const void *>::getHashValue(
+                            Val.End.getOpaquePointerValue()));
+  }
+
+  static bool isEqual(const swift::SourceRange &LHS,
+                      const swift::SourceRange &RHS) {
+    return LHS == RHS;
+  }
+};
+} // namespace llvm
+
 #endif // SWIFT_BASIC_SOURCELOC_H