kate/BadFish
1
0
Fork 0
mirror of https://github.com/sockspls/badfish synced 2025-04-30 08:43:09 +00:00
Code Issues Releases Wiki Activity

Streamline calc_symlen()

Browse source
This commit is contained in:
Marco Costalba 2016-05-14 12:09:00 +02:00
parent 976581339f
commit 1a88103633
1 changed files with 12 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/syzygy/tbprobe.cpp
View file

@ -926,25 +926,24 @@ void set_norms(T* p, int num, const uint8_t pawns[])
++p->norm[i]; ++p->norm[i];
} }
void calc_symlen(PairsData* d, size_t s, std::vector<uint8_t>& tmp) uint8_t set_symlen(PairsData* d, size_t s, std::vector<bool>& visited)
{ {
visited[s] = true; // We can set now because tree is acyclic
int sr = d->btree[s].get<LR::Right>(); int sr = d->btree[s].get<LR::Right>();
if (sr == 0xFFF) if (sr == 0xFFF)
d->symlen[s] = 0; return 0;
else { else {
int sl = d->btree[s].get<LR::Left>(); int sl = d->btree[s].get<LR::Left>();
if (!tmp[sl]) if (!visited[sl])
calc_symlen(d, sl, tmp); d->symlen[sl] = set_symlen(d, sl, visited);
if (!tmp[sr]) if (!visited[sr])
calc_symlen(d, sr, tmp); d->symlen[sr] = set_symlen(d, sr, visited);
d->symlen[s] = d->symlen[sl] + d->symlen[sr] + 1; return d->symlen[sl] + d->symlen[sr] + 1;
} }
tmp[s] = 1;
} }
uint8_t* set_sizes(PairsData* d, uint8_t* data, uint64_t tb_size) uint8_t* set_sizes(PairsData* d, uint8_t* data, uint64_t tb_size)
@ -982,11 +981,11 @@ uint8_t* set_sizes(PairsData* d, uint8_t* data, uint64_t tb_size)
d->symlen.resize(number<uint16_t, LittleEndian>(data)); data += sizeof(uint16_t); d->symlen.resize(number<uint16_t, LittleEndian>(data)); data += sizeof(uint16_t);
d->btree = (LR*)data; d->btree = (LR*)data;
std::vector<uint8_t> tmp(d->symlen.size()); std::vector<bool> visited(d->symlen.size());
for (size_t i = 0; i < d->symlen.size(); ++i) for (size_t sym = 0; sym < d->symlen.size(); ++sym)
if (!tmp[i]) if (!visited[sym])
calc_symlen(d, i, tmp); d->symlen[sym] = set_symlen(d, sym, visited);
return data + 3 * d->symlen.size() + (d->symlen.size() & 1); return data + 3 * d->symlen.size() + (d->symlen.size() & 1);
} }