Implement Countable and IteratorAggregate in *Collection classes

Author: sebastianbergmann

src/CodeCoverage/Data/BranchCollection.php

@@ -54,7 +54,7 @@
  * @link       http://github.com/sebastianbergmann/php-code-coverage
  * @since      Class available since Release 2.1.0
  */
-class PHP_CodeCoverage_Data_BranchCollection
+class PHP_CodeCoverage_Data_BranchCollection implements Countable, IteratorAggregate
 {
     /**
      * @var PHP_CodeCoverage_Data_Branch[]
@@ -68,4 +68,20 @@ public function addBranch(PHP_CodeCoverage_Data_Branch $branch)
     {
         $this->branches[] = $branch;
     }
+
+    /**
+     * @return integer
+     */
+    public function count()
+    {
+        return count($this->branches);
+    }
+
+    /**
+     * @return ArrayIterator
+     */
+    public function getIterator()
+    {
+        return new ArrayIterator($this->branches);
+    }
 }

src/CodeCoverage/Data/FileCollection.php

@@ -54,7 +54,7 @@
  * @link       http://github.com/sebastianbergmann/php-code-coverage
  * @since      Class available since Release 2.1.0
  */
-class PHP_CodeCoverage_Data_FileCollection
+class PHP_CodeCoverage_Data_FileCollection implements Countable, IteratorAggregate
 {
     /**
      * @var PHP_CodeCoverage_Data_File[]
@@ -130,4 +130,20 @@ private function createFile($path, array $lines)
 
         $this->files[$path] = new PHP_CodeCoverage_Data_File($path, $functions, $_lines);
     }
+
+    /**
+     * @return integer
+     */
+    public function count()
+    {
+        return count($this->files);
+    }
+
+    /**
+     * @return ArrayIterator
+     */
+    public function getIterator()
+    {
+        return new ArrayIterator($this->files);
+    }
 }

src/CodeCoverage/Data/FunctionCollection.php

@@ -54,7 +54,7 @@
  * @link       http://github.com/sebastianbergmann/php-code-coverage
  * @since      Class available since Release 2.1.0
  */
-class PHP_CodeCoverage_Data_FunctionCollection
+class PHP_CodeCoverage_Data_FunctionCollection implements Countable, IteratorAggregate
 {
     /**
      * @var PHP_CodeCoverage_Data_Function[]
@@ -68,4 +68,20 @@ public function addFunction(PHP_CodeCoverage_Data_Function $function)
     {
         $this->functions[] = $function;
     }
+
+    /**
+     * @return integer
+     */
+    public function count()
+    {
+        return count($this->functions);
+    }
+
+    /**
+     * @return ArrayIterator
+     */
+    public function getIterator()
+    {
+        return new ArrayIterator($this->functions);
+    }
 }

src/CodeCoverage/Data/LineCollection.php

@@ -54,7 +54,7 @@
  * @link       http://github.com/sebastianbergmann/php-code-coverage
  * @since      Class available since Release 2.1.0
  */
-class PHP_CodeCoverage_Data_LineCollection
+class PHP_CodeCoverage_Data_LineCollection implements Countable, IteratorAggregate
 {
     /**
      * @var PHP_CodeCoverage_Data_Line[]
@@ -80,4 +80,20 @@ public function addCoveringTest($id, array $lines)
             $this->lines[$line]->addCoveringTest($id);
         }
     }
+
+    /**
+     * @return integer
+     */
+    public function count()
+    {
+        return count($this->lines);
+    }
+
+    /**
+     * @return ArrayIterator
+     */
+    public function getIterator()
+    {
+        return new ArrayIterator($this->lines);
+    }
 }

src/CodeCoverage/Data/OpcodeCollection.php

@@ -54,7 +54,7 @@
  * @link       http://github.com/sebastianbergmann/php-code-coverage
  * @since      Class available since Release 2.1.0
  */
-class PHP_CodeCoverage_Data_OpcodeCollection
+class PHP_CodeCoverage_Data_OpcodeCollection implements Countable, IteratorAggregate
 {
     /**
      * @var PHP_CodeCoverage_Data_Opcode[]
@@ -68,4 +68,20 @@ public function addOpcode(PHP_CodeCoverage_Data_Opcode $opcode)
     {
         $this->opcodes[] = $opcode;
     }
+
+    /**
+     * @return integer
+     */
+    public function count()
+    {
+        return count($this->opcodes);
+    }
+
+    /**
+     * @return ArrayIterator
+     */
+    public function getIterator()
+    {
+        return new ArrayIterator($this->opcodes);
+    }
 }

src/CodeCoverage/Data/PathCollection.php

@@ -54,7 +54,7 @@
  * @link       http://github.com/sebastianbergmann/php-code-coverage
  * @since      Class available since Release 2.1.0
  */
-class PHP_CodeCoverage_Data_PathCollection
+class PHP_CodeCoverage_Data_PathCollection implements Countable, IteratorAggregate
 {
     /**
      * @var PHP_CodeCoverage_Data_Path[]
@@ -68,4 +68,20 @@ public function addPath(PHP_CodeCoverage_Data_Path $path)
     {
         $this->paths[] = $path;
     }
+
+    /**
+     * @return integer
+     */
+    public function count()
+    {
+        return count($this->paths);
+    }
+
+    /**
+     * @return ArrayIterator
+     */
+    public function getIterator()
+    {
+        return new ArrayIterator($this->paths);
+    }
 }