diff options
Diffstat (limited to 'Demo/classes/Vec.py')
-rwxr-xr-x | Demo/classes/Vec.py | 82 |
1 files changed, 41 insertions, 41 deletions
diff --git a/Demo/classes/Vec.py b/Demo/classes/Vec.py index 8289bc8..0a56ddb 100755 --- a/Demo/classes/Vec.py +++ b/Demo/classes/Vec.py @@ -2,63 +2,63 @@ def vec(*v): - return apply(Vec, v) + return apply(Vec, v) class Vec: - def __init__(self, *v): - self.v = [] - for x in v: - self.v.append(x) + def __init__(self, *v): + self.v = [] + for x in v: + self.v.append(x) - def fromlist(self, v): - self.v = [] - if type(v) <> type([]): - raise TypeError - self.v = v[:] - return self + def fromlist(self, v): + self.v = [] + if type(v) <> type([]): + raise TypeError + self.v = v[:] + return self - def __repr__(self): - return 'vec(' + `self.v`[1:-1] + ')' + def __repr__(self): + return 'vec(' + `self.v`[1:-1] + ')' - def __len__(self): - return len(self.v) + def __len__(self): + return len(self.v) - def __getitem__(self, i): - return self.v[i] + def __getitem__(self, i): + return self.v[i] - def __add__(a, b): - # Element-wise addition - v = [] - for i in range(len(a)): - v.append(a[i] + b[i]) - return Vec().fromlist(v) + def __add__(a, b): + # Element-wise addition + v = [] + for i in range(len(a)): + v.append(a[i] + b[i]) + return Vec().fromlist(v) - def __sub__(a, b): - # Element-wise subtraction - v = [] - for i in range(len(a)): - v.append(a[i] - b[i]) - return Vec().fromlist(v) + def __sub__(a, b): + # Element-wise subtraction + v = [] + for i in range(len(a)): + v.append(a[i] - b[i]) + return Vec().fromlist(v) - def __mul__(self, scalar): - # Multiply by scalar - v = [] - for i in range(len(self.v)): - v.append(self.v[i]*scalar) - return Vec().fromlist(v) + def __mul__(self, scalar): + # Multiply by scalar + v = [] + for i in range(len(self.v)): + v.append(self.v[i]*scalar) + return Vec().fromlist(v) def test(): - a = vec(1, 2, 3) - b = vec(3, 2, 1) - print a - print b - print a+b - print a*3.0 + a = vec(1, 2, 3) + b = vec(3, 2, 1) + print a + print b + print a+b + print a*3.0 test() |