- Implemented automatic label setting from datasets for labeled_axis

1 files changed, 44 insertions, 1 deletions

diff --git a/src/datasets/base.php b/src/datasets/base.php
index 2a28116..7cda08d 100644
--- a/src/datasets/base.php
+++ b/src/datasets/base.php
@@ -12,7 +12,7 @@
  *
  * @package Graph
  */
-class ezcGraphDataset implements ArrayAccess
+class ezcGraphDataset implements ArrayAccess, Iterator
 {
 
     protected $label;
@@ -23,6 +23,8 @@ class ezcGraphDataset implements ArrayAccess
 
     protected $data;
 
+    protected $current;
+
     public function __construct()
     {
         $this->label = new ezcGraphDatasetStringProperty( $this );
@@ -136,6 +138,47 @@ class ezcGraphDataset implements ArrayAccess
     {
         unset( $this->data[$key] );
     }
+
+    final public function current()
+    {
+        $keys = array_keys( $this->data );
+        if ( !isset( $this->current ) )
+        {
+            $this->current = 0;
+        }
+
+        return $this->data[$keys[$this->current]];
+    }
+
+    final public function next()
+    {
+        $keys = array_keys( $this->data );
+        if ( ++$this->current >= count( $keys ) )
+        {
+            return false;
+        }
+        else 
+        {
+            return $this->data[$keys[$this->current]];
+        }
+    }
+
+    final public function key()
+    {
+        $keys = array_keys( $this->data );
+        return $keys[$this->current];
+    }
+
+    final public function valid()
+    {
+        $keys = array_keys( $this->data );
+        return isset( $keys[$this->current] );
+    }
+
+    final public function rewind()
+    {
+        $this->current = 0;
+    }
 }
 
 ?>