package BACK

import chisel3._
import chisel3.util._

class ManualParamBundle extends Bundle{
  val vendorID = UInt( 16.W )
  val moduleID = UInt( 16.W )
  val hwVersion = UInt( 16.W )
  val swVersion = UInt( 16.W )
  val batchID  = UInt( 16.W )
  val serial   = UInt( 48.W )
}


abstract class BackBoardSlvLocal extends BackBoardSlvBase with BackBoardSlvDown with BackBoardSlvUp{

  def DATA_SINWR_ADDR   : UInt = "h00".U
  def DATA_PERWR_ADDR   : UInt = "h02".U
  def STATUSWR_ADDR     : UInt = "h04".U

  def DATA_SINRD_ADDR   : UInt = "h00".U
  def DATA_PERRD_ADDR   : UInt = "h02".U
  def STATUSRD_ADDR     : UInt = "h04".U

  def VENDORID_ADDR     : UInt = "h10".U
  def MODULEID_ADDR     : UInt = "h12".U
  def HVERSION_ADDR     : UInt = "h14".U
  def SVERSION_ADDR     : UInt = "h16".U
  def BATCHNUM_ADDR     : UInt = "h18".U
  def SERIAL0_ADDR      : UInt = "h1a".U
  def SERIAL1_ADDR      : UInt = "h1c".U
  def SERIAL2_ADDR      : UInt = "h1e".U

  val param = IO(Input( new ManualParamBundle ))

  val rankNum    = downReg(4)
  val statusLed  = downReg(5)(1,0)

  val (us10Counter, us10Trigger) = Counter(Range(0, 500))

  when( true.B ){


    upReg(16) := param.vendorID(7,0)
    upReg(17) := param.vendorID(15,8)

    upReg(18) := param.moduleID(7,0)
    upReg(19) := param.moduleID(15,8)

    upReg(20) := param.hwVersion(7,0)
    upReg(21) := param.hwVersion(15,8)

    upReg(22) := param.swVersion(7,0)
    upReg(23) := param.swVersion(15,8)

    upReg(24) := param.batchID(7,0)
    upReg(25) := param.batchID(15,8)

    upReg(26) := param.serial(7,0)
    upReg(27) := param.serial(15,8)
    upReg(28) := param.serial(23,16)
    upReg(29) := param.serial(31,24)
    upReg(30) := param.serial(39,32)
    upReg(31) := param.serial(47,40)    
  }


}

trait BackBoardSlvStatusLed { this: BackBoardSlvLocal =>
  val flashCounter = RegInit( 0.U( 20.W ) )

  when( us10Trigger ){
    when( flashCounter === 100000.U ){
      flashCounter := 0.U
    } .otherwise{
      flashCounter := flashCounter + 1.U      
    }
  }

  val flash1HZ = flashCounter.extract(19)
  val flash2HZ = flashCounter.extract(18)

  val LED_PR = IO(Output( Bool() ))

  LED_PR := Mux1H(Seq(
    ( statusLed === "b00".U ) -> flash2HZ,
    ( statusLed === "b01".U ) -> flash1HZ, 
    ( statusLed === "b10".U ) -> true.B,
    ( statusLed === "b11".U ) -> false.B,
    ))
}



trait BackBoardSlvIn { this: BackBoardSlvLocal =>

  def gp: Int

  val flitCfg = Seq(
    ( 0.U  -> 0.S ),
    ( 1.U  -> 10.S),
    ( 2.U  -> 25.S),
    ( 3.U  -> 50.S),
    ( 4.U  -> 100.S),
    ( 5.U  -> 200.S),
    ( 6.U  -> 400.S),
    ( 7.U  -> 800.S),
    ( 8.U  -> 1600.S),
    ( 9.U  -> 3200.S),
    ( 10.U -> 6400.S),
  )

  val in = IO(Input(UInt((8*gp).W)))

  val digitalIn = Reg( Vec( 8*gp, Bool() ) )
  

  for( j <- 0 until gp ) {
    when( true.B ) { upReg( 2+j ) := Cat( ( 8*j+7 to 8*j ).map{ k => digitalIn(k) } ) }

    val digitalFilter = downReg(0+j)

    for( i <- 0 until 8 ) {
      val toCnt = RegInit( 0.S( 16.W ) )

      flitCfg.map{ ( cfg: (UInt,SInt) ) => 
        when( digitalFilter === cfg._1 ){
          when( in(i+8*j) ){
            when( toCnt >= cfg._2 ){
              toCnt := cfg._2
              digitalIn(i+8*j) := true.B
            } .otherwise{
              toCnt := toCnt + 1.S
            }
          } .otherwise{
            when( toCnt <= -cfg._2 ){
              toCnt := -cfg._2
              digitalIn(i+8*j) := false.B
            } .otherwise{
              toCnt := toCnt - 1.S
            }
          }
        }        
        
      }
    }
  }

}




class DIn16 extends BackBoardSlvLocal
with BackBoardSlvStatusLed
with BackBoardSlvIn{
  def gp = 2
}







class BackBoardSlvOut16 extends BackBoardSlvLocal{
  val in  = IO(Input(UInt(32.W)))
  val out = IO(Output(UInt(32.W)))
  val oe  = IO(Output(UInt(32.W)))



  out := Cat( (0 until 4).map{i => downReg(i)}.reverse )
  oe  := Cat( (4 until 8).map{i => downReg(i)}.reverse )

  for( i <- 0 until 4 ) {
    upReg(i) := in( i*8+7,8*i )
  }
}


// trait LocalRegisters { this: BackBoardSlvBase =>
//   val led = IO(Output(UInt(3.W)))
//   val sw  = IO( Input(UInt(1.W)))



//   val digitalDir = RegInit(0.U(32.W))
//   val digitalOut = RegInit(0.U(32.W))
//   val digitalIn  = RegInit(VecInit(Seq.fill(32){false.B}))
//   val digitalFilter = RegInit(0.U(2.W))


//   val clk100usCnt = RegInit(0.U(13.W))

//   when( clk100usCnt < 5000.U ){
//     clk100usCnt := clk100usCnt + 1.U
//   } .otherwise{
//     clk100usCnt := 0.U
//   }

//   for( i <- 0 until 32 ) {
//     val hiCnt = RegInit( 0.U( 7.W ) )
//     val loCnt = RegInit( 0.U( 7.W ) )
//     val toCnt = RegInit( 0.U( 7.W ) )

//     when( digitalFilter === "b00".U ){
//       digitalIn(i) := sw
//     } .elsewhen( digitalFilter === "b01".U ){
//       when( clk100usCnt === 5000.U ){
//         when( toCnt === 10.U ){
//           hiCnt := 0.U
//           loCnt := 0.U
//           toCnt := 0.U
//           when( hiCnt >= 8.U & digitalIn(i) === false.B){ digitalIn(i) := true.B }
//           when( loCnt >= 8.U & digitalIn(i) === true.B) { digitalIn(i) := false.B }
//         } .elsewhen( toCnt > 10.U ){
//           hiCnt := 0.U
//           loCnt := 0.U
//           toCnt := 0.U
//         } .otherwise{
//           when(sw === true.B)  { hiCnt := hiCnt + 1.U }
//           when(sw === false.B) { loCnt := loCnt + 1.U }
//           toCnt := toCnt + 1.U
//         }
//       }
//     } .elsewhen( digitalFilter === "b10".U ){
//       when( clk100usCnt === 5000.U ){
//         when( toCnt === 40.U ){
//           hiCnt := 0.U
//           loCnt := 0.U
//           toCnt := 0.U
//           when( hiCnt >= 32.U & digitalIn(i) === false.B){ digitalIn(i) := true.B }
//           when( loCnt >= 32.U & digitalIn(i) === true.B) { digitalIn(i) := false.B }
//         } .elsewhen( toCnt > 40.U ){
//           hiCnt := 0.U
//           loCnt := 0.U
//           toCnt := 0.U
//         } .otherwise{
//           when(sw === true.B)  { hiCnt := hiCnt + 1.U }
//           when(sw === false.B) { loCnt := loCnt + 1.U }
//           toCnt := toCnt + 1.U
//         }
//       }
//     } .elsewhen( digitalFilter === "b11".U ){
//       when( clk100usCnt === 5000.U ){
//         when( toCnt === 80.U ){
//           hiCnt := 0.U
//           loCnt := 0.U
//           toCnt := 0.U
//           when( hiCnt >= 64.U & digitalIn(i) === false.B){ digitalIn(i) := true.B }
//           when( loCnt >= 64.U & digitalIn(i) === true.B) { digitalIn(i) := false.B }
//         } .elsewhen( toCnt > 80.U ){
//           hiCnt := 0.U
//           loCnt := 0.U
//           toCnt := 0.U
//         } .otherwise{
//           when(sw === true.B)  { hiCnt := hiCnt + 1.U }
//           when(sw === false.B) { loCnt := loCnt + 1.U }
//           toCnt := toCnt + 1.U
//         }
//       }
//     }
//   }




  
//   val pluseCnt = RegInit(0.U(32.W))

//   val pluseShift = ShiftRegisters( sw, 3, false.B, true.B )
//   when( ~pluseShift(2).asBool & pluseShift(1).asBool ){
//     pluseCnt := pluseCnt + 1.U
//   }

//   val pluseAutoCnt = RegInit(0.U(32.W))
//   val pluseAutoFre = RegInit(0.U(32.W))
//   when( clk100usCnt === 5000.U ){
//     pluseAutoCnt := 0.U
//     pluseAutoFre := pluseAutoCnt
//   } .elsewhen( ~pluseShift(2).asBool & pluseShift(1).asBool ){
//     pluseAutoCnt := pluseAutoCnt + 1.U
//   }

//   val qeiA = IO(Input(Bool()))
//   val qeiB = IO(Input(Bool()))

//   val qeiAShift = ShiftRegisters( qeiA, 2, false.B, true.B )
//   val qeiBShift = ShiftRegisters( qeiB, 3, false.B, true.B )

//   val qeiCnt = RegInit(0.U(32.W))
//   val qeiAutoCnt = RegInit(0.U(32.W))
//   val qeiAutoFre = RegInit(0.U(32.W))

//   when( ~qeiBShift(2) & qeiBShift(1) ){
//     qeiCnt := Mux( qeiAShift(1), qeiCnt + 1.U, qeiCnt - 1.U )
//     qeiAutoCnt := Mux( qeiAShift(1), qeiCnt + 1.U, qeiCnt - 1.U )
//   }


//   when( clk100usCnt === 5000.U ){
//     qeiAutoCnt := 0.U
//     qeiAutoFre := qeiAutoCnt
//   } .elsewhen( ~qeiBShift(2) & qeiBShift(1) ){
//     qeiAutoCnt := Mux( qeiAShift(1), qeiAutoCnt + 1.U, qeiAutoCnt - 1.U )
//   }
  
  
//   registersRD := Mux1H(Seq(
//     (upStreamReq.io.latDat.bits.register === 0.U) -> digitalDir,
//     (upStreamReq.io.latDat.bits.register === 1.U) -> digitalIn.asUInt,
//     (upStreamReq.io.latDat.bits.register === 2.U) -> digitalFilter,
//     (upStreamReq.io.latDat.bits.register === 3.U) -> pluseCnt,
//     (upStreamReq.io.latDat.bits.register === 4.U) -> pluseAutoFre,
//     (upStreamReq.io.latDat.bits.register === 5.U) -> qeiCnt,
//     (upStreamReq.io.latDat.bits.register === 6.U) -> qeiAutoFre,
//   ))

//   when( upStreamReq.io.latDat.fire & upStreamReq.io.latDat.bits.address(5,1) === io.localHost & upStreamReq.io.latDat.bits.address.extract(0) === 1.U ){
//     when(upStreamReq.io.latDat.bits.register === 0.U){
//       digitalDir := upStreamReq.io.latDat.bits.data
//     }
//     when(upStreamReq.io.latDat.bits.register === 1.U){
//       digitalOut := upStreamReq.io.latDat.bits.data
//     }
//     when(upStreamReq.io.latDat.bits.register === 2.U){
//       digitalFilter := upStreamReq.io.latDat.bits.data
//     }
//     when(upStreamReq.io.latDat.bits.register === 3.U){
//       pluseCnt := upStreamReq.io.latDat.bits.data
//     }
//     when(upStreamReq.io.latDat.bits.register === 4.U){
//       ;
//     }
//     when(upStreamReq.io.latDat.bits.register === 5.U){
//       qeiCnt := upStreamReq.io.latDat.bits.data
//     }
//     when(upStreamReq.io.latDat.bits.register === 6.U){
//       ;
//     }
//   }




//   led := digitalOut(2,0)
//   // digitalIn := sw
// }