/*
 * Copyright (C) 2015 Jared Boone, ShareBrained Technology, Inc.
 * Copyright (C) 2017 Furrtek
 * 
 * This file is part of PortaPack.
 *
 * 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, 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; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */

#include "ui.hpp"
#include "ui_navigation.hpp"
#include "encoders.hpp"

using namespace portapack;

namespace encoders {

size_t make_bitstream(std::string& fragments) {
	uint8_t byte = 0;
	size_t bitstream_length = 0;
	uint8_t * bitstream = shared_memory.bb_data.data;
	
	for (auto c : fragments) {
		byte <<= 1;
		if (c != '0')
			byte |= 1;
		
		if ((bitstream_length & 7) == 7)
			bitstream[bitstream_length >> 3] = byte;
		
		bitstream_length++;
	}
	
	// Finish last byte if needed
	size_t padding =  8 - (bitstream_length & 7);
	if (padding != 8) {
		byte <<= padding;
		bitstream[(bitstream_length + padding - 1) >> 3] = byte;
		padding++;
	}
	
	return bitstream_length;
}

void bitstream_append(size_t& bitstream_length, uint32_t bit_count, uint32_t bits) {
	uint8_t * bitstream = shared_memory.bb_data.data;
	uint32_t bit_mask = 1 << (bit_count - 1);
	uint32_t bit_index;
	uint8_t byte = 0;
	
	if (bitstream_length & 7)
		byte = bitstream[bitstream_length >> 3];
	
	bit_index = 7 - (bitstream_length & 7);
	
	for (size_t i = 0; i < bit_count; i++) {
		if (bits & bit_mask)
			byte |= (1 << bit_index);
		
		if (!bit_index) {
			bitstream[bitstream_length >> 3] = byte;
			byte = 0;
		}
				
		bit_index = (bit_index - 1) & 7;
		bits <<= 1;
		
		bitstream_length++;
	}
	
	bitstream[bitstream_length >> 3] = byte;
}
	
} /* namespace encoders */