scalachess/bench/src/main/scala/benchmarks/InsufficientMaterialBench.scala at 0f84b3f9bc9ec3f8e3bd4ecb5503ffad7633aff3 · lichess-org/scalachess · 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
package benchmarks

import org.openjdk.jmh.annotations.*

import java.util.concurrent.TimeUnit
import chess.format.{ FullFen, Fen }
import chess.variant.Horde
import chess.variant.Standard
import chess.InsufficientMatingMaterial

@State(Scope.Thread)
@BenchmarkMode(Array(Mode.Throughput))
@OutputTimeUnit(TimeUnit.SECONDS)
@Measurement(iterations = 15, timeUnit = TimeUnit.SECONDS, time = 3)
@Warmup(iterations = 15, timeUnit = TimeUnit.SECONDS, time = 3)
@Fork(value = 3)
@Threads(value = 1)
class InsufficientMaterialBench:

  var hordeGames = List(
    "k7/ppP5/brp5/8/8/8/8/8 b - -",
    "8/2k5/3q4/8/8/8/1P6/8 b - -",
    "8/2k5/3q4/8/8/8/1P6/8 w - -",
    "r7/2Bb4/q3k3/8/8/3q4/8/5qqr b - -",
    "8/2k5/3q4/8/8/1Q6/8/8 b - -",
    "8/2k5/3q4/8/8/1Q6/8/8 w - -",
    "8/2k5/3q4/8/8/1B2N3/8/8 b - -",
    "8/2k5/3q4/8/8/1B2N3/8/8 w - -",
    "8/2k5/3q4/8/8/3B4/4NB2/8 b - -",
    "8/5k2/7q/7P/6rP/6P1/6P1/8 b - - 0 52",
    "8/p7/pk6/P7/P7/8/8/8 b - -",
    "QNBRRBNQ/PPpPPpPP/P1P2PkP/8/8/8/8/8 b - -",
    "b7/pk6/P7/P7/8/8/8/8 b - - 0 1",
    "8/1b5r/1P6/1Pk3q1/1PP5/r1P5/P1P5/2P5 b - - 0 52",
    "7B/6k1/8/8/8/8/8/8 b - -",
    "k7/5p2/4p2P/3p2P1/2p2P2/1p2P2P/p2P2P1/2P2P2 w - - 0 1",
    "8/N7/8/8/8/8/bqnnbqbr/k7 b - - 0 1",
    "8/6PP/8/8/8/8/3npqrn/7k b - - 0 1",
    "8/P1P5/8/8/8/8/bbnb4/k7 b - - 0 1",
    "8/6PP/8/8/8/8/5rrb/7k b - - 0 1"
  ).map(FullFen(_)).map(Fen.read(Horde, _).get)

  @Benchmark
  def horde() =
    hordeGames.map: board =>
      board.variant.isInsufficientMaterial(board)

  var fens = List(
    "4k3/8/8/8/8/8/8/4K3 w - - 0 1",
    "4k3/8/8/8/8/8/4P3/4K3 w - - 0 1",
    "4k3/p4p2/p4p2/p4p2/Pp2pPp1/1Pp1P1Pp/2P1P2P/4K3 w - - 0 1",
    "4k3/p4p2/p4p2/p4p2/Pp2pPp1/1Pp1P1Pp/2P1P2P/4K3 b - f3 0 1"
  ).map(FullFen(_)).map(Fen.read(Standard, _).get)

  @Benchmark
  def pawnsLocked() =
    fens.map: position =>
      InsufficientMatingMaterial.allPawnsLocked(position.board)

  @Benchmark
  def kingPawnFortress() =
    fens.map: position =>
      InsufficientMatingMaterial.kingPawnFortress(position)

  @Benchmark
  def insufficientMatingMaterial() =
    fens.map: position =>
      InsufficientMatingMaterial(position)

  @Benchmark
  def isInsufficientMaterial() =
    fens.map: position =>
      position.variant.isInsufficientMaterial(position)

  @Benchmark
  def playerHasInsufficientMaterial() =
    fens.map: position =>
      position.variant.playerHasInsufficientMaterial(position)

  @Benchmark
  def opponentHasInsufficientMaterial() =
    fens.map: position =>
      position.variant.opponentHasInsufficientMaterial(position)