Properly print MP_OBJ_QSTR objects.

This commit is contained in:
Paul Sokolovsky 2014-01-20 00:59:25 +02:00
parent 2c7a1b2ad1
commit bb33cc66fb
3 changed files with 11 additions and 4 deletions

View File

@ -44,6 +44,8 @@ void printf_wrapper(void *env, const char *fmt, ...) {
void mp_obj_print_helper(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t o_in, mp_print_kind_t kind) {
if (MP_OBJ_IS_SMALL_INT(o_in)) {
print(env, "%d", (int)MP_OBJ_SMALL_INT_VALUE(o_in));
} else if (MP_OBJ_IS_QSTR(o_in)) {
mp_obj_str_print_qstr(print, env, MP_OBJ_QSTR_VALUE(o_in), kind);
} else {
mp_obj_base_t *o = o_in;
if (o->type->print != NULL) {

View File

@ -280,6 +280,7 @@ void mp_obj_exception_get_traceback(mp_obj_t self_in, machine_uint_t *n, machine
// str
extern const mp_obj_type_t str_type;
qstr mp_obj_str_get(mp_obj_t self_in);
void mp_obj_str_print_qstr(void (*print)(void *env, const char *fmt, ...), void *env, qstr q, mp_print_kind_t kind);
#if MICROPY_ENABLE_FLOAT
// float

View File

@ -22,16 +22,20 @@ static mp_obj_t mp_obj_new_str_iterator(mp_obj_str_t *str, int cur);
/******************************************************************************/
/* str */
void str_print(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t self_in, mp_print_kind_t kind) {
mp_obj_str_t *self = self_in;
void mp_obj_str_print_qstr(void (*print)(void *env, const char *fmt, ...), void *env, qstr q, mp_print_kind_t kind) {
if (kind == PRINT_STR) {
print(env, "%s", qstr_str(self->qstr));
print(env, "%s", qstr_str(q));
} else {
// TODO need to escape chars etc
print(env, "'%s'", qstr_str(self->qstr));
print(env, "'%s'", qstr_str(q));
}
}
void str_print(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t self_in, mp_print_kind_t kind) {
mp_obj_str_t *self = self_in;
mp_obj_str_print_qstr(print, env, self->qstr, kind);
}
mp_obj_t str_binary_op(int op, mp_obj_t lhs_in, mp_obj_t rhs_in) {
mp_obj_str_t *lhs = lhs_in;
const char *lhs_str = qstr_str(lhs->qstr);