"""
sage: attach("~/recherche/opahc/presentation/figures/figures/arbres_pw/script.sage")
sage: for n in range(1,5):
....:     for p in PackedWords(n):
....:         script(p)
"""

def script(pw):
    """
    sage: script(PackedWord([2,3,1]))
    """
    load('/home/hugo/recherche/opahc/perso/tree_latex.py')
    fichier = open("arbre"+str(Words()(pw))+".tex", 'w')
    l = str([tree_latex(t) for t in pw.packed_word_to_labelled_forest()])
    l = l[1:-1]
    l = l.replace("$[1]$", "")
    l = l.replace("$[", "$")
    l = l.replace("]$", "$")
    l = l.replace("}, {", "}\n{ ")

    # TODO essayer de faire ces savebox ici!!!...
    # l = l.replace("$[", "$")
    # l = l.replace("]$", "\\bigstar$")
    # l = l.replace("$o$", "")
    #return l

    fichier.write(l)
    fichier.close()