From 0115b1856623a3e06bd341c9bef3dcd1ba0c7b10 Mon Sep 17 00:00:00 2001
From: Yann Salmon <nomail@yannsalmon.fr>
Date: Tue, 22 Feb 2022 14:18:10 +0100
Subject: [PATCH] =?UTF-8?q?code=20plus=20agr=C3=A9able?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 latex/conversion.py | 101 +++++++++++++++++++++-----------------------
 1 file changed, 48 insertions(+), 53 deletions(-)

diff --git a/latex/conversion.py b/latex/conversion.py
index 75ba20b..feba10e 100644
--- a/latex/conversion.py
+++ b/latex/conversion.py
@@ -1,10 +1,23 @@
 import os, shutil
 
-def extraire_caml(nf_tex, mlref, nf_texo, tag) :
+def nomFonctionCaml(ligne) :
+    lstrip = ligne.strip()
+    if lstrip.startswith("let") :
+        nomfonc = lstrip[4:]
+        if nomfonc.startswith("rec ") :
+            nomfonc = nomfonc[4:]
+        nomfonc = nomfonc.split()[0]
+        return nomfonc.split("=")[0]
+    else :
+        return None
+
+def extraireCaml(nf_tex, mlref, tag) :
     nf_tex = nf_tex.replace("file://", "")
+    nf_tex_bak = nf_tex + ".bak"
+    shutil.copy2(nf_tex, nf_tex_bak)
     nf_ml = os.path.join(os.path.dirname(nf_tex), mlref)
-    nf_texo = nf_texo.replace("file://", "")
-    output_tex = open(nf_texo, "w")
+    contenu = open(nf_tex).readlines()
+    output_tex = open(nf_tex, "w")
     output_ml = open(nf_ml, "w")
     incode = False
     nomfonc = ""
@@ -17,54 +30,36 @@ def extraire_caml(nf_tex, mlref, nf_texo, tag) :
         else :
             compteursNoms[s] += 1
             return "{}__{}".format(s, compteursNoms[s])
-    for l in open(nf_tex) :
-        lstrip = l.strip()
-        if lstrip.startswith(r"\begin{"+tag+"}") :
-            incode = True
-            nomfonc = ""
-            complement = None
-            if "[" in lstrip :
-                complement = lstrip.split("[")[1].split("]")[0]
-            continue
-        if incode :
-            if nomfonc == "" :
-                if lstrip == "" :
-                    continue
-                if "let" in lstrip :
-                    nomfonc = l[4:]
-                    if nomfonc.startswith("rec ") :
-                        nomfonc = nomfonc[4:]
-                    nomfonc = nomfonc.split()[0]
-                    nomfonc = setNom(nomfonc)
-                else :
-                    nomfonc = setNom("anon")
-                output_ml.write("(*** " + nomfonc + " ***)\n")
-            if lstrip == r"\end{"+tag+"}" :
-                output_ml.write("(*** " + nomfonc + "_fin ***)\n\n")
-                if complement is not None :
-                    output_tex.write(r"\inputcaml["+nomfonc+"]{"+ mlref + "}["+complement+"]\n")
-                else :
-                    output_tex.write(r"\inputcaml["+nomfonc+"]{"+ mlref + "}\n")
-                incode = False
-                continue
-            output_ml.write(l)
-        else :
-            output_tex.write(l)
-
-def caml(s, start=0) :
     try :
-        pos_let = s.index("\nlet", start)
-        fin = s.index(r";;", pos_let)
-        print(s[pos_let:fin+3])
-        if s[pos_let+4:].startswith(" rec") :
-            delta = 5
-        else :
-            delta = 0
-        pos_nom = s.index(" ", pos_let + delta) + 1
-        fin_nom = s.index(" ", pos_nom)
-        nom = s[pos_nom:fin_nom]
-        print(nom)
-        news = s[:pos_let] + "(*** " + nom + " ***)" + s[pos_let:fin+3] + "(*** " + nom + "_fin ***)\n" + s[fin+3:]
-        return caml(news, fin + 3 + len("(*** " + nom + " ***)")+ len("(*** " + nom + "_fin ***)\n"))
-    except ValueError :
-        return s
\ No newline at end of file
+        for l in contenu :
+            lstrip = l.strip()
+            if lstrip.startswith(r"\begin{"+tag+"}") :
+                incode = True
+                nomfonc = ""
+                complement = None
+                if "[" in lstrip :
+                    complement = lstrip.split("[")[1].split("]")[0]
+                continue
+            if incode :
+                if nomfonc == "" :
+                    if lstrip == "" :
+                        continue
+                    nomfonc = nomFonctionCaml(lstrip)
+                    if nomfonc is None :
+                        nomfonc = "anon"
+                    nomfonc = setNom(nomfonc)
+                    output_ml.write("(*** " + nomfonc + " ***)\n")
+                if lstrip == r"\end{"+tag+"}" :
+                    output_ml.write("(*** " + nomfonc + "_fin ***)\n\n")
+                    if complement is not None :
+                        output_tex.write(r"\inputcaml["+nomfonc+"]{"+ mlref + "}["+complement+"]\n")
+                    else :
+                        output_tex.write(r"\inputcaml["+nomfonc+"]{"+ mlref + "}\n")
+                    incode = False
+                    continue
+                output_ml.write(l)
+            else :
+                output_tex.write(l)
+    finally :
+        output_ml.close()
+        output_tex.close()
\ No newline at end of file