mattr 0.0.3

Utility for transposing multi-dimensional data
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

use std::mem::transmute;

fn gen_data(width: usize, height: usize) -> Vec<usize> {
    (0..width*height).collect()
}

const BLOCK_SIZE: usize = 16;

#[test]
fn test_out_of_place_transpose() {
    let sizes = [
    	0, 1, 2,
    	BLOCK_SIZE - 1, BLOCK_SIZE, BLOCK_SIZE + 1, 
    	BLOCK_SIZE * 4 - 1, BLOCK_SIZE * 5, BLOCK_SIZE * 4 + 1
    	];

    for &width in &sizes {
        for &height in &sizes {
            let input = gen_data(width, height);
            let mut output = vec![0; width * height];

            unsafe { mattr::transpose(&input,transmute(output.as_mut_slice()), width, height) };

            for x in 0..width {
                for y in 0..height {
                    assert_eq!(input[x + y * width], output[y + x * height], "x = {}, y = {}", x, y);
                }
            }
        }
    }
}