git.sesse.net Git - stockfish/blob - src/client.cpp

   1 #include <iostream>
   2 #include <memory>
   3 #include <string>
   4
   5 #include <grpc++/grpc++.h>
   6
   7 #include "hashprobe.grpc.pb.h"
   8 #include "types.h"
   9 #include "uci.h"
  10
  11 using grpc::Channel;
  12 using grpc::ClientContext;
  13 using grpc::Status;
  14 using namespace hashprobe;
  15
  16 std::string FormatMove(const HashProbeMove &move) {
  17   if (move.pretty().empty()) return "MOVE_NONE";
  18   return move.pretty();
  19 }
  20
  21 int main(int argc, char** argv) {
  22   std::shared_ptr<Channel> channel(grpc::CreateChannel(
  23       "localhost:50051", grpc::InsecureChannelCredentials()));
  24   std::unique_ptr<HashProbe::Stub> stub(HashProbe::NewStub(channel));
  25
  26   for ( ;; ) {
  27     char buf[256];
  28     if (fgets(buf, sizeof(buf), stdin) == nullptr || buf[0] == '\n') {
  29       exit(0);
  30     }
  31
  32     char *ptr = strchr(buf, '\n');
  33     if (ptr != nullptr) *ptr = 0;
  34
  35     HashProbeRequest request;
  36     request.set_fen(buf);
  37
  38     HashProbeResponse response;
  39     ClientContext context;
  40     Status status = stub->Probe(&context, request, &response);
  41
  42     if (status.ok()) {
  43       for (const HashProbeLine &line : response.line()) {
  44         std::cout << FormatMove(line.move()) << " ";
  45         std::cout << line.found() << " ";
  46         for (const HashProbeMove &move : line.pv()) {
  47           std::cout << FormatMove(move) << ",";
  48         }
  49         std::cout << " ";
  50         switch (line.bound()) {
  51         case HashProbeLine::BOUND_NONE:
  52           std::cout << "?";
  53           break;
  54         case HashProbeLine::BOUND_EXACT:
  55           std::cout << "==";
  56           break;
  57         case HashProbeLine::BOUND_UPPER:
  58           std::cout << "<=";
  59           break;
  60         case HashProbeLine::BOUND_LOWER:
  61           std::cout << ">=";
  62           break;
  63         }
  64         switch (line.value().score_type()) {
  65         case HashProbeScore::SCORE_CP:
  66           std::cout << " cp " << line.value().score_cp() << " ";
  67           break;
  68         case HashProbeScore::SCORE_MATE:
  69           std::cout << " mate " << line.value().score_mate() << " ";
  70           break;
  71         }
  72         std::cout << line.depth() << std::endl;
  73       }
  74       std::cout << "END" << std::endl;
  75     } else {
  76       std::cout << "ERROR" << std::endl;
  77     }
  78   }
  79
  80   return 0;
  81 }