Skip to content

Commit b3c325c

Browse files
infinite loop prevention
- thx @Noggog
1 parent e7d4a81 commit b3c325c

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

HalgarisRPGLoot/ArmorAnalyzer.cs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -232,9 +232,10 @@ private FormKey GenerateEnchantment(
232232
{
233233
var level = item.Entry.Data.Level;
234234
var forLevel = ByLevelIndexed[level];
235+
var takeMin = Math.Min(rarityEnchCount, forLevel.Length);
235236
var effects = Extensions.Repeatedly(() => forLevel.RandomItem())
236237
.Distinct()
237-
.Take(rarityEnchCount)
238+
.Take(takeMin)
238239
.Shuffle();
239240

240241
var oldench = effects.First().Enchantment;

HalgarisRPGLoot/WeaponAnalyer.cs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,9 +233,10 @@ private FormKey GenerateEnchantment(
233233
{
234234
var level = item.Entry.Data.Level;
235235
var forLevel = ByLevelIndexed[level];
236+
var takeMin = Math.Min(rarityEnchCount, forLevel.Length);
236237
var effects = Extensions.Repeatedly(() => forLevel.RandomItem())
237238
.Distinct()
238-
.Take(rarityEnchCount)
239+
.Take(takeMin)
239240
.Shuffle();
240241

241242
var oldench = effects.First().Enchantment;

0 commit comments

Comments
 (0)