-- Mathlib/.lake/packages/lean4/src/lean/Std/Tactic/BVDecide/Bitblast/BVExpr/Circuit/Lemmas/Operations/ShiftRight.lean
theorem twoPowShift_eq (aig : AIG α) (target : TwoPowShiftTarget aig w) (lhs : BitVec w)
    (rhs : BitVec target.n) (assign : α → Bool)
    (hleft : ∀ (idx : Nat) (hidx : idx < w), ⟦aig, target.lhs.get idx hidx, assign⟧ = lhs.getLsbD idx)
    (hright : ∀ (idx : Nat) (hidx : idx < target.n), ⟦aig, target.rhs.get idx hidx, assign⟧ = rhs.getLsbD idx) :
    ∀ (idx : Nat) (hidx : idx < w),
        ⟦
          (twoPowShift aig target).aig,
          (twoPowShift aig target).vec.get idx hidx,
          assign
        ⟧
          =
        (lhs >>> (rhs &&& BitVec.twoPow target.n target.pow)).getLsbD idx

Proof state:
α : Type
inst✝¹ : Hashable α
inst✝ : DecidableEq α
w : Nat
aig : AIG α
lhs : BitVec w
assign : α → Bool
idx : Nat
hidx : idx < w
res : RefVecEntry α w
n : Nat
lvec : aig.RefVec w
rvec : aig.RefVec n
pow : Nat
rhs : BitVec { n := n, lhs := lvec, rhs := rvec, pow := pow }.n
hleft :
  ∀ (idx : Nat) (hidx : idx < w),
    ⟦assign, { aig := aig, ref := { n := n, lhs := lvec, rhs := rvec, pow := pow }.lhs.get idx hidx }⟧ = lhs.getLsbD idx
hright :
  ∀ (idx : Nat) (hidx : idx < { n := n, lhs := lvec, rhs := rvec, pow := pow }.n),
    ⟦assign, { aig := aig, ref := { n := n, lhs := lvec, rhs := rvec, pow := pow }.rhs.get idx hidx }⟧ = rhs.getLsbD idx
h✝ : pow < n
hg :
  RefVec.ite (blastShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).aig
      { discr := (rvec.cast ⋯).get pow h✝, lhs := (blastShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).vec,
        rhs := lvec.cast ⋯ } =
    res
hif1 : rhs.getLsbD pow = false
⊢ ⟦assign, { aig := res.aig, ref := res.vec.get idx hidx }⟧ = (lhs >>> 0#n).getLsbD idx

Tactic:
rw [← hg]

Result:
α : Type
inst✝¹ : Hashable α
inst✝ : DecidableEq α
w : Nat
aig : AIG α
lhs : BitVec w
assign : α → Bool
idx : Nat
hidx : idx < w
res : RefVecEntry α w
n : Nat
lvec : aig.RefVec w
rvec : aig.RefVec n
pow : Nat
rhs : BitVec { n := n, lhs := lvec, rhs := rvec, pow := pow }.n
hleft :
  ∀ (idx : Nat) (hidx : idx < w),
    ⟦assign, { aig := aig, ref := { n := n, lhs := lvec, rhs := rvec, pow := pow }.lhs.get idx hidx }⟧ = lhs.getLsbD idx
hright :
  ∀ (idx : Nat) (hidx : idx < { n := n, lhs := lvec, rhs := rvec, pow := pow }.n),
    ⟦assign, { aig := aig, ref := { n := n, lhs := lvec, rhs := rvec, pow := pow }.rhs.get idx hidx }⟧ = rhs.getLsbD idx
h✝ : pow < n
hg :
  RefVec.ite (blastShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).aig
      { discr := (rvec.cast ⋯).get pow h✝, lhs := (blastShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).vec,
        rhs := lvec.cast ⋯ } =
    res
hif1 : rhs.getLsbD pow = false
⊢ ⟦assign,
      {
        aig :=
          (RefVec.ite (blastShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).aig
              { discr := (rvec.cast ⋯).get pow h✝,
                lhs := (blastShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).vec, rhs := lvec.cast ⋯ }).aig,
        ref :=
          (RefVec.ite (blastShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).aig
                  { discr := (rvec.cast ⋯).get pow h✝,
                    lhs := (blastShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).vec,
                    rhs := lvec.cast ⋯ }).vec.get
            idx hidx }⟧ =
    (lhs >>> 0#n).getLsbD idx

Proof state:
case mk.isTrue.isFalse
α : Type
inst✝¹ : Hashable α
inst✝ : DecidableEq α
w : Nat
aig : AIG α
lhs : BitVec w
assign : α → Bool
idx : Nat
hidx : idx < w
res : RefVecEntry α w
n : Nat
lvec : aig.RefVec w
rvec : aig.RefVec n
pow : Nat
rhs : BitVec { n := n, lhs := lvec, rhs := rvec, pow := pow }.n
hleft :
  ∀ (idx : Nat) (hidx : idx < w),
    ⟦assign, { aig := aig, ref := { n := n, lhs := lvec, rhs := rvec, pow := pow }.lhs.get idx hidx }⟧ = lhs.getLsbD idx
hright :
  ∀ (idx : Nat) (hidx : idx < { n := n, lhs := lvec, rhs := rvec, pow := pow }.n),
    ⟦assign, { aig := aig, ref := { n := n, lhs := lvec, rhs := rvec, pow := pow }.rhs.get idx hidx }⟧ = rhs.getLsbD idx
h✝¹ : pow < n
hg :
  RefVec.ite (blastShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).aig
      { discr := (rvec.cast ⋯).get pow h✝¹, lhs := (blastShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).vec,
        rhs := lvec.cast ⋯ } =
    res
h✝ : ¬rhs.getLsbD pow = true
⊢ ⟦assign, { aig := res.aig, ref := res.vec.get idx hidx }⟧ = (lhs >>> 0#n).getLsbD idx

Tactic:
next hif1 =>
simp only [Bool.not_eq_true] at hif1
rw [← hg]
simp only [RefVec.denote_ite, RefVec.get_cast, Ref.cast_eq,
  denote_blastShiftRightConst]
rw [AIG.LawfulVecOperator.denote_mem_prefix (f := blastShiftRightConst)]
rw [hright]
simp only [hif1, Bool.false_eq_true, ↓reduceIte]
rw [AIG.LawfulVecOperator.denote_mem_prefix (f := blastShiftRightConst)]
rw [hleft]
simp

Result:
no goals

Proof state:
case mk
α : Type
inst✝¹ : Hashable α
inst✝ : DecidableEq α
w : Nat
aig : AIG α
lhs : BitVec w
assign : α → Bool
idx : Nat
hidx : idx < w
res : RefVecEntry α w
n : Nat
lvec : aig.RefVec w
rvec : aig.RefVec n
pow : Nat
rhs : BitVec { n := n, lhs := lvec, rhs := rvec, pow := pow }.n
hleft :
  ∀ (idx : Nat) (hidx : idx < w),
    ⟦assign, { aig := aig, ref := { n := n, lhs := lvec, rhs := rvec, pow := pow }.lhs.get idx hidx }⟧ = lhs.getLsbD idx
hright :
  ∀ (idx : Nat) (hidx : idx < { n := n, lhs := lvec, rhs := rvec, pow := pow }.n),
    ⟦assign, { aig := aig, ref := { n := n, lhs := lvec, rhs := rvec, pow := pow }.rhs.get idx hidx }⟧ = rhs.getLsbD idx
hg : twoPowShift aig { n := n, lhs := lvec, rhs := rvec, pow := pow } = res
⊢ ⟦assign, { aig := res.aig, ref := res.vec.get idx hidx }⟧ =
    (lhs >>> if rhs.getLsbD pow = true then BitVec.twoPow n pow else 0#n).getLsbD idx

Tactic:
unfold twoPowShift at hg

Result:
case mk
α : Type
inst✝¹ : Hashable α
inst✝ : DecidableEq α
w : Nat
aig : AIG α
lhs : BitVec w
assign : α → Bool
idx : Nat
hidx : idx < w
res : RefVecEntry α w
n : Nat
lvec : aig.RefVec w
rvec : aig.RefVec n
pow : Nat
rhs : BitVec { n := n, lhs := lvec, rhs := rvec, pow := pow }.n
hleft :
  ∀ (idx : Nat) (hidx : idx < w),
    ⟦assign, { aig := aig, ref := { n := n, lhs := lvec, rhs := rvec, pow := pow }.lhs.get idx hidx }⟧ = lhs.getLsbD idx
hright :
  ∀ (idx : Nat) (hidx : idx < { n := n, lhs := lvec, rhs := rvec, pow := pow }.n),
    ⟦assign, { aig := aig, ref := { n := n, lhs := lvec, rhs := rvec, pow := pow }.rhs.get idx hidx }⟧ = rhs.getLsbD idx
hg :
  (match { n := n, lhs := lvec, rhs := rvec, pow := pow } with
    | { n := n, lhs := lhs, rhs := rhs, pow := pow } =>
      if h : pow < n then
        let res := blastShiftRightConst aig { vec := lhs, distance := 2 ^ pow };
        let aig_1 := res.aig;
        let shifted := res.vec;
        let_fun this := ⋯;
        let rhs := rhs.cast this;
        let lhs_1 := lhs.cast this;
        RefVec.ite aig_1 { discr := rhs.get pow h, lhs := shifted, rhs := lhs_1 }
      else { aig := aig, vec := lhs }) =
    res
⊢ ⟦assign, { aig := res.aig, ref := res.vec.get idx hidx }⟧ =
    (lhs >>> if rhs.getLsbD pow = true then BitVec.twoPow n pow else 0#n).getLsbD idx