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()
    let path = Bundle.main.path(forResource: "Shaders", ofType: "metal")
    do {
      let input = try String(contentsOfFile: path!, encoding: String.Encoding.utf8)
      let library = try device!.makeLibrary(source: input, options: nil)
      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()
  }
}