/*
 * Copyright (C) 2007-2012 Siemens AG
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

/*******************************************************************
 *
 * @author Daniel.Peintner.EXT@siemens.com
 * @version 0.6 
 * @contact Joerg.Heuer@siemens.com
 *
 * <p>Code generated by EXIdizer</p>
 ********************************************************************/



#include "EXITypes.h"
#include "BitOutputStream.h"

#ifndef BIT_OUTPUT_STREAM_C
#define BIT_OUTPUT_STREAM_C

/*	NOTE: nbits <= 8 */
int writeBits(bitstream_t* stream, uint16_t nbits, uint32_t val) {
	/*  is there enough space in the buffer */
	if (nbits <= stream->capacity) {
		/* all bits fit into the current buffer */
		stream->buffer = (stream->buffer << (nbits)) | (val & (0xff
				>> (BITS_IN_BYTE - nbits)));
		stream->capacity -= nbits;
		/* if the buffer is full write byte */
		if (stream->capacity == 0) {
#if EXI_STREAM == BYTE_ARRAY
			if ((*stream->pos) >= stream->size) {
				return EXI_ERROR_OUTPUT_STREAM_EOF;
			}
			stream->data[(*stream->pos)++] = stream->buffer;
#endif
#if EXI_STREAM == FILE_STREAM
			if ( putc(stream->buffer, stream->file) == EOF ) {
				return EXI_ERROR_OUTPUT_STREAM_EOF;
			}
#endif
			stream->capacity = BITS_IN_BYTE;
			stream->buffer = 0;
		}
	} else {
		/* the buffer is not enough
		 * fill the buffer */
		stream->buffer = (stream->buffer << stream->capacity) | ((val >> (nbits
				- stream->capacity)) & (0xff >> (BITS_IN_BYTE
				- stream->capacity)));

		nbits -= stream->capacity;
#if EXI_STREAM == BYTE_ARRAY
		if ((*stream->pos) >= stream->size) {
			return EXI_ERROR_OUTPUT_STREAM_EOF;
		}
		stream->data[(*stream->pos)++] = stream->buffer;
#endif
#if EXI_STREAM == FILE_STREAM
		if ( putc(stream->buffer, stream->file) == EOF ) {
			return EXI_ERROR_OUTPUT_STREAM_EOF;
		}
#endif
		stream->buffer = 0;

		/* write whole bytes */
		while (nbits >= BITS_IN_BYTE) {
			nbits -= BITS_IN_BYTE;
#if EXI_STREAM == BYTE_ARRAY
			if ((*stream->pos) >= stream->size) {
				return EXI_ERROR_OUTPUT_STREAM_EOF;
			}
			stream->data[(*stream->pos)++] = (val >> (nbits));
#endif
#if EXI_STREAM == FILE_STREAM
			if ( putc((val >> (nbits)), stream->file) == EOF ) {
				return EXI_ERROR_OUTPUT_STREAM_EOF;
			}
#endif
		}

		/* spared bits are kept in the buffer */
		stream->buffer = val; /* Note: the high bits will be shifted out during further filling */
		stream->capacity = BITS_IN_BYTE - (nbits);
	}

	return 0;
}

/**
 * Flush output
 */
int flush(bitstream_t* stream) {
	if (stream->capacity == BITS_IN_BYTE) {
		/* nothing to do, no bits in buffer */
		return 0;
	} else {
		return writeBits(stream, stream->capacity, 0);
	}
}

#endif