| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153 |
- <?php
- /*
- * This file is part of the Recursion Context package.
- *
- * (c) Sebastian Bergmann <sebastian@phpunit.de>
- *
- * For the full copyright and license information, please view the LICENSE
- * file that was distributed with this source code.
- */
- namespace SebastianBergmann\RecursionContext;
- /**
- * A context containing previously processed arrays and objects
- * when recursively processing a value.
- */
- final class Context
- {
- /**
- * @var array[]
- */
- private $arrays;
- /**
- * @var \SplObjectStorage
- */
- private $objects;
- /**
- * Initialises the context
- */
- public function __construct()
- {
- $this->arrays = array();
- $this->objects = new \SplObjectStorage;
- }
- /**
- * Adds a value to the context.
- *
- * @param array|object $value The value to add.
- * @return int|string The ID of the stored value, either as
- * a string or integer.
- * @throws InvalidArgumentException Thrown if $value is not an array or
- * object
- */
- public function add(&$value)
- {
- if (is_array($value)) {
- return $this->addArray($value);
- }
- else if (is_object($value)) {
- return $this->addObject($value);
- }
- throw new InvalidArgumentException(
- 'Only arrays and objects are supported'
- );
- }
- /**
- * Checks if the given value exists within the context.
- *
- * @param array|object $value The value to check.
- * @return int|string|false The string or integer ID of the stored
- * value if it has already been seen, or
- * false if the value is not stored.
- * @throws InvalidArgumentException Thrown if $value is not an array or
- * object
- */
- public function contains(&$value)
- {
- if (is_array($value)) {
- return $this->containsArray($value);
- }
- else if (is_object($value)) {
- return $this->containsObject($value);
- }
- throw new InvalidArgumentException(
- 'Only arrays and objects are supported'
- );
- }
- /**
- * @param array $array
- * @return bool|int
- */
- private function addArray(array &$array)
- {
- $key = $this->containsArray($array);
- if ($key !== false) {
- return $key;
- }
- $this->arrays[] = &$array;
- return count($this->arrays) - 1;
- }
- /**
- * @param object $object
- * @return string
- */
- private function addObject($object)
- {
- if (!$this->objects->contains($object)) {
- $this->objects->attach($object);
- }
- return spl_object_hash($object);
- }
- /**
- * @param array $array
- * @return int|false
- */
- private function containsArray(array &$array)
- {
- $keys = array_keys($this->arrays, $array, true);
- $hash = '_Key_' . microtime(true);
- foreach ($keys as $key) {
- $this->arrays[$key][$hash] = $hash;
- if (isset($array[$hash]) && $array[$hash] === $hash) {
- unset($this->arrays[$key][$hash]);
- return $key;
- }
- unset($this->arrays[$key][$hash]);
- }
- return false;
- }
- /**
- * @param object $value
- * @return string|false
- */
- private function containsObject($value)
- {
- if ($this->objects->contains($value)) {
- return spl_object_hash($value);
- }
- return false;
- }
- }
|