MarcoEngine/uci.py at main · MarcoNITE/MarcoEngine · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
import json # json dictionaries
import os # Operating System

import chess.engine # chess engine
from colorama import init, Fore # color text

import engine_main # engine
from utils import * # utils
from config import * # config
from python_checking import check # checking Operating System

if not os.path.exists("stockfish"): # if we haven't stockfish
    os.system('./makefish.sh') # download it!

init(autoreset=True) # initing colorama

weights = [] # weights list

board = chess.Board() # board
_engine = chess.engine.SimpleEngine.popen_uci(ENGINE_DIR) # chess engine
uci_conf = json.load(open('./settings/uci_config.json', 'r')) # uci config
uci_conf2 = json.load(open('./settings/conf.json', 'r')) # id's uci config
uci_default_conf = json.load(open('./settings/uci_default_config.json', 'r')) # default uci config
uci_min_conf = json.load(open('./settings/uci_min_config.json', 'r')) # minimal uci config
uci_max_conf = json.load(open('./settings/uci_max_config.json', 'r')) # maximal uci config


def show_intro():
    """Showing intro"""
    intro = Fore.GREEN + f""" _______ _______               __
|   |   |    ___|.-----.-----.|__|.-----.-----.
|       |    ___||     |  _  ||  ||     |  -__|
|__|_|__|_______||__|__|___  ||__||__|__|_____|
                       |_____|
-----------------------------------------------
{Fore.MAGENTA} MarcoEngine by Mark Kim. {Fore.LIGHTRED_EX} Neural chess network.
                       """ # intro

    print(intro) # printing intro


def if_havent_weights(dir):
    """Check for user have weights"""
    dir_list = os.listdir(str(dir)) # dir of list weights
    if dir_list == ["weights_norm.default.json"]: # if dir list is default weights
        print_l("""
                    You want to rename "weights_norm.json"(in directory
                    "weights") to "weights_norm.json".

        """, type="CRITICAL") # printing error about


def finding_weights():
    # finding weights file
    for f in os.listdir("./weights"):  # cycle of files in "weights" directory
        if f.endswith(".json"):  # if file ends with ".json"..
            weights.append(json.load(open(f'./weights/{f}', 'r')))  # append to weights that file


def uci_commander(command):
    # UCI-protocol

    if command.startswith('go'):
        engine_main.go(command=command, board=board, _engine=_engine) # analyzing

    elif command.startswith('uciok') or command == 'uci':  # if command starts with "uciok", or command is "uci"..
        for key in list(uci_conf2.keys()):  # easy config(name, author)
            print(f"id {str(key)}  {uci_conf2[key]}")  # printing

        for key in list(uci_conf.keys()):  # uci config
            type_option = str(type(uci_default_conf[str(key)]))  # type of option
            type_option = type_option.replace("<class '", "")  # deleting <class ' from string
            type_option = type_option.replace("'>", "")  # deleting '> from string
            print(
                f"option name {str(key)} type {type_option} default {uci_default_conf[key]} min {uci_min_conf[key]} max {uci_max_conf[key]}")  # printing

            del type_option  # and deleting

        print('uciok', flush=True)  # uci is ok :)

    elif command.startswith('ucinewgame'):  # if command starts with "ucinewgame"..
        new_board(old_board=board)  # deleting board, create new

    elif command.startswith('position fen'):  # if command starts with "position fen"
        new_board(command.split()[2])  # new board with new fen

    elif command.startswith('position startpos moves'):
        moves = command.split(" ")[3:]  # parse moves
        board.clear()  # clear board
        board.set_fen(chess.STARTING_FEN)  # set starting fen
        for move in moves:  # cycle of moves
            board.push(chess.Move.from_uci(move))  # and make this moves

    elif command.startswith('isready'):  # hey, you ready?
        print('readyok')  # yes, ready

    elif command.startswith('quit'):  # exit
        sys.exit(1)  # exit

    # Stockfish-like commands
    elif command == 'd':
        print(board.unicode()) # prints unicode board

    elif command == 'flip':
        print(board.mirror().unicode()) # prints fliped board

    elif command == 'bench':
        for i in range(0, 11): # cycle: from 0 to 10
            engine_main.go(command=command, board=board, _engine=_engine,
                            depth=DEFAULT_DEPTH) # analyzing

            print("------") # line

    elif command == "":  # random enter
        pass  # passing, new iteration

    else:  # if command unkown...
        print(f'[?] Unknow command: {command}')  # printing


show_intro()  # showing intro
if_havent_weights('./weights')  # check for weights
check() # checking Operating system

while True:  # endless cycle
    cmd = input()  # command
    uci_commander(cmd)  # send command