Vector2 by eisterman

vector.rs

extern crate num;

use num::Float;
use std::ops::{Add,Sub,Mul};

#[derive(Debug, Eq, PartialEq)]
struct Vector2<T> {
    x: T,
    y: T,
}

impl<T: Add<Output=T>> Add for Vector2<T> {
    type Output = Vector2<T>;

    fn add(self, other: Vector2<T>) -> Vector2<T> {
        Vector2::<T> { x: self.x + other.x, y: self.y + other.y }
    }
}

impl<T: Sub<Output=T>> Sub for Vector2<T> {
    type Output = Vector2<T>;

    fn sub(self, other: Vector2<T>) -> Vector2<T> {
        Vector2::<T> { x: self.x - other.x, y: self.y - other.y }
    }
}

impl<T> Mul<T> for Vector2<T> where T: Mul<Output=T> + Copy {
    type Output = Vector2<T>;

    fn mul(self, rhs: T) -> Vector2<T> {
        Vector2::<T> { x: self.x * rhs, y: self.y * rhs }
    }
}

impl<T> Vector2<T> where T: Float {
    fn norm(self) -> T {
        (self.x * self.x + self.y * self.y).sqrt()
    }
}

//TODO: Come fare il 5 * Vec ?

#[test]
fn add_trait() {
    assert_eq!(Vector2::<f64> { x: 4., y: 1.}, Vector2::<f64> {x:3., y:1.5} + Vector2::<f64> {x:1., y:-0.5} );
}

#[test]
fn sub_trait() {
    assert_eq!(Vector2::<f64> { x: 2., y: 2.}, Vector2::<f64> {x:3., y:1.5} - Vector2::<f64> {x:1., y:-0.5} );
}

#[test]
fn mul_trait() {
    assert_eq!(Vector2::<f64> { x: 6., y: 3.}, Vector2::<f64> {x:3., y:1.5} * 2. ); 
}

#[test]
fn norm_test() {
    assert_eq!(Vector2::<f64> { x: 4., y: -3.}.norm(), 5_f64);
}