build.sbt

lazy val scalaVer = "2.13.6"
name := "kojo-ai"

version := "0.8.0"

scalaVersion := scalaVer

scalacOptions := Seq("-feature", "-deprecation")

val djlVersion = "0.30.0"
val pytorchVersion = "2.4.0"
val javacppVersion = "1.5.10"


// Platform classifier for native library dependencies
val platform = org.bytedeco.javacpp.Loader.Detector.getPlatform

// JavaCPP-Preset libraries with native dependencies
val presetLibs = Seq(
  "opencv"   -> "4.9.0",
  "ffmpeg"   -> "6.1.1",
  "openblas" -> "0.3.26"
).flatMap { case (lib, ver) =>
  Seq(
    "org.bytedeco" % lib % s"$ver-$javacppVersion",
    "org.bytedeco" % lib % s"$ver-$javacppVersion" classifier platform
  )
}

libraryDependencies ++= Seq(
  "ai.djl" % "api" % djlVersion,
  "ai.djl" % "basicdataset" % djlVersion,
  "ai.djl" % "model-zoo" % djlVersion,
  "ai.djl.pytorch" % "pytorch-model-zoo" % djlVersion,
  "ai.djl.pytorch" % "pytorch-native-cpu" % pytorchVersion % Runtime classifier platform,
  "ai.djl.pytorch" % "pytorch-jni" % s"$pytorchVersion-$djlVersion" % Runtime,
//  "ai.djl.tensorflow" % "tensorflow-model-zoo" % djlVersion,
//  "ai.djl.tensorflow" % "tensorflow-native-cpu" % "2.7.4" % Runtime classifier platform,
  "org.apache.logging.log4j" % "log4j-slf4j-impl" % "2.23.1",
  "commons-cli" % "commons-cli" % "1.5.0",
  "org.knowm.xchart" % "xchart" % "3.8.5",
  "org.scala-lang" % "scala-reflect" % scalaVer,
  "tech.tablesaw" % "tablesaw-core" % "0.43.1",
  "org.apache.commons" % "commons-math3" % "3.6.1",
  "org.bytedeco" % "javacpp" % javacppVersion,
  "org.bytedeco" % "javacpp" % javacppVersion classifier platform,
  "org.bytedeco" % "javacv" % javacppVersion,
) ++ presetLibs

//Build distribution
val distOutpath             = settingKey[File]("Where to copy all dependencies and kojo")
val buildDist  = taskKey[Unit]("Copy runtime dependencies and built kojo to 'distOutpath'")

lazy val dist = project
  .in(file("."))
  .settings(
    distOutpath              := baseDirectory.value / "dist",
    buildDist   := {
      val allLibs:                List[File]          = (Runtime / dependencyClasspath).value.map(_.data).filter(_.isFile).toList
      val buildArtifact:          File                = (Runtime / packageBin).value
      val jars:                   List[File]          = buildArtifact :: allLibs
      val `mappings src->dest`:   List[(File, File)]  = jars.map(f => (f, distOutpath.value / f.getName))
      val log                                         = streams.value.log
      log.info(s"Copying to ${distOutpath.value}:")
      log.info(s"${`mappings src->dest`.map(f => s" * ${f._1}").mkString("\n")}")
      IO.copy(`mappings src->dest`)
    }
  )