lib/tai/utils/decimal.ex

defmodule Tai.Utils.Decimal do

  @spec cast!(term, :normalize | term) :: Decimal.t | no_return
  def cast!(value, normalize \\ :ignore) do
    with {:ok, d} <- Decimal.cast(value) do
      case normalize do
        :normalize -> Decimal.normalize(d)
        _ -> d
      end
    else
      :error ->
        raise("#{inspect value} cannot be converted to Decimal")
    end
  end

  @spec round_up(Decimal.t(), Decimal.t()) :: Decimal.t()
  def round_up(val, increment) do
    {vc, ic, e} = same_order_of(val, increment)
    r = rem(vc, ic)

    if val.sign == 1 and r > 0 do
      v = vc + (ic - r)
      Decimal.new(val.sign, v, e)
    else
      v = vc - r
      Decimal.new(val.sign, v, e)
    end
  end

  @spec round_down(Decimal.t(), Decimal.t()) :: Decimal.t()
  def round_down(val, increment) do
    {vc, ic, e} = same_order_of(val, increment)
    r = rem(vc, ic)

    if val.sign == 1 or r == 0 do
      v = vc - r
      Decimal.new(val.sign, v, e)
    else
      v = vc + (ic - r)
      Decimal.new(val.sign, v, e)
    end
  end

  defp same_order_of(val, increment) do
    cond do
      val.exp < increment.exp ->
        s = abs(val.exp - increment.exp)
        ic = trunc(Tai.Utils.Math.pow(10, s)) * increment.coef
        {val.coef, ic, val.exp}

      increment.exp < val.exp ->
        s = abs(increment.exp - val.exp)
        vc = trunc(Tai.Utils.Math.pow(10, s)) * val.coef
        {vc, increment.coef, increment.exp}

      true ->
        {val.coef, increment.coef, val.exp}
    end
  end
end