#pythonista 3
#chess engine based on piece influence

from array import array

n = 9
m = ""    #move string
#         0123456789012"
pieces = ".prnbqkPRNBQK"
adj = 0
x = y = 0

board = array('B')
board = [[0] * n for i in range(n)]
inf   = array('B')
inf   = [[0] * n for i in range(n+1)]

#king influence field
kf = [[-1,-1],[ 0,-1],[ 1,-1],
      [-1, 0],        [ 1, 0],
      [-1, 1],[ 0, 1],[ 1, 1]]

nf = [[-1,-2],[ 1,-2],
      [-2,-1],[ 2,-1],
      [-2, 1],[ 2, 1],
      [-1, 2],[ 1, 2]]

'''
#test board
board[8] = [0, 0, 0, 0, 0, 0, 0, 0, 0]# R  8 - 2
board[7] = [0, 0, 0, 0, 0, 0, 0, 0, 0]# N  9 - 3
board[6] = [0, 0, 0, 0, 0, 0, 0, 0, 0]# B 10 - 4
board[5] = [0, 0, 0, 0, 0, 0, 0, 0, 0]# Q 11 - 5
board[4] = [0, 0, 0, 0, 0, 0, 0, 0, 0]# K 12 - 6
board[3] = [0, 0, 0, 0, 0, 0, 0, 0, 0]# P  7 - 1
board[2] = [0, 0, 0, 0, 0, 0, 0, 0, 0]
board[1] = [0, 0, 0, 0, 0, 0, 0, 0, 0]
#              1  2  3  4  5  6  7  8
'''

#standard board                            W   B
board[8] = [0, 2, 3, 4, 5, 6, 4, 3, 2]# R  8 - 2
board[7] = [0, 1, 1, 1, 1, 1, 1, 1, 1]# N  9 - 3
board[6] = [0, 0, 0, 0, 0, 0, 0, 0, 0]# B 10 - 4
board[5] = [0, 0, 0, 0, 0, 0, 0, 0, 0]# Q 11 - 5
board[4] = [0, 0, 0, 0, 0, 0, 0, 0, 0]# K 12 - 6
board[3] = [0, 0, 0, 0, 0, 0, 0, 0, 0]# P  7 - 1
board[2] = [0, 7, 7, 7, 7, 7, 7, 7, 7]
board[1] = [0, 8, 9,10,11,12,10, 9, 8]

def on(h,v): # on board
    return(int((h>0) and (h<9) and (v>0) and (v<9)))

def enter_move(m):
    m = input(":    ")
    if(len(m)==3):
        board[int(m[2])][int(m[1])] = pieces.find(m[0])
    if(len(m)==4):
        board[int(m[3])][int(m[2])] = board[int(m[1])][int(m[0])]
        board[int(m[1])][int(m[0])] = 0

def move_gen():
    pass

def disp_ascii(): # print out ascii representation
    print("\n          ascii board")
    for j in range(1,n):
        print(n-j,"| ", end="")
        for i in range(1,n):
            print(pieces[board[9-j][i]], end = "  ")
        print()
    print("  ","-"*22)
    print("    1  2  3  4  5  6  7  8")

def disp_board():#debug - display numeric board
    print("\n         numeric board")
    for j in range(1,n):
        for i in range(1,n):
            print("%02d" % board[9-j][i], end = " ")
        print()
        
def disp_inf():#debug - display numeric board
    print("\n           influence")
    for j in range(1,n):
        print(n-j,"| ", end="")
        for i in range(1,n):
            print("%02d" % inf[9-j][i], end = " ")
        print()
    print("  ","-"*22)
    print("    1  2  3  4  5  6  7  8")
    print("\ntotal influence: ", sum(sum(inf,[])))

def calc_inf():
#this part uses the on() function
#reset influence array
    for j in range(n):
        for i in range(n):
            inf[i][j] = 0
        total_inf = 0

    for j in range(1,n):
        for i in range(1,n):
            if(board[i][j]==12 or board[i][j]==6):#kings
                adj=int(board[i][j]==12) or int(not(board[i][j]==6))-1
                for w in range(len(kf)):# k-king
                    if(on(i+kf[w][0],j+kf[w][1])):inf[i+kf[w][0]][j+kf[w][1]]+=adj

            if(board[i][j]==9 or board[i][j]==3):#knights
                adj=int(board[i][j]==9) or int(not(board[i][j]==3))-1
                for w in range(len(nf)):# k-king
                    if(on(i+nf[w][0],j+nf[w][1])):inf[i+nf[w][0]][j+nf[w][1]]+=adj

            if(board[i][j]==7):#white pawn
                adj=1
                if(on(i+1,j-1)):inf[i+1][j-1]+=adj
                if(on(i+1,j+1)):inf[i+1][j+1]+=adj
                
            if(board[i][j]==1):#black pawn
                adj=-1
                if(on(i-1,j-1)):inf[i-1][j-1]+=adj
                if(on(i-1,j+1)):inf[i-1][j+1]+=adj
                
            if(board[i][j]==8 or board[i][j]==2):#rooks
                adj=int(board[i][j]==8) or int(not(board[i][j]==2))-1
                for w in range(-7,8):
                    if(on(i,j+w)):inf[i][j+w]+=adj
                    if(on(i+w,j)):inf[i+w][j]+=adj
                inf[i][j]-=2*adj

            if(board[i][j]==10 or board[i][j]==4):#bishops
                adj=int(board[i][j]==10) or int(not(board[i][j]==4))-1
                for w in range(-7,8):
                    if(on(i+w,j+w)):inf[i+w][j+w]+=adj
                    if(on(i+w,j-w)):inf[i+w][j-w]+=adj
                inf[i][j]-=2*adj

            if(board[i][j]==11 or board[i][j]==5):#queens
                adj=int(board[i][j]==11) or int(not(board[i][j]==5))-1
                for w in range(-7,8):
                    if(on(i,j+w)):inf[i][j+w]+=adj     #rook part
                    if(on(i+w,j)):inf[i+w][j]+=adj
                    if(on(i+w,j+w)):inf[i+w][j+w]+=adj #bishop part
                    if(on(i+w,j-w)):inf[i+w][j-w]+=adj
                inf[i][j]-=4*adj

calc_inf()
disp_inf()
disp_ascii()
while(True):
    enter_move(m)
    calc_inf()
    disp_inf()
    disp_ascii()
#    disp_board()