PyTouhou: src/util/math.rs annotate

author	Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
date	Fri, 02 Aug 2019 20:24:45 +0200
parents
children

rev	line source
643 01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	1 //! Various helpers to deal with vectors and matrices.
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	2
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	3 /// A 4×4 f32 matrix type.
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	4 pub struct Mat4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	5 inner: [[f32; 4]; 4]
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	6 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	7
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	8 impl Mat4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	9 /// Create a new matrix from a set of 16 f32.
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	10 pub fn new(inner: [[f32; 4]; 4]) -> Mat4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	11 Mat4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	12 inner
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	13 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	14 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	15
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	16 fn zero() -> Mat4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	17 Mat4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	18 inner: [[0.; 4]; 4]
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	19 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	20 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	21
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	22 fn identity() -> Mat4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	23 Mat4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	24 inner: [[1., 0., 0., 0.],
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	25 [0., 1., 0., 0.],
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	26 [0., 0., 1., 0.],
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	27 [0., 0., 0., 1.]]
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	28 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	29 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	30
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	31 /// Immutably borrow the array of f32 inside this matrix.
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	32 pub fn borrow_inner(&self) -> &[[f32; 4]; 4] {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	33 &self.inner
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	34 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	35
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	36 /// Scale the matrix in 2D.
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	37 pub fn scale2d(&mut self, x: f32, y: f32) {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	38 for i in 0..4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	39 self.inner[0][i] *= x;
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	40 self.inner[1][i] *= y;
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	41 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	42 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	43
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	44 /// Flip the matrix.
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	45 pub fn flip(&mut self) {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	46 for i in 0..4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	47 self.inner[0][i] = -self.inner[0][i];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	48 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	49 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	50
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	51 /// Rotate the matrix around its x angle (in radians).
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	52 pub fn rotate_x(&mut self, angle: f32) {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	53 let mut lines: [f32; 8] = [0.; 8];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	54 let cos_a = angle.cos();
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	55 let sin_a = angle.sin();
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	56 for i in 0..4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	57 lines[ i] = self.inner[0][i];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	58 lines[4 + i] = self.inner[1][i];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	59 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	60 for i in 0..4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	61 self.inner[1][i] = cos_a * lines[i] - sin_a * lines[4+i];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	62 self.inner[2][i] = sin_a * lines[i] + cos_a * lines[4+i];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	63 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	64 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	65
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	66 /// Rotate the matrix around its y angle (in radians).
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	67 pub fn rotate_y(&mut self, angle: f32) {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	68 let mut lines: [f32; 8] = [0.; 8];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	69 let cos_a = angle.cos();
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	70 let sin_a = angle.sin();
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	71 for i in 0..4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	72 lines[ i] = self.inner[0][i];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	73 lines[4 + i] = self.inner[2][i];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	74 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	75 for i in 0..4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	76 self.inner[0][i] = cos_a * lines[i] + sin_a * lines[4+i];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	77 self.inner[2][i] = -sin_a * lines[i] + cos_a * lines[4+i];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	78 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	79 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	80
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	81 /// Rotate the matrix around its z angle (in radians).
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	82 pub fn rotate_z(&mut self, angle: f32) {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	83 let mut lines: [f32; 8] = [0.; 8];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	84 let cos_a = angle.cos();
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	85 let sin_a = angle.sin();
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	86 for i in 0..4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	87 lines[ i] = self.inner[0][i];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	88 lines[4 + i] = self.inner[1][i];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	89 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	90 for i in 0..4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	91 self.inner[0][i] = cos_a * lines[i] - sin_a * lines[4+i];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	92 self.inner[1][i] = sin_a * lines[i] + cos_a * lines[4+i];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	93 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	94 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	95
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	96 /// Translate the matrix by a 3D offset.
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	97 pub fn translate(&mut self, offset: [f32; 3]) {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	98 let mut item: [f32; 3] = [0.; 3];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	99 for i in 0..3 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	100 item[i] = self.inner[3][i] * offset[i];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	101 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	102 for i in 0..3 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	103 for j in 0..4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	104 self.inner[i][j] += item[i];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	105 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	106 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	107 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	108
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	109 /// Translate the matrix by a 2D offset.
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	110 pub fn translate_2d(&mut self, x: f32, y: f32) {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	111 let offset = [x, y, 0.];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	112 self.translate(offset);
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	113 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	114 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	115
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	116 impl std::ops::Mul<Mat4> for Mat4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	117 type Output = Mat4;
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	118 fn mul(self, rhs: Mat4) -> Mat4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	119 let mut tmp = Mat4::zero();
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	120 for i in 0..4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	121 for j in 0..4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	122 for k in 0..4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	123 tmp.inner[i][j] += self.inner[i][k] * rhs.inner[k][j];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	124 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	125 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	126 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	127 tmp
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	128 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	129 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	130
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	131 /// Create an orthographic projection matrix.
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	132 pub fn ortho_2d(left: f32, right: f32, bottom: f32, top: f32) -> Mat4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	133 let mut mat = Mat4::identity();
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	134 mat.inner[0][0] = 2. / (right - left);
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	135 mat.inner[1][1] = 2. / (top - bottom);
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	136 mat.inner[2][2] = -1.;
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	137 mat.inner[3][0] = -(right + left) / (right - left);
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	138 mat.inner[3][1] = -(top + bottom) / (top - bottom);
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	139 mat
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	140 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	141
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	142 /// Setup a camera view matrix.
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	143 pub fn setup_camera(dx: f32, dy: f32, dz: f32) -> Mat4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	144 // Some explanations on the magic constants:
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	145 // 192. = 384. / 2. = width / 2.
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	146 // 224. = 448. / 2. = height / 2.
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	147 // 835.979370 = 224./math.tan(math.radians(15)) = (height/2.)/math.tan(math.radians(fov/2))
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	148 // This is so that objects on the (O, x, y) plane use pixel coordinates
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	149 look_at([192., 224., -835.979370 * dz], [192. + dx, 224. - dy, 0.], [0., -1., 0.])
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	150 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	151
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	152 /// Creates a perspective projection matrix.
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	153 pub fn perspective(fov_y: f32, aspect: f32, z_near: f32, z_far: f32) -> Mat4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	154 let top = (fov_y / 2.).tan() * z_near;
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	155 let bottom = -top;
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	156 let left = -top * aspect;
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	157 let right = top * aspect;
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	158
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	159 let mut mat = Mat4::identity();
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	160 mat.inner[0][0] = (2. * z_near) / (right - left);
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	161 mat.inner[1][1] = (2. * z_near) / (top - bottom);
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	162 mat.inner[2][2] = -(z_far + z_near) / (z_far - z_near);
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	163 mat.inner[2][3] = -1.;
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	164 mat.inner[3][2] = -(2. * z_far * z_near) / (z_far - z_near);
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	165 mat.inner[3][3] = 0.;
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	166 mat
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	167 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	168
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	169 type Vec3 = [f32; 3];
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	170
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	171 fn look_at(eye: Vec3, center: Vec3, up: Vec3) -> Mat4 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	172 let f = normalize(sub(center, eye));
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	173 let u = normalize(up);
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	174 let s = normalize(cross(f, u));
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	175 let u = cross(s, f);
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	176
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	177 Mat4::new([[s[0], u[0], -f[0], 0.],
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	178 [s[1], u[1], -f[1], 0.],
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	179 [s[2], u[2], -f[2], 0.],
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	180 [-dot(s, eye), -dot(u, eye), dot(f, eye), 1.]])
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	181 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	182
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	183 fn sub(a: Vec3, b: Vec3) -> Vec3 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	184 [a[0] - b[0],
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	185 a[1] - b[1],
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	186 a[2] - b[2]]
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	187 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	188
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	189 fn normalize(vec: Vec3) -> Vec3 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	190 let normal = 1. / (vec[0] * vec[0] + vec[1] * vec[1] + vec[2] * vec[2]).sqrt();
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	191 [vec[0] * normal, vec[1] * normal, vec[2] * normal]
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	192 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	193
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	194 fn cross(a: Vec3, b: Vec3) -> Vec3 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	195 [a[1] * b[2] - b[1] * a[2],
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	196 a[2] * b[0] - b[2] * a[0],
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	197 a[0] * b[1] - b[0] * a[1]]
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	198 }
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	199
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	200 fn dot(a: Vec3, b: Vec3) -> f32 {
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	201 a[0] * b[0] + a[1] * b[1] + a[2] * b[2]
01849ffd0180 Add an anmrenderer binary. Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> parents: diff changeset	202 }

643

01849ffd0180 Add an anmrenderer binary.