3 files changed, 15 insertions, 4 deletions
diff --git a/client/src/shaders/fragment_pbr.wgsl b/client/src/shaders/fragment_pbr.wgsl
index f3ed37e..f3387a1 100644
--- a/client/src/shaders/fragment_pbr.wgsl
+++ b/client/src/shaders/fragment_pbr.wgsl
@@ -17,7 +17,8 @@ struct VertexOut {
     @builtin(position) clip: vec4<f32>,
     @location(0) normal: vec3<f32>,
     @location(1) tangent: vec3<f32>,
-    @location(2) texcoord: vec2<f32>,
+    @location(2) view: vec3<f32>,
+    @location(3) texcoord: vec2<f32>,
 }
 
 struct Material {
@@ -44,9 +45,15 @@ fn main(vo: VertexOut) -> @location(0) vec4<f32> {
     let normal = tangent_basis * (t_normal.rgb * 2. - 1.);
 
     let alpha = t_albedo.a;
-    let lighting = mix(1., saturate(dot(LIGHT, normal)), 0.9);
+
+    let ambient = 0.1;
+    let diffuse = saturate(dot(LIGHT, normal));
+    let specular = pow(dot(reflect(-LIGHT, normal), vo.view), material.roughness);
+
+    let lighting = ambient + diffuse; // + specular;
 
     let color = t_albedo.rgb * lighting;
+    // let color = vec3(dot(normalize(vo.normal), normalize(vo.view)) * 0.5 + 0.5) ;
 
     // TODO better (and faster?) randomness for alpha dither
     if fract(dot(sin(vo.clip * 123.) * 1213., vec4(3., 2., 1., 4.))) > alpha {
diff --git a/client/src/shaders/vertex_world.wgsl b/client/src/shaders/vertex_world.wgsl
index 5d69acd..d7bf445 100644
--- a/client/src/shaders/vertex_world.wgsl
+++ b/client/src/shaders/vertex_world.wgsl
@@ -23,7 +23,8 @@ struct VertexOut {
     @builtin(position) clip: vec4<f32>,
     @location(0) normal: vec3<f32>,
     @location(1) tangent: vec3<f32>,
-    @location(2) texcoord: vec2<f32>,
+    @location(2) world: vec3<f32>,
+    @location(3) texcoord: vec2<f32>,
 }
 
 struct PushConst {
@@ -40,6 +41,7 @@ fn main(vi: VertexIn) -> VertexOut {
         clip,
         normalize(pc.model_basis * vi.normal),
         normalize(pc.model_basis * vi.tangent),
+        vi.position
         vi.texcoord
     );
     return vo;
diff --git a/client/src/shaders/vertex_world_skin.wgsl b/client/src/shaders/vertex_world_skin.wgsl
index 288950d..4b45a6f 100644
--- a/client/src/shaders/vertex_world_skin.wgsl
+++ b/client/src/shaders/vertex_world_skin.wgsl
@@ -25,7 +25,8 @@ struct VertexOut {
     @builtin(position) clip: vec4<f32>,
     @location(0) normal: vec3<f32>,
     @location(1) tangent: vec3<f32>,
-    @location(2) texcoord: vec2<f32>,
+    @location(2) world: vec3<f32>,
+    @location(3) texcoord: vec2<f32>,
 }
 
 struct PushConst {
@@ -43,6 +44,7 @@ fn main(vi: VertexIn) -> VertexOut {
         clip,
         normalize(pc.model_basis * vi.normal),
         normalize(pc.model_basis * vi.tangent),
+        vi.position,
         vi.texcoord
     );
     return vo;