optional keys selection in it::map

1 files changed, 8 insertions, 4 deletions

diff --git a/it.class b/it.class
index a9b6662..1731330 100644
--- a/it.class
+++ b/it.class
@@ -950,18 +950,22 @@ static function date($format = "", $stamp = null)
  * Iterate over an array, replacing every element by function value or expression
  * @param $code The function or expression to apply. Expression may contain $k for keys and $v for values
  * @param $array The array to iterate over
+ * @param $p['keys'] Only modify elements with given keys (keys can be array or comma separated)
  */
-static function map($code, $array)
+static function map($code, $array, $p = null)
 {
 	static $cache = array();
 
+	if (is_string($code) && it::match('^[\w:]+$', $code))
+		$code .= '($v)';
+
 	if (!is_callable($func = $code) && !($func = $cache[$code]))
 		$func = $cache[$code] = create_function('$k,$v', "return $code;");
 
-	foreach ($array as $k => $v)
-		$result[$k] = $func($k, $v);
+	foreach ($p['keys'] ? it::filter_keys($array, $p['keys']) : $array as $k => $v)
+		$array[$k] = $func($k, $v);
 
-	return (array)$result;
+	return (array)$array;
 }
 
 /**