refactor main App for better code re-use

project/build.scala

@@ -37,7 +37,7 @@ object BuildSettings extends Build {
       a.split(" ")
     },
     unmanagedSourceDirectories in Compile ++= addons.value.map(baseDirectory.value / _ / "src/main/scala"),
-    mainClass in (Compile, run) := Some("rocketchip.TestGenerator"),
+    mainClass in (Compile, run) := Some("rocketchip.RocketChipGenerator"),
     make := {
       val jobs = java.lang.Runtime.getRuntime.availableProcessors
       val (makeDir, target) = setMake.parsed

src/main/scala/Generator.scala

@@ -4,17 +4,14 @@ package rocketchip
 
 import Chisel._
 import scala.collection.mutable.{LinkedHashSet,LinkedHashMap}
-import cde.{Parameters, ParameterDump, Config, Field, CDEMatchError}
+import cde.{Parameters, ParameterDump, Config, Field, CDEMatchError, World}
 import coreplex._
 
-object TestGenerator extends App {
+object GeneratorUtils {
-  val projectName = args(0)
+  def getConfig(projectName: String, configClassName: String): Config = {
-  val topModuleName = args(1)
-  val configClassName = args(2)
-
     val aggregateConfigs = configClassName.split('_')
 
-  val finalConfig = aggregateConfigs.foldRight(new Config()) { case (currentConfigName, finalConfig) =>
+    aggregateConfigs.foldRight(new Config()) { case (currentConfigName, finalConfig) =>
       val currentConfig = try {
         Class.forName(s"$projectName.$currentConfigName").newInstance.asInstanceOf[Config]
       } catch {
@@ -25,37 +22,67 @@ object TestGenerator extends App {
       }
       currentConfig ++ finalConfig
     }
-  val world = finalConfig.toInstance
+  }
 
-  val paramsFromConfig: Parameters = Parameters.root(world)
+  def getParameters(config: Config): Parameters =
+    Parameters.root(config.toInstance)
 
+  def getParameters(projectName: String, configClassName: String): Parameters =
+    getParameters(getConfig(projectName, configClassName))
+
+  def elaborate(fullName: String, args: Array[String], params: Parameters) {
     val gen = () =>
-    Class.forName(s"$projectName.$topModuleName")
+      Class.forName(fullName)
         .getConstructor(classOf[cde.Parameters])
-      .newInstance(paramsFromConfig)
+        .newInstance(params)
         .asInstanceOf[Module]
 
-  chiselMain.run(args.drop(3), gen)
+    chiselMain.run(args, gen)
-  //Driver.elaborate(gen, configName = configClassName)
+  }
+
+  def dumpParameters(fname: String) {
+    val pdFile = TestGeneration.createOutputFile(fname)
+    pdFile.write(ParameterDump.getDump)
+    pdFile.close
+  }
+
+  def dumpKnobs(configClassName: String, world: World) {
+    val knbFile = TestGeneration.createOutputFile(configClassName + ".knb")
+    knbFile.write(world.getKnobs)
+    knbFile.close
+
+    val cstFile = TestGeneration.createOutputFile(configClassName + ".cst")
+    cstFile.write(world.getConstraints)
+    cstFile.close
+  }
+
+  def dumpConfigString(fname: String, params: Parameters) {
+    val cfgFile = new java.io.FileOutputStream(Driver.targetDir + "/" + fname)
+    cfgFile.write(params(ConfigString))
+    cfgFile.close
+  }
+}
+import GeneratorUtils._
+
+object RocketChipGenerator extends App {
+  val projectName = args(0)
+  val topModuleName = args(1)
+  val configClassName = args(2)
+
+  val config = getConfig(projectName, configClassName)
+  val world = config.toInstance
+  val paramsFromConfig = Parameters.root(world)
+
+  elaborate(s"$projectName.$topModuleName", args.drop(3), paramsFromConfig)
 
   TestGeneration.addSuite(new RegressionTestSuite(paramsFromConfig(RegressionTestNames)))
-
   TestGeneration.generateMakefrag(topModuleName, configClassName)
   TestBenchGeneration.generateVerilogFragment(
     topModuleName, configClassName, paramsFromConfig)
   TestBenchGeneration.generateCPPFragment(
     topModuleName, configClassName, paramsFromConfig)
 
-  val pdFile = TestGeneration.createOutputFile(s"$topModuleName.$configClassName.prm")
+  dumpParameters(s"$topModuleName.$configClassName.prm")
-  pdFile.write(ParameterDump.getDump)
+  dumpKnobs(configClassName, world)
-  pdFile.close
+  dumpConfigString(s"$configClassName.cfg", paramsFromConfig)
-  val v = TestGeneration.createOutputFile(configClassName + ".knb")
-  v.write(world.getKnobs)
-  v.close
-  val d = new java.io.FileOutputStream(Driver.targetDir + "/" + configClassName + ".cfg")
-  d.write(paramsFromConfig(ConfigString))
-  d.close
-  val w = TestGeneration.createOutputFile(configClassName + ".cst")
-  w.write(world.getConstraints)
-  w.close
 }