From de279d5104ecf824eaed110e9eb3ef34afe3fcee Mon Sep 17 00:00:00 2001
From: mrhid6 <dominic.richardson1@hotmail.co.uk>
Date: Wed, 13 Nov 2024 16:51:47 +0000
Subject: [PATCH] feat: ds helper functions

---
 .../Private/Subsystems/RRDADataAssetSubsystem.cpp         | 8 ++++++++
 .../Public/Subsystems/RRDADataAssetSubsystem.h            | 6 ++++++
 2 files changed, 14 insertions(+)

diff --git a/Source/RefinedRDApi/Private/Subsystems/RRDADataAssetSubsystem.cpp b/Source/RefinedRDApi/Private/Subsystems/RRDADataAssetSubsystem.cpp
index c11e0d5..d0c3e03 100644
--- a/Source/RefinedRDApi/Private/Subsystems/RRDADataAssetSubsystem.cpp
+++ b/Source/RefinedRDApi/Private/Subsystems/RRDADataAssetSubsystem.cpp
@@ -262,6 +262,10 @@ TArray<URRDADataAssetDSModAdapter*> URRDADataAssetSubsystem::GetAllModAdapterDat
 	return Assets;
 }
 
+bool URRDADataAssetSubsystem::HasDSModAdapterForBuildingClass(TSubclassOf<AFGBuildableFactory> BuildingClass) {
+	return IsValid(GetModAdapterDataAssetForBuildingClass(BuildingClass));
+}
+
 URRDADataAssetDSAdapterCovers* URRDADataAssetSubsystem::GetDSAdapterCoverDataAssetForBuildingClass(TSubclassOf<AFGBuildableFactory> BuildingClass) const {
 	if (!BuildingClass) return nullptr;
 
@@ -286,6 +290,10 @@ TArray<URRDADataAssetDSAdapterCovers*> URRDADataAssetSubsystem::GetAllDSAdapterC
 	return Assets;
 }
 
+bool URRDADataAssetSubsystem::HasDSAdaperCoversForBuildingClass(TSubclassOf<AFGBuildableFactory> BuildingClass) {
+	return IsValid(GetDSAdapterCoverDataAssetForBuildingClass(BuildingClass));
+}
+
 bool URRDADataAssetSubsystem::GetAllTurbineItems(TArray<TSubclassOf<UFGItemDescriptor>>& Items, int32 Tier) const {
 	if (!mTurbineAssets.Contains(Tier)) {
 		return false;
diff --git a/Source/RefinedRDApi/Public/Subsystems/RRDADataAssetSubsystem.h b/Source/RefinedRDApi/Public/Subsystems/RRDADataAssetSubsystem.h
index 212eb48..f5ba402 100644
--- a/Source/RefinedRDApi/Public/Subsystems/RRDADataAssetSubsystem.h
+++ b/Source/RefinedRDApi/Public/Subsystems/RRDADataAssetSubsystem.h
@@ -120,6 +120,9 @@ public:
 	UFUNCTION(BlueprintPure, Category = "RRDA|DataAsset|DS")
 	TArray<URRDADataAssetDSModAdapter*> GetAllModAdapterDataAssets() const;
 
+	UFUNCTION(BlueprintCallable, Category = "RRDA|DataAsset|DS")
+	bool HasDSModAdapterForBuildingClass(TSubclassOf<AFGBuildableFactory> BuildingClass);
+
 	// DS Adapter Covers Functions
 	UFUNCTION(BlueprintCallable, Category = "RRDA|DataAsset|DS")
 	URRDADataAssetDSAdapterCovers* GetDSAdapterCoverDataAssetForBuildingClass(TSubclassOf<AFGBuildableFactory> BuildingClass) const;
@@ -127,6 +130,9 @@ public:
 	UFUNCTION(BlueprintPure, Category = "RRDA|DataAsset|DS")
 	TArray<URRDADataAssetDSAdapterCovers*> GetAllDSAdapterCoverDataAssets() const;
 
+	UFUNCTION(BlueprintCallable, Category = "RRDA|DataAsset|DS")
+	bool HasDSAdaperCoversForBuildingClass(TSubclassOf<AFGBuildableFactory> BuildingClass);
+
 	// Turbine Functions
 	UFUNCTION(BlueprintPure, Category = "RRDA|DataAsset|RP|MP")
 	bool GetAllTurbineItems(TArray<TSubclassOf<UFGItemDescriptor>>& Items, int32 Tier) const;