kotlin-fork/kotlin-native/backend.native/tests/interop/basics/vectors.kt

import kotlinx.cinterop.*
import kotlin.native.*
import kotlin.test.*
import cvectors.*

fun isWin32() = Platform.osFamily == OsFamily.WINDOWS && Platform.cpuArchitecture == CpuArchitecture.X86

fun main(args: Array<String>) {
    val mimalloc = args.takeIf { it.size == 1 }?.get(0) == "mimalloc"

    // See KT-37272. Fixed in mimalloc
    if (mimalloc || !isWin32()) {
        produceComplex().useContents {
            assertEquals(vec4f, vectorOf(1.0f, 1.0f, 1.0f, 1.0f))
            vec4f = vectorOf(0.0f, 0.0f, 0.0f, 0.0f)
            assertEquals(vec4f, vectorOf(0.0f, 0.0f, 0.0f, 0.0f))
        }
    }

    // FIXME: KT-36285
    if (Platform.osFamily != OsFamily.LINUX || Platform.cpuArchitecture != CpuArchitecture.ARM32) {
        assertEquals(49, sendV4I(vectorOf(1, 2, 3, 4)))
    }
    assertEquals(49, (sendV4F(vectorOf(1f, 2f, 3f, 4f)) + 0.00001).toInt())

    if (mimalloc || !isWin32()) {
        memScoped {
            val vector = alloc<KVector4i32Var>().also {
                it.value = vectorOf(1, 2, 3, 4)
            }
            assertEquals(vector.value, vectorOf(1, 2, 3, 4))
        }
    }
}