mirror of
https://github.com/sockspls/badfish
synced 2025-04-29 16:23:09 +00:00
RootMoveList sorting: be compatible with std::sort
sort() and sort_multipv() are almost the same, so use only one implementation. Also introduce the natural RootMove::operator<() to compare the moves instead of compare_root_moves(), this will allow to use std::sort instead of our home grown bubble-sort. Signed-off-by: Marco Costalba <mcostalba@gmail.com>
This commit is contained in:
parent
bd3fd6501b
commit
731a9f3806
1 changed files with 25 additions and 31 deletions
|
@ -51,10 +51,11 @@ namespace {
|
||||||
// root move, we store a score, a node count, and a PV (really a refutation
|
// root move, we store a score, a node count, and a PV (really a refutation
|
||||||
// in the case of moves which fail low).
|
// in the case of moves which fail low).
|
||||||
|
|
||||||
class RootMove {
|
struct RootMove {
|
||||||
|
|
||||||
public:
|
|
||||||
RootMove();
|
RootMove();
|
||||||
|
bool operator<(const RootMove&); // used to sort
|
||||||
|
|
||||||
Move move;
|
Move move;
|
||||||
Value score;
|
Value score;
|
||||||
int64_t nodes, cumulativeNodes;
|
int64_t nodes, cumulativeNodes;
|
||||||
|
@ -78,11 +79,10 @@ namespace {
|
||||||
int64_t get_move_cumulative_nodes(int moveNum);
|
int64_t get_move_cumulative_nodes(int moveNum);
|
||||||
int move_count() const;
|
int move_count() const;
|
||||||
Move scan_for_easy_move() const;
|
Move scan_for_easy_move() const;
|
||||||
void sort();
|
inline void sort();
|
||||||
void sort_multipv(int n);
|
void sort_multipv(int n);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
static bool compare_root_moves(const RootMove &rm1, const RootMove &rm2);
|
|
||||||
static const int MaxRootMoves = 500;
|
static const int MaxRootMoves = 500;
|
||||||
RootMove moves[MaxRootMoves];
|
RootMove moves[MaxRootMoves];
|
||||||
int count;
|
int count;
|
||||||
|
@ -1599,6 +1599,18 @@ namespace {
|
||||||
nodes = cumulativeNodes = 0ULL;
|
nodes = cumulativeNodes = 0ULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// RootMove::operator<() is the comparison function used when
|
||||||
|
// sorting the moves. A move m1 is considered to be better
|
||||||
|
// than a move m2 if it has a higher score, or if the moves
|
||||||
|
// have equal score but m1 has the higher node count.
|
||||||
|
|
||||||
|
bool RootMove::operator<(const RootMove& m) {
|
||||||
|
|
||||||
|
if (score != m.score)
|
||||||
|
return (score < m.score);
|
||||||
|
|
||||||
|
return nodes <= m.nodes;
|
||||||
|
}
|
||||||
|
|
||||||
/// The RootMoveList class
|
/// The RootMoveList class
|
||||||
|
|
||||||
|
@ -1702,45 +1714,27 @@ namespace {
|
||||||
return MOVE_NONE;
|
return MOVE_NONE;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// RootMoveList::compare_root_moves() is the comparison function used by
|
|
||||||
// RootMoveList::sort when sorting the moves. A move m1 is considered to
|
|
||||||
// be better than a move m2 if it has a higher score, or if the moves have
|
|
||||||
// equal score but m1 has the higher node count.
|
|
||||||
|
|
||||||
bool RootMoveList::compare_root_moves(const RootMove &rm1,
|
|
||||||
const RootMove &rm2) {
|
|
||||||
if (rm1.score != rm2.score)
|
|
||||||
return (rm1.score < rm2.score);
|
|
||||||
|
|
||||||
return rm1.nodes <= rm2.nodes;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// RootMoveList::sort() sorts the root move list at the beginning of a new
|
// RootMoveList::sort() sorts the root move list at the beginning of a new
|
||||||
// iteration.
|
// iteration.
|
||||||
|
|
||||||
void RootMoveList::sort() {
|
inline void RootMoveList::sort() {
|
||||||
for(int i = 1; i < count; i++) {
|
|
||||||
RootMove rm = moves[i];
|
sort_multipv(count - 1); // all items
|
||||||
int j;
|
|
||||||
for(j = i; j > 0 && compare_root_moves(moves[j-1], rm); j--)
|
|
||||||
moves[j] = moves[j-1];
|
|
||||||
moves[j] = rm;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// RootMoveList::sort_multipv() sorts the first few moves in the root move
|
// RootMoveList::sort_multipv() sorts the first few moves in the root move
|
||||||
// list by their scores and depths. It is used to order the different PVs
|
// list by their scores and depths. It is used to order the different PVs
|
||||||
// correctly in MultiPV mode.
|
// correctly in MultiPV mode.
|
||||||
|
|
||||||
void RootMoveList::sort_multipv(int n) {
|
void RootMoveList::sort_multipv(int n) {
|
||||||
for(int i = 1; i <= n; i++) {
|
|
||||||
|
for (int i = 1; i <= n; i++)
|
||||||
|
{
|
||||||
RootMove rm = moves[i];
|
RootMove rm = moves[i];
|
||||||
int j;
|
int j;
|
||||||
for(j = i; j > 0 && moves[j-1].score < rm.score; j--)
|
for (j = i; j > 0 && moves[j-1] < rm; j--)
|
||||||
moves[j] = moves[j-1];
|
moves[j] = moves[j-1];
|
||||||
moves[j] = rm;
|
moves[j] = rm;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Reference in a new issue