changes
This commit is contained in:
Edwin Jakobs
@@ -1,128 +1,128 @@
|
||||
package org.openrndr.extra.jumpfill
|
||||
|
||||
import org.openrndr.color.ColorRGBa
|
||||
import org.openrndr.draw.*
|
||||
import org.openrndr.math.Matrix44
|
||||
import org.openrndr.math.Vector2
|
||||
import org.openrndr.resourceUrl
|
||||
|
||||
class EncodePoints : Filter(filterShaderFromUrl(resourceUrl("/shaders/gl3/encode-points.frag")))
|
||||
class JumpFlood : Filter(filterShaderFromUrl(resourceUrl("/shaders/gl3/jumpflood.frag"))) {
|
||||
var maxSteps: Int by parameters
|
||||
var step: Int by parameters
|
||||
}
|
||||
|
||||
class PixelDistance : Filter(filterShaderFromUrl(resourceUrl("/shaders/gl3/pixel-distance.frag")))
|
||||
class ContourPoints : Filter(filterShaderFromUrl(resourceUrl("/shaders/gl3/contour-points.frag")))
|
||||
class Threshold : Filter(filterShaderFromUrl(resourceUrl("/shaders/gl3/threshold.frag"))) {
|
||||
var threshold by parameters
|
||||
|
||||
init {
|
||||
threshold = 0.5
|
||||
}
|
||||
}
|
||||
|
||||
val encodePoints by lazy { EncodePoints() }
|
||||
val jumpFlood by lazy { JumpFlood() }
|
||||
val pixelDistance by lazy { PixelDistance() }
|
||||
val contourPoints by lazy { ContourPoints() }
|
||||
val threshold by lazy { Threshold() }
|
||||
|
||||
|
||||
class JumpFlooder(val width: Int, val height: Int) {
|
||||
private val dimension = Math.max(width, height)
|
||||
private val exp = Math.ceil(Math.log(dimension.toDouble()) / Math.log(2.0)).toInt()
|
||||
private val squareDim = Math.pow(2.0, exp.toDouble()).toInt()
|
||||
|
||||
private val coordinates =
|
||||
listOf(colorBuffer(squareDim, squareDim, format = ColorFormat.RGB, type = ColorType.FLOAT32),
|
||||
colorBuffer(squareDim, squareDim, format = ColorFormat.RGB, type = ColorType.FLOAT32))
|
||||
|
||||
private val final = renderTarget(width, height) {
|
||||
colorBuffer(type = ColorType.FLOAT32)
|
||||
}
|
||||
|
||||
val result: ColorBuffer get() = final.colorBuffer(0)
|
||||
|
||||
private val square = renderTarget(squareDim, squareDim) {
|
||||
colorBuffer()
|
||||
}
|
||||
|
||||
private var contourUsed = false
|
||||
private val thresholded by lazy { colorBuffer(width, height) }
|
||||
private val edges by lazy { colorBuffer(width, height) }
|
||||
|
||||
fun distanceToContour(drawer: Drawer, input: ColorBuffer, thresholdValue: Double = 0.5): ColorBuffer {
|
||||
threshold.threshold = thresholdValue
|
||||
threshold.apply(input, thresholded)
|
||||
contourPoints.apply(thresholded, edges)
|
||||
contourUsed = true
|
||||
return jumpFlood(drawer, edges)
|
||||
}
|
||||
|
||||
fun directions(xRange: IntProgression = 0 until width, yRange: IntProgression = 0 until height): Array<List<Vector2>> {
|
||||
result.shadow.download()
|
||||
return result.shadow.mapIndexed(xRange, yRange) { _, _, r, g, _, _ -> Vector2(r, g) }
|
||||
}
|
||||
|
||||
|
||||
fun jumpFlood(drawer: Drawer, input: ColorBuffer): ColorBuffer {
|
||||
if (input.width != width || input.height != height) {
|
||||
throw IllegalArgumentException("dimensions mismatch")
|
||||
}
|
||||
|
||||
drawer.isolatedWithTarget(square) {
|
||||
drawer.background(ColorRGBa.BLACK)
|
||||
drawer.ortho(square)
|
||||
drawer.view = Matrix44.IDENTITY
|
||||
drawer.model = Matrix44.IDENTITY
|
||||
drawer.image(input)
|
||||
}
|
||||
encodePoints.apply(square.colorBuffer(0), coordinates[0])
|
||||
|
||||
for (i in 0 until exp) {
|
||||
jumpFlood.step = i
|
||||
jumpFlood.apply(coordinates[i % 2], coordinates[(i + 1) % 2])
|
||||
}
|
||||
|
||||
pixelDistance.apply( arrayOf(coordinates[exp % 2], thresholded), coordinates[exp % 2])
|
||||
drawer.isolatedWithTarget(final) {
|
||||
drawer.background(ColorRGBa.BLACK)
|
||||
drawer.ortho(final)
|
||||
drawer.view = Matrix44.IDENTITY
|
||||
drawer.model = Matrix44.IDENTITY
|
||||
drawer.image(coordinates[exp % 2])
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
fun destroy(destroyFinal: Boolean = true) {
|
||||
coordinates.forEach { it.destroy() }
|
||||
|
||||
square.colorBuffer(0).destroy()
|
||||
square.detachColorBuffers()
|
||||
square.destroy()
|
||||
|
||||
if (destroyFinal) {
|
||||
final.colorBuffer(0).destroy()
|
||||
}
|
||||
final.detachColorBuffers()
|
||||
|
||||
final.destroy()
|
||||
|
||||
if (contourUsed) {
|
||||
edges.destroy()
|
||||
thresholded.destroy()
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
fun jumpFlood(drawer: Drawer, points: ColorBuffer): ColorBuffer {
|
||||
val jumpFlooder = JumpFlooder(points.width, points.height)
|
||||
jumpFlooder.jumpFlood(drawer, points)
|
||||
val result = jumpFlooder.result
|
||||
jumpFlooder.destroy(false)
|
||||
return result
|
||||
package org.openrndr.extra.jumpfill
|
||||
|
||||
import org.openrndr.color.ColorRGBa
|
||||
import org.openrndr.draw.*
|
||||
import org.openrndr.math.Matrix44
|
||||
import org.openrndr.math.Vector2
|
||||
import org.openrndr.resourceUrl
|
||||
|
||||
class EncodePoints : Filter(filterShaderFromUrl(resourceUrl("/shaders/gl3/encode-points.frag")))
|
||||
class JumpFlood : Filter(filterShaderFromUrl(resourceUrl("/shaders/gl3/jumpflood.frag"))) {
|
||||
var maxSteps: Int by parameters
|
||||
var step: Int by parameters
|
||||
}
|
||||
|
||||
class PixelDistance : Filter(filterShaderFromUrl(resourceUrl("/shaders/gl3/pixel-distance.frag")))
|
||||
class ContourPoints : Filter(filterShaderFromUrl(resourceUrl("/shaders/gl3/contour-points.frag")))
|
||||
class Threshold : Filter(filterShaderFromUrl(resourceUrl("/shaders/gl3/threshold.frag"))) {
|
||||
var threshold by parameters
|
||||
|
||||
init {
|
||||
threshold = 0.5
|
||||
}
|
||||
}
|
||||
|
||||
val encodePoints by lazy { EncodePoints() }
|
||||
val jumpFlood by lazy { JumpFlood() }
|
||||
val pixelDistance by lazy { PixelDistance() }
|
||||
val contourPoints by lazy { ContourPoints() }
|
||||
val threshold by lazy { Threshold() }
|
||||
|
||||
|
||||
class JumpFlooder(val width: Int, val height: Int) {
|
||||
private val dimension = Math.max(width, height)
|
||||
private val exp = Math.ceil(Math.log(dimension.toDouble()) / Math.log(2.0)).toInt()
|
||||
private val squareDim = Math.pow(2.0, exp.toDouble()).toInt()
|
||||
|
||||
private val coordinates =
|
||||
listOf(colorBuffer(squareDim, squareDim, format = ColorFormat.RGB, type = ColorType.FLOAT32),
|
||||
colorBuffer(squareDim, squareDim, format = ColorFormat.RGB, type = ColorType.FLOAT32))
|
||||
|
||||
private val final = renderTarget(width, height) {
|
||||
colorBuffer(type = ColorType.FLOAT32)
|
||||
}
|
||||
|
||||
val result: ColorBuffer get() = final.colorBuffer(0)
|
||||
|
||||
private val square = renderTarget(squareDim, squareDim) {
|
||||
colorBuffer()
|
||||
}
|
||||
|
||||
private var contourUsed = false
|
||||
val thresholded by lazy { colorBuffer(width, height) }
|
||||
val edges by lazy { colorBuffer(width, height) }
|
||||
|
||||
fun distanceToContour(drawer: Drawer, input: ColorBuffer, thresholdValue: Double = 0.5): ColorBuffer {
|
||||
threshold.threshold = thresholdValue
|
||||
threshold.apply(input, thresholded)
|
||||
contourPoints.apply(thresholded, edges)
|
||||
contourUsed = true
|
||||
return jumpFlood(drawer, edges)
|
||||
}
|
||||
|
||||
fun directions(xRange: IntProgression = 0 until width, yRange: IntProgression = 0 until height): Array<List<Vector2>> {
|
||||
result.shadow.download()
|
||||
return result.shadow.mapIndexed(xRange, yRange) { _, _, r, g, _, _ -> Vector2(r, g) }
|
||||
}
|
||||
|
||||
|
||||
fun jumpFlood(drawer: Drawer, input: ColorBuffer): ColorBuffer {
|
||||
if (input.width != width || input.height != height) {
|
||||
throw IllegalArgumentException("dimensions mismatch")
|
||||
}
|
||||
|
||||
drawer.isolatedWithTarget(square) {
|
||||
drawer.background(ColorRGBa.BLACK)
|
||||
drawer.ortho(square)
|
||||
drawer.view = Matrix44.IDENTITY
|
||||
drawer.model = Matrix44.IDENTITY
|
||||
drawer.image(input)
|
||||
}
|
||||
encodePoints.apply(square.colorBuffer(0), coordinates[0])
|
||||
|
||||
for (i in 0 until exp) {
|
||||
jumpFlood.step = i
|
||||
jumpFlood.apply(coordinates[i % 2], coordinates[(i + 1) % 2])
|
||||
}
|
||||
|
||||
pixelDistance.apply(arrayOf(coordinates[exp % 2], thresholded), coordinates[exp % 2])
|
||||
drawer.isolatedWithTarget(final) {
|
||||
drawer.background(ColorRGBa.BLACK)
|
||||
drawer.ortho(final)
|
||||
drawer.view = Matrix44.IDENTITY
|
||||
drawer.model = Matrix44.IDENTITY
|
||||
drawer.image(coordinates[exp % 2])
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
fun destroy(destroyFinal: Boolean = true) {
|
||||
coordinates.forEach { it.destroy() }
|
||||
|
||||
square.colorBuffer(0).destroy()
|
||||
square.detachColorBuffers()
|
||||
square.destroy()
|
||||
|
||||
if (destroyFinal) {
|
||||
final.colorBuffer(0).destroy()
|
||||
}
|
||||
final.detachColorBuffers()
|
||||
|
||||
final.destroy()
|
||||
|
||||
if (contourUsed) {
|
||||
edges.destroy()
|
||||
thresholded.destroy()
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
fun jumpFlood(drawer: Drawer, points: ColorBuffer): ColorBuffer {
|
||||
val jumpFlooder = JumpFlooder(points.width, points.height)
|
||||
jumpFlooder.jumpFlood(drawer, points)
|
||||
val result = jumpFlooder.result
|
||||
jumpFlooder.destroy(false)
|
||||
return result
|
||||
}
|
||||
Reference in New Issue
Block a user