HCha/Stockfish
1
0
Fork 0
mirror of https://github.com/HChaZZY/Stockfish.git synced 2025-12-24 19:16:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add king square counter

Browse Source
This commit is contained in:
Tomasz Sobczyk
2021-04-05 16:15:37 +02:00
parent 570a0f6f3c
commit eda51f19a2
1 changed files with 29 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/learn/stats.cpp
View File

@@ -147,6 +147,32 @@ namespace Learner::Stats
std::uint64_t m_num_positions;
};
struct KingSquareCounter : StatisticGathererBase
{
KingSquareCounter() :
m_white{},
m_black{}
{
}
void on_position(const Position& pos) override
{
m_white[pos.square<KING>(WHITE)] += 1;
m_black[pos.square<KING>(BLACK)] += 1;
}
void reset() override
{
m_white = StatPerSquare<std::uint64_t>{};
m_black = StatPerSquare<std::uint64_t>{};
}
private:
StatPerSquare<std::uint64_t> m_white;
StatPerSquare<std::uint64_t> m_black;
};
/*
This function provides factories for all possible statistic gatherers.
Each new statistic gatherer needs to be added there.
@@ -158,6 +184,9 @@ namespace Learner::Stats
reg.add<PositionCounter>("position_count");
reg.add<KingSquareCounter>("king");
reg.add<KingSquareCounter>("king_square_count");
return reg;
}();