Finding the closest locations with GKRTree in Swift

I have recently faced a problem where I had to efficiently look up locations that are geographically close to a specified point. As the naive approach, including computing a distance between dozens of point pairs, seem not so efficient to me — I made a little research and gave a try to Apple-provided R-tree implementation from GameKit.

This story is a quick dive into R-tree basics as well as features a short Apple-provided GKRTree usage example.

R-tree basics

R-tree is a data structure that has a wide application for indexing spatial data. The concept was first proposed by Antonin Guttman in 1984. The key idea behind R-tree is grouping nearby objects and representing them with their minimum bounding rectangles or boxes. The minimum bounding rectangle, often referred to as MBR, is the smallest rectangle that contains a given group of objects. The MBRs of higher-level group MBRs of child nodes. That makes searches in the tree efficient as the query that does not intersect the bounding rectangle on a higher level also cannot intersect any of the contained objects MBRs. That makes filtering out non-matching elements faster.

To have a better image of how the data are structured, let’s have a short look at that diagram from Wikipedia. We can see top-level, black marked regions that cover the biggest parts of the surface and contain smaller blue areas. We can observe that when searching for an element lying inside R15 box we can filter out all R1 children in just one comparison significantly reducing the amount of computation needed for a search operation.

R-Tree example diagram — Wikipedia

R-tree is also a balanced search tree, which means that all leaves are on the same level. Additionally, that structure is also optimized for disk storage, so it found wide application in the database field.

R-trees can be useful for many tasks:

  • Finding POI’s that are within a certain distance from a given location
  • Deciding which game objects are close enough to trigger interactions
  • Searching closest nodes in the coordinates grid
  • And many more…

R-tree in Swift

GKRTree is an implementation of R-tree concept provided by Apple in GameKit. As I am rather a type of person that relies on deeply-tested and battle-proven implementations, so rather than implementing one myself, I gave it a try in my use case.

Let’s assume that our data are locations represented by the following class. It’s required for an object being GKRTree element to be NSObject’s subclasses.

For creating an instance of GKRTree it’s needed to specify maximum capacity. Let’s then specify a couple of Location instances and create a GKRTree object.

Our points may be drawn using the below diagram, which may be helpful to visualise the spatial relationships between them.

Elements visualised in coordinate grid

Once we have an R-tree instance and elements created, it’s time to put them into the actual structure. The API for that looks as follow

It takes an element with its bounding box minimum and maximum coordinates. As in our scenario, we are using points as locations we need to somehow translate them into bounding boxes that are being accepted by the GKRTree API. We can achieve that by simply adding some padding to their coordinates. We can do it by adding computed variables to the SomeLocation class.

The paddings we have just added can be now visualized on the diagram.

Elements with their bounding boxes

We can now add the elements to the tree.

The last parameter used when adding elements is the splitStrategy. It defines how the tree reorganizes its internal structure. Four different strategies can be applied. The overall result depends on the organization of data you process, so it's valuable to check different strategies and compare the performance to pick the most efficient one.

Now, it’s time to search the closest neighbors for a selected point. Let’s assume that it will be at (1,2). Looking at the GKRTree’s search method we can see that it accepts, similarly to the one used for adding elements, bounding box boundaries.

That leads us to specify the area in which we want to find points.

Search point location

The search bounding box can be added to our diagram as well. As we can see below — we should expect getting one location back as there is only one element falling into the bounding box we just defined.

We can also observe that modifying the search criteria by increasing the box size gives us more results which can be later sorted, filtered and processed depending on the actual use case.

While using R-tree it’s worth being aware that this data structure operates in a way that keeps the tree balanced. It means that no branch of the R-tree contains significantly more objects or sub-branches than any other branch. It results in increasing the amount of time required to perform insertion and deletion operations while decreasing the time needed to search for elements.

Complete example

The complete code for the discussed example looks as follows.

Wrap up

R-trees are a widely-applied data structure that may be used for implementing various sorts of spatial data searches. For Apple’s platforms, there is a built-in implementation provided in the GameKit framework that makes implementing search on spatial data really quick. If you happen to need a 3D equivalent, be sure to check GKOctree.

I hope that the presented example was a useful base for implementing your own solutions.

Thanks for reading!

Originally published at https://matrejek.eu.

Senior Software Engineer @ Pegasystems

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store