added chapter 13
This commit is contained in:
parent
ea49c7202c
commit
5885a67b73
|
|
@ -14,3 +14,4 @@ Repository to accompany the following blog posts:
|
||||||
- [Using MetalKit part 10](http://mhorga.org/2016/05/02/using-metalkit-part-10.html)
|
- [Using MetalKit part 10](http://mhorga.org/2016/05/02/using-metalkit-part-10.html)
|
||||||
- [Using MetalKit part 11](http://mhorga.org/2016/05/10/using-metalkit-part-11.html)
|
- [Using MetalKit part 11](http://mhorga.org/2016/05/10/using-metalkit-part-11.html)
|
||||||
- [Using MetalKit part 12](http://mhorga.org/2016/05/18/using-metalkit-part-12.html)
|
- [Using MetalKit part 12](http://mhorga.org/2016/05/18/using-metalkit-part-12.html)
|
||||||
|
- [Using MetalKit part 13](http://mhorga.org/2016/05/25/using-metalkit-part-13.html)
|
||||||
|
|
@ -0,0 +1,8 @@
|
||||||
|
|
||||||
|
import Cocoa
|
||||||
|
import XCPlayground
|
||||||
|
|
||||||
|
let device = MTLCreateSystemDefaultDevice()!
|
||||||
|
let frame = NSRect(x:0, y:0, width:400, height:400)
|
||||||
|
let view = MetalView(frame: frame, device: device)
|
||||||
|
XCPlaygroundPage.currentPage.liveView = view
|
||||||
|
|
@ -0,0 +1,30 @@
|
||||||
|
|
||||||
|
#include <metal_stdlib>
|
||||||
|
|
||||||
|
using namespace metal;
|
||||||
|
|
||||||
|
kernel void compute(texture2d<float, access::write> output [[texture(0)]],
|
||||||
|
constant float &timer [[buffer(1)]],
|
||||||
|
constant float2 &mouse [[buffer(2)]],
|
||||||
|
uint2 gid [[thread_position_in_grid]])
|
||||||
|
{
|
||||||
|
int width = output.get_width();
|
||||||
|
int height = output.get_height();
|
||||||
|
float2 uv = float2(gid) / float2(width, height);
|
||||||
|
uv = uv * 2.0 - 1.0;
|
||||||
|
float radius = 0.5;
|
||||||
|
float distance = length(uv) - radius;
|
||||||
|
// output.write(distance < 0 ? float4(1) : float4(0), gid);
|
||||||
|
|
||||||
|
float planet = float(sqrt(radius * radius - uv.x * uv.x - uv.y * uv.y));
|
||||||
|
// planet /= radius;
|
||||||
|
// output.write(distance < 0 ? float4(planet) : float4(0), gid);
|
||||||
|
|
||||||
|
float3 normal = normalize(float3(uv.x, uv.y, planet));
|
||||||
|
// output.write(distance < 0 ? float4(float3(normal), 1) : float4(0), gid);
|
||||||
|
|
||||||
|
float3 source = normalize(float3(cos(timer), sin(timer), 1));
|
||||||
|
// float3 source = normalize(float3(-1, 0, 1));
|
||||||
|
float light = dot(normal, source);
|
||||||
|
output.write(distance < 0 ? float4(float3(light), 1) : float4(0), gid);
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,71 @@
|
||||||
|
|
||||||
|
import MetalKit
|
||||||
|
|
||||||
|
public class MetalView: MTKView, NSWindowDelegate {
|
||||||
|
|
||||||
|
var queue: MTLCommandQueue! = nil
|
||||||
|
var cps: MTLComputePipelineState! = nil
|
||||||
|
var timer: Float = 0
|
||||||
|
var timerBuffer: MTLBuffer!
|
||||||
|
var mouseBuffer: MTLBuffer!
|
||||||
|
var pos: NSPoint!
|
||||||
|
|
||||||
|
override public func mouseDown(event: NSEvent) {
|
||||||
|
pos = convertPointToLayer(convertPoint(event.locationInWindow, fromView: nil))
|
||||||
|
let scale = layer!.contentsScale
|
||||||
|
pos.x *= scale
|
||||||
|
pos.y *= scale
|
||||||
|
}
|
||||||
|
|
||||||
|
required public init(coder: NSCoder) {
|
||||||
|
super.init(coder: coder)
|
||||||
|
}
|
||||||
|
|
||||||
|
override public init(frame frameRect: CGRect, device: MTLDevice?) {
|
||||||
|
super.init(frame: frameRect, device: device)
|
||||||
|
registerShaders()
|
||||||
|
}
|
||||||
|
|
||||||
|
override public func drawRect(dirtyRect: NSRect) {
|
||||||
|
super.drawRect(dirtyRect)
|
||||||
|
if let drawable = currentDrawable {
|
||||||
|
let commandBuffer = queue.commandBuffer()
|
||||||
|
let commandEncoder = commandBuffer.computeCommandEncoder()
|
||||||
|
commandEncoder.setComputePipelineState(cps)
|
||||||
|
commandEncoder.setTexture(drawable.texture, atIndex: 0)
|
||||||
|
commandEncoder.setBuffer(mouseBuffer, offset: 0, atIndex: 2)
|
||||||
|
commandEncoder.setBuffer(timerBuffer, offset: 0, atIndex: 1)
|
||||||
|
update()
|
||||||
|
let threadGroupCount = MTLSizeMake(8, 8, 1)
|
||||||
|
let threadGroups = MTLSizeMake(drawable.texture.width / threadGroupCount.width, drawable.texture.height / threadGroupCount.height, 1)
|
||||||
|
commandEncoder.dispatchThreadgroups(threadGroups, threadsPerThreadgroup: threadGroupCount)
|
||||||
|
commandEncoder.endEncoding()
|
||||||
|
commandBuffer.presentDrawable(drawable)
|
||||||
|
commandBuffer.commit()
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func update() {
|
||||||
|
timer += 0.01
|
||||||
|
var bufferPointer = timerBuffer.contents()
|
||||||
|
memcpy(bufferPointer, &timer, sizeof(Float))
|
||||||
|
bufferPointer = mouseBuffer.contents()
|
||||||
|
memcpy(bufferPointer, &pos, sizeof(NSPoint))
|
||||||
|
}
|
||||||
|
|
||||||
|
func registerShaders() {
|
||||||
|
queue = device!.newCommandQueue()
|
||||||
|
let path = NSBundle.mainBundle().pathForResource("Shaders", ofType: "metal")
|
||||||
|
do {
|
||||||
|
let input = try String(contentsOfFile: path!, encoding: NSUTF8StringEncoding)
|
||||||
|
let library = try device!.newLibraryWithSource(input, options: nil)
|
||||||
|
let kernel = library.newFunctionWithName("compute")!
|
||||||
|
cps = try device!.newComputePipelineStateWithFunction(kernel)
|
||||||
|
} catch let e {
|
||||||
|
Swift.print("\(e)")
|
||||||
|
}
|
||||||
|
timerBuffer = device!.newBufferWithLength(sizeof(Float), options: [])
|
||||||
|
mouseBuffer = device!.newBufferWithLength(sizeof(NSPoint), options: [])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,4 @@
|
||||||
|
<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
|
||||||
|
<playground version='5.0' target-platform='osx'>
|
||||||
|
<timeline fileName='timeline.xctimeline'/>
|
||||||
|
</playground>
|
||||||
7
ch13/chapter13.playground/playground.xcworkspace/contents.xcworkspacedata
generated
Executable file
7
ch13/chapter13.playground/playground.xcworkspace/contents.xcworkspacedata
generated
Executable file
|
|
@ -0,0 +1,7 @@
|
||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<Workspace
|
||||||
|
version = "1.0">
|
||||||
|
<FileRef
|
||||||
|
location = "self:">
|
||||||
|
</FileRef>
|
||||||
|
</Workspace>
|
||||||
BIN
ch13/chapter13.playground/playground.xcworkspace/xcuserdata/marius.xcuserdatad/UserInterfaceState.xcuserstate
generated
Executable file
BIN
ch13/chapter13.playground/playground.xcworkspace/xcuserdata/marius.xcuserdatad/UserInterfaceState.xcuserstate
generated
Executable file
Binary file not shown.
|
|
@ -0,0 +1,6 @@
|
||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<Timeline
|
||||||
|
version = "3.0">
|
||||||
|
<TimelineItems>
|
||||||
|
</TimelineItems>
|
||||||
|
</Timeline>
|
||||||
Loading…
Reference in New Issue