1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
uniform mat4 ProjectionMatrix;
uniform mat4 ViewMatrix;
uniform sampler2D strokeColor;
uniform sampler2D strokeDepth;
uniform vec2 Viewport;
uniform vec4 loc;
uniform float energy;
uniform float ambient;
uniform float pixsize; /* rv3d->pixsize */
uniform float pixfactor;
out vec4 FragColor;
float defaultpixsize = pixsize * (1000.0 / pixfactor);
#define height loc.w
/* project 3d point to 2d on screen space */
vec2 toScreenSpace(vec4 vertex)
{
/* need to calculate ndc because this is not done by vertex shader */
vec3 ndc = vec3(vertex).xyz / vertex.w;
vec2 sc;
sc.x = ((ndc.x + 1.0) / 2.0) * Viewport.x;
sc.y = ((ndc.y + 1.0) / 2.0) * Viewport.y;
return sc;
}
void main()
{
float stroke_depth;
vec4 objcolor;
vec4 light_loc = ProjectionMatrix * ViewMatrix * vec4(loc.xyz, 1.0);
vec2 light2d = toScreenSpace(light_loc);
/* calc pixel scale */
float pxscale = (ProjectionMatrix[3][3] == 0.0) ? (10.0 / (light_loc.z * defaultpixsize)) :
(10.0 / defaultpixsize);
pxscale = max(pxscale, 0.000001);
/* the height over plane is received in the w component of the loc
* and needs a factor to adapt to pixels
*/
float peak = height * 10.0 * pxscale;
vec3 light3d = vec3(light2d.x, light2d.y, peak);
vec2 uv = vec2(gl_FragCoord.xy);
vec3 frag_loc = vec3(uv.x, uv.y, 0);
vec3 norm = vec3(0, 0, 1.0); /* always z-up */
ivec2 iuv = ivec2(uv.x, uv.y);
stroke_depth = texelFetch(strokeDepth, iuv, 0).r;
objcolor = texelFetch(strokeColor, iuv, 0);
/* diffuse light */
vec3 lightdir = normalize(light3d - frag_loc);
float diff = max(dot(norm, lightdir), 0.0);
float dist = length(light3d - frag_loc) / pxscale;
float factor = diff * ((energy * 100.0) / (dist * dist));
vec3 result = factor * max(ambient, 0.1) * vec3(objcolor);
gl_FragDepth = stroke_depth;
FragColor = vec4(result.r, result.g, result.b, objcolor.a);
}
|
