djsundog/circuitpython
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

py/stream: Simplify arg extraction logic for stream_ioctl.

Browse Source 
Saves 16 bytes of code.

Also, use mp_obj_get_int_truncated to allow integers as big as a machine
word to be passed as the value.
This commit is contained in:
Damien George 2016-04-10 12:37:59 +01:00
parent 6e87aeb841
commit 657aef66ff
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
py/stream.c
View File

@ -416,11 +416,10 @@ STATIC mp_obj_t stream_ioctl(size_t n_args, const mp_obj_t *args) {
mp_buffer_info_t bufinfo; mp_buffer_info_t bufinfo;
uintptr_t val = 0; uintptr_t val = 0;
if (n_args > 2) { if (n_args > 2) {
if (MP_OBJ_IS_INT(args[2])) { if (mp_get_buffer(args[2], &bufinfo, MP_BUFFER_WRITE)) {
val = mp_obj_get_int(args[2]);
} else {
mp_get_buffer_raise(args[2], &bufinfo, MP_BUFFER_WRITE);
val = (uintptr_t)bufinfo.buf; val = (uintptr_t)bufinfo.buf;
} else {
val = mp_obj_get_int_truncated(args[2]);
} }
} }