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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
|
/*
* Copyright (C) 2015 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#pragma version(1)
#pragma rs java_package_name(com.example.android.rs.vr.engine)
#pragma rs_fp_relaxed
int brick_dimx;
int brick_dimy;
int brick_dimz;
rs_allocation volume;
rs_allocation opacity;
int z_offset;
// output a single bit per pixel volume based on opacity
uint __attribute__((kernel)) pack_chunk(uint32_t x) {
int brick = x / (32 * 32);
int bx = brick % brick_dimx;
int yz = brick / brick_dimx;
int by = yz % brick_dimy;
int bz = yz / brick_dimy;
int in_brick = x % (32 * 32);
int in_br_y = in_brick % 32;
int in_br_z = in_brick / 32;
int pz = (bz << 5) | in_br_z;
int py = (by << 5) | in_br_y;
uint out = 0;
if (pz >= rsAllocationGetDimZ(volume)) {
return out;
}
if (py >= rsAllocationGetDimY(volume)) {
return out;
}
for (int in_br_x = 0; in_br_x < 32; in_br_x++) {
int px = (bx << 5) | in_br_x;
int intensity = 0xFFFF & rsGetElementAt_short(volume, px, py, pz);
uchar op = rsGetElementAt_uchar(opacity, intensity);
uint v = (op > 0) ? 1 : 0;
out |= v << in_br_x;
}
return out;
}
rs_allocation bricks; // input bricks
uint __attribute__((kernel)) dilate(uint in, uint32_t x) {
int BRICK_SIZE = 32 * 32;
int brick = x / (BRICK_SIZE);
int bx = brick % brick_dimx;
int yz = brick / brick_dimx;
int by = yz % brick_dimy;
int bz = yz / brick_dimy;
int in_brick = x % (BRICK_SIZE);
int in_br_y = in_brick % 32;
int in_br_z = in_brick / 32;
uint slice;
uint out = in;
out |= in >> 1;
out |= in << 1;
int base_brick = bx + (by + brick_dimy * bz) * brick_dimx;
int base_offset = (in_br_z) * 32 + (in_br_y);
int slice_pos = base_brick * BRICK_SIZE + base_offset;
// +/- in x
if (bx > 0) {
slice = 0x80000000 & rsGetElementAt_uint(bricks, slice_pos - BRICK_SIZE);
out |= slice >> 31;
}
if (bx < brick_dimx - 1) {
slice = 1 & rsGetElementAt_uint(bricks, slice_pos + BRICK_SIZE);
out |= slice << 31;
}
// - in Y
int off_neg_y = -1; // simple case -1 slice;
if (in_br_y == 0) { // att the edge in brick go to y-1 brick
if (by > 0) { // edge of screen
off_neg_y = 31 - BRICK_SIZE * brick_dimx;
} else {// edge of volume
off_neg_y = 0;
}
}
slice = rsGetElementAt_uint(bricks, slice_pos + off_neg_y);
out |= slice;
// + in Y
int off_pos_y = 1;
if (in_br_y == 31) {
if (by < brick_dimy - 1) {
off_pos_y = BRICK_SIZE * brick_dimx - 31;
} else {
off_pos_y = 0;
}
}
slice = rsGetElementAt_uint(bricks, slice_pos + off_pos_y);
out |= slice;
int off_neg_z = -32;
if (in_br_z == 0) {
if (bz > 0) { // edge of screen
off_neg_z = 31*32 - brick_dimx * brick_dimy* BRICK_SIZE;
} else {
off_neg_z = 0;
}
}
slice = rsGetElementAt_uint(bricks, slice_pos + off_neg_z);
out |= slice;
int off_pos_z = 32;
if (in_br_z == 31) {
if (bz < brick_dimz - 1) {
off_pos_z = brick_dimx * brick_dimy * BRICK_SIZE - 31*32;
} else {
off_pos_z = 0;
}
}
slice = rsGetElementAt_uint(bricks, slice_pos + off_pos_z);
out |= slice;
return out;
}
int z;
void __attribute__((kernel)) copy(short in, uint32_t x, uint32_t y) {
rsSetElementAt_short(volume, in, x, y, z);
}
int size;
static float3 nylander(float3 p, int n) {
float3 out;
float r = length(p);
float phi = atan2(p.y, p.x);
float theta = acos(p.z / r);
float rn = pow(r, n);
out.x = sin(n * theta) * cos(n * phi);
out.y = sin(n * theta) * sin(n * phi);
out.z = cos(n * theta);
return out * rn;
}
/**
* 8 x faster than the above for n = 3
*/
static float3 nylander3(float3 p) {
float3 out = (float3){0.f, 0.f, 0.f};
float xy2 = p.x * p.x + p.y * p.y;
if (xy2 == 0) return out;
float z23x2y2 = (3 * p.z * p.z - p.x * p.x - p.y * p.y);
out.x = (z23x2y2 * p.x * (p.x * p.x - 3 * p.y * p.y)) / xy2;
out.y = (z23x2y2 * p.y * (3 * p.x * p.x - p.y * p.y)) / xy2;
out.z = p.z * (p.z * p.z - 3 * p.x * p.x - 3 * p.y * p.y);
return out;
}
static float vsize(float3 p) {
return sqrt(p.x * p.x + p.y * p.y + p.z * p.z);
}
short __attribute__((kernel)) mandelbulb(uint32_t x, uint32_t y) {
int size2 = size / 2;
if (z < size2) {
return 256-4*(size2-z+4)*hypot((float)x-size2,(float)y-size2) / size2 ;
}
float3 c = (float3) {(float) x, (float) y, (float) z};
c = ((c - size2) / (size2 * .9f));
int loop = 25;
float3 p = c;
float len;
for (int i = 0; i < loop; i++) {
// p = nylander(p, 3) + c;
p = nylander3(p) + c;
len = fast_length(p);
if (len > 2.f) return 255 - loop*10;
if (len < .3f) return loop*10;
}
len = length(p);
return (short) (255 - (len * 255) / 4);
}
|
