////////////////////////////////////////////////////////////////////////////////
//    Scorched3D (c) 2000-2011
//
//    This file is part of Scorched3D.
//
//    Scorched3D is free software; you can redistribute it and/or modify
//    it under the terms of the GNU General Public License as published by
//    the Free Software Foundation; either version 2 of the License, or
//    (at your option) any later version.
//
//    Scorched3D is distributed in the hope that it will be useful,
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//    GNU General Public License for more details.
//
//    You should have received a copy of the GNU General Public License along
//    with this program; if not, write to the Free Software Foundation, Inc.,
//    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
////////////////////////////////////////////////////////////////////////////////

#include <coms/ComsSyncCheckMessage.hpp>
#include <common/Logger.hpp>
#include <common/OptionsTransient.hpp>
#include <engine/Simulator.hpp>
#ifndef S3D_SERVER
	#include <client/ScorchedClient.hpp>
#endif
#include <server/ScorchedServer.hpp>
#include <target/TargetContainer.hpp>
#include <target/TargetState.hpp>
#include <target/TargetLife.hpp>
#include <tank/TankState.hpp>
#include <tank/TankScore.hpp>
#include <tank/Tank.hpp>
#include <landscapemap/LandscapeMaps.hpp>
#include <set>

ComsMessageType ComsSyncCheckMessage::ComsSyncCheckMessageType("ComsSyncCheckMessageType");

ComsSyncCheckMessage::ComsSyncCheckMessage() :
	ComsMessage(ComsSyncCheckMessageType)
{
}

ComsSyncCheckMessage::ComsSyncCheckMessage(unsigned int syncId,
	ScorchedContext &context) :
	ComsMessage(ComsSyncCheckMessageType),
	syncId_(syncId)
{
	// Send the height map data
	{
		HeightMap &map = context.getLandscapeMaps().getGroundMaps().getHeightMap();
		for (int y=0; y<map.getMapHeight(); y++)
		{
			for (int x=0; x<map.getMapWidth(); x++)
			{
				fixed height = map.getHeight(x, y);
				landscapeBuffer_.addToBuffer(height);
				FixedVector &normal = map.getNormal(x, y);
				landscapeBuffer_.addToBuffer(normal);
			}
		}
	}
	if (context.getLandscapeMaps().getRoofMaps().getRoofOn())
	{
		HeightMap &map = context.getLandscapeMaps().getRoofMaps().getRoofMap();
		for (int y=0; y<map.getMapHeight(); y++)
		{
			for (int x=0; x<map.getMapWidth(); x++)
			{
				fixed height = map.getHeight(x, y);
				roofBuffer_.addToBuffer(height);
				FixedVector &normal = map.getNormal(x, y);
				roofBuffer_.addToBuffer(normal);
			}
		}
	}

	// Send target data
	std::map<unsigned int, Target *> &possibletargets =
		context.getTargetContainer().getTargets();
	std::map<unsigned int, Target *>::iterator itor;
	targetsBuffer_.addToBuffer((int) possibletargets.size());
	for (itor = possibletargets.begin();
		itor != possibletargets.end();
		++itor)
	{
		Target *target = (*itor).second;
		targetsBuffer_.addToBuffer(target->getPlayerId());
		targetsBuffer_.addToBuffer((int) target->getType());

		NetBuffer tmpBuffer;
		target->writeMessage(tmpBuffer);
		targetsBuffer_.addToBuffer(tmpBuffer);
	}

	// Add sync checks
	syncChecks_ = context.getSimulator().getSyncCheck();
}

ComsSyncCheckMessage::~ComsSyncCheckMessage()
{
}

bool ComsSyncCheckMessage::writeMessage(NetBuffer &buffer)
{
	buffer.addToBuffer(syncId_);
	buffer.addToBuffer(landscapeBuffer_);
	buffer.addToBuffer(roofBuffer_);
	buffer.addToBuffer(targetsBuffer_);
	int checks = (int) syncChecks_.size();
	buffer.addToBuffer(checks);
	std::vector<std::string>::iterator itor;
	for (itor = syncChecks_.begin();
		itor != syncChecks_.end();
		++itor)
	{
		buffer.addToBuffer(*itor);
	}
	return true;
}

bool ComsSyncCheckMessage::readMessage(NetBufferReader &reader)
{
	if (!reader.getFromBuffer(syncId_)) return false;
	if (!reader.getFromBuffer(landscapeBuffer_)) return false;
	if (!reader.getFromBuffer(roofBuffer_)) return false;
	if (!reader.getFromBuffer(targetsBuffer_)) return false;
	int checks = 0;
	if (!reader.getFromBuffer(checks)) return false;
	for (int c=0; c<checks; c++)
	{
		std::string check;
		if (!reader.getFromBuffer(check)) return false;
		syncChecks_.push_back(check);
	}
	return true;
}

