diff --git a/bindings/python/clang/cindex.py b/bindings/python/clang/cindex.py index 48a107cd33..474740a64d 100644 --- a/bindings/python/clang/cindex.py +++ b/bindings/python/clang/cindex.py @@ -146,6 +146,12 @@ class SourceLocation(Structure): """Get the file offset represented by this source location.""" return self._get_instantiation()[3] + def __eq__(self, other): + return SourceLocation_equalLocations(self, other) + + def __ne__(self, other): + return not self.__eq__(other) + def __repr__(self): if self.file: filename = self.file.name @@ -186,6 +192,12 @@ class SourceRange(Structure): """ return SourceRange_end(self) + def __eq__(self, other): + return SourceRange_equalRanges(self, other) + + def __ne__(self, other): + return not self.__eq__(other) + def __repr__(self): return "" % (self.start, self.end) @@ -1613,6 +1625,10 @@ SourceLocation_getLocation = lib.clang_getLocation SourceLocation_getLocation.argtypes = [TranslationUnit, File, c_uint, c_uint] SourceLocation_getLocation.restype = SourceLocation +SourceLocation_equalLocations = lib.clang_equalLocations +SourceLocation_equalLocations.argtypes = [SourceLocation, SourceLocation] +SourceLocation_equalLocations.restype = bool + # Source Range Functions SourceRange_getRange = lib.clang_getRange SourceRange_getRange.argtypes = [SourceLocation, SourceLocation] @@ -1626,6 +1642,10 @@ SourceRange_end = lib.clang_getRangeEnd SourceRange_end.argtypes = [SourceRange] SourceRange_end.restype = SourceLocation +SourceRange_equalRanges = lib.clang_equalRanges +SourceRange_equalRanges.argtypes = [SourceRange, SourceRange] +SourceRange_equalRanges.restype = bool + # CursorKind Functions CursorKind_is_decl = lib.clang_isDeclaration CursorKind_is_decl.argtypes = [CursorKind] diff --git a/bindings/python/tests/cindex/test_location.py b/bindings/python/tests/cindex/test_location.py index 300136f0cd..1707f01aea 100644 --- a/bindings/python/tests/cindex/test_location.py +++ b/bindings/python/tests/cindex/test_location.py @@ -1,4 +1,4 @@ -from clang.cindex import Index, File, SourceLocation, Cursor +from clang.cindex import Index, File, SourceLocation, SourceRange, Cursor baseInput="int one;\nint two;\n" @@ -47,6 +47,12 @@ def test_location(): if n.spelling == 'one': assert n == cursor + # Ensure locations referring to the same entity are equivalent. + location2 = SourceLocation.from_position(tu, file, 1, 5) + assert location == location2 + location3 = SourceLocation.from_position(tu, file, 1, 4) + assert location2 != location3 + def test_extent(): index = Index.create() tu = index.parse('t.c', unsaved_files = [('t.c',baseInput)]) @@ -60,3 +66,15 @@ def test_extent(): assert_location(n.extent.start,line=2,column=1,offset=9) assert_location(n.extent.end,line=2,column=8,offset=16) assert baseInput[n.extent.start.offset:n.extent.end.offset] == "int two" + + file = File.from_name(tu, 't.c') + location1 = SourceLocation.from_position(tu, file, 1, 1) + location2 = SourceLocation.from_position(tu, file, 1, 8) + + range1 = SourceRange.from_locations(location1, location2) + range2 = SourceRange.from_locations(location1, location2) + assert range1 == range2 + + location3 = SourceLocation.from_position(tu, file, 1, 6) + range3 = SourceRange.from_locations(location1, location3) + assert range1 != range3