minecraft-kajetans-mod/src/main/java/me/km/utils/Location.java

package me.km.utils;

import java.util.Objects;
import java.util.regex.PatternSyntaxException;
import net.minecraft.block.BlockState;
import net.minecraft.entity.Entity;
import net.minecraft.server.MinecraftServer;
import net.minecraft.util.RegistryKey;
import net.minecraft.util.ResourceLocation;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.MathHelper;
import net.minecraft.util.math.vector.Vector3d;
import net.minecraft.util.registry.Registry;
import net.minecraft.world.IWorld;
import net.minecraft.world.World;

public class Location {
    private final World w;
    private double x;
    private double y;
    private double z;
    private float yaw;
    private float pitch;

    public Location(World w, double x, double y, double z, float yaw, float pitch) {
        this.w = w;
        this.x = x;
        this.y = y;
        this.z = z;
        this.yaw = yaw;
        this.pitch = pitch;
    }

    public Location(World w, Vector3d pos, float yaw, float pitch) {
        this(w, pos.x, pos.y, pos.z, yaw, pitch);
    }

    public Location(World w, Vector3d pos) {
        this(w, pos, 0, 0);
    }

    public Location(World w, BlockPos pos, float yaw, float pitch) {
        this(w, pos.getX(), pos.getY(), pos.getZ(), yaw, pitch);
    }

    public Location(World w, BlockPos pos) {
        this(w, pos, 0, 0);
    }

    public Location(Entity ent) {
        this(ent.world, ent.getPosX(), ent.getPosY(), ent.getPosZ(), ent.rotationYaw, ent.rotationPitch);
    }

    public Location(MinecraftServer server, String location) {
        try {
            String[] parts = location.split(":");
            if(parts.length >= 6) {
                this.w = Utils.getWorldFromName(server, location);
                this.x = Double.parseDouble(parts[1]);
                this.y = Double.parseDouble(parts[2]);
                this.z = Double.parseDouble(parts[3]);
                this.yaw = Float.parseFloat(parts[4]);
                this.pitch = Float.parseFloat(parts[5]);
                return;
            } else if(parts.length >= 4) {
                this.w = Utils.getWorldFromName(server, parts[0]);
                this.x = Double.parseDouble(parts[1]);
                this.y = Double.parseDouble(parts[2]);
                this.z = Double.parseDouble(parts[3]);
                this.yaw = 0;
                this.pitch = 0;
                return;
            }
            throw new IllegalArgumentException(String.format("'%s' is not a valid location string", location));
        } catch(PatternSyntaxException | NumberFormatException ex) {
            throw new IllegalArgumentException(String.format("'%s' is not a valid location string", location));
        }
    }

    @Override
    public int hashCode() {
        int hash = 7;
        hash = 17 * hash + Objects.hashCode(this.w);
        hash = 17 * hash + (int) (Double.doubleToLongBits(this.x) ^ (Double.doubleToLongBits(this.x) >>> 32));
        hash = 17 * hash + (int) (Double.doubleToLongBits(this.y) ^ (Double.doubleToLongBits(this.y) >>> 32));
        hash = 17 * hash + (int) (Double.doubleToLongBits(this.z) ^ (Double.doubleToLongBits(this.z) >>> 32));
        return hash;
    }

    @Override
    public boolean equals(Object obj) {
        if(this == obj) {
            return true;
        }
        if(obj == null || getClass() != obj.getClass()) {
            return false;
        }
        final Location other = (Location) obj;
        return Double.doubleToLongBits(this.x) == Double.doubleToLongBits(other.x)
                && Double.doubleToLongBits(this.y) == Double.doubleToLongBits(other.y)
                && Double.doubleToLongBits(this.z) == Double.doubleToLongBits(other.z)
                && Objects.equals(this.w, other.w);
    }

    public void add(double x, double y, double z) {
        this.x += x;
        this.y += y;
        this.z += z;
    }

    public Location copyAdd(double x, double y, double z) {
        return new Location(w, this.x + x, this.y + y, this.z + z, this.yaw, this.pitch);
    }

    public void addX(double x) {
        this.x += x;
    }

    public void addY(double y) {
        this.y += y;
    }

    public void addZ(double z) {
        this.z += z;
    }

    public void setX(double x) {
        this.x = x;
    }

    public void setY(double y) {
        this.y = y;
    }

    public void setZ(double z) {
        this.z = z;
    }

    public void set(double x, double y, double z) {
        this.x = x;
        this.y = y;
        this.z = z;
    }

    public double getX() {
        return x;
    }

    public double getY() {
        return y;
    }

    public double getZ() {
        return z;
    }

    public void round() {
        this.x = MathHelper.floor(this.x);
        this.y = MathHelper.floor(this.y);
        this.z = MathHelper.floor(this.z);
    }

    public World getWorld() {
        return w;
    }

    public Vector3d getPos() {
        return new Vector3d(x, y, z);
    }

    public BlockPos getBlockPos() {
        return new BlockPos(x, y, z);
    }

    public float getYaw() {
        return yaw;
    }

    public float getPitch() {
        return pitch;
    }

    public void setPitch(float pitch) {
        this.pitch = pitch;
    }

    public void setYaw(float yaw) {
        this.yaw = yaw;
    }

    public BlockState getBlockState() {
        return w.getBlockState(new BlockPos(x, y, z));
    }

    public BlockState getRelativeBlockState(int x, int y, int z) {
        return w.getBlockState(new BlockPos(this.x + x, this.y + y, this.z + z));
    }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder(Utils.getWorldName(w));
        sb.append(":");
        sb.append(x);
        sb.append(":");
        sb.append(y);
        sb.append(":");
        sb.append(z);
        sb.append(":");
        sb.append(yaw);
        sb.append(":");
        sb.append(pitch);
        return sb.toString();
    }

    public String toBlockString() {
        StringBuilder sb = new StringBuilder(Utils.getWorldName(w));
        sb.append(":");
        sb.append(MathHelper.floor(x));
        sb.append(":");
        sb.append(MathHelper.floor(y));
        sb.append(":");
        sb.append(MathHelper.floor(z));
        return sb.toString();
    }

    public Location copy() {
        return new Location(w, x, y, z, yaw, pitch);
    }
}