superb-warfare/src/main/java/com/atsuishio/superbwarfare/perk/AmmoPerk.java
2025-05-26 04:52:54 +08:00

130 lines
4.3 KiB
Java

package com.atsuishio.superbwarfare.perk;
import com.atsuishio.superbwarfare.data.gun.DamageReduce;
import com.atsuishio.superbwarfare.data.gun.GunData;
import com.atsuishio.superbwarfare.entity.projectile.ProjectileEntity;
import net.minecraft.core.Holder;
import net.minecraft.util.Mth;
import net.minecraft.world.effect.MobEffect;
import net.minecraft.world.effect.MobEffectInstance;
import net.minecraft.world.entity.Entity;
import java.util.ArrayList;
import java.util.function.Supplier;
public class AmmoPerk extends Perk {
public float bypassArmorRate;
public float damageRate;
public float speedRate;
public boolean slug;
public float[] rgb;
public Supplier<ArrayList<MobEffect>> mobEffects;
public AmmoPerk(Builder builder) {
super(builder.descriptionId, builder.type);
this.bypassArmorRate = builder.bypassArmorRate;
this.damageRate = builder.damageRate;
this.speedRate = builder.speedRate;
this.slug = builder.slug;
this.rgb = builder.rgb;
this.mobEffects = () -> builder.mobEffects;
}
public AmmoPerk(String descriptionId, Type type) {
super(descriptionId, type);
this.rgb = new float[]{1, 222 / 255f, 39 / 255f};
this.mobEffects = ArrayList::new;
}
@Override
public void modifyProjectile(GunData data, PerkInstance instance, Entity entity) {
if (!(entity instanceof ProjectileEntity projectile)) return;
projectile.setRGB(this.rgb);
projectile.bypassArmorRate((float) Math.max(this.bypassArmorRate + data.bypassArmor(), 0));
if (this.slug) {
projectile.setDamage((float) (data.damage() * data.projectileAmount()));
}
if (!this.mobEffects.get().isEmpty()) {
int amplifier = this.getEffectAmplifier(instance);
ArrayList<MobEffectInstance> mobEffectInstances = new ArrayList<>();
for (MobEffect effect : this.mobEffects.get()) {
mobEffectInstances.add(new MobEffectInstance(Holder.direct(effect), 70 + 30 * level, amplifier));
}
projectile.effect(mobEffectInstances);
}
}
public int getEffectAmplifier(PerkInstance instance) {
return instance.level() - 1;
}
public double getModifiedVelocity(GunData data, PerkInstance instance) {
return data.velocity() * this.speedRate;
}
@Override
public double getModifiedDamageReduceRate(DamageReduce reduce) {
if (this.slug && reduce.type == DamageReduce.ReduceType.SHOTGUN) {
return 0.015;
}
return super.getModifiedDamageReduceRate(reduce);
}
@Override
public double getModifiedDamageReduceMinDistance(DamageReduce reduce) {
if (this.slug && reduce.type == DamageReduce.ReduceType.SHOTGUN) {
return super.getModifiedDamageReduceMinDistance(reduce) * 2;
}
return super.getModifiedDamageReduceMinDistance(reduce);
}
public static class Builder {
String descriptionId;
Type type;
float bypassArmorRate = 0.0f;
float damageRate = 1.0f;
float speedRate = 1.0f;
boolean slug = false;
float[] rgb = {1, 222 / 255f, 39 / 255f};
public ArrayList<MobEffect> mobEffects = new ArrayList<>();
public Builder(String descriptionId, Type type) {
this.descriptionId = descriptionId;
this.type = type;
}
public Builder bypassArmorRate(float bypassArmorRate) {
this.bypassArmorRate = Mth.clamp(bypassArmorRate, -1, 1);
return this;
}
public Builder damageRate(float damageRate) {
this.damageRate = Mth.clamp(damageRate, 0, Float.POSITIVE_INFINITY);
return this;
}
public Builder speedRate(float speedRate) {
this.speedRate = Mth.clamp(speedRate, 0, Float.POSITIVE_INFINITY);
return this;
}
public Builder slug(boolean slug) {
this.slug = slug;
return this;
}
public Builder rgb(int r, int g, int b) {
this.rgb[0] = r / 255f;
this.rgb[1] = g / 255f;
this.rgb[2] = b / 255f;
return this;
}
public Builder mobEffect(Supplier<MobEffect> mobEffect) {
this.mobEffects.add(mobEffect.get());
return this;
}
}
}