Allow all kinds of string to be processed in open/4

This commit is contained in:
Adrián Arroyo Calle
2021-11-28 12:37:16 +01:00
parent ca28f24e42
commit 86161ccf2a
2 changed files with 8 additions and 25 deletions

View File

@@ -1475,7 +1475,11 @@ open(SourceSink, Mode, Stream, StreamOptions) :-
( SourceSink = stream(S0) ->
'$set_stream_options'(S0, Alias, EOFAction, Reposition, Type),
Stream = S0
; '$open'(SourceSink, Mode, Stream, Alias, EOFAction, Reposition, Type)
; (
atom(SourceSink) ->
atom_chars(SourceSink, SourceSinkString)
; SourceSink = SourceSinkString
), '$open'(SourceSinkString, Mode, Stream, Alias, EOFAction, Reposition, Type)
)
).

View File

@@ -3169,31 +3169,10 @@ impl MachineState {
let options = self.to_stream_options(alias, eof_action, reposition, stream_type);
let mut stream = match self.store(self.deref(self[temp_v!(1)])) {
Addr::Con(h) if self.heap.atom_at(h) => match &self.heap[h] {
&HeapCellValue::Atom(ref atom, _) => {
self.stream_from_file_spec(atom.clone(), indices, &options)?
}
_ => {
unreachable!()
}
},
Addr::Char(c) => {
let atom = clause_name!(c.to_string(), self.atom_tbl);
self.stream_from_file_spec(atom, indices, &options)?
}
Addr::PStrLocation(h, n) => match &self.heap[h] {
&HeapCellValue::PartialString(_, _has_tail @ false) => {
let mut heap_pstr_iter = self.heap_pstr_iter(Addr::PStrLocation(h, n));
let mut iter = self.heap_pstr_iter(self[temp_v!(1)]);
let file_spec = clause_name!(iter.to_string(), self.atom_tbl);
let file_spec = clause_name!(heap_pstr_iter.to_string(), self.atom_tbl);
self.stream_from_file_spec(file_spec, indices, &options)?
}
_ => self.stream_from_file_spec(clause_name!(""), indices, &options)?,
},
_ => self.stream_from_file_spec(clause_name!(""), indices, &options)?,
};
let mut stream = self.stream_from_file_spec(file_spec, indices, &options)?;
*stream.options_mut() = options;