// See LICENSE for license details.

package unittest

import Chisel._
import cde.{Parameters, Config, CDEMatchError}
import rocketchip.{BaseConfig, BasePlatformConfig}

class WithJunctionsUnitTests extends Config(
  (pname, site, here) => pname match {
    case junctions.PAddrBits => 32
    case rocket.XLen => 64
    case UnitTests => (p: Parameters) => Seq(
      Module(new junctions.MultiWidthFifoTest),
      Module(new junctions.HastiTest()(p)))
    case _ => throw new CDEMatchError
  })

class JunctionsUnitTestConfig extends Config(new WithJunctionsUnitTests ++ new BasePlatformConfig)

class WithUncoreUnitTests extends Config(
  (pname, site, here) => pname match {
    case rocketchip.NCoreplexExtClients => 0
    case uncore.tilelink.TLId => "L1toL2"
    case UnitTests => (p: Parameters) => Seq(
      Module(new uncore.devices.ROMSlaveTest()(p)),
      Module(new uncore.devices.TileLinkRAMTest()(p)),
      Module(new uncore.tilelink2.TLFuzzRAMTest))
    case _ => throw new CDEMatchError
  }
)

class UncoreUnitTestConfig extends Config(new WithUncoreUnitTests ++ new BaseConfig)

class WithTL2UnitTests extends Config(
  (pname, site, here) => pname match {
    case UnitTests => (p: Parameters) => {
      Seq(
        //Module(new uncore.tilelink2.TLRAMSimpleTest(1)),
        //Module(new uncore.tilelink2.TLRAMSimpleTest(4)),
        //Module(new uncore.tilelink2.TLRAMSimpleTest(16)),
        Module(new uncore.tilelink2.TLRAMFragmenterTest(4, 256)),
        Module(new uncore.tilelink2.TLRAMFragmenterTest(16, 64)),
        Module(new uncore.tilelink2.TLRAMFragmenterTest(4, 16)),
        Module(new uncore.tilelink2.TLRAMXbarTest(1)),
        Module(new uncore.tilelink2.TLRAMXbarTest(2)),
        Module(new uncore.tilelink2.TLRAMXbarTest(8)),
        //Module(new uncore.tilelink2.TLMulticlientXbarTest(4,4)),
        //Module(new uncore.tilelink2.TLMulticlientXbarTest(1,4)),
        Module(new uncore.tilelink2.TLRAMWidthWidgetTest(1,1)),
        Module(new uncore.tilelink2.TLRAMWidthWidgetTest(4,4)),
        Module(new uncore.tilelink2.TLRAMWidthWidgetTest(16,16)),
        Module(new uncore.tilelink2.TLRAMWidthWidgetTest(4,64)),
        Module(new uncore.tilelink2.TLRAMWidthWidgetTest(64,4)),
        Module(new uncore.tilelink2.TLRR0Test),
        Module(new uncore.tilelink2.TLRR1Test),
        Module(new uncore.tilelink2.TLRAMCrossingTest)
      )
    }
    case _ => throw new CDEMatchError
  })

class TL2UnitTestConfig extends Config(new WithTL2UnitTests ++ new BasePlatformConfig)