//
// 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 : Sat Dec  9 09:31:23 2017
// Update Count     : 275
//

#include "fstream"

#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.h>

#define IO_MSG "I/O error: "

void ?{}( ofstream & os, void * file, _Bool sepDefault, _Bool sepOnOff, const char * separator, const char * tupleSeparator ) {
	os.file = file;
	os.sepDefault = sepDefault;
	os.sepOnOff = sepOnOff;
	sepSet( os, separator );
	sepSetCur( os, sepGet( os ) );
	sepSetTuple( os, 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 ?{}( ofstream & os ) { os.file = 0; }

void ?{}( ofstream & os, const char * name, const char * mode ) {
	open( os, name, mode );
}
void ?{}( ofstream & os, const char * name ) {
	open( os, name, "w" );
}

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 os.file == 0 || 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 open( ofstream & os, const char * name ) {
	open( os, name, "w" );
} // 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;


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

// private
void ?{}( ifstream & is, void * file ) {
	is.file = file;
}

// public
void ?{}( ifstream & is ) {	is.file = 0; }

void ?{}( ifstream & is, const char * name, const char * mode ) {
	open( is, name, mode );
}
void ?{}( ifstream & is, const char * name ) {
	open( is, name, "r" );
}

int fail( ifstream & is ) {
	return is.file == 0 || 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 open( ifstream & is, const char * name ) {
	open( is, name, "r" );
} // 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: //