// Test to make sure that Longhair works well with input ordered like this for
// k = 4 and m = 2
// 0
// 2
// 3
// 5
// 6
void order_test() {
int block_bytes = 8 * 162; // a multiple of 8
Abyssinian prng;
prng.Initialize(m_clock.msec(), Clock::cycles());
int block_count = 4;
int recovery_block_count = 2;
u8 *data = new u8[block_bytes * block_count];
u8 *recovery_blocks = new u8[block_bytes * recovery_block_count];
Block *blocks = new Block[block_count];
const u8 *data_ptrs[256];
for (int ii = 0; ii < block_count; ++ii) {
data_ptrs[ii] = data + ii * block_bytes;
}
for (int ii = 0; ii < block_bytes * block_count; ++ii) {
data[ii] = (u8)prng.Next();
}
assert(!cauchy_256_encode(block_count, recovery_block_count, data_ptrs, recovery_blocks, block_bytes));
for (int ii = 0; ii < block_count; ++ii) {
blocks[ii].data = (u8*)data_ptrs[ii];
blocks[ii].row = (u8)ii;
}
int rem = block_count;
for (int ii = 0; ii < recovery_block_count; ++ii) {
int jj = prng.Next() % rem;
--rem;
for (int kk = jj; kk < rem; ++kk) {
blocks[kk].data = blocks[kk + 1].data;
blocks[kk].row = blocks[kk + 1].row;
}
blocks[rem].data = recovery_blocks + ii * block_bytes;
blocks[rem].row = block_count + ii;
}
cout << "Before decode:" << endl;
for (int ii = 0; ii < block_count; ++ii) {
cout << (int)blocks[ii].row << endl;
}
assert(!cauchy_256_decode(block_count, recovery_block_count, blocks, block_bytes));
cout << "After decode:" << endl;
for (int ii = 0; ii < block_count; ++ii) {
cout << (int)blocks[ii].row << endl;
}
for (int ii = 0; ii < block_count; ++ii) {
assert(!memcmp(blocks[ii].data, data_ptrs[blocks[ii].row], block_bytes));
}
}
int main() {
m_clock.OnInitialize();
cauchy_256_init();
m_clock.usec();
cout << "Cauchy RS Codec Unit Tester" << endl;
order_test();
int block_bytes = 8 * 162; // a multiple of 8
cout << "Using " << block_bytes << " bytes per block (ie. packet/chunk size); must be a multiple of 8 bytes" << endl;
Abyssinian prng;
prng.Initialize(m_clock.msec(), Clock::cycles());
u8 heat_map[256 * 256] = { 0 };
for (int block_count = 1; block_count < 256; ++block_count) {
#ifdef CAT_REASONABLE_RECOVERY_COUNT
for (int recovery_block_count = 1; recovery_block_count <= (block_count / 2) && recovery_block_count < (256 - block_count); ++recovery_block_count) {
#else
for (int recovery_block_count = 1; recovery_block_count < (256 - block_count); ++recovery_block_count) {
#endif
u8 *data = new u8[block_bytes * block_count];
u8 *recovery_blocks = new u8[block_bytes * recovery_block_count];
Block *blocks = new Block[block_count];
const u8 *data_ptrs[256];
for (int ii = 0; ii < block_count; ++ii) {
data_ptrs[ii] = data + ii * block_bytes;
}
double sum_encode = 0;
int erasures_count;
#ifdef CAT_WORST_CASE_BENCHMARK
erasures_count = recovery_block_count;
if (block_count < erasures_count)
{
erasures_count = block_count;
}
{
#else
# ifdef CAT_ENCODE_TIMES_ONLY
for (erasures_count = 1; erasures_count <= 4 && erasures_count <= recovery_block_count && erasures_count <= block_count; ++erasures_count) {
# else
for (erasures_count = 1; erasures_count <= recovery_block_count && erasures_count <= block_count; ++erasures_count) {
# endif
#endif
for (int ii = 0; ii < block_bytes * block_count; ++ii) {
data[ii] = (u8)prng.Next();
}
double t0 = m_clock.usec();
assert(!cauchy_256_encode(block_count, recovery_block_count, data_ptrs, recovery_blocks, block_bytes));
double t1 = m_clock.usec();
double encode_time = t1 - t0;
sum_encode += encode_time;
cout << "Encoded k=" << block_count << " data blocks with m=" << recovery_block_count << " recovery blocks in " << encode_time << " usec : " << (block_bytes * block_count / encode_time) << " MB/s" << endl;
#ifndef CAT_ENCODE_TIMES_ONLY
for (int ii = 0; ii < erasures_count; ++ii) {
int erasure_index = recovery_block_count - ii - 1;
blocks[ii].data = recovery_blocks + erasure_index * block_bytes;
blocks[ii].row = block_count + erasure_index;
}
for (int ii = erasures_count; ii < block_count; ++ii) {
blocks[ii].data = data + ii * block_bytes;
blocks[ii].row = ii;
}
t0 = m_clock.usec();
assert(!cauchy_256_decode(block_count, recovery_block_count, blocks, block_bytes));
t1 = m_clock.usec();
double decode_time = t1 - t0;
cout << "+ Decoded " << erasures_count << " erasures in " << decode_time << " usec : " << (block_bytes * block_count / decode_time) << " MB/s" << endl;
/*
for (int ii = 0; ii < erasures_count; ++ii) {
const u8 *orig = data + ii * block_bytes;
print(orig, block_bytes / 8);
print(blocks[ii].data, block_bytes / 8);
}
*/
for (int ii = 0; ii < erasures_count; ++ii) {
const u8 *orig = data + ii * block_bytes;
assert(!memcmp(blocks[ii].data, orig, block_bytes));
}
#endif // CAT_ENCODE_TIMES_ONLY
}
double avg_encode = sum_encode / erasures_count;
int speed = block_bytes * block_count / avg_encode;
u8 map_value = 0;
if (speed < 10) {
map_value = 1;
} else if (speed < 50) {
map_value = 2;
} else if (speed < 100) {
map_value = 3;
} else if (speed < 200) {
map_value = 4;
} else if (speed < 300) {
map_value = 5;
} else if (speed < 400) {
map_value = 6;
} else if (speed < 500) {
map_value = 7;
} else {
map_value = 8;
}
heat_map[block_count * 256 + recovery_block_count] = map_value;
delete []data;
delete []recovery_blocks;
delete []blocks;
}
}
ofstream file;
file.open("docs/heatmap.txt");
for (int ii = 0; ii < 256; ++ii) {
for (int jj = 0; jj < 256; ++jj) {
u8 map_value = heat_map[ii * 256 + jj];
file << (int)map_value << " ";
}
file << endl;
}
m_clock.OnFinalize();
return 0;
}
