// Fill out your copyright notice in the Description page of Project Settings. #pragma once #include "CoreMinimal.h" #include "Unit.h" #include "Upgrade.h" #include "UnitUpgrade.generated.h" /** * */ UCLASS() class SPACEBATTLER_API UUnitUpgrade : public UUpgrade { GENERATED_BODY() public: void ApplyToUnit(AUnit* Unit); };