Add `zip()`.

This commit is contained in:
CismonX 2019-01-18 21:37:42 +08:00
parent fddc87e95d
commit cbc03da7fb
2 changed files with 64 additions and 2 deletions

View File

@ -3325,7 +3325,49 @@ PHP_METHOD(Collection, windowed)
PHP_METHOD(Collection, zip)
{
zval* elements;
zend_fcall_info fci;
zend_fcall_info_cache fcc;
ZEND_PARSE_PARAMETERS_START(1, 2)
Z_PARAM_ZVAL(elements)
Z_PARAM_OPTIONAL
Z_PARAM_FUNC(fci, fcc)
ZEND_PARSE_PARAMETERS_END();
ELEMENTS_VALIDATE(elements, ERR_BAD_ARGUMENT_TYPE, return);
zend_array* current = THIS_COLLECTION;
if (UNEXPECTED(!HT_IS_PACKED(current)) || UNEXPECTED(!HT_IS_PACKED(elements_arr))) {
ERR_NOT_PACKED();
RETURN_NULL();
}
zend_bool has_transform = EX_NUM_ARGS() > 1;
INIT_FCI(&fci, 2);
uint32_t num_elements = zend_hash_num_elements(current);
uint32_t num_other = zend_hash_num_elements(elements_arr);
if (num_other < num_elements) {
num_elements = num_other;
}
ARRAY_NEW(zipped, num_elements);
Bucket* bucket_this = current->arData;
Bucket* bucket_other = elements_arr->arData;
uint32_t idx;
for (idx = 0; idx < num_elements; ++idx) {
if (has_transform) {
ZVAL_COPY_VALUE(&params[0], &(bucket_this + idx)->val);
ZVAL_COPY_VALUE(&params[1], &(bucket_other + idx)->val);
zend_call_function(&fci, &fcc);
} else {
zend_object* obj = create_pair_obj();
Z_TRY_ADDREF((bucket_this + idx)->val);
Z_TRY_ADDREF((bucket_other + idx)->val);
pair_update_first(obj, &(bucket_this + idx)->val);
pair_update_second(obj, &(bucket_other + idx)->val);
zval pair;
ZVAL_OBJ(&pair, obj);
ZVAL_COPY_VALUE(&retval, &pair);
}
zend_hash_next_index_insert(zipped, &retval);
}
RETVAL_NEW_COLLECTION(zipped);
}
PHP_METHOD(Collection, zipWithNext)
@ -3337,7 +3379,7 @@ PHP_METHOD(Collection, zipWithNext)
Z_PARAM_FUNC(fci, fcc)
ZEND_PARSE_PARAMETERS_END();
zend_array* current = THIS_COLLECTION;
if (!HT_IS_PACKED(current)) {
if (UNEXPECTED(!HT_IS_PACKED(current))) {
ERR_NOT_PACKED();
RETURN_NULL();
}

20
tests/061-zip.phpt Normal file
View File

@ -0,0 +1,20 @@
--TEST--
Test Collection::zip().
--FILE--
<?php
$array = ['a', 'b', 'c'];
$array1 = ['d', 'e'];
$zipped = Collection::init($array)->zip($array1)->toArray();
if ($zipped != [new Pair('a', 'd'), new Pair('b', 'e')]) {
echo 'Collection::zip() failed.', PHP_EOL;
}
$transform = function ($v1, $v2) {
return $v1.$v2;
};
$zipped = Collection::init($array1)->zip($array, $transform)->toArray();
if ($zipped != ['da', 'eb']) {
echo 'Collection::zip() failed.', PHP_EOL;
}
?>
--EXPECT--