swift - GPUImage2使用SolidColorGenerator作为RectangleGenerator

类似于simpleVideoFilter和面部检测，我想用彩色矩形突出显示我的GPUImage2视频提要的一部分。 SolidColorGenerator似乎是一个不错的开始，但有两个例外。首先，我不确定如何将SolidColorGenerator输出转置到原始图像上的正确位置。其次，我不确定如何在原始图像中添加N个矩形。同时，我在LineGenerator上取得了成功，但我认为矩形看起来会更好。

奖金：
具有圆角和单个像素边框的矩形。

最佳答案

对我有用的是编写RectangleGenerator。新RectangleGenerator的关键是绘制GL_TRIANGLES。

glDrawArrays(GLenum(GL_TRIANGLES), 0, GLsizei(positions.count) * 2)

可以进行的一种优化是使用GL_Elements和GL_TRIANGLE_STRIP，尽管我无法弄清楚。



完整的解决方案应支持border，borderWidth和cornerRadius。现在，简单的矩形就足够了。完整代码如下。

let vertex = "attribute vec4 position; void main(){ gl_Position = position; }"

let fragment = "uniform lowp vec3 fillColor; void main(){ gl_FragColor = vec4(fillColor, 1.0); }"

public class RectangleGenerator: ImageGenerator {

let rectangleShader:ShaderProgram
var uniformSettings = ShaderUniformSettings()

public var fillColor:Color = Color.green { didSet { uniformSettings["fillColor"] = fillColor } }

public override init(size:Size) {
    rectangleShader = crashOnShaderCompileFailure("RectangleGenerator"){try sharedImageProcessingContext.programForVertexShader(vertex, fragmentShader:fragment)}
    super.init(size:size)

    ({fillColor = Color.green})()
}

public func renderRectangles(_ positions:[Position]) {
    guard positions.count > 3 else { return }

    imageFramebuffer.activateFramebufferForRendering()

    rectangleShader.use()
    uniformSettings.restoreShaderSettings(rectangleShader)
    clearFramebufferWithColor(Color.transparent)

    guard let positionAttribute = rectangleShader.attributeIndex("position") else { fatalError("A position attribute was missing from the shader program during rendering.") }

    let convertedPositions = positions.flatMap{$0.toGLArray()}
    glVertexAttribPointer(positionAttribute, 2, GLenum(GL_FLOAT), 0, 0, convertedPositions)

    glBlendEquation(GLenum(GL_FUNC_ADD))
    glBlendFunc(GLenum(GL_ONE), GLenum(GL_ONE))
    glEnable(GLenum(GL_BLEND))

    glDrawArrays(GLenum(GL_TRIANGLES), 0, GLsizei(positions.count) * 2)
    glDisable(GLenum(GL_BLEND))

    notifyTargets()
}
}

快乐编码