def __init__(self, inputfile, reportfile=None, outputfile=None): self._model = PySWMM(inputfile, reportfile, outputfile) self._model.swmm_open() self._isOpen = True self._advance_seconds = None self._isStarted = False self._initial_conditions = None
def test_pyswmm_exception(): swmmobject = PySWMM(*get_model_files(MODEL_WEIR_SETTING_PATH)) with pytest.raises(PYSWMMException): swmmobject.swmm_open() print(swmmobject.fileLoaded) swmmobject.swmm_open() swmmobject.swmm_close()
def __init__(self, inputfile, reportfile=None, outputfile=None, swmm_lib_path=None): self._model = PySWMM(inputfile, reportfile, outputfile, swmm_lib_path) self._model.swmm_open() self._isOpen = True self._advance_seconds = None self._isStarted = False self._callbacks = { "before_start": None, "before_step": None, "after_step": None, "before_end": None, "after_end": None, "after_close": None }
def __init__(self, inputfile, reportfile=None, outputfile=None, swmm_lib_path=None): self._model = PySWMM(inputfile, reportfile, outputfile, swmm_lib_path) self._model.swmm_open() self._isOpen = True self._advance_seconds = None self._isStarted = False self._callbacks = { "before_start": None, "before_step": None, "after_step": None, "before_end": None, "after_end": None, "after_close": None }
class Simulation(object): """ Base class for a SWMM Simulation. The model object provides several options to run a simulation. User can specified SWMM library path. Uses default lib if not provided. Initialize the Simulation class. :param str inpfile: Name of SWMM input file (default '') :param str rptfile: Report file to generate (default None) :param str binfile: Optional binary output file (default None) :param str swmm_lib_path: User-specified SWMM library path (default None). Examples: Intialize a simulation and iterate through a simulation. This approach requires some clean up. >>> from pyswmm import Simulation >>> >>> sim = Simulation('tests/data/TestModel1_weirSetting.inp') >>> for step in sim: ... pass >>> >>> sim.report() >>> sim.close() Intialize using with statement. This automatically cleans up after a simulation >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... pass ... sim.report() Initialize the simulation and execute. This style does not allow the user to interact with the simulation. However, this approach tends to be the fastest. >>> from pyswmm import Simulation >>> >>> sim = Simulation('tests/data/TestModel1_weirSetting.inp') >>> sim.execute() """ def __init__(self, inputfile, reportfile=None, outputfile=None, swmm_lib_path=None): self._model = PySWMM(inputfile, reportfile, outputfile, swmm_lib_path) self._model.swmm_open() self._isOpen = True self._advance_seconds = None self._isStarted = False self._callbacks = { "before_start": None, "before_step": None, "after_step": None, "before_end": None, "after_end": None, "after_close": None } def __enter__(self): """ Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... print(sim.current_time) ... sim.report() >>> >>> 2015-11-01 14:00:30 >>> 2015-11-01 14:01:00 >>> 2015-11-01 14:01:30 >>> 2015-11-01 14:02:00 """ return self def __iter__(self): """Iterator over Simulation""" return self def start(self): """Start Simulation""" if not self._isStarted: # Set Model Initial Conditions # (This Will be Deprecated with Time) if hasattr(self, "_initial_conditions"): self._initial_conditions() # Execute Callback Hooks Before Simulation self._execute_callback(self.before_start()) self._model.swmm_start(True) self._isStarted = True def __next__(self): """Next""" # Start Simulation self.start() # Execute Callback Hooks Before Simulation Step self._execute_callback(self.before_step()) # Simulation Step Amount if self._advance_seconds is None: time = self._model.swmm_step() else: time = self._model.swmm_stride(self._advance_seconds) # Execute Callback Hooks After Simulation Step self._execute_callback(self.after_step()) if time <= 0.0: self._execute_callback(self.before_end()) raise StopIteration return self._model next = __next__ # Python 2 def __exit__(self, *a): """close""" if self._isStarted: self._model.swmm_end() self._isStarted = False # Execute Callback Hooks After Simulation End self._execute_callback(self.after_end()) if self._isOpen: self._model.swmm_close() self._isOpen = False # Execute Callback Hooks After Simulation Closes self._execute_callback(self.after_close()) @staticmethod def _is_callback(callable_object): """Checks if arugment is a function/method.""" if not callable(callable_object): error_msg = 'Requires Callable Object, not {}'.format( type(callable_object)) raise (PYSWMMException(error_msg)) else: return True def _execute_callback(self, callback): """Runs the callback.""" if callback: try: callback() except PYSWMMException: error_msg = "Callback Failed" raise PYSWMMException((error_msg)) def initial_conditions(self, init_conditions): """ Intial Conditions for Hydraulics and Hydrology can be set from within the api by setting a function to the initial_conditions property. >>> from pyswmm import Simulation >>> >>> with Simulation('./TestModel1_weirSetting.inp') as sim: ... nodeJ1 = Nodes(sim)["J1"] ... ... def init_conditions(): ... nodeJ1.initial_depth = 4 ... ... sim.initial_conditions(init_conditions) ... ... for step in sim: ... pass ... sim.report() """ if self._is_callback(init_conditions): self._initial_conditions = init_conditions def before_start(self): """Get Before Start Callback. :return: Callbacks """ return self._callbacks["before_start"] def add_before_start(self, callback): """ Add callback function/method/object to execute before the simlation starts. Needs to be callable. :param func callback: Callable Object >>> from pyswmm import Simulation >>> >>> def test_callback(): ... print("CALLBACK - Executed") >>> >>> with Simulation('./TestModel1_weirSetting.inp') as sim: ... ... sim.before_start(test_callback) #<- pass function handle. ... print("Waiting to Start") ... for ind, step in enumerate(sim): ... print("Step {}".format(ind)) ... print("Complete!") ... print("Closed") >>> >>> "Waiting to Start" >>> "CALLBACK - Executed" >>> "Step 0" >>> "Step 1" >>> ... >>> "Complete!" >>> "Closed" """ if self._is_callback(callback): self._callbacks["before_start"] = callback def before_step(self): """Get Before Step Callback. :return: Callbacks """ return self._callbacks["before_step"] def add_before_step(self, callback): """ Add callback function/method/object to execute before a simlation step. Needs to be callable. :param func callback: Callable Object (See self.add_before_start() for more details) """ if self._is_callback(callback): self._callbacks["before_step"] = callback def after_step(self): """Get After Step Callback. :return: Callbacks """ return self._callbacks["after_step"] def add_after_step(self, callback): """ Add callback function/method/object to execute after a simlation step. Needs to be callable. :param func callback: Callable Object (See self.add_before_start() for more details) """ if self._is_callback(callback): self._callbacks["after_step"] = callback def before_end(self): """Get Before End Callback. :return: Callbacks """ return self._callbacks["before_end"] def add_before_end(self, callback): """ Add callback function/method/object to execute after the simulation ends. Needs to be callable. :param func callback: Callable Object (See self.add_before_start() for more details) """ if self._is_callback(callback): self._callbacks["before_end"] = callback def after_end(self): """Get After End Callback. :return: Callbacks """ return self._callbacks["after_end"] def add_after_end(self, callback): """ Add callback function/method/object to execute after the simulation ends. Needs to be callable. :param func callback: Callable Object (See self.add_before_start() for more details) """ if self._is_callback(callback): self._callbacks["after_end"] = callback def after_close(self): """Get After Close Callback. :return: Callbacks """ return self._callbacks["after_close"] def add_after_close(self, callback): """ Add callback function/method/object to execute after the simulation is closed. Needs to be callable. :param func callback: Callable Object (See self.add_before_start() for more details) """ if self._is_callback(callback): self._callbacks["after_close"] = callback def step_advance(self, advance_seconds): """ Advances the model by X number of seconds instead of intervening at every routing step. This does not change the routing step for the simulation; only lets python take back control after each advance period. :param int advance_seconds: Seconds to Advance simulation Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... sim.step_advance(300) ... for step in sim: ... print(step.current_time) ... # or here! sim.step_advance(newvalue) ... sim.report() >>> >>> 2015-11-01 14:00:30 >>> 2015-11-01 14:01:00 >>> 2015-11-01 14:01:30 >>> 2015-11-01 14:02:00 """ self._advance_seconds = advance_seconds def report(self): """ Writes to report file after simulation. Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... pass ... sim.report() """ self._model.swmm_report() def close(self): """ Intialize a simulation and iterate through a simulation. This approach requires some clean up. Examples: >>> from pyswmm import Simulation >>> >>> sim = Simulation('./TestModel1_weirSetting.inp') >>> for step in sim: ... pass >>> >>> sim.report() >>> sim.close() """ self.__exit__() def execute(self): """ Open an input file, run SWMM, then close the file. Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.execute() """ self._model.swmmExec() @property def engine_version(self): """ Retrieves the SWMM Engine Version. :return: Engine Version :rtype: StrictVersion Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.engine_version StrictVersion("5.1.13") """ return self._model.swmm_getVersion() @property def runoff_error(self): """ Retrieves the Runoff Mass Balance Error. :return: Runoff Mass Balance Error :rtype: float Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.execute() >>> sim.runoff_error 0.01 """ return self._model.swmm_getMassBalErr()[0] @property def flow_routing_error(self): """ Retrieves the Flow Routing Mass Balance Error. :return: Flow Routing Mass Balance Error :rtype: float Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.execute() >>> sim.flow_routing_error 0.01 """ return self._model.swmm_getMassBalErr()[1] @property def quality_error(self): """ Retrieves the Quality Routing Mass Balance Error. :return: Quality Routing Mass Balance Error :rtype: float Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.execute() >>> sim.quality_error 0.01 """ return self._model.swmm_getMassBalErr()[2] @property def start_time(self): """Get/set Simulation start time. Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... print sim.start_time ... sim.start_time = datetime(2015,5,10,15,15,1) >>> >>> datetime.datetime(2015,5,10,15,15,1) """ return self._model.getSimulationDateTime( SimulationTime.StartDateTime.value) @start_time.setter def start_time(self, dtimeval): """Set simulation Start time""" self._model.setSimulationDateTime(SimulationTime.StartDateTime.value, dtimeval) @property def end_time(self): """Get/set Simulation end time. Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... print sim.end_time ... sim.end_time = datetime(2016,5,10,15,15,1) >>> >>> datetime.datetime(2016,5,10,15,15,1) """ return self._model.getSimulationDateTime( SimulationTime.EndDateTime.value) @end_time.setter def end_time(self, dtimeval): """Set simulation End time.""" self._model.setSimulationDateTime(SimulationTime.EndDateTime.value, dtimeval) @property def report_start(self): """Get/set Simulation report start time. Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... print sim.report_start ... sim.report_start = datetime(2015,5,10,15,15,1) >>> >>> datetime.datetime(2015,5,10,15,15,1) """ return self._model.getSimulationDateTime( SimulationTime.ReportStart.value) @report_start.setter def report_start(self, dtimeval): """Set simulation report start time.""" self._model.setSimulationDateTime(SimulationTime.ReportStart.value, dtimeval) @property def flow_units(self): """ Get Simulation Units (CFS, GPM, MGD, CMS, LPS, MLD). :return: Flow Unit :rtype: str Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... print sim.flow_units >>> >>> CFS """ return self._model.getSimUnit(SimulationUnits.FlowUnits.value) @property def system_units(self): """Get system units (US, SI). :return: System Unit :rtype: str Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... print sim.system_units >>> >>> US """ return self._model.getSimUnit(SimulationUnits.UnitSystem.value) @property def current_time(self): """Get Simulation Current Time. :return: Current Simulation Time :rtype: Datetime Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... print(sim.current_time) ... sim.report() >>> >>> 2015-11-01 14:00:30 >>> 2015-11-01 14:01:00 >>> 2015-11-01 14:01:30 >>> 2015-11-01 14:02:00 """ return self._model.getCurrentSimulationTime() @property def percent_complete(self): """Get Simulation Percent Complete. :return: Current Percent Complete :rtype: Datetime Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... print(sim.percent_complete) ... sim.report() >>> >>> 0.01 >>> 0.25 >>> 0.50 >>> 0.75 """ dt = self.current_time - self.start_time total_time = self.end_time - self.start_time return float(dt.total_seconds()) / total_time.total_seconds()
class Simulation(object): """ Base class for a SWMM Simulation. The model object provides several options to run a simulation. Examples: Intialize a simulation and iterate through a simulation. This approach requires some clean up. >>> from pyswmm import Simulation >>> >>> sim = Simulation('./TestModel1_weirSetting.inp') >>> for step in sim: ... pass >>> >>> sim.report() >>> sim.close() Intialize using with statement. This automatically cleans up after a simulation >>> from pyswmm import Simulation >>> >>> with Simulation('./TestModel1_weirSetting.inp') as sim: ... for step in sim: ... pass ... sim.report() Initialize the simulation and execute. This style does not allow the user to interact with the simulation. However, this approach tends to be the fastest. >>> from pyswmm import Simulation >>> >>> sim = Simulation('./TestModel1_weirSetting.inp') >>> sim.execute() """ def __init__(self, inputfile, reportfile=None, outputfile=None): self._model = PySWMM(inputfile, reportfile, outputfile) self._model.swmm_open() self._advance_seconds = None self._isStarted = False def __enter__(self): """ Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... print(sim.current_time) ... sim.report() >>> >>> 2015-11-01 14:00:30 >>> 2015-11-01 14:01:00 >>> 2015-11-01 14:01:30 >>> 2015-11-01 14:02:00 """ return self def __iter__(self): """Iterator over Simulation""" return self def next(self): """Next""" # Start Simulation if not self._isStarted: self._model.swmm_start(True) self._isStarted = True # Simulation Step Amount if self._advance_seconds is None: time = self._model.swmm_step() else: time = self._model.swmm_stride(self._advance_seconds) if time <= 0.0: self._model.swmm_end() raise StopIteration return self._model def __exit__(self, *a): """close""" self._model.swmm_end() self._model.swmm_close() def step_advance(self, advance_seconds): """ Advances the model by X number of seconds instead of intervening at every routing step. This does not change the routing step for the simulation; only lets python take back control after each advance period. :param int advance_seconds: Seconds to Advance simulation Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... sim.step_advance(300) ... for step in sim: ... print(step.current_time) ... # or here! sim.step_advance(newvalue) ... sim.report() >>> >>> 2015-11-01 14:00:30 >>> 2015-11-01 14:01:00 >>> 2015-11-01 14:01:30 >>> 2015-11-01 14:02:00 """ self._advance_seconds = advance_seconds def report(self): """ Writes to report file after simulation Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... pass ... sim.report() """ self._model.swmm_report() def close(self): """ Intialize a simulation and iterate through a simulation. This approach requires some clean up. Examples: >>> from pyswmm import Simulation >>> >>> sim = Simulation('./TestModel1_weirSetting.inp') >>> for step in sim: ... pass >>> >>> sim.report() >>> sim.close() """ self.__exit__() def execute(self): """ Open an input file, run SWMM, then close the file. Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.execute() """ self._model.swmmExec() @property def start_time(self): """Get/set Simulation start time Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... print sim.start_time ... sim.start_time = datetime(2015,5,10,15,15,1) >>> >>> datetime.datetime(2015,5,10,15,15,1) """ return self._model.getSimulationDateTime( SimulationTime.StartDateTime.value) @start_time.setter def start_time(self, dtimeval): """Set simulation Start time""" self._model.setSimulationDateTime(SimulationTime.StartDateTime.value, dtimeval) @property def end_time(self): """Get/set Simulation end time Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... print sim.end_time ... sim.end_time = datetime(2016,5,10,15,15,1) >>> >>> datetime.datetime(2016,5,10,15,15,1) """ return self._model.getSimulationDateTime( SimulationTime.EndDateTime.value) @end_time.setter def end_time(self, dtimeval): """Set simulation End time""" self._model.setSimulationDateTime(SimulationTime.EndDateTime.value, dtimeval) @property def report_start(self): """Get/set Simulation report start time Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... print sim.report_start ... sim.report_start = datetime(2015,5,10,15,15,1) >>> >>> datetime.datetime(2015,5,10,15,15,1) """ return self._model.getSimulationDateTime( SimulationTime.ReportStart.value) @report_start.setter def report_start(self, dtimeval): """Set simulation report start time""" self._model.setSimulationDateTime(SimulationTime.ReportStart.value, dtimeval) @property def flow_units(self): """ Get Simulation Units (CFS, GPM, MGD, CMS, LPS, MLD). :return: Flow Unit :rtype: str Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... print sim.flow_units >>> >>> CFS """ return self._model.getSimUnit(SimulationUnits.FlowUnits.value) @property def system_units(self): """Get system units (US, SI). :return: System Unit :rtype: str Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... print sim.system_units >>> >>> US """ return self._model.getSimUnit(SimulationUnits.UnitSystem.value) @property def current_time(self): """Get Simulation Current Time. :return: Current Simulation Time :rtype: Datetime Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... print(sim.current_time) ... sim.report() >>> >>> 2015-11-01 14:00:30 >>> 2015-11-01 14:01:00 >>> 2015-11-01 14:01:30 >>> 2015-11-01 14:02:00 """ return self._model.getCurrentSimualationTime()
def test_error_rpt_out(): swmmobject = PySWMM(MODEL_WEIR_SETTING_PATH) swmmobject.swmm_open()
def test_nodes_1(): swmmobject = PySWMM(MODEL_WEIR_SETTING_PATH) swmmobject.swmm_open() node = Node(swmmobject, "J1") print(node.invert_elevation)
def test_plot_weir_setting(): swmmobject = PySWMM(*get_model_files(MODEL_WEIR_SETTING_PATH)) swmmobject.swmm_open() swmmobject.swmm_start(True) fig = plt.figure() ax = fig.add_subplot(221) ax.set_ylabel('Flow Rate') line, = ax.plot([], [], label='C3') ax.grid() ax.legend() ax2 = fig.add_subplot(223, sharex=ax) ax2.set_ylabel('Setting') line2, = ax2.plot([], [], label='C3') ax2.grid() xdata, ydata = [], [] ydata2 = [] ax3 = fig.add_subplot(2, 2, (2, 4)) arr_lena = read_png(IMAGE_WEIR_SETTING_PATH) imagebox = OffsetImage(arr_lena, zoom=0.45) ab = AnnotationBbox( imagebox, (0.5, 0.5), xybox=(0.5, 0.5), xycoords='data', boxcoords="offset points", pad=0.0, ) ax3.add_artist(ab) ax3.axis('off') def data_gen(t=0): i = 0 while (True): time = swmmobject.swmm_stride(300) i += 1 if i == 80: swmmobject.setLinkSetting('C3', 0.9) elif i == 90: swmmobject.setLinkSetting('C3', 0.8) elif i == 100: swmmobject.setLinkSetting('C3', 0.7) elif i == 110: swmmobject.setLinkSetting('C3', 0.6) elif i == 120: swmmobject.setLinkSetting('C3', 0.5) elif i == 130: swmmobject.setLinkSetting('C3', 0.4) elif i == 140: swmmobject.setLinkSetting('C3', 0.3) elif i == 150: swmmobject.setLinkSetting('C3', 0.2) elif i == 160: swmmobject.setLinkSetting('C3', 0.1) elif i == 170: swmmobject.setLinkSetting('C3', 0.0) elif i == 220: swmmobject.setLinkSetting('C3', 1.0) if i > 0 and time == 0.0: break if i > 0 and time > 0: yield time def run(t): xdata.append(t) new_y = swmmobject.getLinkResult('C3', tka.LinkResults.newFlow) ydata.append(new_y) new_y2 = swmmobject.getLinkResult('C3', tka.LinkResults.setting) ydata2.append(new_y2) xmin, xmax = ax.get_xlim() ymin, ymax = ax.get_ylim() ymin2, ymax2 = ax2.get_ylim() # ax if new_y > ymax: ax.set_ylim(-0.1, 1.5 * ymax) if t >= xmax: ax.set_xlim(xmin, 1.5 * xmax) ax.figure.canvas.draw() line.set_data(xdata, ydata) # ax1 if new_y2 > ymax2: ax2.set_ylim(-0.1, 1.2 * ymax2) line2.set_data(xdata, ydata2) ani = animation.FuncAnimation(fig, run, data_gen, blit=False, repeat=False, save_count=800, interval=10) show_fig = False if show_fig: plt.show() else: movie_path = os.path.join(DATA_PATH, "weir_setting.mp4") print(movie_path, ani) # ani.save(movie_path, fps=20, dpi=170, bitrate=50000) plt.close() swmmobject.swmm_end() swmmobject.swmm_close() print("Check Passed")
def test_plot_node_inflows(): swmmobject = PySWMM(*get_model_files(MODEL_NODE_INFLOWS_PATH)) swmmobject.swmm_open() swmmobject.swmm_start() fig = plt.figure() ax = fig.add_subplot(2, 3, (1, 2)) ax.set_ylabel('Flow Rate') line, = ax.plot([], [], label='C3') ax.grid() ax.legend() ax2 = fig.add_subplot(2, 3, (4, 5), sharex=ax) ax2.set_ylabel('Toolkit API\nInflow') line2, = ax2.plot([], [], label='J1') ax2.grid() xdata, ydata = [], [] ydata2 = [] ax3 = fig.add_subplot(2, 3, (3, 6)) arr_lena = read_png(IMAGE_NODE_INFLOWS_PATH) imagebox = OffsetImage(arr_lena, zoom=0.67) ab = AnnotationBbox( imagebox, (0.5, 0.5), xybox=(0.5, 0.5), xycoords='data', boxcoords="offset points", pad=0.0, ) ax3.add_artist(ab) ax3.axis('off') def data_gen(t=0): i = 0 while (True): if i >= 80: swmmobject.setNodeInflow('J1', random.randint(1, 5)) time = swmmobject.swmm_stride(1500) i += 1 if i > 0 and time == 0.0: break if i > 0 and time > 0: yield time def run(t): xdata.append(t) new_y = swmmobject.getLinkResult('C2', tka.LinkResults.newFlow) ydata.append(new_y) new_y2 = swmmobject.getNodeResult('J1', tka.NodeResults.newLatFlow) ydata2.append(new_y2) xmin, xmax = ax.get_xlim() ymin, ymax = ax.get_ylim() ymin2, ymax2 = ax2.get_ylim() # ax if new_y > ymax: ax.set_ylim(-0.1, 1.5 * ymax) if t >= xmax: ax.set_xlim(xmin, 1.5 * xmax) ax.figure.canvas.draw() line.set_data(xdata, ydata) # ax1 if new_y2 > ymax2: ax2.set_ylim(-0.1, 1.2 * ymax2) line2.set_data(xdata, ydata2) ani = animation.FuncAnimation( fig, run, data_gen, blit=False, repeat=False, save_count=800, interval=10, ) show_figure = False if show_figure: plt.show() else: movie_path = os.path.join(DATA_PATH, "node_inflows.mp4") print(movie_path, ani) # ani.save(movie_path, fps=20, dpi=170, bitrate=50000) plt.close() swmmobject.swmm_end() swmmobject.swmm_report() swmmobject.swmm_close() print("Check Passed")
def test_warning(): swmmobject = PySWMM(*get_model_files(MODEL_WEIR_SETTING_PATH)) swmmobject.swmm_open() swmmobject.getLinkResult('C2', 0) swmmobject.swmm_close()
# Simulation Time Steps episode_count = 198 timesteps = episode_count * 576 epi_start = 0 epi_end = 1 epsilon_value = np.linspace(epi_start, epi_end, timesteps + 10) episode_timer = 0 while episode_timer < episode_count: #run SWMM swmm_model = PySWMM('test.inp', 'test.rpt', 'test.out') swmm_model.swmm_open() swmm_model.swmm_start() n1 = "Tank_LaoDongLu" n2 = "Tank_ChengXi" p1 = "Pump_LaoDongLu1" p2 = "Pump_LaoDongLu2" p3 = "Pump_LaoDongLu3" p4 = "Pump_ChengXi1" p5 = "Pump_ChengXi2" p6 = "Pump_ChengXi3" p7 = "Pump_ChengXi4" node = [n1, n2] pump = [p1, p2, p3, p4, p5, p6, p7] for i in node_controlled:
def test_set_current_time(): swmmobject = PySWMM(*get_model_files(MODEL_WEIR_SETTING_PATH)) swmmobject.swmm_open() print(swmmobject.getSimulationDateTime(0)) print(swmmobject.getSimulationDateTime(0) - timedelta(days=10)) swmmobject.setSimulationDateTime( 0, swmmobject.getSimulationDateTime(0) - timedelta(days=10), ) print(swmmobject.getSimulationDateTime(0)) print(swmmobject.getSimulationDateTime(1)) print(swmmobject.getSimulationDateTime(1) - timedelta(days=10)) swmmobject.setSimulationDateTime( 1, swmmobject.getSimulationDateTime(1) - timedelta(days=10), ) print(swmmobject.getSimulationDateTime(1)) print(swmmobject.getSimulationDateTime(2)) print(swmmobject.getSimulationDateTime(2) - timedelta(days=10)) swmmobject.setSimulationDateTime( 2, swmmobject.getSimulationDateTime(2) - timedelta(days=10), ) print(swmmobject.getSimulationDateTime(2)) swmmobject.swmm_start() i = 0 while (True): time = swmmobject.swmm_stride(600) i += 1 print(swmmobject.getCurrentSimualationTime()) print(type(swmmobject.getCurrentSimualationTime())) if (time <= 0.0): break if i % 144 == 0: print(i) swmmobject.swmm_end() swmmobject.swmm_close()
def test_current_time(): swmmobject = PySWMM(*get_model_files(MODEL_WEIR_SETTING_PATH)) swmmobject.swmm_open() date = swmmobject.getSimulationDateTime(0) print(date) swmmobject.swmm_start() i = 0 while (True): time = swmmobject.swmm_stride(600) print(time) i += 1 current_time = swmmobject.getCurrentSimualationTime() print(current_time) print(type(swmmobject.getCurrentSimualationTime())) if (time <= 0.0): break if i % 144 == 0: print(i) swmmobject.swmm_end() swmmobject.swmm_report() swmmobject.swmm_close()
def test_nodes_1(): ''' pytest pyswmm/tests/test_nodes.py -k `test_nodes_1` ''' swmmobject = PySWMM(MODEL_WEIR_SETTING_PATH) swmmobject.swmm_open() node = Node(swmmobject, "J1") assert(node.invert_elevation == 20.728)
episode_timer = 0 while episode_timer < episode_count: """ i=random.randint(1,3) """ i = 1 sim_num = i #run SWMM rain = rain_generation(i) swmm_model = PySWMM(r'../model/test' + str(i) + '.inp', r'../model/test' + str(i) + '.rpt', r'../model/test' + str(i) + '.out') swmm_model.swmm_open() swmm_model.swmm_start() n1 = "Tank_LaoDongLu" n2 = "Tank_ChengXi" p1 = "Pump_LaoDongLu1" p2 = "Pump_LaoDongLu2" p3 = "Pump_LaoDongLu3" p4 = "Pump_ChengXi1" p5 = "Pump_ChengXi2" p6 = "Pump_ChengXi3" p7 = "Pump_ChengXi4" node = [n1, n2] pump = [p1, p2, p3, p4, p5, p6, p7]
class Simulation(object): """ Base class for a SWMM Simulation. The model object provides several options to run a simulation. Examples: Intialize a simulation and iterate through a simulation. This approach requires some clean up. >>> from pyswmm import Simulation >>> >>> sim = Simulation('./TestModel1_weirSetting.inp') >>> for step in sim: ... pass >>> >>> sim.report() >>> sim.close() Intialize using with statement. This automatically cleans up after a simulation >>> from pyswmm import Simulation >>> >>> with Simulation('./TestModel1_weirSetting.inp') as sim: ... for step in sim: ... pass ... sim.report() Initialize the simulation and execute. This style does not allow the user to interact with the simulation. However, this approach tends to be the fastest. >>> from pyswmm import Simulation >>> >>> sim = Simulation('./TestModel1_weirSetting.inp') >>> sim.execute() """ def __init__(self, inputfile, reportfile=None, outputfile=None): self._model = PySWMM(inputfile, reportfile, outputfile) self._model.swmm_open() self._isOpen = True self._advance_seconds = None self._isStarted = False self._initial_conditions = None def __enter__(self): """ Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... print(sim.current_time) ... sim.report() >>> >>> 2015-11-01 14:00:30 >>> 2015-11-01 14:01:00 >>> 2015-11-01 14:01:30 >>> 2015-11-01 14:02:00 """ return self def __iter__(self): """Iterator over Simulation""" return self def start(self): """Start Simulation""" if not self._isStarted: # Set Model Initial Conditions if self._initial_conditions: self._initial_conditions() self._model.swmm_start(True) self._isStarted = True def __next__(self): """Next""" # Start Simulation self.start() # Simulation Step Amount if self._advance_seconds is None: time = self._model.swmm_step() else: time = self._model.swmm_stride(self._advance_seconds) if time <= 0.0: self._model.swmm_end() self._isStarted = False raise StopIteration return self._model next = __next__ # Python 2 def __exit__(self, *a): """close""" if self._isStarted: self._model.swmm_end() self._isStarted = False if self._isOpen: self._model.swmm_close() self._isOpen = False def initial_conditions(self, init_conditions): """ Intial Conditions for Hydraulics and Hydrology can be set from within the api by setting a function to the initial_conditions property. >>> from pyswmm import Simulation >>> >>> with Simulation('./TestModel1_weirSetting.inp') as sim: ... nodeJ1 = Nodes(sim)["J1"] ... ... def init_conditions(): ... nodeJ1.initial_depth = 4 ... ... sim.initial_conditions(init_conditions) ... ... for step in sim: ... pass ... sim.report() """ if hasattr(init_conditions, '__call__'): self._initial_conditions = init_conditions else: error_msg = 'Requires Type Function, not {}'.format( type(init_conditions)) raise (PYSWMMException(error_msg)) def step_advance(self, advance_seconds): """ Advances the model by X number of seconds instead of intervening at every routing step. This does not change the routing step for the simulation; only lets python take back control after each advance period. :param int advance_seconds: Seconds to Advance simulation Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... sim.step_advance(300) ... for step in sim: ... print(step.current_time) ... # or here! sim.step_advance(newvalue) ... sim.report() >>> >>> 2015-11-01 14:00:30 >>> 2015-11-01 14:01:00 >>> 2015-11-01 14:01:30 >>> 2015-11-01 14:02:00 """ self._advance_seconds = advance_seconds def report(self): """ Writes to report file after simulation. Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... pass ... sim.report() """ self._model.swmm_report() def close(self): """ Intialize a simulation and iterate through a simulation. This approach requires some clean up. Examples: >>> from pyswmm import Simulation >>> >>> sim = Simulation('./TestModel1_weirSetting.inp') >>> for step in sim: ... pass >>> >>> sim.report() >>> sim.close() """ self.__exit__() def execute(self): """ Open an input file, run SWMM, then close the file. Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.execute() """ self._model.swmmExec() @property def engine_version(self): """ Retrieves the SWMM Engine Version. :return: Engine Version :rtype: StrictVersion Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.engine_version StrictVersion("5.1.13") """ return self._model.swmm_getVersion() @property def runoff_error(self): """ Retrieves the Runoff Mass Balance Error. :return: Runoff Mass Balance Error :rtype: float Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.execute() >>> sim.runoff_error 0.01 """ return self._model.swmm_getMassBalErr()[0] @property def flow_routing_error(self): """ Retrieves the Flow Routing Mass Balance Error. :return: Flow Routing Mass Balance Error :rtype: float Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.execute() >>> sim.flow_routing_error 0.01 """ return self._model.swmm_getMassBalErr()[1] @property def quality_error(self): """ Retrieves the Quality Routing Mass Balance Error. :return: Quality Routing Mass Balance Error :rtype: float Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.execute() >>> sim.quality_error 0.01 """ return self._model.swmm_getMassBalErr()[2] @property def start_time(self): """Get/set Simulation start time. Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... print sim.start_time ... sim.start_time = datetime(2015,5,10,15,15,1) >>> >>> datetime.datetime(2015,5,10,15,15,1) """ return self._model.getSimulationDateTime( SimulationTime.StartDateTime.value) @start_time.setter def start_time(self, dtimeval): """Set simulation Start time""" self._model.setSimulationDateTime(SimulationTime.StartDateTime.value, dtimeval) @property def end_time(self): """Get/set Simulation end time. Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... print sim.end_time ... sim.end_time = datetime(2016,5,10,15,15,1) >>> >>> datetime.datetime(2016,5,10,15,15,1) """ return self._model.getSimulationDateTime( SimulationTime.EndDateTime.value) @end_time.setter def end_time(self, dtimeval): """Set simulation End time.""" self._model.setSimulationDateTime(SimulationTime.EndDateTime.value, dtimeval) @property def report_start(self): """Get/set Simulation report start time. Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... print sim.report_start ... sim.report_start = datetime(2015,5,10,15,15,1) >>> >>> datetime.datetime(2015,5,10,15,15,1) """ return self._model.getSimulationDateTime( SimulationTime.ReportStart.value) @report_start.setter def report_start(self, dtimeval): """Set simulation report start time.""" self._model.setSimulationDateTime(SimulationTime.ReportStart.value, dtimeval) @property def flow_units(self): """ Get Simulation Units (CFS, GPM, MGD, CMS, LPS, MLD). :return: Flow Unit :rtype: str Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... print sim.flow_units >>> >>> CFS """ return self._model.getSimUnit(SimulationUnits.FlowUnits.value) @property def system_units(self): """Get system units (US, SI). :return: System Unit :rtype: str Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... print sim.system_units >>> >>> US """ return self._model.getSimUnit(SimulationUnits.UnitSystem.value) @property def current_time(self): """Get Simulation Current Time. :return: Current Simulation Time :rtype: Datetime Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... print(sim.current_time) ... sim.report() >>> >>> 2015-11-01 14:00:30 >>> 2015-11-01 14:01:00 >>> 2015-11-01 14:01:30 >>> 2015-11-01 14:02:00 """ return self._model.getCurrentSimulationTime() @property def percent_complete(self): """Get Simulation Percent Complete. :return: Current Percent Complete :rtype: Datetime Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('../test/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... print(sim.percent_complete) ... sim.report() >>> >>> 0.01 >>> 0.25 >>> 0.50 >>> 0.75 """ dt = self.current_time - self.start_time total_time = self.end_time - self.start_time return float(dt.total_seconds()) / total_time.total_seconds()
def test_nodes_4(): swmmobject = PySWMM(*get_model_files(MODEL_WEIR_SETTING_PATH)) swmmobject.swmm_open() print(swmmobject.getNodeParam('J2', tka.NodeParams.invertElev)) swmmobject.setNodeParam('J2', tka.NodeParams.invertElev, 19) print(swmmobject.getNodeParam('J2', tka.NodeParams.invertElev)) print(swmmobject.getLinkParam('C1:C2', tka.LinkParams.offset1)) swmmobject.setLinkParam('C1:C2', tka.LinkParams.offset1, 19) print(swmmobject.getLinkParam('C1:C2', tka.LinkParams.offset1)) print(swmmobject.getSubcatchParam('S2', tka.SubcParams.area)) swmmobject.setSubcatchParam('S2', tka.SubcParams.area, 19) print(swmmobject.getSubcatchParam('S2', tka.SubcParams.area)) swmmobject.swmm_start() i = 0 while (True): time = swmmobject.swmm_stride(600) i += 1 if (time <= 0.0): break if i % 144 == 0: print(i) swmmobject.setLinkParam('C1:C2', tka.LinkParams.qLimit, 0.25) swmmobject.swmm_end() swmmobject.swmm_close() print("swmm_step() Check Passed")
class Simulation(object): """ Base class for a SWMM Simulation. The model object provides several options to run a simulation. User can specified SWMM library path. Uses default lib if not provided. Initialize the Simulation class. :param str inpfile: Name of SWMM input file (default '') :param str rptfile: Report file to generate (default None) :param str binfile: Optional binary output file (default None) :param str swmm_lib_path: User-specified SWMM library path (default None). Examples: Intialize a simulation and iterate through a simulation. This approach requires some clean up. >>> from pyswmm import Simulation >>> >>> sim = Simulation('tests/data/TestModel1_weirSetting.inp') >>> for step in sim: ... pass >>> >>> sim.report() >>> sim.close() Intialize using with statement. This automatically cleans up after a simulation >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... pass ... sim.report() Initialize the simulation and execute. This style does not allow the user to interact with the simulation. However, this approach tends to be the fastest. >>> from pyswmm import Simulation >>> >>> sim = Simulation('tests/data/TestModel1_weirSetting.inp') >>> sim.execute() """ def __init__(self, inputfile, reportfile=None, outputfile=None, swmm_lib_path=None): self._model = PySWMM(inputfile, reportfile, outputfile, swmm_lib_path) self._model.swmm_open() self._isOpen = True self._advance_seconds = None self._isStarted = False self._callbacks = { "before_start": None, "before_step": None, "after_step": None, "before_end": None, "after_end": None, "after_close": None } def __enter__(self): """ Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... print(sim.current_time) ... sim.report() >>> >>> 2015-11-01 14:00:30 >>> 2015-11-01 14:01:00 >>> 2015-11-01 14:01:30 >>> 2015-11-01 14:02:00 """ return self def __iter__(self): """Iterator over Simulation""" return self def start(self): """Start Simulation""" if not self._isStarted: # Set Model Initial Conditions # (This Will be Deprecated with Time) if hasattr(self, "_initial_conditions"): self._initial_conditions() # Execute Callback Hooks Before Simulation self._execute_callback(self.before_start()) self._model.swmm_start(True) self._isStarted = True def __next__(self): """Next""" # Start Simulation self.start() # Execute Callback Hooks Before Simulation Step self._execute_callback(self.before_step()) # Simulation Step Amount if self._advance_seconds is None: time = self._model.swmm_step() else: time = self._model.swmm_stride(self._advance_seconds) # Execute Callback Hooks After Simulation Step self._execute_callback(self.after_step()) if time <= 0.0: self._execute_callback(self.before_end()) raise StopIteration return self._model next = __next__ # Python 2 def __exit__(self, *a): """close""" if self._isStarted: self._model.swmm_end() self._isStarted = False # Execute Callback Hooks After Simulation End self._execute_callback(self.after_end()) if self._isOpen: self._model.swmm_close() self._isOpen = False # Execute Callback Hooks After Simulation Closes self._execute_callback(self.after_close()) @staticmethod def _is_callback(callable_object): """Checks if arugment is a function/method.""" if not callable(callable_object): error_msg = 'Requires Callable Object, not {}'.format( type(callable_object)) raise (PYSWMMException(error_msg)) else: return True def _execute_callback(self, callback): """Runs the callback.""" if callback: try: callback() except PYSWMMException: error_msg = "Callback Failed" raise PYSWMMException((error_msg)) def initial_conditions(self, init_conditions): """ Intial Conditions for Hydraulics and Hydrology can be set from within the api by setting a function to the initial_conditions property. >>> from pyswmm import Simulation >>> >>> with Simulation('./TestModel1_weirSetting.inp') as sim: ... nodeJ1 = Nodes(sim)["J1"] ... ... def init_conditions(): ... nodeJ1.initial_depth = 4 ... ... sim.initial_conditions(init_conditions) ... ... for step in sim: ... pass ... sim.report() """ if self._is_callback(init_conditions): self._initial_conditions = init_conditions def before_start(self): """Get Before Start Callback. :return: Callbacks """ return self._callbacks["before_start"] def add_before_start(self, callback): """ Add callback function/method/object to execute before the simlation starts. Needs to be callable. :param func callback: Callable Object >>> from pyswmm import Simulation >>> >>> def test_callback(): ... print("CALLBACK - Executed") >>> >>> with Simulation('./TestModel1_weirSetting.inp') as sim: ... ... sim.before_start(test_callback) #<- pass function handle. ... print("Waiting to Start") ... for ind, step in enumerate(sim): ... print("Step {}".format(ind)) ... print("Complete!") ... print("Closed") >>> >>> "Waiting to Start" >>> "CALLBACK - Executed" >>> "Step 0" >>> "Step 1" >>> ... >>> "Complete!" >>> "Closed" """ if self._is_callback(callback): self._callbacks["before_start"] = callback def before_step(self): """Get Before Step Callback. :return: Callbacks """ return self._callbacks["before_step"] def add_before_step(self, callback): """ Add callback function/method/object to execute before a simlation step. Needs to be callable. :param func callback: Callable Object (See self.add_before_start() for more details) """ if self._is_callback(callback): self._callbacks["before_step"] = callback def after_step(self): """Get After Step Callback. :return: Callbacks """ return self._callbacks["after_step"] def add_after_step(self, callback): """ Add callback function/method/object to execute after a simlation step. Needs to be callable. :param func callback: Callable Object (See self.add_before_start() for more details) """ if self._is_callback(callback): self._callbacks["after_step"] = callback def before_end(self): """Get Before End Callback. :return: Callbacks """ return self._callbacks["before_end"] def add_before_end(self, callback): """ Add callback function/method/object to execute after the simulation ends. Needs to be callable. :param func callback: Callable Object (See self.add_before_start() for more details) """ if self._is_callback(callback): self._callbacks["before_end"] = callback def after_end(self): """Get After End Callback. :return: Callbacks """ return self._callbacks["after_end"] def add_after_end(self, callback): """ Add callback function/method/object to execute after the simulation ends. Needs to be callable. :param func callback: Callable Object (See self.add_before_start() for more details) """ if self._is_callback(callback): self._callbacks["after_end"] = callback def after_close(self): """Get After Close Callback. :return: Callbacks """ return self._callbacks["after_close"] def add_after_close(self, callback): """ Add callback function/method/object to execute after the simulation is closed. Needs to be callable. :param func callback: Callable Object (See self.add_before_start() for more details) """ if self._is_callback(callback): self._callbacks["after_close"] = callback def step_advance(self, advance_seconds): """ Advances the model by X number of seconds instead of intervening at every routing step. This does not change the routing step for the simulation; only lets python take back control after each advance period. :param int advance_seconds: Seconds to Advance simulation Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... sim.step_advance(300) ... for step in sim: ... print(step.current_time) ... # or here! sim.step_advance(newvalue) ... sim.report() >>> >>> 2015-11-01 14:00:30 >>> 2015-11-01 14:01:00 >>> 2015-11-01 14:01:30 >>> 2015-11-01 14:02:00 """ self._advance_seconds = advance_seconds def report(self): """ Writes to report file after simulation. Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... pass ... sim.report() """ self._model.swmm_report() def close(self): """ Intialize a simulation and iterate through a simulation. This approach requires some clean up. Examples: >>> from pyswmm import Simulation >>> >>> sim = Simulation('./TestModel1_weirSetting.inp') >>> for step in sim: ... pass >>> >>> sim.report() >>> sim.close() """ self.__exit__() def execute(self): """ Open an input file, run SWMM, then close the file. Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.execute() """ self._model.swmmExec() @property def engine_version(self): """ Retrieves the SWMM Engine Version. :return: Engine Version :rtype: StrictVersion Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.engine_version StrictVersion("5.1.13") """ return self._model.swmm_getVersion() @property def runoff_error(self): """ Retrieves the Runoff Mass Balance Error. :return: Runoff Mass Balance Error :rtype: float Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.execute() >>> sim.runoff_error 0.01 """ return self._model.swmm_getMassBalErr()[0] @property def flow_routing_error(self): """ Retrieves the Flow Routing Mass Balance Error. :return: Flow Routing Mass Balance Error :rtype: float Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.execute() >>> sim.flow_routing_error 0.01 """ return self._model.swmm_getMassBalErr()[1] @property def quality_error(self): """ Retrieves the Quality Routing Mass Balance Error. :return: Quality Routing Mass Balance Error :rtype: float Examples: >>> sim = PYSWMM(r'\\test.inp') >>> sim.execute() >>> sim.quality_error 0.01 """ return self._model.swmm_getMassBalErr()[2] @property def start_time(self): """Get/set Simulation start time. Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... print sim.start_time ... sim.start_time = datetime(2015,5,10,15,15,1) >>> >>> datetime.datetime(2015,5,10,15,15,1) """ return self._model.getSimulationDateTime( SimulationTime.StartDateTime.value) @start_time.setter def start_time(self, dtimeval): """Set simulation Start time""" self._model.setSimulationDateTime(SimulationTime.StartDateTime.value, dtimeval) @property def end_time(self): """Get/set Simulation end time. Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... print sim.end_time ... sim.end_time = datetime(2016,5,10,15,15,1) >>> >>> datetime.datetime(2016,5,10,15,15,1) """ return self._model.getSimulationDateTime( SimulationTime.EndDateTime.value) @end_time.setter def end_time(self, dtimeval): """Set simulation End time.""" self._model.setSimulationDateTime(SimulationTime.EndDateTime.value, dtimeval) @property def report_start(self): """Get/set Simulation report start time. Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... print sim.report_start ... sim.report_start = datetime(2015,5,10,15,15,1) >>> >>> datetime.datetime(2015,5,10,15,15,1) """ return self._model.getSimulationDateTime( SimulationTime.ReportStart.value) @report_start.setter def report_start(self, dtimeval): """Set simulation report start time.""" self._model.setSimulationDateTime(SimulationTime.ReportStart.value, dtimeval) @property def flow_units(self): """ Get Simulation Units (CFS, GPM, MGD, CMS, LPS, MLD). :return: Flow Unit :rtype: str Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... print sim.flow_units >>> >>> CFS """ return self._model.getSimUnit(SimulationUnits.FlowUnits.value) @property def system_units(self): """Get system units (US, SI). :return: System Unit :rtype: str Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... print sim.system_units >>> >>> US """ return self._model.getSimUnit(SimulationUnits.UnitSystem.value) @property def current_time(self): """Get Simulation Current Time. :return: Current Simulation Time :rtype: Datetime Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... print(sim.current_time) ... sim.report() >>> >>> 2015-11-01 14:00:30 >>> 2015-11-01 14:01:00 >>> 2015-11-01 14:01:30 >>> 2015-11-01 14:02:00 """ return self._model.getCurrentSimulationTime() @property def percent_complete(self): """Get Simulation Percent Complete. :return: Current Percent Complete :rtype: Datetime Examples: >>> from pyswmm import Simulation >>> >>> with Simulation('tests/data/TestModel1_weirSetting.inp') as sim: ... for step in sim: ... print(sim.percent_complete) ... sim.report() >>> >>> 0.01 >>> 0.25 >>> 0.50 >>> 0.75 """ dt = self.current_time - self.start_time total_time = self.end_time - self.start_time return float(dt.total_seconds()) / total_time.total_seconds()