Added memory module

2 files changed, 47 insertions, 6 deletions

diff --git a/src/hwdmodule.cpp b/src/gpio.cpp
index 6d3a574..478bcc7 100644
--- a/src/hwdmodule.cpp
+++ b/src/gpio.cpp
@@ -36,7 +36,7 @@ static PyObject * gpio_read(PyObject * self, PyObject * args)
 }
 
 
-static PyMethodDef HwdMethods[] = {
+static PyMethodDef gpio_methods[] = {
 	{"set_mode", gpio_set_mode, METH_VARARGS, "set_mode(port, mode)\n--\n\nSets *mode* of a *port*."},
 	{"write", gpio_write, METH_VARARGS, "write(port, value)\n--\n\nWrites *value* to a *port*."},
 	{"read", gpio_read, METH_VARARGS, "read(port)\n--\n\nReads a value from a *port*."},
@@ -44,16 +44,16 @@ static PyMethodDef HwdMethods[] = {
 };
 
 
-static struct PyModuleDef hwdmodule = {
+static struct PyModuleDef gpio_module = {
 	PyModuleDef_HEAD_INIT,
-	"hwd",
+	"gpio",
 	NULL,
 	-1,
-	HwdMethods
+	gpio_methods
 };
 
 
-PyMODINIT_FUNC PyInit_hwd(void)
+PyMODINIT_FUNC PyInit_gpio(void)
 {
-	return PyModule_Create(&hwdmodule);
+	return PyModule_Create(&gpio_module);
 }
diff --git a/src/memory.cpp b/src/memory.cpp
new file mode 100644
index 0000000..415695d
--- /dev/null
+++ b/src/memory.cpp
@@ -0,0 +1,41 @@
+#define PY_SSIZE_T_CLEAN
+#include <Python.h>
+
+#include <cstddef>
+#include <vector>
+
+#include <hwd.h>
+
+
+static PyObject * memory_write(PyObject * self, PyObject * args)
+{
+	Py_ssize_t offset;
+	Py_buffer data;
+	if (!PyArg_ParseTuple(args, "y*n", &data, &offset))
+		return NULL;
+	hwd::memory::write(
+		std::vector<char> {static_cast<char *>(data.buf), static_cast<char *>(data.buf) + data.len},
+		static_cast<std::size_t>(offset));
+	Py_RETURN_NONE;
+}
+
+
+static PyMethodDef memory_methods[] = {
+	{"write", memory_write, METH_VARARGS, "mem_write(data, offset)\n--\n\nWrites *data* to memory at *offset*."},
+	{NULL, NULL, 0, NULL}
+};
+
+
+static struct PyModuleDef memory_module = {
+	PyModuleDef_HEAD_INIT,
+	"memory",
+	NULL,
+	-1,
+	memory_methods
+};
+
+
+PyMODINIT_FUNC PyInit_memory(void)
+{
+	return PyModule_Create(&memory_module);
+}