123456// This function returns a counter def genCounter(n: Int) = { val cntReg = RegInit(0.U(8.W)) cntReg := Mux(cntReg === n.U, 0.U, cntReg + 1.U) cntReg}