Protect Child from returning Node of nil pointer

arl · Oct 21, 2016 · 267a755 · 267a755
1 parent 8c30015
commit 267a755
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/basicnode.go b/basicnode.go
@@ -24,6 +24,9 @@ func (n *basicNode) Parent() Node {
 
 // Child returns current node child at specified quadrant.
 func (n *basicNode) Child(q Quadrant) Node {
+	if n.c[q] == nil {
+		return nil
+	}
 	return n.c[q]
 }
 

diff --git a/cnnode.go b/cnnode.go
@@ -50,6 +50,9 @@ func (n *cnNode) Parent() Node {
 
 // Child returns current node child at specified quadrant.
 func (n *cnNode) Child(q Quadrant) Node {
+	if n.c[q] == nil {
+		return nil
+	}
 	return n.c[q]
 }