package xiangshan.backend.fu.NewCSR

import chisel3._
import chisel3.util._
import chisel3.experimental.BundleLiterals._
import org.chipsalliance.cde.config.Parameters
import xiangshan.backend.fu.NewCSR.CSRDefines.{CSRROField => RO, CSRRWField => RW, CSRWARLField => WARL, _}
import xiangshan.backend.fu.NewCSR.CSRFunc._

object CSRBundles {
  class XtvecBundle extends CSRBundle {
    val mode = XtvecMode(1, 0, wNoFilter)
    val addr = WARL(63, 2, wNoFilter)
  }

  class CauseBundle extends CSRBundle {
    val Interrupt = RW(63)
    val ExceptionCode = RW(62, 0)
  }

  class Counteren extends CSRBundle {
    val CY = RW(0)
    val TM = RW(1)
    val IR = RW(2)
    val HPM = RW(31, 3)
  }

  class OneFieldBundle extends CSRBundle {
    val ALL = RW(63, 0)
  }

  class Envcfg extends CSRBundle {
    val STCE  = RO(    63).withReset(0.U)
    val PBMTE = RO(    62).withReset(0.U)
    val ADUE  = RO(    61).withReset(0.U)
    val PMM   = RO(33, 32).withReset(0.U)
    val CBZE  = RO(     7).withReset(0.U)
    val CBCFE = RO(     6).withReset(0.U)
    val CBIE  = RO( 5,  4).withReset(0.U)
    val FIOM  = RO(     0).withReset(0.U)
  }

  class PrivState extends Bundle {
    val PRVM = PrivMode(0)
    val V    = VirtMode(0)

    def isModeHU: Bool = this.V === VirtMode.Off && this.PRVM === PrivMode.U

    def isModeHS: Bool = this.V === VirtMode.Off && this.PRVM === PrivMode.S

    def isModeHUorHS: Bool = this.V === VirtMode.Off && this.PRVM.isOneOf(PrivMode.S, PrivMode.U)

    def isModeM: Bool = this.V === VirtMode.Off && this.PRVM === PrivMode.M

    def isModeVU: Bool = this.V === VirtMode.On && this.PRVM === PrivMode.U

    def isModeVS: Bool = this.V === VirtMode.On && this.PRVM === PrivMode.S

    def isModeHUorVU: Bool = this.PRVM === PrivMode.U

    def isVirtual: Bool = this.V === VirtMode.On

    // VU < VS < HS < M
    // HU < HS < M
    def < (that: PrivState): Bool = {
      (this.isVirtual && (that.isModeM || that.isModeHS)) ||
        (this.V === that.V && this.PRVM < that.PRVM)
    }

    def > (that: PrivState): Bool = {
      (that.isVirtual && (this.isModeM || this.isModeHS)) ||
        (that.V === this.V && that.PRVM < this.PRVM)
    }
  }

  object PrivState {
    def ModeM: PrivState = WireInit((new PrivState).Lit(
      _.PRVM -> PrivMode.M,
      _.V    -> VirtMode.Off,
    ))

    def ModeHS: PrivState = WireInit((new PrivState).Lit(
      _.PRVM -> PrivMode.S,
      _.V    -> VirtMode.Off,
    ))

    def ModeHU: PrivState = WireInit((new PrivState).Lit(
      _.PRVM -> PrivMode.U,
      _.V    -> VirtMode.Off,
    ))

    def ModeVS: PrivState = WireInit((new PrivState).Lit(
      _.PRVM -> PrivMode.S,
      _.V    -> VirtMode.On,
    ))

    def ModeVU: PrivState = WireInit((new PrivState).Lit(
      _.PRVM -> PrivMode.U,
      _.V    -> VirtMode.On,
    ))
  }
}