75 lines
1.9 KiB
Go
75 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/notnil/chess"
|
|
"github.com/notnil/chess/uci"
|
|
"math"
|
|
)
|
|
|
|
// stolen^H^H inspired from lichess https://github.com/ornicar/lila/blob/master/modules/analyse/src/main/Advice.scala#L79
|
|
func WinningChance(cp int) float64 {
|
|
winning_chance := 2 / (1 + math.Exp(-0.004 * float64(cp))) - 1
|
|
return winning_chance
|
|
}
|
|
|
|
func main() {
|
|
engine, err := uci.New("stockfish")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer engine.Close()
|
|
|
|
if err := engine.Run(uci.CmdUCI, uci.CmdIsReady, uci.CmdUCINewGame); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
game := chess.NewGame()
|
|
prevprev_winning_chance := 0.0
|
|
prev_winning_chance := 0.0
|
|
for game.Outcome() == chess.NoOutcome {
|
|
if err := engine.Run(uci.CmdPosition{Position: game.Position()}, uci.CmdGo{Depth: 12}); err != nil {
|
|
panic(err)
|
|
}
|
|
search_results := engine.SearchResults()
|
|
cp := search_results.Info.Score.CP
|
|
winning_chance := WinningChance(cp)
|
|
num_of_moves := len(game.Moves())
|
|
if (num_of_moves > 0) {
|
|
delta := prevprev_winning_chance - winning_chance
|
|
if (num_of_moves % 2 == 0) {
|
|
delta *= -1;
|
|
}
|
|
if delta > 0.3 {
|
|
fmt.Print("B-b-b-blunder!!")
|
|
} else if delta > 0.2 {
|
|
fmt.Print("That was a mistake.")
|
|
} else if delta > 0.1 {
|
|
fmt.Print("Meh...")
|
|
} else {
|
|
fmt.Print("Ok")
|
|
}
|
|
fmt.Printf(" (%0.2f)\n", -delta)
|
|
}
|
|
prevprev_winning_chance = prev_winning_chance
|
|
prev_winning_chance = winning_chance
|
|
fmt.Println(game.Position().Board().Draw())
|
|
fmt.Println("Score (centipawns):", cp, "Winning chance:", winning_chance, "Best Move: ", search_results.BestMove)
|
|
fmt.Println("Move: ", search_results.BestMove)
|
|
if err := game.Move(search_results.BestMove); err != nil {
|
|
panic(err)
|
|
}
|
|
// for {
|
|
// var move string
|
|
// fmt.Print("Move: ")
|
|
// fmt.Scanln(&move)
|
|
// if err := game.MoveStr(move); err == nil {
|
|
// break
|
|
// }
|
|
// fmt.Println("Illegal move!")
|
|
// }
|
|
}
|
|
fmt.Println(game.Outcome())
|
|
fmt.Println(game.Position().Board().Draw())
|
|
}
|