iosMain.net.iriscan.sdk.utils.ImageExt.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of biometric-sdk Show documentation

Biometric SDK Kotlin Multiplatform library (android + ios)

There is a newer version: 0.6.0

package net.iriscan.sdk.utils

import cnames.structs.CGImage
import kotlinx.cinterop.*
import net.iriscan.sdk.core.image.*
import platform.CoreGraphics.*

/**
 * @author Slava Gornostal
 */
internal fun imageToCGImage(image: Image): CGImage {
    val width = image.width.toULong()
    val height = image.height.toULong()
    val pixelCount = width * height
    val data = nativeHeap.allocArray(pixelCount.toInt())
    for (i in 0 until pixelCount.toInt()) {
        val rgb = image.colors[i]
        val rgba = (rgb.red() shl 24) or (rgb.green() shl 16) or (rgb.blue() shl 8) or 0
        data[i] = rgba.toUInt()
    }
    val colorSpace = CGColorSpaceCreateDeviceRGB()
    val bitmapInfo = CGImageAlphaInfo.kCGImageAlphaNoneSkipFirst.value or kCGBitmapByteOrder32Big
    val context = CGBitmapContextCreate(data, width, height, 8u, width * 4u, colorSpace, bitmapInfo)
    val cgImage = CGBitmapContextCreateImage(context)
    CGContextRelease(context)
    nativeHeap.free(data)
    return cgImage!!.pointed
}

internal fun cgImageToImage(image: CGImage): Image {
    val width = CGImageGetWidth(image.ptr)
    val height = CGImageGetHeight(image.ptr)
    val pixelCount = width * height
    val data = nativeHeap.allocArray(pixelCount.toInt())
    val colorSpace = CGColorSpaceCreateDeviceRGB()
    val bitmapInfo = CGImageAlphaInfo.kCGImageAlphaNoneSkipFirst.value or kCGBitmapByteOrder32Big
    val context = CGBitmapContextCreate(data, width, height, 8u, width * 4u, colorSpace, bitmapInfo)
    CGContextDrawImage(context, CGRectMake(0.0, 0.0, width.toDouble(), height.toDouble()), image.ptr)
    val pixels = IntArray(pixelCount.toInt())
    for (i in pixels.indices) {
        val rgba = data[i]
        val r = (rgba and 0xFF000000u).toInt() shr 24
        val g = (rgba and 0x00FF0000u).toInt() shr 16
        val b = (rgba and 0x0000FF00u).toInt() shr 8
        pixels[i] = createColor(r, g, b)
    }
    CGContextRelease(context)
    nativeHeap.free(data)
    return Image(width.toInt(), height.toInt(), ImageColorType.RGB, pixels)
}