Weapon CPP

#include "EnterAurora.h"
#include "AuroraBaseWeapon.h"

// Sets default values
AAuroraBaseWeapon::AAuroraBaseWeapon(const FObjectInitializer& ObjectInitializer) : Super(ObjectInitializer)
{
    // Set this actor to call Tick() every frame.  You can turn this off to improve performance if you don't need it.
    PrimaryActorTick.bCanEverTick = true;

    // Create Weapon Mesh
    weaponMesh = ObjectInitializer.CreateDefaultSubobject<USkeletalMeshComponent>(this, TEXT("Weapon Mesh"));
    RootComponent = weaponMesh;

    // Create Audio Component
    audioSource = ObjectInitializer.CreateDefaultSubobject<UAudioComponent>(this, TEXT("Audio Source"));
    audioSource->AttachTo(RootComponent);
    audioSource->bAutoActivate = false;
}

// Called when the game starts or when spawned
void AAuroraBaseWeapon::BeginPlay()
{
    Super::BeginPlay();

    // Clamp cryo level
    currentCryoLevel = FMath::Clamp(currentCryoLevel, 0.0f, maxCryoLevel);
}

// Called every frame
void AAuroraBaseWeapon::Tick(float DeltaTime)
{
    Super::Tick(DeltaTime);

    if (currentCryoLevel > 0.0f)
    {
        currentCryoLevel -= cryoMultiplier * DeltaTime;
        GEngine->AddOnScreenDebugMessage(-1, 2.0f, FColor::Green, TEXT("Current Cryo level: ") + FString::SanitizeFloat(currentCryoLevel));
    }
}

void AAuroraBaseWeapon::OnWeaponFireStart()
{
    if (!bExhausted)
    {
        if (projectileType == EWeaponProjectile::EBullet)
        {
            // Debug Message
            GEngine->AddOnScreenDebugMessage(-1, 2.0f, FColor::Green, TEXT("Bullet"));
            Instant_Fire();
            audioSource->Activate(true);
            audioSource->Play(0.0f);
            currentCryoLevel += cryoMultiplier;
            GEngine->AddOnScreenDebugMessage(-1, 2.0f, FColor::Green, TEXT("Current Cryo level: ") + FString::SanitizeFloat(currentCryoLevel));
            CheckCryoLevel();
            GetWorld()->GetTimerManager().SetTimer(exhaustedHandler, this, &AAuroraBaseWeapon::ExhaustedWaitTimer, 2.0f, false);
        }
        if (projectileType == EWeaponProjectile::ESpread)
        {
            // Debug Message
            GEngine->AddOnScreenDebugMessage(-1, 2.0f, FColor::Green, TEXT("Spread"));

            // For loop for spread
            for (int32 i = 0; i <= weaponSpread; i++)
            {
                Instant_Fire();
                audioSource->Activate(true);
                audioSource->Play(0.0f);
            }
        }
    }

    GetWorld()->GetTimerManager().SetTimer(fireHandler, this, &AAuroraBaseWeapon::OnWeaponFireStart, fireRate, false);
}

void AAuroraBaseWeapon::OnWeaponFireStop()
{
    GetWorld()->GetTimerManager().ClearTimer(fireHandler);
}

void AAuroraBaseWeapon::Instant_Fire()
{
    const int32 RandomSeed = FMath::Rand();
    FRandomStream weaponRandomStream(RandomSeed);
    const float currentSpread = weaponSpread;
    const float spreadCone = FMath::DegreesToRadians(weaponSpread * 0.5f);
    const FVector aimDir = weaponMesh->GetSocketRotation(muzzleSocket).Vector();
    const FVector startTrace = weaponMesh->GetSocketLocation(muzzleSocket);
    const FVector shootDir = weaponRandomStream.VRandCone(aimDir, spreadCone, spreadCone);
    const FVector endTrace = startTrace + shootDir * weaponRange;
    const FHitResult impact = WeaponTrace(startTrace, endTrace);
    ProcessInstantHit(impact, startTrace, shootDir, RandomSeed, currentSpread);
}

void AAuroraBaseWeapon::CheckCryoLevel()
{
    if (currentCryoLevel >= maxCryoLevel)
    {
        bExhausted = true;
    }
    else
        bExhausted = false;
}

void AAuroraBaseWeapon::ExhaustedWaitTimer()
{
    if (currentCryoLevel < 10.0f)
    {
        bExhausted = false;
    }
}

FHitResult AAuroraBaseWeapon::WeaponTrace(const FVector & TraceFrom, const FVector & TraceTo) const
{
    static FName weaponFireTag = FName(TEXT("WeaponTrace"));

    FCollisionQueryParams traceParams(weaponFireTag, true, Instigator);
    traceParams.bTraceAsyncScene = true;
    traceParams.bReturnPhysicalMaterial = true;
    traceParams.AddIgnoredActor(this);

    FHitResult hit(ForceInit);

    GetWorld()->LineTraceSingleByChannel(hit, TraceFrom, TraceTo, Trace_Weapon, traceParams);

    return hit;
}

void AAuroraBaseWeapon::ProcessInstantHit(const FHitResult & Impact, const FVector & Origin, const FVector & ShootDir, int32 RandomSeed, float ReticleSpread)
{
    const FVector endTrace = Origin + ShootDir * weaponRange;
    const FVector endPoint = Impact.GetActor() ? Impact.ImpactPoint : endTrace;
    DrawDebugLine(this->GetWorld(), Origin, Impact.TraceEnd, FColor::Red, true, 1000.0f, 10.0f);

    FDamageEvent damageEvent;
}