57 lines
1.8 KiB
Swift
Executable File
57 lines
1.8 KiB
Swift
Executable File
|
|
import MetalKit
|
|
|
|
public class Renderer: NSObject, MTKViewDelegate {
|
|
|
|
public var device: MTLDevice!
|
|
var queue: MTLCommandQueue!
|
|
var pipelineState: MTLComputePipelineState!
|
|
var timerBuffer: MTLBuffer!
|
|
var timer: Float = 0
|
|
|
|
override public init() {
|
|
super.init()
|
|
initializeMetal()
|
|
}
|
|
|
|
func initializeMetal() {
|
|
device = MTLCreateSystemDefaultDevice()
|
|
queue = device!.makeCommandQueue()
|
|
do {
|
|
let library = device.makeDefaultLibrary()!
|
|
let kernel = library.makeFunction(name: "compute")!
|
|
pipelineState = try device!.makeComputePipelineState(function: kernel)
|
|
} catch {
|
|
print(error)
|
|
}
|
|
timerBuffer = device!.makeBuffer(length: MemoryLayout<Float>.size, options: [])
|
|
}
|
|
|
|
func update() {
|
|
timer += 0.01
|
|
let bufferPointer = timerBuffer.contents()
|
|
memcpy(bufferPointer, &timer, MemoryLayout<Float>.size)
|
|
}
|
|
|
|
public func mtkView(_ view: MTKView, drawableSizeWillChange size: CGSize) {}
|
|
|
|
public func draw(in view: MTKView) {
|
|
update()
|
|
guard let commandBuffer = queue.makeCommandBuffer(),
|
|
let commandEncoder = commandBuffer.makeComputeCommandEncoder(),
|
|
let drawable = view.currentDrawable else { fatalError() }
|
|
commandEncoder.setComputePipelineState(pipelineState)
|
|
commandEncoder.setTexture(drawable.texture, index: 0)
|
|
commandEncoder.setBuffer(timerBuffer, offset: 0, index: 0)
|
|
let w = pipelineState.threadExecutionWidth
|
|
let h = pipelineState.maxTotalThreadsPerThreadgroup / w
|
|
let threadsPerGroup = MTLSizeMake(w, h, 1)
|
|
let threadsPerGrid = MTLSizeMake(Int(view.drawableSize.width),
|
|
Int(view.drawableSize.height), 1)
|
|
commandEncoder.dispatchThreads(threadsPerGrid, threadsPerThreadgroup: threadsPerGroup)
|
|
commandEncoder.endEncoding()
|
|
commandBuffer.present(drawable)
|
|
commandBuffer.commit()
|
|
}
|
|
}
|