Mapytex/pymath/fraction.py

342 lines
8.4 KiB
Python
Raw Normal View History

#!/usr/bin/env python
# encoding: utf-8
2014-02-21 05:01:34 +00:00
from .arithmetic import gcd
2014-02-27 17:02:34 +00:00
__all__ = ['Fraction']
class Fraction(object):
2013-08-07 23:12:11 +00:00
"""Fractions!"""
2013-08-07 23:12:11 +00:00
def __init__(self, num, denom = 1):
"""To initiate a fraction we need a numerator and a denominator
2013-08-07 23:12:11 +00:00
:param num: the numerator
:param denom: the denominator
2013-08-07 23:12:11 +00:00
"""
self._num = num
self._denom = denom
self.isNumber = 1
2013-08-07 23:12:11 +00:00
def simplify(self):
"""Simplify the fraction
2013-08-07 23:12:11 +00:00
:returns: steps to simplify the fraction or the fraction if there is nothing to do
"""
steps = []
2014-02-21 10:05:25 +00:00
if self._num == 0:
steps.append(0)
return steps
2013-08-07 23:12:11 +00:00
if self._denom < 0:
n_frac = Fraction(-self._num, -self._denom)
steps.append(n_frac)
else:
n_frac = self
gcd_ = gcd(abs(n_frac._num), abs(n_frac._denom))
if gcd_ == n_frac._denom:
n_frac = n_frac._num // gcd_
steps.append(n_frac)
2013-08-07 23:12:11 +00:00
elif gcd_ != 1:
n_frac = Fraction(n_frac._num // gcd_ , n_frac._denom // gcd_)
steps.append([n_frac._num, gcd_, '*', n_frac._denom, gcd_, '*', '/' ])
steps.append(n_frac)
2013-08-07 23:12:11 +00:00
return steps
2013-08-07 23:12:11 +00:00
def __str__(self):
if self._denom == 1:
return str(self._num)
else:
return str(self._num) + " / " + str(self._denom)
def __repr__(self):
return "< Fraction " + self.__str__() + ">"
2014-02-21 10:08:55 +00:00
def __txt__(self):
return str(self)
def __tex__(self):
if self._denom == 1:
return str(self._num)
else:
return "\\frac{{ {a} }}{{ {b} }}".format(a = self._num, b = self._denom)
2014-02-21 10:08:55 +00:00
def __float__(self):
return self._num / self._denom
2014-05-28 16:39:08 +00:00
def convert2fraction(self, other):
""" Convert a other into a fraction """
2013-08-07 23:12:11 +00:00
if type(other) == Fraction:
#cool
number = other
else:
number = Fraction(other)
2014-05-28 16:39:08 +00:00
return number
def __add__(self, other):
if other == 0:
return [self]
2014-05-28 16:39:08 +00:00
number = self.convert2fraction(other)
steps = []
2013-08-07 23:12:11 +00:00
if self._denom == number._denom:
com_denom = self._denom
num1 = self._num
num2 = number._num
2013-08-07 23:12:11 +00:00
else:
gcd_denom = gcd(self._denom, number._denom)
coef1 = number._denom // gcd_denom
coef2 = self._denom // gcd_denom
steps.append([self._num, coef1, "*", self._denom, coef1, "*", '/', number._num, coef2, "*", number._denom, coef2, "*", "/",'+'])
2013-08-07 23:12:11 +00:00
com_denom = self._denom * coef1
num1 = self._num * coef1
num2 = number._num * coef2
steps.append([num1, num2, '+', com_denom, '/'])
2013-08-07 23:12:11 +00:00
num = num1 + num2
2013-08-07 23:12:11 +00:00
ans_frac = Fraction(num, com_denom)
steps.append(ans_frac)
2013-08-07 23:12:11 +00:00
steps += ans_frac.simplify()
2013-08-07 23:12:11 +00:00
return steps
def __radd__(self, other):
if other == 0:
return [self]
2014-05-28 16:39:08 +00:00
number = self.convert2fraction(other)
2014-05-28 16:49:19 +00:00
return number + self
2013-08-07 23:12:11 +00:00
def __sub__(self, other):
if other == 0:
return [self]
2014-05-28 16:39:08 +00:00
number = self.convert2fraction(other)
steps = []
2013-08-07 23:12:11 +00:00
if self._denom == number._denom:
com_denom = self._denom
num1 = self._num
num2 = number._num
2013-08-07 23:12:11 +00:00
else:
gcd_denom = gcd(self._denom, number._denom)
coef1 = number._denom // gcd_denom
coef2 = self._denom // gcd_denom
steps.append([self._num, coef1, "*", self._denom, coef1, "*", '/', number._num, coef2, "*", number._denom, coef2, "*", "/",'-'])
2013-08-07 23:12:11 +00:00
com_denom = self._denom * coef1
num1 = self._num * coef1
num2 = number._num * coef2
steps.append([num1, num2, '-', com_denom, '/'])
2013-08-07 23:12:11 +00:00
num = num1 - num2
2013-08-07 23:12:11 +00:00
ans_frac = Fraction(num, com_denom)
steps.append(ans_frac)
2013-08-07 23:12:11 +00:00
steps += ans_frac.simplify()
2013-08-07 23:12:11 +00:00
return steps
2014-01-17 17:10:38 +00:00
def __rsub__(self, other):
if other == 0:
return [self]
2014-05-28 16:39:08 +00:00
number = self.convert2fraction(other)
2014-05-28 16:49:19 +00:00
return number - self
2014-01-17 17:10:38 +00:00
def __neg__(self):
return Fraction(-self._num,self._denom).simplify()
2013-08-07 23:12:11 +00:00
def __mul__(self, other):
steps = []
if other == 0:
return [0]
elif other == 1:
return [self]
elif type(other) == int:
gcd1 = gcd(other, self._denom)
if gcd1 != 1:
num = [self._num, int(other/gcd1), "*", gcd1,"*"]
denom = [int(self._denom/gcd1), gcd1, "*"]
else:
num = [self._num, other, "*"]
denom = [self._denom]
steps.append(num + denom + ["/"])
num = int(self._num * other / gcd1)
denom = int(self._denom / gcd1)
else:
number = self.convert2fraction(other)
gcd1 = gcd(self._num, number._denom)
if gcd1 != 1:
num1 = [int(self._num/ gcd1), gcd1, "*"]
denom2 = [int(number._denom/ gcd1), gcd1, "*"]
else:
num1 = [self._num]
denom2 = [number._denom]
gcd2 = gcd(self._denom, number._num)
if gcd2 != 1:
num2 = [int(number._num/ gcd2), gcd2, "*"]
denom1 = [int(self._denom/ gcd2), gcd2, "*"]
else:
num2 = [number._num]
denom1 = [self._denom]
steps.append(num1 + num2 + [ '*'] + denom1 + denom2 + ['*', '/'])
num = int(self._num * number._num / (gcd1 * gcd2))
denom = int(self._denom * number._denom / (gcd1 * gcd2))
2013-08-07 23:12:11 +00:00
ans_frac = Fraction(num, denom)
steps.append(ans_frac)
2013-08-07 23:12:11 +00:00
steps += ans_frac.simplify()
2013-08-07 23:12:11 +00:00
return steps
def __rmul__(self, other):
2014-06-27 09:01:44 +00:00
return self * other
2013-08-07 23:12:11 +00:00
def __truediv__(self, other):
if other == 0:
raise ZeroDivisionError("division by zero")
elif other == 1:
return [self]
2014-05-28 16:39:08 +00:00
number = self.convert2fraction(other)
steps = []
2013-08-07 23:12:11 +00:00
number = Fraction(number._denom, number._num)
steps.append([self, number, "*"])
2013-08-07 23:12:11 +00:00
steps += self * number
return steps
def __rtruediv__(self, other):
2014-05-28 16:39:08 +00:00
number = self.convert2fraction(other)
2014-05-28 16:49:19 +00:00
return number / self
2014-02-28 07:49:03 +00:00
def __abs__(self):
return Fraction(abs(self._num), abs(self._denom))
2014-02-21 17:02:34 +00:00
def __eq__(self, other):
""" == """
2014-05-28 16:39:08 +00:00
number = self.convert2fraction(other)
2014-02-21 17:02:34 +00:00
return self._num * number._denom == self._denom * number._num
def __lt__(self, other):
2014-02-21 17:02:34 +00:00
""" < """
2014-05-28 16:52:46 +00:00
return float(self) < float(other)
def __le__(self, other):
2014-02-21 17:02:34 +00:00
""" <= """
2014-05-28 16:52:46 +00:00
return float(self) <= float(other)
def __gt__(self, other):
""" > """
return float(self) > float(other)
def __ge__(self, other):
""" >= """
return float(self) >= float(other)
if __name__ == '__main__':
f = Fraction(1, 12)
g = Fraction(1, 12)
h = Fraction(1,-5)
t = Fraction(10,3)
2013-08-07 23:12:11 +00:00
print("---------")
2014-05-28 16:49:19 +00:00
print("1 + ", str(h))
for i in (1 + h):
2013-08-07 23:12:11 +00:00
print(i)
print("---------")
2014-05-28 16:49:19 +00:00
print(str(f) , "+", str(t))
for i in (f + t):
print(i)
print("---------")
print(str(f) , "+", str(g))
for i in (f + g):
print(i)
print("---------")
print(str(f) , "-", str(g))
for i in (f - g):
print(i)
print("---------")
print(str(f) , "*", str(g))
for i in (f * g):
print(i)
print("---------")
print(str(h) , "+", str(t))
for i in (h + t):
print(i)
print("---------")
print(str(h) , "-", str(t))
for i in (h - t):
print(i)
print("---------")
print(str(h) , "*", str(t))
for i in (h * t):
print(i)
print("---------")
print("-", str(h) )
for i in (-h):
print(i)
print("---------")
2014-05-28 16:49:19 +00:00
print(str(h) , "/", str(t))
for i in (h / t):
print(i)
print("---------")
print(str(h) , "+", str(0))
for i in (h + 0):
print(i)
print("---------")
print(str(h) , "*", str(1))
for i in (h * 1):
print(i)
print("---------")
print(str(h) , "*", str(0))
for i in (h * 0):
print(i)
print("---------")
print(str(h) , "*", str(4))
for i in (h * 4):
print(i)
2013-08-07 23:12:11 +00:00
#print(f.simplify())
# -----------------------------
# Reglages pour 'vim'
# vim:set autoindent expandtab tabstop=4 shiftwidth=4:
# cursor: 16 del