mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
Refactor "%f" % Inf/NaN
* sprintf.c (rb_str_format): as for non-finite float, calculate the exact needed size with the space flag. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@58454 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
This commit is contained in:
parent
6f8f950cc3
commit
be7309ddf6
1 changed files with 17 additions and 18 deletions
35
sprintf.c
35
sprintf.c
|
@ -1128,6 +1128,8 @@ rb_str_format(int argc, const VALUE *argv, VALUE fmt)
|
|||
fval = RFLOAT_VALUE(rb_Float(val));
|
||||
if (isnan(fval) || isinf(fval)) {
|
||||
const char *expr;
|
||||
int elen;
|
||||
char sign = '\0';
|
||||
|
||||
if (isnan(fval)) {
|
||||
expr = "NaN";
|
||||
|
@ -1136,31 +1138,28 @@ rb_str_format(int argc, const VALUE *argv, VALUE fmt)
|
|||
expr = "Inf";
|
||||
}
|
||||
need = (int)strlen(expr);
|
||||
if ((!isnan(fval) && fval < 0.0) || (flags & (FPLUS|FSPACE)))
|
||||
need++;
|
||||
elen = need;
|
||||
i = 0;
|
||||
if (!isnan(fval) && fval < 0.0)
|
||||
sign = '-';
|
||||
else if (flags & (FPLUS|FSPACE))
|
||||
sign = (flags & FPLUS) ? '+' : ' ';
|
||||
if (sign)
|
||||
++need;
|
||||
if ((flags & FWIDTH) && need < width)
|
||||
need = width;
|
||||
|
||||
CHECK(need + 1);
|
||||
snprintf(&buf[blen], need + 1, "%*s", need, "");
|
||||
FILL(' ', need);
|
||||
if (flags & FMINUS) {
|
||||
if (!isnan(fval) && fval < 0.0)
|
||||
buf[blen++] = '-';
|
||||
else if (flags & FPLUS)
|
||||
buf[blen++] = '+';
|
||||
else if (flags & FSPACE)
|
||||
blen++;
|
||||
memcpy(&buf[blen], expr, strlen(expr));
|
||||
if (sign)
|
||||
buf[blen - need--] = sign;
|
||||
memcpy(&buf[blen - need], expr, elen);
|
||||
}
|
||||
else {
|
||||
if (!isnan(fval) && fval < 0.0)
|
||||
buf[blen + need - strlen(expr) - 1] = '-';
|
||||
else if (flags & FPLUS)
|
||||
buf[blen + need - strlen(expr) - 1] = '+';
|
||||
memcpy(&buf[blen + need - strlen(expr)], expr,
|
||||
strlen(expr));
|
||||
if (sign)
|
||||
buf[blen - elen - 1] = sign;
|
||||
memcpy(&buf[blen - elen], expr, elen);
|
||||
}
|
||||
blen += strlen(&buf[blen]);
|
||||
break;
|
||||
}
|
||||
|
||||
|
|
Loading…
Add table
Reference in a new issue