lean4-htt/library/init/data/nat/bitwise.lean

24 lines
687 B
Text

/-
Copyright (c) 2019 Microsoft Corporation. All rights reserved.
Released under Apache 2.0 license as described in the file LICENSE.
Authors: Leonardo de Moura
-/
prelude
import init.data.nat.basic init.data.nat.div init.coe
namespace Nat
def bitwise (f : Bool → Bool → Bool) : Nat → Nat → Nat | n m :=
if n = 0 then (if f false true then m else 0)
else if m = 0 then (if f true false then n else 0)
else
let n' := n / 2 in
let m' := m / 2 in
let b₁ := n % 2 = 1 in
let b₂ := m % 2 = 1 in
let r := bitwise n' m' in
if f b₁ b₂ then bit1 r else bit0 r
def land : Nat → Nat → Nat := bitwise and
def lor : Nat → Nat → Nat := bitwise or
end Nat