【UE5】TMapに複合キーを指定する方法

// int32をキー、FStringを値として扱う.
TMap<int32, FString> MyMap;
MyMap.Add(1, TEXT("Apple"));
MyMap.Add(2, TEXT("Banana"));

// 敵のカテゴリーを定義する列挙型.
UENUM(BlueprintType)
enum class EEnemyCategory : uint8
{
  None,
  Slime,
  Dragon,
};

/**
 * @brief 敵データテーブル.
 */
USTRUCT(BlueprintType)
struct FEnemyData : public FTableRowBase
{
  GENERATED_BODY()

  // カテゴリー.
  UPROPERTY(EditAnywhere, BlueprintReadWrite)
  EEnemyCategory Category;

  // ID.
  UPROPERTY(EditAnywhere, BlueprintReadWrite)
  int32 ID;

  // 名前.
  UPROPERTY(EditAnywhere, BlueprintReadWrite)
  FName Name;

  // HP.
  UPROPERTY(EditAnywhere, BlueprintReadWrite)
  int32 HP;

  // 攻撃力.
  UPROPERTY(EditAnywhere, BlueprintReadWrite)
  int32 Attack;
};

/**
 * @brief 敵の複合キー構造体.
 */
USTRUCT(BlueprintType)
struct FEnemyKey
{
  GENERATED_BODY()

  // カテゴリー.
  UPROPERTY(EditAnywhere, BlueprintReadWrite)
  EEnemyCategory Category;

  // ID.
  UPROPERTY(EditAnywhere, BlueprintReadWrite)
  int32 ID;

  // 構造体同士の比較演算子の定義が必要.
  bool operator==(const FEnemyKey& Other) const
  {
    // カテゴリとIDの2つが一致するかどうか.
    return Category == Other.Category && ID == Other.ID;
  }
};

// ハッシュ関数の定義が必要.
FORCEINLINE uint32 GetTypeHash(const FEnemyKey& Key)
{
  return HashCombine(GetTypeHash(Key.Category), GetTypeHash(Key.ID));
}

// 敵データテーブルの参照.
UPROPERTY(EditAnywhere, BlueprintReadWrite, Category = "EnemyData")
UDataTable* EnemyDataTable;

// 敵データの検索を高速化するためのマップ.
UPROPERTY()
TMap<FEnemyKey, FEnemyData> EnemyDataMap;

void ACPP_TestMultiKey::BeginPlay()
{
  Super::BeginPlay();
  
  // 敵データテーブルをロードする.
  for(const auto& LRowName : EnemyDataTable->GetRowNames()) {
    // 敵データを取得する.
    FEnemyData* LEnemyData = EnemyDataTable->FindRow<FEnemyData>(LRowName, TEXT(""));
    if(LEnemyData) {
      // 敵データのカテゴリーとIDをキーにして、マップに登録する.
      FEnemyKey LKey;
      LKey.Category = LEnemyData->Category;
      LKey.ID = LEnemyData->ID;
      // マップに登録.
      EnemyDataMap.Add(LKey, *LEnemyData);
    }
  }
}

// 敵データを取得する関数.
bool ACPP_TestMultiKey::GetEnemyData(EEnemyCategory Category, int32 ID, FEnemyData& OutEnemy)
{
  FEnemyKey LKey;
  LKey.Category = Category;
  LKey.ID = ID;
  if(EnemyDataMap.Contains(LKey) == false) {
    return false; // 存在しない.
  }
  // マップから敵データを取得.
  OutEnemy = EnemyDataMap[LKey];
  return true;
}

#include "NiagaraComponent.h"
#include "UObject/ObjectKey.h" // FObjectKeyを使うために必要.

// FObjectKeyをキーとして使う.
// TWeakObjectPtrをつけることで参照カウンタを増やさない.
TMap<FObjectKey, TWeakObjectPtr<UNiagaraComponent>> Map;

// 内包するオブジェクト.
void Add(UNiagaraComponent* Comp)
{
    if (!IsValid(Comp)) return;
    Map.Add(FObjectKey(Comp), Comp);
}

// 指定のオブジェクトを内包しているか？
bool Contains(UNiagaraComponent* Comp) const
{
    if (!IsValid(Comp)) return false;
    return Map.Contains(FObjectKey(Comp));
}

// 無効化された要素をクリーンアップ.
void Compact()
{
    for (auto It = Map.CreateIterator(); It; ++It)
    {
        if (!It->Value.IsValid()) // 無効なオブジェクト.
        {
            It.RemoveCurrent(); // 削除.
        }
    }
}