Calculate each pollutant's concentration with Output module

[Karakurko]

I define 3 pollutants in SWMM inp file and execute it successfully. Then I want to extract the results of each pollutant's concentration of certain nodes with Pyswmm Output module. I check swmm.toolkit.shared_enum.py and find a relevant parameter named NodeAttribute.POLLUT_CONC_0, which is defined as concentration of each pollutant, but with the codes below only one sequence can be extracted, which looks alike the CODR result I get directly from SWMM using the same output file. So I wonder is it possilble to extract the results of all the pollutants I define with current version of Pyswmm?

BTW my SWMM version is 5.1.14 and Pyswmm version is 1.4.0, if it helps.

from pyswmm import Simulation, Output
from swmm.toolkit.shared_enum import NodeAttribute
import matplotlib.pyplot as plt

sim = Simulation('D:/study/PKU/SWMM/test/before_LID.inp')
sim.execute()

with Output('D:/study/PKU/SWMM/test/before_LID.out') as out: 
    S1_pollution = list(out.node_series('J1', NodeAttribute.POLLUT_CONC_0).values())
    time_stamps = out.times

fig = plt.figure(figsize=(8,4),dpi=200, constrained_layout=True) 
ax = fig.add_subplot(111)
ax.plot(time_stamps, S1_pollution, color= 'b')
ax.set_ylabel('Concentration(mg/L)')
ax.set_xlabel('Time')
fig.autofmt_xdate()
plt.tight_layout()
plt.show()

[bemcdonnell]

@Karakurko, I agree that this might be a bug. Let me have a look at the accessors and see if we overlooked something. That being said, if you want to extract info and it’s not a super long simulation, you can get the WQ data while pyswmm is running for all your polluts

[bemcdonnell]

I understand. let met look at it and see what I can do!

[bemcdonnell]

@Karakurko, can you send me a test model with your 3 pollutants? I might have a simple solution that you can use today. It's not the most incredible, but it could be functional. I just want to try :)

[Karakurko]

Sure, but it seems that inp file is not supported to be delivered on this page, I will send it to ur email instead.

[bemcdonnell]

@Karakurko just zip it up and drop it in here (and put your python code as well)

[Karakurko]

Sry my connection is bad or restricted so it took me so long to upload
test.zip

[bemcdonnell]

@Karakurko, ok i have a very simple work around for you!

You'll need to make a very simple modification to swmm.toolkit python code. Depending if you are using a VENV or just have pyswmm/swmm-toolkit globally navigate to:

... /site-packages/swmm/toolkit/shared_enum.py

Then add these following lines:

class NodeAttribute(Enum, start = 0):
    """Node Attribute enum class.

    .. rubric:: Enum Members

    ========================= ===============================
    :attr:`~INVERT_DEPTH`     water depth above invert
    :attr:`~HYDRAULIC_HEAD`   hydraulic head
    :attr:`~PONDED_VOLUME`    volume stored and ponded
    :attr:`~LATERAL_INFLOW`   lateral inflow rate
    :attr:`~TOTAL_INFLOW`     total inflow rate
    :attr:`~FLOODING_LOSSES`  overflow rate
    :attr:`~POLLUT_CONC_0`    concentration of each pollutant
    ========================= ===============================
    """
    INVERT_DEPTH = 0
    HYDRAULIC_HEAD = 1
    PONDED_VOLUME = 2
    LATERAL_INFLOW = 3
    TOTAL_INFLOW = 4
    FLOODING_LOSSES = 5
    POLLUT_CONC_0 = 6
    POLLUT_CONC_1 = 7
    POLLUT_CONC_2 = 8

Finally, let's use those enums

with Output('test.out') as out:

    S1_pollution = list(out.node_series('J1', NodeAttribute.POLLUT_CONC_0).values())
    S2_pollution = list(out.node_series('J1', NodeAttribute.POLLUT_CONC_1).values())
    S3_pollution = list(out.node_series('J1', NodeAttribute.POLLUT_CONC_2).values())
    time_stamps = out.times

creates:

[Karakurko]

Oh this really does work! Thank you so much!!