jepler/circuitpython
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

mpprint: Rework integer vararg handling.

Browse source 
This adds support for %llx (needed by XINT_FMT for printing cell
objects) and incidentally support for capitalized output of %P.

Signed-off-by: Jeff Epler <jepler@gmail.com>
This commit is contained in:
Jeff Epler 2025-07-05 15:09:06 +01:00
parent a7842a7866
commit 198876f66f
1 changed files with 50 additions and 39 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

89
py/mpprint.c
View file

@ -446,16 +446,26 @@ int mp_vprintf(const mp_print_t *print, const char *fmt, va_list args) {
}
}
// parse long specifiers (only for LP64 model where they make a difference)
#ifndef __LP64__
const
#endif
// parse long and long long specifiers (only where they make a difference)
#if LONG_MAX > INT_MAX
bool long_arg = false;
#endif
#if (MICROPY_OBJ_REPR == MICROPY_OBJ_REPR_D) || defined(_WIN64)
bool long_long_arg = false;
#endif
if (*fmt == 'l') {
++fmt;
#ifdef __LP64__
#if LONG_MAX > INT_MAX
long_arg = true;
#endif
#if (MICROPY_OBJ_REPR == MICROPY_OBJ_REPR_D) || defined(_WIN64)
if (*fmt == 'l') {
++fmt;
long_long_arg = true;
}
#endif
}
if (*fmt == '\0') {
@ -501,35 +511,48 @@ int mp_vprintf(const mp_print_t *print, const char *fmt, va_list args) {
chrs += mp_print_strn(print, str, len, flags, fill, width);
break;
}
case 'd': {
mp_int_t val;
if (long_arg) {
val = va_arg(args, long int);
} else {
val = va_arg(args, int);
}
chrs += mp_print_int(print, val, 1, 10, 'a', flags, fill, width);
break;
}
case 'p':
case 'P':
case 'd':
case 'u':
case 'x':
case 'X': {
int base = 16 - ((*fmt + 1) & 6); // maps char u/x/X to base 10/16/16
char fmt_c = (*fmt & 0xf0) - 'P' + 'A'; // maps char u/x/X to char a/a/A
char fmt_chr = *fmt;
mp_uint_t val;
if (long_arg) {
val = va_arg(args, unsigned long int);
} else {
val = va_arg(args, unsigned int);
if (fmt_chr == 'p' || fmt_chr == 'P') {
val = va_arg(args, intptr_t);
}
chrs += mp_print_int(print, val, 0, base, fmt_c, flags, fill, width);
#if (MICROPY_OBJ_REPR == MICROPY_OBJ_REPR_D) || defined(_WIN64)
else if (long_long_arg) {
val = va_arg(args, unsigned long long);
}
#endif
#if LONG_MAX > INT_MAX
else if (long_arg) {
if (sizeof(long) != sizeof(mp_uint_t) && fmt_chr == 'd') {
val = va_arg(args, long);
} else {
val = va_arg(args, unsigned long);
}
}
#endif
else {
if (sizeof(int) != sizeof(mp_uint_t) && fmt_chr == 'd') {
val = va_arg(args, int);
} else {
val = va_arg(args, unsigned);
}
}
int base;
char fmt_c = (fmt_chr & 0xf0) - 'P' + 'A'; // maps char u/x/X to char a/a/A
if (fmt_chr == 'd' || fmt_chr == 'u') {
base = 10;
} else {
base = 16;
}
chrs += mp_print_int(print, val, fmt_chr == 'd', base, fmt_c, flags, fill, width);
break;
}
case 'p':
case 'P': // don't bother to handle upcase for 'P'
// Use unsigned long int to work on both ILP32 and LP64 systems
chrs += mp_print_int(print, va_arg(args, unsigned long int), 0, 16, 'a', flags, fill, width);
break;
#if MICROPY_PY_BUILTINS_FLOAT
case 'e':
case 'E':
@ -545,18 +568,6 @@ int mp_vprintf(const mp_print_t *print, const char *fmt, va_list args) {
#endif
break;
}
#endif
// Because 'l' is eaten above, another 'l' means %ll. We need to support
// this length specifier for OBJ_REPR_D (64-bit NaN boxing).
// TODO Either enable this unconditionally, or provide a specific config var.
#if (MICROPY_OBJ_REPR == MICROPY_OBJ_REPR_D) || defined(_WIN64)
case 'l': {
unsigned long long int arg_value = va_arg(args, unsigned long long int);
++fmt;
assert(*fmt == 'u' || *fmt == 'd' || !"unsupported fmt char");
chrs += mp_print_int(print, arg_value, *fmt == 'd', 10, 'a', flags, fill, width);
break;
}
#endif
default:
// if it's not %% then it's an unsupported format character