103 lines
3.7 KiB
Python
103 lines
3.7 KiB
Python
#
|
|
# Sam Hadow - Huffman-py
|
|
# Copyright (C) 2023
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
# -*- encoding: utf-8 -*-
|
|
from huffman_py.Sommets import *
|
|
|
|
class Arbre(object):
|
|
def __init__(self, racine = None):
|
|
# racine de l'arbre (un objet Sommet)
|
|
self.racine = racine
|
|
|
|
# les sommets dans l'arbre (une liste d'objets Sommet)
|
|
self.sommets = []
|
|
self.gen_arbre()
|
|
|
|
self.set_id()
|
|
|
|
|
|
# getters et setters des attributs avec _attribut pour stocker l'attribut dans un espace non public
|
|
# on utilisera obligatoirement ces méthodes pour manipuler les attributs
|
|
|
|
def get_racine(self):
|
|
return self.__racine
|
|
def set_racine(self, valeur):
|
|
self.__racine = valeur
|
|
racine = property(fget=get_racine, fset=set_racine, doc="racine de l'arbre")
|
|
|
|
def get_sommets(self):
|
|
return self.__sommets
|
|
def set_sommets(self, valeur):
|
|
self.__sommets = valeur
|
|
sommets = property(fget=get_sommets, fset=set_sommets, doc="liste des sommets dans l'arbre")
|
|
|
|
def gen_arbre(self):
|
|
# genère la liste des sommets d'un arbre à partir de la racine de celui ci
|
|
L = list()
|
|
L.append(self.racine)
|
|
while len(L) > 0:
|
|
traitement = L.pop(0)
|
|
self.sommets.append(traitement)
|
|
if traitement.left:
|
|
L.append(traitement.left)
|
|
if traitement.right:
|
|
L.append(traitement.right)
|
|
|
|
def set_id(self):
|
|
# attribue des identifiants aux sommets d'un arbre
|
|
for i,elem in enumerate(self.sommets):
|
|
elem.identifiant = i
|
|
|
|
def __isub__(self, noeud):
|
|
# suppression d'un sommet en surchargeant un opérateur
|
|
if self.recherche(noeud) != None:
|
|
self.sommets.remove(noeud)
|
|
# on supprime également les fils
|
|
if noeud.right != None:
|
|
self.__isub__(noeud.right)
|
|
if noeud.left != None:
|
|
self.__isub__(noeud.left)
|
|
return self
|
|
|
|
def __iadd__(self, arbre2):
|
|
# fusion d'arbres
|
|
# on crée une nouvelle racined
|
|
occurrence = self.racine.occurrence
|
|
ancienne_racine = self.racine
|
|
if occurrence > arbre2.racine.occurrence:
|
|
self.racine = Sommets(occurrence, '', left = arbre2.racine, right = ancienne_racine)
|
|
else:
|
|
self.racine = Sommets(occurrence, '', left = ancienne_racine, right = arbre2.racine)
|
|
|
|
# on ajoute la nouvelle racine à notre liste de sommets
|
|
self.sommets.append(self.racine)
|
|
|
|
# on ajoute les sommets de arbre2
|
|
self.sommets += arbre2.sommets
|
|
|
|
# on crée les id
|
|
self.set_id()
|
|
|
|
return self
|
|
|
|
def recherche(self, noeud):
|
|
for sommet in self.sommets:
|
|
if sommet == noeud:
|
|
return sommet
|
|
return None
|