mirror of
https://github.com/shopspring/decimal.git
synced 2024-11-22 20:40:48 +01:00
feat: RoundHalfTowardZero()
This commit is contained in:
parent
840ef45ca8
commit
54e62db847
2 changed files with 63 additions and 3 deletions
37
decimal.go
37
decimal.go
|
@ -1513,7 +1513,7 @@ func (d Decimal) StringFixedCash(interval uint8) string {
|
||||||
return rounded.string(false)
|
return rounded.string(false)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Round rounds the decimal to places decimal places.
|
// Round rounds the decimal to places decimal places (half away from zero).
|
||||||
// If places < 0, it will round the integer part to the nearest 10^(-places).
|
// If places < 0, it will round the integer part to the nearest 10^(-places).
|
||||||
//
|
//
|
||||||
// Example:
|
// Example:
|
||||||
|
@ -1544,6 +1544,37 @@ func (d Decimal) Round(places int32) Decimal {
|
||||||
return ret
|
return ret
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// RoundHalfTowardZero rounds the decimal to places decimal places (half toward zero).
|
||||||
|
// If places < 0, it will round the integer part to the nearest 10^(-places).
|
||||||
|
//
|
||||||
|
// Example:
|
||||||
|
//
|
||||||
|
// NewFromFloat(5.45).RoundHalfTowardZero(1).String() // output: "5.4"
|
||||||
|
// NewFromFloat(545).RoundHalfTowardZero(-1).String() // output: "540"
|
||||||
|
func (d Decimal) RoundHalfTowardZero(places int32) Decimal {
|
||||||
|
if d.exp == -places {
|
||||||
|
return d
|
||||||
|
}
|
||||||
|
// truncate to places + 1
|
||||||
|
ret := d.rescale(-places - 1)
|
||||||
|
|
||||||
|
// add sign(d) * 0.4
|
||||||
|
if ret.value.Sign() < 0 {
|
||||||
|
ret.value.Sub(ret.value, fourInt)
|
||||||
|
} else {
|
||||||
|
ret.value.Add(ret.value, fourInt)
|
||||||
|
}
|
||||||
|
|
||||||
|
// floor for positive numbers, ceil for negative numbers
|
||||||
|
_, m := ret.value.DivMod(ret.value, tenInt, new(big.Int))
|
||||||
|
ret.exp++
|
||||||
|
if ret.value.Sign() < 0 && m.Cmp(zeroInt) != 0 {
|
||||||
|
ret.value.Add(ret.value, oneInt)
|
||||||
|
}
|
||||||
|
|
||||||
|
return ret
|
||||||
|
}
|
||||||
|
|
||||||
// RoundHalfUp rounds the decimal half towards +infinity.
|
// RoundHalfUp rounds the decimal half towards +infinity.
|
||||||
//
|
//
|
||||||
// Example:
|
// Example:
|
||||||
|
@ -1560,7 +1591,7 @@ func (d Decimal) RoundHalfUp(places int32) Decimal {
|
||||||
// truncate to places + 1
|
// truncate to places + 1
|
||||||
ret := d.rescale(-places - 1)
|
ret := d.rescale(-places - 1)
|
||||||
|
|
||||||
// add sign(d) * 0.5
|
// add sign(d) * 0.5 if sign(d) >= 0 else sign(d) * 0.4
|
||||||
if ret.value.Sign() < 0 {
|
if ret.value.Sign() < 0 {
|
||||||
ret.value.Sub(ret.value, fourInt)
|
ret.value.Sub(ret.value, fourInt)
|
||||||
} else {
|
} else {
|
||||||
|
@ -1593,7 +1624,7 @@ func (d Decimal) RoundHalfDown(places int32) Decimal {
|
||||||
// truncate to places + 1
|
// truncate to places + 1
|
||||||
ret := d.rescale(-places - 1)
|
ret := d.rescale(-places - 1)
|
||||||
|
|
||||||
// add sign(d) * 0.5
|
// add sign(d) * 0.5 if sign(d) < 0 else sign(d) * 0.4
|
||||||
if ret.value.Sign() < 0 {
|
if ret.value.Sign() < 0 {
|
||||||
ret.value.Sub(ret.value, fiveInt)
|
ret.value.Sub(ret.value, fiveInt)
|
||||||
} else {
|
} else {
|
||||||
|
|
|
@ -3753,3 +3753,32 @@ func TestDecimal_RoundHalfDown(t *testing.T) {
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestDecimal_RoundHalfTowardZero(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
d Decimal
|
||||||
|
places int32
|
||||||
|
want Decimal
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "5.45",
|
||||||
|
d: NewFromFloat(5.45),
|
||||||
|
places: 1,
|
||||||
|
want: NewFromFloat(5.4),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "545",
|
||||||
|
d: NewFromInt(545),
|
||||||
|
places: -1,
|
||||||
|
want: NewFromInt(540),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
if got := tt.d.RoundHalfTowardZero(tt.places); !got.Equal(tt.want) {
|
||||||
|
t.Errorf("RoundHalfDown() = %v, want %v", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in a new issue