aicodix___modem/freezer.cc
2023-01-04 12:13:04 +01:00

40 lines
1.1 KiB
C++

/*
Table generator for frozen bits
Copyright 2021 Ahmet Inan <inan@aicodix.de>
*/
#include <cmath>
#include <cassert>
#include <iostream>
#include <algorithm>
#include <functional>
#include "polar_freezer.hh"
template <int M>
void code(int N, int K)
{
long double erasure_probability = (long double)(N - K) / N;
double design_SNR = 10 * std::log10(-std::log(erasure_probability));
std::cerr << "design SNR: " << design_SNR << std::endl;
auto freeze = new CODE::PolarCodeConst0<M>;
double better_SNR = design_SNR + 1.59175;
std::cerr << "better SNR: " << better_SNR << std::endl;
long double better_probability = std::exp(-pow(10.0, better_SNR / 10));
auto frozen = new uint32_t[1<<(M-5)];
(*freeze)(frozen, M, K+(1<<M)-N, better_probability);
delete freeze;
std::cout << "static const uint32_t frozen_" << std::dec << N << "_" << K << "[" << (1<<(M-5)) << "] = { " << std::hex;
for (int i = 0; i < 1<<(M-5); ++i)
std::cout << "0x" << frozen[i] << ", ";
std::cout << "};" << std::endl;
}
int main()
{
code<11>(2048, 1360+32);
code<11>(2048, 1024+32);
code<11>(2048, 680+32);
return 0;
}