X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Fmovegen.h;h=6f1e24a4313107680c0508bd9a95722a3773af0d;hp=f8c4f22c8b43ae1cde8f013c9dac2c85c0366e2d;hb=60c121f3b1ee7d5ced3435cc1718e4e6e6fd8383;hpb=d3608c4e79a29110f4c4a369d7207c6dd8e01f34 diff --git a/src/movegen.h b/src/movegen.h index f8c4f22c..6f1e24a4 100644 --- a/src/movegen.h +++ b/src/movegen.h @@ -1,7 +1,7 @@ /* Stockfish, a UCI chess playing engine derived from Glaurung 2.1 Copyright (C) 2004-2008 Tord Romstad (Glaurung author) - Copyright (C) 2008-2013 Marco Costalba, Joona Kiiski, Tord Romstad + Copyright (C) 2008-2015 Marco Costalba, Joona Kiiski, Tord Romstad Stockfish is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by @@ -17,11 +17,13 @@ along with this program. If not, see . */ -#if !defined(MOVEGEN_H_INCLUDED) +#ifndef MOVEGEN_H_INCLUDED #define MOVEGEN_H_INCLUDED #include "types.h" +class Position; + enum GenType { CAPTURES, QUIETS, @@ -31,28 +33,34 @@ enum GenType { LEGAL }; -class Position; +struct ExtMove { + Move move; + Value value; +}; + +inline bool operator<(const ExtMove& f, const ExtMove& s) { + return f.value < s.value; +} template -MoveStack* generate(const Position& pos, MoveStack* mlist); +ExtMove* generate(const Position& pos, ExtMove* moveList); -/// The MoveList struct is a simple wrapper around generate(), sometimes comes -/// handy to use this class instead of the low level generate() function. +/// The MoveList struct is a simple wrapper around generate(). It sometimes comes +/// in handy to use this class instead of the low level generate() function. template struct MoveList { - explicit MoveList(const Position& pos) : cur(mlist), last(generate(pos, mlist)) { last->move = MOVE_NONE; } - void operator++() { cur++; } - Move operator*() const { return cur->move; } - size_t size() const { return last - mlist; } + 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 MoveStack* it(mlist); it != last; ++it) if (it->move == m) return true; + for (const ExtMove& ms : *this) if (ms.move == m) return true; return false; } private: - MoveStack mlist[MAX_MOVES]; - MoveStack *cur, *last; + ExtMove moveList[MAX_MOVES], *last; }; -#endif // !defined(MOVEGEN_H_INCLUDED) +#endif // #ifndef MOVEGEN_H_INCLUDED