package rocket import Chisel._ import Node._; class ioQueue[T <: Data](flushable: Boolean)(data: => T) extends Bundle { val flush = if (flushable) Bool(INPUT) else null val enq = new ioDecoupled()(data).flip val deq = new ioDecoupled()(data) } class queue[T <: Data](entries: Int, pipe: Boolean = false, flushable: Boolean = false)(data: => T) extends Component { val io = new ioQueue(flushable)(data) val do_enq = io.enq.ready && io.enq.valid val do_deq = io.deq.ready && io.deq.valid var enq_ptr = UFix(0) var deq_ptr = UFix(0) if (entries > 1) { enq_ptr = Reg(resetVal = UFix(0, log2up(entries))) deq_ptr = Reg(resetVal = UFix(0, log2up(entries))) val pow2 = Bool((entries & (entries-1)) == 0) when (do_deq) { deq_ptr := Mux(!pow2 && deq_ptr === UFix(entries-1), UFix(0), deq_ptr + UFix(1)) } when (do_enq) { enq_ptr := Mux(!pow2 && enq_ptr === UFix(entries-1), UFix(0), enq_ptr + UFix(1)) } if (flushable) { when (io.flush) { deq_ptr := UFix(0) enq_ptr := UFix(0) } } } val maybe_full = Reg(resetVal = Bool(false)) when (do_enq != do_deq) { maybe_full := do_enq } if (flushable) { when (io.flush) { maybe_full := Bool(false) } } io.deq.valid := maybe_full || enq_ptr != deq_ptr io.enq.ready := !maybe_full || enq_ptr != deq_ptr || (if (pipe) io.deq.ready else Bool(false)) io.deq.bits <> Mem(entries, do_enq, enq_ptr, io.enq.bits).read(deq_ptr) } object Queue { def apply[T <: Data](enq: ioDecoupled[T], entries: Int = 2, pipe: Boolean = false) = { val q = (new queue(entries, pipe)) { enq.bits.clone } q.io.enq <> enq q.io.deq } } class pipereg[T <: Data]()(data: => T) extends Component { val io = new Bundle { val enq = new ioPipe()(data).flip val deq = new ioPipe()(data) } //val bits = Reg() { io.enq.bits.clone } //when (io.enq.valid) { // bits := io.enq.bits //} io.deq.valid := Reg(io.enq.valid, resetVal = Bool(false)) io.deq.bits <> Mem(1, io.enq.valid, UFix(0), io.enq.bits).read(UFix(0)) } object Pipe { def apply[T <: Data](enq: ioPipe[T], latency: Int = 1): ioPipe[T] = { val q = (new pipereg) { enq.bits.clone } q.io.enq <> enq q.io.deq if (latency > 1) Pipe(q.io.deq, latency-1) else q.io.deq } }