////////////////////////////////////////////////////////////////////////////////
//    Scorched3D (c) 2000-2011, 2025, 2026
//
//    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 <weapons/WeaponRoller.hpp>
#include <weapons/AccessoryStore.hpp>
#include <weapons/Shield.hpp>
#include <actions/ShotBounce.hpp>
#include <common/Defines.hpp>
#include <engine/ActionController.hpp>
#include <engine/Simulator.hpp>
#include <target/Target.hpp>
#include <target/TargetContainer.hpp>
#include <target/TargetShield.hpp>
#include <target/TargetLife.hpp>
#include <landscapemap/LandscapeMaps.hpp>
#include <math.h>

REGISTER_ACCESSORY_SOURCE(WeaponRoller);

WeaponRoller::WeaponRoller() :
	collisionAction_(nullptr),
	shieldHurtFactorExp_("WeaponRoller::shieldHurtFactorExp", 0),
	windFactorExp_("WeaponRoller::windFactorExp", 1),
	gravityFactorExp_("WeaponRoller::gravityFactorExp", 1),
	dampenVelocityExp_("WeaponRoller::dampenVelocityExp", 1),
	timeout_("WeaponRoller::timeout", 0),
	scale_("WeaponRoller::scale", 1),
	timeExp_("WeaponRoller::timeExp"),
	numberRollers_("WeaponRoller::numberRollers"),
	stepSize_(true, 100),
	maintainVelocity_(false),
	roll_(true),
	stickyShields_(false),
	landscapeCollision_(true),
	shieldCollision_(true),
	tankCollision_(true),
	targetCollision_(true),
	noCameraTrack_(false)
{}

WeaponRoller::~WeaponRoller()
{
	delete collisionAction_;
	collisionAction_ = nullptr;
}

bool WeaponRoller::parseXML(AccessoryCreateContext &context, XMLNode *accessoryNode)
{
	if (!Weapon::parseXML(context, accessoryNode)) return false;

	// Get number of rollers
	if (!accessoryNode->getNamedChild("numberrollers", numberRollers_)) return false;

	accessoryNode->getNamedChild("scale", scale_, false);

	// Get life time
	if (!accessoryNode->getNamedChild("time", timeExp_)) return false;
	accessoryNode->getNamedChild("stepsize", stepSize_, false);
	accessoryNode->getNamedChild("timeout", timeout_, false);

	accessoryNode->getNamedChild("landscapecollision", landscapeCollision_, false);
	accessoryNode->getNamedChild("shieldcollision", shieldCollision_, false);
	accessoryNode->getNamedChild("tankcollision", tankCollision_, false);
	accessoryNode->getNamedChild("targetcollision", targetCollision_, false);

    // Get the hurt factor (if any)
    accessoryNode->getNamedChild("shieldhurtfactor", shieldHurtFactorExp_, false);

	// Get the wind factor (if any)
	accessoryNode->getNamedChild("windfactor", windFactorExp_, false);
	accessoryNode->getNamedChild("gravityfactor", gravityFactorExp_, false);

	// Get the maintianvelocity (if any)
	accessoryNode->getNamedChild("maintainvelocity", maintainVelocity_, false);

	// Get the velocity dampening factor (if any)
	accessoryNode->getNamedChild("dampenvelocity", dampenVelocityExp_, false);

	// Get if we are to roll
	accessoryNode->getNamedChild("roll", roll_, false);

	// Get if we are to roll
	accessoryNode->getNamedChild("stickyshields", stickyShields_, false);

	accessoryNode->getNamedChild("nocameratrack", noCameraTrack_, false);

	// Groups
	if (!localGroups_.readXML(accessoryNode, "localgroupname")) return false;
	if (!globalGroups_.readXML(accessoryNode, "globalgroupname")) return false;

	XMLNode* subNode = nullptr;
	if (!accessoryNode->getNamedChild("collisionaction", subNode)) return false;

	AccessoryPart *accessory = context.getAccessoryStore().createAccessoryPart(context, parent_, subNode);
	if (!accessory || accessory->getType() != AccessoryPart::AccessoryWeapon)
	{
		return subNode->returnError("Failed to find sub weapon, not a weapon");
	}
	collisionAction_ = (Weapon*) accessory;

	// Get the weapon model
	XMLNode* modelNode = nullptr;
	if (!accessoryNode->getNamedChild("rollermodel", modelNode)) return false;
	if (!rollerModelId_.initFromNode(modelNode)) return false;

	return true;
}

void WeaponRoller::fireWeapon(
	ScorchedContext &context,
	WeaponFireContext &weaponContext,
	FixedVector &oldposition,
	FixedVector &velocity
)
{
	fixed minHeight = context.getLandscapeMaps().getGroundMaps().getInterpHeight(
		oldposition[0], oldposition[1]
	);

	// Make sure position is not underground
	if (oldposition[2] < minHeight)
	{
		if (minHeight - oldposition[2] > 10) // Give room for shields as well
		{
			return;
		}
	}

	RandomGenerator &random = context.getSimulator().getRandomGenerator();
	int numberRollers = numberRollers_.getUInt(context);
	for (int i=0; i<numberRollers; i++)
	{
		FixedVector position = oldposition;
		position[2] += fixed(true, 1500);
		
		// Make a slightly different starting position
		position[0] += random.getRandFixed("WeaponRoller") * 2 - 1;
		position[1] += random.getRandFixed("WeaponRoller") * 2 - 1;
		fixed minHeight = context.getLandscapeMaps().getGroundMaps().getInterpHeight(position[0], position[1]) + 1;
		if (position[2] < minHeight) position[2] = minHeight;
				
		// Check if we have hit the roof (quite litteraly)
		{
			fixed maxHeight = context.getLandscapeMaps().getRoofMaps().getInterpRoofHeight(position[0], position[1]);
			if (position[2] > maxHeight - 1)
			{
				position[2] = maxHeight - 1;
			}
		}

		// Make sure this new position is not inside a tank's shields
		bool ok = false;
		while (!ok)
		{
			ok = true;
			std::map<unsigned int, Target *> &targets = 
				context.getTargetContainer().getTargets();
			std::map<unsigned int, Target *>::iterator itor;
			for (itor = targets.begin();
				itor != targets.end();
				++itor)
			{
				Target *current = (*itor).second;
				FixedVector &tankPos = 
					current->getLife().getTargetPosition();
				Accessory *accessory = 
					current->getShield().getCurrentShield();
				if (accessory)
				{
					Shield *shield = (Shield *) accessory->getAction();
					FixedVector offset = position - tankPos;
					if (shield->inShield(offset))
					{
						ok = false;
						position[2] += 1;
					}
				}
			}
		}
		
		addRoller(context, weaponContext, position, velocity);
	}
}

void WeaponRoller::addRoller(
	ScorchedContext &context,
	WeaponFireContext &weaponContext,
	FixedVector &position,
	FixedVector &velocity
)
{
	RandomGenerator &random = context.getSimulator().getRandomGenerator();

	FixedVector newVelocity;
	if (maintainVelocity_)
	{
		newVelocity = velocity * dampenVelocityExp_.getValue(context);
	}
	else
	{
		newVelocity[0] = random.getRandFixed("WeaponRoller") - fixed(true, 5000);
		newVelocity[1] = random.getRandFixed("WeaponRoller") - fixed(true, 5000);
		newVelocity[2] = random.getRandFixed("WeaponRoller") * 2;
	}
	
	context.getActionController().addAction(
		new ShotBounce(this, position, newVelocity, weaponContext)
	);
}
