org.jruby.ext.bigdecimal.RubyBigDecimal Maven / Gradle / Ivy
/***** BEGIN LICENSE BLOCK *****
* Version: EPL 1.0/GPL 2.0/LGPL 2.1
*
* The contents of this file are subject to the Eclipse Public
* License Version 1.0 (the "License"); you may not use this file
* except in compliance with the License. You may obtain a copy of
* the License at http://www.eclipse.org/legal/epl-v10.html
*
* Software distributed under the License is distributed on an "AS
* IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
* implied. See the License for the specific language governing
* rights and limitations under the License.
*
* Copyright (C) 2006 Ola Bini
* Copyright (C) 2009 Joseph LaFata
*
* Alternatively, the contents of this file may be used under the terms of
* either of the GNU General Public License Version 2 or later (the "GPL"),
* or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
* in which case the provisions of the GPL or the LGPL are applicable instead
* of those above. If you wish to allow use of your version of this file only
* under the terms of either the GPL or the LGPL, and not to allow others to
* use your version of this file under the terms of the EPL, indicate your
* decision by deleting the provisions above and replace them with the notice
* and other provisions required by the GPL or the LGPL. If you do not delete
* the provisions above, a recipient may use your version of this file under
* the terms of any one of the EPL, the GPL or the LGPL.
***** END LICENSE BLOCK *****/
package org.jruby.ext.bigdecimal;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.math.MathContext;
import java.math.RoundingMode;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.jruby.CompatVersion;
import org.jruby.Ruby;
import org.jruby.RubyArray;
import org.jruby.RubyBignum;
import org.jruby.RubyBoolean;
import org.jruby.RubyClass;
import org.jruby.RubyFixnum;
import org.jruby.RubyFloat;
import org.jruby.RubyInteger;
import org.jruby.RubyModule;
import org.jruby.RubyNumeric;
import org.jruby.RubyObject;
import org.jruby.RubyRational;
import org.jruby.RubyString;
import org.jruby.RubySymbol;
import org.jruby.anno.JRubyClass;
import org.jruby.anno.JRubyConstant;
import org.jruby.anno.JRubyMethod;
import org.jruby.runtime.Arity;
import org.jruby.runtime.Block;
import org.jruby.runtime.ObjectAllocator;
import org.jruby.runtime.ThreadContext;
import org.jruby.runtime.Visibility;
import org.jruby.runtime.builtin.IRubyObject;
import org.jruby.util.Numeric;
import org.jruby.util.SafeDoubleParser;
/**
* @author Ola Bini
*/
@JRubyClass(name="BigDecimal", parent="Numeric")
public class RubyBigDecimal extends RubyNumeric {
private static final ObjectAllocator ALLOCATOR = new ObjectAllocator() {
public RubyBigDecimal allocate(Ruby runtime, RubyClass klass) {
return new RubyBigDecimal(runtime, klass);
}
};
@JRubyConstant
public final static int ROUND_DOWN = BigDecimal.ROUND_DOWN;
@JRubyConstant
public final static int ROUND_CEILING = BigDecimal.ROUND_CEILING;
@JRubyConstant
public final static int ROUND_UP = BigDecimal.ROUND_UP;
@JRubyConstant
public final static int ROUND_HALF_DOWN = BigDecimal.ROUND_HALF_DOWN;
@JRubyConstant
public final static int ROUND_HALF_EVEN = BigDecimal.ROUND_HALF_EVEN;
@JRubyConstant
public final static int ROUND_HALF_UP = BigDecimal.ROUND_HALF_UP;
@JRubyConstant
public final static int ROUND_FLOOR = BigDecimal.ROUND_FLOOR;
@JRubyConstant
public final static int SIGN_POSITIVE_INFINITE = 3;
@JRubyConstant
public final static int EXCEPTION_OVERFLOW = 8;
@JRubyConstant
public final static int SIGN_POSITIVE_ZERO = 1;
@JRubyConstant
public final static int EXCEPTION_ALL = 255;
@JRubyConstant
public final static int SIGN_NEGATIVE_FINITE = -2;
@JRubyConstant
public final static int EXCEPTION_UNDERFLOW = 4;
@JRubyConstant
public final static int SIGN_NaN = 0;
@JRubyConstant
public final static int BASE = 10000;
@JRubyConstant
public final static int ROUND_MODE = 256;
@JRubyConstant
public final static int SIGN_POSITIVE_FINITE = 2;
@JRubyConstant
public final static int EXCEPTION_INFINITY = 1;
@JRubyConstant
public final static int SIGN_NEGATIVE_INFINITE = -3;
@JRubyConstant
public final static int EXCEPTION_ZERODIVIDE = 1;
@JRubyConstant
public final static int SIGN_NEGATIVE_ZERO = -1;
@JRubyConstant
public final static int EXCEPTION_NaN = 2;
// Static constants
private static final BigDecimal TWO = new BigDecimal(2);
private static final double SQRT_10 = 3.162277660168379332;
public static RubyClass createBigDecimal(Ruby runtime) {
RubyClass bigDecimal = runtime.defineClass("BigDecimal", runtime.getNumeric(), ALLOCATOR);
runtime.getKernel().defineAnnotatedMethods(BigDecimalKernelMethods.class);
bigDecimal.setInternalModuleVariable("vpPrecLimit", RubyFixnum.zero(runtime));
bigDecimal.setInternalModuleVariable("vpExceptionMode", RubyFixnum.zero(runtime));
bigDecimal.setInternalModuleVariable("vpRoundingMode", runtime.newFixnum(ROUND_HALF_UP));
bigDecimal.defineAnnotatedMethods(RubyBigDecimal.class);
bigDecimal.defineAnnotatedConstants(RubyBigDecimal.class);
if (runtime.is1_9()) {
RubyModule bigMath = runtime.defineModule("BigMath");
// TODO: BigMath.exp and BigMath.pow in native code
bigDecimal.defineConstant("NAN", newNaN(runtime));
bigDecimal.defineConstant("INFINITY", newInfinity(runtime, 1));
}
return bigDecimal;
}
private final boolean isNaN;
private final int infinitySign;
private final int zeroSign;
private BigDecimal value;
public BigDecimal getValue() {
return value;
}
public RubyBigDecimal(Ruby runtime, RubyClass klass) {
super(runtime, klass);
this.isNaN = false;
this.infinitySign = 0;
this.zeroSign = 0;
this.value = BigDecimal.ZERO;
}
public RubyBigDecimal(Ruby runtime, BigDecimal value) {
super(runtime, runtime.getClass("BigDecimal"));
this.isNaN = false;
this.infinitySign = 0;
this.zeroSign = 0;
this.value = value;
}
public RubyBigDecimal(Ruby runtime, RubyClass klass, BigDecimal value) {
super(runtime, klass);
this.isNaN = false;
this.infinitySign = 0;
this.zeroSign = 0;
this.value = value;
}
public RubyBigDecimal(Ruby runtime, BigDecimal value, int infinitySign) {
super(runtime, runtime.getClass("BigDecimal"));
this.isNaN = false;
this.infinitySign = infinitySign;
this.zeroSign = 0;
this.value = value;
}
public RubyBigDecimal(Ruby runtime, BigDecimal value, int infinitySign, int zeroSign) {
super(runtime, runtime.getClass("BigDecimal"));
this.isNaN = false;
this.infinitySign = infinitySign;
this.zeroSign = zeroSign;
this.value = value;
}
public RubyBigDecimal(Ruby runtime, BigDecimal value, boolean isNan) {
super(runtime, runtime.getClass("BigDecimal"));
this.isNaN = isNan;
this.infinitySign = 0;
this.zeroSign = 0;
this.value = value;
}
public RubyBigDecimal(Ruby runtime, RubyBigDecimal rbd) {
this(runtime, runtime.getClass("BigDecimal"), rbd);
}
public RubyBigDecimal(Ruby runtime, RubyClass klass, RubyBigDecimal rbd) {
super(runtime, klass);
this.isNaN = rbd.isNaN;
this.infinitySign = rbd.infinitySign;
this.zeroSign = rbd.zeroSign;
this.value = rbd.value;
}
public static class BigDecimalKernelMethods {
@JRubyMethod(name = "BigDecimal", required = 1, optional = 1, module = true, visibility = Visibility.PRIVATE)
public static IRubyObject newBigDecimal(ThreadContext context, IRubyObject recv, IRubyObject[] args) {
if (args.length == 1) return newInstance(context, context.runtime.getClass("BigDecimal"), args[0]);
return newInstance(context, context.runtime.getClass("BigDecimal"), args[0], args[1]);
}
}
public static RubyBigDecimal newBigDecimal(IRubyObject recv, IRubyObject[] args, Block unusedBlock) {
return newInstance(recv.getRuntime().getClass("BigDecimal"), args);
}
@JRubyMethod(name = "ver", meta = true)
public static IRubyObject ver(IRubyObject recv) {
return recv.getRuntime().newString("1.0.1");
}
@JRubyMethod(name = "_dump", optional = 1)
public IRubyObject dump(IRubyObject[] args, Block unusedBlock) {
RubyString precision = RubyString.newUnicodeString(getRuntime(), "0:");
return precision.append(asString());
}
@JRubyMethod(name = "_load", required = 1, meta = true)
public static RubyBigDecimal load(IRubyObject recv, IRubyObject from, Block block) {
RubyBigDecimal rubyBigDecimal = (RubyBigDecimal) (((RubyClass) recv).allocate());
String precisionAndValue = from.convertToString().asJavaString();
String value = precisionAndValue.substring(precisionAndValue.indexOf(':') + 1);
rubyBigDecimal.value = new BigDecimal(value);
return rubyBigDecimal;
}
@JRubyMethod(name = "double_fig", meta = true)
public static IRubyObject double_fig(IRubyObject recv) {
return recv.getRuntime().newFixnum(20);
}
@JRubyMethod(name = "limit", optional = 1, meta = true)
public static IRubyObject limit(final IRubyObject self, IRubyObject[] args) {
final RubyModule BigDecimal = (RubyModule) self;
IRubyObject current = BigDecimal.searchInternalModuleVariable("vpPrecLimit");
if (args.length > 0) {
IRubyObject arg = args[0];
if (!arg.isNil()) {
final Ruby runtime = self.getRuntime();
if (!(arg instanceof RubyFixnum)) {
throw runtime.newTypeError(arg, runtime.getFixnum());
}
if (0 > ((RubyFixnum)arg).getLongValue()) {
throw runtime.newArgumentError("argument must be positive");
}
BigDecimal.setInternalModuleVariable("vpPrecLimit", arg);
}
}
return current;
}
@JRubyMethod(name = "save_limit", meta = true)
public static IRubyObject save_limit(ThreadContext context, IRubyObject self, Block block) {
return saveVariable(context, (RubyModule) self, block, "vpPrecLimit");
}
@JRubyMethod(name = "save_exception_mode", meta = true)
public static IRubyObject save_exception_mode(ThreadContext context, IRubyObject self, Block block) {
return saveVariable(context, (RubyModule) self, block, "vpExceptionMode");
}
@JRubyMethod(name = "save_rounding_mode", meta = true)
public static IRubyObject save_rounding_mode(ThreadContext context, IRubyObject self, Block block) {
return saveVariable(context, (RubyModule) self, block, "vpRoundingMode");
}
private static IRubyObject saveVariable(final ThreadContext context, final RubyModule BigDecimal,
final Block block, final String intVariableName) {
IRubyObject current = BigDecimal.searchInternalModuleVariable(intVariableName);
try {
return block.yieldSpecific(context);
}
finally {
BigDecimal.setInternalModuleVariable(intVariableName, current);
}
}
@JRubyMethod(name = "mode", required = 1, optional = 1, meta = true)
public static IRubyObject mode(ThreadContext context, IRubyObject recv, IRubyObject[] args) {
// FIXME: I doubt any of the constants referenced in this method
// are ever redefined -- should compare to the known values, rather
// than do an expensive constant lookup.
Ruby runtime = recv.getRuntime();
RubyClass clazz = runtime.getClass("BigDecimal");
RubyModule c = (RubyModule)recv;
args = Arity.scanArgs(runtime, args, 1, 1);
IRubyObject mode = args[0];
IRubyObject value = args[1];
if (!(mode instanceof RubyFixnum)) {
throw runtime.newTypeError("wrong argument type " + mode.getMetaClass() + " (expected Fixnum)");
}
long longMode = ((RubyFixnum)mode).getLongValue();
long EXCEPTION_ALL = ((RubyFixnum)clazz.getConstant("EXCEPTION_ALL")).getLongValue();
if ((longMode & EXCEPTION_ALL) != 0) {
if (value.isNil()) {
return c.searchInternalModuleVariable("vpExceptionMode");
}
if (!(value.isNil()) && !(value instanceof RubyBoolean)) {
throw runtime.newArgumentError("second argument must be true or false");
}
RubyFixnum currentExceptionMode = (RubyFixnum)c.searchInternalModuleVariable("vpExceptionMode");
RubyFixnum newExceptionMode = new RubyFixnum(runtime, currentExceptionMode.getLongValue());
RubyFixnum EXCEPTION_INFINITY = (RubyFixnum)clazz.getConstant("EXCEPTION_INFINITY");
if ((longMode & EXCEPTION_INFINITY.getLongValue()) != 0) {
newExceptionMode = (value.isTrue()) ? (RubyFixnum)currentExceptionMode.callCoerced(context, "|", EXCEPTION_INFINITY)
: (RubyFixnum)currentExceptionMode.callCoerced(context, "&", new RubyFixnum(runtime, ~(EXCEPTION_INFINITY).getLongValue()));
}
RubyFixnum EXCEPTION_NaN = (RubyFixnum)clazz.getConstant("EXCEPTION_NaN");
if ((longMode & EXCEPTION_NaN.getLongValue()) != 0) {
newExceptionMode = (value.isTrue()) ? (RubyFixnum)currentExceptionMode.callCoerced(context, "|", EXCEPTION_NaN)
: (RubyFixnum)currentExceptionMode.callCoerced(context, "&", new RubyFixnum(runtime, ~(EXCEPTION_NaN).getLongValue()));
}
RubyFixnum EXCEPTION_UNDERFLOW = (RubyFixnum)clazz.getConstant("EXCEPTION_UNDERFLOW");
if ((longMode & EXCEPTION_UNDERFLOW.getLongValue()) != 0) {
newExceptionMode = (value.isTrue()) ? (RubyFixnum)currentExceptionMode.callCoerced(context, "|", EXCEPTION_UNDERFLOW)
: (RubyFixnum)currentExceptionMode.callCoerced(context, "&", new RubyFixnum(runtime, ~(EXCEPTION_UNDERFLOW).getLongValue()));
}
RubyFixnum EXCEPTION_OVERFLOW = (RubyFixnum)clazz.getConstant("EXCEPTION_OVERFLOW");
if ((longMode & EXCEPTION_OVERFLOW.getLongValue()) != 0) {
newExceptionMode = (value.isTrue()) ? (RubyFixnum)currentExceptionMode.callCoerced(context, "|", EXCEPTION_OVERFLOW)
: (RubyFixnum)currentExceptionMode.callCoerced(context, "&", new RubyFixnum(runtime, ~(EXCEPTION_OVERFLOW).getLongValue()));
}
c.setInternalModuleVariable("vpExceptionMode", newExceptionMode);
return newExceptionMode;
}
long ROUND_MODE = ((RubyFixnum)clazz.getConstant("ROUND_MODE")).getLongValue();
if (longMode == ROUND_MODE) {
if (value.isNil()) {
return c.searchInternalModuleVariable("vpRoundingMode");
}
RoundingMode javaRoundingMode = javaRoundingModeFromRubyRoundingMode(runtime, value);
RubyFixnum roundingMode = runtime.newFixnum(javaRoundingMode.ordinal());
c.setInternalModuleVariable("vpRoundingMode", roundingMode);
return c.searchInternalModuleVariable("vpRoundingMode");
}
throw runtime.newTypeError("first argument for BigDecimal#mode invalid");
}
private static RoundingMode getRoundingMode(Ruby runtime) {
RubyClass BigDecimal = runtime.getClass("BigDecimal");
RubyFixnum roundingMode = (RubyFixnum) BigDecimal
.searchInternalModuleVariable("vpRoundingMode");
return RoundingMode.valueOf((int) roundingMode.getLongValue());
}
private static boolean isNaNExceptionMode(Ruby runtime) {
RubyClass BigDecimal = runtime.getClass("BigDecimal");
RubyFixnum currentExceptionMode = (RubyFixnum) BigDecimal
.searchInternalModuleVariable("vpExceptionMode");
RubyFixnum EXCEPTION_NaN = (RubyFixnum) BigDecimal.getConstant("EXCEPTION_NaN");
return (currentExceptionMode.getLongValue() & EXCEPTION_NaN.getLongValue()) != 0;
}
private static boolean isInfinityExceptionMode(Ruby runtime) {
RubyClass BigDecimal = runtime.getClass("BigDecimal");
RubyFixnum currentExceptionMode = (RubyFixnum) BigDecimal
.searchInternalModuleVariable("vpExceptionMode");
RubyFixnum EXCEPTION_INFINITY = (RubyFixnum) BigDecimal.getConstant("EXCEPTION_INFINITY");
return (currentExceptionMode.getLongValue() & EXCEPTION_INFINITY.getLongValue()) != 0;
}
private static boolean isOverflowExceptionMode(Ruby runtime) {
RubyClass BigDecimal = runtime.getClass("BigDecimal");
RubyFixnum currentExceptionMode = (RubyFixnum) BigDecimal
.searchInternalModuleVariable("vpExceptionMode");
RubyFixnum EXCEPTION_OVERFLOW = (RubyFixnum) BigDecimal.getConstant("EXCEPTION_OVERFLOW");
return (currentExceptionMode.getLongValue() & EXCEPTION_OVERFLOW.getLongValue()) != 0;
}
private static RubyBigDecimal cannotBeCoerced(ThreadContext context, IRubyObject value, boolean must) {
if (must) {
throw context.runtime.newTypeError(
errMessageType(context, value) + " can't be coerced into BigDecimal"
);
}
return null;
}
private static String errMessageType(ThreadContext context, IRubyObject value) {
if (value == null || value.isNil()) return "nil";
if (value.isImmediate()) return RubyObject.inspect(context, value).toString();
return value.getMetaClass().getBaseName();
}
private static RubyBigDecimal getVpValue19(ThreadContext context, IRubyObject v, boolean must) {
long precision;
if (v instanceof RubyFloat) {
precision = 0;
} else if (v instanceof RubyRational) {
precision = 0;
} else {
precision = -1;
}
return getVpValueWithPrec19(context, v, precision, must);
}
private static IRubyObject getVpRubyObjectWithPrec19Inner(ThreadContext context, RubyRational value, long precision, boolean must) {
BigDecimal numerator = BigDecimal.valueOf(RubyNumeric.num2long(value.numerator(context)));
BigDecimal denominator = BigDecimal.valueOf(RubyNumeric.num2long(value.denominator(context)));
int len = numerator.precision() + denominator.precision();
int pow = len / 4;
MathContext mathContext = new MathContext((pow + 1) * 4, getRoundingMode(context.runtime));
return new RubyBigDecimal(context.runtime, numerator.divide(denominator, mathContext));
}
private static RubyBigDecimal getVpValueWithPrec19(ThreadContext context, IRubyObject value, long precision, boolean must) {
while (true) {
if (value instanceof RubyFloat) {
if (precision > Long.MAX_VALUE) cannotBeCoerced(context, value, must);
RubyFloat f = (RubyFloat) value;
value = new RubyBigDecimal(context.runtime, BigDecimal.valueOf(f.getDoubleValue()));
continue;
}
else if (value instanceof RubyRational) {
if (precision < 0) {
if (must) {
throw context.runtime.newArgumentError(value.getMetaClass().getBaseName() + " can't be coerced into BigDecimal without a precision");
}
return null;
}
value = getVpRubyObjectWithPrec19Inner(context, (RubyRational) value, precision, must);
continue;
}
else if (value instanceof RubyBigDecimal) {
return (RubyBigDecimal) value;
}
else if (value instanceof RubyFixnum || value instanceof RubyBignum) {
return newInstance(context, context.runtime.getClass("BigDecimal"), value.asString());
}
return cannotBeCoerced(context, value, must);
}
}
private static RubyBigDecimal getVpValue(ThreadContext context, IRubyObject value, boolean must) {
if (value instanceof RubyBigDecimal) return (RubyBigDecimal) value;
if (value instanceof RubyFixnum || value instanceof RubyBignum) {
return newInstance(context, context.runtime.getClass("BigDecimal"), value.asString());
}
if ((value instanceof RubyRational) || (value instanceof RubyFloat)) {
return newInstance(context, context.runtime.getClass("BigDecimal"), value.asString(), RubyFixnum.newFixnum(context.runtime, RubyFloat.DIG));
}
return cannotBeCoerced(context, value, must);
}
@JRubyMethod(name = "induced_from", required = 1, meta = true)
public static IRubyObject induced_from(ThreadContext context, IRubyObject recv, IRubyObject arg) {
return getVpValue(context, arg, true);
}
private static RubyBigDecimal newInstance(Ruby runtime, IRubyObject recv, RubyBigDecimal arg) {
return new RubyBigDecimal(runtime, (RubyClass) recv, arg);
}
private static RubyBigDecimal newInstance(Ruby runtime, IRubyObject recv, RubyFixnum arg, MathContext mathContext) {
return new RubyBigDecimal(runtime, (RubyClass) recv, new BigDecimal(arg.getLongValue(), mathContext));
}
private static RubyBigDecimal newInstance(ThreadContext context, RubyRational arg, MathContext mathContext) {
BigDecimal num = new BigDecimal(arg.numerator(context).convertToInteger().getLongValue());
BigDecimal den = new BigDecimal(arg.denominator(context).convertToInteger().getLongValue());
BigDecimal value = num.divide(den, mathContext);
return new RubyBigDecimal(context.runtime, value);
}
private static RubyBigDecimal newInstance(Ruby runtime, IRubyObject recv, RubyFloat arg, MathContext mathContext) {
// precision can be no more than float digits
if (mathContext.getPrecision() > RubyFloat.DIG + 1) throw runtime.newArgumentError("precision too large");
double dblVal = arg.getDoubleValue();
if(Double.isInfinite(dblVal) || Double.isNaN(dblVal)) throw runtime.newFloatDomainError("NaN");
return new RubyBigDecimal(runtime, (RubyClass) recv, new BigDecimal(dblVal, mathContext));
}
private static RubyBigDecimal newInstance(Ruby runtime, IRubyObject recv, RubyBignum arg, MathContext mathContext) {
return new RubyBigDecimal(runtime, (RubyClass) recv, new BigDecimal(arg.getBigIntegerValue(), mathContext));
}
private final static Pattern NUMBER_PATTERN = Pattern.compile("^([+-]?\\d*\\.?\\d*([eE]?)([+-]?\\d*)).*");
private static RubyBigDecimal newInstance(ThreadContext context, IRubyObject recv, IRubyObject arg, MathContext mathContext) {
String strValue = arg.convertToString().toString().trim();
int sign = 1;
switch ( strValue.length() > 0 ? strValue.charAt(0) : ' ' ) {
case '_' : return newZero(context.runtime, 1); // leading "_" are not allowed
case 'N' :
if ( "NaN".equals(strValue) ) return newNaN(context.runtime);
break;
case 'I' :
if ( "Infinity".equals(strValue) ) return newInfinity(context.runtime, 1);
break;
case '-' :
if ( "-Infinity".equals(strValue) ) return newInfinity(context.runtime, -1);
sign = -1;
break;
case '+' :
if ( "+Infinity".equals(strValue) ) return newInfinity(context.runtime, +1);
break;
}
// Convert String to Java understandable format (for BigDecimal).
strValue = strValue.replaceFirst("[dD]", "E"); // 1. MRI allows d and D as exponent separators
strValue = strValue.replaceAll("_", ""); // 2. MRI allows underscores anywhere
Matcher matcher = NUMBER_PATTERN.matcher(strValue);
strValue = matcher.replaceFirst("$1"); // 3. MRI ignores the trailing junk
String exp = matcher.group(2); int idx;
if ( exp != null && ! exp.isEmpty() ) {
String expValue = matcher.group(3);
if (expValue.isEmpty() || expValue.equals("-") || expValue.equals("+")) {
strValue = strValue.concat("0"); // 4. MRI allows 1E, 1E-, 1E+
}
else if (isExponentOutOfRange(expValue)) {
// Handle infinity (Integer.MIN_VALUE + 1) < expValue < Integer.MAX_VALUE
return newInfinity(context.runtime, sign);
}
}
else if ( ( idx = matcher.start(3) ) > 0 ) {
strValue = strValue.substring(0, idx); // ignored tail junk e.g. "5-6" -> "-6"
}
BigDecimal decimal;
try {
decimal = new BigDecimal(strValue, mathContext);
}
catch (NumberFormatException e) {
if (isOverflowExceptionMode(context.runtime)) throw context.runtime.newFloatDomainError("exponent overflow");
decimal = new BigDecimal(0);
}
// MRI behavior: -0 and +0 are two different things
if (decimal.signum() == 0) return newZero(context.runtime, sign);
return new RubyBigDecimal(context.runtime, (RubyClass) recv, decimal);
}
private static boolean isExponentOutOfRange(final String expValue) {
int num = 0;
int sign = 1;
final int len = expValue.length();
final char ch = expValue.charAt(0);
if (ch == '-') {
sign = -1;
} else if (ch != '+') {
num = '0' - ch;
}
int i = 1;
final int max = (sign == 1) ? -Integer.MAX_VALUE : Integer.MIN_VALUE + 1;
final int multmax = max / 10;
while (i < len) {
int d = expValue.charAt(i++) - '0';
if (num < multmax) {
return true;
}
num *= 10;
if (num < (max + d)) {
return true;
}
num -= d;
}
return false;
}
@Deprecated
public static RubyBigDecimal newInstance(IRubyObject recv, IRubyObject[] args) {
final ThreadContext context = recv.getRuntime().getCurrentContext();
switch (args.length) {
case 1: return newInstance(context, recv, args[0]);
case 2: return newInstance(context, recv, args[0], args[1]);
}
throw new IllegalArgumentException("unexpected argument count: " + args.length);
}
@JRubyMethod(name = "new", meta = true)
public static RubyBigDecimal newInstance(ThreadContext context, IRubyObject recv, IRubyObject arg) {
if ( context.runtime.is1_9() ) {
if (arg instanceof RubyBigDecimal) return newInstance(context.runtime, recv, (RubyBigDecimal) arg);
if (arg instanceof RubyRational) throw context.runtime.newArgumentError("can't omit precision for a Rational.");
if (arg instanceof RubyFloat) throw context.runtime.newArgumentError("can't omit precision for a Float.");
if (arg instanceof RubyFixnum) return newInstance(context.runtime, recv, (RubyFixnum) arg, MathContext.UNLIMITED);
if (arg instanceof RubyBignum) return newInstance(context.runtime, recv, (RubyBignum) arg, MathContext.UNLIMITED);
}
return newInstance(context, recv, arg, MathContext.UNLIMITED);
}
@JRubyMethod(name = "new", meta = true)
public static RubyBigDecimal newInstance(ThreadContext context, IRubyObject recv, IRubyObject arg, IRubyObject mathArg) {
int digits = (int) mathArg.convertToInteger().getLongValue();
if (digits < 0) throw context.runtime.newArgumentError("argument must be positive");
MathContext mathContext = new MathContext(digits);
if ( context.runtime.is1_9() ) {
if (arg instanceof RubyBigDecimal) return newInstance(context.runtime, recv, (RubyBigDecimal) arg);
if (arg instanceof RubyFloat) return newInstance(context.runtime, recv, (RubyFloat) arg, mathContext);
if (arg instanceof RubyRational) return newInstance(context, (RubyRational) arg, mathContext);
if (arg instanceof RubyFixnum) return newInstance(context.runtime, recv, (RubyFixnum) arg, mathContext);
if (arg instanceof RubyBignum) return newInstance(context.runtime, recv, (RubyBignum) arg, mathContext);
mathContext = MathContext.UNLIMITED;
}
return newInstance(context, recv, arg, mathContext);
}
private static RubyBigDecimal newZero(final Ruby runtime, final int sign) {
return new RubyBigDecimal(runtime, BigDecimal.ZERO, 0, sign < 0 ? -1 : 1);
}
private static RubyBigDecimal newNaN(Ruby runtime) {
if ( isNaNExceptionMode(runtime) ) {
throw runtime.newFloatDomainError("Computation results to 'NaN'(Not a Number)");
}
return new RubyBigDecimal(runtime, BigDecimal.ZERO, true);
}
private static RubyBigDecimal newInfinity(final Ruby runtime, final int sign) {
if ( isInfinityExceptionMode(runtime) ) {
throw runtime.newFloatDomainError("Computation results to 'Infinity'");
}
return new RubyBigDecimal(runtime, BigDecimal.ZERO, sign < 0 ? -1 : 1);
}
private RubyBigDecimal setResult() {
return setResult(0);
}
private RubyBigDecimal setResult(int scale) {
int prec = RubyFixnum.fix2int(getRuntime().getClass("BigDecimal").searchInternalModuleVariable("vpPrecLimit"));
int prec2 = Math.max(scale, prec);
if (prec2 > 0 && this.value.scale() > (prec2-getExponent())) {
this.value = this.value.setScale(prec2-getExponent(), BigDecimal.ROUND_HALF_UP);
}
return this;
}
@Override
@JRubyMethod(name = "hash")
public RubyFixnum hash() {
return getRuntime().newFixnum(value.stripTrailingZeros().hashCode());
}
@JRubyMethod(name = {"%", "modulo"}, required = 1, compat = CompatVersion.RUBY1_8)
public IRubyObject op_mod(ThreadContext context, IRubyObject arg) {
// TODO: full-precision remainder is 1000x slower than MRI!
Ruby runtime = context.runtime;
if (isInfinity() || isNaN()) {
return newNaN(runtime);
}
RubyBigDecimal val = getVpValue(context, arg, false);
if (val == null) {
return callCoerced(context, "%", arg, true);
}
if (val.isInfinity() || val.isNaN() || val.isZero()) {
return newNaN(runtime);
}
// Java and MRI definitions of modulo are different.
BigDecimal modulo = value.remainder(val.value);
if (modulo.signum() * val.value.signum() < 0) {
modulo = modulo.add(val.value);
}
return new RubyBigDecimal(runtime, modulo).setResult();
}
@JRubyMethod(name = {"%", "modulo"}, required = 1, compat = CompatVersion.RUBY1_9)
public IRubyObject op_mod19(ThreadContext context, IRubyObject other) {
// TODO: full-precision divmod is 1000x slower than MRI!
Ruby runtime = context.runtime;
RubyBigDecimal val = getVpValue19(context, other, false);
if (val == null) {
return callCoerced(context, "%", other, true);
}
if (isNaN() || val.isNaN() || (isInfinity() && val.isInfinity())) {
return newNaN(runtime);
}
if (val.isZero()) {
throw context.runtime.newZeroDivisionError();
}
if (isInfinity()) {
return newNaN(runtime);
}
if (val.isInfinity()) {
return this;
}
if (isZero()) {
return newZero(runtime, value.signum());
}
// Java and MRI definitions of modulo are different.
BigDecimal modulo = value.remainder(val.value);
if (modulo.signum() * val.value.signum() < 0) {
modulo = modulo.add(val.value);
}
return new RubyBigDecimal(runtime, modulo).setResult();
}
@Override
@JRubyMethod(name = "remainder", required = 1, compat = CompatVersion.RUBY1_8)
public IRubyObject remainder(ThreadContext context, IRubyObject arg) {
RubyBigDecimal val = getVpValue(context, arg, false);
return remainderInternal(context, val, arg);
}
@JRubyMethod(name = "remainder", required = 1, compat = CompatVersion.RUBY1_9)
public IRubyObject remainder19(ThreadContext context, IRubyObject arg) {
RubyBigDecimal val = getVpValue19(context, arg, false);
return remainderInternal(context, val, arg);
}
private IRubyObject remainderInternal(ThreadContext context, RubyBigDecimal val, IRubyObject arg) {
// TODO: full-precision remainder is 1000x slower than MRI!
Ruby runtime = context.runtime;
if (isInfinity() || isNaN()) {
return newNaN(runtime);
}
if (val == null) {
return callCoerced(context, "remainder", arg, true);
}
if (val.isInfinity() || val.isNaN() || val.isZero()) {
return newNaN(runtime);
}
// Java and MRI definitions of remainder are the same.
return new RubyBigDecimal(runtime, value.remainder(val.value)).setResult();
}
@JRubyMethod(name = "*", required = 1, compat = CompatVersion.RUBY1_8)
public IRubyObject op_mul(ThreadContext context, IRubyObject arg) {
return mult(context, arg, vpPrecLimit(context.runtime));
}
@JRubyMethod(name = "*", required = 1, compat = CompatVersion.RUBY1_9)
public IRubyObject op_mul19(ThreadContext context, IRubyObject arg) {
return mult19(context, arg, vpPrecLimit(context.runtime));
}
private static IRubyObject vpPrecLimit(final Ruby runtime) {
return runtime.getClass("BigDecimal").searchInternalModuleVariable("vpPrecLimit");
}
@JRubyMethod(name = "mult", required = 2, compat = CompatVersion.RUBY1_8)
public IRubyObject mult(ThreadContext context, IRubyObject b, IRubyObject n) {
RubyBigDecimal val = getVpValue(context, b, false);
if (val == null) { // TODO: what about n arg?
return callCoerced(context, "*", b, true);
}
return multInternal(context.runtime, val, n);
}
@JRubyMethod(name = "mult", required = 2, compat = CompatVersion.RUBY1_9)
public IRubyObject mult19(ThreadContext context, IRubyObject b, IRubyObject n) {
RubyBigDecimal val = getVpValue19(context, b, false);
if (val == null) { // TODO: what about n arg?
return callCoerced(context, "*", b, true);
}
return multInternal(context.runtime, val, n);
}
private RubyBigDecimal multInternal(final Ruby runtime, RubyBigDecimal val, IRubyObject n) {
int digits = RubyNumeric.fix2int(n);
if ( isNaN() || val.isNaN() ) {
return newNaN(runtime);
}
if ( ( isInfinity() && val.isZero() ) || ( isZero() && val.isInfinity() ) ) {
return newNaN(runtime);
}
if ( isZero() || val.isZero() ) {
int sign1 = isZero()? zeroSign : value.signum();
int sign2 = val.isZero() ? val.zeroSign : val.value.signum();
return newZero(runtime, sign1 * sign2);
}
if ( isInfinity() || val.isInfinity() ) {
int sign1 = isInfinity() ? infinitySign : value.signum();
int sign2 = val.isInfinity() ? val.infinitySign : val.value.signum();
return newInfinity(runtime, sign1 * sign2);
}
BigDecimal res = value.multiply(val.value);
if (res.precision() > digits) {
// TODO: rounding mode should not be hard-coded. See #mode.
res = res.round(new MathContext(digits, RoundingMode.HALF_UP));
}
return new RubyBigDecimal(runtime, res).setResult();
}
// @Deprecated
public IRubyObject op_pow(IRubyObject arg) {
return op_pow(getRuntime().getCurrentContext(), arg);
}
@JRubyMethod(name = {"**", "power"}, required = 1, compat = CompatVersion.RUBY1_8)
public RubyBigDecimal op_pow(final ThreadContext context, IRubyObject arg) {
if ( ! (arg instanceof RubyFixnum) ) {
throw context.runtime.newTypeError("wrong argument type " + arg.getMetaClass() + " (expected Fixnum)");
}
if (isNaN() || isInfinity()) return newNaN(context.runtime);
int times = RubyNumeric.fix2int(arg.convertToInteger());
if (times < 0) {
if (isZero()) return newInfinity(context.runtime, value.signum());
return new RubyBigDecimal(context.runtime, powNegative(times));
}
return new RubyBigDecimal(context.runtime, value.pow(times));
}
// @Deprecated
public IRubyObject op_pow19(IRubyObject exp) {
return op_pow19(getRuntime().getCurrentContext(), exp);
}
@JRubyMethod(name = {"**", "power"}, required = 1, compat = CompatVersion.RUBY1_9)
public RubyBigDecimal op_pow19(ThreadContext context, IRubyObject exp) {
final Ruby runtime = context.runtime;
if ( ! (exp instanceof RubyNumeric) ) {
throw context.runtime.newTypeError("wrong argument type " + exp.getMetaClass() + " (expected scalar Numeric)");
}
if (isNaN()) return newNaN(runtime);
if (isInfinity()) {
if (Numeric.f_negative_p(context, exp)) {
if (infinitySign < 0) {
if ( Numeric.f_integer_p(context, exp).isTrue() ) {
/* (-Infinity) ** (-even_integer) -> +0 */
/* (-Infinity) ** (-odd_integer) -> -0 */
return newZero(runtime, is_even(exp) ? +1 : -1);
}
/* (-Infinity) ** (-non_integer) -> -0 */
return newZero(runtime, -1);
}
else {
return newZero(runtime, 0);
}
} else {
if (infinitySign < 0) {
if ( Numeric.f_integer_p(context, exp).isTrue() ) {
return newInfinity(runtime, is_even(exp) ? +1 : -1);
}
throw runtime.newMathDomainError("a non-integral exponent for a negative base");
}
return newInfinity(runtime, 1);
}
}
final int times; final double rem; // exp's decimal part
// when pow is not an integer we're play the oldest trick :
// X pow (T+R) = X pow T * X pow R
if ( ! ( exp instanceof RubyInteger ) ) {
BigDecimal expVal = BigDecimal.valueOf( ((RubyNumeric) exp).getDoubleValue() );
BigDecimal[] divAndRem = expVal.divideAndRemainder(BigDecimal.ONE);
times = divAndRem[0].intValueExact(); rem = divAndRem[1].doubleValue();
}
else {
times = RubyNumeric.fix2int(exp); rem = 0;
}
BigDecimal pow;
if ( times < 0 ) {
if (isZero()) return newInfinity(context.runtime, value.signum());
pow = powNegative(times);
}
else {
pow = value.pow(times);
}
if ( rem > 0 ) {
// TODO of course this assumes we fit into double (and we loose some precision)
double remPow = Math.pow(value.doubleValue(), rem);
pow = pow.multiply( BigDecimal.valueOf(remPow) );
}
return new RubyBigDecimal(runtime, pow);
}
private BigDecimal powNegative(final int times) {
// Note: MRI has a very non-trivial way of calculating the precision,
// so we use very simple approximation here:
int precision = (-times + 4) * (getAllDigits().length() + 4);
return value.pow(times, new MathContext(precision, RoundingMode.HALF_UP));
}
@JRubyMethod(name = "+", compat = CompatVersion.RUBY1_8)
public IRubyObject op_plus(ThreadContext context, IRubyObject b) {
RubyBigDecimal val = getVpValue(context, b, false);
return addInternal(context, val, b, vpPrecLimit(context.runtime));
}
@JRubyMethod(name = "+", compat = CompatVersion.RUBY1_9)
public IRubyObject op_plus19(ThreadContext context, IRubyObject b) {
RubyBigDecimal val = getVpValue19(context, b, false);
return addInternal(context, val, b, vpPrecLimit(context.runtime));
}
@JRubyMethod(name = "add", compat = CompatVersion.RUBY1_8)
public IRubyObject add2(ThreadContext context, IRubyObject b, IRubyObject digits) {
RubyBigDecimal val = getVpValue(context, b, false);
return addInternal(context, val, b, digits);
}
@JRubyMethod(name = "add", compat = CompatVersion.RUBY1_9)
public IRubyObject add219(ThreadContext context, IRubyObject b, IRubyObject digits) {
RubyBigDecimal val = getVpValue19(context, b, false);
return addInternal(context, val, b, digits);
}
private IRubyObject addInternal(ThreadContext context, RubyBigDecimal val, IRubyObject b, IRubyObject digits) {
Ruby runtime = context.runtime;
int prec = getPositiveInt(context, digits);
if (val == null) {
// TODO:
// MRI behavior: Call "+" or "add", depending on the call.
// But this leads to exceptions when Floats are added. See:
// http://blade.nagaokaut.ac.jp/cgi-bin/scat.rb/ruby/ruby-core/17374
// return callCoerced(context, op, b, true); -- this is MRI behavior.
// We'll use ours for now, thus providing an ability to add Floats.
return callCoerced(context, "+", b, true);
}
RubyBigDecimal res = handleAddSpecialValues(context, val);
if ( res != null ) return res;
RoundingMode roundMode = getRoundingMode(runtime);
return new RubyBigDecimal(runtime, value.add(
val.value, new MathContext(prec, roundMode))); // TODO: why this: .setResult();
}
private static int getPositiveInt(ThreadContext context, IRubyObject arg) {
final Ruby runtime = context.runtime;
if ( arg instanceof RubyFixnum ) {
int value = RubyNumeric.fix2int(arg);
if (value < 0) {
throw runtime.newArgumentError("argument must be positive");
}
return value;
}
throw runtime.newTypeError(arg, runtime.getFixnum());
}
private RubyBigDecimal handleAddSpecialValues(ThreadContext context, RubyBigDecimal val) {
if (isNaN() || val.isNaN) {
return newNaN(context.runtime);
}
int sign = infinitySign * val.infinitySign;
if (sign > 0) {
return isInfinity() ? this : val;
}
if (sign < 0) {
return newNaN(context.runtime);
}
if (sign == 0) {
sign = infinitySign + val.infinitySign;
if (sign != 0) {
return newInfinity(context.runtime, sign);
}
}
return null;
}
@Override
@JRubyMethod(name = "+@")
public IRubyObject op_uplus() {
return this;
}
@JRubyMethod(name = "-", required = 1, compat = CompatVersion.RUBY1_8)
public IRubyObject op_minus(ThreadContext context, IRubyObject b) {
RubyBigDecimal val = getVpValue(context, b, false);
return subInternal(context, val, b);
}
@JRubyMethod(name = "-", required = 1, compat = CompatVersion.RUBY1_9)
public IRubyObject op_minus19(ThreadContext context, IRubyObject b) {
return subInternal(context, getVpValue19(context, b, true), b);
}
@JRubyMethod(name = "sub", required = 2, compat = CompatVersion.RUBY1_8)
public IRubyObject sub2(ThreadContext context, IRubyObject b, IRubyObject n) {
RubyBigDecimal val = getVpValue(context, b, false);
return subInternal(context, val, b);
}
@JRubyMethod(name = "sub", required = 2, compat = CompatVersion.RUBY1_9)
public IRubyObject sub219(ThreadContext context, IRubyObject b, IRubyObject n) {
RubyBigDecimal val = getVpValue19(context, b, false);
return subInternal(context, val, b);
}
private IRubyObject subInternal(ThreadContext context, RubyBigDecimal val, IRubyObject b) {
if (val == null) {
return callCoerced(context, "-", b);
}
RubyBigDecimal res = handleMinusSpecialValues(context, val);
if (res != null) {
return res;
}
return new RubyBigDecimal(context.runtime, value.subtract(val.value)).setResult();
}
private RubyBigDecimal handleMinusSpecialValues(ThreadContext context, RubyBigDecimal val) {
if (isNaN() || val.isNaN()) {
return newNaN(context.runtime);
}
int sign = infinitySign * val.infinitySign;
if (sign > 0) {
return newNaN(context.runtime);
}
if (sign < 0) {
return this;
}
if (sign == 0) {
if (isInfinity()) {
return this;
}
if (val.isInfinity()) {
return newInfinity(context.runtime, val.infinitySign * -1);
}
sign = infinitySign + val.infinitySign;
if (sign != 0) {
return newInfinity(context.runtime, sign);
}
}
return null;
}
@JRubyMethod(name = "-@")
public IRubyObject op_uminus() {
final Ruby runtime = getRuntime();
if (isNaN()) {
return newNaN(runtime);
}
if (isInfinity()) {
return newInfinity(runtime, -infinitySign);
}
if (isZero()) {
return newZero(runtime, -zeroSign);
}
return new RubyBigDecimal(runtime, value.negate());
}
@JRubyMethod(name = {"/", "quo"}, compat = CompatVersion.RUBY1_8)
public IRubyObject op_quo(ThreadContext context, IRubyObject other) {
RubyBigDecimal val = getVpValue(context, other, false);
if (val == null) return callCoerced(context, "/", other, true);
// regular division with some default precision
// TODO: proper algorithm to set the precision
IRubyObject result = op_div(context, other, context.runtime.newFixnum(200));
if ( other instanceof RubyFloat && result instanceof RubyBigDecimal ) {
return ((RubyBigDecimal) result).convertToFloat();
}
return result;
}
@JRubyMethod(name = {"/", "quo"}, compat = CompatVersion.RUBY1_9)
public IRubyObject op_quo19(ThreadContext context, IRubyObject other) {
return op_quo19_20(context, other);
}
@JRubyMethod(name = {"/", "quo"}, compat = CompatVersion.RUBY2_0)
public IRubyObject op_quo20(ThreadContext context, IRubyObject other) {
return op_quo19_20(context, other);
}
private IRubyObject op_quo19_20(ThreadContext context, IRubyObject other) {
RubyBigDecimal val = getVpValue19(context, other, false);
if (val == null) return callCoerced(context, "/", other, true);
// regular division with some default precision
// proper algorithm to set the precision
// the precision is multiple of 4
// and the precision is larger than len * 2
int len = value.precision() + val.value.precision();
int pow = len / 4;
int precision = (pow + 1) * 4 * 2;
return op_div(context, val, context.runtime.newFixnum(precision));
}
@JRubyMethod(name = "div", compat = CompatVersion.RUBY1_8)
public IRubyObject op_div(ThreadContext context, IRubyObject other) {
// integer division
RubyBigDecimal val = getVpValue(context, other, false);
if (val == null) {
return callCoerced(context, "div", other);
}
if (isNaN() || val.isZero() || val.isNaN()) {
return newNaN(context.runtime);
}
if (isInfinity() || val.isInfinity()) {
return newNaN(context.runtime);
}
return new RubyBigDecimal(context.runtime,
this.value.divideToIntegralValue(val.value)).setResult();
}
@JRubyMethod(name = "div", compat = CompatVersion.RUBY1_9)
public IRubyObject op_div19(ThreadContext context, IRubyObject r) {
RubyBigDecimal val = getVpValue19(context, r, false);
if (val == null) return callCoerced(context, "div", r, true);
if (isNaN() || val.isNaN()) {
throw context.runtime.newFloatDomainError("Computation results to 'NaN'");
}
if (isInfinity() && val.isOne()) {
throw context.runtime.newFloatDomainError("Computation results to 'Infinity'");
}
if (val.isInfinity()) {
return newZero(context.runtime, val.infinitySign);
}
if (isZero() || val.isZero()) {
throw context.runtime.newZeroDivisionError();
}
return op_div(context, r);
}
@JRubyMethod(name = "div", compat = CompatVersion.RUBY1_8)
public IRubyObject op_div(ThreadContext context, IRubyObject other, IRubyObject digits) {
// TODO: take BigDecimal.mode into account.
int scale = RubyNumeric.fix2int(digits);
RubyBigDecimal val = getVpValue(context, other, false);
if (val == null) return callCoerced(context, "div", other, true);
if (isNaN() || (isZero() && val.isZero()) || val.isNaN()) {
return newNaN(context.runtime);
}
if (val.isZero()) {
int sign1 = isInfinity() ? infinitySign : value.signum();
return newInfinity(context.runtime, sign1 * val.zeroSign);
}
if (isInfinity() && !val.isInfinity()) {
return newInfinity(context.runtime, infinitySign * val.value.signum());
}
if (!isInfinity() && val.isInfinity()) {
return newZero(context.runtime, value.signum() * val.infinitySign);
}
if (isInfinity() && val.isInfinity()) {
return newNaN(context.runtime);
}
if (isZero()) {
return newZero(context.runtime, zeroSign * val.value.signum());
}
if (scale == 0) {
// MRI behavior: "If digits is 0, the result is the same as the / operator."
return op_quo(context, other);
}
MathContext mathContext = new MathContext(scale, getRoundingMode(context.runtime));
return new RubyBigDecimal(context.runtime, value.divide(val.value, mathContext)).setResult(scale);
}
@JRubyMethod(name = "div", compat = CompatVersion.RUBY1_9)
public IRubyObject op_div19(ThreadContext context, IRubyObject other, IRubyObject digits) {
RubyBigDecimal val = getVpValue(context, other, false);
if (val == null) return callCoerced(context, "div", other, true);
if (isNaN() || val.isNaN()) {
throw context.runtime.newFloatDomainError("Computation results to 'NaN'");
}
return op_div(context, other, digits);
}
private IRubyObject cmp(ThreadContext context, final IRubyObject arg, final char op) {
final int e;
RubyBigDecimal rb = getVpValue(context, arg, false);
if (rb == null) {
IRubyObject cmp = callCoerced(context, "<=>", arg, false);
if ( cmp.isNil() ) { // arg.coerce failed
if (op == '*') return context.nil;
if (op == '=' || isNaN()) return context.runtime.getFalse();
throw context.runtime.newArgumentError("comparison of BigDecimal with "+ errMessageType(context, arg) +" failed");
}
e = RubyNumeric.fix2int(cmp);
} else {
if (isNaN() | rb.isNaN()) {
return (op == '*') ? context.nil : context.runtime.getFalse();
}
if (infinitySign != 0 || rb.infinitySign != 0) {
e = infinitySign - rb.infinitySign;
} else {
e = value.compareTo(rb.value);
}
}
switch(op) {
case '*': return context.runtime.newFixnum(e);
case '=': return context.runtime.newBoolean(e == 0);
case '!': return context.runtime.newBoolean(e != 0);
case 'G': return context.runtime.newBoolean(e >= 0);
case '>': return context.runtime.newBoolean(e > 0);
case 'L': return context.runtime.newBoolean(e <= 0);
case '<': return context.runtime.newBoolean(e < 0);
}
return context.nil;
}
@Override
@JRubyMethod(name = "<=>", required = 1)
public IRubyObject op_cmp(ThreadContext context, IRubyObject arg) {
return cmp(context, arg, '*');
}
@Override
@JRubyMethod(name = {"eql?", "==", "==="}, required = 1)
public IRubyObject eql_p(ThreadContext context, IRubyObject arg) {
return cmp(context, arg, '=');
}
@JRubyMethod(name = "<", required = 1)
public IRubyObject op_lt(ThreadContext context, IRubyObject arg) {
return cmp(context, arg, '<');
}
@JRubyMethod(name = "<=", required = 1)
public IRubyObject op_le(ThreadContext context, IRubyObject arg) {
return cmp(context, arg, 'L');
}
@JRubyMethod(name = ">", required = 1)
public IRubyObject op_gt(ThreadContext context, IRubyObject arg) {
return cmp(context, arg, '>');
}
@JRubyMethod(name = ">=", required = 1)
public IRubyObject op_ge(ThreadContext context, IRubyObject arg) {
return cmp(context, arg, 'G');
}
@JRubyMethod(name = "abs")
public IRubyObject abs() {
Ruby runtime = getRuntime();
if (isNaN) {
return newNaN(runtime);
}
if (isInfinity()) {
return newInfinity(runtime, 1);
}
return new RubyBigDecimal(getRuntime(), value.abs()).setResult();
}
@JRubyMethod(name = "ceil", optional = 1)
public IRubyObject ceil(IRubyObject[] args) {
if (isNaN) {
return newNaN(getRuntime());
}
if (isInfinity()) {
return newInfinity(getRuntime(), infinitySign);
}
int n = 0;
if (args.length > 0) {
n = RubyNumeric.fix2int(args[0]);
}
if (value.scale() > n) { // rounding neccessary
return new RubyBigDecimal(getRuntime(),
value.setScale(n, RoundingMode.CEILING));
} else {
return this;
}
}
@JRubyMethod(name = "ceil", optional = 1, compat = CompatVersion.RUBY1_9)
public IRubyObject ceil19(IRubyObject[] args) {
checkFloatDomain();
if (args.length == 0) {
BigInteger ceil = value.setScale(0, RoundingMode.CEILING).toBigInteger();
if (ceil.compareTo(BigInteger.valueOf((long)ceil.intValue())) == 0) {
// it fits in Fixnum
return RubyInteger.int2fix(getRuntime(), ceil.intValue());
}
return RubyBignum.newBignum(getRuntime(), ceil);
}
return ceil(args);
}
@Override
public IRubyObject coerce(IRubyObject other) {
return coerce(getRuntime().getCurrentContext(), other);
}
@JRubyMethod(name = "coerce", required = 1)
public RubyArray coerce(ThreadContext context, IRubyObject other) {
return context.runtime.newArray(getVpValue(context, other, true), this);
}
@Override
public double getDoubleValue() { return SafeDoubleParser.doubleValue(value); }
@Override
public long getLongValue() { return value.longValue(); }
@Override
public BigInteger getBigIntegerValue() {
return value.toBigInteger();
}
public BigDecimal getBigDecimalValue() {
return value;
}
public RubyNumeric multiplyWith(ThreadContext context, RubyInteger value) {
return (RubyNumeric)op_mul(context, value);
}
public RubyNumeric multiplyWith(ThreadContext context, RubyFloat value) {
return (RubyNumeric)op_mul(context, value);
}
public RubyNumeric multiplyWith(ThreadContext context, RubyBignum value) {
return (RubyNumeric)op_mul(context, value);
}
@Override
@JRubyMethod(name = "divmod", required = 1, compat = CompatVersion.RUBY1_8)
public IRubyObject divmod(ThreadContext context, IRubyObject other) {
// TODO: full-precision divmod is 1000x slower than MRI!
Ruby runtime = context.runtime;
if (isInfinity() || isNaN()) {
return RubyArray.newArray(runtime, newNaN(runtime), newNaN(runtime));
}
RubyBigDecimal val = getVpValue(context, other, false);
if (val == null) {
return callCoerced(context, "divmod", other, true);
}
if (val.isInfinity() || val.isNaN() || val.isZero()) {
return RubyArray.newArray(runtime, newNaN(runtime), newNaN(runtime));
}
// Java and MRI definitions of divmod are different.
BigDecimal[] divmod = value.divideAndRemainder(val.value);
BigDecimal div = divmod[0];
BigDecimal mod = divmod[1];
if (mod.signum() * val.value.signum() < 0) {
div = div.subtract(BigDecimal.ONE);
mod = mod.add(val.value);
}
return RubyArray.newArray(runtime,
new RubyBigDecimal(runtime, div),
new RubyBigDecimal(runtime, mod));
}
@Override
@JRubyMethod(name = "divmod", required = 1, compat = CompatVersion.RUBY1_9)
public IRubyObject divmod19(ThreadContext context, IRubyObject other) {
// TODO: full-precision divmod is 1000x slower than MRI!
Ruby runtime = context.runtime;
RubyBigDecimal val = getVpValue19(context, other, false);
if (val == null) {
return callCoerced(context, "divmod", other, true);
}
if (isNaN() || val.isNaN() || (isInfinity() && val.isInfinity())) {
return RubyArray.newArray(runtime, newNaN(runtime), newNaN(runtime));
}
if (val.isZero()) {
throw context.runtime.newZeroDivisionError();
}
if (isInfinity()) {
int sign = (infinitySign == val.value.signum()) ? 1 : -1;
return RubyArray.newArray(runtime, newInfinity(runtime, sign), newNaN(runtime));
}
if (val.isInfinity()) {
return RubyArray.newArray(runtime, newZero(runtime, val.value.signum()), this);
}
if (isZero()) {
return RubyArray.newArray(runtime, newZero(runtime,
value.signum()), newZero(runtime, value.signum()));
}
// Java and MRI definitions of divmod are different.
BigDecimal[] divmod = value.divideAndRemainder(val.value);
BigDecimal div = divmod[0];
BigDecimal mod = divmod[1];
if (mod.signum() * val.value.signum() < 0) {
div = div.subtract(BigDecimal.ONE);
mod = mod.add(val.value);
}
return RubyArray.newArray(runtime, new RubyBigDecimal(runtime, div), new RubyBigDecimal(runtime, mod));
}
@JRubyMethod(name = "exponent")
public IRubyObject exponent() {
return getRuntime().newFixnum(getExponent());
}
@JRubyMethod(name = "finite?")
public IRubyObject finite_p() {
if (isNaN()) {
return getRuntime().getFalse();
}
return getRuntime().newBoolean(!isInfinity());
}
@JRubyMethod(name = "floor", optional = 1)
public IRubyObject floor(IRubyObject[] args) {
if (isNaN) {
return newNaN(getRuntime());
}
if (isInfinity()) {
return newInfinity(getRuntime(), infinitySign);
}
int n = 0;
if (args.length > 0) {
n = RubyNumeric.fix2int(args[0]);
}
final RubyBigDecimal floor;
if (value.scale() > n) { // rounding neccessary
floor = new RubyBigDecimal(getRuntime(),
value.setScale(n, RoundingMode.FLOOR));
} else {
floor = this;
}
if (getRuntime().is1_8()) return floor;
if (args.length == 0) {
return floor.to_int19();
}
return floor;
}
@JRubyMethod(name = "floor", optional = 1, compat = CompatVersion.RUBY1_9)
public IRubyObject floor19(IRubyObject[] args) {
if (isNaN || isInfinity()) {
throw getRuntime().newFloatDomainError("Computation results to '" + to_s(args).asJavaString() + "'");
}
return floor(args);
}
@JRubyMethod(name = "frac")
public IRubyObject frac() {
if (isNaN) {
return newNaN(getRuntime());
}
if (isInfinity()) {
return newInfinity(getRuntime(), infinitySign);
}
if (value.scale() > 0 && value.precision() < value.scale()) {
return new RubyBigDecimal(getRuntime(), value);
}
BigDecimal val = value.subtract(((RubyBigDecimal)fix()).value);
return new RubyBigDecimal(getRuntime(), val);
}
@JRubyMethod(name = "infinite?")
public IRubyObject infinite_p() {
if (infinitySign == 0) {
return getRuntime().getNil();
}
return getRuntime().newFixnum(infinitySign);
}
@JRubyMethod(name = "inspect")
public IRubyObject inspect(ThreadContext context) {
StringBuilder val = new StringBuilder("#");
return getRuntime().newString(val.toString());
}
@JRubyMethod(name = "nan?")
public IRubyObject nan_p() {
return getRuntime().newBoolean(isNaN);
}
@JRubyMethod(name = "nonzero?")
public IRubyObject nonzero_p() {
return isZero() ? getRuntime().getNil() : this;
}
@JRubyMethod(name = "precs")
public IRubyObject precs() {
final Ruby runtime = getRuntime();
final IRubyObject[] array = new IRubyObject[2];
array[0] = runtime.newFixnum(getSignificantDigits().length());
int len = getAllDigits().length();
int pow = len / 4;
array[1] = runtime.newFixnum((pow + 1) * 4);
return RubyArray.newArrayNoCopy(runtime, array);
}
@Deprecated
public IRubyObject round(IRubyObject[] args) {
return round(getRuntime().getCurrentContext(), args);
}
@JRubyMethod(name = "round", optional = 2)
public IRubyObject round(ThreadContext context, IRubyObject[] args) {
final int scale = args.length > 0 ? num2int(args[0]) : 0;
// Special treatment for BigDecimal::NAN and BigDecimal::INFINITY
//
// If round is called without any argument, we should raise a
// FloatDomainError. Otherwise, we don't have to call round ;
// we can simply return the number itself.
if (scale == 0 && isInfinity()) {
StringBuilder message = new StringBuilder("Computation results to ");
message.append('\'').append(callMethod(context, "to_s")).append('\'');
throw context.runtime.newFloatDomainError(message.toString());
} else {
if (isNaN()) return newNaN(context.runtime);
if (isInfinity()) {
return newInfinity(context.runtime, infinitySign);
}
}
RoundingMode mode = (args.length > 1) ? javaRoundingModeFromRubyRoundingMode(context.runtime, args[1]) : getRoundingMode(context.runtime);
// JRUBY-914: Java 1.4 BigDecimal does not allow a negative scale, so we have to simulate it
final RubyBigDecimal bigDecimal;
if (scale < 0) {
// shift the decimal point just to the right of the digit to be rounded to (divide by 10**(abs(scale)))
// -1 -> 10's digit, -2 -> 100's digit, etc.
BigDecimal normalized = value.movePointRight(scale);
// ...round to that digit
BigDecimal rounded = normalized.setScale(0, mode);
// ...and shift the result back to the left (multiply by 10**(abs(scale)))
bigDecimal = new RubyBigDecimal(context.runtime, rounded.movePointLeft(scale));
} else {
bigDecimal = new RubyBigDecimal(context.runtime, value.setScale(scale, mode));
}
if (context.runtime.is1_8()) {
return bigDecimal;
}
return args.length == 0 ? bigDecimal.to_int19() : bigDecimal;
}
//this relies on the Ruby rounding enumerations == Java ones, which they (currently) all are
private static RoundingMode javaRoundingModeFromRubyRoundingMode(Ruby runtime, IRubyObject arg) {
if (arg instanceof RubySymbol) {
RubySymbol roundingModeSymbol = (RubySymbol) arg;
String roundingModeString = roundingModeSymbol.asJavaString();
if (roundingModeString.equals("up")) {
return RoundingMode.UP;
} else if (roundingModeString.equals("down") || roundingModeString.equals("truncate")) {
return RoundingMode.DOWN;
} else if (roundingModeString.equals("half_up") || roundingModeString.equals("default")) {
return RoundingMode.HALF_UP;
} else if (roundingModeString.equals("half_down")) {
return RoundingMode.HALF_DOWN;
} else if (roundingModeString.equals("half_even") || roundingModeString.equals("banker")) {
return RoundingMode.HALF_EVEN;
} else if (roundingModeString.equals("ceiling") || roundingModeString.equals("ceil")) {
return RoundingMode.CEILING;
} else if (roundingModeString.equals("floor")) {
return RoundingMode.FLOOR;
} else {
throw runtime.newArgumentError("invalid rounding mode");
}
} else {
try {
return RoundingMode.valueOf(num2int(arg));
} catch (IllegalArgumentException iae) {
throw runtime.newArgumentError("invalid rounding mode");
}
}
}
@JRubyMethod(name = "sign")
public IRubyObject sign() {
if (isNaN()) return getMetaClass().getConstant("SIGN_NaN");
if (isInfinity()) return getMetaClass().getConstant(infinitySign < 0 ? "SIGN_NEGATIVE_INFINITE" : "SIGN_POSITIVE_INFINITE");
if (isZero()) return getMetaClass().getConstant(zeroSign < 0 ? "SIGN_NEGATIVE_ZERO" : "SIGN_POSITIVE_ZERO");
return getMetaClass().getConstant(value.signum() < 0 ? "SIGN_NEGATIVE_FINITE" : "SIGN_POSITIVE_FINITE");
}
@JRubyMethod(name = "split")
public RubyArray split() {
final Ruby runtime = getRuntime();
final IRubyObject[] array = new IRubyObject[4];
// sign
final RubyFixnum sign;
if (isNaN) {
sign = RubyFixnum.zero(runtime);
} else if (isInfinity()) {
sign = runtime.newFixnum(infinitySign);
} else if (isZero()){
sign = runtime.newFixnum(zeroSign);
} else {
sign = runtime.newFixnum(value.signum());
}
array[0] = sign;
// significant digits and exponent
final RubyString digits;
final RubyFixnum exp;
if (isNaN()) {
digits = runtime.newString("NaN");
exp = RubyFixnum.zero(runtime);
} else if (isInfinity()) {
digits = runtime.newString("Infinity");
exp = RubyFixnum.zero(runtime);
} else if (isZero()){
digits = runtime.newString("0");
exp = RubyFixnum.zero(runtime);
} else {
// normalize the value
digits = runtime.newString(getSignificantDigits());
exp = runtime.newFixnum(getExponent());
}
array[1] = digits;
array[3] = exp;
// base
array[2] = runtime.newFixnum(10);
return RubyArray.newArrayNoCopy(runtime, array);
}
// it doesn't handle special cases
private String getSignificantDigits() {
// TODO: no need to calculate every time.
BigDecimal val = value.abs().stripTrailingZeros();
return val.unscaledValue().toString();
}
private String getAllDigits() {
// TODO: no need to calculate every time.
BigDecimal val = value.abs();
return val.unscaledValue().toString();
}
// it doesn't handle special cases
private int getExponent() {
if (isZero() || isNaN() || isInfinity()) return 0;
BigDecimal val = value.abs().stripTrailingZeros();
return val.precision() - val.scale();
}
@JRubyMethod(name = "sqrt", required = 1)
public IRubyObject sqrt(IRubyObject arg) {
Ruby runtime = getRuntime();
if (isNaN()) {
throw runtime.newFloatDomainError("(VpSqrt) SQRT(NaN value)");
}
if ((isInfinity() && infinitySign < 0) || value.signum() < 0) {
throw runtime.newFloatDomainError("(VpSqrt) SQRT(negative value)");
}
if (isInfinity() && infinitySign > 0) {
return newInfinity(runtime, 1);
}
// NOTE: MRI's sqrt precision is limited by 100,
// but we allow values more than 100.
int n = RubyNumeric.fix2int(arg);
if (n < 0) {
throw runtime.newArgumentError("argument must be positive");
}
n += 4; // just in case, add a bit of extra precision
return new RubyBigDecimal(getRuntime(),
bigSqrt(this.value, new MathContext(n, RoundingMode.HALF_UP))).setResult();
}
@JRubyMethod(name = "to_f")
public IRubyObject to_f() {
if (isNaN()) {
return RubyFloat.newFloat(getRuntime(), Double.NaN);
}
if (isInfinity()) {
return RubyFloat.newFloat(getRuntime(),
infinitySign < 0 ? Double.NEGATIVE_INFINITY : Double.POSITIVE_INFINITY);
}
if (isZero()) {
return RubyFloat.newFloat(getRuntime(),
zeroSign < 0 ? -0.0 : 0.0);
}
if (-value.scale() > RubyFloat.MAX_10_EXP) {
switch (value.signum()) {
case -1:
return RubyFloat.newFloat(getRuntime(), Double.NEGATIVE_INFINITY);
case 0:
return RubyFloat.newFloat(getRuntime(), 0);
case 1:
return RubyFloat.newFloat(getRuntime(), Double.POSITIVE_INFINITY);
default:
// eh?!
}
}
return RubyFloat.newFloat(getRuntime(), SafeDoubleParser.doubleValue(value));
}
@JRubyMethod(name = {"to_i", "to_int"}, compat = CompatVersion.RUBY1_8)
public IRubyObject to_int() {
if (isNaN() || infinitySign != 0) {
return getRuntime().getNil();
}
try {
return RubyNumeric.int2fix(getRuntime(), value.longValueExact());
} catch (ArithmeticException ae) {
return RubyBignum.bignorm(getRuntime(), value.toBigInteger());
}
}
@JRubyMethod(name = "to_r")
public IRubyObject to_r(ThreadContext context) {
checkFloatDomain();
RubyArray i = split();
long sign = ((Long)i.get(0));
String digits = (String)i.get(1).toString();
long base = ((Long)i.get(2));
long power = ((Long)i.get(3));
long denomi_power = power - digits.length();
IRubyObject bigDigits = RubyBignum.newBignum(getRuntime(), (String)digits).op_mul(context, sign);
RubyBignum numerator;
if(bigDigits instanceof RubyBignum) {
numerator = (RubyBignum)bigDigits;
}
else {
numerator = RubyBignum.newBignum(getRuntime(), bigDigits.toString());
}
IRubyObject num, den;
if(denomi_power < 0) {
num = numerator;
den = RubyFixnum.newFixnum(getRuntime(), base).op_mul(context, RubyFixnum.newFixnum(getRuntime(), -denomi_power));
}
else {
num = numerator.op_pow(context, RubyFixnum.newFixnum(getRuntime(), base).op_mul(context, RubyFixnum.newFixnum(getRuntime(), denomi_power)));
den = RubyFixnum.newFixnum(getRuntime(), 1);
}
return RubyRational.newInstance(context, context.runtime.getRational(), num, den);
}
@JRubyMethod(name = {"to_i", "to_int"}, compat = CompatVersion.RUBY1_9)
public IRubyObject to_int19() {
checkFloatDomain();
return to_int();
}
private String removeTrailingZeroes(String in) {
while(in.length() > 0 && in.charAt(in.length()-1)=='0') {
in = in.substring(0, in.length()-1);
}
return in;
}
public static boolean formatHasLeadingPlus(String format) {
return format.startsWith("+");
}
public static boolean formatHasLeadingSpace(String format) {
return format.startsWith(" ");
}
public static boolean formatHasFloatingPointNotation(String format) {
return format.endsWith("F");
}
public static int formatFractionalDigitGroups(String format) {
int groups = 0;
Pattern p = Pattern.compile("(\\+| )?(\\d+)(E|F)?");
Matcher m = p.matcher(format);
if (m.matches()) {
groups = Integer.parseInt(m.group(2));
}
return groups;
}
private static String firstArgument(IRubyObject[] args) {
if ( args.length == 0 ) return null;
final IRubyObject arg = args[0];
return arg.isNil() ? null : arg.toString();
}
private static boolean posSpace(String arg) {
if ( arg == null ) return false;
return formatHasLeadingSpace(arg);
}
private static boolean posSign(String arg) {
if ( arg == null ) return false;
return formatHasLeadingPlus(arg) || posSpace(arg);
}
private static boolean asEngineering(String arg) {
if ( arg == null ) return true;
return ! formatHasFloatingPointNotation(arg);
}
private static int groups(String arg) {
if (arg == null) return 0;
return formatFractionalDigitGroups(arg);
}
private boolean isZero() {
return !isNaN() && !isInfinity() && (value.signum() == 0);
}
private boolean isOne() {
return value.abs().compareTo(BigDecimal.ONE) == 0;
}
private boolean isNaN() {
return isNaN;
}
private boolean isInfinity() {
return infinitySign != 0;
}
private String unscaledValue() {
return value.abs().unscaledValue().toString();
}
private String sign(String arg, int signum) {
return signum == -1 ? "-" : (signum == 1 ? (posSign(arg) ? (posSpace(arg) ? " " : "+") : "") : "");
}
private CharSequence engineeringValue(String arg) {
StringBuilder build = new StringBuilder();
build.append( sign(arg, value.signum()) ).append("0.");
String s = removeTrailingZeroes(unscaledValue());
if (groups(arg) == 0) {
build.append("".equals(s) ? "0" : s);
} else {
int length = s.length();
String sep = "";
for (int index = 0; index < length; index += groups(arg)) {
int next = index + groups(arg);
build.append(sep).append(s.substring(index, next > length ? length : next));
sep = " ";
}
}
build.append('E').append(getExponent());
return build;
}
private CharSequence floatingPointValue(String arg) {
String values[] = value.abs().stripTrailingZeros().toPlainString().split("\\.");
String whole = "0";
if (values.length > 0) {
whole = values[0];
}
String after = "0";
if (values.length > 1) {
after = values[1];
}
int signum = value.signum();
StringBuilder build = new StringBuilder();
build.append(signum == -1 ? "-" : (signum == 1 ? (posSign(arg) ? (posSpace(arg) ? " " : "+") : "") : ""));
if (groups(arg) == 0) {
build.append(whole);
if (null != after) {
build.append(".").append(after);
}
} else {
int index = 0;
String sep = "";
while (index < whole.length()) {
int next = index + groups(arg);
if (next > whole.length()) {
next = whole.length();
}
build.append(sep).append(whole.substring(index, next));
sep = " ";
index += groups(arg);
}
if (null != after) {
build.append(".");
index = 0;
sep = "";
while (index < after.length()) {
int next = index + groups(arg);
if (next > after.length()) {
next = after.length();
}
build.append(sep).append(after.substring(index, next));
sep = " ";
index += groups(arg);
}
}
}
return build;
}
@JRubyMethod(name = "to_s", optional = 1)
public RubyString to_s(IRubyObject[] args) {
String arg = firstArgument(args);
if ( isNaN() ) return getRuntime().newString("NaN");
if ( infinitySign != 0 ) {
return getRuntime().newString(infinitySign == -1 ? "-Infinity" : "Infinity");
}
if ( isZero() ) {
return getRuntime().newString(zeroSign < 0 ? "-0.0" : "0.0");
}
return getRuntime().newString(
( asEngineering(arg) ? engineeringValue(arg) : floatingPointValue(arg) ).toString()
);
}
// Note: #fix has only no-arg form, but truncate allows optional parameter.
@JRubyMethod
public IRubyObject fix() {
IRubyObject[] ary = { RubyFixnum.zero(getRuntime()) };
return truncate(ary);
}
@JRubyMethod(name = "truncate", optional = 1, compat = CompatVersion.RUBY1_8)
public IRubyObject truncate(IRubyObject[] args) {
if (args.length == 0) {
IRubyObject[] ary = { RubyFixnum.zero(getRuntime()) };
return truncate(ary);
}
if (isNaN) {
return newNaN(getRuntime());
}
if (isInfinity()) {
return newInfinity(getRuntime(), infinitySign);
}
int n = RubyNumeric.fix2int(args[0]);
int precision = value.precision() - value.scale() + n;
if (precision > 0) {
return new RubyBigDecimal(getRuntime(),
value.round(new MathContext(precision, RoundingMode.DOWN)));
} else {
// TODO: proper sign
return new RubyBigDecimal(getRuntime(), BigDecimal.ZERO);
}
}
@JRubyMethod(name = "truncate", optional = 1, compat = CompatVersion.RUBY1_9)
public IRubyObject truncate19(IRubyObject[] args) {
if (args.length == 0) {
return ((RubyBigDecimal) truncate(args)).to_int19();
} else {
return truncate(args);
}
}
@JRubyMethod(name = "zero?")
public IRubyObject zero_p() {
return getRuntime().newBoolean(isZero());
}
/**
* Returns the correctly rounded square root of a positive
* BigDecimal. This method performs the fast Square Root by
* Coupled Newton Iteration algorithm by Timm Ahrendt, from
* the book "Pi, unleashed" by Jörg Arndt in a neat loop.
*
* The code is based on Frans Lelieveld's code , used here with
* permission.
*
* @param squarD The number to get the root from.
* @param rootMC Precision and rounding mode.
* @return the root of the argument number
* @throws ArithmeticException
* if the argument number is negative
* @throws IllegalArgumentException
* if rootMC has precision 0
* @see http://oldblog.novaloka.nl/blogger.xs4all.nl/novaloka/archive/2007/09/15/295396.html
*/
public static BigDecimal bigSqrt(BigDecimal squarD, MathContext rootMC) {
// General number and precision checking
int sign = squarD.signum();
if (sign == -1) {
throw new ArithmeticException("Square root of a negative number: " + squarD);
} else if (sign == 0) {
return squarD.round(rootMC);
}
int prec = rootMC.getPrecision(); // the requested precision
if (prec == 0) {
throw new IllegalArgumentException("Most roots won't have infinite precision = 0");
}
// Initial precision is that of double numbers 2^63/2 ~ 4E18
int BITS = 62; // 63-1 an even number of number bits
int nInit = 16; // precision seems 16 to 18 digits
MathContext nMC = new MathContext(18, RoundingMode.HALF_DOWN);
// Iteration variables, for the square root x and the reciprocal v
BigDecimal x = null, e = null; // initial x: x0 ~ sqrt()
BigDecimal v = null, g = null; // initial v: v0 = 1/(2*x)
// Estimate the square root with the foremost 62 bits of squarD
BigInteger bi = squarD.unscaledValue(); // bi and scale are a tandem
int biLen = bi.bitLength();
int shift = Math.max(0, biLen - BITS + (biLen%2 == 0 ? 0 : 1)); // even shift..
bi = bi.shiftRight(shift); // ..floors to 62 or 63 bit BigInteger
double root = Math.sqrt(SafeDoubleParser.doubleValue(bi));
BigDecimal halfBack = new BigDecimal(BigInteger.ONE.shiftLeft(shift/2));
int scale = squarD.scale();
if (scale % 2 == 1) {
root *= SQRT_10; // 5 -> 2, -5 -> -3 need half a scale more..
}
scale = (int) Math.ceil(scale/2.); // ..where 100 -> 10 shifts the scale
// Initial x - use double root - multiply by halfBack to unshift - set new scale
x = new BigDecimal(root, nMC);
x = x.multiply(halfBack, nMC); // x0 ~ sqrt()
if (scale != 0) {
x = x.movePointLeft(scale);
}
if (prec < nInit) { // for prec 15 root x0 must surely be OK
return x.round(rootMC); // return small prec roots without iterations
}
// Initial v - the reciprocal
v = BigDecimal.ONE.divide(TWO.multiply(x), nMC); // v0 = 1/(2*x)
// Collect iteration precisions beforehand
List nPrecs = new ArrayList();
assert nInit > 3 : "Never ending loop!"; // assume nInit = 16 <= prec
// Let m be the exact digits precision in an earlier! loop
for (int m = prec + 1; m > nInit; m = m/2 + (m > 100 ? 1 : 2)) {
nPrecs.add(m);
}
// The loop of "Square Root by Coupled Newton Iteration"
for (int i = nPrecs.size() - 1; i > -1; i--) {
// Increase precision - next iteration supplies n exact digits
nMC = new MathContext(nPrecs.get(i), (i%2 == 1) ? RoundingMode.HALF_UP :
RoundingMode.HALF_DOWN);
// Next x // e = d - x^2
e = squarD.subtract(x.multiply(x, nMC), nMC);
if (i != 0) {
x = x.add(e.multiply(v, nMC)); // x += e*v ~ sqrt()
} else {
x = x.add(e.multiply(v, rootMC), rootMC); // root x is ready!
break;
}
// Next v // g = 1 - 2*x*v
g = BigDecimal.ONE.subtract(TWO.multiply(x).multiply(v, nMC));
v = v.add(g.multiply(v, nMC)); // v += g*v ~ 1/2/sqrt()
}
return x; // return sqrt(squarD) with precision of rootMC
}
private void checkFloatDomain() {
if (isNaN) {
throw this.getRuntime().newFloatDomainError("NaN");
}
if (infinitySign != 0) {
if (infinitySign == -1) {
throw getRuntime().newFloatDomainError("-Infinity");
} else {
throw getRuntime().newFloatDomainError("Infinity");
}
}
}
private boolean is_even(IRubyObject x) {
if (x instanceof RubyFixnum) {
return RubyNumeric.fix2long((RubyFixnum) x) % 2 == 0;
}
if (x instanceof RubyBignum) {
return RubyBignum.big2long((RubyBignum) x) % 2 == 0;
}
return false;
}
}// RubyBigdecimal
© 2015 - 2025 Weber Informatics LLC | Privacy Policy