check for sr end > num grid cells

spatialmodel · Jul 9, 2016 · 4e89c24 · 4e89c24
1 parent d4361c9
commit 4e89c24
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sr/sr.go b/sr/sr.go
@@ -132,8 +132,8 @@ func (sr *SR) Run(outfile string, layers []int, begin, end int) error {
 	for _, l := range layers {
 		layersMap[l] = Empty{}
 	}
-	if end < 0 {
-		end = len(sr.d.Cells())
+	if l := len(sr.d.Cells()); end < 0 || end > l {
+		end = l
 	}
 	for i := 0; i < len(sr.d.Cells()); i++ {