diff --git a/include/epanet2.bas b/include/epanet2.bas index 615f045..585d556 100644 --- a/include/epanet2.bas +++ b/include/epanet2.bas @@ -56,6 +56,7 @@ Public Const EN_EFFICIENCY = 16 Public Const EN_HEADCURVE = 17 Public Const EN_EFFICIENCYCURVE = 18 Public Const EN_PRICEPATTERN = 19 +Public Const EN_STATE = 20 Public Const EN_DURATION = 0 ' Time parameters Public Const EN_HYDSTEP = 1 diff --git a/include/epanet2.h b/include/epanet2.h index 0470d9e..3b26116 100644 --- a/include/epanet2.h +++ b/include/epanet2.h @@ -115,7 +115,8 @@ typedef enum { EN_EFFICIENCY = 16, EN_HEADCURVE = 17, EN_EFFICIENCYCURVE = 18, - EN_PRICEPATTERN = 19 + EN_PRICEPATTERN = 19, + EN_STATE = 20 } EN_LinkProperty; /// Time parameter codes diff --git a/src/epanet.c b/src/epanet.c index 72f59d8..0fe9b72 100644 --- a/src/epanet.c +++ b/src/epanet.c @@ -1994,6 +1994,7 @@ int DLLEXPORT EN_getlinknodes(EN_Project *p, int index, int *node1, int DLLEXPORT EN_getlinkvalue(EN_Project *p, int index, EN_LinkProperty code, EN_API_FLOAT_TYPE *value) { double a, h, q, v = 0.0; int returnValue = 0; + int pmp; EN_Network *net = &p->network; hydraulics_t *hyd = &p->hydraulics; @@ -2007,8 +2008,6 @@ int DLLEXPORT EN_getlinkvalue(EN_Project *p, int index, EN_LinkProperty code, EN double *LinkFlows = hyd->LinkFlows; double *LinkSetting = hyd->LinkSetting; - - /* Check for valid arguments */ *value = 0.0; if (!p->Openflag) @@ -2124,6 +2123,20 @@ int DLLEXPORT EN_getlinkvalue(EN_Project *p, int index, EN_LinkProperty code, EN else v = 1.0; break; + + case EN_STATE: + v = hyd->LinkStatus[index]; + + if (Link[index].Type == EN_PUMP) { + pmp = findpump(net, index); + if (hyd->LinkStatus[index] >= OPEN) { + if (hyd->LinkFlows[index] > hyd->LinkSetting[index] * Pump[pmp].Qmax) + v = XFLOW; + if (hyd->LinkFlows[index] < 0.0) + v = XHEAD; + } + } + break; case EN_SETTING: if (Link[index].Type == EN_PIPE || Link[index].Type == EN_CVPIPE) {