/*
 * This file is part of the MicroPython project, http://micropython.org/
 *
 * The MIT License (MIT)
 *
 * Copyright (c) 2013-2015 Damien P. George
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

#include <assert.h>
#include <stdarg.h>
#include <stdint.h>
#include <stdio.h>
#include <string.h>

#include "py/mphal.h"
#include "py/mpprint.h"
#include "py/obj.h"
#include "py/objint.h"
#include "py/runtime.h"

#if MICROPY_PY_BUILTINS_FLOAT
#include "py/formatfloat.h"
#endif

static const char pad_spaces[] = " ";
static const char pad_zeroes[] = "0000000000000000";

STATIC void plat_print_strn(void *env, const char *str, size_t len) {
 (void)env;
 MP_PLAT_PRINT_STRN(str, len);
}

const mp_print_t mp_plat_print = {NULL, plat_print_strn};

int mp_print_str(const mp_print_t *print, const char *str) {
 size_t len = strlen(str);
 if (len) {
 print->print_strn(print->data, str, len);
 }
 return len;
}

int mp_print_strn(const mp_print_t *print, const char *str, size_t len, int flags, char fill, int width) {
 int left_pad = 0;
 int right_pad = 0;
 int pad = width - len;
 int pad_size;
 int total_chars_printed = 0;
 const char *pad_chars;

 if (!fill || fill == ' ') {
 pad_chars = pad_spaces;
 pad_size = sizeof(pad_spaces) - 1;
 } else if (fill == '0') {
 pad_chars = pad_zeroes;
 pad_size = sizeof(pad_zeroes) - 1;
 } else {
 // Other pad characters are fairly unusual, so we'll take the hit
 // and output them 1 at a time.
 pad_chars = &fill;
 pad_size = 1;
 }

 if (flags & PF_FLAG_CENTER_ADJUST) {
 left_pad = pad / 2;
 right_pad = pad - left_pad;
 } else if (flags & PF_FLAG_LEFT_ADJUST) {
 right_pad = pad;
 } else {
 left_pad = pad;
 }

 if (left_pad > 0) {
 total_chars_printed += left_pad;
 while (left_pad > 0) {
 int p = left_pad;
 if (p > pad_size) {
 p = pad_size;
 }
 print->print_strn(print->data, pad_chars, p);
 left_pad -= p;
 }
 }
 if (len) {
 print->print_strn(print->data, str, len);
 total_chars_printed += len;
 }
 if (right_pad > 0) {
 total_chars_printed += right_pad;
 while (right_pad > 0) {
 int p = right_pad;
 if (p > pad_size) {
 p = pad_size;
 }
 print->print_strn(print->data, pad_chars, p);
 right_pad -= p;
 }
 }
 return total_chars_printed;
}

// 32-bits is 10 digits, add 3 for commas, 1 for sign, 1 for terminating null
// We can use 16 characters for 32-bit and 32 characters for 64-bit
#define INT_BUF_SIZE (sizeof(mp_int_t) * 4)

// Our mp_vprintf function below does not support the '#' format modifier to
// print the prefix of a non-base-10 number, so we don't need code for this.
#define SUPPORT_INT_BASE_PREFIX (0)

// This function is used exclusively by mp_vprintf to format ints.
// It needs to be a separate function to mp_print_mp_int, since converting to a mp_int looses the MSB.
STATIC int mp_print_int(const mp_print_t *print, mp_uint_t x, int sgn, int base, int base_char, int flags, char fill, int width) {
 char sign = 0;
 if (sgn) {
 if ((mp_int_t)x < 0) {
 sign = '-';
 x = -x;
 } else if (flags & PF_FLAG_SHOW_SIGN) {
 sign = '+';
 } else if (flags & PF_FLAG_SPACE_SIGN) {
 sign = ' ';
 }
 }

 char buf[INT_BUF_SIZE];
 char *b = buf + INT_BUF_SIZE;

 if (x == 0) {
 *(--b) = '0';
 } else {
 do {
 int c = x % base;
 x /= base;
 if (c >= 10) {
 c += base_char - 10;
 } else {
 c += '0';
 }
 *(--b) = c;
 } while (b > buf && x != 0);
 }

 #if SUPPORT_INT_BASE_PREFIX
 char prefix_char = '0円';

 if (flags & PF_FLAG_SHOW_PREFIX) {
 if (base == 2) {
 prefix_char = base_char + 'b' - 'a';
 } else if (base == 8) {
 prefix_char = base_char + 'o' - 'a';
 } else if (base == 16) {
 prefix_char = base_char + 'x' - 'a';
 }
 }
 #endif

 int len = 0;
 if (flags & PF_FLAG_PAD_AFTER_SIGN) {
 if (sign) {
 len += mp_print_strn(print, &sign, 1, flags, fill, 1);
 width--;
 }
 #if SUPPORT_INT_BASE_PREFIX
 if (prefix_char) {
 len += mp_print_strn(print, "0", 1, flags, fill, 1);
 len += mp_print_strn(print, &prefix_char, 1, flags, fill, 1);
 width -= 2;
 }
 #endif
 } else {
 #if SUPPORT_INT_BASE_PREFIX
 if (prefix_char && b > &buf[1]) {
 *(--b) = prefix_char;
 *(--b) = '0';
 }
 #endif
 if (sign && b > buf) {
 *(--b) = sign;
 }
 }

 len += mp_print_strn(print, b, buf + INT_BUF_SIZE - b, flags, fill, width);
 return len;
}

int mp_print_mp_int(const mp_print_t *print, mp_obj_t x, int base, int base_char, int flags, char fill, int width, int prec) {
 // These are the only values for "base" that are required to be supported by this
 // function, since Python only allows the user to format integers in these bases.
 // If needed this function could be generalised to handle other values.
 assert(base == 2 || base == 8 || base == 10 || base == 16);

 if (!mp_obj_is_int(x)) {
 // This will convert booleans to int, or raise an error for
 // non-integer types.
 x = MP_OBJ_NEW_SMALL_INT(mp_obj_get_int(x));
 }

 if ((flags & (PF_FLAG_LEFT_ADJUST | PF_FLAG_CENTER_ADJUST)) == 0 && fill == '0') {
 if (prec > width) {
 width = prec;
 }
 prec = 0;
 }
 char prefix_buf[4];
 char *prefix = prefix_buf;

 if (mp_obj_int_sign(x) >= 0) {
 if (flags & PF_FLAG_SHOW_SIGN) {
 *prefix++ = '+';
 } else if (flags & PF_FLAG_SPACE_SIGN) {
 *prefix++ = ' ';
 }
 }

 if (flags & PF_FLAG_SHOW_PREFIX) {
 if (base == 2) {
 *prefix++ = '0';
 *prefix++ = base_char + 'b' - 'a';
 } else if (base == 8) {
 *prefix++ = '0';
 if (flags & PF_FLAG_SHOW_OCTAL_LETTER) {
 *prefix++ = base_char + 'o' - 'a';
 }
 } else if (base == 16) {
 *prefix++ = '0';
 *prefix++ = base_char + 'x' - 'a';
 }
 }
 *prefix = '0円';
 int prefix_len = prefix - prefix_buf;
 prefix = prefix_buf;

 char comma = '0円';
 if (flags & PF_FLAG_SHOW_COMMA) {
 comma = ',';
 }

 // The size of this buffer is rather arbitrary. If it's not large
 // enough, a dynamic one will be allocated.
 char stack_buf[sizeof(mp_int_t) * 4];
 char *buf = stack_buf;
 size_t buf_size = sizeof(stack_buf);
 size_t fmt_size = 0;
 char *str;

 if (prec > 1) {
 flags |= PF_FLAG_PAD_AFTER_SIGN;
 }
 char sign = '0円';
 if (flags & PF_FLAG_PAD_AFTER_SIGN) {
 // We add the pad in this function, so since the pad goes after
 // the sign & prefix, we format without a prefix
 str = mp_obj_int_formatted(&buf, &buf_size, &fmt_size,
 x, base, NULL, base_char, comma);
 if (*str == '-') {
 sign = *str++;
 fmt_size--;
 }
 } else {
 str = mp_obj_int_formatted(&buf, &buf_size, &fmt_size,
 x, base, prefix, base_char, comma);
 }

 int spaces_before = 0;
 int spaces_after = 0;

 if (prec > 1) {
 // If prec was specified, then prec specifies the width to zero-pad the
 // the number to. This zero-padded number then gets left or right
 // aligned in width characters.

 int prec_width = fmt_size; // The digits
 if (prec_width < prec) {
 prec_width = prec;
 }
 if (flags & PF_FLAG_PAD_AFTER_SIGN) {
 if (sign) {
 prec_width++;
 }
 prec_width += prefix_len;
 }
 if (prec_width < width) {
 if (flags & PF_FLAG_LEFT_ADJUST) {
 spaces_after = width - prec_width;
 } else {
 spaces_before = width - prec_width;
 }
 }
 fill = '0';
 flags &= ~PF_FLAG_LEFT_ADJUST;
 }

 int len = 0;
 if (spaces_before) {
 len += mp_print_strn(print, "", 0, 0, ' ', spaces_before);
 }
 if (flags & PF_FLAG_PAD_AFTER_SIGN) {
 // pad after sign implies pad after prefix as well.
 if (sign) {
 len += mp_print_strn(print, &sign, 1, 0, 0, 1);
 width--;
 }
 if (prefix_len) {
 len += mp_print_strn(print, prefix, prefix_len, 0, 0, 1);
 width -= prefix_len;
 }
 }
 if (prec > 1) {
 width = prec;
 }

 len += mp_print_strn(print, str, fmt_size, flags, fill, width);

 if (spaces_after) {
 len += mp_print_strn(print, "", 0, 0, ' ', spaces_after);
 }

 if (buf != stack_buf) {
 m_del(char, buf, buf_size);
 }
 return len;
}

#if MICROPY_PY_BUILTINS_FLOAT
int mp_print_float(const mp_print_t *print, mp_float_t f, char fmt, int flags, char fill, int width, int prec) {
 char buf[32];
 char sign = '0円';
 int chrs = 0;

 if (flags & PF_FLAG_SHOW_SIGN) {
 sign = '+';
 } else
 if (flags & PF_FLAG_SPACE_SIGN) {
 sign = ' ';
 }

 int len = mp_format_float(f, buf, sizeof(buf), fmt, prec, sign);

 char *s = buf;

 if ((flags & PF_FLAG_ADD_PERCENT) && (size_t)(len + 1) < sizeof(buf)) {
 buf[len++] = '%';
 buf[len] = '0円';
 }

 // buf[0] < '0' returns true if the first character is space, + or -
 if ((flags & PF_FLAG_PAD_AFTER_SIGN) && buf[0] < '0') {
 // We have a sign character
 s++;
 chrs += mp_print_strn(print, &buf[0], 1, 0, 0, 1);
 width--;
 len--;
 }

 chrs += mp_print_strn(print, s, len, flags, fill, width);

 return chrs;
}
#endif

int mp_printf(const mp_print_t *print, const char *fmt, ...) {
 va_list ap;
 va_start(ap, fmt);
 int ret = mp_vprintf(print, fmt, ap);
 va_end(ap);
 return ret;
}

int mp_vprintf(const mp_print_t *print, const char *fmt, va_list args) {
 int chrs = 0;
 for (;;) {
 {
 const char *f = fmt;
 while (*f != '0円' && *f != '%') {
 ++f; // XXX UTF8 advance char
 }
 if (f > fmt) {
 print->print_strn(print->data, fmt, f - fmt);
 chrs += f - fmt;
 fmt = f;
 }
 }

 if (*fmt == '0円') {
 break;
 }

 // move past % character
 ++fmt;

 // parse flags, if they exist
 int flags = 0;
 char fill = ' ';
 while (*fmt != '0円') {
 if (*fmt == '-') {
 flags |= PF_FLAG_LEFT_ADJUST;
 } else if (*fmt == '+') {
 flags |= PF_FLAG_SHOW_SIGN;
 } else if (*fmt == ' ') {
 flags |= PF_FLAG_SPACE_SIGN;
 } else if (*fmt == '!') {
 flags |= PF_FLAG_NO_TRAILZ;
 } else if (*fmt == '0') {
 flags |= PF_FLAG_PAD_AFTER_SIGN;
 fill = '0';
 } else {
 break;
 }
 ++fmt;
 }

 // parse width, if it exists
 int width = 0;
 for (; '0' <= *fmt && *fmt <= '9'; ++fmt) {
 width = width * 10 + *fmt - '0';
 }

 // parse precision, if it exists
 int prec = -1;
 if (*fmt == '.') {
 ++fmt;
 if (*fmt == '*') {
 ++fmt;
 prec = va_arg(args, int);
 } else {
 prec = 0;
 for (; '0' <= *fmt && *fmt <= '9'; ++fmt) {
 prec = prec * 10 + *fmt - '0';
 }
 }
 if (prec < 0) {
 prec = 0;
 }
 }

 // parse long specifiers (only for LP64 model where they make a difference)
 #ifndef __LP64__
 const
 #endif
 bool long_arg = false;
 if (*fmt == 'l') {
 ++fmt;
 #ifdef __LP64__
 long_arg = true;
 #endif
 }

 if (*fmt == '0円') {
 break;
 }

 switch (*fmt) {
 case 'b':
 if (va_arg(args, int)) {
 chrs += mp_print_strn(print, "true", 4, flags, fill, width);
 } else {
 chrs += mp_print_strn(print, "false", 5, flags, fill, width);
 }
 break;
 case 'c': {
 char str = va_arg(args, int);
 chrs += mp_print_strn(print, &str, 1, flags, fill, width);
 break;
 }
 case 'q': {
 qstr qst = va_arg(args, qstr);
 size_t len;
 const char *str = (const char *)qstr_data(qst, &len);
 if (prec < 0) {
 prec = len;
 }
 chrs += mp_print_strn(print, str, prec, flags, fill, width);
 break;
 }
 case 's': {
 const char *str = va_arg(args, const char *);
 #ifndef NDEBUG
 // With debugging enabled, catch printing of null string pointers
 if (prec != 0 && str == NULL) {
 chrs += mp_print_strn(print, "(null)", 6, flags, fill, width);
 break;
 }
 #endif
 if (prec < 0) {
 prec = strlen(str);
 }
 chrs += mp_print_strn(print, str, prec, 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 '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
 mp_uint_t val;
 if (long_arg) {
 val = va_arg(args, unsigned long int);
 } else {
 val = va_arg(args, unsigned int);
 }
 chrs += mp_print_int(print, val, 0, 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':
 case 'f':
 case 'F':
 case 'g':
 case 'G': {
 #if ((MICROPY_FLOAT_IMPL == MICROPY_FLOAT_IMPL_FLOAT) || (MICROPY_FLOAT_IMPL == MICROPY_FLOAT_IMPL_DOUBLE))
 mp_float_t f = (mp_float_t)va_arg(args, double);
 chrs += mp_print_float(print, f, *fmt, flags, fill, width, prec);
 #else
 #error Unknown MICROPY FLOAT IMPL
 #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;
 if (*fmt == 'u' || *fmt == 'd') {
 chrs += mp_print_int(print, arg_value, *fmt == 'd', 10, 'a', flags, fill, width);
 break;
 }
 assert(!"unsupported fmt char");
 }
 #endif
 default:
 // if it's not %% then it's an unsupported format character
 assert(*fmt == '%' || !"unsupported fmt char");
 print->print_strn(print->data, fmt, 1);
 chrs += 1;
 break;
 }
 ++fmt;
 }
 return chrs;
}