Amaranth in Hy

counter.hy

(import amaranth *)

(defn domain [domain #* expr] (domain.__iadd__ expr))

(defmacro mwhen [m condition #* body]
  `(do
    (with [(m.If ~condition)]
      ~body)))

(defmacro mif [m condition pos neg]
  `(do
    (with [(m.If ~condition)]
      ~pos)
    (with [(m.Else)]
      ~neg)))
      

(defclass UpCounter [Elaboratable]
  (defn __init__ [self limit]
    (setv
      self.limit limit
      self.en (Signal)
      self.ovf (Signal)
      self.count (Signal 16)))
  (defn elaborate [self platform]
    (let [m (Module)]
      (domain m.d.comb (self.ovf.eq (= self.count self.limit)))
      (mwhen m self.en
        (mif m self.ovf
          (domain m.d.sync (self.count.eq 0))
          (domain m.d.sync (self.count.eq (+ self.count 1)))))
      m)))