Add support for Deep Java Library NDArray

Author: Martmists-GH

README.md

@@ -12,7 +12,7 @@ repositories {
 }
 
 dependencies {
-    implementation("com.martmists.ndarray-simd:ndarray-simd:1.5.5")
+    implementation("com.martmists.ndarray-simd:ndarray-simd:1.6.0")
 }
 ```
 

build.gradle.kts

@@ -17,7 +17,7 @@ plugins {
 }
 
 group = "com.martmists.ndarray-simd"
-version = "1.5.5"
+version = "1.6.0"
 val isProduction = (findProperty("production") ?: System.getProperty("production")) != null
 
 repositories {
@@ -282,6 +282,9 @@ kotlin {
 
                 // Compat: kotlinx.dataframe
                 compileOnly("org.jetbrains.kotlinx:dataframe-core:0.13.1")
+
+                // Compat: Deep Java Library
+                compileOnly("ai.djl:api:0.32.0")
             }
         }
 
@@ -293,10 +296,12 @@ kotlin {
                 implementation("com.sksamuel.scrimage:scrimage-core:4.1.3")
                 implementation("dev.langchain4j:langchain4j:0.32.0")
                 implementation("org.jetbrains.kotlinx:dataframe-core:0.13.1")
+                implementation("ai.djl:api:0.32.0")
 
                 implementation("org.jetbrains.exposed:exposed-core:0.60.0")
                 implementation("org.jetbrains.exposed:exposed-jdbc:0.60.0")
                 implementation("org.xerial:sqlite-jdbc:3.44.1.0")
+
             }
         }
 

src/jvmMain/kotlin/com/martmists/ndarray/simd/compat/djl.kt

@@ -0,0 +1,28 @@
+package com.martmists.ndarray.simd.compat
+
+import ai.djl.ndarray.NDArray
+import ai.djl.ndarray.NDManager
+import ai.djl.ndarray.types.Shape
+import com.martmists.ndarray.simd.F64Array
+
+private val fallbackManager by lazy {
+    NDManager.newBaseManager()
+}
+
+/**
+ * Converts an [NDArray] to an [F64Array].
+ * @since 1.6.0
+ */
+fun NDArray.toF64Array(): F64Array {
+    val shape = IntArray(shape.dimension()) { shape.get(it).toInt() }
+    return F64Array.of(toDoubleArray()).reshape(*shape)
+}
+
+/**
+ * Converts an [F64Array] to a [NDArray].
+ * @param manager An optional manager to use. Falls back to [NDManager.newBaseManager] if null.
+ * @since 1.6.0
+ */
+fun F64Array.toDJL(manager: NDManager? = null): NDArray {
+    return (manager ?: fallbackManager).create(flatten().toDoubleArray(), Shape(shape.map(Int::toLong)))
+}

src/jvmTest/kotlin/DJLTest.kt

@@ -0,0 +1,23 @@
+import com.martmists.ndarray.simd.F64Array
+import com.martmists.ndarray.simd.compat.fromImage
+import com.martmists.ndarray.simd.compat.image
+import com.martmists.ndarray.simd.compat.toDJL
+import com.martmists.ndarray.simd.compat.toF64Array
+import com.martmists.ndarray.simd.compat.toImage
+import com.martmists.ndarray.simd.compat.toOpenCVMat
+import org.opencv.core.CvType
+import java.io.File
+import kotlin.test.Test
+import kotlin.test.assertEquals
+
+class DJLTest {
+    private val image = F64Array.fromImage(File(this::class.java.getResource("sample.png")!!.toURI()))
+
+    @Test
+    fun `Test DJL conversion`() {
+        val nd = image.toDJL()
+        val img = nd.toF64Array()
+        img.neqInPlace(image)
+        assertEquals(0.0, img.sum())
+    }
+}