update

Source/Luckyrobots/Private/MainScreenUserWidget.cpp

@@ -4,7 +4,133 @@
 #include "MainScreenUserWidget.h"
 #include "Engine/DataTable.h"
 
+
 void UMainScreenUserWidget::InitData()
 {
+	InitRobotData();
+	InitLevelData();
+}
 
+void UMainScreenUserWidget::InitRobotData()
+{
+	FLevelData CurrentLevelData = GetCurrentLevelData();
+	if (CurrentLevelData.ID > 0)
+	{
+		if (RobotDataDataTable)
+		{
+			RobotDataList.Empty();
+
+			FString ContextString;
+			TArray<FName> RowNames = RobotDataDataTable->GetRowNames();
+			for (auto RowString : RowNames)
+			{
+				FRobotData* pRow = RobotDataDataTable->FindRow<FRobotData>(FName(RowString), ContextString);
+				if (pRow)
+				{
+					if (pRow->bActive)
+					{
+						if (CurrentLevelData.RobotTypeList.Find(pRow->RobotType) >= 0)
+						{
+							RobotDataList.Add(*pRow);
+						}
+					}
+				}
+			}
+		}
+	}
+	else
+	{
+		if (RobotDataDataTable)
+		{
+			RobotDataList.Empty();
+
+			FString ContextString;
+			TArray<FName> RowNames = RobotDataDataTable->GetRowNames();
+			for (auto RowString : RowNames)
+			{
+				FRobotData* pRow = RobotDataDataTable->FindRow<FRobotData>(FName(RowString), ContextString);
+				if (pRow)
+				{
+					if (pRow->bActive)
+					{
+						RobotDataList.Add(*pRow);
+					}
+				}
+			}
+		}
+	}
+
+	BPUpdateSelectRobot();
+}
+
+void UMainScreenUserWidget::InitLevelData()
+{
+	FRobotData CurrentRobotData = GetCurrentRobotData();
+	if (!CurrentRobotData.Name.IsEmpty())
+	{
+		if (LevelDataTable)
+		{
+			LevelDataList.Empty();
+
+			FString ContextString;
+			TArray<FName> RowNames = LevelDataTable->GetRowNames();
+			for (auto RowString : RowNames)
+			{
+				FLevelData* pRow = LevelDataTable->FindRow<FLevelData>(FName(RowString), ContextString);
+				if (pRow)
+				{
+					if (pRow->bActive)
+					{
+						if (pRow->RobotTypeList.Find(CurrentRobotData.RobotType) >= 0)
+						{
+							LevelDataList.Add(*pRow);
+						}
+					}
+				}
+			}
+		}
+	}
+	else
+	{
+		if (LevelDataTable)
+		{
+			LevelDataList.Empty();
+
+			FString ContextString;
+			TArray<FName> RowNames = LevelDataTable->GetRowNames();
+			for (auto RowString : RowNames)
+			{
+				FLevelData* pRow = LevelDataTable->FindRow<FLevelData>(FName(RowString), ContextString);
+				if (pRow)
+				{
+					if (pRow->bActive)
+					{
+						LevelDataList.Add(*pRow);
+					}
+				}
+			}
+		}
+	}
+
+	BPUpdateSelectLevel();
+}
+
+FRobotData UMainScreenUserWidget::GetCurrentRobotData()
+{
+	FRobotData CurrentRobotData;
+	if (RobotDataList.IsValidIndex(iCurrentSelectRobot))
+	{
+		CurrentRobotData = RobotDataList[iCurrentSelectRobot];
+	}
+	return CurrentRobotData;
+}
+
+FLevelData UMainScreenUserWidget::GetCurrentLevelData()
+{
+	FLevelData CurrentLevelData;
+	if (LevelDataList.IsValidIndex(iCurrentSelectLevel))
+	{
+		CurrentLevelData = LevelDataList[iCurrentSelectLevel];
+	}
+	return CurrentLevelData;
 }

Source/Luckyrobots/Public/MainScreenUserWidget.h

@@ -4,6 +4,7 @@
 
 #include "CoreMinimal.h"
 #include "Blueprint/UserWidget.h"
+#include "SharedDef.h"
 #include "MainScreenUserWidget.generated.h"
 
 class UDataTable;
@@ -21,6 +22,32 @@ public:
 
 	UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "Config")
 	UDataTable* LevelDataTable;
+
 public:
+	UPROPERTY(EditAnywhere, BlueprintReadWrite)
+	TArray<FRobotData> RobotDataList;
+
+	UPROPERTY(EditAnywhere, BlueprintReadWrite)
+	TArray<FLevelData> LevelDataList;
+
+	UPROPERTY(EditAnywhere, BlueprintReadWrite)
+	int iCurrentSelectRobot;
+	UPROPERTY(EditAnywhere, BlueprintReadWrite)
+	int iCurrentSelectLevel;
+public:
+	UFUNCTION(BlueprintCallable)
 	void InitData();
+	UFUNCTION(BlueprintCallable)
+	void InitRobotData();
+	UFUNCTION(BlueprintCallable)
+	void InitLevelData();
+	UFUNCTION(BlueprintCallable)
+	FRobotData GetCurrentRobotData();
+	UFUNCTION(BlueprintCallable)
+	FLevelData GetCurrentLevelData();
+public:
+	UFUNCTION(BlueprintImplementableEvent)
+	void BPUpdateSelectRobot();
+	UFUNCTION(BlueprintImplementableEvent)
+	void BPUpdateSelectLevel();
 };

Source/Luckyrobots/Public/SharedDef.h

@@ -1,6 +1,5 @@
 #pragma once
 
-#include "Engine/DataTable.h"
 #include "SharedDef.generated.h"