summaryrefslogtreecommitdiff
path: root/examples/natmod/zlib
diff options
context:
space:
mode:
Diffstat (limited to 'examples/natmod/zlib')
-rw-r--r--examples/natmod/zlib/Makefile13
-rw-r--r--examples/natmod/zlib/zlib.c35
2 files changed, 48 insertions, 0 deletions
diff --git a/examples/natmod/zlib/Makefile b/examples/natmod/zlib/Makefile
new file mode 100644
index 000000000..e4b4e0712
--- /dev/null
+++ b/examples/natmod/zlib/Makefile
@@ -0,0 +1,13 @@
+# Location of top-level MicroPython directory
+MPY_DIR = ../../..
+
+# Name of module (different to built-in zlib so it can coexist)
+MOD = zlib_$(ARCH)
+
+# Source files (.c or .py)
+SRC = zlib.c
+
+# Architecture to build for (x86, x64, armv7m, xtensa, xtensawin)
+ARCH = x64
+
+include $(MPY_DIR)/py/dynruntime.mk
diff --git a/examples/natmod/zlib/zlib.c b/examples/natmod/zlib/zlib.c
new file mode 100644
index 000000000..2b0dbd5ca
--- /dev/null
+++ b/examples/natmod/zlib/zlib.c
@@ -0,0 +1,35 @@
+#define MICROPY_PY_ZLIB (1)
+
+#include "py/dynruntime.h"
+
+#if !defined(__linux__)
+void *memset(void *s, int c, size_t n) {
+ return mp_fun_table.memset_(s, c, n);
+}
+#endif
+
+mp_obj_full_type_t decompio_type;
+
+#include "extmod/modzlib.c"
+
+mp_map_elem_t decompio_locals_dict_table[3];
+STATIC MP_DEFINE_CONST_DICT(decompio_locals_dict, decompio_locals_dict_table);
+
+mp_obj_t mpy_init(mp_obj_fun_bc_t *self, size_t n_args, size_t n_kw, mp_obj_t *args) {
+ MP_DYNRUNTIME_INIT_ENTRY
+
+ decompio_type.base.type = mp_fun_table.type_type;
+ decompio_type.name = MP_QSTR_DecompIO;
+ MP_OBJ_TYPE_SET_SLOT(&decompio_type, make_new, &decompio_make_new, 0);
+ MP_OBJ_TYPE_SET_SLOT(&decompio_type, protocol, &decompio_stream_p, 1);
+ decompio_locals_dict_table[0] = (mp_map_elem_t){ MP_OBJ_NEW_QSTR(MP_QSTR_read), MP_OBJ_FROM_PTR(&mp_stream_read_obj) };
+ decompio_locals_dict_table[1] = (mp_map_elem_t){ MP_OBJ_NEW_QSTR(MP_QSTR_readinto), MP_OBJ_FROM_PTR(&mp_stream_readinto_obj) };
+ decompio_locals_dict_table[2] = (mp_map_elem_t){ MP_OBJ_NEW_QSTR(MP_QSTR_readline), MP_OBJ_FROM_PTR(&mp_stream_unbuffered_readline_obj) };
+ MP_OBJ_TYPE_SET_SLOT(&decompio_type, locals_dict, (void*)&decompio_locals_dict, 2);
+
+ mp_store_global(MP_QSTR___name__, MP_OBJ_NEW_QSTR(MP_QSTR_zlib));
+ mp_store_global(MP_QSTR_decompress, MP_OBJ_FROM_PTR(&mod_zlib_decompress_obj));
+ mp_store_global(MP_QSTR_DecompIO, MP_OBJ_FROM_PTR(&decompio_type));
+
+ MP_DYNRUNTIME_INIT_EXIT
+}