|
1
|
1 // file-io.cc -*- C++ -*- |
|
|
2 /* |
|
|
3 |
|
332
|
4 Copyright (C) 1993, 1994 John W. Eaton |
|
1
|
5 |
|
|
6 This file is part of Octave. |
|
|
7 |
|
|
8 Octave is free software; you can redistribute it and/or modify it |
|
|
9 under the terms of the GNU General Public License as published by the |
|
|
10 Free Software Foundation; either version 2, or (at your option) any |
|
|
11 later version. |
|
|
12 |
|
|
13 Octave is distributed in the hope that it will be useful, but WITHOUT |
|
|
14 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
|
|
15 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
|
|
16 for more details. |
|
|
17 |
|
|
18 You should have received a copy of the GNU General Public License |
|
|
19 along with Octave; see the file COPYING. If not, write to the Free |
|
|
20 Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. |
|
|
21 |
|
|
22 */ |
|
|
23 |
|
|
24 // Written by John C. Campbell <jcc@che.utexas.edu>. |
|
|
25 |
|
240
|
26 #ifdef HAVE_CONFIG_H |
|
|
27 #include "config.h" |
|
1
|
28 #endif |
|
|
29 |
|
|
30 #include <DLList.h> |
|
|
31 #include <unistd.h> |
|
|
32 #include <string.h> |
|
|
33 #include <stdio.h> |
|
444
|
34 #include <errno.h> |
|
1
|
35 #include <stdlib.h> |
|
|
36 #include <strstream.h> |
|
|
37 #include <ctype.h> |
|
|
38 |
|
456
|
39 #include "dMatrix.h" |
|
444
|
40 |
|
1
|
41 #include "statdefs.h" |
|
|
42 #include "file-io.h" |
|
|
43 #include "input.h" |
|
|
44 #include "octave-hist.h" |
|
|
45 #include "tree-const.h" |
|
|
46 #include "error.h" |
|
542
|
47 #include "help.h" |
|
1
|
48 #include "utils.h" |
|
|
49 #include "pager.h" |
|
529
|
50 #include "defun.h" |
|
444
|
51 #include "sysdep.h" |
|
|
52 #include "mappers.h" |
|
529
|
53 #include "variables.h" |
|
1
|
54 |
|
|
55 // keeps a count of how many files are open and in the file list |
|
|
56 static int file_count = 0; |
|
|
57 |
|
|
58 // keeps a count of args sent to printf or scanf |
|
|
59 static int fmt_arg_count = 0; |
|
|
60 |
|
240
|
61 class file_info |
|
1
|
62 { |
|
|
63 public: |
|
240
|
64 file_info (void); |
|
|
65 file_info (int num, const char *nm, FILE *t, const char *md); |
|
|
66 file_info (const file_info& f); |
|
1
|
67 |
|
240
|
68 file_info& operator = (const file_info& f); |
|
1
|
69 |
|
240
|
70 ~file_info (void); |
|
1
|
71 |
|
|
72 int number (void) const; |
|
240
|
73 const char *name (void) const; |
|
1
|
74 FILE *fptr (void) const; |
|
240
|
75 const char *mode (void) const; |
|
1
|
76 |
|
444
|
77 int eof (void) const; |
|
|
78 int error (void) const; |
|
|
79 |
|
1
|
80 private: |
|
334
|
81 int file_number; |
|
|
82 char *file_name; |
|
|
83 FILE *file_fptr; |
|
|
84 char *file_mode; |
|
1
|
85 }; |
|
|
86 |
|
240
|
87 file_info::file_info (void) |
|
1
|
88 { |
|
334
|
89 file_number = -1; |
|
529
|
90 file_name = 0; |
|
|
91 file_fptr = 0; |
|
|
92 file_mode = 0; |
|
1
|
93 } |
|
|
94 |
|
240
|
95 file_info::file_info (int n, const char *nm, FILE *t, const char *md) |
|
|
96 { |
|
334
|
97 file_number = n; |
|
|
98 file_name = strsave (nm); |
|
|
99 file_fptr = t; |
|
|
100 file_mode = strsave (md); |
|
240
|
101 } |
|
|
102 |
|
|
103 file_info::file_info (const file_info& f) |
|
1
|
104 { |
|
367
|
105 file_number = f.file_number; |
|
|
106 file_name = strsave (f.file_name); |
|
|
107 file_fptr = f.file_fptr; |
|
|
108 file_mode = strsave (f.file_mode); |
|
1
|
109 } |
|
|
110 |
|
240
|
111 file_info& |
|
|
112 file_info::operator = (const file_info& f) |
|
1
|
113 { |
|
240
|
114 if (this != & f) |
|
|
115 { |
|
367
|
116 file_number = f.file_number; |
|
334
|
117 delete [] file_name; |
|
|
118 file_name = strsave (f.file_name); |
|
|
119 file_fptr = f.file_fptr; |
|
|
120 delete [] file_mode; |
|
|
121 file_mode = strsave (f.file_mode); |
|
240
|
122 } |
|
1
|
123 return *this; |
|
|
124 } |
|
|
125 |
|
240
|
126 file_info::~file_info (void) |
|
1
|
127 { |
|
334
|
128 delete [] file_name; |
|
|
129 delete [] file_mode; |
|
1
|
130 } |
|
|
131 |
|
|
132 int |
|
240
|
133 file_info::number (void) const |
|
1
|
134 { |
|
334
|
135 return file_number; |
|
1
|
136 } |
|
|
137 |
|
240
|
138 const char * |
|
|
139 file_info::name (void) const |
|
1
|
140 { |
|
334
|
141 return file_name; |
|
1
|
142 } |
|
|
143 |
|
|
144 FILE * |
|
240
|
145 file_info::fptr (void) const |
|
1
|
146 { |
|
334
|
147 return file_fptr; |
|
1
|
148 } |
|
|
149 |
|
240
|
150 const char * |
|
|
151 file_info::mode (void) const |
|
1
|
152 { |
|
334
|
153 return file_mode; |
|
1
|
154 } |
|
|
155 |
|
|
156 // double linked list containing relevant information about open files |
|
240
|
157 static DLList <file_info> file_list; |
|
1
|
158 |
|
|
159 void |
|
164
|
160 initialize_file_io (void) |
|
1
|
161 { |
|
334
|
162 file_info octave_stdin (0, "stdin", stdin, "r"); |
|
|
163 file_info octave_stdout (1, "stdout", stdout, "w"); |
|
|
164 file_info octave_stderr (2, "stderr", stderr, "w"); |
|
1
|
165 |
|
334
|
166 file_list.append (octave_stdin); |
|
|
167 file_list.append (octave_stdout); |
|
|
168 file_list.append (octave_stderr); |
|
1
|
169 |
|
|
170 file_count = 3; |
|
|
171 } |
|
|
172 |
|
444
|
173 /* |
|
|
174 * Given a file name or number, return a pointer to the corresponding |
|
|
175 * open file. If the file has not already been opened, return NULL. |
|
|
176 */ |
|
1
|
177 Pix |
|
164
|
178 return_valid_file (const tree_constant& arg) |
|
1
|
179 { |
|
|
180 if (arg.is_string_type ()) |
|
|
181 { |
|
|
182 Pix p = file_list.first (); |
|
240
|
183 file_info file; |
|
1
|
184 for (int i = 0; i < file_count; i++) |
|
|
185 { |
|
|
186 char *file_name = arg.string_value (); |
|
|
187 file = file_list (p); |
|
|
188 if (strcmp (file.name (), file_name) == 0) |
|
|
189 return p; |
|
|
190 file_list.next (p); |
|
|
191 } |
|
|
192 } |
|
|
193 else if (arg.is_scalar_type ()) |
|
|
194 { |
|
|
195 double file_num = arg.double_value (); |
|
|
196 if ((double) NINT (file_num) != file_num) |
|
|
197 error ("file number not an integer value"); |
|
|
198 else |
|
|
199 { |
|
|
200 Pix p = file_list.first (); |
|
240
|
201 file_info file; |
|
1
|
202 for (int i = 0; i < file_count; i++) |
|
|
203 { |
|
|
204 file = file_list (p); |
|
|
205 if (file.number () == file_num) |
|
|
206 return p; |
|
|
207 file_list.next (p); |
|
|
208 } |
|
|
209 error ("no file with that number"); |
|
|
210 } |
|
|
211 } |
|
|
212 else |
|
|
213 error ("inapproriate file specifier"); |
|
|
214 |
|
529
|
215 return 0; |
|
1
|
216 } |
|
|
217 |
|
|
218 static Pix |
|
370
|
219 fopen_file_for_user (const tree_constant& arg, const char *mode, |
|
|
220 const char *warn_for) |
|
1
|
221 { |
|
|
222 char *file_name = arg.string_value (); |
|
|
223 |
|
|
224 FILE *file_ptr = fopen (file_name, mode); |
|
529
|
225 if (file_ptr) |
|
1
|
226 { |
|
240
|
227 file_info file (++file_count, file_name, file_ptr, mode); |
|
1
|
228 file_list.append (file); |
|
|
229 |
|
|
230 Pix p = file_list.first (); |
|
240
|
231 file_info file_from_list; |
|
1
|
232 |
|
|
233 for (int i = 0; i < file_count; i++) |
|
|
234 { |
|
|
235 file_from_list = file_list (p); |
|
|
236 if (strcmp (file_from_list.name (), file_name) == 0) |
|
|
237 return p; |
|
|
238 file_list.next (p); |
|
|
239 } |
|
|
240 } |
|
|
241 |
|
370
|
242 error ("%s: unable to open file `%s'", warn_for, file_name); |
|
240
|
243 |
|
529
|
244 return 0; |
|
1
|
245 } |
|
|
246 |
|
368
|
247 static Pix |
|
|
248 file_io_get_file (const tree_constant arg, const char *mode, |
|
|
249 const char *warn_for) |
|
|
250 { |
|
|
251 Pix p = return_valid_file (arg); |
|
|
252 |
|
529
|
253 if (! p) |
|
368
|
254 { |
|
|
255 if (arg.is_string_type ()) |
|
|
256 { |
|
|
257 char *name = arg.string_value (); |
|
|
258 |
|
|
259 struct stat buffer; |
|
|
260 int status = stat (name, &buffer); |
|
|
261 |
|
372
|
262 if (status == 0) |
|
368
|
263 { |
|
|
264 if ((buffer.st_mode & S_IFREG) == S_IFREG) |
|
370
|
265 p = fopen_file_for_user (arg, mode, warn_for); |
|
368
|
266 else |
|
|
267 error ("%s: invalid file type", warn_for); |
|
|
268 } |
|
372
|
269 else if (status < 0 && *mode != 'r') |
|
|
270 p = fopen_file_for_user (arg, mode, warn_for); |
|
368
|
271 else |
|
|
272 error ("%s: can't stat file `%s'", warn_for, name); |
|
|
273 } |
|
|
274 else |
|
|
275 error ("%s: invalid file specifier", warn_for); |
|
|
276 } |
|
|
277 |
|
|
278 return p; |
|
|
279 } |
|
1
|
280 |
|
529
|
281 DEFUN ("fclose", Ffclose, Sfclose, 2, 1, |
|
|
282 "fclose (FILENAME or FILENUM): close a file") |
|
|
283 { |
|
|
284 Octave_object retval; |
|
|
285 |
|
|
286 int nargin = args.length (); |
|
|
287 |
|
|
288 if (nargin != 2) |
|
|
289 print_usage ("fclose"); |
|
|
290 else |
|
|
291 retval = fclose_internal (args); |
|
|
292 |
|
|
293 return retval; |
|
|
294 } |
|
|
295 |
|
497
|
296 Octave_object |
|
|
297 fclose_internal (const Octave_object& args) |
|
1
|
298 { |
|
497
|
299 Octave_object retval; |
|
1
|
300 |
|
497
|
301 Pix p = return_valid_file (args(1)); |
|
1
|
302 |
|
529
|
303 if (! p) |
|
1
|
304 return retval; |
|
|
305 |
|
240
|
306 file_info file = file_list (p); |
|
1
|
307 |
|
|
308 if (file.number () < 3) |
|
|
309 { |
|
|
310 warning ("fclose: can't close stdin, stdout, or stderr!"); |
|
|
311 return retval; |
|
|
312 } |
|
|
313 |
|
|
314 int success = fclose (file.fptr ()); |
|
|
315 file_list.del (p); |
|
|
316 file_count--; |
|
|
317 |
|
497
|
318 retval.resize (1); |
|
1
|
319 if (success == 0) |
|
497
|
320 retval(0) = tree_constant (1.0); // succeeded |
|
1
|
321 else |
|
|
322 { |
|
|
323 error ("fclose: error on closing file"); |
|
497
|
324 retval(0) = tree_constant (0.0); // failed |
|
1
|
325 } |
|
|
326 |
|
|
327 return retval; |
|
|
328 } |
|
|
329 |
|
529
|
330 DEFUN ("fflush", Ffflush, Sfflush, 2, 1, |
|
|
331 "fflush (FILENAME or FILENUM): flush buffered data to output file") |
|
|
332 { |
|
|
333 Octave_object retval; |
|
|
334 |
|
|
335 int nargin = args.length (); |
|
|
336 |
|
|
337 if (nargin != 2) |
|
|
338 print_usage ("fflush"); |
|
|
339 else |
|
|
340 retval = fflush_internal (args); |
|
|
341 |
|
|
342 return retval; |
|
|
343 } |
|
|
344 |
|
497
|
345 Octave_object |
|
|
346 fflush_internal (const Octave_object& args) |
|
1
|
347 { |
|
497
|
348 Octave_object retval; |
|
1
|
349 |
|
497
|
350 Pix p = return_valid_file (args(1)); |
|
1
|
351 |
|
529
|
352 if (! p) |
|
1
|
353 return retval; |
|
|
354 |
|
240
|
355 file_info file = file_list (p); |
|
1
|
356 |
|
|
357 if (strcmp (file.mode (), "r") == 0) |
|
|
358 { |
|
|
359 warning ("can't flush an input stream"); |
|
|
360 return retval; |
|
|
361 } |
|
|
362 |
|
|
363 int success = 0; |
|
240
|
364 |
|
1
|
365 if (file.number () == 1) |
|
|
366 flush_output_to_pager (); |
|
|
367 else |
|
|
368 success = fflush (file.fptr ()); |
|
|
369 |
|
497
|
370 retval.resize (1); |
|
1
|
371 if (success == 0) |
|
497
|
372 retval(0) = tree_constant (1.0); // succeeded |
|
1
|
373 else |
|
|
374 { |
|
|
375 error ("fflush: write error"); |
|
497
|
376 retval(0) = tree_constant (0.0); // failed |
|
1
|
377 } |
|
|
378 |
|
|
379 return retval; |
|
|
380 } |
|
|
381 |
|
|
382 static int |
|
164
|
383 valid_mode (const char *mode) |
|
1
|
384 { |
|
529
|
385 if (mode) |
|
1
|
386 { |
|
|
387 char m = mode[0]; |
|
|
388 if (m == 'r' || m == 'w' || m == 'a') |
|
|
389 { |
|
|
390 m = mode[1]; |
|
|
391 return (m == '\0' || (m == '+' && mode[2] == '\0')); |
|
|
392 } |
|
|
393 } |
|
|
394 return 0; |
|
|
395 } |
|
|
396 |
|
529
|
397 DEFUN ("fgets", Ffgets, Sfgets, 3, 2, |
|
|
398 "[STRING, LENGTH] = fgets (FILENAME or FILENUM, LENGTH)\n\ |
|
|
399 \n\ |
|
|
400 read a string from a file") |
|
|
401 { |
|
|
402 Octave_object retval; |
|
|
403 |
|
|
404 int nargin = args.length (); |
|
|
405 |
|
|
406 if (nargin != 3) |
|
|
407 print_usage ("fgets"); |
|
|
408 else |
|
|
409 retval = fgets_internal (args, nargout); |
|
|
410 |
|
|
411 return retval; |
|
|
412 } |
|
|
413 |
|
497
|
414 Octave_object |
|
|
415 fgets_internal (const Octave_object& args, int nargout) |
|
1
|
416 { |
|
497
|
417 Octave_object retval; |
|
1
|
418 |
|
497
|
419 Pix p = file_io_get_file (args(1), "r", "fgets"); |
|
1
|
420 |
|
529
|
421 if (! p) |
|
368
|
422 return retval; |
|
|
423 |
|
1
|
424 int length = 0; |
|
497
|
425 if (args(2).is_scalar_type ()) |
|
1
|
426 { |
|
497
|
427 length = (int) args(2).double_value (); |
|
1
|
428 if ((double) NINT (length) != length) |
|
|
429 { |
|
|
430 error ("fgets: length not an integer value"); |
|
|
431 return retval; |
|
|
432 } |
|
|
433 } |
|
|
434 |
|
368
|
435 file_info file = file_list (p); |
|
|
436 |
|
1
|
437 char string[length+1]; |
|
|
438 char *success = fgets (string, length+1, file.fptr ()); |
|
|
439 |
|
529
|
440 if (! success) |
|
1
|
441 { |
|
497
|
442 retval.resize (1); |
|
|
443 retval(0) = tree_constant (-1.0); |
|
1
|
444 return retval; |
|
|
445 } |
|
|
446 |
|
|
447 if (nargout == 2) |
|
|
448 { |
|
497
|
449 retval.resize (2); |
|
|
450 retval(1) = tree_constant ((double) strlen (string)); |
|
1
|
451 } |
|
|
452 else |
|
497
|
453 retval.resize (1); |
|
1
|
454 |
|
497
|
455 retval(0) = tree_constant (string); |
|
1
|
456 |
|
|
457 return retval; |
|
|
458 } |
|
|
459 |
|
529
|
460 DEFUN ("fopen", Ffopen, Sfopen, 3, 1, |
|
|
461 "FILENUM = fopen (FILENAME, MODE): open a file\n\ |
|
|
462 \n\ |
|
|
463 Valid values for mode include:\n\ |
|
|
464 \n\ |
|
|
465 r : open text file for reading\n\ |
|
|
466 w : open text file for writing; discard previous contents if any\n\ |
|
|
467 a : append; open or create text file for writing at end of file\n\ |
|
|
468 r+ : open text file for update (i.e., reading and writing)\n\ |
|
|
469 w+ : create text file for update; discard previous contents if any\n\ |
|
|
470 a+ : append; open or create text file for update, writing at end\n\n\ |
|
|
471 Update mode permits reading from and writing to the same file.") |
|
|
472 { |
|
|
473 Octave_object retval; |
|
|
474 |
|
|
475 int nargin = args.length (); |
|
|
476 |
|
|
477 if (nargin != 3) |
|
|
478 print_usage ("fopen"); |
|
|
479 else |
|
|
480 retval = fopen_internal (args); |
|
|
481 |
|
|
482 return retval; |
|
|
483 } |
|
|
484 |
|
497
|
485 Octave_object |
|
|
486 fopen_internal (const Octave_object& args) |
|
1
|
487 { |
|
497
|
488 Octave_object retval; |
|
1
|
489 Pix p; |
|
|
490 |
|
497
|
491 if (! args(1).is_string_type ()) |
|
1
|
492 { |
|
|
493 error ("fopen: file name must be a string"); |
|
|
494 return retval; |
|
|
495 } |
|
|
496 |
|
497
|
497 p = return_valid_file (args(1)); |
|
1
|
498 |
|
529
|
499 if (p) |
|
1
|
500 { |
|
240
|
501 file_info file = file_list (p); |
|
1
|
502 |
|
497
|
503 retval.resize (1); |
|
|
504 retval(0) = tree_constant ((double) file.number ()); |
|
1
|
505 |
|
|
506 return retval; |
|
|
507 } |
|
|
508 |
|
497
|
509 if (! args(2).is_string_type ()) |
|
1
|
510 { |
|
370
|
511 error ("fopen: file mode must be a string"); |
|
1
|
512 return retval; |
|
|
513 } |
|
|
514 |
|
497
|
515 char *name = args(1).string_value (); |
|
|
516 char *mode = args(2).string_value (); |
|
1
|
517 |
|
|
518 if (! valid_mode (mode)) |
|
|
519 { |
|
|
520 error ("fopen: invalid mode"); |
|
|
521 return retval; |
|
|
522 } |
|
|
523 |
|
|
524 struct stat buffer; |
|
|
525 if (stat (name, &buffer) == 0 && (buffer.st_mode & S_IFDIR) == S_IFDIR) |
|
|
526 { |
|
|
527 error ("fopen: can't open directory"); |
|
|
528 return retval; |
|
|
529 } |
|
|
530 |
|
|
531 FILE *file_ptr = fopen (name, mode); |
|
|
532 |
|
529
|
533 if (! file_ptr) |
|
1
|
534 { |
|
370
|
535 error ("fopen: unable to open file `%s'", name); |
|
1
|
536 return retval; |
|
|
537 } |
|
|
538 |
|
|
539 int number = file_count++; |
|
|
540 |
|
240
|
541 file_info file (number, name, file_ptr, mode); |
|
1
|
542 file_list.append (file); |
|
|
543 |
|
497
|
544 retval.resize (1); |
|
|
545 retval(0) = tree_constant ((double) number); |
|
1
|
546 |
|
|
547 return retval; |
|
|
548 } |
|
|
549 |
|
529
|
550 DEFUN ("freport", Ffreport, Sfreport, 1, 1, |
|
|
551 "freport (): list open files and their status") |
|
|
552 { |
|
|
553 Octave_object retval; |
|
|
554 |
|
|
555 int nargin = args.length (); |
|
|
556 |
|
|
557 if (nargin > 1) |
|
|
558 warning ("freport: ignoring extra arguments"); |
|
|
559 |
|
|
560 retval = freport_internal (); |
|
|
561 |
|
|
562 return retval; |
|
|
563 } |
|
|
564 |
|
497
|
565 Octave_object |
|
164
|
566 freport_internal (void) |
|
1
|
567 { |
|
497
|
568 Octave_object retval; |
|
1
|
569 Pix p = file_list.first (); |
|
|
570 |
|
|
571 ostrstream output_buf; |
|
|
572 |
|
|
573 output_buf << "\n number mode name\n\n"; |
|
|
574 for (int i = 0; i < file_count; i++) |
|
|
575 { |
|
240
|
576 file_info file = file_list (p); |
|
1
|
577 output_buf.form ("%7d%6s %s\n", file.number (), file.mode (), |
|
|
578 file.name ()); |
|
|
579 file_list.next (p); |
|
|
580 } |
|
|
581 |
|
|
582 output_buf << "\n" << ends; |
|
|
583 maybe_page_output (output_buf); |
|
|
584 |
|
|
585 return retval; |
|
|
586 } |
|
|
587 |
|
529
|
588 DEFUN ("frewind", Ffrewind, Sfrewind, 2, 1, |
|
|
589 "frewind (FILENAME or FILENUM): set file position at beginning of file") |
|
|
590 { |
|
|
591 Octave_object retval; |
|
|
592 |
|
|
593 int nargin = args.length (); |
|
|
594 |
|
|
595 if (nargin != 2) |
|
|
596 print_usage ("frewind"); |
|
|
597 else |
|
|
598 retval = frewind_internal (args); |
|
|
599 |
|
|
600 return retval; |
|
|
601 } |
|
|
602 |
|
497
|
603 Octave_object |
|
|
604 frewind_internal (const Octave_object& args) |
|
1
|
605 { |
|
497
|
606 Octave_object retval; |
|
1
|
607 |
|
497
|
608 Pix p = file_io_get_file (args(1), "a+", "frewind"); |
|
1
|
609 |
|
529
|
610 if (p) |
|
368
|
611 { |
|
|
612 file_info file = file_list (p); |
|
|
613 rewind (file.fptr ()); |
|
|
614 } |
|
1
|
615 |
|
|
616 return retval; |
|
|
617 } |
|
|
618 |
|
529
|
619 DEFUN ("fseek", Ffseek, Sfseek, 4, 1, |
|
|
620 "fseek (FILENAME or FILENUM, OFFSET [, ORIGIN])\n\ |
|
|
621 \n\ |
|
|
622 set file position for reading or writing") |
|
|
623 { |
|
|
624 Octave_object retval; |
|
|
625 |
|
|
626 int nargin = args.length (); |
|
|
627 |
|
|
628 if (nargin != 3 && nargin != 4) |
|
|
629 print_usage ("fseek"); |
|
|
630 else |
|
|
631 retval = fseek_internal (args); |
|
|
632 |
|
|
633 return retval; |
|
|
634 } |
|
|
635 |
|
497
|
636 Octave_object |
|
506
|
637 fseek_internal (const Octave_object& args) |
|
1
|
638 { |
|
497
|
639 Octave_object retval; |
|
1
|
640 |
|
506
|
641 int nargin = args.length (); |
|
|
642 |
|
497
|
643 Pix p = file_io_get_file (args(1), "a+", "fseek"); |
|
1
|
644 |
|
529
|
645 if (! p) |
|
368
|
646 return retval; |
|
1
|
647 |
|
|
648 long origin = SEEK_SET; |
|
|
649 long offset = 0; |
|
497
|
650 if (args(2).is_scalar_type ()) |
|
1
|
651 { |
|
497
|
652 offset = (long) args(2).double_value (); |
|
1
|
653 if ((double) NINT (offset) != offset) |
|
|
654 { |
|
|
655 error ("fseek: offset not an integer value"); |
|
|
656 return retval; |
|
|
657 } |
|
|
658 } |
|
|
659 |
|
497
|
660 if (nargin == 4 && args(3).is_scalar_type ()) |
|
1
|
661 { |
|
497
|
662 origin = (long) args(3).double_value (); |
|
1
|
663 if (origin == -1) |
|
|
664 origin = SEEK_CUR; |
|
|
665 else if (origin == -2) |
|
|
666 origin = SEEK_END; |
|
|
667 else |
|
|
668 { |
|
|
669 if ((double) NINT (origin) != origin) |
|
|
670 { |
|
|
671 error ("fseek: origin not an integer value"); |
|
|
672 return retval; |
|
|
673 } |
|
|
674 } |
|
|
675 } |
|
|
676 |
|
240
|
677 file_info file = file_list (p); |
|
1
|
678 int success = fseek (file.fptr (), offset, origin); |
|
497
|
679 retval.resize (1); |
|
1
|
680 |
|
|
681 if (success == 0) |
|
497
|
682 retval(0) = tree_constant (1.0); // succeeded |
|
1
|
683 else |
|
|
684 { |
|
|
685 error ("fseek: file error"); |
|
497
|
686 retval(0) = tree_constant (0.0); // failed |
|
1
|
687 } |
|
|
688 |
|
|
689 return retval; |
|
|
690 } |
|
|
691 |
|
529
|
692 /* |
|
|
693 * Tell current position of file. |
|
|
694 */ |
|
|
695 DEFUN ("ftell", Fftell, Sftell, 2, 1, |
|
|
696 "POSITION = ftell (FILENAME or FILENUM): returns the current file position") |
|
|
697 { |
|
|
698 Octave_object retval; |
|
|
699 |
|
|
700 int nargin = args.length (); |
|
|
701 |
|
|
702 if (nargin != 2) |
|
|
703 print_usage ("ftell"); |
|
|
704 else |
|
|
705 retval = ftell_internal (args); |
|
|
706 |
|
|
707 return retval; |
|
|
708 } |
|
|
709 |
|
497
|
710 Octave_object |
|
|
711 ftell_internal (const Octave_object& args) |
|
1
|
712 { |
|
497
|
713 Octave_object retval; |
|
1
|
714 |
|
497
|
715 Pix p = file_io_get_file (args(1), "a+", "ftell"); |
|
1
|
716 |
|
529
|
717 if (p) |
|
368
|
718 { |
|
|
719 file_info file = file_list (p); |
|
|
720 long offset = ftell (file.fptr ()); |
|
497
|
721 retval.resize (1); |
|
|
722 retval(0) = tree_constant ((double) offset); |
|
1
|
723 |
|
368
|
724 if (offset == -1L) |
|
|
725 error ("ftell: write error"); |
|
|
726 } |
|
1
|
727 |
|
|
728 return retval; |
|
|
729 } |
|
|
730 |
|
|
731 void |
|
164
|
732 close_files (void) |
|
1
|
733 { |
|
|
734 Pix p = file_list.first (); |
|
|
735 |
|
|
736 for (int i = 0; i < file_count; i++) |
|
|
737 { |
|
240
|
738 file_info file = file_list (p); |
|
1
|
739 if (i > 2) // do not close stdin, stdout, stderr! |
|
|
740 { |
|
|
741 int success = fclose (file.fptr ()); |
|
|
742 if (success != 0) |
|
|
743 error ("closing %s", file.name ()); |
|
|
744 } |
|
|
745 file_list.del (p); |
|
|
746 } |
|
|
747 } |
|
|
748 |
|
|
749 static int |
|
497
|
750 process_printf_format (const char *s, const Octave_object& args, |
|
506
|
751 ostrstream& sb, const char *type) |
|
1
|
752 { |
|
|
753 ostrstream fmt; |
|
|
754 |
|
506
|
755 int nargin = args.length (); |
|
|
756 |
|
1
|
757 fmt << "%"; // do_printf() already blew past this one... |
|
|
758 |
|
|
759 tree_constant_rep::constant_type arg_type; |
|
|
760 |
|
|
761 int chars_from_fmt_str = 0; |
|
|
762 |
|
|
763 again: |
|
|
764 switch (*s) |
|
|
765 { |
|
|
766 case '+': case '-': case ' ': case '0': case '#': |
|
|
767 chars_from_fmt_str++; |
|
|
768 fmt << *s++; |
|
|
769 goto again; |
|
|
770 |
|
|
771 case '\0': |
|
|
772 goto invalid_format; |
|
|
773 |
|
|
774 default: |
|
|
775 break; |
|
|
776 } |
|
|
777 |
|
|
778 if (*s == '*') |
|
|
779 { |
|
|
780 if (fmt_arg_count >= nargin) |
|
|
781 { |
|
218
|
782 error ("%s: not enough arguments", type); |
|
1
|
783 return -1; |
|
|
784 } |
|
|
785 |
|
497
|
786 if (args(fmt_arg_count).const_type () |
|
1
|
787 != tree_constant_rep::scalar_constant) |
|
|
788 { |
|
218
|
789 error ("%s: `*' must be replaced by an integer", type); |
|
1
|
790 return -1; |
|
|
791 } |
|
|
792 |
|
497
|
793 fmt << NINT (args(fmt_arg_count++).double_value ()); |
|
1
|
794 s++; |
|
|
795 chars_from_fmt_str++; |
|
|
796 } |
|
|
797 else |
|
|
798 { |
|
|
799 while (*s != '\0' && isdigit (*s)) |
|
|
800 { |
|
|
801 chars_from_fmt_str++; |
|
|
802 fmt << *s++; |
|
|
803 } |
|
|
804 } |
|
|
805 |
|
|
806 if (*s == '\0') |
|
|
807 goto invalid_format; |
|
|
808 |
|
|
809 if (*s == '.') |
|
|
810 { |
|
|
811 chars_from_fmt_str++; |
|
|
812 fmt << *s++; |
|
|
813 } |
|
|
814 |
|
|
815 if (*s == '*') |
|
|
816 { |
|
|
817 if (*(s-1) == '*') |
|
|
818 goto invalid_format; |
|
|
819 |
|
|
820 if (fmt_arg_count >= nargin) |
|
|
821 { |
|
218
|
822 error ("%s: not enough arguments", type); |
|
1
|
823 return -1; |
|
|
824 } |
|
|
825 |
|
497
|
826 if (args(fmt_arg_count).const_type () |
|
1
|
827 != tree_constant_rep::scalar_constant) |
|
|
828 { |
|
218
|
829 error ("%s: `*' must be replaced by an integer", type); |
|
1
|
830 return -1; |
|
|
831 } |
|
|
832 |
|
497
|
833 fmt << NINT (args(fmt_arg_count++).double_value ()); |
|
1
|
834 s++; |
|
|
835 chars_from_fmt_str++; |
|
|
836 } |
|
|
837 else |
|
|
838 { |
|
|
839 while (*s != '\0' && isdigit (*s)) |
|
|
840 { |
|
|
841 chars_from_fmt_str++; |
|
|
842 fmt << *s++; |
|
|
843 } |
|
|
844 } |
|
|
845 |
|
|
846 if (*s == '\0') |
|
|
847 goto invalid_format; |
|
|
848 |
|
|
849 if (*s != '\0' && (*s == 'h' || *s == 'l' || *s == 'L')) |
|
|
850 { |
|
|
851 chars_from_fmt_str++; |
|
|
852 fmt << *s++; |
|
|
853 } |
|
|
854 |
|
|
855 if (*s == '\0') |
|
|
856 goto invalid_format; |
|
|
857 |
|
|
858 if (fmt_arg_count >= nargin) |
|
|
859 { |
|
218
|
860 error ("%s: not enough arguments", type); |
|
1
|
861 return -1; |
|
|
862 } |
|
|
863 |
|
497
|
864 arg_type = args(fmt_arg_count).const_type (); |
|
1
|
865 |
|
|
866 switch (*s) |
|
|
867 { |
|
|
868 case 'd': case 'i': case 'o': case 'u': case 'x': case 'X': |
|
|
869 |
|
|
870 if (arg_type != tree_constant_rep::scalar_constant) |
|
|
871 goto invalid_conversion; |
|
|
872 else |
|
|
873 { |
|
|
874 chars_from_fmt_str++; |
|
|
875 fmt << *s << ends; |
|
497
|
876 double d = args(fmt_arg_count++).double_value (); |
|
1
|
877 if ((int) d != d) |
|
|
878 goto invalid_conversion; |
|
|
879 else |
|
|
880 { |
|
|
881 char *s = fmt.str (); |
|
|
882 sb.form (s, (int) d); |
|
|
883 delete [] s; |
|
|
884 return chars_from_fmt_str; |
|
|
885 } |
|
|
886 } |
|
|
887 |
|
|
888 case 'e': case 'E': case 'f': case 'g': case 'G': |
|
|
889 |
|
|
890 if (arg_type != tree_constant_rep::scalar_constant) |
|
|
891 goto invalid_conversion; |
|
|
892 else |
|
|
893 { |
|
|
894 chars_from_fmt_str++; |
|
|
895 fmt << *s << ends; |
|
|
896 char *s = fmt.str (); |
|
497
|
897 sb.form (s, args(fmt_arg_count++).double_value ()); |
|
1
|
898 delete [] s; |
|
|
899 return chars_from_fmt_str; |
|
|
900 } |
|
|
901 |
|
|
902 case 's': |
|
|
903 |
|
|
904 if (arg_type != tree_constant_rep::string_constant) |
|
|
905 goto invalid_conversion; |
|
|
906 else |
|
|
907 { |
|
|
908 chars_from_fmt_str++; |
|
|
909 fmt << *s << ends; |
|
|
910 char *s = fmt.str (); |
|
497
|
911 sb.form (s, args(fmt_arg_count++).string_value ()); |
|
1
|
912 delete [] s; |
|
|
913 return chars_from_fmt_str; |
|
|
914 } |
|
|
915 |
|
|
916 case 'c': |
|
|
917 |
|
|
918 if (arg_type != tree_constant_rep::string_constant) |
|
|
919 goto invalid_conversion; |
|
|
920 else |
|
|
921 { |
|
|
922 chars_from_fmt_str++; |
|
|
923 fmt << *s << ends; |
|
497
|
924 char *str = args(fmt_arg_count++).string_value (); |
|
1
|
925 if (strlen (str) != 1) |
|
|
926 goto invalid_conversion; |
|
|
927 else |
|
|
928 { |
|
|
929 char *s = fmt.str (); |
|
|
930 sb.form (s, *str); |
|
|
931 delete [] s; |
|
|
932 return chars_from_fmt_str; |
|
|
933 } |
|
|
934 } |
|
|
935 |
|
|
936 default: |
|
|
937 goto invalid_format; |
|
|
938 } |
|
|
939 |
|
|
940 invalid_conversion: |
|
218
|
941 error ("%s: invalid conversion", type); |
|
1
|
942 return -1; |
|
|
943 |
|
|
944 invalid_format: |
|
218
|
945 error ("%s: invalid format", type); |
|
1
|
946 return -1; |
|
|
947 } |
|
|
948 |
|
|
949 |
|
529
|
950 /* |
|
|
951 * Formatted printing to a file. |
|
|
952 */ |
|
|
953 DEFUN ("fprintf", Ffprintf, Sfprintf, -1, 1, |
|
|
954 "fprintf (FILENAME or FILENUM, FORMAT, ...)") |
|
|
955 { |
|
|
956 Octave_object retval; |
|
|
957 |
|
|
958 int nargin = args.length (); |
|
|
959 |
|
|
960 if (nargin < 3) |
|
|
961 print_usage ("fprintf"); |
|
|
962 else |
|
|
963 retval = do_printf ("fprintf", args, nargout); |
|
|
964 |
|
|
965 return retval; |
|
|
966 } |
|
|
967 |
|
|
968 /* |
|
|
969 * Formatted printing. |
|
|
970 */ |
|
|
971 DEFUN ("printf", Fprintf, Sprintf, -1, 1, |
|
|
972 "printf (FORMAT, ...)") |
|
|
973 { |
|
|
974 Octave_object retval; |
|
|
975 |
|
|
976 int nargin = args.length (); |
|
|
977 |
|
|
978 if (nargin < 2) |
|
|
979 print_usage ("printf"); |
|
|
980 else |
|
|
981 retval = do_printf ("printf", args, nargout); |
|
|
982 |
|
|
983 return retval; |
|
|
984 } |
|
|
985 |
|
|
986 /* |
|
|
987 * Formatted printing to a string. |
|
|
988 */ |
|
|
989 DEFUN ("sprintf", Fsprintf, Ssprintf, -1, 1, |
|
|
990 "s = sprintf (FORMAT, ...)") |
|
|
991 { |
|
|
992 Octave_object retval; |
|
|
993 |
|
|
994 int nargin = args.length (); |
|
|
995 |
|
|
996 if (nargin < 2) |
|
|
997 print_usage ("sprintf"); |
|
|
998 else |
|
|
999 retval = do_printf ("sprintf", args, nargout); |
|
|
1000 |
|
|
1001 return retval; |
|
|
1002 } |
|
|
1003 |
|
497
|
1004 Octave_object |
|
506
|
1005 do_printf (const char *type, const Octave_object& args, int nargout) |
|
1
|
1006 { |
|
497
|
1007 Octave_object retval; |
|
1
|
1008 fmt_arg_count = 1; |
|
|
1009 char *fmt; |
|
240
|
1010 file_info file; |
|
1
|
1011 |
|
|
1012 if (strcmp (type, "fprintf") == 0) |
|
|
1013 { |
|
497
|
1014 if (args(2).is_string_type ()) |
|
1
|
1015 { |
|
497
|
1016 fmt = args(2).string_value (); |
|
1
|
1017 fmt_arg_count++; |
|
|
1018 } |
|
|
1019 else |
|
|
1020 { |
|
|
1021 error ("%s: format must be a string", type); |
|
|
1022 return retval; |
|
|
1023 } |
|
|
1024 |
|
497
|
1025 Pix p = file_io_get_file (args(1), "a+", type); |
|
368
|
1026 |
|
529
|
1027 if (! p) |
|
368
|
1028 return retval; |
|
1
|
1029 |
|
|
1030 file = file_list (p); |
|
368
|
1031 |
|
1
|
1032 if (file.mode () == "r") |
|
|
1033 { |
|
|
1034 error ("%s: file is read only", type); |
|
|
1035 return retval; |
|
|
1036 } |
|
368
|
1037 |
|
497
|
1038 fmt = args(2).string_value (); |
|
368
|
1039 |
|
1
|
1040 fmt_arg_count++; |
|
|
1041 } |
|
497
|
1042 else if (args(1).is_string_type ()) |
|
1
|
1043 { |
|
497
|
1044 fmt = args(1).string_value (); |
|
1
|
1045 fmt_arg_count++; |
|
|
1046 } |
|
|
1047 else |
|
|
1048 { |
|
|
1049 error ("%s: invalid format string", type); |
|
|
1050 return retval; |
|
|
1051 } |
|
|
1052 |
|
|
1053 // Scan fmt for % escapes and print out the arguments. |
|
|
1054 |
|
|
1055 ostrstream output_buf; |
|
|
1056 |
|
|
1057 char *ptr = fmt; |
|
|
1058 |
|
|
1059 for (;;) |
|
|
1060 { |
|
|
1061 char c; |
|
|
1062 while ((c = *ptr++) != '\0' && c != '%') |
|
|
1063 output_buf << c; |
|
|
1064 |
|
|
1065 if (c == '\0') |
|
|
1066 break; |
|
|
1067 |
|
|
1068 if (*ptr == '%') |
|
|
1069 { |
|
|
1070 ptr++; |
|
|
1071 output_buf << c; |
|
|
1072 continue; |
|
|
1073 } |
|
|
1074 |
|
|
1075 // We must be looking at a format specifier. Extract it or fail. |
|
|
1076 |
|
|
1077 |
|
506
|
1078 int status = process_printf_format (ptr, args, output_buf, type); |
|
1
|
1079 |
|
|
1080 if (status < 0) |
|
|
1081 return retval; |
|
|
1082 |
|
|
1083 ptr += status; |
|
|
1084 } |
|
|
1085 |
|
|
1086 output_buf << ends; |
|
|
1087 if (strcmp (type, "printf") == 0 |
|
|
1088 || (strcmp (type, "fprintf") == 0 && file.number () == 1)) |
|
|
1089 { |
|
|
1090 maybe_page_output (output_buf); |
|
|
1091 } |
|
|
1092 else if (strcmp (type, "fprintf") == 0) |
|
|
1093 { |
|
|
1094 char *msg = output_buf.str (); |
|
|
1095 int success = fputs (msg, file.fptr ()); |
|
|
1096 if (success == EOF) |
|
218
|
1097 warning ("%s: unknown failure writing to file", type); |
|
1
|
1098 delete [] msg; |
|
|
1099 } |
|
|
1100 else if (strcmp (type, "sprintf") == 0) |
|
|
1101 { |
|
497
|
1102 retval.resize (1); |
|
1
|
1103 char *msg = output_buf.str (); |
|
497
|
1104 retval(0) = tree_constant (msg); |
|
1
|
1105 delete [] msg; |
|
|
1106 } |
|
|
1107 |
|
|
1108 return retval; |
|
|
1109 } |
|
|
1110 |
|
|
1111 static int |
|
368
|
1112 process_scanf_format (const char *s, ostrstream& fmt, |
|
|
1113 const char *type, int nargout, FILE* fptr, |
|
497
|
1114 Octave_object& values) |
|
1
|
1115 { |
|
|
1116 fmt << "%"; |
|
|
1117 |
|
|
1118 int chars_from_fmt_str = 0; |
|
|
1119 int store_value = 1; |
|
|
1120 int string_width = -1; |
|
|
1121 int success = 1; |
|
|
1122 |
|
|
1123 if (*s == '*') |
|
|
1124 { |
|
|
1125 store_value = 0; |
|
|
1126 s++; |
|
|
1127 chars_from_fmt_str++; |
|
|
1128 } |
|
|
1129 |
|
|
1130 if (isdigit (*s)) |
|
|
1131 { |
|
|
1132 ostrstream str_number; |
|
|
1133 while (*s != '\0' && isdigit (*s)) |
|
|
1134 { |
|
|
1135 chars_from_fmt_str++; |
|
|
1136 str_number << *s; |
|
|
1137 fmt << *s++; |
|
|
1138 } |
|
|
1139 str_number << ends; |
|
|
1140 char *number = str_number.str (); |
|
|
1141 string_width = atoi (number); |
|
|
1142 delete [] number; |
|
|
1143 } |
|
|
1144 |
|
|
1145 if (*s == '\0') |
|
|
1146 goto invalid_format; |
|
|
1147 |
|
|
1148 if (*s != '\0' && (*s == 'h' || *s == 'l' || *s == 'L')) |
|
|
1149 { |
|
|
1150 chars_from_fmt_str++; |
|
|
1151 s++; |
|
|
1152 } |
|
|
1153 |
|
|
1154 if (*s == '\0') |
|
|
1155 goto invalid_format; |
|
|
1156 |
|
368
|
1157 // Even if we don't have a place to store them, attempt to convert |
|
|
1158 // everything specified by the format string. |
|
1
|
1159 |
|
497
|
1160 if (fmt_arg_count >= (nargout ? nargout : 1)) |
|
368
|
1161 store_value = 0; |
|
1
|
1162 |
|
|
1163 switch (*s) |
|
|
1164 { |
|
|
1165 case 'd': case 'i': case 'o': case 'u': case 'x': case 'X': |
|
|
1166 { |
|
|
1167 chars_from_fmt_str++; |
|
|
1168 fmt << *s << ends; |
|
|
1169 int temp; |
|
|
1170 char *str = fmt.str (); |
|
|
1171 success = fscanf (fptr, str, &temp); |
|
|
1172 delete [] str; |
|
|
1173 if (success > 0 && store_value) |
|
497
|
1174 values(fmt_arg_count++) = tree_constant ((double) temp); |
|
1
|
1175 } |
|
|
1176 break; |
|
|
1177 case 'e': case 'E': case 'f': case 'g': case 'G': |
|
|
1178 { |
|
|
1179 chars_from_fmt_str++; |
|
|
1180 fmt << 'l' << *s << ends; |
|
|
1181 double temp; |
|
|
1182 char *str = fmt.str (); |
|
|
1183 success = fscanf (fptr, str, &temp); |
|
|
1184 delete [] str; |
|
|
1185 if (success > 0 && store_value) |
|
497
|
1186 values(fmt_arg_count++) = tree_constant (temp); |
|
1
|
1187 } |
|
|
1188 break; |
|
|
1189 case 's': |
|
|
1190 { |
|
|
1191 if (string_width < 1) |
|
|
1192 { |
|
|
1193 string_width = 0; |
|
|
1194 long original_position = ftell (fptr); |
|
|
1195 int c; |
|
|
1196 |
|
|
1197 while ((c = getc (fptr)) != EOF |
|
|
1198 && (c == ' ' || c == '\n' || c != '\t')) |
|
|
1199 ; // Don't count leading whitespace. |
|
|
1200 |
|
|
1201 if (c != EOF) |
|
|
1202 string_width++; |
|
|
1203 |
|
|
1204 for (;;) |
|
|
1205 { |
|
|
1206 c = getc (fptr); |
|
|
1207 if (c != EOF && c != ' ' && c != '\n' && c != '\t') |
|
|
1208 string_width++; |
|
|
1209 else |
|
|
1210 break; |
|
|
1211 } |
|
|
1212 |
|
|
1213 fseek (fptr, original_position, SEEK_SET); |
|
|
1214 } |
|
|
1215 chars_from_fmt_str++; |
|
|
1216 char temp[string_width+1]; |
|
|
1217 fmt << *s << ends; |
|
|
1218 char *str = fmt.str (); |
|
|
1219 success = fscanf (fptr, str, temp); |
|
|
1220 delete [] str; |
|
|
1221 if (success && store_value) |
|
497
|
1222 values(fmt_arg_count++) = tree_constant (temp); |
|
1
|
1223 } |
|
|
1224 break; |
|
|
1225 case 'c': |
|
|
1226 { |
|
|
1227 if (string_width < 1) |
|
|
1228 string_width = 1; |
|
|
1229 chars_from_fmt_str++; |
|
|
1230 char temp[string_width+1]; |
|
|
1231 memset (temp, '\0', string_width+1); |
|
|
1232 fmt << *s << ends; |
|
|
1233 char *str = fmt.str (); |
|
|
1234 success = fscanf (fptr, str, temp); |
|
|
1235 delete [] str; |
|
|
1236 temp[string_width] = '\0'; |
|
|
1237 if (success > 0 && store_value) |
|
497
|
1238 values(fmt_arg_count++) = tree_constant (temp); |
|
1
|
1239 } |
|
|
1240 break; |
|
|
1241 default: |
|
|
1242 goto invalid_format; |
|
|
1243 } |
|
|
1244 |
|
368
|
1245 if (success > 0) |
|
1
|
1246 return chars_from_fmt_str; |
|
368
|
1247 else if (success == 0) |
|
218
|
1248 warning ("%s: invalid conversion", type); |
|
1
|
1249 else if (success == EOF) |
|
|
1250 { |
|
|
1251 if (strcmp (type, "fscanf") == 0) |
|
218
|
1252 warning ("%s: end of file reached before final conversion", type); |
|
1
|
1253 else if (strcmp (type, "sscanf") == 0) |
|
218
|
1254 warning ("%s: end of string reached before final conversion", type); |
|
1
|
1255 else if (strcmp (type, "scanf") == 0) |
|
218
|
1256 warning ("%s: end of input reached before final conversion", type); |
|
1
|
1257 } |
|
|
1258 else |
|
|
1259 { |
|
|
1260 invalid_format: |
|
218
|
1261 warning ("%s: invalid format", type); |
|
1
|
1262 } |
|
|
1263 |
|
|
1264 return -1; |
|
|
1265 } |
|
|
1266 |
|
529
|
1267 /* |
|
|
1268 * Formatted reading from a file. |
|
|
1269 */ |
|
|
1270 DEFUN ("fscanf", Ffscanf, Sfscanf, 3, -1, |
|
|
1271 "[A, B, C, ...] = fscanf (FILENAME or FILENUM, FORMAT)") |
|
|
1272 { |
|
|
1273 Octave_object retval; |
|
|
1274 |
|
|
1275 int nargin = args.length (); |
|
|
1276 |
|
|
1277 if (nargin != 2 && nargin != 3) |
|
|
1278 print_usage ("fscanf"); |
|
|
1279 else |
|
|
1280 retval = do_scanf ("fscanf", args, nargout); |
|
|
1281 |
|
|
1282 return retval; |
|
|
1283 } |
|
|
1284 |
|
|
1285 /* |
|
|
1286 * Formatted reading. |
|
|
1287 */ |
|
|
1288 DEFUN ("scanf", Fscanf, Sscanf, 2, -1, |
|
|
1289 "[A, B, C, ...] = scanf (FORMAT)") |
|
|
1290 { |
|
|
1291 Octave_object retval; |
|
|
1292 |
|
|
1293 int nargin = args.length (); |
|
|
1294 |
|
|
1295 if (nargin != 2) |
|
|
1296 print_usage ("scanf"); |
|
|
1297 else |
|
|
1298 retval = do_scanf ("scanf", args, nargout); |
|
|
1299 |
|
|
1300 return retval; |
|
|
1301 } |
|
|
1302 |
|
|
1303 /* |
|
|
1304 * Formatted reading from a string. |
|
|
1305 */ |
|
|
1306 DEFUN ("sscanf", Fsscanf, Ssscanf, 3, -1, |
|
|
1307 "[A, B, C, ...] = sscanf (STRING, FORMAT)") |
|
|
1308 { |
|
|
1309 Octave_object retval; |
|
|
1310 |
|
|
1311 int nargin = args.length (); |
|
|
1312 |
|
|
1313 if (nargin != 3) |
|
|
1314 print_usage ("sscanf"); |
|
|
1315 else |
|
|
1316 retval = do_scanf ("sscanf", args, nargout); |
|
|
1317 |
|
|
1318 return retval; |
|
|
1319 } |
|
|
1320 |
|
497
|
1321 Octave_object |
|
506
|
1322 do_scanf (const char *type, const Octave_object& args, int nargout) |
|
1
|
1323 { |
|
497
|
1324 Octave_object retval; |
|
529
|
1325 char *scanf_fmt = 0; |
|
|
1326 char *tmp_file = 0; |
|
1
|
1327 int tmp_file_open = 0; |
|
529
|
1328 FILE *fptr = 0; |
|
240
|
1329 file_info file; |
|
1
|
1330 |
|
|
1331 fmt_arg_count = 0; |
|
|
1332 |
|
|
1333 if (strcmp (type, "scanf") != 0) |
|
|
1334 { |
|
497
|
1335 if (args(2).is_string_type ()) |
|
|
1336 scanf_fmt = args(2).string_value (); |
|
1
|
1337 else |
|
|
1338 { |
|
|
1339 error ("%s: format must be a string", type); |
|
|
1340 return retval; |
|
|
1341 } |
|
|
1342 } |
|
|
1343 |
|
|
1344 int doing_fscanf = (strcmp (type, "fscanf") == 0); |
|
|
1345 |
|
|
1346 if (doing_fscanf) |
|
|
1347 { |
|
497
|
1348 Pix p = file_io_get_file (args(1), "r", type); |
|
368
|
1349 |
|
529
|
1350 if (! p) |
|
368
|
1351 return retval; |
|
1
|
1352 |
|
|
1353 file = file_list (p); |
|
|
1354 |
|
|
1355 if (strcmp (file.mode (), "w") == 0 || strcmp (file.mode (), "a") == 0) |
|
|
1356 { |
|
|
1357 error ("%s: this file is opened for writing only", type); |
|
|
1358 return retval; |
|
|
1359 } |
|
|
1360 |
|
|
1361 fptr = file.fptr (); |
|
|
1362 } |
|
|
1363 |
|
529
|
1364 if ((! fptr && args(1).is_string_type ()) |
|
368
|
1365 || (doing_fscanf && file.number () == 0)) |
|
1
|
1366 { |
|
|
1367 char *string; |
|
|
1368 |
|
|
1369 if (strcmp (type, "scanf") == 0) |
|
497
|
1370 scanf_fmt = args(1).string_value (); |
|
1
|
1371 |
|
|
1372 if (strcmp (type, "scanf") == 0 |
|
|
1373 || (doing_fscanf && file.number () == 0)) |
|
|
1374 { |
|
|
1375 string = gnu_readline (""); |
|
|
1376 if (string && *string) |
|
|
1377 maybe_save_history (string); |
|
|
1378 } |
|
|
1379 else |
|
497
|
1380 string = args(1).string_value (); |
|
1
|
1381 |
|
529
|
1382 tmp_file = tmpnam (0); |
|
1
|
1383 |
|
|
1384 fptr = fopen (tmp_file, "w+"); |
|
529
|
1385 if (! fptr) |
|
1
|
1386 { |
|
|
1387 error ("%s: error opening temporary file", type); |
|
|
1388 return retval; |
|
|
1389 } |
|
|
1390 tmp_file_open = 1; |
|
|
1391 unlink (tmp_file); |
|
|
1392 |
|
529
|
1393 if (! string) |
|
1
|
1394 panic_impossible (); |
|
|
1395 |
|
|
1396 int success = fputs (string, fptr); |
|
|
1397 fflush (fptr); |
|
|
1398 rewind (fptr); |
|
|
1399 |
|
|
1400 if (success < 0) |
|
|
1401 { |
|
|
1402 error ("%s: trouble writing temporary file", type); |
|
|
1403 fclose (fptr); |
|
|
1404 return retval; |
|
|
1405 } |
|
|
1406 } |
|
|
1407 else if (! doing_fscanf) |
|
|
1408 { |
|
|
1409 error ("%s: first argument must be a string", type); |
|
|
1410 return retval; |
|
|
1411 } |
|
|
1412 |
|
|
1413 // Scan scanf_fmt for % escapes and assign the arguments. |
|
|
1414 |
|
497
|
1415 retval.resize (nargout ? nargout : 1); |
|
1
|
1416 |
|
|
1417 char *ptr = scanf_fmt; |
|
|
1418 |
|
|
1419 for (;;) |
|
|
1420 { |
|
|
1421 ostrstream fmt; |
|
|
1422 char c; |
|
|
1423 while ((c = *ptr++) != '\0' && c != '%') |
|
|
1424 fmt << c; |
|
|
1425 |
|
|
1426 if (c == '\0') |
|
|
1427 break; |
|
|
1428 |
|
|
1429 if (*ptr == '%') |
|
|
1430 { |
|
|
1431 ptr++; |
|
|
1432 fmt << c; |
|
|
1433 continue; |
|
|
1434 } |
|
|
1435 |
|
|
1436 // We must be looking at a format specifier. Extract it or fail. |
|
|
1437 |
|
368
|
1438 int status = process_scanf_format (ptr, fmt, type, nargout, |
|
|
1439 fptr, retval); |
|
1
|
1440 |
|
|
1441 if (status < 0) |
|
|
1442 { |
|
|
1443 if (fmt_arg_count == 0) |
|
497
|
1444 retval.resize (0); |
|
1
|
1445 break; |
|
|
1446 } |
|
|
1447 |
|
|
1448 ptr += status; |
|
|
1449 } |
|
|
1450 |
|
|
1451 if (tmp_file_open) |
|
|
1452 fclose (fptr); |
|
|
1453 |
|
|
1454 return retval; |
|
|
1455 } |
|
|
1456 |
|
|
1457 /* |
|
471
|
1458 * Find out how many elements are left to read. |
|
444
|
1459 */ |
|
|
1460 static long |
|
471
|
1461 num_items_remaining (FILE *fptr, char *type) |
|
444
|
1462 { |
|
471
|
1463 size_t size; |
|
|
1464 |
|
|
1465 if (strcasecmp (type, "uchar") == 0) |
|
|
1466 size = sizeof (u_char); |
|
|
1467 else if (strcasecmp (type, "char") == 0) |
|
|
1468 size = sizeof (char); |
|
|
1469 else if (strcasecmp (type, "short") == 0) |
|
|
1470 size = sizeof (short); |
|
|
1471 else if (strcasecmp (type, "ushort") == 0) |
|
|
1472 size = sizeof (u_short); |
|
|
1473 else if (strcasecmp (type, "int") == 0) |
|
|
1474 size = sizeof (int); |
|
|
1475 else if (strcasecmp (type, "uint") == 0) |
|
|
1476 size = sizeof (u_int); |
|
|
1477 else if (strcasecmp (type, "long") == 0) |
|
|
1478 size = sizeof (long); |
|
|
1479 else if (strcasecmp (type, "ulong") == 0) |
|
|
1480 size = sizeof (u_long); |
|
|
1481 else if (strcasecmp (type, "float") == 0) |
|
|
1482 size = sizeof (float); |
|
|
1483 else if (strcasecmp (type, "double") == 0) |
|
|
1484 size = sizeof (double); |
|
|
1485 else |
|
|
1486 return 0; |
|
|
1487 |
|
444
|
1488 long curr_pos = ftell (fptr); |
|
|
1489 |
|
|
1490 fseek (fptr, 0, SEEK_END); |
|
|
1491 long end_of_file = ftell (fptr); |
|
|
1492 |
|
471
|
1493 fseek (fptr, curr_pos, SEEK_SET); |
|
444
|
1494 |
|
|
1495 long len = end_of_file - curr_pos; |
|
|
1496 |
|
471
|
1497 return len / size; |
|
444
|
1498 } |
|
|
1499 |
|
529
|
1500 DEFUN ("fread", Ffread, Sfread, 4, 2, |
|
|
1501 "[DATA, COUNT] = fread (FILENUM, SIZE, PRECISION)\n\ |
|
|
1502 \n\ |
|
|
1503 Reads data in binary form of type PRECISION from a file.\n\ |
|
|
1504 \n\ |
|
|
1505 FILENUM : file number from fopen\n\ |
|
|
1506 SIZE : size specification for the Data matrix\n\ |
|
|
1507 PRECISION : type of data to read, valid types are\n\ |
|
|
1508 \n\ |
|
|
1509 'char', 'schar', 'short', 'int', 'long', 'float'\n\ |
|
|
1510 'double', 'uchar', 'ushort', 'uint', 'ulong'\n\ |
|
|
1511 \n\ |
|
|
1512 DATA : matrix in which the data is stored\n\ |
|
|
1513 COUNT : number of elements read") |
|
|
1514 { |
|
|
1515 Octave_object retval; |
|
|
1516 |
|
|
1517 int nargin = args.length (); |
|
|
1518 |
|
|
1519 if (nargin < 2 || nargin > 4) |
|
|
1520 print_usage ("fread"); |
|
|
1521 else |
|
|
1522 retval = fread_internal (args, nargout); |
|
|
1523 |
|
|
1524 return retval; |
|
|
1525 } |
|
|
1526 |
|
444
|
1527 /* |
|
|
1528 * Read binary data from a file. |
|
|
1529 * |
|
|
1530 * [data, count] = fread (fid, size, 'precision') |
|
|
1531 * |
|
|
1532 * fid : the file id from fopen |
|
|
1533 * size : the size of the matrix or vector or scaler to read |
|
|
1534 * |
|
|
1535 * n : reads n elements of a column vector |
|
|
1536 * inf : reads to the end of file (default) |
|
|
1537 * [m, n] : reads enough elements to fill the matrix |
|
|
1538 * the number of columns can be inf |
|
|
1539 * |
|
|
1540 * precision : type of the element. Can be: |
|
|
1541 * |
|
|
1542 * char, uchar, schar, short, ushort, int, uint, |
|
|
1543 * long, ulong, float, double |
|
|
1544 * |
|
|
1545 * Default is uchar. |
|
|
1546 * |
|
|
1547 * data : output data |
|
|
1548 * count : number of elements read |
|
|
1549 */ |
|
497
|
1550 Octave_object |
|
506
|
1551 fread_internal (const Octave_object& args, int nargout) |
|
444
|
1552 { |
|
497
|
1553 Octave_object retval; |
|
444
|
1554 |
|
506
|
1555 int nargin = args.length (); |
|
|
1556 |
|
497
|
1557 Pix p = file_io_get_file (args(1), "r", "fread"); |
|
444
|
1558 |
|
529
|
1559 if (! p) |
|
444
|
1560 return retval; |
|
|
1561 |
|
|
1562 // Get type and number of bytes per element to read. |
|
|
1563 char *prec = "uchar"; |
|
|
1564 if (nargin > 3) |
|
|
1565 { |
|
497
|
1566 if (args(3).is_string_type ()) |
|
|
1567 prec = args(3).string_value (); |
|
444
|
1568 else |
|
|
1569 { |
|
|
1570 error ("fread: precision must be a specified as a string"); |
|
|
1571 return retval; |
|
|
1572 } |
|
|
1573 } |
|
|
1574 |
|
471
|
1575 // Get file info. |
|
444
|
1576 |
|
|
1577 file_info file = file_list (p); |
|
471
|
1578 |
|
|
1579 FILE *fptr = file.fptr (); |
|
444
|
1580 |
|
|
1581 // Set up matrix to read into. If specified in arguments use that |
|
|
1582 // number, otherwise read everyting left in file. |
|
|
1583 |
|
471
|
1584 double dnr = 0.0; |
|
|
1585 double dnc = 0.0; |
|
|
1586 int nr; |
|
|
1587 int nc; |
|
444
|
1588 |
|
|
1589 if (nargin > 2) |
|
|
1590 { |
|
497
|
1591 if (args(2).is_scalar_type ()) |
|
444
|
1592 { |
|
497
|
1593 tree_constant tmpa = args(2).make_numeric (); |
|
471
|
1594 dnr = tmpa.double_value (); |
|
|
1595 dnc = 1.0; |
|
444
|
1596 } |
|
497
|
1597 else if (args(2).is_matrix_type ()) |
|
444
|
1598 { |
|
497
|
1599 ColumnVector tmp = args(2).to_vector (); |
|
444
|
1600 |
|
471
|
1601 if (tmp.length () == 2) |
|
|
1602 { |
|
|
1603 dnr = tmp.elem (0); |
|
|
1604 dnc = tmp.elem (1); |
|
|
1605 } |
|
|
1606 else |
|
|
1607 { |
|
|
1608 error ("fread: invalid size specification\n"); |
|
|
1609 return retval; |
|
|
1610 } |
|
|
1611 } |
|
|
1612 |
|
|
1613 if ((xisinf (dnr)) && (xisinf (dnc))) |
|
444
|
1614 { |
|
471
|
1615 error ("fread: number of rows and columns cannot both be infinite"); |
|
444
|
1616 return retval; |
|
|
1617 } |
|
|
1618 |
|
471
|
1619 if (xisinf (dnr)) |
|
|
1620 { |
|
|
1621 nc = NINT (dnc); |
|
|
1622 int n = num_items_remaining (fptr, prec); |
|
|
1623 nr = n / nc; |
|
|
1624 if (n > nr * nc) |
|
|
1625 nr++; |
|
|
1626 } |
|
|
1627 else if (xisinf (dnc)) |
|
|
1628 { |
|
|
1629 nr = NINT (dnr); |
|
|
1630 int n = num_items_remaining (fptr, prec); |
|
|
1631 nc = n / nr; |
|
|
1632 if (n > nc * nr) |
|
|
1633 nc++; |
|
|
1634 } |
|
|
1635 else |
|
|
1636 { |
|
|
1637 nr = NINT (dnr); |
|
|
1638 nc = NINT (dnc); |
|
|
1639 } |
|
444
|
1640 } |
|
|
1641 else |
|
|
1642 { |
|
|
1643 // No size parameter, read what's left of the file. |
|
471
|
1644 nc = 1; |
|
|
1645 int n = num_items_remaining (fptr, prec); |
|
|
1646 nr = n / nc; |
|
|
1647 if (n > nr * nc) |
|
|
1648 nr++; |
|
444
|
1649 } |
|
|
1650 |
|
|
1651 Matrix m (nr, nc, octave_NaN); |
|
|
1652 |
|
|
1653 // Read data. |
|
|
1654 |
|
471
|
1655 int count = m.read (fptr, prec); |
|
444
|
1656 |
|
|
1657 if (nargout > 1) |
|
|
1658 { |
|
497
|
1659 retval.resize (2); |
|
|
1660 retval(1) = tree_constant ((double) count); |
|
444
|
1661 } |
|
|
1662 else |
|
497
|
1663 retval.resize (1); |
|
444
|
1664 |
|
497
|
1665 retval(0) = tree_constant (m); |
|
444
|
1666 |
|
|
1667 return retval; |
|
|
1668 } |
|
|
1669 |
|
529
|
1670 DEFUN ("fwrite", Ffwrite, Sfwrite, 4, 1, |
|
|
1671 "COUNT = fwrite (FILENUM, DATA, PRECISION)\n\ |
|
|
1672 \n\ |
|
|
1673 Writes data to a file in binary form of size PRECISION\n\ |
|
|
1674 \n\ |
|
|
1675 FILENUM : file number from fopen\n\ |
|
|
1676 DATA : matrix of elements to be written\n\ |
|
|
1677 PRECISION : type of data to read, valid types are\n\ |
|
|
1678 \n\ |
|
|
1679 'char', 'schar', 'short', 'int', 'long', 'float'\n\ |
|
|
1680 'double', 'uchar', 'ushort', 'uint', 'ulong'\n\ |
|
|
1681 \n\ |
|
|
1682 COUNT : number of elements written") |
|
|
1683 { |
|
|
1684 Octave_object retval; |
|
|
1685 |
|
|
1686 int nargin = args.length (); |
|
|
1687 |
|
|
1688 if (nargin < 3 || nargin > 4) |
|
|
1689 print_usage ("fwrite"); |
|
|
1690 else |
|
|
1691 retval = fwrite_internal (args, nargout); |
|
|
1692 |
|
|
1693 return retval; |
|
|
1694 } |
|
|
1695 |
|
444
|
1696 /* |
|
|
1697 * Write binary data to a file. |
|
|
1698 * |
|
|
1699 * count = fwrite (fid, data, 'precision') |
|
|
1700 * |
|
|
1701 * fid : file id from fopen |
|
|
1702 * Data : data to be written |
|
|
1703 * precision : type of output element. Can be: |
|
|
1704 * |
|
|
1705 * char, uchar, schar, short, ushort, int, uint, |
|
|
1706 * long, float, double |
|
|
1707 * |
|
|
1708 * Default is uchar. |
|
|
1709 * |
|
|
1710 * count : the number of elements written |
|
|
1711 */ |
|
497
|
1712 Octave_object |
|
506
|
1713 fwrite_internal (const Octave_object& args, int nargout) |
|
444
|
1714 { |
|
497
|
1715 Octave_object retval; |
|
444
|
1716 |
|
506
|
1717 int nargin = args.length (); |
|
|
1718 |
|
497
|
1719 Pix p = file_io_get_file (args(1), "a+", "fwrite"); |
|
444
|
1720 |
|
529
|
1721 if (! p) |
|
444
|
1722 return retval; |
|
|
1723 |
|
|
1724 // Get type and number of bytes per element to read. |
|
|
1725 char *prec = "uchar"; |
|
|
1726 if (nargin > 3) |
|
|
1727 { |
|
497
|
1728 if (args(3).is_string_type ()) |
|
|
1729 prec = args(3).string_value (); |
|
444
|
1730 else |
|
|
1731 { |
|
|
1732 error ("fwrite: precision must be a specified as a string"); |
|
|
1733 return retval; |
|
|
1734 } |
|
|
1735 } |
|
|
1736 |
|
|
1737 file_info file = file_list (p); |
|
|
1738 |
|
497
|
1739 Matrix m = args(2).to_matrix (); |
|
444
|
1740 |
|
471
|
1741 int count = m.write (file.fptr (), prec); |
|
444
|
1742 |
|
497
|
1743 retval.resize (1); |
|
|
1744 retval(0) = tree_constant ((double) count); |
|
444
|
1745 |
|
|
1746 return retval; |
|
|
1747 } |
|
|
1748 |
|
529
|
1749 DEFUN ("feof", Ffeof, Sfeof, 2, 1, |
|
|
1750 "ERROR = feof (FILENAME or FILENUM)\n\ |
|
|
1751 \n\ |
|
|
1752 Returns a non zero value for an end of file condition for the\n\ |
|
|
1753 file specified by FILENAME or FILENUM from fopen") |
|
|
1754 { |
|
|
1755 Octave_object retval; |
|
|
1756 |
|
|
1757 int nargin = args.length (); |
|
|
1758 |
|
|
1759 if (nargin != 2) |
|
|
1760 print_usage ("feof"); |
|
|
1761 else |
|
|
1762 retval = feof_internal (args, nargout); |
|
|
1763 |
|
|
1764 return retval; |
|
|
1765 } |
|
|
1766 |
|
444
|
1767 /* |
|
|
1768 * Check for an EOF condition on a file opened by fopen. |
|
|
1769 * |
|
|
1770 * eof = feof (fid) |
|
|
1771 * |
|
|
1772 * fid : file id from fopen |
|
|
1773 * eof : non zero for an end of file condition |
|
|
1774 */ |
|
497
|
1775 Octave_object |
|
506
|
1776 feof_internal (const Octave_object& args, int nargout) |
|
444
|
1777 { |
|
497
|
1778 Octave_object retval; |
|
444
|
1779 |
|
|
1780 // Get file info. |
|
497
|
1781 Pix p = return_valid_file (args(1)); |
|
444
|
1782 |
|
529
|
1783 if (! p) |
|
444
|
1784 return retval; |
|
|
1785 |
|
|
1786 file_info file = file_list (p); |
|
|
1787 |
|
497
|
1788 retval.resize (1); |
|
529
|
1789 retval(0) = (double) feof (file.fptr ()); |
|
|
1790 |
|
|
1791 return retval; |
|
|
1792 } |
|
|
1793 |
|
|
1794 DEFUN ("ferror", Fferror, Sferror, 2, 1, |
|
|
1795 "ERROR = ferror (FILENAME or FILENUM)\n\ |
|
|
1796 \n\ |
|
|
1797 Returns a non zero value for an error condition on the\n\ |
|
|
1798 file specified by FILENAME or FILENUM from fopen") |
|
|
1799 { |
|
|
1800 Octave_object retval; |
|
|
1801 |
|
|
1802 int nargin = args.length (); |
|
|
1803 |
|
|
1804 if (nargin != 2) |
|
|
1805 print_usage ("ferror"); |
|
|
1806 else |
|
|
1807 retval = ferror_internal (args, nargout); |
|
444
|
1808 |
|
|
1809 return retval; |
|
|
1810 } |
|
|
1811 |
|
|
1812 /* |
|
|
1813 * Check for an error condition on a file opened by fopen. |
|
|
1814 * |
|
|
1815 * [message, errnum] = ferror (fid) |
|
|
1816 * |
|
|
1817 * fid : file id from fopen |
|
|
1818 * message : system error message |
|
|
1819 * errnum : error number |
|
|
1820 */ |
|
497
|
1821 Octave_object |
|
506
|
1822 ferror_internal (const Octave_object& args, int nargout) |
|
444
|
1823 { |
|
497
|
1824 Octave_object retval; |
|
444
|
1825 |
|
|
1826 // Get file info. |
|
497
|
1827 Pix p = return_valid_file (args(1)); |
|
444
|
1828 |
|
529
|
1829 if (! p) |
|
444
|
1830 return retval; |
|
|
1831 |
|
|
1832 file_info file = file_list (p); |
|
|
1833 |
|
|
1834 int ierr = ferror (file.fptr ()); |
|
|
1835 |
|
|
1836 if (nargout > 1) |
|
|
1837 { |
|
497
|
1838 retval.resize (2); |
|
|
1839 retval(1) = tree_constant ((double) ierr); |
|
444
|
1840 } |
|
|
1841 else |
|
497
|
1842 retval.resize (1); |
|
444
|
1843 |
|
497
|
1844 retval(0) = tree_constant (strsave (strerror (ierr))); |
|
444
|
1845 |
|
|
1846 return retval; |
|
|
1847 } |
|
|
1848 |
|
|
1849 /* |
|
1
|
1850 ;;; Local Variables: *** |
|
|
1851 ;;; mode: C++ *** |
|
|
1852 ;;; page-delimiter: "^/\\*" *** |
|
|
1853 ;;; End: *** |
|
|
1854 */ |
