vampire-like/Assets/GameMain/Scripts/Utility/RarityUtility.cs

90 lines
2.5 KiB
C#

using DataTable;
using Definition.Enum;
using GameFramework.DataTable;
using UnityEngine;
namespace CustomUtility
{
public static class RarityUtility
{
public static ItemRarity SelectRarityForLevel(IDataTable<DRLevelRarity> table, int level)
{
DRLevelRarity row = GetLevelRarityRow(table, level);
if (row == null)
{
return ItemRarity.White;
}
int white = Mathf.Max(0, row.WhiteWeight);
int green = Mathf.Max(0, row.GreenWeight);
int blue = Mathf.Max(0, row.BlueWeight);
int purple = Mathf.Max(0, row.PurpleWeight);
int red = Mathf.Max(0, row.RedWeight);
int total = white + green + blue + purple + red;
if (total <= 0)
{
return ItemRarity.White;
}
int roll = Random.Range(1, total + 1);
if (roll <= white) return ItemRarity.White;
roll -= white;
if (roll <= green) return ItemRarity.Green;
roll -= green;
if (roll <= blue) return ItemRarity.Blue;
roll -= blue;
if (roll <= purple) return ItemRarity.Purple;
return ItemRarity.Red;
}
private static DRLevelRarity GetLevelRarityRow(IDataTable<DRLevelRarity> table, int level)
{
if (table == null)
{
return null;
}
DRLevelRarity[] rows = table.GetAllDataRows();
if (rows == null || rows.Length == 0)
{
return null;
}
int safeLevel = Mathf.Max(1, level);
DRLevelRarity below = null;
DRLevelRarity above = null;
foreach (DRLevelRarity row in rows)
{
if (row == null)
{
continue;
}
if (safeLevel >= row.LevelMin && safeLevel <= row.LevelMax)
{
return row;
}
if (safeLevel < row.LevelMin)
{
if (above == null || row.LevelMin < above.LevelMin)
{
above = row;
}
}
else if (safeLevel > row.LevelMax)
{
if (below == null || row.LevelMax > below.LevelMax)
{
below = row;
}
}
}
return above ?? below;
}
}
}