-- 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
        ⟧
          =
        (BitVec.sshiftRight' 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 (blastArithShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).aig
      { discr := (rvec.cast ⋯).get pow h✝,
        lhs := (blastArithShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).vec, rhs := lvec.cast ⋯ } =
    res
hif1 : rhs.getLsbD pow = true
⊢ (if ⟦assign, { aig := aig, ref := { gate := (rvec.get pow h✝).gate, hgate := ?m.184267 } }⟧ = true then
      if h : 2 ^ pow + idx < w then ⟦assign, { aig := aig, ref := lvec.get (2 ^ pow + idx) ⋯ }⟧
      else ⟦assign, { aig := aig, ref := lvec.get (w - 1) ⋯ }⟧
    else
      ⟦assign,
        { aig := (blastArithShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).aig,
          ref := { gate := (lvec.get idx hidx).gate, hgate := ⋯ } }⟧) =
    (lhs.sshiftRight' (BitVec.twoPow n pow)).getLsbD idx

α : 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 (blastArithShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).aig
      { discr := (rvec.cast ⋯).get pow h✝,
        lhs := (blastArithShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).vec, rhs := lvec.cast ⋯ } =
    res
hif1 : rhs.getLsbD pow = true
⊢ (rvec.get pow h✝).gate < aig.decls.size

Tactic:
rw [hright]

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 (blastArithShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).aig
      { discr := (rvec.cast ⋯).get pow h✝,
        lhs := (blastArithShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).vec, rhs := lvec.cast ⋯ } =
    res
hif1 : rhs.getLsbD pow = true
⊢ (if rhs.getLsbD pow = true then
      if h : 2 ^ pow + idx < w then ⟦assign, { aig := aig, ref := lvec.get (2 ^ pow + idx) ⋯ }⟧
      else ⟦assign, { aig := aig, ref := lvec.get (w - 1) ⋯ }⟧
    else
      ⟦assign,
        { aig := (blastArithShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).aig,
          ref := { gate := (lvec.get idx hidx).gate, hgate := ⋯ } }⟧) =
    (lhs.sshiftRight' (BitVec.twoPow n 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 (blastArithShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).aig
      { discr := (rvec.cast ⋯).get pow h✝,
        lhs := (blastArithShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).vec, rhs := lvec.cast ⋯ } =
    res
hif1 : rhs.getLsbD pow = true
⊢ (if h : 2 ^ pow + idx < w then ⟦assign, { aig := aig, ref := lvec.get (2 ^ pow + idx) ⋯ }⟧
    else ⟦assign, { aig := aig, ref := lvec.get (w - 1) ⋯ }⟧) =
    (lhs.sshiftRight' (BitVec.twoPow n pow)).getLsbD idx

Tactic:
have hmod : 2 ^ pow % 2 ^ n = 2 ^ pow := by
  apply Nat.mod_eq_of_lt
  apply Nat.pow_lt_pow_of_lt <;> omega

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 (blastArithShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).aig
      { discr := (rvec.cast ⋯).get pow h✝,
        lhs := (blastArithShiftRightConst aig { vec := lvec, distance := 2 ^ pow }).vec, rhs := lvec.cast ⋯ } =
    res
hif1 : rhs.getLsbD pow = true
hmod : 2 ^ pow % 2 ^ n = 2 ^ pow
⊢ (if h : 2 ^ pow + idx < w then ⟦assign, { aig := aig, ref := lvec.get (2 ^ pow + idx) ⋯ }⟧
    else ⟦assign, { aig := aig, ref := lvec.get (w - 1) ⋯ }⟧) =
    (lhs.sshiftRight' (BitVec.twoPow n pow)).getLsbD idx