Initial Commit

Co-Authored-By: Damien Robert <Damien.Olivier.Robert+git@gmail.com>
Co-Authored-By: Frederik Vercauteren <frederik.vercauteren@gmail.com>
Co-Authored-By: Jonathan Komada Eriksen <jonathan.eriksen97@gmail.com>
Co-Authored-By: Pierrick Dartois <pierrickdartois@icloud.com>
Co-Authored-By: Riccardo Invernizzi <nidadoni@gmail.com>
Co-Authored-By: Ryan Rueger <git@rueg.re>                                                                                                                                           [0.01s]
Co-Authored-By: Benjamin Wesolowski <benjamin@pasch.umpa.ens-lyon.fr>
Co-Authored-By: Arthur Herlédan Le Merdy <ahlm@riseup.net>
Co-Authored-By: Boris Fouotsa <tako.fouotsa@epfl.ch>

1 files changed, 28 insertions, 0 deletions

diff --git a/theta_lib/isogenies/tree.py b/theta_lib/isogenies/tree.py
new file mode 100644
index 0000000..a6e3da3
--- /dev/null
+++ b/theta_lib/isogenies/tree.py
@@ -0,0 +1,28 @@
+from sage.all import *
+
+class Tree:
+	def __init__(self,node):
+		self._node=node
+		self._edges=[]
+		self._children=[]
+
+	def add_child(self,child,edge):
+		self._children.append(child)
+		self._edges.append(edge)
+
+	def look_node(self,node):
+		if self._node==node:
+			return self
+		elif len(self._children)>0:
+			for child in self._children:
+				t_node=child.look_node(node)
+				if t_node!=None:
+					return t_node
+
+	def edge_product(self,L_factors,factor_node=ZZ(1)):
+		n=len(self._children)
+		L_prod=[(factor_node,self._node)]
+		for i in range(n):
+			L_prod+=self._children[i].edge_product(L_factors,factor_node*L_factors[self._edges[i]])
+		return L_prod
+