-
Notifications
You must be signed in to change notification settings - Fork 754
/
MeleeHarvestToolStatsBuilder.java
111 lines (98 loc) · 5.28 KB
/
MeleeHarvestToolStatsBuilder.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package slimeknights.tconstruct.tools;
import com.google.common.annotations.VisibleForTesting;
import lombok.AccessLevel;
import lombok.Getter;
import net.minecraft.world.item.Tier;
import net.minecraftforge.common.TierSortingRegistry;
import slimeknights.tconstruct.library.tools.definition.ToolDefinition;
import slimeknights.tconstruct.library.tools.definition.ToolDefinitionData;
import slimeknights.tconstruct.library.tools.definition.module.material.MaterialStatsToolHook;
import slimeknights.tconstruct.library.tools.definition.module.material.MaterialStatsToolHook.WeightedStatType;
import slimeknights.tconstruct.library.tools.nbt.MaterialNBT;
import slimeknights.tconstruct.library.tools.nbt.StatsNBT;
import slimeknights.tconstruct.library.tools.stat.IToolStat;
import slimeknights.tconstruct.library.tools.stat.ToolStats;
import slimeknights.tconstruct.library.tools.stat.ToolStatsBuilder;
import slimeknights.tconstruct.library.utils.HarvestTiers;
import slimeknights.tconstruct.tools.stats.ExtraMaterialStats;
import slimeknights.tconstruct.tools.stats.HandleMaterialStats;
import slimeknights.tconstruct.tools.stats.HeadMaterialStats;
import java.util.Comparator;
import java.util.List;
/**
* Standard stat builder for melee and harvest tools. Calculates the five main stat types, and handles the bonuses for other types
*/
@Getter(AccessLevel.PROTECTED)
public final class MeleeHarvestToolStatsBuilder extends ToolStatsBuilder {
private final List<HeadMaterialStats> heads;
private final List<HandleMaterialStats> handles;
private final List<ExtraMaterialStats> extras;
@VisibleForTesting
public MeleeHarvestToolStatsBuilder(ToolDefinitionData toolData, List<HeadMaterialStats> heads, List<HandleMaterialStats> handles, List<ExtraMaterialStats> extras) {
super(toolData);
this.heads = heads;
this.handles = handles;
this.extras = extras;
}
/** Creates a builder from the definition and materials */
public static ToolStatsBuilder from(ToolDefinition toolDefinition, MaterialNBT materials) {
ToolDefinitionData data = toolDefinition.getData();
List<WeightedStatType> requiredComponents = MaterialStatsToolHook.stats(toolDefinition);
// if the NBT is invalid, at least we can return the default stats builder, as an exception here could kill itemstacks
if (materials.size() != requiredComponents.size()) {
return ToolStatsBuilder.noParts(toolDefinition);
}
return new MeleeHarvestToolStatsBuilder(data,
listOfCompatibleWith(HeadMaterialStats.ID, materials, requiredComponents),
listOfCompatibleWith(HandleMaterialStats.ID, materials, requiredComponents),
listOfCompatibleWith(ExtraMaterialStats.ID, materials, requiredComponents)
);
}
@Override
protected void setStats(StatsNBT.Builder builder) {
// add in specific stat types handled by our materials
builder.set(ToolStats.DURABILITY, buildDurability());
builder.set(ToolStats.HARVEST_TIER, buildHarvestLevel());
builder.set(ToolStats.ATTACK_DAMAGE, buildAttackDamage());
builder.set(ToolStats.ATTACK_SPEED, buildAttackSpeed());
builder.set(ToolStats.MINING_SPEED, buildMiningSpeed());
}
@Override
protected boolean handles(IToolStat<?> stat) {
return stat == ToolStats.DURABILITY || stat == ToolStats.HARVEST_TIER
|| stat == ToolStats.ATTACK_DAMAGE || stat == ToolStats.ATTACK_SPEED || stat == ToolStats.MINING_SPEED;
}
/** Builds durability for the tool */
public float buildDurability() {
double averageHeadDurability = getAverageValue(heads, HeadMaterialStats::getDurability) + getStatOrDefault(ToolStats.DURABILITY, 0f);
double averageHandleModifier = getAverageValue(handles, HandleMaterialStats::getDurability, 1);
// durability should never be below 1
return Math.max(1, (int)(averageHeadDurability * averageHandleModifier));
}
/** Builds mining speed for the tool */
public float buildMiningSpeed() {
double averageHeadSpeed = getAverageValue(heads, HeadMaterialStats::getMiningSpeed) + getStatOrDefault(ToolStats.MINING_SPEED, 0f);
double averageHandleModifier = getAverageValue(handles, HandleMaterialStats::getMiningSpeed, 1);
return (float)Math.max(0.1d, averageHeadSpeed * averageHandleModifier);
}
/** Builds attack speed for the tool */
public float buildAttackSpeed() {
float baseSpeed = toolData.getBaseStat(ToolStats.ATTACK_SPEED);
double averageHandleModifier = getAverageValue(handles, HandleMaterialStats::getAttackSpeed, 1);
return (float)Math.max(0, baseSpeed * averageHandleModifier);
}
/** Builds the harvest level for the tool */
public Tier buildHarvestLevel() {
List<Tier> sortedTiers = TierSortingRegistry.getSortedTiers();
return heads.stream()
.map(HeadMaterialStats::getTier)
.max(Comparator.comparingInt(sortedTiers::indexOf))
.orElse(HarvestTiers.minTier());
}
/** Builds attack damage for the tool */
public float buildAttackDamage() {
double averageHeadAttack = getAverageValue(heads, HeadMaterialStats::getAttack) + getStatOrDefault(ToolStats.ATTACK_DAMAGE, 0f);
double averageHandle = getAverageValue(handles, HandleMaterialStats::getAttackDamage, 1.0f);
return (float)Math.max(0.0d, averageHeadAttack * averageHandle);
}
}