112 lines
2.3 KiB
C++
112 lines
2.3 KiB
C++
// Copyright Pacificao. All Rights Reserved.
|
|
|
|
#include "AgrarianInventoryComponent.h"
|
|
#include "Net/UnrealNetwork.h"
|
|
|
|
UAgrarianInventoryComponent::UAgrarianInventoryComponent()
|
|
{
|
|
PrimaryComponentTick.bCanEverTick = false;
|
|
SetIsReplicatedByDefault(true);
|
|
}
|
|
|
|
void UAgrarianInventoryComponent::GetLifetimeReplicatedProps(TArray<FLifetimeProperty>& OutLifetimeProps) const
|
|
{
|
|
Super::GetLifetimeReplicatedProps(OutLifetimeProps);
|
|
DOREPLIFETIME(UAgrarianInventoryComponent, Items);
|
|
}
|
|
|
|
bool UAgrarianInventoryComponent::HasItem(FName ItemId, int32 Quantity) const
|
|
{
|
|
return GetItemCount(ItemId) >= Quantity;
|
|
}
|
|
|
|
int32 UAgrarianInventoryComponent::GetItemCount(FName ItemId) const
|
|
{
|
|
int32 Count = 0;
|
|
for (const FAgrarianItemStack& Stack : Items)
|
|
{
|
|
if (Stack.ItemId == ItemId)
|
|
{
|
|
Count += Stack.Quantity;
|
|
}
|
|
}
|
|
return Count;
|
|
}
|
|
|
|
bool UAgrarianInventoryComponent::AddItem(const FAgrarianItemStack& Stack)
|
|
{
|
|
if (!GetOwner() || !GetOwner()->HasAuthority() || !Stack.IsValidStack())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
for (FAgrarianItemStack& Existing : Items)
|
|
{
|
|
if (Existing.ItemId == Stack.ItemId)
|
|
{
|
|
Existing.Quantity += Stack.Quantity;
|
|
BroadcastInventoryChanged();
|
|
return true;
|
|
}
|
|
}
|
|
|
|
if (Items.Num() >= MaxSlots)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
Items.Add(Stack);
|
|
BroadcastInventoryChanged();
|
|
return true;
|
|
}
|
|
|
|
bool UAgrarianInventoryComponent::RemoveItem(FName ItemId, int32 Quantity)
|
|
{
|
|
if (!GetOwner() || !GetOwner()->HasAuthority() || Quantity <= 0 || !HasItem(ItemId, Quantity))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
int32 Remaining = Quantity;
|
|
for (int32 Index = Items.Num() - 1; Index >= 0 && Remaining > 0; --Index)
|
|
{
|
|
FAgrarianItemStack& Stack = Items[Index];
|
|
if (Stack.ItemId != ItemId)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
const int32 Removed = FMath::Min(Stack.Quantity, Remaining);
|
|
Stack.Quantity -= Removed;
|
|
Remaining -= Removed;
|
|
|
|
if (Stack.Quantity <= 0)
|
|
{
|
|
Items.RemoveAt(Index);
|
|
}
|
|
}
|
|
|
|
BroadcastInventoryChanged();
|
|
return true;
|
|
}
|
|
|
|
void UAgrarianInventoryComponent::ServerAddItem_Implementation(const FAgrarianItemStack& Stack)
|
|
{
|
|
AddItem(Stack);
|
|
}
|
|
|
|
void UAgrarianInventoryComponent::ServerRemoveItem_Implementation(FName ItemId, int32 Quantity)
|
|
{
|
|
RemoveItem(ItemId, Quantity);
|
|
}
|
|
|
|
void UAgrarianInventoryComponent::OnRep_Items()
|
|
{
|
|
BroadcastInventoryChanged();
|
|
}
|
|
|
|
void UAgrarianInventoryComponent::BroadcastInventoryChanged()
|
|
{
|
|
OnInventoryChanged.Broadcast();
|
|
}
|