<?php

/*
 *
 *  ____            _        _   __  __ _                  __  __ ____
 * |  _ \ ___   ___| | _____| |_|  \/  (_)_ __   ___      |  \/  |  _ \
 * | |_) / _ \ / __| |/ / _ \ __| |\/| | | '_ \ / _ \_____| |\/| | |_) |
 * |  __/ (_) | (__|   <  __/ |_| |  | | | | | |  __/_____| |  | |  __/
 * |_|   \___/ \___|_|\_\___|\__|_|  |_|_|_| |_|\___|     |_|  |_|_|
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * @author PocketMine Team
 * @link http://www.pocketmine.net/
 *
 *
*/

namespace pocketmine\level\format\leveldb;

use pocketmine\level\format\generic\BaseFullChunk;
use pocketmine\level\format\LevelProvider;
use pocketmine\nbt\NBT;
use pocketmine\Player;
use pocketmine\utils\Binary;

class Chunk extends BaseFullChunk{

	const DATA_LENGTH = 16384 * (2 + 1 + 1 + 1) + 256 + 1024;

	protected $isLightPopulated = \false;
	protected $isPopulated = \false;
	protected $isGenerated = \false;

	public function __construct($level, $chunkX, $chunkZ, $terrain, array $entityData = \null, array $tileData = \null){
		$offset = 0;

		$blocks = \substr($terrain, $offset, 32768);
		$offset += 32768;
		$data = \substr($terrain, $offset, 16384);
		$offset += 16384;
		$skyLight = \substr($terrain, $offset, 16384);
		$offset += 16384;
		$blockLight = \substr($terrain, $offset, 16384);
		$offset += 16384;

		$heightMap = [];
		foreach(\unpack("C*", \substr($terrain, $offset, 256)) as $c){
			$heightMap[] = $c;
		}
		$offset += 256;

		$biomeColors = [];
		foreach(\unpack("N*", \substr($terrain, $offset, 1024)) as $c){
			$biomeColors[] = $c;
		}
		$offset += 1024;

		parent::__construct($level, $chunkX, $chunkZ, $blocks, $data, $skyLight, $blockLight, $biomeColors, $heightMap, $entityData === \null ? [] : $entityData, $tileData === \null ? [] : $tileData);
	}

	public function getBlockId($x, $y, $z){
		return \ord($this->blocks{($x << 11) | ($z << 7) | $y});
	}

	public function setBlockId($x, $y, $z, $id){
		$this->blocks{($x << 11) | ($z << 7) | $y} = \chr($id);
		$this->hasChanged = \true;
	}

	public function getBlockData($x, $y, $z){
		$m = \ord($this->data{($x << 10) | ($z << 6) | ($y >> 1)});
		if(($y & 1) === 0){
			return $m & 0x0F;
		}else{
			return $m >> 4;
		}
	}

	public function setBlockData($x, $y, $z, $data){
		$i = ($x << 10) | ($z << 6) | ($y >> 1);
		$old_m = \ord($this->data{$i});
		if(($y & 1) === 0){
			$this->data{$i} = \chr(($old_m & 0xf0) | ($data & 0x0f));
		}else{
			$this->data{$i} = \chr((($data & 0x0f) << 4) | ($old_m & 0x0f));
		}
		$this->hasChanged = \true;
	}

	public function getFullBlock($x, $y, $z){
		$i = ($x << 11) | ($z << 7) | $y;
		if(($y & 1) === 0){
			return (\ord($this->blocks{$i}) << 4) | (\ord($this->data{$i >> 1}) & 0x0F);
		}else{
			return (\ord($this->blocks{$i}) << 4) | (\ord($this->data{$i >> 1}) >> 4);
		}
	}

	public function getBlock($x, $y, $z, &$blockId, &$meta = \null){
		$full = $this->getFullBlock($x, $y, $z);
		$blockId = $full >> 4;
		$meta = $full & 0x0f;
	}

	public function setBlock($x, $y, $z, $blockId = \null, $meta = \null){
		$i = ($x << 11) | ($z << 7) | $y;

		$changed = \false;

		if($blockId !== \null){
			$blockId = \chr($blockId);
			if($this->blocks{$i} !== $blockId){
				$this->blocks{$i} = $blockId;
				$changed = \true;
			}
		}

		if($meta !== \null){
			$i >>= 1;
			$old_m = \ord($this->data{$i});
			if(($y & 1) === 0){
				$this->data{$i} = \chr(($old_m & 0xf0) | ($meta & 0x0f));
				if(($old_m & 0x0f) !== $meta){
					$changed = \true;
				}
			}else{
				$this->data{$i} = \chr((($meta & 0x0f) << 4) | ($old_m & 0x0f));
				if((($old_m & 0xf0) >> 4) !== $meta){
					$changed = \true;
				}
			}
		}

		if($changed){
			$this->hasChanged = \true;
		}

		return $changed;
	}

	public function getBlockSkyLight($x, $y, $z){
		$sl = \ord($this->skyLight{($x << 10) | ($z << 6) | ($y >> 1)});
		if(($y & 1) === 0){
			return $sl & 0x0F;
		}else{
			return $sl >> 4;
		}
	}

	public function setBlockSkyLight($x, $y, $z, $level){
		$i = ($x << 10) | ($z << 6) | ($y >> 1);
		$old_sl = \ord($this->skyLight{$i});
		if(($y & 1) === 0){
			$this->skyLight{$i} = \chr(($old_sl & 0xf0) | ($level & 0x0f));
		}else{
			$this->skyLight{$i} = \chr((($level & 0x0f) << 4) | ($old_sl & 0x0f));
		}
		$this->hasChanged = \true;
	}

	public function getBlockLight($x, $y, $z){
		$l = \ord($this->blockLight{($x << 10) | ($z << 6) | ($y >> 1)});
		if(($y & 1) === 0){
			return $l & 0x0F;
		}else{
			return $l >> 4;
		}
	}

	public function setBlockLight($x, $y, $z, $level){
		$i = ($x << 10) | ($z << 6) | ($y >> 1);
		$old_l = \ord($this->blockLight{$i});
		if(($y & 1) === 0){
			$this->blockLight{$i} = \chr(($old_l & 0xf0) | ($level & 0x0f));
		}else{
			$this->blockLight{$i} = \chr((($level & 0x0f) << 4) | ($old_l & 0x0f));
		}
		$this->hasChanged = \true;
	}

	public function getBlockIdColumn($x, $z){
		return \substr($this->blocks, ($x << 11) + ($z << 7), 128);
	}

	public function getBlockDataColumn($x, $z){
		return \substr($this->data, ($x << 10) + ($z << 6), 64);
	}

	public function getBlockSkyLightColumn($x, $z){
		return \substr($this->skyLight, ($x << 10) + ($z << 6), 64);
	}

	public function getBlockLightColumn($x, $z){
		return \substr($this->blockLight, ($x << 10) + ($z << 6), 64);
	}

	/**
	 * @return bool
	 */
	public function isLightPopulated(){
		return $this->isLightPopulated;
	}

	/**
	 * @param int $value
	 */
	public function setLightPopulated($value = 1){
		$this->isLightPopulated = (bool) $value;
	}

	/**
	 * @return bool
	 */
	public function isPopulated(){
		return $this->isPopulated;
	}

	/**
	 * @param int $value
	 */
	public function setPopulated($value = 1){
		$this->isPopulated = (bool) $value;
	}

	/**
	 * @return bool
	 */
	public function isGenerated(){
		return $this->isGenerated;
	}

	/**
	 * @param int $value
	 */
	public function setGenerated($value = 1){
		$this->isGenerated = (bool) $value;
	}

	public static function fromFastBinary($data, LevelProvider $provider = \null){
		return self::fromBinary($data, $provider);
	}

	/**
	 * @param string        $data
	 * @param LevelProvider $provider
	 *
	 * @return Chunk
	 */
	public static function fromBinary($data, LevelProvider $provider = \null){
		try{
			$chunkX = (\PHP_INT_SIZE === 8 ? \unpack("V", \substr($data, 0, 4))[1] << 32 >> 32 : \unpack("V", \substr($data, 0, 4))[1]);
			$chunkZ = (\PHP_INT_SIZE === 8 ? \unpack("V", \substr($data, 4, 4))[1] << 32 >> 32 : \unpack("V", \substr($data, 4, 4))[1]);
			$chunkData = \substr($data, 8, -1);

			$flags = \ord(\substr($data, -1));

			$entities = \null;
			$tiles = \null;

			if($provider instanceof LevelDB){
				$nbt = new NBT(NBT::LITTLE_ENDIAN);

				$entityData = $provider->getDatabase()->get(\substr($data, 0, 8) . LevelDB::ENTRY_ENTITIES);
				if($entityData !== \false and \strlen($entityData) > 0){
					$nbt->read($entityData, \true);
					$entities = $nbt->getData();
					if(!\is_array($entities)){
						$entities = [$entities];
					}
				}
				$tileData = $provider->getDatabase()->get(\substr($data, 0, 8) . LevelDB::ENTRY_TILES);
				if($tileData !== \false and \strlen($tileData) > 0){
					$nbt->read($tileData, \true);
					$tiles = $nbt->getData();
					if(!\is_array($tiles)){
						$tiles = [$tiles];
					}
				}
			}

			$chunk = new Chunk($provider instanceof LevelProvider ? $provider : LevelDB::class, $chunkX, $chunkZ, $chunkData, $entities, $tiles);
			if($flags & 0x01){
				$chunk->setGenerated();
			}
			if($flags & 0x02){
				$chunk->setPopulated();
			}
			if($flags & 0x04){
				$chunk->setLightPopulated();
			}
			return $chunk;
		}catch(\Exception $e){
			return \null;
		}
	}

	public function toFastBinary(){
		return $this->toBinary(\false);
	}

	public function toBinary($saveExtra = \false){
		$chunkIndex = LevelDB::chunkIndex($this->getX(), $this->getZ());

		$provider = $this->getProvider();
		if($saveExtra and $provider instanceof LevelDB){
			$nbt = new NBT(NBT::LITTLE_ENDIAN);
			$entities = [];

			foreach($this->getEntities() as $entity){
				if(!($entity instanceof Player) and !$entity->closed){
					$entity->saveNBT();
					$entities[] = $entity->namedtag;
				}
			}

			if(\count($entities) > 0){
				$nbt->setData($entities);
				$provider->getDatabase()->put($chunkIndex . LevelDB::ENTRY_ENTITIES, $nbt->write());
			}else{
				$provider->getDatabase()->delete($chunkIndex . LevelDB::ENTRY_ENTITIES);
			}


			$tiles = [];

			foreach($this->getTiles() as $tile){
				if(!$tile->closed){
					$tile->saveNBT();
					$tiles[] = $tile->namedtag;
				}
			}

			if(\count($tiles) > 0){
				$nbt->setData($tiles);
				$provider->getDatabase()->put($chunkIndex . LevelDB::ENTRY_TILES, $nbt->write());
			}else{
				$provider->getDatabase()->delete($chunkIndex . LevelDB::ENTRY_TILES);
			}


		}

		$heightmap = \pack("C*", ...$this->getHeightMapArray());
		$biomeColors = \pack("N*", ...$this->getBiomeColorArray());

		return $chunkIndex .
			$this->getBlockIdArray() .
			$this->getBlockDataArray() .
			$this->getBlockSkyLightArray() .
			$this->getBlockLightArray() .
			$heightmap .
			$biomeColors . \chr(
				($this->isLightPopulated() ? 0x04 : 0) | ($this->isPopulated() ? 0x02 : 0) | ($this->isGenerated() ? 0x01 : 0)
			);
	}

	/**
	 * @param int           $chunkX
	 * @param int           $chunkZ
	 * @param LevelProvider $provider
	 *
	 * @return Chunk
	 */
	public static function getEmptyChunk($chunkX, $chunkZ, LevelProvider $provider = \null){
		try{
			$chunk = new Chunk($provider instanceof LevelProvider ? $provider : LevelDB::class, $chunkX, $chunkZ, \str_repeat("\x00", self::DATA_LENGTH));
			$chunk->skyLight = \str_repeat("\xff", 16384);
			return $chunk;
		}catch(\Exception $e){
			return \null;
		}
	}
}
