Custom hydrator for Doctrine2, allows to access array elements using getters, similar to the ObjectHydrator

ArrayObjectHydrator.php

<?php
namespace My\Doctrine\Hydration;

use Doctrine\ORM\Internal\Hydration\ArrayHydrator,
    My\ArrayObject as ArrayObject;

/**
 * Custom Doctrine2 hydrator, allows to access array properties using getters like
 * in ObjectHydrator
 * 
 * 
 * @author Bernard Baltrusaitis <bernard@runawaylover.info> 
 */
class ArrayObjectHydrator extends ArrayHydrator
{
    protected function hydrateAllData()
    {
        $result = parent::hydrateAllData();
        
        return $this->toArrayObject($result);
    }
    
    protected function toArrayObject(array $result)
    {
        $return = new ArrayObject($result);
        
        foreach ($result as $k => $v) {
            if (is_array($v)) {
                $v = $this->toArrayObject($v);
            }
            
            $return[$k] = $v;
        }
        
        return $return;
    }    
}