BFFramework/BFFramework/Classes/PQGPUImage/Source/OpenGLRendering.swift

#if os(Linux)
    #if GLES
        import COpenGLES.gles2
        let GL_DEPTH24_STENCIL8 = GL_DEPTH24_STENCIL8_OES
        let GL_TRUE = GLboolean(1)
        let GL_FALSE = GLboolean(0)
    #else
        import COpenGL
    #endif
#else
    #if GLES
        import OpenGLES
    #else
        import OpenGL.GL3
    #endif
#endif

import Foundation

public enum InputTextureStorageFormat {
    case textureCoordinates([GLfloat])
    case textureVBO(GLuint)
}

public struct InputTextureProperties {
    public let textureStorage: InputTextureStorageFormat
    public let texture: GLuint

    public init(textureCoordinates: [GLfloat]? = nil, textureVBO: GLuint? = nil, texture: GLuint) {
        self.texture = texture
        switch (textureCoordinates, textureVBO) {
        case let (.some(coordinates), .none): self.textureStorage = .textureCoordinates(coordinates)
        case let (.none, .some(vbo)): textureStorage = .textureVBO(vbo)
        case (.none, .none): fatalError("Need to specify either texture coordinates or a VBO to InputTextureProperties")
        case (.some, .some): fatalError("Can't specify both texture coordinates and a VBO to InputTextureProperties")
        }
    }
}

public struct GLSize {
    public let width: GLint
    public let height: GLint

    public init(width: GLint, height: GLint) {
        self.width = width
        self.height = height
    }

    public init(_ size: Size) {
        width = size.glWidth()
        height = size.glHeight()
    }
}

extension Size {
    init(_ size: GLSize) {
        width = Float(size.width)
        height = Float(size.height)
    }
}

public let standardImageVertices: [GLfloat] = [-1.0, -1.0, 1.0, -1.0, -1.0, 1.0, 1.0, 1.0]
public let verticallyInvertedImageVertices: [GLfloat] = [-1.0, 1.0, 1.0, 1.0, -1.0, -1.0, 1.0, -1.0]

// "position" and "inputTextureCoordinate", "inputTextureCoordinate2" attribute naming follows the convention of the old GPUImage
public func renderQuadWithShader(_ shader: ShaderProgram, uniformSettings: ShaderUniformSettings? = nil, vertices: [GLfloat]? = nil, vertexBufferObject: GLuint? = nil, inputTextures: [InputTextureProperties], context: OpenGLContext = sharedImageProcessingContext) {
    switch (vertices, vertexBufferObject) {
    case (.none, .some): break
    case (.some, .none): break
    case (.some, .some): fatalError("Can't specify both vertices and a VBO in renderQuadWithShader()")
    case (.none, .none): fatalError("Can't specify both vertices and a VBO in renderQuadWithShader()")
    }

    context.makeCurrentContext()
    shader.use()
    uniformSettings?.restoreShaderSettings(shader)

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

    if let boundVBO = vertexBufferObject {
        glBindBuffer(GLenum(GL_ARRAY_BUFFER), boundVBO)
        glVertexAttribPointer(positionAttribute, 2, GLenum(GL_FLOAT), 0, 0, nil)
        glBindBuffer(GLenum(GL_ARRAY_BUFFER), 0)
    } else {
        glVertexAttribPointer(positionAttribute, 2, GLenum(GL_FLOAT), 0, 0, vertices!)
    }

    for (index, inputTexture) in inputTextures.enumerated() {
        if let textureCoordinateAttribute = shader.attributeIndex("inputTextureCoordinate".withNonZeroSuffix(index)) {
            switch inputTexture.textureStorage {
            case let .textureCoordinates(textureCoordinates):
                glVertexAttribPointer(textureCoordinateAttribute, 2, GLenum(GL_FLOAT), 0, 0, textureCoordinates)
            case let .textureVBO(textureVBO):
                glBindBuffer(GLenum(GL_ARRAY_BUFFER), textureVBO)
                glVertexAttribPointer(textureCoordinateAttribute, 2, GLenum(GL_FLOAT), 0, 0, nil)
                glBindBuffer(GLenum(GL_ARRAY_BUFFER), 0)
            }
        } else if index == 0 {
            fatalError("The required attribute named inputTextureCoordinate was missing from the shader program during rendering.")
        }

        glActiveTexture(textureUnitForIndex(index))
        glBindTexture(GLenum(GL_TEXTURE_2D), inputTexture.texture)

        shader.setValue(GLint(index), forUniform: "inputImageTexture".withNonZeroSuffix(index))
    }

    glDrawArrays(GLenum(GL_TRIANGLE_STRIP), 0, 4)

    if vertexBufferObject != nil {
        glBindBuffer(GLenum(GL_ARRAY_BUFFER), 0)
    }

    for (index, _) in inputTextures.enumerated() {
        glActiveTexture(textureUnitForIndex(index))
        glBindTexture(GLenum(GL_TEXTURE_2D), 0)
    }
}

public func clearFramebufferWithColor(_ color: Color) {
    glClearColor(GLfloat(color.redComponent), GLfloat(color.greenComponent), GLfloat(color.blueComponent), GLfloat(color.alphaComponent))
    glClear(GLenum(GL_COLOR_BUFFER_BIT))
}

func renderStencilMaskFromFramebuffer(_ framebuffer: Framebuffer) {
    let inputTextureProperties = framebuffer.texturePropertiesForOutputRotation(.noRotation)
    glEnable(GLenum(GL_STENCIL_TEST))
    glClearStencil(0)
    glClear(GLenum(GL_DEPTH_BUFFER_BIT | GL_STENCIL_BUFFER_BIT))
    glColorMask(GLboolean(GL_FALSE), GLboolean(GL_FALSE), GLboolean(GL_FALSE), GLboolean(GL_FALSE))
    glStencilFunc(GLenum(GL_ALWAYS), 1, 1)
    glStencilOp(GLenum(GL_KEEP), GLenum(GL_KEEP), GLenum(GL_REPLACE))

    #if GL
        glEnable(GLenum(GL_ALPHA_TEST))
        glAlphaFunc(GLenum(GL_NOTEQUAL), 0.0)
        renderQuadWithShader(sharedImageProcessingContext.passthroughShader, vertices: standardImageVertices, inputTextures: [inputTextureProperties])
    #else
        let alphaTestShader = crashOnShaderCompileFailure("Stencil") { try sharedImageProcessingContext.programForVertexShader(OneInputVertexShader, fragmentShader: AlphaTestFragmentShader) }
        renderQuadWithShader(alphaTestShader, vertices: standardImageVertices, inputTextures: [inputTextureProperties])
    #endif

    glColorMask(GLboolean(GL_TRUE), GLboolean(GL_TRUE), GLboolean(GL_TRUE), GLboolean(GL_TRUE))

    glStencilFunc(GLenum(GL_EQUAL), 1, 1)
    glStencilOp(GLenum(GL_KEEP), GLenum(GL_KEEP), GLenum(GL_KEEP))

    #if GL
        glDisable(GLenum(GL_ALPHA_TEST))
    #endif
}

func disableStencil() {
    glDisable(GLenum(GL_STENCIL_TEST))
}

func textureUnitForIndex(_ index: Int) -> GLenum {
    switch index {
    case 0: return GLenum(GL_TEXTURE0)
    case 1: return GLenum(GL_TEXTURE1)
    case 2: return GLenum(GL_TEXTURE2)
    case 3: return GLenum(GL_TEXTURE3)
    case 4: return GLenum(GL_TEXTURE4)
    case 5: return GLenum(GL_TEXTURE5)
    case 6: return GLenum(GL_TEXTURE6)
    case 7: return GLenum(GL_TEXTURE7)
    case 8: return GLenum(GL_TEXTURE8)
    default: fatalError("Attempted to address too high a texture unit")
    }
}

public func generateTexture(minFilter: Int32, magFilter: Int32, wrapS: Int32, wrapT: Int32) -> GLuint {
    var texture: GLuint = 0

    glActiveTexture(GLenum(GL_TEXTURE1))
    glGenTextures(1, &texture)
    glBindTexture(GLenum(GL_TEXTURE_2D), texture)
    glTexParameteri(GLenum(GL_TEXTURE_2D), GLenum(GL_TEXTURE_MIN_FILTER), minFilter)
    glTexParameteri(GLenum(GL_TEXTURE_2D), GLenum(GL_TEXTURE_MAG_FILTER), magFilter)
    glTexParameteri(GLenum(GL_TEXTURE_2D), GLenum(GL_TEXTURE_WRAP_S), wrapS)
    glTexParameteri(GLenum(GL_TEXTURE_2D), GLenum(GL_TEXTURE_WRAP_T), wrapT)

    glBindTexture(GLenum(GL_TEXTURE_2D), 0)

    return texture
}

public func uploadLocalArray(data: [GLfloat], into texture: GLuint, size: GLSize) {
    glActiveTexture(GLenum(GL_TEXTURE1))
    glBindTexture(GLenum(GL_TEXTURE_2D), texture)
    glTexImage2D(GLenum(GL_TEXTURE_2D), 0, GL_RGBA, size.width, size.height, 0, GLenum(GL_RGBA), GLenum(GL_FLOAT), data)
    glBindTexture(GLenum(GL_TEXTURE_2D), 0)
}

func generateFramebufferForTexture(_ texture: GLuint, width: GLint, height: GLint, internalFormat: Int32, format: Int32, type: Int32, stencil: Bool) throws -> (GLuint, GLuint?) {
    var framebuffer: GLuint = 0
    glActiveTexture(GLenum(GL_TEXTURE1))

    glGenFramebuffers(1, &framebuffer)
    glBindFramebuffer(GLenum(GL_FRAMEBUFFER), framebuffer)
    glBindTexture(GLenum(GL_TEXTURE_2D), texture)

    glTexImage2D(GLenum(GL_TEXTURE_2D), 0, internalFormat, width, height, 0, GLenum(format), GLenum(type), nil)
    glFramebufferTexture2D(GLenum(GL_FRAMEBUFFER), GLenum(GL_COLOR_ATTACHMENT0), GLenum(GL_TEXTURE_2D), texture, 0)

    let status = glCheckFramebufferStatus(GLenum(GL_FRAMEBUFFER))
    if status != GLenum(GL_FRAMEBUFFER_COMPLETE) {
        throw FramebufferCreationError(errorCode: status)
    }

    let stencilBuffer: GLuint?
    if stencil {
        stencilBuffer = try attachStencilBuffer(width: width, height: height)
    } else {
        stencilBuffer = nil
    }

    glBindTexture(GLenum(GL_TEXTURE_2D), 0)
    glBindFramebuffer(GLenum(GL_FRAMEBUFFER), 0)
    return (framebuffer, stencilBuffer)
}

func attachStencilBuffer(width: GLint, height: GLint) throws -> GLuint {
    var stencilBuffer: GLuint = 0
    glGenRenderbuffers(1, &stencilBuffer)
    glBindRenderbuffer(GLenum(GL_RENDERBUFFER), stencilBuffer)
    glRenderbufferStorage(GLenum(GL_RENDERBUFFER), GLenum(GL_DEPTH24_STENCIL8), width, height) // iOS seems to only support combination depth + stencil, from references
    #if os(iOS)
        glFramebufferRenderbuffer(GLenum(GL_FRAMEBUFFER), GLenum(GL_DEPTH_ATTACHMENT), GLenum(GL_RENDERBUFFER), stencilBuffer)
    #endif
    glFramebufferRenderbuffer(GLenum(GL_FRAMEBUFFER), GLenum(GL_STENCIL_ATTACHMENT), GLenum(GL_RENDERBUFFER), stencilBuffer)

    glBindRenderbuffer(GLenum(GL_RENDERBUFFER), 0)

    let status = glCheckFramebufferStatus(GLenum(GL_FRAMEBUFFER))
    if status != GLenum(GL_FRAMEBUFFER_COMPLETE) {
        throw FramebufferCreationError(errorCode: status)
    }

    return stencilBuffer
}

public func enableAdditiveBlending() {
    glBlendEquation(GLenum(GL_FUNC_ADD))
    glBlendFunc(GLenum(GL_ONE), GLenum(GL_ONE))
    glEnable(GLenum(GL_BLEND))
}

public func disableBlending() {
    glDisable(GLenum(GL_BLEND))
}

public func generateVBO(for vertices: [GLfloat]) -> GLuint {
    var newBuffer: GLuint = 0
    glGenBuffers(1, &newBuffer)
    glBindBuffer(GLenum(GL_ARRAY_BUFFER), newBuffer)
    glBufferData(GLenum(GL_ARRAY_BUFFER), MemoryLayout<GLfloat>.size * vertices.count, vertices, GLenum(GL_STATIC_DRAW))
    glBindBuffer(GLenum(GL_ARRAY_BUFFER), 0)
    return newBuffer
}

public func deleteVBO(_ vbo: GLuint) {
    var deletedVBO = vbo
    glDeleteBuffers(1, &deletedVBO)
}

extension String {
    func withNonZeroSuffix(_ suffix: Int) -> String {
        if suffix == 0 {
            return self
        } else {
            return "\(self)\(suffix + 1)"
        }
    }

    func withGLChar(_ operation: (UnsafePointer<GLchar>) -> Void) {
        withCString { pointer in
            operation(UnsafePointer<GLchar>(pointer))
        }
    }
}