Mercurial > cgi-bin > hgweb.cgi > SimpleResizer
diff app/src/main/java/com/bartsent/simpleresizer/lib/Resizer.kt @ 7:9374d044a132 concur
Concurrency attempt, NOT WORKING.
author | David Barts <n5jrn@me.com> |
---|---|
date | Wed, 17 Feb 2021 07:24:26 -0800 |
parents | |
children | 884092efe31a |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/app/src/main/java/com/bartsent/simpleresizer/lib/Resizer.kt Wed Feb 17 07:24:26 2021 -0800 @@ -0,0 +1,140 @@ +package com.bartsent.simpleresizer.lib + +import android.graphics.Bitmap +import android.graphics.Color +import java.io.Closeable +import kotlin.math.ceil +import kotlin.math.floor + +class Resizer: Closeable { + private val NWORKERS = maxOf(1, Runtime.getRuntime().availableProcessors() - 2) + private val workers = Channel<Worker>(NWORKERS) + init { + for (i in 0 until NWORKERS) + workers.write(Worker(workers)) + } + + override fun close() { + for (i in 0 until NWORKERS) + workers.read().run { + interrupt() + join() + } + } + + private fun drain(): Unit { + val saved = Array<Worker>(NWORKERS) { workers.read() } + saved.forEach { workers.write(it) } + } + + private data class IndexWeight(var index: Int, var weight: Double) + + private fun precomputeWeights(dstSize: Int, srcSize: Int, filter: ScalingKernel): Array<Array<IndexWeight>> { + val du = srcSize.toDouble() / dstSize.toDouble() + val scale = maxOf(1.0, du) + val ru = ceil(scale * filter.size) + val TEMPLATE = arrayOf<IndexWeight>() + val out = Array<Array<IndexWeight>>(dstSize) { TEMPLATE } + val tmp = ArrayList<IndexWeight>((ru.toInt()+2)*2) + val emax = srcSize - 1 + + for (v in 0 until dstSize) { + val fu = (v.toDouble()+0.5)*du - 0.5 + val begin = maxOf(0, ceil(fu - ru).toInt()) + val end = minOf(emax, floor(fu + ru).toInt()) + var sum: Double = 0.0 + for (u in begin..end) { + val w = filter.weight((u.toDouble() - fu) / scale) + if (w != 0.0) { + sum += w + tmp.add(IndexWeight(u, w)) + } + } + if (sum != 0.0) { + tmp.forEach { + it.weight /= sum + } + } + out[v] = tmp.toArray(TEMPLATE) + tmp.clear() + } + return out + } + + private data class WorkUnit(val target: Bitmap, val x: Int, val y: Int, + val weights: Array<IndexWeight>, val scanLine: IntArray) + + private class Worker(private val workers: Channel<Worker>): Thread() { + private val _input = Channel<WorkUnit>(1) + + override fun run() { + while (true) { + val data = _input.read() + if (isInterrupted) + return + resample(data.target, data.x, data.y, data.weights, data.scanLine) + workers.write(this) + } + } + + fun send(data: WorkUnit) { + _input.write(data) + } + + private fun clamp(v: Double): Int = minOf(255, maxOf(0, (v + 0.5).toInt())) + + private fun resample(target: Bitmap, x: Int, y: Int, weights: Array<IndexWeight>, scanLine: IntArray): Unit { + var r = 0.0; var g = 0.0; var b = 0.0; var a = 0.0 + weights.forEach { + val c = scanLine[it.index] + val aw = Color.alpha(c).toDouble() * it.weight + r += Color.red(c).toDouble() * aw + g += Color.green(c).toDouble() * aw + b += Color.blue(c).toDouble() * aw + a += aw + } + if (a == 0.0) + return + val argb = Color.argb(clamp(a), clamp(r/a), clamp(g/a), clamp(b/a)) + synchronized(target) { + target.setPixel(x, y, argb) + } + } + } + + private fun resample(target: Bitmap, x: Int, y: Int, weights: Array<IndexWeight>, scanLine: IntArray): Unit { + workers.read().send(WorkUnit(target, x, y, weights, scanLine)) + } + + fun horizontal(image: Bitmap, newWidth: Int, kernel: ScalingKernel): Bitmap { + val dst = Bitmap.createBitmap(newWidth, image.height, Bitmap.Config.ARGB_8888) + val weights = precomputeWeights(newWidth, image.width, kernel) + val scanLine = IntArray(image.width) + for (y in 0 until image.height) { + for (x in 0 until image.width) { + scanLine[x] = image.getPixel(x, y) + } + for (x in weights.indices) { + resample(dst, x, y, weights[x], scanLine) + } + } + drain() + return dst + } + + fun vertical(image: Bitmap, newHeight: Int, kernel: ScalingKernel): Bitmap { + val dst = Bitmap.createBitmap(image.width, newHeight, Bitmap.Config.ARGB_8888) + val weights = precomputeWeights(newHeight, image.height, kernel) + val scanLine = IntArray(image.height) + for (x in 0 until image.width) { + for (y in 0 until image.height) { + scanLine[y] = image.getPixel(x, y) + } + for (y in weights.indices) { + resample(dst, x, y, weights[y], scanLine) + } + } + drain() + return dst + } +} \ No newline at end of file