com.yungnickyoung.minecraft.yungsapi.math.ColPos Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of YungsApi Show documentation
A common API for YUNG's Minecraft mods
The newest version!
package com.yungnickyoung.minecraft.yungsapi.math;

import net.minecraft.core.BlockPos;
import net.minecraft.core.Direction;
import net.minecraft.world.level.block.Rotation;

/**
 * Similar to the BlockPos class, but only provides two axes (x and z).
 * Designed to work well with BlockPos.
 */
public class ColPos {
    protected int x, z;

    private static final int NUM_X_BITS = 32;
    private static final int NUM_Z_BITS = NUM_X_BITS;
    private static final long X_MASK = (1L << NUM_X_BITS) - 1L;
    private static final long Z_MASK = (1L << NUM_Z_BITS) - 1L;

    public ColPos() {
        this(0, 0);
    }

    public ColPos(int x, int z) {
        this.x = x;
        this.z = z;
    }

    public ColPos(BlockPos source) {
        this.x = source.getX();
        this.z = source.getZ();
    }

    public int getX() {
        return x;
    }

    public int getZ() {
        return z;
    }

    public long toLong() {
        long i = 0L;
        i = i | ((long)this.getX() & X_MASK);
        i = i | ((long)this.getZ() & Z_MASK) << NUM_X_BITS;
        return i;
    }

    public static ColPos fromLong(long packedPos) {
        int x = (int)(packedPos << 64 - NUM_X_BITS >> 64 - NUM_X_BITS);
        int z = (int)(packedPos >> 64 - NUM_Z_BITS);
        return new ColPos(x, z);
    }

    public ColPos up() {
        return this.up(1);
    }

    public ColPos up(int n) {
        return this.offset(Direction.UP, n);
    }

    public ColPos down() {
        return this.down(1);
    }

    public ColPos down(int n) {
        return this.offset(Direction.DOWN, n);
    }

    public ColPos north() {
        return this.north(1);
    }

    public ColPos north(int n) {
        return this.offset(Direction.NORTH, n);
    }

    public ColPos south() {
        return this.south(1);
    }

    public ColPos south(int n) {
        return this.offset(Direction.SOUTH, n);
    }

    public ColPos west() {
        return this.west(1);
    }

    public ColPos west(int n) {
        return this.offset(Direction.WEST, n);
    }

    public ColPos east() {
        return this.east(1);
    }

    public ColPos east(int n) {
        return this.offset(Direction.EAST, n);
    }

    public ColPos offset(Direction facing) {
        return this.offset(facing, 1);
    }

    public ColPos offset(Direction facing, int n) {
        return n == 0 ? this : new ColPos(this.getX() + facing.getStepX() * n, this.getZ() + facing.getStepZ() * n);
    }

    public ColPos rotate(Rotation rotationIn) {
        switch(rotationIn) {
            case NONE:
            default:
                return this;
            case CLOCKWISE_90:
                return new ColPos(-this.getZ(), this.getX());
            case CLOCKWISE_180:
                return new ColPos(-this.getX(), -this.getZ());
            case COUNTERCLOCKWISE_90:
                return new ColPos(this.getZ(), -this.getX());
        }
    }

    public BlockPos toBlockPos() {
        return new BlockPos(getX(), 1, getZ());
    }

    public static ColPos fromBlockPos(BlockPos blockPos) {
        return new ColPos(blockPos);
    }

    public static class Mutable extends ColPos {
        public Mutable() {
            super();
        }

        public Mutable(BlockPos source) {
            this(source.getX(), source.getZ());
        }

        public Mutable(ColPos pos) {
            this(pos.getX(), pos.getZ());
        }

        public Mutable(int x, int z) {
            super(x, z);
        }

        public Mutable set(int x, int z) {
            this.x = x;
            this.z = z;
            return this;
        }

        public Mutable set(ColPos source) {
            set(source.getX(), source.getZ());
            return this;
        }

        public Mutable set(BlockPos source) {
            set(source.getX(), source.getZ());
            return this;
        }

        public Mutable move(Direction facing, int n) {
            return this.set(this.x + facing.getStepX() * n, this.z + facing.getStepZ() * n);
        }

        public Mutable move(Direction facing) {
            return this.move(facing, 1);
        }

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

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