[orx-compositor] convert to MPP
This commit is contained in:
Edwin Jakobs
272
orx-compositor/src/commonMain/kotlin/Compositor.kt
Normal file
272
orx-compositor/src/commonMain/kotlin/Compositor.kt
Normal file
@@ -0,0 +1,272 @@
|
||||
package org.openrndr.extra.compositor
|
||||
|
||||
import org.openrndr.Extension
|
||||
import org.openrndr.Program
|
||||
import org.openrndr.color.ColorRGBa
|
||||
import org.openrndr.draw.*
|
||||
import org.openrndr.extra.fx.blend.SourceIn
|
||||
import org.openrndr.extra.fx.blend.SourceOut
|
||||
import org.openrndr.extra.parameters.BooleanParameter
|
||||
import org.openrndr.extra.parameters.Description
|
||||
import org.openrndr.math.Matrix44
|
||||
|
||||
private val postBufferCache = mutableListOf<ColorBuffer>()
|
||||
|
||||
fun RenderTarget.deepDestroy() {
|
||||
val cbcopy = colorAttachments.map { it }
|
||||
val dbcopy = depthBuffer
|
||||
detachDepthBuffer()
|
||||
detachColorAttachments()
|
||||
cbcopy.forEach {
|
||||
when (it) {
|
||||
is ColorBufferAttachment -> it.colorBuffer.destroy()
|
||||
is CubemapAttachment -> it.cubemap.destroy()
|
||||
is ArrayTextureAttachment -> it.arrayTexture.destroy()
|
||||
is ArrayCubemapAttachment -> it.arrayCubemap.destroy()
|
||||
}
|
||||
}
|
||||
dbcopy?.destroy()
|
||||
destroy()
|
||||
}
|
||||
|
||||
/**
|
||||
* A single layer representation
|
||||
*/
|
||||
@Description("Layer")
|
||||
open class Layer internal constructor() {
|
||||
var copyLayers: List<Layer> = listOf()
|
||||
var sourceOut = SourceOut()
|
||||
var sourceIn = SourceIn()
|
||||
var maskLayer: Layer? = null
|
||||
var drawFunc: () -> Unit = {}
|
||||
val children: MutableList<Layer> = mutableListOf()
|
||||
var blendFilter: Pair<Filter, Filter.() -> Unit>? = null
|
||||
val postFilters: MutableList<Pair<Filter, Filter.() -> Unit>> = mutableListOf()
|
||||
var colorType = ColorType.UINT8
|
||||
var accumulation: ColorBuffer? = null
|
||||
|
||||
@BooleanParameter("enabled")
|
||||
var enabled = true
|
||||
|
||||
@BooleanParameter("Invert mask")
|
||||
var invertMask = false
|
||||
var clearColor: ColorRGBa? = ColorRGBa.TRANSPARENT
|
||||
private var layerTarget: RenderTarget? = null
|
||||
|
||||
val result: ColorBuffer?
|
||||
get() {
|
||||
return layerTarget?.colorBuffer(0)
|
||||
}
|
||||
|
||||
/**
|
||||
* draw the layer
|
||||
*/
|
||||
fun drawLayer(drawer: Drawer) {
|
||||
if (!enabled) {
|
||||
return
|
||||
}
|
||||
|
||||
val activeRenderTarget = RenderTarget.active
|
||||
|
||||
accumulation = if (activeRenderTarget !is ProgramRenderTarget) {
|
||||
activeRenderTarget.colorBuffer(0)
|
||||
} else {
|
||||
null
|
||||
}
|
||||
|
||||
if (shouldCreateLayerTarget(activeRenderTarget)) {
|
||||
createLayerTarget(activeRenderTarget, drawer)
|
||||
}
|
||||
|
||||
layerTarget?.let { target ->
|
||||
if (copyLayers.isNotEmpty()) {
|
||||
copyLayers.forEach {
|
||||
drawer.isolatedWithTarget(target) {
|
||||
clearColor?.let {
|
||||
drawer.clear(it)
|
||||
}
|
||||
|
||||
it.layerTarget?.let { copyTarget ->
|
||||
drawer.image(copyTarget.colorBuffer(0))
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
maskLayer?.let {
|
||||
if (it.shouldCreateLayerTarget(activeRenderTarget)) {
|
||||
it.createLayerTarget(activeRenderTarget, drawer)
|
||||
}
|
||||
|
||||
it.layerTarget?.let { maskRt ->
|
||||
drawer.isolatedWithTarget(maskRt) {
|
||||
if (copyLayers.isEmpty()) {
|
||||
clearColor?.let { color ->
|
||||
drawer.clear(color)
|
||||
}
|
||||
}
|
||||
drawer.fill = ColorRGBa.WHITE
|
||||
drawer.stroke = ColorRGBa.WHITE
|
||||
it.drawFunc()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
drawer.isolatedWithTarget(target) {
|
||||
if (copyLayers.isEmpty()) {
|
||||
clearColor?.let {
|
||||
drawer.clear(it)
|
||||
}
|
||||
}
|
||||
drawFunc()
|
||||
children.forEach {
|
||||
it.drawLayer(drawer)
|
||||
}
|
||||
}
|
||||
|
||||
if (postFilters.size > 0) {
|
||||
val sizeMismatch = if (postBufferCache.isNotEmpty()) {
|
||||
postBufferCache[0].width != activeRenderTarget.width || postBufferCache[0].height != activeRenderTarget.height
|
||||
} else {
|
||||
false
|
||||
}
|
||||
|
||||
if (sizeMismatch) {
|
||||
postBufferCache.forEach { it.destroy() }
|
||||
postBufferCache.clear()
|
||||
}
|
||||
|
||||
if (postBufferCache.isEmpty()) {
|
||||
postBufferCache += persistent { colorBuffer(activeRenderTarget.width, activeRenderTarget.height, type = colorType) }
|
||||
postBufferCache += persistent { colorBuffer(activeRenderTarget.width, activeRenderTarget.height, type = colorType) }
|
||||
}
|
||||
}
|
||||
|
||||
val layerPost = postFilters.let { filters ->
|
||||
val targets = postBufferCache
|
||||
val result = filters.foldIndexed(target.colorBuffer(0)) { i, source, filter ->
|
||||
val localTarget = targets[i % targets.size]
|
||||
filter.first.apply(filter.second)
|
||||
filter.first.apply(source, localTarget)
|
||||
localTarget
|
||||
}
|
||||
result
|
||||
}
|
||||
|
||||
maskLayer?.let {
|
||||
val maskFilter = if (invertMask) sourceOut else sourceIn
|
||||
|
||||
maskFilter.apply(arrayOf(layerPost, it.layerTarget!!.colorBuffer(0)), layerPost)
|
||||
}
|
||||
|
||||
val localBlendFilter = blendFilter
|
||||
if (localBlendFilter == null) {
|
||||
drawer.isolatedWithTarget(activeRenderTarget) {
|
||||
drawer.ortho()
|
||||
drawer.view = Matrix44.IDENTITY
|
||||
drawer.model = Matrix44.IDENTITY
|
||||
drawer.image(layerPost, layerPost.bounds, drawer.bounds)
|
||||
}
|
||||
} else {
|
||||
localBlendFilter.first.apply(localBlendFilter.second)
|
||||
localBlendFilter.first.apply(arrayOf(activeRenderTarget.colorBuffer(0), layerPost), activeRenderTarget.colorBuffer(0))
|
||||
}
|
||||
|
||||
accumulation?.copyTo(target.colorBuffer(0))
|
||||
}
|
||||
}
|
||||
|
||||
private fun shouldCreateLayerTarget(activeRenderTarget: RenderTarget): Boolean {
|
||||
return layerTarget == null || (layerTarget?.width != activeRenderTarget.width || layerTarget?.height != activeRenderTarget.height)
|
||||
}
|
||||
|
||||
private fun createLayerTarget(activeRenderTarget: RenderTarget, drawer: Drawer) {
|
||||
layerTarget?.deepDestroy()
|
||||
layerTarget = renderTarget(activeRenderTarget.width, activeRenderTarget.height) {
|
||||
colorBuffer(type = colorType)
|
||||
depthBuffer()
|
||||
}
|
||||
layerTarget?.let {
|
||||
drawer.withTarget(it) {
|
||||
drawer.clear(ColorRGBa.TRANSPARENT)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* create a layer within the composition
|
||||
*/
|
||||
fun Layer.layer(function: Layer.() -> Unit): Layer {
|
||||
val layer = Layer().apply { function() }
|
||||
children.add(layer)
|
||||
return layer
|
||||
}
|
||||
|
||||
/**
|
||||
* set the draw contents of the layer
|
||||
*/
|
||||
fun Layer.draw(function: () -> Unit) {
|
||||
drawFunc = function
|
||||
}
|
||||
|
||||
/**
|
||||
* use the layer as a base
|
||||
*/
|
||||
fun Layer.use(vararg layer: Layer) {
|
||||
copyLayers = layer.toList()
|
||||
}
|
||||
|
||||
/**
|
||||
* the drawing acts as a mask on the layer
|
||||
*/
|
||||
fun Layer.mask(function: () -> Unit) {
|
||||
maskLayer = Layer().apply {
|
||||
this.drawFunc = function
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* add a post-processing filter to the layer
|
||||
*/
|
||||
fun <F : Filter> Layer.post(filter: F, configure: F.() -> Unit = {}): F {
|
||||
@Suppress("UNCHECKED_CAST")
|
||||
postFilters.add(Pair(filter as Filter, configure as Filter.() -> Unit))
|
||||
return filter
|
||||
}
|
||||
|
||||
/**
|
||||
* add a blend filter to the layer
|
||||
*/
|
||||
fun <F : Filter> Layer.blend(filter: F, configure: F.() -> Unit = {}): F {
|
||||
@Suppress("UNCHECKED_CAST")
|
||||
blendFilter = Pair(filter as Filter, configure as Filter.() -> Unit)
|
||||
return filter
|
||||
}
|
||||
|
||||
class Composite : Layer() {
|
||||
fun draw(drawer: Drawer) {
|
||||
drawLayer(drawer)
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* create a layered composition
|
||||
*/
|
||||
fun compose(function: Layer.() -> Unit): Composite {
|
||||
val root = Composite()
|
||||
root.function()
|
||||
return root
|
||||
}
|
||||
|
||||
class Compositor : Extension {
|
||||
override var enabled: Boolean = true
|
||||
var composite = Composite()
|
||||
|
||||
override fun afterDraw(drawer: Drawer, program: Program) {
|
||||
drawer.isolated {
|
||||
drawer.defaults()
|
||||
composite.drawLayer(drawer)
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user