diff --git a/src/movegen.h b/src/movegen.h index bbb35b391..3047bf6ce 100644 --- a/src/movegen.h +++ b/src/movegen.h @@ -55,12 +55,37 @@ inline bool operator<(const ExtMove& f, const ExtMove& s) { template ExtMove* generate(const Position& pos, ExtMove* moveList); +constexpr size_t moveListSize = sizeof(ExtMove) * MAX_MOVES; + /// 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) : last(generate(pos, moveList)) {} + +#ifdef USE_HEAP_INSTEAD_OF_STACK_FOR_MOVE_LIST + explicit MoveList(const Position& pos) + { + this->moveList = (ExtMove*)malloc(moveListSize); + if (this->moveList == 0) + { + printf("Error: Failed to allocate memory in heap."); + exit(1); + } + this->last = generate(pos, this->moveList); + } + + ~MoveList() + { + free(this->moveList); + } +#else + explicit MoveList(const Position& pos) : last(generate(pos, moveList)) + { + ; + } +#endif + const ExtMove* begin() const { return moveList; } const ExtMove* end() const { return last; } size_t size() const { return last - moveList; } @@ -69,7 +94,12 @@ struct MoveList { } private: - ExtMove moveList[MAX_MOVES], *last; + ExtMove* last; +#ifdef USE_HEAP_INSTEAD_OF_STACK_FOR_MOVE_LIST + ExtMove* moveList = 0; +#else + ExtMove moveList[MAX_MOVES]; +#endif }; } // namespace Stockfish diff --git a/src/types.h b/src/types.h index 5abfac80b..b049736b1 100644 --- a/src/types.h +++ b/src/types.h @@ -227,13 +227,22 @@ typedef uint64_t Bitboard; constexpr int SQUARE_BITS = 6; #endif +//When defined, move list will be stored in heap. Delete this if you want to use stack to store move list. Using stack can cause overflow (Segmentation Fault) when the search is too deep. +#define USE_HEAP_INSTEAD_OF_STACK_FOR_MOVE_LIST + #ifdef ALLVARS constexpr int MAX_MOVES = 8192; -constexpr int MAX_PLY = 60; +#ifdef USE_HEAP_INSTEAD_OF_STACK_FOR_MOVE_LIST +constexpr int MAX_PLY = 246; +#else +constexpr int MAX_PLY = 60; +#endif +/// endif USE_HEAP_INSTEAD_OF_STACK_FOR_MOVE_LIST #else constexpr int MAX_MOVES = 1024; -constexpr int MAX_PLY = 246; +constexpr int MAX_PLY = 246; #endif +/// endif ALLVARS /// A move needs 16 bits to be stored ///