def expand_term(self, term):
"""distributes a coefficient to all of the terms within a group within
parentheses."""
if self.phrase == '0':
return Function('0', variables = self.variables, constants\
= self.constants, asimp = self.asimp)
if True:
coefficient = Function(\
eqn_helper.get_coefficient(term), variables = self.variables,\
constants = self.constants, asimp = self.asimp)\
.simplify_parentheses()
core = Function(eqn_helper.get_core(term),\
variables = self.variables, asimp = self.asimp,\
constants = self.constants).simplify_parentheses()
#most of the expansion really happens within the Function's
#multiplication method
core *= coefficient
return core
else:
return term
def combine_like_terms(self):
"""Takes terms with the same cores and combines their coefficients."""
result = []
terms = self.terms.copy()
#False if no substitution has been made in the current loop
subbed = False
#makes everything a function so that it's easier to work with
for i in range(len(terms)):
terms[i] = Function(terms[i], variables = self.variables,\
constants = self.constants, asimp = self.asimp)
#trims away the first term each time it checks for like terms
while len(terms) > 0:
subbed = False
for j in range(len(result)):
#checks that the cores of the terms are the same
if len(terms) > 0 and eqn_helper.get_core(result[j].phrase)\
== eqn_helper.get_core(terms[0].phrase):
try:
if eqn_helper.get_core(result[j].phrase) != '':
result[j] = Function(\
str(float(eqn_helper.get_coefficient(terms[0].phrase))\
+ float(eqn_helper.get_coefficient(result[j].phrase)))\
+ '*' + eqn_helper.get_core(result[j].phrase),\
asimp = self.asimp, constants = self.constants)
else:
result[j] = Function(\
str(float(eqn_helper.get_coefficient(terms[0].phrase))\
+ float(eqn_helper.get_coefficient(result[j].phrase))), \
asimp = self.asimp, constants = self.constants)
del terms[0]
subbed = True
except ValueError:
if eqn_helper.get_core(result[j].phrase) != '':
result[j] = Function(\
'('+str(Function(eqn_helper.get_coefficient(terms[0].phrase))\
+ Function(eqn_helper.get_coefficient(result[j].phrase)))\
+ ')*' + eqn_helper.get_core(result[j].phrase), \
asimp = self.asimp, constants = self.constants)
del terms[0]
subbed = True
except TypeError:
result [j] = Function(\
str(complex(terms[0].evaluate())\
+ complex(result[j].evaluate())), asimp = False,\
constants = self.constants)
del terms[0]
subbed = True
if not subbed:
result.append(terms[0])
del terms[0]
subbed = False
phrase = ''
for i in result:
try:
if float(i.phrase) != 0:
phrase = phrase + i.get_phrase() + '+'
except ValueError:
phrase = phrase + i.get_phrase() + '+'
i = 0
while i < len(result)-1:
if Function(result[i].get_phrase()).get_variables() \
!= Function(result[i+1].get_phrase()).get_variables():
return Function('(' + phrase[:len(phrase)-1] + ')',\
variables = self.variables, asimp = self.asimp,
constants = self.constants)
i += 1
return Function(phrase[:len(phrase)-1], variables = self.variables,
asimp = self.asimp, constants = self.constants)
