X-Git-Url: http://git.shiar.net/perl/list-index.git/blobdiff_plain/2af0622fdc633320a7172164ddb9c7436e41998f..12c2437bb737f40da7d7514fc31c641ef0110cf3:/lib/List/Index.pm

diff --git a/lib/List/Index.pm b/lib/List/Index.pm
index 4ce9c25..f703d97 100644
--- a/lib/List/Index.pm
+++ b/lib/List/Index.pm
@@ -22,17 +22,17 @@ sub ranges {
 	my $length   = $options->{length  } || 4;
 	my $pages    = $options->{pages   } || 1 + int $#$self / $pagesize;
 
-	$pagesize = @$self / $pages;
-	my $offset = $pagesize + .5;
-	my $shrunk = 0;
-	my $enlarged = 0;
+	$pagesize = $pages >= $#$self ? 1 : @$self / $pages;
+	my $lookbehind = -$context;
+	my $lookahead  =  $context;
+
 	my @links = ('');
-	while ($offset < @$self) {
+	for (my $offset = $pagesize + .5; $offset < @$self; $offset += $pagesize) {
 		my $link = substr $self->[$offset], 0, $length;
 		if ($context) {
 			my $penalty = 0;
 			# take a value slightly before the current offset
-			if ((my $before = $offset - $context + $shrunk) > 0) {
+			if ((my $before = $offset + $lookbehind) > 0) {
 				# see how much of it matches the current link
 				my $trim = 1;
 				for my $match (split //, $self->[$before - 1]) {
@@ -49,9 +49,10 @@ sub ranges {
 				}
 			}
 
-			$shrunk = 0;
+			$lookbehind = -$context;
+
 			# take a value after the current offset
-			if ((my $after = $offset + $context - $enlarged) < $#$self) {
+			if ((my $after = $offset + $lookahead) < $#$self) {
 				# see how much of it matches the current link
 				my $trim = 1;
 				for my $match (split //, $self->[$after]) {
@@ -63,17 +64,17 @@ sub ranges {
 					$link = substr $self->[$after], 0, $trim;
 					# advance lookbehind offset on the next page
 					$penalty = 0;
-					for ($offset .. $#$self) {
-						last if $self->[$offset + $shrunk] =~ /^\Q$link/;
-						$shrunk++;
+					for ($offset .. $after) {
+						last if $self->[$_] =~ /^\Q$link/;
+						$lookbehind++;
 					}
 				}
 			}
-			$enlarged = $penalty;
+
+			$lookahead = $context - $penalty;
 		}
 
 		push @links, $link;
-		$offset += $pagesize;
 	}
 
 	use List::MoreUtils 'uniq';