10000 py/stream: Move uPy func obj wrappers to below their respective funcs. · lapsule/micropython@358e5d8 · GitHub
[go: up one dir, main page]

Skip to content

Commit 358e5d8

Browse files
committed
py/stream: Move uPy func obj wrappers to below their respective funcs.
1 parent 657aef6 commit 358e5d8

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

py/stream.c

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -182,6 +182,7 @@ STATIC mp_obj_t stream_read(size_t n_args, const mp_obj_t *args) {
182182
return mp_obj_new_str_from_vstr(STREAM_CONTENT_TYPE(stream_p), &vstr);
183183
}
184184
}
185+
MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mp_stream_read_obj, 1, 2, stream_read);
185186

186187
mp_obj_t mp_stream_write(mp_obj_t self_in, const void *buf, size_t len) {
187188
const mp_stream_p_t *stream_p = mp_get_stream_raise(self_in, MP_STREAM_OP_WRITE);
@@ -228,6 +229,7 @@ STATIC mp_obj_t stream_write_method(mp_obj_t self_in, mp_obj_t arg) {
228229
mp_get_buffer_raise(arg, &bufinfo, MP_BUFFER_READ);
229230
return mp_stream_write(self_in, bufinfo.buf, bufinfo.len);
230231
}
232+
MP_DEFINE_CONST_FUN_OBJ_2(mp_stream_write_obj, stream_write_method);
231233

232234
STATIC mp_obj_t stream_readinto(size_t n_args, const mp_obj_t *args) {
233235
const mp_stream_p_t *stream_p = mp_get_stream_raise(args[0], MP_STREAM_OP_READ);
@@ -256,6 +258,7 @@ STATIC mp_obj_t stream_readinto(size_t n_args, const mp_obj_t *args) {
256258
return MP_OBJ_NEW_SMALL_INT(out_sz);
257259
}
258260
}
261+
MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mp_stream_readinto_obj, 2, 3, stream_readinto);
259262

260263
STATIC mp_obj_t stream_readall(mp_obj_t self_in) {
261264
const mp_stream_p_t *stream_p = mp_get_stream_raise(self_in, MP_STREAM_OP_READ);
@@ -300,6 +303,7 @@ STATIC mp_obj_t stream_readall(mp_obj_t self_in) {
300303
vstr.len = total_size;
301304
return mp_obj_new_str_from_vstr(STREAM_CONTENT_TYPE(stream_p), &vstr);
302305
}
306+
MP_DEFINE_CONST_FUN_OBJ_1(mp_stream_readall_obj, stream_readall);
303307

304308
// Unbuffered, inefficient implementation of readline() for raw I/O files.
305309
STATIC mp_obj_t stream_unbuffered_readline(size_t n_args, const mp_obj_t *args) {
@@ -357,6 +361,7 @@ STATIC mp_obj_t stream_unbuffered_readline(size_t n_args, const mp_obj_t *args)
357361

358362
return mp_obj_new_str_from_vstr(STREAM_CONTENT_TYPE(stream_p), &vstr);
359363
}
364+
MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mp_stream_unbuffered_readline_obj, 1, 2, stream_unbuffered_readline);
360365

361366
// TODO take an optional extra argument (what does it do exactly?)
362367
STATIC mp_obj_t stream_unbuffered_readlines(mp_obj_t self) {
@@ -432,9 +437,3 @@ STATIC mp_obj_t stream_ioctl(size_t n_args, const mp_obj_t *args) {
432437
return mp_obj_new_int(res);
433438
}
434439
MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mp_stream_ioctl_obj, 2, 3, stream_ioctl);
435-
436-
MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mp_stream_read_obj, 1, 2, stream_read);
437-
MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mp_stream_readinto_obj, 2, 3, stream_readinto);
438-
MP_DEFINE_CONST_FUN_OBJ_1(mp_stream_readall_obj, stream_readall);
439-
MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mp_stream_unbuffered_readline_obj, 1, 2, stream_unbuffered_readline);
440-
MP_DEFINE_CONST_FUN_OBJ_2(mp_stream_write_obj, stream_write_method);

0 commit comments

Comments
 (0)
0