Austin Schuh | 208337d | 2022-01-01 14:29:11 -0800 | [diff] [blame^] | 1 | /////////////////////////////////////////////////////////////////////////////// |
| 2 | // \author (c) Marco Paland (info@paland.com) |
| 3 | // 2014-2019, PALANDesign Hannover, Germany |
| 4 | // |
| 5 | // \license The MIT License (MIT) |
| 6 | // |
| 7 | // Permission is hereby granted, free of charge, to any person obtaining a copy |
| 8 | // of this software and associated documentation files (the "Software"), to deal |
| 9 | // in the Software without restriction, including without limitation the rights |
| 10 | // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
| 11 | // copies of the Software, and to permit persons to whom the Software is |
| 12 | // furnished to do so, subject to the following conditions: |
| 13 | // |
| 14 | // The above copyright notice and this permission notice shall be included in |
| 15 | // all copies or substantial portions of the Software. |
| 16 | // |
| 17 | // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| 18 | // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
| 19 | // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
| 20 | // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
| 21 | // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
| 22 | // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
| 23 | // THE SOFTWARE. |
| 24 | // |
| 25 | // \brief Tiny printf, sprintf and (v)snprintf implementation, optimized for speed on |
| 26 | // embedded systems with a very limited resources. These routines are thread |
| 27 | // safe and reentrant! |
| 28 | // Use this instead of the bloated standard/newlib printf cause these use |
| 29 | // malloc for printf (and may not be thread safe). |
| 30 | // |
| 31 | /////////////////////////////////////////////////////////////////////////////// |
| 32 | |
| 33 | #include <stdbool.h> |
| 34 | #include <stdint.h> |
| 35 | #include <stdio.h> |
| 36 | |
| 37 | #include "pico/platform.h" |
| 38 | #include "pico/printf.h" |
| 39 | |
| 40 | // PICO_CONFIG: PICO_PRINTF_NTOA_BUFFER_SIZE, Define printf ntoa buffer size, min=0, max=128, default=32, group=pico_printf |
| 41 | // 'ntoa' conversion buffer size, this must be big enough to hold one converted |
| 42 | // numeric number including padded zeros (dynamically created on stack) |
| 43 | #ifndef PICO_PRINTF_NTOA_BUFFER_SIZE |
| 44 | #define PICO_PRINTF_NTOA_BUFFER_SIZE 32U |
| 45 | #endif |
| 46 | |
| 47 | // PICO_CONFIG: PICO_PRINTF_FTOA_BUFFER_SIZE, Define printf ftoa buffer size, min=0, max=128, default=32, group=pico_printf |
| 48 | // 'ftoa' conversion buffer size, this must be big enough to hold one converted |
| 49 | // float number including padded zeros (dynamically created on stack) |
| 50 | #ifndef PICO_PRINTF_FTOA_BUFFER_SIZE |
| 51 | #define PICO_PRINTF_FTOA_BUFFER_SIZE 32U |
| 52 | #endif |
| 53 | |
| 54 | // PICO_CONFIG: PICO_PRINTF_SUPPORT_FLOAT, Enable floating point printing, type=bool, default=1, group=pico_printf |
| 55 | // support for the floating point type (%f) |
| 56 | #ifndef PICO_PRINTF_SUPPORT_FLOAT |
| 57 | #define PICO_PRINTF_SUPPORT_FLOAT 1 |
| 58 | #endif |
| 59 | |
| 60 | // PICO_CONFIG: PICO_PRINTF_SUPPORT_EXPONENTIAL, Enable exponential floating point printing, type=bool, default=1, group=pico_printf |
| 61 | // support for exponential floating point notation (%e/%g) |
| 62 | #ifndef PICO_PRINTF_SUPPORT_EXPONENTIAL |
| 63 | #define PICO_PRINTF_SUPPORT_EXPONENTIAL 1 |
| 64 | #endif |
| 65 | |
| 66 | // PICO_CONFIG: PICO_PRINTF_DEFAULT_FLOAT_PRECISION, Define default floating point precision, min=1, max=16, default=6, group=pico_printf |
| 67 | #ifndef PICO_PRINTF_DEFAULT_FLOAT_PRECISION |
| 68 | #define PICO_PRINTF_DEFAULT_FLOAT_PRECISION 6U |
| 69 | #endif |
| 70 | |
| 71 | // PICO_CONFIG: PICO_PRINTF_MAX_FLOAT, Define the largest float suitable to print with %f, min=1, max=1e9, default=1e9, group=pico_printf |
| 72 | #ifndef PICO_PRINTF_MAX_FLOAT |
| 73 | #define PICO_PRINTF_MAX_FLOAT 1e9 |
| 74 | #endif |
| 75 | |
| 76 | // PICO_CONFIG: PICO_PRINTF_SUPPORT_LONG_LONG, Enable support for long long types (%llu or %p), type=bool, default=1, group=pico_printf |
| 77 | #ifndef PICO_PRINTF_SUPPORT_LONG_LONG |
| 78 | #define PICO_PRINTF_SUPPORT_LONG_LONG 1 |
| 79 | #endif |
| 80 | |
| 81 | // PICO_CONFIG: PICO_PRINTF_SUPPORT_PTRDIFF_T, Enable support for the ptrdiff_t type (%t), type=bool, default=1, group=pico_printf |
| 82 | // ptrdiff_t is normally defined in <stddef.h> as long or long long type |
| 83 | #ifndef PICO_PRINTF_SUPPORT_PTRDIFF_T |
| 84 | #define PICO_PRINTF_SUPPORT_PTRDIFF_T 1 |
| 85 | #endif |
| 86 | |
| 87 | /////////////////////////////////////////////////////////////////////////////// |
| 88 | |
| 89 | // internal flag definitions |
| 90 | #define FLAGS_ZEROPAD (1U << 0U) |
| 91 | #define FLAGS_LEFT (1U << 1U) |
| 92 | #define FLAGS_PLUS (1U << 2U) |
| 93 | #define FLAGS_SPACE (1U << 3U) |
| 94 | #define FLAGS_HASH (1U << 4U) |
| 95 | #define FLAGS_UPPERCASE (1U << 5U) |
| 96 | #define FLAGS_CHAR (1U << 6U) |
| 97 | #define FLAGS_SHORT (1U << 7U) |
| 98 | #define FLAGS_LONG (1U << 8U) |
| 99 | #define FLAGS_LONG_LONG (1U << 9U) |
| 100 | #define FLAGS_PRECISION (1U << 10U) |
| 101 | #define FLAGS_ADAPT_EXP (1U << 11U) |
| 102 | |
| 103 | // import float.h for DBL_MAX |
| 104 | #if PICO_PRINTF_SUPPORT_FLOAT |
| 105 | |
| 106 | #include <float.h> |
| 107 | |
| 108 | #endif |
| 109 | |
| 110 | /** |
| 111 | * Output a character to a custom device like UART, used by the printf() function |
| 112 | * This function is declared here only. You have to write your custom implementation somewhere |
| 113 | * \param character Character to output |
| 114 | */ |
| 115 | static void _putchar(char character) { |
| 116 | putchar(character); |
| 117 | } |
| 118 | |
| 119 | // output function type |
| 120 | typedef void (*out_fct_type)(char character, void *buffer, size_t idx, size_t maxlen); |
| 121 | |
| 122 | #if !PICO_PRINTF_ALWAYS_INCLUDED |
| 123 | // we don't have a way to specify a truly weak symbol reference (the linker will always include targets in a single link step, |
| 124 | // so we make a function pointer that is initialized on the first printf called... if printf is not included in the binary |
| 125 | // (or has never been called - we can't tell) then this will be null. the assumption is that if you are using printf |
| 126 | // you are likely to have printed something. |
| 127 | static int (*lazy_vsnprintf)(out_fct_type out, char *buffer, const size_t maxlen, const char *format, va_list va); |
| 128 | #endif |
| 129 | |
| 130 | // wrapper (used as buffer) for output function type |
| 131 | typedef struct { |
| 132 | void (*fct)(char character, void *arg); |
| 133 | void *arg; |
| 134 | } out_fct_wrap_type; |
| 135 | |
| 136 | // internal buffer output |
| 137 | static inline void _out_buffer(char character, void *buffer, size_t idx, size_t maxlen) { |
| 138 | if (idx < maxlen) { |
| 139 | ((char *) buffer)[idx] = character; |
| 140 | } |
| 141 | } |
| 142 | |
| 143 | // internal null output |
| 144 | static inline void _out_null(char character, void *buffer, size_t idx, size_t maxlen) { |
| 145 | (void) character; |
| 146 | (void) buffer; |
| 147 | (void) idx; |
| 148 | (void) maxlen; |
| 149 | } |
| 150 | |
| 151 | // internal _putchar wrapper |
| 152 | static inline void _out_char(char character, void *buffer, size_t idx, size_t maxlen) { |
| 153 | (void) buffer; |
| 154 | (void) idx; |
| 155 | (void) maxlen; |
| 156 | if (character) { |
| 157 | _putchar(character); |
| 158 | } |
| 159 | } |
| 160 | |
| 161 | |
| 162 | // internal output function wrapper |
| 163 | static inline void _out_fct(char character, void *buffer, size_t idx, size_t maxlen) { |
| 164 | (void) idx; |
| 165 | (void) maxlen; |
| 166 | if (character) { |
| 167 | // buffer is the output fct pointer |
| 168 | ((out_fct_wrap_type *) buffer)->fct(character, ((out_fct_wrap_type *) buffer)->arg); |
| 169 | } |
| 170 | } |
| 171 | |
| 172 | |
| 173 | // internal secure strlen |
| 174 | // \return The length of the string (excluding the terminating 0) limited by 'maxsize' |
| 175 | static inline unsigned int _strnlen_s(const char *str, size_t maxsize) { |
| 176 | const char *s; |
| 177 | for (s = str; *s && maxsize--; ++s); |
| 178 | return (unsigned int) (s - str); |
| 179 | } |
| 180 | |
| 181 | |
| 182 | // internal test if char is a digit (0-9) |
| 183 | // \return true if char is a digit |
| 184 | static inline bool _is_digit(char ch) { |
| 185 | return (ch >= '0') && (ch <= '9'); |
| 186 | } |
| 187 | |
| 188 | |
| 189 | // internal ASCII string to unsigned int conversion |
| 190 | static unsigned int _atoi(const char **str) { |
| 191 | unsigned int i = 0U; |
| 192 | while (_is_digit(**str)) { |
| 193 | i = i * 10U + (unsigned int) (*((*str)++) - '0'); |
| 194 | } |
| 195 | return i; |
| 196 | } |
| 197 | |
| 198 | |
| 199 | // output the specified string in reverse, taking care of any zero-padding |
| 200 | static size_t _out_rev(out_fct_type out, char *buffer, size_t idx, size_t maxlen, const char *buf, size_t len, |
| 201 | unsigned int width, unsigned int flags) { |
| 202 | const size_t start_idx = idx; |
| 203 | |
| 204 | // pad spaces up to given width |
| 205 | if (!(flags & FLAGS_LEFT) && !(flags & FLAGS_ZEROPAD)) { |
| 206 | for (size_t i = len; i < width; i++) { |
| 207 | out(' ', buffer, idx++, maxlen); |
| 208 | } |
| 209 | } |
| 210 | |
| 211 | // reverse string |
| 212 | while (len) { |
| 213 | out(buf[--len], buffer, idx++, maxlen); |
| 214 | } |
| 215 | |
| 216 | // append pad spaces up to given width |
| 217 | if (flags & FLAGS_LEFT) { |
| 218 | while (idx - start_idx < width) { |
| 219 | out(' ', buffer, idx++, maxlen); |
| 220 | } |
| 221 | } |
| 222 | |
| 223 | return idx; |
| 224 | } |
| 225 | |
| 226 | |
| 227 | // internal itoa format |
| 228 | static size_t _ntoa_format(out_fct_type out, char *buffer, size_t idx, size_t maxlen, char *buf, size_t len, |
| 229 | bool negative, unsigned int base, unsigned int prec, unsigned int width, |
| 230 | unsigned int flags) { |
| 231 | // pad leading zeros |
| 232 | if (!(flags & FLAGS_LEFT)) { |
| 233 | if (width && (flags & FLAGS_ZEROPAD) && (negative || (flags & (FLAGS_PLUS | FLAGS_SPACE)))) { |
| 234 | width--; |
| 235 | } |
| 236 | while ((len < prec) && (len < PICO_PRINTF_NTOA_BUFFER_SIZE)) { |
| 237 | buf[len++] = '0'; |
| 238 | } |
| 239 | while ((flags & FLAGS_ZEROPAD) && (len < width) && (len < PICO_PRINTF_NTOA_BUFFER_SIZE)) { |
| 240 | buf[len++] = '0'; |
| 241 | } |
| 242 | } |
| 243 | |
| 244 | // handle hash |
| 245 | if (flags & FLAGS_HASH) { |
| 246 | if (!(flags & FLAGS_PRECISION) && len && ((len == prec) || (len == width))) { |
| 247 | len--; |
| 248 | if (len && (base == 16U)) { |
| 249 | len--; |
| 250 | } |
| 251 | } |
| 252 | if ((base == 16U) && !(flags & FLAGS_UPPERCASE) && (len < PICO_PRINTF_NTOA_BUFFER_SIZE)) { |
| 253 | buf[len++] = 'x'; |
| 254 | } else if ((base == 16U) && (flags & FLAGS_UPPERCASE) && (len < PICO_PRINTF_NTOA_BUFFER_SIZE)) { |
| 255 | buf[len++] = 'X'; |
| 256 | } else if ((base == 2U) && (len < PICO_PRINTF_NTOA_BUFFER_SIZE)) { |
| 257 | buf[len++] = 'b'; |
| 258 | } |
| 259 | if (len < PICO_PRINTF_NTOA_BUFFER_SIZE) { |
| 260 | buf[len++] = '0'; |
| 261 | } |
| 262 | } |
| 263 | |
| 264 | if (len < PICO_PRINTF_NTOA_BUFFER_SIZE) { |
| 265 | if (negative) { |
| 266 | buf[len++] = '-'; |
| 267 | } else if (flags & FLAGS_PLUS) { |
| 268 | buf[len++] = '+'; // ignore the space if the '+' exists |
| 269 | } else if (flags & FLAGS_SPACE) { |
| 270 | buf[len++] = ' '; |
| 271 | } |
| 272 | } |
| 273 | |
| 274 | return _out_rev(out, buffer, idx, maxlen, buf, len, width, flags); |
| 275 | } |
| 276 | |
| 277 | |
| 278 | // internal itoa for 'long' type |
| 279 | static size_t _ntoa_long(out_fct_type out, char *buffer, size_t idx, size_t maxlen, unsigned long value, bool negative, |
| 280 | unsigned long base, unsigned int prec, unsigned int width, unsigned int flags) { |
| 281 | char buf[PICO_PRINTF_NTOA_BUFFER_SIZE]; |
| 282 | size_t len = 0U; |
| 283 | |
| 284 | // no hash for 0 values |
| 285 | if (!value) { |
| 286 | flags &= ~FLAGS_HASH; |
| 287 | } |
| 288 | |
| 289 | // write if precision != 0 and value is != 0 |
| 290 | if (!(flags & FLAGS_PRECISION) || value) { |
| 291 | do { |
| 292 | const char digit = (char) (value % base); |
| 293 | buf[len++] = (char)(digit < 10 ? '0' + digit : (flags & FLAGS_UPPERCASE ? 'A' : 'a') + digit - 10); |
| 294 | value /= base; |
| 295 | } while (value && (len < PICO_PRINTF_NTOA_BUFFER_SIZE)); |
| 296 | } |
| 297 | |
| 298 | return _ntoa_format(out, buffer, idx, maxlen, buf, len, negative, (unsigned int) base, prec, width, flags); |
| 299 | } |
| 300 | |
| 301 | |
| 302 | // internal itoa for 'long long' type |
| 303 | #if PICO_PRINTF_SUPPORT_LONG_LONG |
| 304 | |
| 305 | static size_t _ntoa_long_long(out_fct_type out, char *buffer, size_t idx, size_t maxlen, unsigned long long value, |
| 306 | bool negative, unsigned long long base, unsigned int prec, unsigned int width, |
| 307 | unsigned int flags) { |
| 308 | char buf[PICO_PRINTF_NTOA_BUFFER_SIZE]; |
| 309 | size_t len = 0U; |
| 310 | |
| 311 | // no hash for 0 values |
| 312 | if (!value) { |
| 313 | flags &= ~FLAGS_HASH; |
| 314 | } |
| 315 | |
| 316 | // write if precision != 0 and value is != 0 |
| 317 | if (!(flags & FLAGS_PRECISION) || value) { |
| 318 | do { |
| 319 | const char digit = (char) (value % base); |
| 320 | buf[len++] = (char)(digit < 10 ? '0' + digit : (flags & FLAGS_UPPERCASE ? 'A' : 'a') + digit - 10); |
| 321 | value /= base; |
| 322 | } while (value && (len < PICO_PRINTF_NTOA_BUFFER_SIZE)); |
| 323 | } |
| 324 | |
| 325 | return _ntoa_format(out, buffer, idx, maxlen, buf, len, negative, (unsigned int) base, prec, width, flags); |
| 326 | } |
| 327 | |
| 328 | #endif // PICO_PRINTF_SUPPORT_LONG_LONG |
| 329 | |
| 330 | |
| 331 | #if PICO_PRINTF_SUPPORT_FLOAT |
| 332 | |
| 333 | #if PICO_PRINTF_SUPPORT_EXPONENTIAL |
| 334 | // forward declaration so that _ftoa can switch to exp notation for values > PICO_PRINTF_MAX_FLOAT |
| 335 | static size_t _etoa(out_fct_type out, char *buffer, size_t idx, size_t maxlen, double value, unsigned int prec, |
| 336 | unsigned int width, unsigned int flags); |
| 337 | #endif |
| 338 | |
| 339 | #define is_nan __builtin_isnan |
| 340 | |
| 341 | // internal ftoa for fixed decimal floating point |
| 342 | static size_t _ftoa(out_fct_type out, char *buffer, size_t idx, size_t maxlen, double value, unsigned int prec, |
| 343 | unsigned int width, unsigned int flags) { |
| 344 | char buf[PICO_PRINTF_FTOA_BUFFER_SIZE]; |
| 345 | size_t len = 0U; |
| 346 | double diff = 0.0; |
| 347 | |
| 348 | // powers of 10 |
| 349 | static const double pow10[] = {1, 10, 100, 1000, 10000, 100000, 1000000, 10000000, 100000000, 1000000000}; |
| 350 | |
| 351 | // test for special values |
| 352 | if (is_nan(value)) |
| 353 | return _out_rev(out, buffer, idx, maxlen, "nan", 3, width, flags); |
| 354 | if (value < -DBL_MAX) |
| 355 | return _out_rev(out, buffer, idx, maxlen, "fni-", 4, width, flags); |
| 356 | if (value > DBL_MAX) |
| 357 | return _out_rev(out, buffer, idx, maxlen, (flags & FLAGS_PLUS) ? "fni+" : "fni", (flags & FLAGS_PLUS) ? 4U : 3U, |
| 358 | width, flags); |
| 359 | |
| 360 | // test for very large values |
| 361 | // standard printf behavior is to print EVERY whole number digit -- which could be 100s of characters overflowing your buffers == bad |
| 362 | if ((value > PICO_PRINTF_MAX_FLOAT) || (value < -PICO_PRINTF_MAX_FLOAT)) { |
| 363 | #if PICO_PRINTF_SUPPORT_EXPONENTIAL |
| 364 | return _etoa(out, buffer, idx, maxlen, value, prec, width, flags); |
| 365 | #else |
| 366 | return 0U; |
| 367 | #endif |
| 368 | } |
| 369 | |
| 370 | // test for negative |
| 371 | bool negative = false; |
| 372 | if (value < 0) { |
| 373 | negative = true; |
| 374 | value = 0 - value; |
| 375 | } |
| 376 | |
| 377 | // set default precision, if not set explicitly |
| 378 | if (!(flags & FLAGS_PRECISION)) { |
| 379 | prec = PICO_PRINTF_DEFAULT_FLOAT_PRECISION; |
| 380 | } |
| 381 | // limit precision to 9, cause a prec >= 10 can lead to overflow errors |
| 382 | while ((len < PICO_PRINTF_FTOA_BUFFER_SIZE) && (prec > 9U)) { |
| 383 | buf[len++] = '0'; |
| 384 | prec--; |
| 385 | } |
| 386 | |
| 387 | int whole = (int) value; |
| 388 | double tmp = (value - whole) * pow10[prec]; |
| 389 | unsigned long frac = (unsigned long) tmp; |
| 390 | diff = tmp - frac; |
| 391 | |
| 392 | if (diff > 0.5) { |
| 393 | ++frac; |
| 394 | // handle rollover, e.g. case 0.99 with prec 1 is 1.0 |
| 395 | if (frac >= pow10[prec]) { |
| 396 | frac = 0; |
| 397 | ++whole; |
| 398 | } |
| 399 | } else if (diff < 0.5) { |
| 400 | } else if ((frac == 0U) || (frac & 1U)) { |
| 401 | // if halfway, round up if odd OR if last digit is 0 |
| 402 | ++frac; |
| 403 | } |
| 404 | |
| 405 | if (prec == 0U) { |
| 406 | diff = value - (double) whole; |
| 407 | if (!((diff < 0.5) || (diff > 0.5)) && (whole & 1)) { |
| 408 | // exactly 0.5 and ODD, then round up |
| 409 | // 1.5 -> 2, but 2.5 -> 2 |
| 410 | ++whole; |
| 411 | } |
| 412 | } else { |
| 413 | unsigned int count = prec; |
| 414 | // now do fractional part, as an unsigned number |
| 415 | while (len < PICO_PRINTF_FTOA_BUFFER_SIZE) { |
| 416 | --count; |
| 417 | buf[len++] = (char) (48U + (frac % 10U)); |
| 418 | if (!(frac /= 10U)) { |
| 419 | break; |
| 420 | } |
| 421 | } |
| 422 | // add extra 0s |
| 423 | while ((len < PICO_PRINTF_FTOA_BUFFER_SIZE) && (count-- > 0U)) { |
| 424 | buf[len++] = '0'; |
| 425 | } |
| 426 | if (len < PICO_PRINTF_FTOA_BUFFER_SIZE) { |
| 427 | // add decimal |
| 428 | buf[len++] = '.'; |
| 429 | } |
| 430 | } |
| 431 | |
| 432 | // do whole part, number is reversed |
| 433 | while (len < PICO_PRINTF_FTOA_BUFFER_SIZE) { |
| 434 | buf[len++] = (char) (48 + (whole % 10)); |
| 435 | if (!(whole /= 10)) { |
| 436 | break; |
| 437 | } |
| 438 | } |
| 439 | |
| 440 | // pad leading zeros |
| 441 | if (!(flags & FLAGS_LEFT) && (flags & FLAGS_ZEROPAD)) { |
| 442 | if (width && (negative || (flags & (FLAGS_PLUS | FLAGS_SPACE)))) { |
| 443 | width--; |
| 444 | } |
| 445 | while ((len < width) && (len < PICO_PRINTF_FTOA_BUFFER_SIZE)) { |
| 446 | buf[len++] = '0'; |
| 447 | } |
| 448 | } |
| 449 | |
| 450 | if (len < PICO_PRINTF_FTOA_BUFFER_SIZE) { |
| 451 | if (negative) { |
| 452 | buf[len++] = '-'; |
| 453 | } else if (flags & FLAGS_PLUS) { |
| 454 | buf[len++] = '+'; // ignore the space if the '+' exists |
| 455 | } else if (flags & FLAGS_SPACE) { |
| 456 | buf[len++] = ' '; |
| 457 | } |
| 458 | } |
| 459 | |
| 460 | return _out_rev(out, buffer, idx, maxlen, buf, len, width, flags); |
| 461 | } |
| 462 | |
| 463 | |
| 464 | #if PICO_PRINTF_SUPPORT_EXPONENTIAL |
| 465 | |
| 466 | // internal ftoa variant for exponential floating-point type, contributed by Martijn Jasperse <m.jasperse@gmail.com> |
| 467 | static size_t _etoa(out_fct_type out, char *buffer, size_t idx, size_t maxlen, double value, unsigned int prec, |
| 468 | unsigned int width, unsigned int flags) { |
| 469 | // check for NaN and special values |
| 470 | if (is_nan(value) || (value > DBL_MAX) || (value < -DBL_MAX)) { |
| 471 | return _ftoa(out, buffer, idx, maxlen, value, prec, width, flags); |
| 472 | } |
| 473 | |
| 474 | // determine the sign |
| 475 | const bool negative = value < 0; |
| 476 | if (negative) { |
| 477 | value = -value; |
| 478 | } |
| 479 | |
| 480 | // default precision |
| 481 | if (!(flags & FLAGS_PRECISION)) { |
| 482 | prec = PICO_PRINTF_DEFAULT_FLOAT_PRECISION; |
| 483 | } |
| 484 | |
| 485 | // determine the decimal exponent |
| 486 | // based on the algorithm by David Gay (https://www.ampl.com/netlib/fp/dtoa.c) |
| 487 | union { |
| 488 | uint64_t U; |
| 489 | double F; |
| 490 | } conv; |
| 491 | |
| 492 | conv.F = value; |
| 493 | int expval; |
| 494 | if (conv.U) { |
| 495 | int exp2 = (int) ((conv.U >> 52U) & 0x07FFU) - 1023; // effectively log2 |
| 496 | conv.U = (conv.U & ((1ULL << 52U) - 1U)) | (1023ULL << 52U); // drop the exponent so conv.F is now in [1,2) |
| 497 | // now approximate log10 from the log2 integer part and an expansion of ln around 1.5 |
| 498 | expval = (int) (0.1760912590558 + exp2 * 0.301029995663981 + (conv.F - 1.5) * 0.289529654602168); |
| 499 | // now we want to compute 10^expval but we want to be sure it won't overflow |
| 500 | exp2 = (int) (expval * 3.321928094887362 + 0.5); |
| 501 | const double z = expval * 2.302585092994046 - exp2 * 0.6931471805599453; |
| 502 | const double z2 = z * z; |
| 503 | conv.U = (uint64_t) (exp2 + 1023) << 52U; |
| 504 | // compute exp(z) using continued fractions, see https://en.wikipedia.org/wiki/Exponential_function#Continued_fractions_for_ex |
| 505 | conv.F *= 1 + 2 * z / (2 - z + (z2 / (6 + (z2 / (10 + z2 / 14))))); |
| 506 | // correct for rounding errors |
| 507 | if (value < conv.F) { |
| 508 | expval--; |
| 509 | conv.F /= 10; |
| 510 | } |
| 511 | } else { |
| 512 | expval = 0; |
| 513 | } |
| 514 | |
| 515 | // the exponent format is "%+03d" and largest value is "307", so set aside 4-5 characters |
| 516 | unsigned int minwidth = ((expval < 100) && (expval > -100)) ? 4U : 5U; |
| 517 | |
| 518 | // in "%g" mode, "prec" is the number of *significant figures* not decimals |
| 519 | if (flags & FLAGS_ADAPT_EXP) { |
| 520 | // do we want to fall-back to "%f" mode? |
| 521 | if ((conv.U == 0) || ((value >= 1e-4) && (value < 1e6))) { |
| 522 | if ((int) prec > expval) { |
| 523 | prec = (unsigned) ((int) prec - expval - 1); |
| 524 | } else { |
| 525 | prec = 0; |
| 526 | } |
| 527 | flags |= FLAGS_PRECISION; // make sure _ftoa respects precision |
| 528 | // no characters in exponent |
| 529 | minwidth = 0U; |
| 530 | expval = 0; |
| 531 | } else { |
| 532 | // we use one sigfig for the whole part |
| 533 | if ((prec > 0) && (flags & FLAGS_PRECISION)) { |
| 534 | --prec; |
| 535 | } |
| 536 | } |
| 537 | } |
| 538 | |
| 539 | // will everything fit? |
| 540 | unsigned int fwidth = width; |
| 541 | if (width > minwidth) { |
| 542 | // we didn't fall-back so subtract the characters required for the exponent |
| 543 | fwidth -= minwidth; |
| 544 | } else { |
| 545 | // not enough characters, so go back to default sizing |
| 546 | fwidth = 0U; |
| 547 | } |
| 548 | if ((flags & FLAGS_LEFT) && minwidth) { |
| 549 | // if we're padding on the right, DON'T pad the floating part |
| 550 | fwidth = 0U; |
| 551 | } |
| 552 | |
| 553 | // rescale the float value |
| 554 | if (expval) { |
| 555 | value /= conv.F; |
| 556 | } |
| 557 | |
| 558 | // output the floating part |
| 559 | const size_t start_idx = idx; |
| 560 | idx = _ftoa(out, buffer, idx, maxlen, negative ? -value : value, prec, fwidth, flags & ~FLAGS_ADAPT_EXP); |
| 561 | |
| 562 | // output the exponent part |
| 563 | if (minwidth) { |
| 564 | // output the exponential symbol |
| 565 | out((flags & FLAGS_UPPERCASE) ? 'E' : 'e', buffer, idx++, maxlen); |
| 566 | // output the exponent value |
| 567 | idx = _ntoa_long(out, buffer, idx, maxlen, (uint)((expval < 0) ? -expval : expval), expval < 0, 10, 0, minwidth - 1, |
| 568 | FLAGS_ZEROPAD | FLAGS_PLUS); |
| 569 | // might need to right-pad spaces |
| 570 | if (flags & FLAGS_LEFT) { |
| 571 | while (idx - start_idx < width) out(' ', buffer, idx++, maxlen); |
| 572 | } |
| 573 | } |
| 574 | return idx; |
| 575 | } |
| 576 | |
| 577 | #endif // PICO_PRINTF_SUPPORT_EXPONENTIAL |
| 578 | #endif // PICO_PRINTF_SUPPORT_FLOAT |
| 579 | |
| 580 | // internal vsnprintf |
| 581 | static int _vsnprintf(out_fct_type out, char *buffer, const size_t maxlen, const char *format, va_list va) { |
| 582 | #if !PICO_PRINTF_ALWAYS_INCLUDED |
| 583 | lazy_vsnprintf = _vsnprintf; |
| 584 | #endif |
| 585 | unsigned int flags, width, precision, n; |
| 586 | size_t idx = 0U; |
| 587 | |
| 588 | if (!buffer) { |
| 589 | // use null output function |
| 590 | out = _out_null; |
| 591 | } |
| 592 | |
| 593 | while (*format) { |
| 594 | // format specifier? %[flags][width][.precision][length] |
| 595 | if (*format != '%') { |
| 596 | // no |
| 597 | out(*format, buffer, idx++, maxlen); |
| 598 | format++; |
| 599 | continue; |
| 600 | } else { |
| 601 | // yes, evaluate it |
| 602 | format++; |
| 603 | } |
| 604 | |
| 605 | // evaluate flags |
| 606 | flags = 0U; |
| 607 | do { |
| 608 | switch (*format) { |
| 609 | case '0': |
| 610 | flags |= FLAGS_ZEROPAD; |
| 611 | format++; |
| 612 | n = 1U; |
| 613 | break; |
| 614 | case '-': |
| 615 | flags |= FLAGS_LEFT; |
| 616 | format++; |
| 617 | n = 1U; |
| 618 | break; |
| 619 | case '+': |
| 620 | flags |= FLAGS_PLUS; |
| 621 | format++; |
| 622 | n = 1U; |
| 623 | break; |
| 624 | case ' ': |
| 625 | flags |= FLAGS_SPACE; |
| 626 | format++; |
| 627 | n = 1U; |
| 628 | break; |
| 629 | case '#': |
| 630 | flags |= FLAGS_HASH; |
| 631 | format++; |
| 632 | n = 1U; |
| 633 | break; |
| 634 | default : |
| 635 | n = 0U; |
| 636 | break; |
| 637 | } |
| 638 | } while (n); |
| 639 | |
| 640 | // evaluate width field |
| 641 | width = 0U; |
| 642 | if (_is_digit(*format)) { |
| 643 | width = _atoi(&format); |
| 644 | } else if (*format == '*') { |
| 645 | const int w = va_arg(va, int); |
| 646 | if (w < 0) { |
| 647 | flags |= FLAGS_LEFT; // reverse padding |
| 648 | width = (unsigned int) -w; |
| 649 | } else { |
| 650 | width = (unsigned int) w; |
| 651 | } |
| 652 | format++; |
| 653 | } |
| 654 | |
| 655 | // evaluate precision field |
| 656 | precision = 0U; |
| 657 | if (*format == '.') { |
| 658 | flags |= FLAGS_PRECISION; |
| 659 | format++; |
| 660 | if (_is_digit(*format)) { |
| 661 | precision = _atoi(&format); |
| 662 | } else if (*format == '*') { |
| 663 | const int prec = (int) va_arg(va, int); |
| 664 | precision = prec > 0 ? (unsigned int) prec : 0U; |
| 665 | format++; |
| 666 | } |
| 667 | } |
| 668 | |
| 669 | // evaluate length field |
| 670 | switch (*format) { |
| 671 | case 'l' : |
| 672 | flags |= FLAGS_LONG; |
| 673 | format++; |
| 674 | if (*format == 'l') { |
| 675 | flags |= FLAGS_LONG_LONG; |
| 676 | format++; |
| 677 | } |
| 678 | break; |
| 679 | case 'h' : |
| 680 | flags |= FLAGS_SHORT; |
| 681 | format++; |
| 682 | if (*format == 'h') { |
| 683 | flags |= FLAGS_CHAR; |
| 684 | format++; |
| 685 | } |
| 686 | break; |
| 687 | #if PICO_PRINTF_SUPPORT_PTRDIFF_T |
| 688 | case 't' : |
| 689 | flags |= (sizeof(ptrdiff_t) == sizeof(long) ? FLAGS_LONG : FLAGS_LONG_LONG); |
| 690 | format++; |
| 691 | break; |
| 692 | #endif |
| 693 | case 'j' : |
| 694 | flags |= (sizeof(intmax_t) == sizeof(long) ? FLAGS_LONG : FLAGS_LONG_LONG); |
| 695 | format++; |
| 696 | break; |
| 697 | case 'z' : |
| 698 | flags |= (sizeof(size_t) == sizeof(long) ? FLAGS_LONG : FLAGS_LONG_LONG); |
| 699 | format++; |
| 700 | break; |
| 701 | default : |
| 702 | break; |
| 703 | } |
| 704 | |
| 705 | // evaluate specifier |
| 706 | switch (*format) { |
| 707 | case 'd' : |
| 708 | case 'i' : |
| 709 | case 'u' : |
| 710 | case 'x' : |
| 711 | case 'X' : |
| 712 | case 'o' : |
| 713 | case 'b' : { |
| 714 | // set the base |
| 715 | unsigned int base; |
| 716 | if (*format == 'x' || *format == 'X') { |
| 717 | base = 16U; |
| 718 | } else if (*format == 'o') { |
| 719 | base = 8U; |
| 720 | } else if (*format == 'b') { |
| 721 | base = 2U; |
| 722 | } else { |
| 723 | base = 10U; |
| 724 | flags &= ~FLAGS_HASH; // no hash for dec format |
| 725 | } |
| 726 | // uppercase |
| 727 | if (*format == 'X') { |
| 728 | flags |= FLAGS_UPPERCASE; |
| 729 | } |
| 730 | |
| 731 | // no plus or space flag for u, x, X, o, b |
| 732 | if ((*format != 'i') && (*format != 'd')) { |
| 733 | flags &= ~(FLAGS_PLUS | FLAGS_SPACE); |
| 734 | } |
| 735 | |
| 736 | // ignore '0' flag when precision is given |
| 737 | if (flags & FLAGS_PRECISION) { |
| 738 | flags &= ~FLAGS_ZEROPAD; |
| 739 | } |
| 740 | |
| 741 | // convert the integer |
| 742 | if ((*format == 'i') || (*format == 'd')) { |
| 743 | // signed |
| 744 | if (flags & FLAGS_LONG_LONG) { |
| 745 | #if PICO_PRINTF_SUPPORT_LONG_LONG |
| 746 | const long long value = va_arg(va, long long); |
| 747 | idx = _ntoa_long_long(out, buffer, idx, maxlen, |
| 748 | (unsigned long long) (value > 0 ? value : 0 - value), value < 0, base, |
| 749 | precision, width, flags); |
| 750 | #endif |
| 751 | } else if (flags & FLAGS_LONG) { |
| 752 | const long value = va_arg(va, long); |
| 753 | idx = _ntoa_long(out, buffer, idx, maxlen, (unsigned long) (value > 0 ? value : 0 - value), |
| 754 | value < 0, base, precision, width, flags); |
| 755 | } else { |
| 756 | const int value = (flags & FLAGS_CHAR) ? (char) va_arg(va, int) : (flags & FLAGS_SHORT) |
| 757 | ? (short int) va_arg(va, int) |
| 758 | : va_arg(va, int); |
| 759 | idx = _ntoa_long(out, buffer, idx, maxlen, (unsigned int) (value > 0 ? value : 0 - value), |
| 760 | value < 0, base, precision, width, flags); |
| 761 | } |
| 762 | } else { |
| 763 | // unsigned |
| 764 | if (flags & FLAGS_LONG_LONG) { |
| 765 | #if PICO_PRINTF_SUPPORT_LONG_LONG |
| 766 | idx = _ntoa_long_long(out, buffer, idx, maxlen, va_arg(va, unsigned long long), false, base, |
| 767 | precision, width, flags); |
| 768 | #endif |
| 769 | } else if (flags & FLAGS_LONG) { |
| 770 | idx = _ntoa_long(out, buffer, idx, maxlen, va_arg(va, unsigned long), false, base, precision, |
| 771 | width, flags); |
| 772 | } else { |
| 773 | const unsigned int value = (flags & FLAGS_CHAR) ? (unsigned char) va_arg(va, unsigned int) |
| 774 | : (flags & FLAGS_SHORT) |
| 775 | ? (unsigned short int) va_arg(va, |
| 776 | unsigned int) |
| 777 | : va_arg(va, unsigned int); |
| 778 | idx = _ntoa_long(out, buffer, idx, maxlen, value, false, base, precision, width, flags); |
| 779 | } |
| 780 | } |
| 781 | format++; |
| 782 | break; |
| 783 | } |
| 784 | case 'f' : |
| 785 | case 'F' : |
| 786 | #if PICO_PRINTF_SUPPORT_FLOAT |
| 787 | if (*format == 'F') flags |= FLAGS_UPPERCASE; |
| 788 | idx = _ftoa(out, buffer, idx, maxlen, va_arg(va, double), precision, width, flags); |
| 789 | #else |
| 790 | for(int i=0;i<2;i++) out('?', buffer, idx++, maxlen); |
| 791 | va_arg(va, double); |
| 792 | #endif |
| 793 | format++; |
| 794 | break; |
| 795 | case 'e': |
| 796 | case 'E': |
| 797 | case 'g': |
| 798 | case 'G': |
| 799 | #if PICO_PRINTF_SUPPORT_FLOAT && PICO_PRINTF_SUPPORT_EXPONENTIAL |
| 800 | if ((*format == 'g') || (*format == 'G')) flags |= FLAGS_ADAPT_EXP; |
| 801 | if ((*format == 'E') || (*format == 'G')) flags |= FLAGS_UPPERCASE; |
| 802 | idx = _etoa(out, buffer, idx, maxlen, va_arg(va, double), precision, width, flags); |
| 803 | #else |
| 804 | for(int i=0;i<2;i++) out('?', buffer, idx++, maxlen); |
| 805 | va_arg(va, double); |
| 806 | #endif |
| 807 | format++; |
| 808 | break; |
| 809 | case 'c' : { |
| 810 | unsigned int l = 1U; |
| 811 | // pre padding |
| 812 | if (!(flags & FLAGS_LEFT)) { |
| 813 | while (l++ < width) { |
| 814 | out(' ', buffer, idx++, maxlen); |
| 815 | } |
| 816 | } |
| 817 | // char output |
| 818 | out((char) va_arg(va, int), buffer, idx++, maxlen); |
| 819 | // post padding |
| 820 | if (flags & FLAGS_LEFT) { |
| 821 | while (l++ < width) { |
| 822 | out(' ', buffer, idx++, maxlen); |
| 823 | } |
| 824 | } |
| 825 | format++; |
| 826 | break; |
| 827 | } |
| 828 | |
| 829 | case 's' : { |
| 830 | const char *p = va_arg(va, char*); |
| 831 | unsigned int l = _strnlen_s(p, precision ? precision : (size_t) -1); |
| 832 | // pre padding |
| 833 | if (flags & FLAGS_PRECISION) { |
| 834 | l = (l < precision ? l : precision); |
| 835 | } |
| 836 | if (!(flags & FLAGS_LEFT)) { |
| 837 | while (l++ < width) { |
| 838 | out(' ', buffer, idx++, maxlen); |
| 839 | } |
| 840 | } |
| 841 | // string output |
| 842 | while ((*p != 0) && (!(flags & FLAGS_PRECISION) || precision--)) { |
| 843 | out(*(p++), buffer, idx++, maxlen); |
| 844 | } |
| 845 | // post padding |
| 846 | if (flags & FLAGS_LEFT) { |
| 847 | while (l++ < width) { |
| 848 | out(' ', buffer, idx++, maxlen); |
| 849 | } |
| 850 | } |
| 851 | format++; |
| 852 | break; |
| 853 | } |
| 854 | |
| 855 | case 'p' : { |
| 856 | width = sizeof(void *) * 2U; |
| 857 | flags |= FLAGS_ZEROPAD | FLAGS_UPPERCASE; |
| 858 | #if PICO_PRINTF_SUPPORT_LONG_LONG |
| 859 | const bool is_ll = sizeof(uintptr_t) == sizeof(long long); |
| 860 | if (is_ll) { |
| 861 | idx = _ntoa_long_long(out, buffer, idx, maxlen, (uintptr_t) va_arg(va, void*), false, 16U, |
| 862 | precision, width, flags); |
| 863 | } else { |
| 864 | #endif |
| 865 | idx = _ntoa_long(out, buffer, idx, maxlen, (unsigned long) ((uintptr_t) va_arg(va, void*)), false, |
| 866 | 16U, precision, width, flags); |
| 867 | #if PICO_PRINTF_SUPPORT_LONG_LONG |
| 868 | } |
| 869 | #endif |
| 870 | format++; |
| 871 | break; |
| 872 | } |
| 873 | |
| 874 | case '%' : |
| 875 | out('%', buffer, idx++, maxlen); |
| 876 | format++; |
| 877 | break; |
| 878 | |
| 879 | default : |
| 880 | out(*format, buffer, idx++, maxlen); |
| 881 | format++; |
| 882 | break; |
| 883 | } |
| 884 | } |
| 885 | |
| 886 | // termination |
| 887 | out((char) 0, buffer, idx < maxlen ? idx : maxlen - 1U, maxlen); |
| 888 | |
| 889 | // return written chars without terminating \0 |
| 890 | return (int) idx; |
| 891 | } |
| 892 | |
| 893 | |
| 894 | /////////////////////////////////////////////////////////////////////////////// |
| 895 | |
| 896 | int WRAPPER_FUNC(sprintf)(char *buffer, const char *format, ...) { |
| 897 | va_list va; |
| 898 | va_start(va, format); |
| 899 | const int ret = _vsnprintf(_out_buffer, buffer, (size_t) -1, format, va); |
| 900 | va_end(va); |
| 901 | return ret; |
| 902 | } |
| 903 | |
| 904 | int WRAPPER_FUNC(snprintf)(char *buffer, size_t count, const char *format, ...) { |
| 905 | va_list va; |
| 906 | va_start(va, format); |
| 907 | const int ret = _vsnprintf(_out_buffer, buffer, count, format, va); |
| 908 | va_end(va); |
| 909 | return ret; |
| 910 | } |
| 911 | |
| 912 | int WRAPPER_FUNC(vsnprintf)(char *buffer, size_t count, const char *format, va_list va) { |
| 913 | return _vsnprintf(_out_buffer, buffer, count, format, va); |
| 914 | } |
| 915 | |
| 916 | int vfctprintf(void (*out)(char character, void *arg), void *arg, const char *format, va_list va) { |
| 917 | const out_fct_wrap_type out_fct_wrap = {out, arg}; |
| 918 | return _vsnprintf(_out_fct, (char *) (uintptr_t) &out_fct_wrap, (size_t) -1, format, va); |
| 919 | } |
| 920 | |
| 921 | #if LIB_PICO_PRINTF_PICO |
| 922 | #if !PICO_PRINTF_ALWAYS_INCLUDED |
| 923 | bool weak_raw_printf(const char *fmt, ...) { |
| 924 | va_list va; |
| 925 | va_start(va, fmt); |
| 926 | bool rc = weak_raw_vprintf(fmt, va); |
| 927 | va_end(va); |
| 928 | return rc; |
| 929 | } |
| 930 | |
| 931 | bool weak_raw_vprintf(const char *fmt, va_list args) { |
| 932 | if (lazy_vsnprintf) { |
| 933 | char buffer[1]; |
| 934 | lazy_vsnprintf(_out_char, buffer, (size_t) -1, fmt, args); |
| 935 | return true; |
| 936 | } else { |
| 937 | puts(fmt); |
| 938 | return false; |
| 939 | } |
| 940 | } |
| 941 | #endif |
| 942 | #endif |