WeaponFramework.h

Go to the documentation of this file.

// Fill out your copyright notice in the Description page of Project Settings.
 
#pragma once
 
#include "CoreMinimal.h"
#include "GameFramework/Actor.h"
#include "Interfaces/Weapon.h"
#include "WeaponFramework.generated.h"
 
class UProjectileWeaponComponent;
class UHitscanWeaponComponent;
DECLARE_LOG_CATEGORY_EXTERN(LogWeaponFramework, Display, All);
 
 
// class USpotLightComponent;
 
UCLASS()
class WEAPONSYSTEM_API AWeaponFramework : public AActor, public IWeapon
{
    GENERATED_BODY()
    
public: 
    AWeaponFramework();
 
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Weapon|Debugging")
    bool bDebuggingMode = false;
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Weapon|Debugging")
    bool bInfiniteAmmo = false;
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Weapon|Debugging")
    bool bBottomlessClip = false;
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Weapon|Debugging")
    float DrawDebugTime = 2.0f;
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Weapon|Debugging")
    float DrawDebugColorDarkMultiplier = 0.1f;
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Weapon|Debugging")
    float DrawDebugColorLightMultiplier = 2.0f;
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Weapon|Debugging")
    float DrawDebugPointSize = 16.0f;
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Weapon|Debugging")
    FLinearColor MainTraceColor = FLinearColor(0.9f, 0.1f, 0.1f);
    UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Weapon|Debugging")
    FLinearColor BlindFireTraceColor = FLinearColor(0.1f, 0.9f, 0.1f);
 
    UPROPERTY(EditAnywhere, Category = "Weapon|Trace|Debugging")
    bool bTraceComplex = false; 
    UPROPERTY(EditAnywhere, Category = "Weapon|Trace|Debugging")
    bool bReturnPhysicalMaterial = false; 
    //@TODO Get the "too close" number right
    UPROPERTY(EditAnywhere, Category = "Weapon|Trace|Debugging")
    float TraceDistanceTooClose = 0.075f;
protected:
    virtual void BeginPlay() override;
    virtual void SetupWeaponEffects();
 
    UFUNCTION(BlueprintCallable, Category="Weapon|Debugging")
    void LogDebugMessage(const FString& Message, const bool bError = false) const;
 
    UFUNCTION(Server, Reliable, Category="Weapon")
    void Server_InitWeapon();
 
    //~     Weapon Interface Functions      ~//
    virtual void InitWeapon() override;
    
    virtual void Fire(const FVector Location, const FRotator Rotation, const AActor* ActorToIgnore, const float Accuracy) override;
    virtual void Reload(const float ReloadTime) override;
    virtual void CancelReload() override;
    virtual void SwitchAutoMode() override;
    virtual void BlindFireWeapon(const FVector Location,const FRotator Rotation, const AActor* ActorToIgnore, const float Accuracy) override;
 
    virtual void AddAttachment(const FString& AttachmentID) override;
    virtual void ToggleAttachment(const EAttachmentType& AttachmentType) override;
    
    virtual void MoveUMG(bool bIsRightShoulder) override;
    virtual void Aiming(bool bIsAiming) override;
 
    virtual FWeaponData_T GetWeaponData() const override {return WeaponData;};
    virtual FWeaponStats_T GetWeaponStats() const override;
    virtual EWeaponType GetWeaponType() const override{return WeaponType;};
    virtual EWeaponFireMode GetFireMode() const override {return WeaponData.FireMode;};
 
    virtual FTransform GetPivotPoint() const override {return RecoilPivotPoint->GetRelativeTransform();};
    virtual FTransform GetAimDownSightSocket() const override;
    virtual float GetWeaponFOV() const override;
 
    virtual bool IsClipEmpty() const override {return WeaponData.CurrentAmmo <= 0;};
    
    virtual void SetWeaponData(const FWeaponData_T NewWeaponData) override {WeaponData = NewWeaponData;};
    virtual void SetWeaponShouldAutoReload(const bool bValue) override;
    virtual void SetWeaponMesh(USkeletalMesh* SkeletalMesh) override;
    virtual void SetMuzzleOffset(const FVector& Location, const FRotator& Rotation) override;
    virtual void SetRecoilPivotPoint(const FVector& Location, const FRotator& Rotation) override;
 
    virtual void SetWeaponMeshHidden(const bool bValue) override;
    UFUNCTION(NetMulticast, Reliable)
    void Multicast_SetWeaponMeshHidden(const bool bValue);
 
    virtual void SetDebuggingStats(const bool bDebug, const bool bInfinite, const bool bBottomless) override {bDebuggingMode = bDebug; bInfiniteAmmo = bInfinite, bBottomlessClip = bBottomless;};
 
    //~     Weapon Interface Functions      ~//
    
    UFUNCTION(BlueprintCallable, Category="Weapons")
    virtual void ToggleLaser(const bool bForceOff = false);
    
 
    //- Blueprint Functions //
    UFUNCTION(BlueprintImplementableEvent, Category = "Weapon", meta=(DisplayName = "Fire"))
    void ReceiveFire(FHitResult Hit, APawn* InstigatorPawn);
 
    
    UFUNCTION(BlueprintCallable, Category = "Weapon|Attachments")
    TArray<FWeaponModifiers> GetAllWeaponAttachmentModifiers() const;
    UFUNCTION(BlueprintCallable, Category = "Weapon|Attachments")
    TArray<FWeaponAttachment> GetAllWeaponAttachments() const;
    UFUNCTION(BlueprintCallable, Category = "Weapon|Attachments")
    FWeaponAttachment GetWeaponAttachment(const FString& AttachmentID) const;   
    UFUNCTION(BlueprintCallable, Category = "Weapon|Attachments")
    bool RemoveAttachment(const FString AttachmentID, const EAttachmentType AttachmentType);
    UFUNCTION(BlueprintCallable, Category = "Weapon|Attachments")
    void RemoveAllAttachments();
 
    
    bool CanShoot() const;
    float GetAccuracyMultiplier() const {return AccuracyMultiplier;};
    
 
    void ApplyDamageToActor(const FHitResult& Hit, FVector ShotDirection, float Multiplier = 1.0f);
 
    //- Multiplayer //
    UFUNCTION(Server, Reliable)
    void ServerFire(const FVector Location,const FRotator Rotation, const AActor* ActorToIgnore, const float Accuracy);
    UFUNCTION(Server, Reliable, WithValidation)
    virtual void ServerShoot();
    UFUNCTION(Server, Reliable)
    void ServerSetReloading(const bool bReloading);
    UFUNCTION(Server, Reliable)
    void ServerReload();
    UFUNCTION(Server, Reliable, Category = "Weapons|ADS")
    virtual void Server_SetWeaponShouldAutoReload(const bool bValue);
 
    void CalculateBulletSpread(FVector& NewBulletSpread) const;
    
    //@ MOVE TO HITSCAN //
    virtual float CalculateDistanceDamage(const float Distance) const;
    virtual FHitResult BlindFireWeaponTrace(const float Accuracy, const AActor* ActorToIgnore = nullptr);
    virtual bool LineTrace(FHitResult& Hit, FVector& ShotDirection, const FLinearColor& Color = FLinearColor::Green, const FVector& CustomLineEnd = FVector::ZeroVector) const;
 
    
    UPROPERTY(Replicated, VisibleAnywhere, BlueprintReadOnly, Category="Weapon|Components")
    UHitscanWeaponComponent* HitscanWeaponComponent;
    UPROPERTY(Replicated, VisibleAnywhere, BlueprintReadOnly, Category="Weapon|Components")
    UProjectileWeaponComponent* ProjectileWeaponComponent;
 
    UPROPERTY()
    FVector TraceLocation;
    FRotator TraceRotation;
    
    UPROPERTY()
    const AActor* TraceActorToIgnore;
    UPROPERTY()
    TArray<AActor*> ActorsToIgnore;
 
    UPROPERTY()
    APawn* OwnerPawn;
 
    UPROPERTY(Replicated, VisibleAnywhere, BlueprintReadOnly, Category="Weapon|Weapon Data")
    FWeaponData_T WeaponData;
    UPROPERTY(VisibleAnywhere, BlueprintReadOnly, Category="Weapon|General")
    EWeaponType WeaponType = EWeaponType::Default;
 
    
    //- Gun Visual Components   //
    UPROPERTY(VisibleAnywhere, BlueprintReadOnly, Category = "Weapon|Components")
    USceneComponent* SceneRoot; 
    UPROPERTY(VisibleAnywhere, BlueprintReadOnly, Category = "Weapon|Components")
    USkeletalMeshComponent* GunMeshComponent; 
    UPROPERTY(VisibleAnywhere, BlueprintReadOnly, Category = "Weapon|Components")
    USceneComponent* Muzzle;
    UPROPERTY(VisibleAnywhere, BlueprintReadOnly, Category = "Weapon|Components")
    USceneComponent* RecoilPivotPoint;
    UPROPERTY(VisibleAnywhere, BlueprintReadOnly, Category = "Weapon|Components")
    TArray<UStaticMeshComponent*> AttachmentComponents;
    UPROPERTY(VisibleAnywhere, BlueprintReadOnly, Category = "Weapon|Components")
    class UWeaponEffectsComponent* WeaponEffects;
 
    UFUNCTION()
    void OnMeshReplicated();
 
    UFUNCTION()
    void OnRep_WeaponStats();
    
    UPROPERTY(ReplicatedUsing=OnMeshReplicated, VisibleAnywhere, BlueprintReadOnly, Category = "Weapon|Components")
    USkeletalMesh* WeaponMesh;
    
    UPROPERTY(EditAnywhere, BlueprintReadOnly, Category="Projectile")
    TSubclassOf<class AActor> ProjectileClass;
 
 
private:
    void SetWeaponFireMode();
    void AddWeaponAttachments();
 
    void ShootHitScan();
    void ShootProjectile();
    UPROPERTY(ReplicatedUsing=OnRep_WeaponStats, VisibleAnywhere, Category = "Weapon|Status")
    FWeaponStats_T CurrentWeaponStats;
 
    
    bool bWeaponEffectsSetup = false;
    float AccuracyMultiplier = 1.0f;
    bool bLaserOn = false;
 
    bool bIsBeingReloaded;
    FTimerHandle ReloadTimerHandle;
};