From 107c9fb235150d4f18309fb3ca61411e045ae669 Mon Sep 17 00:00:00 2001
From: Damien George <damien.p.george@gmail.com>
Date: Sat, 26 Apr 2014 10:31:15 +0100
Subject: [PATCH] py: Make collections module configurable, enabled by default.

---
 py/builtintables.c  | 2 ++
 py/modcollections.c | 4 ++++
 py/mpconfig.h       | 5 +++++
 py/objnamedtuple.c  | 4 ++++
 4 files changed, 15 insertions(+)

diff --git a/py/builtintables.c b/py/builtintables.c
index 940647d25f..e6461ad80b 100644
--- a/py/builtintables.c
+++ b/py/builtintables.c
@@ -132,7 +132,9 @@ STATIC const mp_map_elem_t mp_builtin_module_table[] = {
 #if MICROPY_ENABLE_MOD_IO
     { MP_OBJ_NEW_QSTR(MP_QSTR_io), (mp_obj_t)&mp_module_io },
 #endif
+#if MICROPY_ENABLE_MOD_COLLECTIONS
     { MP_OBJ_NEW_QSTR(MP_QSTR__collections), (mp_obj_t)&mp_module_collections },
+#endif
 #if MICROPY_ENABLE_MOD_STRUCT
     { MP_OBJ_NEW_QSTR(MP_QSTR_struct), (mp_obj_t)&mp_module_struct },
 #endif
diff --git a/py/modcollections.c b/py/modcollections.c
index 11845a00ce..5278a526fb 100644
--- a/py/modcollections.c
+++ b/py/modcollections.c
@@ -4,6 +4,8 @@
 #include "obj.h"
 #include "builtin.h"
 
+#if MICROPY_ENABLE_MOD_COLLECTIONS
+
 STATIC const mp_map_elem_t mp_module_collections_globals_table[] = {
     { MP_OBJ_NEW_QSTR(MP_QSTR___name__), MP_OBJ_NEW_QSTR(MP_QSTR__collections) },
     { MP_OBJ_NEW_QSTR(MP_QSTR_namedtuple), (mp_obj_t)&mp_namedtuple_obj },
@@ -25,3 +27,5 @@ const mp_obj_module_t mp_module_collections = {
     .name = MP_QSTR__collections,
     .globals = (mp_obj_dict_t*)&mp_module_collections_globals,
 };
+
+#endif // MICROPY_ENABLE_MOD_COLLECTIONS
diff --git a/py/mpconfig.h b/py/mpconfig.h
index b669128ea0..13918e8398 100644
--- a/py/mpconfig.h
+++ b/py/mpconfig.h
@@ -115,6 +115,11 @@ typedef double mp_float_t;
 #define MICROPY_ENABLE_FLOAT (0)
 #endif
 
+// Whether to provide "collections" module
+#ifndef MICROPY_ENABLE_MOD_COLLECTIONS
+#define MICROPY_ENABLE_MOD_COLLECTIONS (1)
+#endif
+
 // Whether to provide "math" module
 #ifndef MICROPY_ENABLE_MOD_MATH
 #define MICROPY_ENABLE_MOD_MATH (1)
diff --git a/py/objnamedtuple.c b/py/objnamedtuple.c
index 10cf40c9ce..4239983589 100644
--- a/py/objnamedtuple.c
+++ b/py/objnamedtuple.c
@@ -7,6 +7,8 @@
 #include "obj.h"
 #include "objtuple.h"
 
+#if MICROPY_ENABLE_MOD_COLLECTIONS
+
 typedef struct _mp_obj_namedtuple_type_t {
     mp_obj_type_t base;
     const char *fields;
@@ -143,3 +145,5 @@ STATIC mp_obj_t new_namedtuple_type(mp_obj_t name_in, mp_obj_t fields_in) {
     return mp_obj_new_namedtuple_type(name, fields);
 }
 MP_DEFINE_CONST_FUN_OBJ_2(mp_namedtuple_obj, new_namedtuple_type);
+
+#endif // MICROPY_ENABLE_MOD_COLLECTIONS