# Constantine
# Copyright (c) 2018-2019    Status Research & Development GmbH
# Copyright (c) 2020-Present Mamy André-Ratsimbazafy
# Licensed and distributed under either of
#   * MIT license (license terms in the root directory or at http://opensource.org/licenses/MIT).
#   * Apache v2 license (license terms in the root directory or at http://www.apache.org/licenses/LICENSE-2.0).
# at your option. This file may not be copied, modified, or distributed except according to those terms.

import
  # Internals
  ../constantine/config/curves,
  ../constantine/arithmetic,
  ../constantine/towers,
  # Helpers
  ../helpers/static_for,
  ./bench_summary_template,
  # Standard library
  std/strutils

# ############################################################
#
#               Benchmark of pairings
#                   for BLS12-381
#
# ############################################################


const Iters = 5000
const AvailableCurves = [
  BN254_Snarks,
]


proc main() =
  separator()
  staticFor i, 0, AvailableCurves.len:
    const curve = AvailableCurves[i]

    mulBench(Fr[curve], Iters)
    sqrBench(Fr[curve], Iters)
    separator()
    mulBench(Fp[curve], Iters)
    sqrBench(Fp[curve], Iters)
    invBench(Fp[curve], Iters)
    sqrtBench(Fp[curve], Iters)
    separator()
    mulBench(Fp2[curve], Iters)
    sqrBench(Fp2[curve], Iters)
    invBench(Fp2[curve], Iters)
    sqrtBench(Fp2[curve], Iters)
    separator()
    addBench(ECP_ShortW_Prj[Fp[curve], NotOnTwist], Iters)
    mixedAddBench(ECP_ShortW_Prj[Fp[curve], NotOnTwist], Iters)
    doublingBench(ECP_ShortW_Prj[Fp[curve], NotOnTwist], Iters)
    scalarMulBench(ECP_ShortW_Prj[Fp[curve], NotOnTwist], Iters)
    separator()
    addBench(ECP_ShortW_Jac[Fp[curve], NotOnTwist], Iters)
    mixedAddBench(ECP_ShortW_Jac[Fp[curve], NotOnTwist], Iters)
    doublingBench(ECP_ShortW_Jac[Fp[curve], NotOnTwist], Iters)
    scalarMulBench(ECP_ShortW_Jac[Fp[curve], NotOnTwist], Iters)
    separator()
    addBench(ECP_ShortW_Prj[Fp2[curve], OnTwist], Iters)
    mixedAddBench(ECP_ShortW_Prj[Fp2[curve], OnTwist], Iters)
    doublingBench(ECP_ShortW_Prj[Fp2[curve], OnTwist], Iters)
    scalarMulBench(ECP_ShortW_Prj[Fp2[curve], OnTwist], Iters)
    separator()
    addBench(ECP_ShortW_Jac[Fp2[curve], OnTwist], Iters)
    mixedAddBench(ECP_ShortW_Jac[Fp2[curve], OnTwist], Iters)
    doublingBench(ECP_ShortW_Jac[Fp2[curve], OnTwist], Iters)
    scalarMulBench(ECP_ShortW_Jac[Fp2[curve], OnTwist], Iters)
    separator()
    mulBench(Fp12[curve], Iters)
    sqrBench(Fp12[curve], Iters)
    invBench(Fp12[curve], Iters)
    separator()
    millerLoopBNBench(curve, Iters)
    finalExpBNBench(curve, Iters)
    pairingBNBench(curve, Iters)
    separator()

main()
notes()