epanet-plus 0.0.1__cp310-cp310-macosx_10_9_x86_64.whl

python-extension/pyepanet2.h

@@ -0,0 +1,141 @@
+#include <Python.h>
+
+
+PyObject* method_ENopen(PyObject* self, PyObject* args);
+PyObject* method_ENclose(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENaddcontrol(PyObject* self, PyObject* args);
+PyObject* method_ENaddcurve(PyObject* self, PyObject* args);
+PyObject* method_ENadddemand(PyObject* self, PyObject* args);
+PyObject* method_ENaddlink(PyObject* self, PyObject* args);
+PyObject* method_ENaddnode(PyObject* self, PyObject* args);
+PyObject* method_ENaddpattern(PyObject* self, PyObject* args);
+PyObject* method_ENaddrule(PyObject* self, PyObject* args);
+PyObject* method_ENclearreport(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENcloseH(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENcloseQ(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENcopyreport(PyObject* self, PyObject* args);
+PyObject* method_ENdeletecontrol(PyObject* self, PyObject* args);
+PyObject* method_ENdeletecurve(PyObject* self, PyObject* args);
+PyObject* method_ENdeletedemand(PyObject* self, PyObject* args);
+PyObject* method_ENdeletelink(PyObject* self, PyObject* args);
+PyObject* method_ENdeletenode(PyObject* self, PyObject* args);
+PyObject* method_ENdeletepattern(PyObject* self, PyObject* args);
+PyObject* method_ENdeleterule(PyObject* self, PyObject* args);
+PyObject* method_ENgetaveragepatternvalue(PyObject* self, PyObject* args);
+PyObject* method_ENgetbasedemand(PyObject* self, PyObject* args);
+PyObject* method_ENgetcomment(PyObject* self, PyObject* args);
+PyObject* method_ENgetcontrol(PyObject* self, PyObject* args);
+PyObject* method_ENgetcoord(PyObject* self, PyObject* args);
+PyObject* method_ENgetcount(PyObject* self, PyObject* args);
+PyObject* method_ENgetcurve(PyObject* self, PyObject* args);
+PyObject* method_ENgetcurveid(PyObject* self, PyObject* args);
+PyObject* method_ENgetcurveindex(PyObject* self, PyObject* args);
+PyObject* method_ENgetcurvelen(PyObject* self, PyObject* args);
+PyObject* method_ENgetcurvetype(PyObject* self, PyObject* args);
+PyObject* method_ENgetcurvevalue(PyObject* self, PyObject* args);
+PyObject* method_ENgetdemandindex(PyObject* self, PyObject* args);
+PyObject* method_ENgetdemandmodel(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENgetdemandname(PyObject* self, PyObject* args);
+PyObject* method_ENgetdemandpattern(PyObject* self, PyObject* args);
+PyObject* method_ENgetelseaction(PyObject* self, PyObject* args);
+PyObject* method_ENgeterror(PyObject* self, PyObject* args);
+PyObject* method_ENgetflowunits(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENgetheadcurveindex(PyObject* self, PyObject* args);
+PyObject* method_ENgetlinkid(PyObject* self, PyObject* args);
+PyObject* method_ENgetlinkindex(PyObject* self, PyObject* args);
+PyObject* method_ENgetlinknodes(PyObject* self, PyObject* args);
+PyObject* method_ENgetlinktype(PyObject* self, PyObject* args);
+PyObject* method_ENgetlinkvalue(PyObject* self, PyObject* args);
+PyObject* method_ENgetnodeid(PyObject* self, PyObject* args);
+PyObject* method_ENgetnodeindex(PyObject* self, PyObject* args);
+PyObject* method_ENgetnodetype(PyObject* self, PyObject* args);
+PyObject* method_ENgetnodevalue(PyObject* self, PyObject* args);
+PyObject* method_ENgetnumdemands(PyObject* self, PyObject* args);
+PyObject* method_ENgetoption(PyObject* self, PyObject* args);
+PyObject* method_ENgetpatternid(PyObject* self, PyObject* args);
+PyObject* method_ENgetpatternindex(PyObject* self, PyObject* args);
+PyObject* method_ENgetpatternlen(PyObject* self, PyObject* args);
+PyObject* method_ENgetpatternvalue(PyObject* self, PyObject* args);
+PyObject* method_ENgetpremise(PyObject* self, PyObject* args);
+PyObject* method_ENgetpumptype(PyObject* self, PyObject* args);
+PyObject* method_ENgetqualinfo(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENgetqualtype(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENgetresultindex(PyObject* self, PyObject* args);
+PyObject* method_ENgetrule(PyObject* self, PyObject* args);
+PyObject* method_ENgetruleID(PyObject* self, PyObject* args);
+PyObject* method_ENgetstatistic(PyObject* self, PyObject* args);
+PyObject* method_ENgetthenaction(PyObject* self, PyObject* args);
+PyObject* method_ENgettimeparam(PyObject* self, PyObject* args);
+PyObject* method_ENgettitle(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENgetversion(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENgetvertex(PyObject* self, PyObject* args);
+PyObject* method_ENgetvertexcount(PyObject* self, PyObject* args);
+PyObject* method_ENinit(PyObject* self, PyObject* args);
+PyObject* method_ENinitH(PyObject* self, PyObject* args);
+PyObject* method_ENinitQ(PyObject* self, PyObject* args);
+PyObject* method_ENnextH(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENnextQ(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENopenH(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENopenQ(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENreport(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENresetreport(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENrunH(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENrunQ(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENsavehydfile(PyObject* self, PyObject* args);
+PyObject* method_ENsaveH(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENsaveinpfile(PyObject* self, PyObject* args);
+PyObject* method_ENsetbasedemand(PyObject* self, PyObject* args);
+PyObject* method_ENsetcomment(PyObject* self, PyObject* args);
+PyObject* method_ENsetcontrol(PyObject* self, PyObject* args);
+PyObject* method_ENsetcurveid(PyObject* self, PyObject* args);
+PyObject* method_ENsetcurve(PyObject* self, PyObject* args);
+PyObject* method_ENsetcoord(PyObject* self, PyObject* args);
+PyObject* method_ENsetcurvevalue(PyObject* self, PyObject* args);
+PyObject* method_ENsetdemandmodel(PyObject* self, PyObject* args);
+PyObject* method_ENsetdemandname(PyObject* self, PyObject* args);
+PyObject* method_ENsetdemandpattern(PyObject* self, PyObject* args);
+PyObject* method_ENsetelseaction(PyObject* self, PyObject* args);
+PyObject* method_ENsetflowunits(PyObject* self, PyObject* args);
+PyObject* method_ENsetheadcurveindex(PyObject* self, PyObject* args);
+PyObject* method_ENsetjuncdata(PyObject* self, PyObject* args);
+PyObject* method_ENsetlinkid(PyObject* self, PyObject* args);
+PyObject* method_ENsetlinknodes(PyObject* self, PyObject* args);
+PyObject* method_ENsetlinktype(PyObject* self, PyObject* args);
+PyObject* method_ENsetlinkvalue(PyObject* self, PyObject* args);
+PyObject* method_ENsetnodeid(PyObject* self, PyObject* args);
+PyObject* method_ENsetnodevalue(PyObject* self, PyObject* args);
+PyObject* method_ENsetoption(PyObject* self, PyObject* args);
+PyObject* method_ENsetpattern(PyObject* self, PyObject* args);
+PyObject* method_ENsetpatternid(PyObject* self, PyObject* args);
+PyObject* method_ENsetpatternvalue(PyObject* self, PyObject* args);
+PyObject* method_ENsetpipedata(PyObject* self, PyObject* args);
+PyObject* method_ENsetpremise(PyObject* self, PyObject* args);
+PyObject* method_ENsetpremiseindex(PyObject* self, PyObject* args);
+PyObject* method_ENsetpremisevalue(PyObject* self, PyObject* args);
+PyObject* method_ENsetpremisestatus(PyObject* self, PyObject* args);
+PyObject* method_ENsetqualtype(PyObject* self, PyObject* args);
+PyObject* method_ENsetreport(PyObject* self, PyObject* args);
+PyObject* method_ENsetrulepriority(PyObject* self, PyObject* args);
+PyObject* method_ENsetstatusreport(PyObject* self, PyObject* args);
+PyObject* method_ENsettankdata(PyObject* self, PyObject* args);
+PyObject* method_ENsetthenaction(PyObject* self, PyObject* args);
+PyObject* method_ENsettimeparam(PyObject* self, PyObject* args);
+PyObject* method_ENsettitle(PyObject* self, PyObject* args);
+PyObject* method_ENsetvertices(PyObject* self, PyObject* args);
+PyObject* method_ENsolveH(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENsolveQ(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENstepQ(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENusehydfile(PyObject* self, PyObject* args);
+PyObject* method_ENwriteline(PyObject* self, PyObject* args);
+PyObject* method_ENgettag(PyObject* self, PyObject* args);
+PyObject* method_ENsettag(PyObject* self, PyObject* args);
+PyObject* method_ENtimetonextevent(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_ENgetnodevalues(PyObject* self, PyObject* args);
+PyObject* method_ENgetlinkvalues(PyObject* self, PyObject* args);
+PyObject* method_ENsetvertex(PyObject* self, PyObject* args);
+PyObject* method_ENloadpatternfile(PyObject* self, PyObject* args);
+PyObject* method_ENsetcurvetype(PyObject* self, PyObject* args);
+PyObject* method_ENgetcontrolenabled(PyObject* self, PyObject* args);
+PyObject* method_ENsetcontrolenabled(PyObject* self, PyObject* args);
+PyObject* method_ENgetruleenabled(PyObject* self, PyObject* args);
+PyObject* method_ENsetruleenabled(PyObject* self, PyObject* args);

python-extension/pyepanet_plus.c

@@ -0,0 +1,37 @@
+#include <Python.h>
+#include "epanet_plus.h"
+
+
+PyObject* method_ENopenfrombuffer(PyObject* self, PyObject* args)
+{
+    char* inpBuffer = NULL;
+    char* inpFile = NULL;
+    char* rptFile = NULL;
+    char* outFile = NULL;
+
+    if(!PyArg_ParseTuple(args, "ssss", &inpBuffer, &inpFile, &rptFile, &outFile)) {
+        return NULL;
+    }
+
+    int err = ENopenfrombuffer(inpBuffer, inpFile, rptFile, outFile);
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_EN_openfrombuffer(PyObject* self, PyObject* args)
+{
+    uintptr_t ptr;
+    char* inpBuffer = NULL;
+    char* inpFile = NULL;
+    char* rptFile = NULL;
+    char* outFile = NULL;
+
+    if(!PyArg_ParseTuple(args, "Kssss", &ptr, &inpBuffer, &inpFile, &rptFile, &outFile)) {
+        return NULL;
+    }
+    EN_Project ph = (EN_Project) ptr;
+
+    int err = EN_openfrombuffer(ph, inpBuffer, inpFile, rptFile, outFile);
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}

python-extension/pyepanet_plus.h

@@ -0,0 +1,4 @@
+#include <Python.h>
+
+PyObject* method_ENopenfrombuffer(PyObject* self, PyObject* args);
+PyObject* method_EN_openfrombuffer(PyObject* self, PyObject* args);

python-extension/pyepanetmsx.c

@@ -0,0 +1,388 @@
+#include <Python.h>
+#include "epanetmsx.h"
+#include "msxtypes.h"
+
+#define   MAXID     31       // Max. # characters in ID name
+
+
+PyObject* method_MSXENopen(PyObject* self, PyObject* args)
+{
+    char *inpFile, *rptFile, *outFile = NULL;
+
+    if(!PyArg_ParseTuple(args, "sss", &inpFile, &rptFile, &outFile)) {
+        return NULL;
+    }
+
+    int err = MSXENopen(inpFile, rptFile, outFile);
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXopen(PyObject* self, PyObject* args)
+{
+    char* fname = NULL;
+    if(!PyArg_ParseTuple(args, "s", &fname)) {
+        return NULL;
+    }
+
+    int err = MSXopen(fname);
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXsolveH(PyObject* self, PyObject* Py_UNUSED(args))
+{
+    int err = MSXsolveH();
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXusehydfile(PyObject* self, PyObject* args)
+{
+    char* fname = NULL;
+    if(!PyArg_ParseTuple(args, "s", &fname)) {
+        return NULL;
+    }
+
+    int err = MSXusehydfile(fname);
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXsolveQ(PyObject* self, PyObject* Py_UNUSED(args))
+{
+    int err = MSXsolveQ();
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXinit(PyObject* self, PyObject* args)
+{
+    int saveFlag;
+    if(!PyArg_ParseTuple(args, "i", &saveFlag)) {
+        return NULL;
+    }   
+
+    int err = MSXinit(saveFlag);
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXstep(PyObject* self, PyObject* args)
+{
+    double t, tleft;
+    int err = MSXstep(&t, &tleft);
+
+    return Py_BuildValue("(idd)", err, t, tleft);
+}
+
+PyObject* method_MSXsaveoutfile(PyObject* self, PyObject* args)
+{
+    char* fname = NULL;
+    if(!PyArg_ParseTuple(args, "s", &fname)) {
+        return NULL;
+    }
+
+    int err = MSXsaveoutfile(fname);
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXsavemsxfile(PyObject* self, PyObject* args)
+{
+    char* fname = NULL;
+    if(!PyArg_ParseTuple(args, "s", &fname)) {
+        return NULL;
+    }
+
+    int err = MSXsavemsxfile(fname);
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXreport(PyObject* self, PyObject* Py_UNUSED(args))
+{
+    int err = MSXreport();
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXclose(PyObject* self, PyObject* Py_UNUSED(args))
+{
+    int err = MSXclose();
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXENclose(PyObject* self, PyObject* Py_UNUSED(args))
+{
+    int err = MSXENclose();
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXgetindex(PyObject* self, PyObject* args)
+{
+    int type, index;
+    char* id = NULL;
+
+    if(!PyArg_ParseTuple(args, "is", &type, &id)) {
+        return NULL;
+    }
+
+    int err = MSXgetindex(type, id, &index);
+
+    return PyTuple_Pack(2, PyLong_FromLong(err), PyLong_FromLong(index));
+}
+
+PyObject* method_MSXgetIDlen(PyObject* self, PyObject* args)
+{
+    int type, index, len;
+    if(!PyArg_ParseTuple(args, "ii", &type, &index)) {
+        return NULL;
+    }
+
+    int err = MSXgetIDlen(type, index, &len);
+
+    return PyTuple_Pack(2, PyLong_FromLong(err), PyLong_FromLong(len));
+}
+
+PyObject* method_MSXgetID(PyObject* self, PyObject* args)
+{
+    int type, index;
+    if(!PyArg_ParseTuple(args, "ii", &type, &index)) {
+        return NULL;
+    }
+
+    char id[MAXID + 1]; // TODO: MSXgetIDlen
+    int err = MSXgetID(type, index, &id[0], MAXID);
+
+    return PyTuple_Pack(2, PyLong_FromLong(err), PyUnicode_FromString(&id[0]));
+}
+
+PyObject* method_MSXgetcount(PyObject* self, PyObject* args)
+{
+    int type, count;
+    if(!PyArg_ParseTuple(args, "i", &type)) {
+        return NULL;
+    }
+
+    int err = MSXgetcount(type, &count);
+
+    return PyTuple_Pack(2, PyLong_FromLong(err), PyLong_FromLong(count));
+}
+
+PyObject* method_MSXgetspecies(PyObject* self, PyObject* args)
+{
+    int index, type;
+    char units[MAXUNITS];
+    double aTol, rTol;
+
+    if(!PyArg_ParseTuple(args, "i", &index)) {
+        return NULL;
+    }
+
+    int err = MSXgetspecies(index, &type, &units[0], &aTol, &rTol);
+
+    return PyTuple_Pack(5, PyLong_FromLong(err), PyLong_FromLong(type), PyUnicode_FromString(&units[0]), PyFloat_FromDouble(aTol), PyFloat_FromDouble(rTol));
+}
+
+PyObject* method_MSXgetconstant(PyObject* self, PyObject* args)
+{
+    int index;
+    if(!PyArg_ParseTuple(args, "i", &index)) {
+        return NULL;
+    }
+
+    double value;
+    int err = MSXgetconstant(index, &value);
+
+    return PyTuple_Pack(2, PyLong_FromLong(err), PyFloat_FromDouble(value));
+}
+
+PyObject* method_MSXgetparameter(PyObject* self, PyObject* args)
+{
+    int type, index, param;
+    if(!PyArg_ParseTuple(args, "iii", &type, &index, &param)) {
+        return NULL;
+    }
+
+    double value;
+    int err = MSXgetparameter(type, index, param, &value);
+
+    return PyTuple_Pack(2, PyLong_FromLong(err), PyFloat_FromDouble(value));
+}
+
+PyObject* method_MSXgetsource(PyObject* self, PyObject* args)
+{
+    int node, species;
+    if(!PyArg_ParseTuple(args, "ii", &node, &species)) {
+        return NULL;
+    }
+
+    int type, pat;
+    double level;
+    int err = MSXgetsource(node, species, &type, &level, &pat);
+
+    return PyTuple_Pack(4, PyLong_FromLong(err), PyLong_FromLong(type), PyFloat_FromDouble(level), PyLong_FromLong(pat));
+}
+
+PyObject* method_MSXgetpatternlen(PyObject* self, PyObject* args)
+{
+    int pat, len;
+    if(!PyArg_ParseTuple(args, "i", &pat)) {
+        return NULL;
+    }
+
+    int err = MSXgetpatternlen(pat, &len);
+
+    return PyTuple_Pack(2, PyLong_FromLong(err), PyLong_FromLong(len));
+}
+
+PyObject* method_MSXgetpatternvalue(PyObject* self, PyObject* args)
+{
+    int pat, period;
+    if(!PyArg_ParseTuple(args, "ii", &pat, &period)) {
+        return NULL;
+    }   
+
+    double value;
+    int err = MSXgetpatternvalue(pat, period, &value);
+
+    return PyTuple_Pack(2, PyLong_FromLong(err), PyFloat_FromDouble(value));
+}
+
+PyObject* method_MSXgetinitqual(PyObject* self, PyObject* args)
+{
+    int type, index, species;
+    if(!PyArg_ParseTuple(args, "iii", &type, &index, &species)) {
+        return NULL;
+    }    
+
+    double value;
+    int err = MSXgetinitqual(type, index, species, &value);
+
+    return PyTuple_Pack(2, PyLong_FromLong(err), PyFloat_FromDouble(value));
+}
+
+PyObject* method_MSXgetqual(PyObject* self, PyObject* args)
+{
+    int type, index, species;
+    if(!PyArg_ParseTuple(args, "iii", &type, &index, &species)) {
+        return NULL;
+    }
+
+    double value;
+    int err = MSXgetqual(type, index, species, &value);
+
+    return PyTuple_Pack(2, PyLong_FromLong(err), PyFloat_FromDouble(value));
+}
+
+PyObject* method_MSXgeterror(PyObject* self, PyObject* args)
+{
+    int code;
+    if(!PyArg_ParseTuple(args, "i", &code)) {
+        return NULL;
+    }
+    
+    char msg[MAXLINE + 1];
+    int err = MSXgeterror(code, &msg[0], MAXLINE);
+
+    return PyTuple_Pack(2, PyLong_FromLong(err), PyUnicode_FromString(&msg[0]));
+}
+
+PyObject* method_MSXsetconstant(PyObject* self, PyObject* args)
+{
+    int index;
+    double value;
+    if(!PyArg_ParseTuple(args, "id", &index, &value)) {
+        return NULL;
+    }
+
+    int err = MSXsetconstant(index, value);
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXsetparameter(PyObject* self, PyObject* args)
+{
+    int type, index, param;
+    double value;
+    if(!PyArg_ParseTuple(args, "iiid", &type, &index, &param, &value)) {
+        return NULL;
+    } 
+
+    int err = MSXsetparameter(type, index, param, value);
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXsetinitqual(PyObject* self, PyObject* args)
+{
+    int type, index, species;
+    double value;
+    if(!PyArg_ParseTuple(args, "iiid", &type, &index, &species, &value)) {
+        return NULL;
+    }
+
+    int err = MSXsetinitqual(type, index, species, value);
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXsetsource(PyObject* self, PyObject* args)
+{
+    int node, species, type, pat;
+    double level;
+    if(!PyArg_ParseTuple(args, "iiidi", &node, &species, &type, &level, &pat)) {
+        return NULL;
+    }
+
+    int err = MSXsetsource(node, species, type, level, pat);
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXsetpatternvalue(PyObject* self, PyObject* args)
+{
+    int pat, period;
+    double value;
+    if(!PyArg_ParseTuple(args, "iid", &pat, &period, &value)) {
+        return NULL;
+    }
+
+    int err = MSXsetpatternvalue(pat, period, value);
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXsetpattern(PyObject* self, PyObject* args)
+{
+    int pat, len;
+    PyListObject* mult = NULL;
+    if(!PyArg_ParseTuple(args, "iOi", &pat, &mult, &len)) {
+        return NULL;
+    }
+
+    double* multRaw = (double*) malloc(sizeof(double) * len);
+    for(int i=0; i != len; i++) {
+        multRaw[i] = PyFloat_AsDouble(PyList_GET_ITEM(mult, i));
+    }
+
+    int err = MSXsetpattern(pat, multRaw, len);
+    free(multRaw);
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}
+
+PyObject* method_MSXaddpattern(PyObject* self, PyObject* args)
+{
+    char* id = NULL;
+    if(!PyArg_ParseTuple(args, "s", &id)) {
+        return NULL;
+    }
+
+    int err = MSXaddpattern(id);
+
+    return PyTuple_Pack(1, PyLong_FromLong(err));
+}

python-extension/pyepanetmsx.h

@@ -0,0 +1,35 @@
+#include <Python.h>
+
+
+PyObject* method_MSXENopen(PyObject* self, PyObject* args);
+PyObject* method_MSXopen(PyObject* self, PyObject* args);
+PyObject* method_MSXsolveH(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_MSXusehydfile(PyObject* self, PyObject* args);
+PyObject* method_MSXsolveQ(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_MSXinit(PyObject* self, PyObject* args);
+PyObject* method_MSXstep(PyObject* self, PyObject* args);
+PyObject* method_MSXsaveoutfile(PyObject* self, PyObject* args);
+PyObject* method_MSXsavemsxfile(PyObject* self, PyObject* args);
+PyObject* method_MSXreport(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_MSXclose(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_MSXENclose(PyObject* self, PyObject* Py_UNUSED(args));
+PyObject* method_MSXgetindex(PyObject* self, PyObject* args);
+PyObject* method_MSXgetIDlen(PyObject* self, PyObject* args);
+PyObject* method_MSXgetID(PyObject* self, PyObject* args);
+PyObject* method_MSXgetcount(PyObject* self, PyObject* args);
+PyObject* method_MSXgetspecies(PyObject* self, PyObject* args);
+PyObject* method_MSXgetconstant(PyObject* self, PyObject* args);
+PyObject* method_MSXgetparameter(PyObject* self, PyObject* args);
+PyObject* method_MSXgetsource(PyObject* self, PyObject* args);
+PyObject* method_MSXgetpatternlen(PyObject* self, PyObject* args);
+PyObject* method_MSXgetpatternvalue(PyObject* self, PyObject* args);
+PyObject* method_MSXgetinitqual(PyObject* self, PyObject* args);
+PyObject* method_MSXgetqual(PyObject* self, PyObject* args);
+PyObject* method_MSXgeterror(PyObject* self, PyObject* args);
+PyObject* method_MSXsetconstant(PyObject* self, PyObject* args);
+PyObject* method_MSXsetparameter(PyObject* self, PyObject* args);
+PyObject* method_MSXsetinitqual(PyObject* self, PyObject* args);
+PyObject* method_MSXsetsource(PyObject* self, PyObject* args);
+PyObject* method_MSXsetpatternvalue(PyObject* self, PyObject* args);
+PyObject* method_MSXsetpattern(PyObject* self, PyObject* args);
+PyObject* method_MSXaddpattern(PyObject* self, PyObject* args);

tests/test_epanet.py

@@ -0,0 +1,16 @@
+"""
+This module tests the EPANET functions.
+"""
+from epanet_plus import EpanetAPI, EpanetConstants
+
+
+def test():
+    epanet_api = EpanetAPI()
+    epanet_api.open("net2-cl2.inp", "net2-cl2.rpt", "")
+
+    assert epanet_api.gettitle() is not None
+    assert epanet_api.getcount(EpanetConstants.EN_NODECOUNT) > 0
+    assert epanet_api.getcount(EpanetConstants.EN_LINKCOUNT) > 0
+    assert epanet_api.getcount(EpanetConstants.EN_TANKCOUNT) > 0
+
+    epanet_api.close()

tests/test_epanetmsx.py

@@ -0,0 +1,36 @@
+"""
+This module tests EPANET-MSX functions.
+"""
+from epanet_plus import EPyT, EpanetAPI, EpanetConstants
+
+
+def test_msx_basic():
+    epanet_api = EpanetAPI()
+    epanet_api.MSXENopen("net2-cl2.inp", "net2-cl2.rpt", "")
+    epanet_api.MSXopen("net2-cl2.msx")
+
+    epanet_api.gettitle()
+    epanet_api.MSXgetspecies(1)
+    epanet_api.MSXgetID(3, 1)
+
+    epanet_api.MSXclose()
+    epanet_api.MSXENclose()
+
+
+def test_simulation():
+    with EPyT("net2-cl2.inp", use_project=False) as epanet_api:
+        epanet_api.load_msx_file("net2-cl2.msx")
+
+        epanet_api.MSXsolveH()
+
+        epanet_api.MSXinit(0)
+        while True:
+            _, tleft = epanet_api.MSXstep()
+
+            for idx in epanet_api.get_all_nodes_idx():
+                assert epanet_api.MSXgetqual(EpanetConstants.MSX_NODE, idx, 1) >= 0
+            for idx in epanet_api.get_all_pipes_idx():
+                assert epanet_api.MSXgetqual(EpanetConstants.MSX_LINK, idx, 1) >= 0
+
+            if tleft == 0:
+                break