-
Notifications
You must be signed in to change notification settings - Fork 754
/
RicochetModifier.java
36 lines (32 loc) · 1.53 KB
/
RicochetModifier.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
package slimeknights.tconstruct.tools.modifiers.upgrades.armor;
import net.minecraftforge.common.MinecraftForge;
import net.minecraftforge.event.entity.living.LivingKnockBackEvent;
import slimeknights.tconstruct.TConstruct;
import slimeknights.tconstruct.common.TinkerTags;
import slimeknights.tconstruct.library.modifiers.Modifier;
import slimeknights.tconstruct.library.modifiers.modules.unserializable.ArmorLevelModule;
import slimeknights.tconstruct.library.module.ModuleHookMap.Builder;
import slimeknights.tconstruct.library.tools.capability.TinkerDataCapability;
import slimeknights.tconstruct.library.tools.capability.TinkerDataCapability.TinkerDataKey;
public class RicochetModifier extends Modifier {
private static final TinkerDataKey<Integer> LEVELS = TConstruct.createKey("ricochet");
public RicochetModifier() {
// TODO: move this out of constructor to generalized logic
MinecraftForge.EVENT_BUS.addListener(this::livingKnockback);
}
@Override
protected void registerHooks(Builder hookBuilder) {
super.registerHooks(hookBuilder);
hookBuilder.addModule(new ArmorLevelModule(LEVELS, false, TinkerTags.Items.HELD_ARMOR));
}
/** Called on knockback to adjust player knockback */
private void livingKnockback(LivingKnockBackEvent event) {
event.getEntity().getCapability(TinkerDataCapability.CAPABILITY).ifPresent(data -> {
int levels = data.get(LEVELS, 0);
if (levels > 0) {
// adds +20% knockback per level
event.setStrength(event.getStrength() * (1 + levels * 0.2f));
}
});
}
}