github.com/arnodel/golua@v0.0.0-20230215163904-e0b5347eaaa1/runtime/arith.go

package runtime

import (
	"errors"
	"fmt"
	"math"
)

// Unm returns (z, true) where z is the value representing -x if x is a number,
// else (NilValue, false).
func Unm(x Value) (Value, bool) {
	switch x.iface.(type) {
	case int64:
		return IntValue(-x.AsInt()), true
	case float64:
		return FloatValue(-x.AsFloat()), true
	default:
		return NilValue, false
	}
}

// Add returns (z, true) where z is the value representing x+y if x and y are
// numbers, else (NilValue, false).
func Add(x, y Value) (Value, bool) {
	switch x.iface.(type) {
	case int64:
		switch y.iface.(type) {
		case int64:
			return IntValue(x.AsInt() + y.AsInt()), true
		case float64:
			return FloatValue(float64(x.AsInt()) + y.AsFloat()), true
		}
	case float64:
		switch y.iface.(type) {
		case int64:
			return FloatValue(x.AsFloat() + float64(y.AsInt())), true
		case float64:
			return FloatValue(x.AsFloat() + y.AsFloat()), true
		}
	}
	return NilValue, false
}

// Sub returns (z, true) where z is the value representing x-y if x and y are
// numbers, else (NilValue, false).
func Sub(x, y Value) (Value, bool) {
	switch x.iface.(type) {
	case int64:
		switch y.iface.(type) {
		case int64:
			return IntValue(x.AsInt() - y.AsInt()), true
		case float64:
			return FloatValue(float64(x.AsInt()) - y.AsFloat()), true
		}
	case float64:
		switch y.iface.(type) {
		case int64:
			return FloatValue(x.AsFloat() - float64(y.AsInt())), true
		case float64:
			return FloatValue(x.AsFloat() - y.AsFloat()), true
		}
	}
	return NilValue, false
}

// Mul returns (z, true) where z is the value representing x*y if x and y are
// numbers, else (NilValue, false).
func Mul(x, y Value) (Value, bool) {
	switch x.iface.(type) {
	case int64:
		switch y.iface.(type) {
		case int64:
			return IntValue(x.AsInt() * y.AsInt()), true
		case float64:
			return FloatValue(float64(x.AsInt()) * y.AsFloat()), true
		}
	case float64:
		switch y.iface.(type) {
		case int64:
			return FloatValue(x.AsFloat() * float64(y.AsInt())), true
		case float64:
			return FloatValue(x.AsFloat() * y.AsFloat()), true
		}
	}
	return NilValue, false
}

// Div returns (z, true) where z is the (float) value representing x/y if x and
// y are numbers, else (NilValue, false).
func Div(x, y Value) (Value, bool) {
	switch x.iface.(type) {
	case int64:
		switch y.iface.(type) {
		case int64:
			return FloatValue(float64(x.AsInt()) / float64(y.AsInt())), true
		case float64:
			return FloatValue(float64(x.AsInt()) / y.AsFloat()), true
		}
	case float64:
		switch y.iface.(type) {
		case int64:
			return FloatValue(x.AsFloat() / float64(y.AsInt())), true
		case float64:
			return FloatValue(x.AsFloat() / y.AsFloat()), true
		}
	}
	return NilValue, false
}

func floordivInt(x, y int64) int64 {
	r := x % y
	q := x / y
	if r != 0 && (r < 0) != (y < 0) {
		q--
	}
	return q
}

func floordivFloat(x, y float64) float64 {
	return math.Floor(x / y)
}

// Div returns (z, true, nil) where z is the (integer) value representing x//y
// if x and y are numbers and y != 0, if y == 0 it returns (NilValue, true,
// div_by_zero_error), else (NilValue, false, nil) if x or y is not a number.
func Idiv(x Value, y Value) (Value, bool, error) {
	switch x.iface.(type) {
	case int64:
		switch y.iface.(type) {
		case int64:
			ny := y.AsInt()
			if ny == 0 {
				return NilValue, true, errors.New("attempt to divide by zero")
			}
			return IntValue(floordivInt(x.AsInt(), ny)), true, nil
		case float64:
			return FloatValue(floordivFloat(float64(x.AsInt()), y.AsFloat())), true, nil
		}
	case float64:
		switch y.iface.(type) {
		case int64:
			return FloatValue(floordivFloat(x.AsFloat(), float64(y.AsInt()))), true, nil
		case float64:
			return FloatValue(floordivFloat(x.AsFloat(), y.AsFloat())), true, nil
		}
	}
	return NilValue, false, nil
}

func modInt(x, y int64) int64 {
	r := x % y
	if r != 0 && (r < 0) != (y < 0) {
		r += y
	}
	return r
}

func modFloat(x, y float64) float64 {
	r := math.Mod(x, y)
	if r != 0 && (r < 0) != (y < 0) {
		r += y
	}
	return r
}

// Mod returns (z, true, nil) where z is the (integer or float) value
// representing x%y if x and y are numbers and y != 0, if y == 0 it returns
// (NilValue, true, mod_by_zero_error), else (NilValue, false, nil) if x or y is
// not a number.
func Mod(x Value, y Value) (Value, bool, error) {
	switch x.iface.(type) {
	case int64:
		switch y.iface.(type) {
		case int64:
			ny := y.AsInt()
			if ny == 0 {
				return NilValue, true, errors.New("attempt to perform 'n%0'")
			}
			return IntValue(modInt(x.AsInt(), ny)), true, nil
		case float64:
			return FloatValue(modFloat(float64(x.AsInt()), y.AsFloat())), true, nil
		}
	case float64:
		switch y.iface.(type) {
		case int64:
			return FloatValue(modFloat(x.AsFloat(), float64(y.AsInt()))), true, nil
		case float64:
			return FloatValue(modFloat(x.AsFloat(), y.AsFloat())), true, nil
		}
	}
	return NilValue, false, nil
}

func powFloat(x, y float64) float64 {
	return math.Pow(x, y)
}

// Pow returns (z, true) where z is the (float) value representing x^y if x and
// y are numbers, else (NilValue, false).
func Pow(x, y Value) (Value, bool) {
	var fx, fy float64
	switch x.iface.(type) {
	case int64:
		fx = float64(x.AsInt())
	case float64:
		fx = x.AsFloat()
	default:
		return NilValue, false
	}
	switch y.iface.(type) {
	case int64:
		fy = float64(y.AsInt())
	case float64:
		fy = y.AsFloat()
	default:
		return NilValue, false
	}
	return FloatValue(powFloat(fx, fy)), true
}

func binaryArithFallback(t *Thread, op string, x, y Value) (Value, error) {
	res, err, ok := metabin(t, op, x, y)
	if ok {
		return res, err
	}
	return NilValue, BinaryArithmeticError(op[2:], x, y)
}

// BinaryArithmeticError returns an error describing the problem with trying to
// perform x op y.
func BinaryArithmeticError(op string, x, y Value) error {
	var wrongVal Value
	switch {
	case numberType(y) != NaN:
		wrongVal = x
	case numberType(x) != NaN:
		wrongVal = y
	default:
		return fmt.Errorf("attempt to %s a '%s' with a '%s'", op, x.CustomTypeName(), y.CustomTypeName())
	}
	return fmt.Errorf("attempt to perform arithmetic on a %s value", wrongVal.CustomTypeName())
}

func unaryArithFallback(t *Thread, op string, x Value) (Value, error) {
	res, err, ok := metaun(t, op, x)
	if ok {
		return res, err
	}
	return NilValue, UnaryArithmeticError(op[2:], x)
}

// UnaryArithmeticError returns an error describing the problem with trying to
// perform the unary operation op(x).
func UnaryArithmeticError(op string, x Value) error {
	return fmt.Errorf("attempt to %s a '%s'", op, x.CustomTypeName())
}