18
18
from itertools import islice
19
19
20
20
# project import
21
- from swmm .toolkit import output , output_enum
21
+ from swmm .toolkit import output , shared_enum
22
22
23
23
24
24
def output_generator (path_ref ):
@@ -40,7 +40,7 @@ def output_generator(path_ref):
40
40
with OutputReader (path_ref ) as reader :
41
41
42
42
for period_index in range (0 , reader .report_periods ()):
43
- for element_type in islice (output_enum .ElementType , 4 ):
43
+ for element_type in islice (shared_enum .ElementType , 4 ):
44
44
for element_index in range (0 , reader .element_count (element_type )):
45
45
46
46
yield (reader .element_result (element_type , period_index , element_index ),
@@ -57,10 +57,10 @@ def __init__(self, filename):
57
57
self .handle = None
58
58
self .count = None
59
59
self .get_element_result = {
60
- output_enum .ElementType .SUBCATCH : output .get_subcatch_result ,
61
- output_enum .ElementType .NODE : output .get_node_result ,
62
- output_enum .ElementType .LINK : output .get_link_result ,
63
- output_enum .ElementType .SYSTEM : output .get_system_result
60
+ shared_enum .ElementType .SUBCATCH : output .get_subcatch_result ,
61
+ shared_enum .ElementType .NODE : output .get_node_result ,
62
+ shared_enum .ElementType .LINK : output .get_link_result ,
63
+ shared_enum .ElementType .SYSTEM : output .get_system_result
64
64
}
65
65
66
66
def __enter__ (self ):
@@ -73,7 +73,7 @@ def __exit__(self, type, value, traceback):
73
73
output .close (self .handle )
74
74
75
75
def report_periods (self ):
76
- return output .get_times (self .handle , output_enum .Time .NUM_PERIODS )
76
+ return output .get_times (self .handle , shared_enum .Time .NUM_PERIODS )
77
77
78
78
def element_count (self , element_type ):
79
79
return self .count [element_type ]
0 commit comments