// Copyright 2016 Mookie. All Rights Reserved.

#include "EBBullet.h"
#include "EBMaterialResponseMap.h"
#include "EBBarrel.h"
#include "Net/UnrealNetwork.h"

void AEBBullet::VelocityChangeBroadcast_Implementation(FVector_NetQuantize NewLocation, FVector NewVelocity) {
	if (!HasAuthority()) {
		FVector RebasedLocation = UGameplayStatics::RebaseZeroOriginOntoLocal(GetWorld(), NewLocation);
		OnTrajectoryUpdateReceived(RebasedLocation, Velocity, NewVelocity);
		SetActorLocation(RebasedLocation);
		Velocity = NewVelocity;
		CanRetrace = false;
	}
}

void AEBBullet::VelocityChangeBroadcastReliable_Implementation(FVector_NetQuantize NewLocation, FVector NewVelocity) {
	if (!HasAuthority()) {
		FVector RebasedLocation = UGameplayStatics::RebaseZeroOriginOntoLocal(GetWorld(), NewLocation);
		OnTrajectoryUpdateReceived(RebasedLocation, Velocity, NewVelocity);
		SetActorLocation(RebasedLocation);
		Velocity = NewVelocity;
		CanRetrace = false;
	}
}

void AEBBullet::SpawnWithExactVelocity(TSubclassOf<class AEBBullet> BulletClass, AActor* BulletOwner, APawn* BulletInstigator, FVector BulletLocation, FVector BulletVelocity) {

	if (BulletClass != nullptr && BulletOwner != nullptr) {
		FActorSpawnParameters spawnParams;
		spawnParams.Owner = BulletOwner;
		spawnParams.Instigator = BulletInstigator;

		AEBBullet* Default = Cast<AEBBullet>(BulletClass->GetDefaultObject());

		FTransform Transform;
		Transform.SetLocation(BulletLocation);
		Transform.SetScale3D(Default->GetActorScale());

		if (Default->RotateActor) {
			FRotator Rotation = UKismetMathLibrary::MakeRotFromX(BulletVelocity);
			if (Default->RotateRandomRoll) Rotation.Add(0, 0, Default->RandomStream.FRandRange(-180.0f, 180.0f));
			Transform.SetRotation(Rotation.Quaternion());
		}
		else {
			Transform.SetRotation(FQuat(1, 0, 0, 1));
		}

		if (!Default->Shotgun) {
			AEBBullet* bullet = SpawnOrReactivate(BulletOwner->GetWorld(), BulletClass, Transform, BulletVelocity, BulletOwner, BulletInstigator);
		}
		else {
			for (int i = 0; i < Default->ShotCount; i++) {
				float Vel = BulletVelocity.Size()*Default->RandomStream.FRandRange(1.0 - Default->ShotVelocitySpread, 1.0 + Default->ShotVelocitySpread);
				FVector SubmunitionVelocity = Default->RandomStream.VRandCone(BulletVelocity, Default->ShotSpread)*Vel;
				AEBBullet* bullet = SpawnOrReactivate(BulletOwner->GetWorld(), BulletClass, Transform, SubmunitionVelocity, BulletOwner, BulletInstigator);
			}
		}
	}
	else {
		UE_LOG(LogTemp, Warning, TEXT("Cannot spawn bullet - invalid class or owner"));
	}
}

void AEBBullet::SpawnWithExactVelocityFromBarrel(TSubclassOf<class AEBBullet> BulletClass, AActor* BulletOwner, APawn* BulletInstigator, FVector BulletLocation, FVector BulletVelocity, class UEBBarrel* SourceBarrel) {

	if (BulletClass != nullptr && BulletOwner != nullptr) {
		FActorSpawnParameters spawnParams;
		spawnParams.Owner = BulletOwner;
		spawnParams.Instigator = BulletInstigator;

		AEBBullet* Default = Cast<AEBBullet>(BulletClass->GetDefaultObject());

		FTransform Transform;
		Transform.SetLocation(BulletLocation);
		Transform.SetScale3D(Default->GetActorScale());

		if (Default->RotateActor) {
			FRotator Rotation = UKismetMathLibrary::MakeRotFromX(BulletVelocity);
			if (Default->RotateRandomRoll) Rotation.Add(0, 0, Default->RandomStream.FRandRange(-180.0f, 180.0f));
			Transform.SetRotation(Rotation.Quaternion());
		}
		else {
			Transform.SetRotation(FQuat(1, 0, 0, 1));
		}

		if (!Default->Shotgun) {
			AEBBullet* bullet = SpawnOrReactivateFromBarrel(BulletOwner->GetWorld(), BulletClass, Transform, BulletVelocity, BulletOwner, BulletInstigator, SourceBarrel);
		}
		else {
			for (int i = 0; i < Default->ShotCount; i++) {
				float Vel = BulletVelocity.Size()*Default->RandomStream.FRandRange(1.0 - Default->ShotVelocitySpread, 1.0 + Default->ShotVelocitySpread);
				FVector SubmunitionVelocity = Default->RandomStream.VRandCone(BulletVelocity, Default->ShotSpread)*Vel;
				AEBBullet* bullet = SpawnOrReactivateFromBarrel(BulletOwner->GetWorld(), BulletClass, Transform, SubmunitionVelocity, BulletOwner, BulletInstigator, SourceBarrel);
			}
		}
	}
	else {
		UE_LOG(LogTemp, Warning, TEXT("Cannot spawn bullet - invalid class or owner"));
	}
}


void AEBBullet::GetLifetimeReplicatedProps(TArray< FLifetimeProperty > & OutLifetimeProps) const
{
	Super::GetLifetimeReplicatedProps(OutLifetimeProps);
	DOREPLIFETIME_CONDITION(AEBBullet, Velocity, COND_InitialOnly);
	DOREPLIFETIME_CONDITION(AEBBullet, RandomStream, COND_InitialOnly);
}

//alternative spawn
void AEBBullet::Spawn(TSubclassOf<class AEBBullet> BulletClass, AActor* BulletOwner, APawn* BulletInstigator, FVector BulletLocation, FVector BulletVelocity) {
	if (BulletClass != nullptr && BulletOwner != nullptr) {

		FActorSpawnParameters spawnParams;
		spawnParams.Owner = BulletOwner;
		spawnParams.Instigator = BulletInstigator;

		AEBBullet* Default = Cast<AEBBullet>(BulletClass->GetDefaultObject());

		FTransform Transform;
		Transform.SetLocation(BulletLocation);
		Transform.SetScale3D(Default->GetActorScale());

		if (Default->RotateActor) {
			if (Default->RotateActor) {
				FRotator Rotation = UKismetMathLibrary::MakeRotFromX(BulletVelocity);
				if (Default->RotateRandomRoll) Rotation.Add(0, 0, Default->RandomStream.FRandRange(-180.0f, 180.0f));
				Transform.SetRotation(Rotation.Quaternion());
			}
		}
		else {
			Transform.SetRotation(FQuat(1, 0, 0, 1));
		}

		//init velocity
		float TotalSpread = Default->Spread;
		if (Default->SpreadBias > 0.0f){
			float SpreadMult = FMath::Pow(FMath::FRand(), Default->SpreadBias);
			TotalSpread *= SpreadMult;
		}

		FVector BulletVelocityNew = Default->RandomStream.VRandCone(BulletVelocity, TotalSpread)*BulletVelocity.Size();
		float VelocityMP = FMath::Lerp(Default->MuzzleVelocityMin, Default->MuzzleVelocityMax, Default->RandomStream.FRand());
		BulletVelocityNew = BulletVelocityNew * VelocityMP;

		if (!Default->Shotgun) {
			AEBBullet* bullet = SpawnOrReactivate(BulletOwner->GetWorld(), BulletClass, Transform, BulletVelocityNew, BulletOwner, BulletInstigator);
		}
		else {
			for (int i = 0; i < Default->ShotCount; i++) {
				float Vel = BulletVelocityNew.Size()*Default->RandomStream.FRandRange(1.0 - Default->ShotVelocitySpread, 1.0 + Default->ShotVelocitySpread);
				FVector SubmunitionVelocity = Default->RandomStream.VRandCone(BulletVelocityNew, Default->ShotSpread)*Vel;
				AEBBullet* bullet = SpawnOrReactivate(BulletOwner->GetWorld(), BulletClass, Transform, SubmunitionVelocity, BulletOwner, BulletInstigator);
			}
		}
	}
	else {
		UE_LOG(LogTemp, Warning, TEXT("Cannot spawn bullet - invalid class or owner"));
	}
}

void AEBBullet::OnImpact_Implementation(bool Ricochet, bool PassedThrough, FVector Location, FVector IncomingVelocity, FVector Normal, FVector ExitLocation, FVector ExitVelocity, FVector Impulse, float PenetrationDepth, AActor* Actor, USceneComponent* Component, FName BoneName, UPhysicalMaterial* PhysMaterial, FHitResult HitResult){
	return;
};

void AEBBullet::OnNetPredictedImpact_Implementation(bool Ricochet, bool PassedThrough, FVector Location, FVector IncomingVelocity, FVector Normal, FVector ExitLocation, FVector ExitVelocity, FVector Impulse, float PenetrationDepth, AActor* Actor, USceneComponent* Component, FName BoneName, UPhysicalMaterial* PhysMaterial, FHitResult HitResult) {
	return;
};

void AEBBullet::OnDeactivated_Implementation() {
	return;
};