source: src/libcfa/fstream.c @ a0b3e32

//
// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
//
// The contents of this file are covered under the licence agreement in the
// file "LICENCE" distributed with Cforall.
//
// fstream.c --
//
// Author           : Peter A. Buhr
// Created On       : Wed May 27 17:56:53 2015
// Last Modified By : Peter A. Buhr
// Last Modified On : Thu Jul  6 18:38:25 2017
// Update Count     : 251
//

#include "fstream"

extern "C" {
#include <stdio.h>                                                                              // vfprintf, vfscanf
#include <stdlib.h>                                                                             // exit
#include <stdarg.h>                                                                             // varargs
#include <string.h>                                                                             // strlen
#include <stdbool.h>                                                                    // true/false
#include <float.h>                                                                              // DBL_DIG, LDBL_DIG
#include <complex.h>                                                                    // creal, cimag
}
#include "assert"

#define IO_MSG "I/O error: "

void ?{}( ofstream * this, void * file, _Bool sepDefault, _Bool sepOnOff, const char * separator, const char * tupleSeparator ) {
        this->file = file;
        this->sepDefault = sepDefault;
        this->sepOnOff = sepOnOff;
        sepSet( this, separator );
        sepSetCur( this, sepGet( this ) );
        sepSetTuple( this, tupleSeparator );
}

// private
_Bool sepPrt( ofstream * os ) { setNL( os, false ); return os->sepOnOff; }
void sepReset( ofstream * os ) { os->sepOnOff = os->sepDefault; }
void sepReset( ofstream * os, _Bool reset ) { os->sepDefault = reset; os->sepOnOff = os->sepDefault; }
const char * sepGetCur( ofstream * os ) { return os->sepCur; }
void sepSetCur( ofstream * os, const char * sepCur ) { os->sepCur = sepCur; }
_Bool getNL( ofstream * os ) { return os->sawNL; }
void setNL( ofstream * os, _Bool state ) { os->sawNL = state; }

// public
void sepOn( ofstream * os ) { os->sepOnOff = ! getNL( os ); }
void sepOff( ofstream * os ) { os->sepOnOff = false; }

_Bool sepDisable( ofstream *os ) {
        _Bool temp = os->sepDefault;
        os->sepDefault = false;
        sepReset( os );
        return temp;
} // sepDisable

_Bool sepEnable( ofstream *os ) {
        _Bool temp = os->sepDefault;
        os->sepDefault = true;
        if ( os->sepOnOff ) sepReset( os );                                     // start of line ?
        return temp;
} // sepEnable

const char * sepGet( ofstream * os ) { return os->separator; }
void sepSet( ofstream * os, const char * s ) {
        assert( s );
        strncpy( os->separator, s, sepSize - 1 );
        os->separator[sepSize - 1] = '\0';
} // sepSet

const char * sepGetTuple( ofstream * os ) { return os->tupleSeparator; }
void sepSetTuple( ofstream * os, const char * s ) {
        assert( s );
        strncpy( os->tupleSeparator, s, sepSize - 1 );
        os->tupleSeparator[sepSize - 1] = '\0';
} // sepSet

int fail( ofstream * os ) {
        return ferror( (FILE *)(os->file) );
} // fail

int flush( ofstream * os ) {
        return fflush( (FILE *)(os->file) );
} // flush

void open( ofstream * os, const char * name, const char * mode ) {
        FILE *file = fopen( name, mode );
        if ( file == 0 ) {                                                                      // do not change unless successful
                fprintf( stderr, IO_MSG "open output file \"%s\", ", name );
                perror( 0 );
                exit( EXIT_FAILURE );
        } // if
        ?{}( os, file, true, false, " ", ", " );
} // open

void close( ofstream * os ) {
        if ( (FILE *)(os->file) == stdout || (FILE *)(os->file) == stderr ) return;

        if ( fclose( (FILE *)(os->file) ) == EOF ) {
                perror( IO_MSG "close output" );
        } // if
} // close

ofstream * write( ofstream * os, const char * data, unsigned long int size ) {
        if ( fail( os ) ) {
                fprintf( stderr, "attempt write I/O on failed stream\n" );
                exit( EXIT_FAILURE );
        } // if

        if ( fwrite( data, 1, size, (FILE *)(os->file) ) != size ) {
                perror( IO_MSG "write" );
                exit( EXIT_FAILURE );
        } // if
        return os;
} // write

int fmt( ofstream * os, const char format[], ... ) {
        va_list args;
        va_start( args, format );
        int len = vfprintf( (FILE *)(os->file), format, args );
        if ( len == EOF ) {
                if ( ferror( (FILE *)(os->file) ) ) {
                        fprintf( stderr, "invalid write\n" );
                        exit( EXIT_FAILURE );
                } // if
        } // if
        va_end( args );

        sepReset( os );                                                                         // reset separator
        return len;
} // fmt

static ofstream soutFile = { (FILE *)(&_IO_2_1_stdout_), true, false, " ", ", " };
ofstream *sout = &soutFile;
static ofstream serrFile = { (FILE *)(&_IO_2_1_stderr_), true, false, " ", ", " };
ofstream *serr = &serrFile;


//---------------------------------------


int fail( ifstream * is ) {
        return ferror( (FILE *)(is->file) );
} // fail

int eof( ifstream * is ) {
        return feof( (FILE *)(is->file) );
} // eof

void open( ifstream * is, const char * name, const char * mode ) {
        FILE *file = fopen( name, mode );
        if ( file == 0 ) {                                                                      // do not change unless successful
                fprintf( stderr, IO_MSG "open input file \"%s\", ", name );
                perror( 0 );
                exit( EXIT_FAILURE );
        } // if
        is->file = file;
} // open

void close( ifstream * is ) {
        if ( (FILE *)(is->file) == stdin ) return;

        if ( fclose( (FILE *)(is->file) ) == EOF ) {
                perror( IO_MSG "close input" );
        } // if
} // close

ifstream * read( ifstream * is, char * data, unsigned long int size ) {
        if ( fail( is ) ) {
                fprintf( stderr, "attempt read I/O on failed stream\n" );
                exit( EXIT_FAILURE );
        } // if

        if ( fread( data, size, 1, (FILE *)(is->file) ) == 0 ) {
                perror( IO_MSG "read" );
                exit( EXIT_FAILURE );
        } // if
        return is;
} // read

ifstream *ungetc( ifstream * is, char c ) {
        if ( fail( is ) ) {
                fprintf( stderr, "attempt ungetc I/O on failed stream\n" );
                exit( EXIT_FAILURE );
        } // if

        if ( ungetc( c, (FILE *)(is->file) ) == EOF ) {
                perror( IO_MSG "ungetc" );
                exit( EXIT_FAILURE );
        } // if
        return is;
} // ungetc

int fmt( ifstream * is, const char format[], ... ) {
        va_list args;

        va_start( args, format );
        int len = vfscanf( (FILE *)(is->file), format, args );
        if ( len == EOF ) {
                if ( ferror( (FILE *)(is->file) ) ) {
                        fprintf( stderr, "invalid read\n" );
                        exit( EXIT_FAILURE );
                } // if
        } // if
        va_end( args );
        return len;
} // fmt


static ifstream sinFile = { (FILE *)(&_IO_2_1_stdin_) };
ifstream *sin = &sinFile;

// Local Variables: //
// tab-width: 4 //
// End: //