/*
 * Copyright (c) 2019-2022 GeyserMC. http://geysermc.org
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 *
 * @author GeyserMC
 * @link https://github.com/GeyserMC/Geyser
 */

package org.geysermc.geyser.util;

import org.cloudburstmc.math.TrigMath;
import org.cloudburstmc.math.vector.Vector3f;

public class MathUtils {
    public static final double SQRT_OF_TWO = Math.sqrt(2);

    public static Vector3f xYRot(Vector3f velocity, float pitch, float yaw) {
        float pitchCos = TrigMath.cos(pitch);
        float pitchSin = TrigMath.sin(pitch);
        float yawCos = TrigMath.cos(yaw);
        float yawSin = TrigMath.sin(yaw);

        float e = velocity.getY() * pitchCos + velocity.getZ() * pitchSin;
        float i = velocity.getZ() * pitchCos - velocity.getY() * pitchSin;
        velocity = Vector3f.from(velocity.getX(), e, i);

        float d1 = velocity.getX() * yawCos + velocity.getZ() * yawSin;
        float i1 = velocity.getZ() * yawCos - velocity.getX() * yawSin;

        return Vector3f.from(d1, e, i1);
    }

    public static Vector3f calculateViewVector(float pitch, float yaw) {
        float var3 = pitch * 0.017453292F;
        float var4 = -yaw * 0.017453292F;
        float var5 = TrigMath.cos(var4);
        float var6 = TrigMath.sin(var4);
        float var7 = TrigMath.cos(var3);
        float var8 = TrigMath.sin(var3);
        return Vector3f.from(var6 * var7, -var8, var5 * var7);
    }

    /**
     * Wrap the given float degrees to be between -180.0 and 180.0.
     * 
     * @param degrees The degrees value to wrap
     * @return The wrapped degrees value between -180.0 and 180.0
     */
    public static float wrapDegrees(float degrees) {
        degrees = degrees % 360.0f;
        if (degrees < -180.0f) {
            degrees += 360.0f;
        } else if (degrees >= 180.0f) {
            degrees -= 360.0f;
        }
        return degrees;
    }

    /**
     * Wrap the given double degrees to be between -180.0 and 180.0.
     * 
     * @param degrees The degrees value to wrap
     * @return The wrapped degrees value between -180.0 and 180.0
     */
    public static float wrapDegrees(double degrees) {
        return wrapDegrees((float) degrees);
    }

    /**
     * Wrap the given degrees to be between -180 and 180 as an integer.
     * 
     * @param degrees The degrees value to wrap
     * @return The wrapped degrees value between -180 and 180 as an integer
     */
    public static int wrapDegreesToInt(float degrees) {
        return (int) wrapDegrees(degrees);
    }

    /**
     * Unwrap the given float degrees to be between 0.0 and 360.0.
     * 
     * @param degrees The degrees value to unwrap
     * @return The unwrapped degrees value between 0.0 and 360.0
     */
    public static float unwrapDegrees(float degrees) {
        return (degrees % 360 + 360) % 360;
    }

    /**
     * Unwrap the given double degrees to be between 0.0 and 360.0.
     * 
     * @param degrees The degrees value to unwrap
     * @return The unwrapped degrees value between 0.0 and 360.0
     */
    public static float unwrapDegrees(double degrees) {
        return unwrapDegrees((float) degrees);
    }

    /**
     * Unwrap the given degrees to be between 0 and 360 as an integer.
     * 
     * @param degrees The degrees value to unwrap
     * @return The unwrapped degrees value between 0 and 360 as an integer
     */
    public static int unwrapDegreesToInt(float degrees) {
        return (int) unwrapDegrees(degrees);
    }

    /**
     * Round the given float to the next whole number
     *
     * @param floatNumber Float to round
     * @return Rounded number
     */
    public static int ceil(float floatNumber) {
        int truncated = (int) floatNumber;
        return floatNumber > truncated ? truncated + 1 : truncated;
    }

    /**
     * If number is greater than the max, set it to max, and if number is lower than low, set it to low.
     *
     * @param num number to calculate
     * @param min the lowest value the number can be
     * @param max the greatest value the number can be
     * @return - min if num is lower than min <br>
     * - max if num is greater than max <br>
     * - num otherwise
     */
    public static double constrain(double num, double min, double max) {
        if (num > max) {
            num = max;
        }

        if (num < min) {
            num = min;
        }

        return num;
    }

    /**
     * If number is greater than the max, set it to max, and if number is lower than low, set it to low.
     *
     * @param num number to calculate
     * @param min the lowest value the number can be
     * @param max the greatest value the number can be
     * @return - min if num is lower than min <br>
     * - max if num is greater than max <br>
     * - num otherwise
     */
    public static int constrain(int num, int min, int max) {
        if (num > max) {
            num = max;
        }

        if (num < min) {
            num = min;
        }

        return num;
    }

    /**
     * Clamps the value between 0 and the given maximum.
     *
     * @param x the value to clamp
     * @param max the highest value the result can be
     * @return 0 if x is less than 0, max if x is greater than max, or x otherwise
     */
    public static double restrain(double x, double max) {
        return constrain(x, 0, max);
    }

    /**
     * Clamps the value between the low and high boundaries
     * Copied from {@link org.cloudburstmc.math.GenericMath} with floats instead.
     *
     * @param value The value to clamp
     * @param low The low bound of the clamp
     * @param high The high bound of the clamp
     * @return the clamped value
     */
    public static float clamp(float value, float low, float high) {
        if (value < low) {
            return low;
        }
        return Math.min(value, high);
    }

    /**
     * Packs a chunk's X and Z coordinates into a single {@code long}.
     *
     * @param x the X coordinate
     * @param z the Z coordinate
     * @return the packed coordinates
     */
    public static long chunkPositionToLong(int x, int z) {
        return ((x & 0xFFFFFFFFL) << 32L) | (z & 0xFFFFFFFFL);
    }
}
