/*
 * LinuxDVB Output handling.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 *
 */

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <memory.h>
#include <asm/types.h>
#include <errno.h>

#include "misc.h"

void PutBits(BitPacker_t * ld, unsigned int code, unsigned int length)
{
	unsigned int bit_buf;
	unsigned int bit_left;

	bit_buf = ld->BitBuffer;
	bit_left = ld->Remaining;

#ifdef DEBUG_PUTBITS
	if (ld->debug)
		dprintf("code = %d, length = %d, bit_buf = 0x%x, bit_left = %d\n",
			code, length, bit_buf, bit_left);
#endif

	if (length < bit_left) {
		/* fits into current buffer */
		bit_buf = (bit_buf << length) | code;
		bit_left -= length;
	} else {
		/* doesn't fit */
		bit_buf <<= bit_left;
		bit_buf |= code >> (length - bit_left);
		ld->Ptr[0] = (uint8_t) (bit_buf >> 24);
		ld->Ptr[1] = (uint8_t) (bit_buf >> 16);
		ld->Ptr[2] = (uint8_t) (bit_buf >> 8);
		ld->Ptr[3] = (uint8_t) bit_buf;
		ld->Ptr += 4;
		length -= bit_left;
		bit_buf = code & ((1 << length) - 1);
		bit_left = 32 - length;
		bit_buf = code;
	}

#ifdef DEBUG_PUTBITS
	if (ld->debug)
		dprintf("bit_left = %d, bit_buf = 0x%x\n", bit_left, bit_buf);
#endif

	/* writeback */
	ld->BitBuffer = bit_buf;
	ld->Remaining = bit_left;
}

void FlushBits(BitPacker_t * ld)
{
	ld->BitBuffer <<= ld->Remaining;
	while (ld->Remaining < 32) {
#ifdef DEBUG_PUTBITS
		if (ld->debug)
			dprintf("flushing 0x%2.2x\n", ld->BitBuffer >> 24);
#endif
		*ld->Ptr++ = ld->BitBuffer >> 24;
		ld->BitBuffer <<= 8;
		ld->Remaining += 8;
	}
	ld->Remaining = 32;
	ld->BitBuffer = 0;
}