source: libcfa/src/fstream.cfa @ 200fcb3

aaron-thesisarm-ehcleanup-dtorsdeferred_resnjacob/cs343-translationjenkins-sandboxnew-astnew-ast-unique-exprno_listpersistent-indexer
Last change on this file since 200fcb3 was 200fcb3, checked in by Peter A. Buhr <pabuhr@…>, 3 years ago

add auto newline to sout, change endl to nl

  • Property mode set to 100644
File size: 6.7 KB
Line 
1//
2// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
3//
4// The contents of this file are covered under the licence agreement in the
5// file "LICENCE" distributed with Cforall.
6//
7// fstream.c --
8//
9// Author           : Peter A. Buhr
10// Created On       : Wed May 27 17:56:53 2015
11// Last Modified By : Peter A. Buhr
12// Last Modified On : Wed Dec 12 08:34:28 2018
13// Update Count     : 298
14//
15
16#include "fstream.hfa"
17
18#include <stdio.h>                                                                              // vfprintf, vfscanf
19#include <stdlib.h>                                                                             // exit
20#include <stdarg.h>                                                                             // varargs
21#include <string.h>                                                                             // strlen
22#include <float.h>                                                                              // DBL_DIG, LDBL_DIG
23#include <complex.h>                                                                    // creal, cimag
24#include <assert.h>
25
26#define IO_MSG "I/O error: "
27
28void ?{}( ofstream & os, void * file, bool sepDefault, bool sepOnOff, bool nlOnOff, bool nonlManip, const char * separator, const char * tupleSeparator ) {
29        os.file = file;
30        os.sepDefault = sepDefault;
31        os.sepOnOff = sepOnOff;
32        os.nlOnOff = nlOnOff;
33        os.nonlManip = nonlManip;
34        sepSet( os, separator );
35        sepSetCur( os, sepGet( os ) );
36        sepSetTuple( os, tupleSeparator );
37}
38
39// private
40bool sepPrt( ofstream & os ) { setNL( os, false ); return os.sepOnOff; }
41void sepReset( ofstream & os ) { os.sepOnOff = os.sepDefault; }
42void sepReset( ofstream & os, bool reset ) { os.sepDefault = reset; os.sepOnOff = os.sepDefault; }
43const char * sepGetCur( ofstream & os ) { return os.sepCur; }
44void sepSetCur( ofstream & os, const char * sepCur ) { os.sepCur = sepCur; }
45bool getNL( ofstream & os ) { return os.sawNL; }
46void setNL( ofstream & os, bool state ) { os.sawNL = state; }
47bool getANL( ofstream & os ) { return os.nlOnOff; }
48bool getNonl( ofstream & os ) { return os.nonlManip; }
49void setNonl( ofstream & os, bool state ) { os.nonlManip = state; }
50
51// public
52void ?{}( ofstream & os ) { os.file = 0; }
53
54void ?{}( ofstream & os, const char * name, const char * mode ) {
55        open( os, name, mode );
56}
57void ?{}( ofstream & os, const char * name ) {
58        open( os, name, "w" );
59}
60
61void sepOn( ofstream & os ) { os.sepOnOff = ! getNL( os ); }
62void sepOff( ofstream & os ) { os.sepOnOff = false; }
63
64bool sepDisable( ofstream & os ) {
65        bool temp = os.sepDefault;
66        os.sepDefault = false;
67        sepReset( os );
68        return temp;
69} // sepDisable
70
71bool sepEnable( ofstream & os ) {
72        bool temp = os.sepDefault;
73        os.sepDefault = true;
74        if ( os.sepOnOff ) sepReset( os );                                      // start of line ?
75        return temp;
76} // sepEnable
77
78void nlOn( ofstream & os ) { os.nlOnOff = true; }
79void nlOff( ofstream & os ) { os.nlOnOff = false; }
80
81const char * sepGet( ofstream & os ) { return os.separator; }
82void sepSet( ofstream & os, const char * s ) {
83        assert( s );
84        strncpy( os.separator, s, sepSize - 1 );
85        os.separator[sepSize - 1] = '\0';
86} // sepSet
87
88const char * sepGetTuple( ofstream & os ) { return os.tupleSeparator; }
89void sepSetTuple( ofstream & os, const char * s ) {
90        assert( s );
91        strncpy( os.tupleSeparator, s, sepSize - 1 );
92        os.tupleSeparator[sepSize - 1] = '\0';
93} // sepSet
94
95int fail( ofstream & os ) {
96        return os.file == 0 || ferror( (FILE *)(os.file) );
97} // fail
98
99int flush( ofstream & os ) {
100        return fflush( (FILE *)(os.file) );
101} // flush
102
103void open( ofstream & os, const char * name, const char * mode ) {
104        FILE *file = fopen( name, mode );
105        #ifdef __CFA_DEBUG__
106        if ( file == 0 ) {
107                fprintf( stderr, IO_MSG "open output file \"%s\", ", name );
108                perror( 0 );
109                exit( EXIT_FAILURE );
110        } // if
111        #endif // __CFA_DEBUG__
112        (os){ file, true, false, true, false, " ", ", " };
113} // open
114
115void open( ofstream & os, const char * name ) {
116        open( os, name, "w" );
117} // open
118
119void close( ofstream & os ) {
120        if ( (FILE *)(os.file) == stdout || (FILE *)(os.file) == stderr ) return;
121
122        if ( fclose( (FILE *)(os.file) ) == EOF ) {
123                perror( IO_MSG "close output" );
124        } // if
125} // close
126
127ofstream & write( ofstream & os, const char * data, size_t size ) {
128        if ( fail( os ) ) {
129                fprintf( stderr, "attempt write I/O on failed stream\n" );
130                exit( EXIT_FAILURE );
131        } // if
132
133        if ( fwrite( data, 1, size, (FILE *)(os.file) ) != size ) {
134                perror( IO_MSG "write" );
135                exit( EXIT_FAILURE );
136        } // if
137        return os;
138} // write
139
140int fmt( ofstream & os, const char format[], ... ) {
141        va_list args;
142        va_start( args, format );
143        int len = vfprintf( (FILE *)(os.file), format, args );
144        if ( len == EOF ) {
145                if ( ferror( (FILE *)(os.file) ) ) {
146                        fprintf( stderr, "invalid write\n" );
147                        exit( EXIT_FAILURE );
148                } // if
149        } // if
150        va_end( args );
151
152        sepReset( os );                                                                         // reset separator
153        return len;
154} // fmt
155
156static ofstream soutFile = { (FILE *)(&_IO_2_1_stdout_), true, false, true, false, " ", ", " };
157ofstream & sout = soutFile;
158static ofstream serrFile = { (FILE *)(&_IO_2_1_stderr_), true, false, true, false, " ", ", " };
159ofstream & serr = serrFile;
160
161
162//---------------------------------------
163
164// private
165void ?{}( ifstream & is, void * file ) {
166        is.file = file;
167}
168
169// public
170void ?{}( ifstream & is ) {     is.file = 0; }
171
172void ?{}( ifstream & is, const char * name, const char * mode ) {
173        open( is, name, mode );
174}
175void ?{}( ifstream & is, const char * name ) {
176        open( is, name, "r" );
177}
178
179int fail( ifstream & is ) {
180        return is.file == 0 || ferror( (FILE *)(is.file) );
181} // fail
182
183int eof( ifstream & is ) {
184        return feof( (FILE *)(is.file) );
185} // eof
186
187void open( ifstream & is, const char * name, const char * mode ) {
188        FILE *file = fopen( name, mode );
189        #ifdef __CFA_DEBUG__
190        if ( file == 0 ) {
191                fprintf( stderr, IO_MSG "open input file \"%s\", ", name );
192                perror( 0 );
193                exit( EXIT_FAILURE );
194        } // if
195        #endif // __CFA_DEBUG__
196        is.file = file;
197} // open
198
199void open( ifstream & is, const char * name ) {
200        open( is, name, "r" );
201} // open
202
203void close( ifstream & is ) {
204        if ( (FILE *)(is.file) == stdin ) return;
205
206        if ( fclose( (FILE *)(is.file) ) == EOF ) {
207                perror( IO_MSG "close input" );
208        } // if
209} // close
210
211ifstream & read( ifstream & is, char * data, size_t size ) {
212        if ( fail( is ) ) {
213                fprintf( stderr, "attempt read I/O on failed stream\n" );
214                exit( EXIT_FAILURE );
215        } // if
216
217        if ( fread( data, size, 1, (FILE *)(is.file) ) == 0 ) {
218                perror( IO_MSG "read" );
219                exit( EXIT_FAILURE );
220        } // if
221        return is;
222} // read
223
224ifstream &ungetc( ifstream & is, char c ) {
225        if ( fail( is ) ) {
226                fprintf( stderr, "attempt ungetc I/O on failed stream\n" );
227                exit( EXIT_FAILURE );
228        } // if
229
230        if ( ungetc( c, (FILE *)(is.file) ) == EOF ) {
231                perror( IO_MSG "ungetc" );
232                exit( EXIT_FAILURE );
233        } // if
234        return is;
235} // ungetc
236
237int fmt( ifstream & is, const char format[], ... ) {
238        va_list args;
239
240        va_start( args, format );
241        int len = vfscanf( (FILE *)(is.file), format, args );
242        if ( len == EOF ) {
243                if ( ferror( (FILE *)(is.file) ) ) {
244                        fprintf( stderr, "invalid read\n" );
245                        exit( EXIT_FAILURE );
246                } // if
247        } // if
248        va_end( args );
249        return len;
250} // fmt
251
252
253static ifstream sinFile = { (FILE *)(&_IO_2_1_stdin_) };
254ifstream & sin = sinFile;
255
256// Local Variables: //
257// tab-width: 4 //
258// End: //
Note: See TracBrowser for help on using the repository browser.