Added count(Enum)

pmmp · Jul 21, 2014 · da884fe · da884fe
1 parent 1772592
commit da884fe
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/tag/Enum.php b/tag/Enum.php
@@ -24,7 +24,7 @@
 use pocketmine\nbt\NBT;
 use pocketmine\nbt\tag\Enum as TagEnum;
 
-class Enum extends NamedTag implements \ArrayAccess{
+class Enum extends NamedTag implements \ArrayAccess, \Countable{
 
 	private $tagType;
 
@@ -73,6 +73,19 @@ public function offsetUnset($offset){
 		unset($this->{$offset});
 	}
 
+	public function count($mode = COUNT_NORMAL){
+		for($i = 0; true; $i++){
+			if(!isset($this->{$i})){
+				return $i;
+			}
+			if($mode === COUNT_RECURSIVE){
+				if($this->{$i} instanceof \Countable){
+					$i += count($this->{$i});
+				}
+			}
+		}
+	}
+
 	public function getType(){
 		return NBT::TAG_Enum;
 	}
@@ -178,4 +191,4 @@ public function write(NBT $nbt){
 			$tag->write($nbt);
 		}
 	}
-}
+}