X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Fmovegen.h;h=d46b8ddb7c9b4d5021fdbbcd13ff5dd59d2872ca;hp=6e4b99cbab215dd4f700dbf6934664d67eda08f8;hb=65f46794af41534e0dc744c6f7531a4c43eca857;hpb=42b48b08e81b55e385e55b3074b7c59d81809a45 diff --git a/src/movegen.h b/src/movegen.h index 6e4b99cb..d46b8ddb 100644 --- a/src/movegen.h +++ b/src/movegen.h @@ -22,6 +22,8 @@ #include "types.h" +class Position; + enum GenType { CAPTURES, QUIETS, @@ -31,7 +33,16 @@ enum GenType { LEGAL }; -class Position; +struct ExtMove { + Move move; + Value value; + + operator Move() const { return move; } +}; + +inline bool operator<(const ExtMove& f, const ExtMove& s) { + return f.value < s.value; +} template ExtMove* generate(const Position& pos, ExtMove* moveList); @@ -41,18 +52,17 @@ ExtMove* generate(const Position& pos, ExtMove* moveList); template struct MoveList { - explicit MoveList(const Position& pos) : cur(moveList), last(generate(pos, moveList)) { last->move = MOVE_NONE; } - void operator++() { ++cur; } - Move operator*() const { return cur->move; } + explicit MoveList(const Position& pos) : last(generate(pos, moveList)) {} + const ExtMove* begin() const { return moveList; } + const ExtMove* end() const { return last; } size_t size() const { return last - moveList; } - bool contains(Move m) const { - for (const ExtMove* it(moveList); it != last; ++it) if (it->move == m) return true; + bool contains(Move move) const { + for (const auto& m : *this) if (m == move) return true; return false; } private: - ExtMove moveList[MAX_MOVES]; - ExtMove *cur, *last; + ExtMove moveList[MAX_MOVES], *last; }; #endif // #ifndef MOVEGEN_H_INCLUDED