mirrors/php-src
0
0
Fork 0
mirror of https://github.com/php/php-src.git synced 2025-01-20 10:43:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

- Added iterator_to_array() and iterator_count()

Browse Source
This commit is contained in:
Marcus Boerger 2004-10-07 23:08:19 +00:00
parent 940c5b393d
commit 6349f46e38
4 changed files with 105 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ext/spl/php_spl.c
View File

@ -45,6 +45,10 @@ function_entry spl_functions[] = {
PHP_FE(spl_classes, NULL)
PHP_FE(class_parents, NULL)
PHP_FE(class_implements, NULL)
#ifdef SPL_ITERATORS_H
PHP_FE(iterator_to_array, NULL)
PHP_FE(iterator_count, NULL)
#endif /* SPL_ITERATORS_H */
{NULL, NULL, NULL}
};
/* }}} */

64
ext/spl/spl_iterators.c
View File

@ -1322,6 +1322,70 @@ static zend_function_entry spl_funcs_CachingRecursiveIterator[] = {
{NULL, NULL, NULL}
};
/* {{{ array iterator_to_array(IteratorAggregate $it)
Copy the iterator into an array */
PHP_FUNCTION(iterator_to_array)
{
zval *obj, **data;
zend_object_iterator *iter;
char *str_key;
uint str_key_len;
ulong int_key;
int key_type;
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "O", &obj, zend_ce_aggregate) == FAILURE) {
RETURN_FALSE;
}
array_init(return_value);
iter = Z_OBJCE_P(obj)->get_iterator(Z_OBJCE_P(obj), obj TSRMLS_CC);
iter->funcs->rewind(iter TSRMLS_CC);
while (iter->funcs->valid(iter TSRMLS_CC) == SUCCESS) {
key_type = iter->funcs->get_current_key(iter, &str_key, &str_key_len, &int_key TSRMLS_CC);
iter->funcs->get_current_data(iter, &data TSRMLS_CC);
(*data)->refcount++;
switch(key_type) {
case HASH_KEY_IS_STRING:
add_assoc_zval_ex(return_value, str_key, str_key_len, *data);
efree(str_key);
break;
case HASH_KEY_IS_LONG:
add_index_zval(return_value, int_key, *data);
break;
}
iter->funcs->move_forward(iter TSRMLS_CC);
}
iter->funcs->dtor(iter TSRMLS_CC);
}
/* }}} */
/* {{{ int iterator_count(IteratorAggregate $it)
Count the elements in an iterator */
PHP_FUNCTION(iterator_count)
{
zval *obj;
zend_object_iterator *iter;
long count = 0;
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "O", &obj, zend_ce_aggregate) == FAILURE) {
RETURN_FALSE;
}
iter = Z_OBJCE_P(obj)->get_iterator(Z_OBJCE_P(obj), obj TSRMLS_CC);
iter->funcs->rewind(iter TSRMLS_CC);
while (iter->funcs->valid(iter TSRMLS_CC) == SUCCESS) {
count++;
iter->funcs->move_forward(iter TSRMLS_CC);
}
iter->funcs->dtor(iter TSRMLS_CC);
RETURN_LONG(count);
}
/* }}} */
/* {{{ PHP_MINIT_FUNCTION(spl_iterators)
*/
PHP_MINIT_FUNCTION(spl_iterators)

3
ext/spl/spl_iterators.h
View File

@ -35,6 +35,9 @@ extern zend_class_entry *spl_ce_CachingRecursiveIterator;
PHP_MINIT_FUNCTION(spl_iterators);
PHP_FUNCTION(iterator_to_array);
PHP_FUNCTION(iterator_count);
typedef enum {
DIT_Default = 0,
DIT_LimitIterator,

34
ext/spl/tests/spl_001.phpt Executable file
View File

@ -0,0 +1,34 @@
--TEST--
SPL: iterator_to_array() and iterator_count()
--FILE--
<?php
$it = new ArrayObject(array("x"=>1, 1=>2, 3=>3, 4, "1"=>5));
$ar = iterator_to_array($it);
var_dump(iterator_count($it));
print_r($ar);
foreach($ar as $v)
{
var_dump($v);
}
?>
===DONE===
--EXPECT--
int(4)
Array
(
[x] => 1
[1] => 5
[3] => 3
[4] => 4
)
int(1)
int(5)
int(3)
int(4)
===DONE===