fix an error

Source/KBFL/Private/Subsystems/ResourceNodes/KBFLActorSpawnDescriptorBase.cpp

@@ -115,16 +115,22 @@ void UKBFLActorSpawnDescriptorBase::RemoveWrongActors(TArray<FTransform>& Transf
 	}
 }
 
-void UKBFLActorSpawnDescriptorBase::ApplyMaterialData(AActor* Actor, TMap<uint8, UMaterial*> MaterialInfo)
+void UKBFLActorSpawnDescriptorBase::ApplyMaterialData(AActor* Actor, TMap<uint8, UMaterialInterface*> MaterialInfo)
 {
-	UStaticMeshComponent* StaticMesh = Cast<UStaticMeshComponent>(Actor->GetComponentByClass(UStaticMeshComponent::StaticClass()));
-	if(StaticMesh && MaterialInfo.Num() > 0)
+	if(Actor && MaterialInfo.Num() > 0)
 	{
-		for (auto Info : MaterialInfo)
+		if(UActorComponent* PrimitiveComponent = Actor->GetComponentByClass(UStaticMeshComponent::StaticClass()))
 		{
-			if( StaticMesh->GetNumMaterials() > Info.Key && Info.Value )
+			UStaticMeshComponent* StaticMesh = Cast<UStaticMeshComponent>(PrimitiveComponent);
+			if(StaticMesh)
 			{
-				StaticMesh->SetMaterial(Info.Key, Info.Value);
+				for (auto Info : MaterialInfo)
+				{
+					if( StaticMesh->GetNumMaterials() > Info.Key && Info.Value )
+					{
+						StaticMesh->SetMaterial(Info.Key, Info.Value);
+					}
+				}
 			}
 		}
 	}

Source/KBFL/Private/Subsystems/ResourceNodes/KBFLResourceNodeDescriptor_ResourceNode.cpp

@@ -13,7 +13,7 @@ void UKBFLResourceNodeDescriptor_ResourceNode::ForeachLocations(TArray<FTransfor
 			AActor* OutActor;
 			if(!CheckActorInRange(Location, OutActor) && IsRangeFree(Location))
 			{
-				SpawnActorAtLocation(Location, GetActorClass());
+				OutActor = SpawnActorAtLocation(Location, GetActorClass());
 			}
 
 			if(OutActor)

Source/KBFL/Public/Subsystems/ResourceNodes/KBFLActorSpawnDescriptorBase.h

@@ -32,7 +32,7 @@ class KBFL_API UKBFLActorSpawnDescriptorBase : public UObject
 	virtual void ModifySpawnedActorPostSpawn(AActor*& InActor);
 	virtual void AfterSpawning() {};
 	
-	void ApplyMaterialData(AActor* Actor, TMap<uint8, UMaterial*> MaterialInfo);
+	void ApplyMaterialData(AActor* Actor, TMap<uint8, UMaterialInterface*> MaterialInfo);
 
 	virtual bool IsAllowedToRemoveActor(AActor* InActor);
 	bool CheckWorld() const;

Source/KBFL/Public/Subsystems/ResourceNodes/KBFLResourceNodeDescriptor_ResourceNode.h

@@ -52,5 +52,5 @@ class KBFL_API UKBFLResourceNodeDescriptor_ResourceNode : public UKBFLResourceNo
 	TSubclassOf<AFGResourceNodeBase> mResourceNodeClass = nullptr;
 
 	UPROPERTY(EditDefaultsOnly, BlueprintReadWrite, Category="Resource Node")
-	TMap<uint8, UMaterial*> mResourceNodeMaterialInfo;
+	TMap<uint8, UMaterialInterface*> mResourceNodeMaterialInfo;
 };