stellar

0 #include <algorithm>

1 #include <iostream>

2 #include <limits>

3 #include <sstream>

4 #include <string>

5

6 #include "engine.hpp"

7 #include "stellar_version.hpp"

8 #include "timer.hpp"

9 #include "uci.hpp"

10

11 namespace uci {

12

13 void communicate(const uci::Settings *settings) {

14 if (!settings->infinite && timer::get_ms() > settings->stoptime) {

15 settings->stopped = true;

16 return;

17 }

18 }

19

20 inline bool parse_move(const Board &board, Move &move, const std::string &move_string) {

21 const Square source = from_coordinates(move_string.substr(0, 2));

22 const Square target = from_coordinates(move_string.substr(2, 2));

23

24 const MoveList list(board);

25 for (int i = 0; i < list.size(); i++) {

26 const Move crnt = list[i];

27 if (crnt.source() == source && crnt.target() == target) {

28 if (move_string[4]) {

29 if (tolower(piece::get_code(crnt.promoted())) != move_string[4]) continue;

30 }

31 move = crnt;

32 return true;

33 }

34 }

35 return false;

36 }

37

38 void loop() {

39 static Settings settings;

40 static std::string line, command;

41 static Move move;

42

43 Board board;

44

45 while (true) {

46 std::getline(std::cin, line);

47 std::istringstream iss(line);

48 iss >> command;

49 if (command == "quit") {

50 break;

51 } else if (command == "uci") {

52 std::cout << "id name Stellar " << getStellarVersion() << "\n";

53 std::cout << "id author Dimitrije Dobrota\n";

54 std::cout << "uciok\n";

55 } else if (command == "debug") {

56 iss >> command;

57 settings.debug = (command == "on");

58 } else if (command == "isready") {

59 std::cout << "readyok\n";

60 } else if (command == "ucinewgame") {

61 settings = Settings();

62 settings.board = Board(start_position);

63 } else if (command == "position") {

64 settings.madeMoves.clear();

65 iss >> command;

66 if (command == "startpos") {

67 settings.board = Board(start_position);

68 } else if (command == "fen") {

69 iss >> command;

70 settings.board = Board(line.substr(13));

71 }

72

73 while (iss >> command)

74 if (command == "moves") break;

75

76 board = settings.board;

77 while (iss >> command) {

78 if (!parse_move(board, move, command)) break;

79 settings.madeMoves.push(move);

80 move.make(board);

81 }

82 } else if (command == "go") {

83 settings.searchMoves.clear();

84 uint64_t wtime = 0, btime = 0, movetime = 0;

85 uint64_t winc = 0, binc = 0, movestogo = 40;

86

87 while (iss >> command) {

88 if (command == "wtime") iss >> wtime;

89 else if (command == "btime")

90 iss >> btime;

91 else if (command == "winc")

92 iss >> winc;

93 else if (command == "binc")

94 iss >> binc;

95 else if (command == "depth")

96 iss >> settings.depth;

97 else if (command == "nodes")

98 iss >> settings.nodes;

99 else if (command == "movetime")

100 iss >> movetime;

101 else if (command == "movestogo")

102 iss >> movestogo;

103 else if (command == "ponder")

104 settings.ponder = true;

105 else if (command == "mate")

106 settings.mate = true;

107 else if (command == "infinite")

108 settings.infinite = true;

109 else if (command == "searchmoves") {

110 while (iss >> command) {

111 if (!parse_move(board, move, command)) break;

112 settings.searchMoves.push(move);

113 }

114 }

115 }

116

117 auto time_left = board.get_side() == WHITE ? wtime : btime;

118 int64_t time = 0;

119

120 if (movetime != 0) {

121 time = movetime;

122 movestogo = 1;

123 } else if (time_left != 0) {

124 auto inc = board.get_side() == WHITE ? winc : binc;

125 time = time_left / movestogo + inc / 2;

126 if (time > time_left) time = time_left - 500;

127 if (time <= 0) time = 100;

128

129 settings.infinite = false;

130 } else {

131 settings.infinite = true;

132 }

133

134 settings.starttime = timer::get_ms();

135 settings.stoptime = settings.starttime + time;

136

137 const Move best = engine::search_position(settings);

138 std::cout << "bestmove " << best << '\n';

139

140 settings.newgame = false;

141 }

142 }

143 }

144 } // namespace uci