package xiangshan.backend.fu.NewCSR.CSREvents

import chisel3._
import chisel3.util._
import utility.{SignExt, ZeroExt}
import xiangshan.ExceptionNO
import xiangshan.ExceptionNO._
import xiangshan.backend.fu.NewCSR.CSRBundles.{CauseBundle, OneFieldBundle, PrivState}
import xiangshan.backend.fu.NewCSR.CSRConfig.{VaddrMaxWidth, XLEN}
import xiangshan.backend.fu.NewCSR.CSRDefines.{PrivMode, SatpMode, VirtMode}
import xiangshan.backend.fu.NewCSR._


class MretEventOutput extends Bundle with EventUpdatePrivStateOutput with EventOutputBase {
  val mstatus  = ValidIO((new MstatusBundle).addInEvent(_.MPP, _.MPV, _.MIE, _.MPIE, _.MPRV))
  val tcontrol = ValidIO((new TcontrolBundle).addInEvent(_.MTE))
  val targetPc = ValidIO(UInt(VaddrMaxWidth.W))

  override def getBundleByName(name: String): ValidIO[CSRBundle] = {
    name match {
      case "mstatus"  => this.mstatus
      case "tcontrol" => this.tcontrol
    }
  }
}

class MretEventInput extends Bundle {
  val mstatus  = Input(new MstatusBundle)
  val mepc     = Input(new Epc())
  val tcontrol = Input(new TcontrolBundle)
}

class MretEventModule extends Module with CSREventBase {
  val in = IO(new MretEventInput)
  val out = IO(new MretEventOutput)

  out := DontCare

  out.privState.valid := valid
  out.mstatus  .valid := valid
  out.tcontrol .valid := valid
  out.targetPc .valid := valid

  out.privState.bits.PRVM := in.mstatus.MPP
  out.privState.bits.V    := Mux(in.mstatus.MPP === PrivMode.M, VirtMode.Off.asUInt, in.mstatus.MPV.asUInt)
  out.mstatus.bits.MPP    := PrivMode.U
  out.mstatus.bits.MIE    := in.mstatus.MPIE
  out.mstatus.bits.MPIE   := 1.U
  out.mstatus.bits.MPRV   := Mux(in.mstatus.MPP =/= PrivMode.M, 0.U, in.mstatus.MPRV.asUInt)
  out.tcontrol.bits.MTE   := in.tcontrol.MPTE
  out.targetPc.bits       := in.mepc.asUInt
}

trait MretEventSinkBundle { self: CSRModule[_] =>
  val retFromM = IO(Flipped(new MretEventOutput))

  private val updateBundle: ValidIO[CSRBundle] = retFromM.getBundleByName(self.modName.toLowerCase())

  (reg.asInstanceOf[CSRBundle].getFields zip updateBundle.bits.getFields).foreach { case (sink, source) =>
    if (updateBundle.bits.eventFields.contains(source)) {
      when(updateBundle.valid) {
        sink := source
      }
    }
  }

}