source: libcfa/src/vec/vec3.hfa @ 1712f542

arm-ehjacob/cs343-translationnew-ast-unique-expr
Last change on this file since 1712f542 was 1712f542, checked in by Dmitry Kobets <dkobets@…>, 2 years ago

Add extended ops for vec3 and tests

  • Property mode set to 100644
File size: 6.3 KB
Line 
1#pragma once
2
3#include <iostream.hfa>
4#include "vec.hfa"
5
6forall (otype T) {
7    struct vec3 {
8        T x, y, z;
9    };
10}
11
12
13forall (otype T) {
14    static inline {
15
16    void ?{}(vec3(T)& v, T x, T y, T z) {
17        v.[x, y, z] = [x, y, z];
18    }
19
20    forall(| zero_assign(T))
21    void ?{}(vec3(T)& vec, zero_t) with (vec) {
22        x = y = z = 0;
23    }
24
25    void ?{}(vec3(T)& vec, T val) with (vec) {
26        x = y = z = val;
27    }
28
29    void ?{}(vec3(T)& vec, vec3(T) other) with (vec) {
30        [x,y,z] = other.[x,y,z];
31    }
32
33    // Assignment
34    void ?=?(vec3(T)& vec, vec3(T) other) with (vec) {
35        [x,y,z] = other.[x,y,z];
36    }
37    forall(| zero_assign(T))
38    void ?=?(vec3(T)& vec, zero_t) with (vec) {
39        x = y = z = 0;
40    }
41
42    // Primitive mathematical operations
43
44    // -
45    forall(| subtract(T)) {
46    vec3(T) ?-?(vec3(T) u, vec3(T) v) { // TODO( can't make this const ref )
47        return [u.x - v.x, u.y - v.y, u.z - v.z];
48    }
49    vec3(T)& ?-=?(vec3(T)& u, vec3(T) v) {
50        u = u - v;
51        return u;
52    }
53    }
54
55    forall(| negate(T)) {
56    vec3(T) -?(vec3(T) v) with (v) {
57        return [-x, -y, -z];
58    }
59    }
60
61    forall(| { T --?(T&); }) {
62    vec3(T)& --?(vec3(T)& v) {
63        --v.x;
64        --v.y;
65        --v.z;
66        return v;
67    }
68    vec3(T) ?--(vec3(T)& v) {
69        vec3(T) copy = v;
70        --v;
71        return copy;
72    }
73    }
74
75    // +
76    forall(| add(T)) {
77    vec3(T) ?+?(vec3(T) u, vec3(T) v) { // TODO( can't make this const ref )
78        return [u.x + v.x, u.y + v.y, u.z + v.z];
79    }
80    vec3(T)& ?+=?(vec3(T)& u, vec3(T) v) {
81        u = u + v;
82        return u;
83    }
84    }
85
86
87    forall(| { T ++?(T&); }) {
88    vec3(T)& ++?(vec3(T)& v) {
89        ++v.x;
90        ++v.y;
91        ++v.z;
92        return v;
93    }
94    vec3(T) ?++(vec3(T)& v) {
95        vec3(T) copy = v;
96        ++v;
97        return copy;
98    }
99    }
100
101    // *
102    forall(| multiply(T)) {
103    vec3(T) ?*?(vec3(T) v, T scalar) with (v) { // TODO (can't make this const ref)
104        return [x * scalar, y * scalar, z * scalar];
105    }
106    vec3(T) ?*?(T scalar, vec3(T) v) { // TODO (can't make this const ref)
107        return v * scalar;
108    }
109    vec3(T) ?*?(vec3(T) u, vec3(T) v) {
110        return [u.x * v.x, u.y * v.y, u.z * v.z];
111    }
112    vec3(T)& ?*=?(vec3(T)& v, T scalar) {
113        v = v * scalar;
114        return v;
115    }
116    vec3(T)& ?*=?(vec3(T)& u, vec3(T) v) {
117        u = u * v;
118        return u;
119    }
120    }
121
122    // /
123    forall(| divide(T)) {
124    vec3(T) ?/?(vec3(T) v, T scalar) with (v) {
125        return [x / scalar, y / scalar, z / scalar];
126    }
127    vec3(T) ?/?(vec3(T) u, vec3(T) v) {
128        return [u.x / v.x, u.y / v.y, u.z / v.z];
129    }
130    vec3(T)& ?/=?(vec3(T)& v, T scalar) with (v) {
131        v = v / scalar;
132        return v;
133    }
134    vec3(T)& ?/=?(vec3(T)& u, vec3(T) v) {
135        u = u / v;
136        return u;
137    }
138    }
139
140    // %
141    forall(| { T ?%?(T,T); }) {
142    vec3(T) ?%?(vec3(T) v, T scalar) with (v) {
143        return [x % scalar, y % scalar, z % scalar];
144    }
145    vec3(T)& ?%=?(vec3(T)& u, T scalar) {
146        u = u % scalar;
147        return u;
148    }
149    vec3(T) ?%?(vec3(T) u, vec3(T) v) {
150        return [u.x % v.x, u.y % v.y, u.z % v.z];
151    }
152    vec3(T)& ?%=?(vec3(T)& u, vec3(T) v) {
153        u = u % v;
154        return u;
155    }
156    }
157
158    // &
159    forall(| { T ?&?(T,T); }) {
160    vec3(T) ?&?(vec3(T) v, T scalar) with (v) {
161        return [x & scalar, y & scalar, z & scalar];
162    }
163    vec3(T)& ?&=?(vec3(T)& u, T scalar) {
164        u = u & scalar;
165        return u;
166    }
167    vec3(T) ?&?(vec3(T) u, vec3(T) v) {
168        return [u.x & v.x, u.y & v.y, u.z & v.z];
169    }
170    vec3(T)& ?&=?(vec3(T)& u, vec3(T) v) {
171        u = u & v;
172        return u;
173    }
174    }
175
176    // |
177    forall(| { T ?|?(T,T); }) {
178    vec3(T) ?|?(vec3(T) v, T scalar) with (v) {
179        return [x | scalar, y | scalar, z | scalar];
180    }
181    vec3(T)& ?|=?(vec3(T)& u, T scalar) {
182        u = u | scalar;
183        return u;
184    }
185    vec3(T) ?|?(vec3(T) u, vec3(T) v) {
186        return [u.x | v.x, u.y | v.y, u.z | v.z];
187    }
188    vec3(T)& ?|=?(vec3(T)& u, vec3(T) v) {
189        u = u | v;
190        return u;
191    }
192    }
193
194    // ^
195    forall(| { T ?^?(T,T); }) {
196    vec3(T) ?^?(vec3(T) v, T scalar) with (v) {
197        return [x ^ scalar, y ^ scalar, z ^ scalar];
198    }
199    vec3(T)& ?^=?(vec3(T)& u, T scalar) {
200        u = u ^ scalar;
201        return u;
202    }
203    vec3(T) ?^?(vec3(T) u, vec3(T) v) {
204        return [u.x ^ v.x, u.y ^ v.y, u.z ^ v.z];
205    }
206    vec3(T)& ?^=?(vec3(T)& u, vec3(T) v) {
207        u = u ^ v;
208        return u;
209    }
210    }
211
212    // <<
213    forall(| { T ?<<?(T,T); }) {
214    vec3(T) ?<<?(vec3(T) v, T scalar) with (v) {
215        return [x << scalar, y << scalar, z << scalar];
216    }
217    vec3(T)& ?<<=?(vec3(T)& u, T scalar) {
218        u = u << scalar;
219        return u;
220    }
221    vec3(T) ?<<?(vec3(T) u, vec3(T) v) {
222        return [u.x << v.x, u.y << v.y, u.z << v.z];
223    }
224    vec3(T)& ?<<=?(vec3(T)& u, vec3(T) v) {
225        u = u << v;
226        return u;
227    }
228    }
229
230    // >>
231    forall(| { T ?>>?(T,T); }) {
232    vec3(T) ?>>?(vec3(T) v, T scalar) with (v) {
233        return [x >> scalar, y >> scalar, z >> scalar];
234    }
235    vec3(T)& ?>>=?(vec3(T)& u, T scalar) {
236        u = u >> scalar;
237        return u;
238    }
239    vec3(T) ?>>?(vec3(T) u, vec3(T) v) {
240        return [u.x >> v.x, u.y >> v.y, u.z >> v.z];
241    }
242    vec3(T)& ?>>=?(vec3(T)& u, vec3(T) v) {
243        u = u >> v;
244        return u;
245    }
246    }
247
248    // ~
249    forall(| { T ~?(T); })
250    vec3(T) ~?(vec3(T) v) with (v) {
251        return [~v.x, ~v.y, ~v.z];
252    }
253
254    // relational
255    forall(| equality(T)) {
256    bool ?==?(vec3(T) u, vec3(T) v) with (u) {
257        return x == v.x && y == v.y && z == v.z;
258    }
259    bool ?!=?(vec3(T) u, vec3(T) v) {
260        return !(u == v);
261    }
262    }
263
264    // Geometric functions
265    forall(| add(T) | multiply(T))
266    T dot(vec3(T) u, vec3(T) v) {
267        return u.x * v.x + u.y * v.y + u.z * v.z;
268    }
269
270    forall(| subtract(T) | multiply(T))
271    vec3(T) cross(vec3(T) u, vec3(T) v) {
272        return (vec3(T)){ u.y * v.z - v.y * u.z,
273                          u.z * v.x - v.z * u.x,
274                          u.x * v.y - v.x * u.y };
275    }
276
277    } // static inline
278}
279
280forall(dtype ostype, otype T | writeable(T, ostype)) {
281    ostype & ?|?(ostype & os, vec3(T) v) with (v) {
282        return os | '<' | x | ',' | y | ',' | z | '>';
283    }
284    void ?|?(ostype & os, vec3(T) v ) with (v) {
285        (ostype &)(os | v); ends(os);
286    }
287}
Note: See TracBrowser for help on using the repository browser.