Add a swizzle! macro
Allows for arbitrarily rearranging the rows of a matrix like in HLSL/GLSL.master
parent
bd1bde1657
commit
61b1d211c1
@ -0,0 +1,119 @@
|
|||||||
|
/** Rearrange the rows of a matrix by the given row identifiers
|
||||||
|
# Arguments
|
||||||
|
|
||||||
|
* `mat`: the matrix to manipulate
|
||||||
|
* `i...`: variadic comma-seperated list of row selectors. The row selectors are each one of:
|
||||||
|
* const expression representing the row index of the input to copy. eg: `0` or `2`
|
||||||
|
* a letter representing the same. you can use `r,g,b,a` or `x,y,z,w` to represent index 0 through 3,
|
||||||
|
or `u,v` to represent indices 0 through 1.
|
||||||
|
* an expression in curly braces, representing a value to be copied to an entire row of
|
||||||
|
the result, eg: `{1.0}` or `{5}`
|
||||||
|
|
||||||
|
note that the number of selectors doesnt need to match the height of the input!
|
||||||
|
|
||||||
|
# Examples
|
||||||
|
```
|
||||||
|
# use vector_victor::{swizzle, Vector};
|
||||||
|
let myvec = Vector::vec([0, 1, 2, 3]);
|
||||||
|
|
||||||
|
// each element can be selected with:
|
||||||
|
// 0: r, x, u, or 0
|
||||||
|
// 1: g, y, v, or 1
|
||||||
|
// 2: b, z, or 2
|
||||||
|
// 3: a, w, or 3
|
||||||
|
// or a result row can be filled by a new value
|
||||||
|
|
||||||
|
assert_eq!(swizzle!(myvec, a, z, v, 0, {7}), Vector::vec([3, 2, 1, 0, 7]));
|
||||||
|
```
|
||||||
|
|
||||||
|
More often you wont mix and match selector "flavors".
|
||||||
|
This example unpacks a [DXT5nm](http://wiki.polycount.com/wiki/Normal_Map_Compression) color
|
||||||
|
into the red and green channels, with blue filled with 0.
|
||||||
|
```
|
||||||
|
# use vector_victor::{swizzle, Vector};
|
||||||
|
let myvec = Vector::vec([0, 120, 0, 255]);
|
||||||
|
assert_eq!(swizzle!(myvec, a, g, {0}), Vector::vec([255, 120, 0]));
|
||||||
|
``` */
|
||||||
|
#[macro_export]
|
||||||
|
macro_rules! swizzle {
|
||||||
|
($mat: expr, $( $i:tt),+) => {{
|
||||||
|
let mut result = $mat.permute_rows(&$crate::Vector::vec([$( $crate::get!($mat, $i), )+]));
|
||||||
|
let mut _p = 0usize;
|
||||||
|
$(
|
||||||
|
$crate::sub_literal!(result, _p, $i);
|
||||||
|
_p += 1;
|
||||||
|
)+
|
||||||
|
result
|
||||||
|
}};
|
||||||
|
}
|
||||||
|
|
||||||
|
#[doc(hidden)]
|
||||||
|
#[macro_export]
|
||||||
|
macro_rules! get {
|
||||||
|
($mat:expr, x) => {
|
||||||
|
0usize
|
||||||
|
};
|
||||||
|
($mat:expr, y) => {
|
||||||
|
1usize
|
||||||
|
};
|
||||||
|
($mat:expr, z) => {
|
||||||
|
2usize
|
||||||
|
};
|
||||||
|
($mat:expr, w) => {
|
||||||
|
3usize
|
||||||
|
};
|
||||||
|
($mat:expr, r) => {
|
||||||
|
0usize
|
||||||
|
};
|
||||||
|
($mat:expr, g) => {
|
||||||
|
1usize
|
||||||
|
};
|
||||||
|
($mat:expr, b) => {
|
||||||
|
2usize
|
||||||
|
};
|
||||||
|
($mat:expr, a) => {
|
||||||
|
3usize
|
||||||
|
};
|
||||||
|
($mat:expr, u) => {
|
||||||
|
0usize
|
||||||
|
};
|
||||||
|
($mat:expr, v) => {
|
||||||
|
1usize
|
||||||
|
};
|
||||||
|
($mat:expr, {$i:expr}) => {
|
||||||
|
0usize /* will be replaced later */
|
||||||
|
};
|
||||||
|
($mat:expr, $i:expr) => {
|
||||||
|
$i
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
#[doc(hidden)]
|
||||||
|
#[macro_export]
|
||||||
|
macro_rules! sub_literal {
|
||||||
|
($result:expr, $p:expr, {$i:expr}) => {
|
||||||
|
$result.set_row($p, &$crate::Vector::fill($i))
|
||||||
|
};
|
||||||
|
($result:expr, $p:expr, $i:expr) => {};
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use crate::Matrix;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_swizzle() {
|
||||||
|
let identity = Matrix::<i32, 4, 4>::identity();
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
swizzle!(identity, a, x, b, (1 + 0), { 2 - 3 }),
|
||||||
|
Matrix::mat([
|
||||||
|
[0, 0, 0, 1], // a
|
||||||
|
[1, 0, 0, 0], // x
|
||||||
|
[0, 0, 1, 0], // b
|
||||||
|
[0, 1, 0, 0], // row 1
|
||||||
|
[-1, -1, -1, -1] // fill(-1)
|
||||||
|
])
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue