# Stint
# Copyright 2018 Status Research & Development GmbH
# Licensed under either of
#
#  * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE) or http://www.apache.org/licenses/LICENSE-2.0)
#  * MIT license ([LICENSE-MIT](LICENSE-MIT) or http://opensource.org/licenses/MIT)
#
# at your option. This file may not be copied, modified, or distributed except according to those terms.

import ../stint, unittest

template chkCountOnes(chk: untyped, bits: int) =
  block:
    var x = 0.stuint(bits)
    chk x.countOnes == 0

    for i in 1 .. bits:
      x = x shl 1
      x = x or 1.stuint(bits)
      chk x.countOnes == i

template chkParity(chk: untyped, bits: int) =
  block:
    var x = 0.stuint(bits)
    chk x.parity == 0

    for i in 1 .. bits:
      x = x shl 1
      x = x or 1.stuint(bits)
      chk x.parity == (i mod 2)

template chkFirstOne(chk: untyped, bits: int) =
  block:
    var x = 0.stuint(bits)
    chk x.firstOne == 0
    x = x + 1
    chk x.firstOne == 1

    for i in 2 .. bits:
      x = x shl 1
      chk x.firstOne == i

template chkLeadingZeros(chk: untyped, bits: int) =
  block:
    var x = 0.stuint(bits)
    chk x.leadingZeros == bits
    x = x + 1
    chk x.leadingZeros == (bits-1)

    for i in 2 .. bits:
      x = x shl 1
      chk x.leadingZeros == (bits-i)

template chkTrailingZeros(chk: untyped, bits: int) =
  block:
    var x = 0.stuint(bits)
    chk x.trailingZeros == bits
    x = x + 1
    chk x.trailingZeros == 0

    for i in 1 .. bits:
      x = x shl 1
      chk x.trailingZeros == i

template ctTest(name: string, body: untyped) =
  body
  echo "[OK] compile time ", name

template testBitOps(chk, tst: untyped) =
  tst "countOnes":
    chk countOnes(0b01000100'u8.stuint(8)) == 2

    chk countOnes(0b01000100'u8.stuint(16)) == 2
    chk countOnes(0b01000100_01000100'u16.stuint(16)) == 4

    chk countOnes(0b01000100'u8.stuint(32)) == 2
    chk countOnes(0b01000100_01000100'u16.stuint(32)) == 4
    chk countOnes(0b01000100_01000100_01000100_01000100'u32.stuint(32)) == 8

    chk countOnes(0b01000100'u8.stuint(64)) == 2
    chk countOnes(0b01000100_01000100'u16.stuint(64)) == 4
    chk countOnes(0b01000100_01000100_01000100_01000100'u32.stuint(64)) == 8
    chk countOnes(0b01000100_01000100_01000100_01000100_01000100_01000100_01000100_01000100'u64.stuint(64)) == 16

    chk countOnes(0b01000100'u8.stuint(128)) == 2
    chk countOnes(0b01000100_01000100'u16.stuint(128)) == 4
    chk countOnes(0b01000100_01000100_01000100_01000100'u32.stuint(128)) == 8
    chk countOnes(0b01000100_01000100_01000100_01000100_01000100_01000100_01000100_01000100'u64.stuint(128)) == 16
    chk countOnes(0b01000100'u8.stuint(128) shl 100) == 2

    chkCountOnes(chk, 128)
    chkCountOnes(chk, 256)

  tst "parity":
    chk parity(0b00000001'u8.stuint(8)) == 1
    chk parity(0b00000011'u8.stuint(8)) == 0

    chk parity(0b00000001'u8.stuint(16)) == 1
    chk parity(0b00000011'u8.stuint(16)) == 0
    chk parity(0b00000001_00000001'u16.stuint(16)) == 0
    chk parity(0b00000011_00000001'u16.stuint(16)) == 1

    chk parity(0b00000001'u8.stuint(32)) == 1
    chk parity(0b00000011'u8.stuint(32)) == 0
    chk parity(0b00000001_00000001'u16.stuint(32)) == 0
    chk parity(0b00000011_00000001'u16.stuint(32)) == 1
    chk parity(0b00000001_00000001_00000001_00000001'u32.stuint(32)) == 0
    chk parity(0b00000011_00000001_00000001_00000001'u32.stuint(32)) == 1

    chk parity(0b00000001'u8.stuint(64)) == 1
    chk parity(0b00000011'u8.stuint(64)) == 0
    chk parity(0b00000001_00000001'u16.stuint(64)) == 0
    chk parity(0b00000011_00000001'u16.stuint(64)) == 1
    chk parity(0b00000001_00000001_00000001_00000001'u32.stuint(64)) == 0
    chk parity(0b00000011_00000001_00000001_00000001'u32.stuint(64)) == 1
    chk parity(0b00000001_00000001_00000001_00000001_00000001_00000001_00000001_00000001'u64.stuint(64)) == 0
    chk parity(0b00000011_00000001_00000001_00000001_00000001_00000001_00000001_00000001'u64.stuint(64)) == 1

    chk parity(0b00000001'u8.stuint(128)) == 1
    chk parity(0b00000011'u8.stuint(128)) == 0
    chk parity(0b00000001_00000001'u16.stuint(128)) == 0
    chk parity(0b00000011_00000001'u16.stuint(128)) == 1
    chk parity(0b00000001_00000001_00000001_00000001'u32.stuint(128)) == 0
    chk parity(0b00000011_00000001_00000001_00000001'u32.stuint(128)) == 1
    chk parity(0b00000001_00000001_00000001_00000001_00000001_00000001_00000001_00000001'u64.stuint(128)) == 0
    chk parity(0b00000011_00000001_00000001_00000001_00000001_00000001_00000001_00000001'u64.stuint(128)) == 1

    chk parity(0b00000001'u8.stuint(128)) == 1
    chk parity(0b00000001'u8.stuint(128) shl 100) == 1

    chkParity(chk, 128)
    chkParity(chk, 256)

  tst "firstOne":
    chk firstOne(0b00000010'u8.stuint(8)) == 2

    chk firstOne(0b00000010'u8.stuint(16)) == 2
    chk firstOne(0b00000010_00000000'u16.stuint(16)) == 10

    chk firstOne(0b00000010'u8.stuint(32)) == 2
    chk firstOne(0b00000010_00000000'u16.stuint(32)) == 10
    chk firstOne(0b00000010_00000000_00000000_00000000'u32.stuint(32)) == 26

    chk firstOne(0b00000010'u8.stuint(64)) == 8*0+2
    chk firstOne(0b00000010_00000000'u16.stuint(64)) == 8*1+2
    chk firstOne(0b00000010_00000000_00000000_00000000'u32.stuint(64)) == 8*3+2
    chk firstOne(0b00000010_00000000_00000000_00000000_00000000_00000000_00000000_00000000'u64.stuint(64)) == 8*7+2

    chk firstOne(0b00000010'u8.stuint(128)) == 2
    chk firstOne(0b00000010'u8.stuint(128) shl 100) == 102
    chk firstOne(0'u8.stuint(128)) == 0

    chkFirstOne(chk, 128)
    chkFirstOne(chk, 256)

  tst "leadingZeros":
    chk leadingZeros(0'u8.stuint(8)) == 8*1
    chk leadingZeros(0b00010000'u8.stuint(8)) == 3

    chk leadingZeros(0'u8.stuint(16)) == 8*2
    chk leadingZeros(0b00010000'u8.stuint(16)) == 8*1+3
    chk leadingZeros(0'u16.stuint(16)) == 8*2
    chk leadingZeros(0b00000000_00010000'u16.stuint(16)) == 8*1+3

    chk leadingZeros(0'u8.stuint(32)) == 8*4
    chk leadingZeros(0b00010000'u8.stuint(32)) == 8*3+3
    chk leadingZeros(0'u16.stuint(32)) == 8*4
    chk leadingZeros(0b00000000_00010000'u16.stuint(32)) == 8*3+3
    chk leadingZeros(0'u32.stuint(32)) == 8*4
    chk leadingZeros(0b00000000_00000000_00000000_00010000'u32.stuint(32)) == 8*3+3

    chk leadingZeros(0'u8.stuint(64)) == 8*8
    chk leadingZeros(0b00010000'u8.stuint(64)) == 8*7+3
    chk leadingZeros(0'u16.stuint(64)) == 8*8
    chk leadingZeros(0b00000000_00010000'u16.stuint(64)) == 8*7+3
    chk leadingZeros(0'u32.stuint(64)) == 8*8
    chk leadingZeros(0b00000000_00000000_00000000_00010000'u32.stuint(64)) == 8*7+3
    chk leadingZeros(0'u64.stuint(64)) == 8*8
    chk leadingZeros(0b00000000_00000000_00000000_00000000_00000000_00000000_00000000_00010000'u64.stuint(64)) == 8*7+3

    chk leadingZeros(0'u8.stuint(128)) == 128
    chk leadingZeros(0b00100000'u8.stuint(128)) == 128 - 6
    chk leadingZeros(0b00100000'u8.stuint(128) shl 100) == 128 - 106

    chkLeadingZeros(chk, 128)
    chkLeadingZeros(chk, 256)

  tst "trailingZeros":
    chk trailingZeros(0'u8.stuint(8)) == 8*1
    chk trailingZeros(0b00010000'u8.stuint(8)) == 4

    chk trailingZeros(0'u8.stuint(16)) == 8*2
    chk trailingZeros(0b00010000'u8.stuint(16)) == 8*0+4
    chk trailingZeros(0'u16.stuint(16)) == 8*2
    chk trailingZeros(0b00010000_00000000'u16.stuint(16)) == 8*1+4

    chk trailingZeros(0'u8.stuint(32)) == 8*4
    chk trailingZeros(0b00010000'u8.stuint(32)) == 8*0+4
    chk trailingZeros(0'u16.stuint(32)) == 8*4
    chk trailingZeros(0b00010000_00000000'u16.stuint(32)) == 8*1+4
    chk trailingZeros(0'u32.stuint(32)) == 8*4
    chk trailingZeros(0b00010000_00000000_00000000_00000000'u32.stuint(32)) == 8*3+4

    chk trailingZeros(0'u8.stuint(64)) == 8*8
    chk trailingZeros(0b00010000'u8.stuint(64)) == 8*0+4
    chk trailingZeros(0'u16.stuint(64)) == 8*8
    chk trailingZeros(0b00010000_00000000'u16.stuint(64)) == 8*1+4
    chk trailingZeros(0'u32.stuint(64)) == 8*8
    chk trailingZeros(0b00010000_00000000_00000000_00000000'u32.stuint(64)) == 8*3+4
    chk trailingZeros(0'u64.stuint(64)) == 8*8
    chk trailingZeros(0b00010000_00000000_00000000_00000000_00000000_00000000_00000000_00000000'u64.stuint(64)) == 8*7+4

    chk trailingZeros(0b00100000'u8.stuint(128)) == 5
    chk trailingZeros(0b00100000'u8.stuint(128) shl 100) == 105
    chk trailingZeros(0'u8.stuint(128)) == 128

    chkTrailingZeros(chk, 128)
    chkTrailingZeros(chk, 256)

static:
  testBitOps(doAssert, ctTest)

suite "Testing bitops2":
  testBitOps(check, test)