build.sbt

name := "LambdaNet"

version := "0.5.0"

ThisBuild/organization := "mrvplusone.github.io"
ThisBuild/scalaVersion := "2.12.13"

scalacOptions ++= Seq(
  "-feature",
  "-Ypartial-unification", // for using cats
  "-language:higherKinds"
//  "-deprecation"
)

// to make the classpath right
run/fork := true
run/connectInput := true  // for StdIn to work

val runOnMac = System.getProperty("os.name") == "Mac OS X"
val memories = {
  if(new File("configs/memory.txt").exists()) {
    val s = scala.io.Source.fromFile("configs/memory.txt")
    val r = s.getLines().toList.map(_.trim.toInt)
    s.close()
    r
  }else List(5,7) // default heap and off-heap limit
}
val heapLimit = memories.head
val offHeapLimit = memories(1)

val nd4jBinary = {
  val useCuda = new File("configs/useCuda.txt").exists()
  if (useCuda) "nd4j-cuda-10.0-platform" else "nd4j-native-platform"
}

javaOptions ++= Seq(
  s"-Xms2G",
  s"-Xmx${heapLimit}G",
  s"-Dorg.bytedeco.javacpp.maxbytes=${offHeapLimit}G",
  s"-Dorg.bytedeco.javacpp.maxphysicalbytes=${offHeapLimit + heapLimit + 1}G"
)

libraryDependencies ++= Seq(
  "com.lihaoyi" %% "fastparse" % "2.0.4",
  "org.scalacheck" %% "scalacheck" % "1.14.0",
  "org.scalatest" %% "scalatest" % "3.0.3" % Test,
  "ch.qos.logback" % "logback-classic" % "1.2.3",
  "com.lihaoyi" %% "ammonite-ops" % "1.0.3",
  "org.json4s" %% "json4s-native" % "3.6.3",
  "com.github.daddykotex" %% "courier" % "1.0.0", // for email notifications
  //  "be.botkop" %% "numsca" % "0.1.4",
  // for building numsca
  "org.nd4j" % nd4jBinary % "1.0.0-beta4",
  "com.typesafe.scala-logging" %% "scala-logging" % "3.7.2",
  "org.typelevel" %% "cats-core" % "2.0.0-M3" withSources(),
  "org.typelevel" %% "cats-effect" % "2.0.0-M3" withSources(),
  "com.github.nscala-time" %% "nscala-time" % "2.22.0",
  "com.lihaoyi" %% "upickle" % "1.3.8",
  "com.lihaoyi" %% "scalatags" % "0.7.0",
  "org.plotly-scala" %% "plotly-render" % "0.8.0",
  "me.tongfei" % "progressbar" % "0.9.2",
)

addCompilerPlugin("com.olegpy" %% "better-monadic-for" % "0.3.0")

val circeVersion = "0.10.0"
libraryDependencies ++= Seq(
  "io.circe" %% "circe-core",
  "io.circe" %% "circe-generic",
  "io.circe" %% "circe-parser"
).map(_ % circeVersion)

//// Add dependency on ScalaFX library
//libraryDependencies += "org.scalafx" %% "scalafx" % "14-R19"
//
//// Determine OS version of JavaFX binaries
//lazy val osName = System.getProperty("os.name") match {
//  case n if n.startsWith("Linux")   => "linux"
//  case n if n.startsWith("Mac")     => "mac"
//  case n if n.startsWith("Windows") => "win"
//  case _                            => throw new Exception("Unknown platform!")
//}

//// Add dependency on JavaFX libraries, OS dependent
//lazy val javaFXModules =
//  Seq("base", "controls", "fxml", "graphics", "media", "swing", "web")
//libraryDependencies ++= javaFXModules.map(m =>
//  "org.openjfx" % s"javafx-$m" % "14.0.1" classifier osName,
//)

// My tasks

val train = taskKey[Unit]("start training")
train :=
  (Compile/runMain).toTask(" lambdanet.train.Training").value

val prepareRepos =
  taskKey[Unit]("parse and prepare data for training and evaluation")
prepareRepos :=
  (Compile/runMain).toTask(" lambdanet.PrepareRepos").value

TaskKey[Unit]("prepareAndTrain") := Def.sequential(prepareRepos, train).value

val runTrained = taskKey[Unit]("run trained model")
runTrained :=
  (Compile/runMain).toTask(" lambdanet.RunTrainedModel").value

val service = taskKey[Unit]("run type inference service")
service :=
  (Compile/runMain).toTask(" lambdanet.TypeInferenceService").value

Compile/discoveredMainClasses += "driver.JavaDriver"