Class SlitScanner

src/main/java/trace/util/SlitScanner.kt:15–176

import java.util.concurrent.TimeUnit
import java.util.stream.Collectors

class SlitScanner(val source: String) {


    var data: MutableList<ByteBuffer>
    var dim: IntArray = intArrayOf(0, 0)

    init {
        var files = Files.list(File(source).toPath()).filter { it.toString().endsWith(".jpg") && !it.fileName.startsWith(".") }.sorted().collect(
            Collectors.toList())

        while (files.size > 500)
            files.toList().filterIndexed { i, b -> i % 2 == 0 }.toList()
        data = files.map {
            dim = FastJPEG.j.dimensions(it.toString())
            val dat = ByteBuffer.allocateDirect(3 * dim[0] * dim[1]).order(ByteOrder.nativeOrder())
            FastJPEG.j.decompress(it.toString(), dat, dim[0], dim[1])

            dat
        }.toMutableList()
    }

    fun access(x: Int, y: Int, n: Int, out: FloatArray, offset: Int) {
        val o = 3 * (y * dim[0] + x)
        out[0 + offset] = (data[n].get(o + 0).toInt() and 255) / 255f
        out[1 + offset] = (data[n].get(o + 1).toInt() and 255) / 255f
        out[2 + offset] = (data[n].get(o + 2).toInt() and 255) / 255f
    }

    fun buildImage(fn: String, w: Int, h: Int, func: (Float, Float, Float, Vec3) -> Unit) {

        val o = ByteBuffer.allocateDirect(3 * w * h).order(ByteOrder.nativeOrder())

        val b = object : ThreadLocal<FloatArray>() {
            override fun initialValue(): FloatArray {
                return FloatArray(24)
            }
        }

        val pool = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors() + 2)

        for (y in 0 until h) {
            pool.submit {
                print("$y / $h ")
                val oo = b.get()
                val a = Vec3()
                try {
                    for (x in 0 until w) {

                        func(x / w.toFloat(), y / h.toFloat(), 0.5f, a)
                        accessI(a.x.toFloat(), a.y.toFloat(), a.z.toFloat(), oo)

                        o.put(3 * y * w + 3 * x + 0, (Math.max(0f, Math.min(1f, oo[0])) * 255).toInt().toByte())
                        o.put(3 * y * w + 3 * x + 1, (Math.max(0f, Math.min(1f, oo[1])) * 255).toInt().toByte())
                        o.put(3 * y * w + 3 * x + 2, (Math.max(0f, Math.min(1f, oo[2])) * 255).toInt().toByte())

                    }
                } catch (e: Exception) {