/*
OFDM modem common bits

Copyright 2025 Ahmet Inan <inan@aicodix.de>
*/

#pragma once

#include "crc.hh"
#include "polar_tables.hh"
#include "hadamard_encoder.hh"

struct Common
{
	static const int mod_max = 8;
	static const int code_max = 15;
	static const int bits_max = 1 << code_max;
	static const int data_max = 2048;
	static const int symbols_max = 22;
	static const int mls0_poly = 0x331;
	static const int mls0_seed = 214;
	static const int mls1_poly = 0x43;
	static const int mls2_poly = 0x163;
	static const int data_tones = 256;
	static const int meta_tones = 32;
	static const int seed_tones = 32;
	static const int tone_count = data_tones + meta_tones + seed_tones;
	static const int block_length = 10;
	static const int block_skew = 3;
	static const int first_meta = 4;
	static const int first_seed = 9;
	static constexpr int slm_poly[16] = {
			0x11d, 0x12b, 0x12d, 0x14d, 0x15f, 0x163, 0x165, 0x169,
			0x171, 0x187, 0x18d, 0x1a9, 0x1c3, 0x1cf, 0x1e7, 0x1f5 };
	CODE::CRC<uint32_t> crc0;
	CODE::HadamardEncoder<6> hadamard_encoder;
	int8_t meta[32];
	int8_t seed[32];
	uint8_t data[data_max];
	const uint32_t *frozen_bits;
	int mod_bits;
	int data_bits;
	int data_bytes;
	int code_order;
	int oper_mode;
	int tone_off;
	int meta_off;
	int seed_off;
	int symbol_count;
	bool differential;

	Common() : crc0(0x8F6E37A0) {}

	void setup(int mode)
	{
		switch (mode) {
		case 0:
			mod_bits = 1;
			symbol_count = 16;
			differential = true;
			code_order = 12;
			data_bits = 2048;
			frozen_bits = frozen_4096_2080;
			break;
		case 1:
			mod_bits = 2;
			symbol_count = 16;
			differential = true;
			code_order = 13;
			data_bits = 4096;
			frozen_bits = frozen_8192_4128;
			break;
		case 2:
			mod_bits = 3;
			symbol_count = 22;
			differential = true;
			code_order = 14;
			data_bits = 8192;
			frozen_bits = frozen_16384_8224;
			break;
		case 3:
			mod_bits = 1;
			symbol_count = 16;
			differential = false;
			code_order = 12;
			data_bits = 2048;
			frozen_bits = frozen_4096_2080;
			break;
		case 4:
			mod_bits = 2;
			symbol_count = 16;
			differential = false;
			code_order = 13;
			data_bits = 4096;
			frozen_bits = frozen_8192_4128;
			break;
		case 5:
			mod_bits = 4;
			symbol_count = 4;
			differential = false;
			code_order = 12;
			data_bits = 2048;
			frozen_bits = frozen_4096_2080;
			break;
		case 6:
			mod_bits = 4;
			symbol_count = 16;
			differential = false;
			code_order = 14;
			data_bits = 8192;
			frozen_bits = frozen_16384_8224;
			break;
		case 7:
			mod_bits = 6;
			symbol_count = 22;
			differential = false;
			code_order = 15;
			data_bits = 16384;
			frozen_bits = frozen_32768_16416;
			break;
		default:
			return;
		}
		oper_mode = mode;
		data_bytes = data_bits / 8;
	}
};