Make range() function smarter

rlerdorf · rlerdorf · commit 45d71e2951ed · 2001-07-09T20:36:47.000Z
@ - Improve range() function to support range('a','z') and range(9,0)
@   types of ranges.  (Rasmus)
diff --git a/ext/standard/array.c b/ext/standard/array.c
@@ -1334,29 +1334,52 @@ PHP_FUNCTION(compact)
 }
 /* }}} */
 
-/* {{{ proto array range(int low, int high)
-   Create an array containing the range of integers from low to high (inclusive) */
+/* {{{ proto array range(mixed low, mixed high)
+   Create an array containing the range of integers or characters from low to high (inclusive) */
 PHP_FUNCTION(range)
 {
 	zval **zlow, **zhigh;
-	int low, high;
 	
 	if (ZEND_NUM_ARGS() != 2 || zend_get_parameters_ex(2,&zlow,&zhigh) == FAILURE) {
 		WRONG_PARAM_COUNT;
 	}
-	convert_to_long_ex(zlow);
-	convert_to_long_ex(zhigh);
-	low = Z_LVAL_PP(zlow);
-	high = Z_LVAL_PP(zhigh);
 
     /* allocate an array for return */
     if (array_init(return_value) == FAILURE) {
 		RETURN_FALSE;
     }
 
-	for (; low <= high; low++) {
-		add_next_index_long(return_value, low);
-	}	
+	if(Z_TYPE_PP(zlow)==IS_STRING && Z_TYPE_PP(zhigh)==IS_STRING) {
+		char *low, *high;
+		convert_to_string_ex(zlow);
+		convert_to_string_ex(zhigh);
+		low = Z_STRVAL_PP(zlow);
+		high = Z_STRVAL_PP(zhigh);
+		if(*low>*high) {
+			for (; *low >= *high; (*low)--) {
+				add_next_index_stringl(return_value, low, 1, 1);
+			}	
+		} else {
+			for (; *low <= *high; (*low)++) {
+				add_next_index_stringl(return_value, low, 1, 1);
+			}	
+		}
+	} else {
+		int low, high;
+		convert_to_long_ex(zlow);
+		convert_to_long_ex(zhigh);
+		low = Z_LVAL_PP(zlow);
+		high = Z_LVAL_PP(zhigh);
+		if(low>high) { 
+			for (; low >= high; low--) {
+				add_next_index_long(return_value, low);
+			}	
+		} else {
+			for (; low <= high; low++) {
+				add_next_index_long(return_value, low);
+			}	
+		}
+	}
 }
 /* }}} */
 
