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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
#ifndef __MATH_INC
#define __MATH_INC
#include "pema99.cginc"
float4 qmul(float4 q1, float4 q2)
{
return float4(
q2.xyz * q1.w + q1.xyz * q2.w + cross(q1.xyz, q2.xyz),
q1.w * q2.w - dot(q1.xyz, q2.xyz)
);
}
// Vector rotation with a quaternion
// http://mathworld.wolfram.com/Quaternion.html
float3 rotate_vector(float3 v, float4 r)
{
float4 r_c = r * float4(-1, -1, -1, 1);
return qmul(r, qmul(float4(v, 0), r_c)).xyz;
}
float4 get_quaternion(float3 axis_normal, float theta) {
return float4(
axis_normal * sin(theta / 2), cos(theta / 2));
}
// Differentiable approximation of the standard `max` function.
float dmax(float a, float b, float k)
{
return log2(exp2(k * a) + exp2(k * b)) / k;
}
// Differentiable approximation of the standard `min` function.
float dmin(float a, float b, float k)
{
return -1.0 * dmax(-1.0 * a, -1.0 * b, k);
}
float dabs(float a, float k)
{
return log2(exp2(k * a) + exp2(-1.0 * k * a));
}
float rand(uint seed) {
seed = seed * 747796405 + 2891336453;
uint result = ((seed >> ((seed >> 28) + 4)) ^ seed) * 277803737;
result = (result >> 22) ^ result;
return result / 4294967295.0;
}
// Generate a random number on [0, 1].
float rand2(float2 p)
{
return glsl_mod(sin(dot(p, float2(561.0, 885.0))) * 776.2, 1.0);
}
// Generate a random number on [0, 1].
float rand3(float3 p)
{
return glsl_mod(sin(dot(p, float3(897.0, 367.0, 197.0))) * 1073.6, 1.0);
}
float length2(float2 p)
{
return p.x * p.x + p.y * p.y;
}
// 3 dimensional value noise. `p` is assumed to be a point inside a unit cube.
// Theory: https://en.wikipedia.org/wiki/Value_noise
float vnoise3d(float3 p)
{
float3 pu = floor(p);
float3 pv = glsl_mod(frac(p), 1.0);
// Assign random numbers to the corner of a cube.
float n000 = rand3(pu + float3(0,0,0));
float n001 = rand3(pu + float3(0,0,1));
float n010 = rand3(pu + float3(0,1,0));
float n011 = rand3(pu + float3(0,1,1));
float n100 = rand3(pu + float3(1,0,0));
float n101 = rand3(pu + float3(1,0,1));
float n110 = rand3(pu + float3(1,1,0));
float n111 = rand3(pu + float3(1,1,1));
float n00 = lerp(n000, n001, pv.z);
float n01 = lerp(n010, n011, pv.z);
float n10 = lerp(n100, n101, pv.z);
float n11 = lerp(n110, n111, pv.z);
float n0 = lerp(n00, n01, pv.y);
float n1 = lerp(n10, n11, pv.y);
float n = lerp(n0, n1, pv.x);
return n;
}
float fbm(float3 p, const int n_octaves, float w)
{
float g = exp2(-w);
float a = 1.0;
float p_scale = 1.0;
float res = 0.0;
for (int i = 0; i < n_octaves; i++) {
res += a * vnoise3d(p * p_scale);
p_scale /= w;
a *= g;
}
return res;
}
#endif // __MATH_INC
|
