Adrien/LeGrandChien
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

optimisation fonction evaluate

Browse source
This commit is contained in:
Didictateur 2026-02-16 13:00:39 +01:00
parent 29b218835c
commit e3d19eedcf
1 changed files with 37 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
legrandchien.py
View file

@ -358,6 +358,7 @@ class Equation:
self.right = right self.right = right
self.value = value self.value = value
self.type = "Equation" self.type = "Equation"
self.compiled_function = None
def copy(self): def copy(self):
if self.op == Operation.CONST: if self.op == Operation.CONST:
@ -408,24 +409,44 @@ class Equation:
return self.value.partial(var_name) * Equation(Operation.COS, None, None, self.value.copy()) return self.value.partial(var_name) * Equation(Operation.COS, None, None, self.value.copy())
if self.op == Operation.COS: if self.op == Operation.COS:
return Const(-1) * self.value.partial(var_name) * Equation(Operation.SIN, None, None, self.value.copy()) return Const(-1) * self.value.partial(var_name) * Equation(Operation.SIN, None, None, self.value.copy())
def compile(self):
if self.compiled_function is not None:
return self.compiled_function
if self.op == Operation.CONST:
self.compiled_function = lambda dico : self.value.value
if self.op == Operation.VAR:
self.compiled_function = lambda dico : dico[self.value.name]
if self.op == Operation.ADD:
left_func = self.left.compile()
right_func = self.right.compile()
self.compiled_function = lambda dico : left_func(dico) + right_func(dico)
if self.op == Operation.SUB:
left_func = self.left.compile()
right_func = self.right.compile()
self.compiled_function = lambda dico : left_func(dico) - right_func(dico)
if self.op == Operation.MULT:
left_func = self.left.compile()
right_func = self.right.compile()
self.compiled_function = lambda dico : left_func(dico) * right_func(dico)
if self.op == Operation.DIV:
left_func = self.left.compile()
right_func = self.right.compile()
self.compiled_function = lambda dico : left_func(dico) / right_func(dico)
if self.op == Operation.SIN:
value_func = self.value.compile()
self.compiled_function = lambda dico : np.sin(value_func(dico))
if self.op == Operation.COS:
value_func = self.value.compile()
self.compiled_function = lambda dico : np.cos(value_func(dico))
return self.compiled_function
def evaluate(self, dico): def evaluate(self, dico):
if self.op == Operation.CONST: if self.compiled_function is None:
return self.value.value self.compile()
if self.op == Operation.VAR: return self.compiled_function(dico)
return dico[self.value.name]
if self.op == Operation.ADD:
return self.left.evaluate(dico) + self.right.evaluate(dico)
if self.op == Operation.SUB:
return self.left.evaluate(dico) - self.right.evaluate(dico)
if self.op == Operation.MULT:
return self.left.evaluate(dico) * self.right.evaluate(dico)
if self.op == Operation.DIV:
return self.left.evaluate(dico) / self.right.evaluate(dico)
if self.op == Operation.SIN:
return np.sin(self.value.evaluate(dico))
if self.op == Operation.COS:
return np.cos(self.value.evaluate(dico))
def solve(self, dico, tmax=10, dt=0.01): def solve(self, dico, tmax=10, dt=0.01):
self.simplify() self.simplify()