31
31
from ansys .dyna .core .lib .format_type import format_type
32
32
from ansys .dyna .core .lib .io_utils import write_or_return
33
33
from ansys .dyna .core .lib .kwd_line_formatter import buffer_to_lines
34
+ from ansys .dyna .core .lib .parameter_set import ParameterSet
34
35
35
36
CHECK_TYPE = True
36
37
@@ -138,7 +139,9 @@ def _get_read_options(self):
138
139
options = {"names" : names , "colspecs" : colspecs , "dtype" : dtype , "comment" : "$" }
139
140
return options
140
141
141
- def _read_buffer_as_dataframe (self , buffer : typing .TextIO , fields : typing .Iterable [Field ]) -> pd .DataFrame :
142
+ def _read_buffer_as_dataframe (
143
+ self , buffer : typing .TextIO , fields : typing .Iterable [Field ], parameter_set : ParameterSet
144
+ ) -> pd .DataFrame :
142
145
read_options = self ._get_read_options ()
143
146
df = pd .read_fwf (buffer , ** read_options )
144
147
return df
@@ -149,32 +152,32 @@ def _get_fields(self) -> typing.List[Field]:
149
152
fields = self ._convert_fields_to_long_format ()
150
153
return fields
151
154
152
- def _load_bounded_from_buffer (self , buf : typing .TextIO ) -> None :
155
+ def _load_bounded_from_buffer (self , buf : typing .TextIO , parameter_set : ParameterSet ) -> None :
153
156
read_options = self ._get_read_options ()
154
157
read_options ["nrows" ] = self ._num_rows ()
155
158
df = pd .read_fwf (buf , ** read_options )
156
159
self ._table = df
157
160
self ._initialized = True
158
161
159
- def _load_unbounded_from_buffer (self , buf : typing .TextIO ) -> None :
162
+ def _load_unbounded_from_buffer (self , buf : typing .TextIO , parameter_set : ParameterSet ) -> None :
160
163
data_lines = buffer_to_lines (buf )
161
- self ._load_lines (data_lines )
164
+ self ._load_lines (data_lines , parameter_set )
162
165
163
- def read (self , buf : typing .TextIO ) -> None :
166
+ def read (self , buf : typing .TextIO , parameter_set : ParameterSet = None ) -> None :
164
167
if self .bounded :
165
168
self ._initialized = True
166
- self ._load_bounded_from_buffer (buf )
169
+ self ._load_bounded_from_buffer (buf , parameter_set )
167
170
else :
168
171
self ._initialize_data (0 )
169
172
self ._initialized = True
170
- self ._load_unbounded_from_buffer (buf )
173
+ self ._load_unbounded_from_buffer (buf , parameter_set )
171
174
172
- def _load_lines (self , data_lines : typing .List [str ]) -> None :
175
+ def _load_lines (self , data_lines : typing .List [str ], parameter_set : ParameterSet ) -> None :
173
176
fields = self ._get_fields ()
174
177
buffer = io .StringIO ()
175
178
[(buffer .write (line ), buffer .write ("\n " )) for line in data_lines ]
176
179
buffer .seek (0 )
177
- self ._table = self ._read_buffer_as_dataframe (buffer , fields )
180
+ self ._table = self ._read_buffer_as_dataframe (buffer , fields , parameter_set )
178
181
self ._initialized = True
179
182
180
183
def write (
0 commit comments