Add test_trou_oblong

Add coord_comp module to find point, curve and surface correspondences between 2 versions of magix3d (with a subsequent impact on ids)

Author: mpoudot

test_link/coord_comp.py

@@ -0,0 +1,151 @@
+import math
+import csv
+import itertools
+
+def ecrire_points(gm, fichier):
+    points = gm.getVertices()
+
+    with open(fichier + "_points.csv", "w", encoding="utf-8") as f:
+        # Écriture d'un en-tête
+        f.write("Nom_Point;X;Y;Z\n")
+        
+        # Parcours de tous les points
+        for nom_point in points:
+            coord = gm.getCoord(nom_point)
+            x = coord.getX()
+            y = coord.getY()
+            z = coord.getZ()
+            
+            # Écriture de la ligne correspondante
+            f.write(f"{nom_point};{x};{y};{z}\n")
+
+
+def distance(p1, p2):
+    """Calcule la distance euclidienne entre deux points 3D."""
+    return math.sqrt((p1[0] - p2[0])**2 +
+                    (p1[1] - p2[1])**2 +
+                    (p1[2] - p2[2])**2)
+
+
+def lire_points(fichier):
+    """Lit un fichier de points et retourne un dict {nom: (x, y, z)}"""
+    points = {}
+    with open(fichier + "_points.csv", newline='', encoding='utf-8') as f:
+        lecteur = csv.DictReader(f, delimiter=';')
+        for ligne in lecteur:
+            nom = ligne['Nom_Point']
+            x = float(ligne['X'])
+            y = float(ligne['Y'])
+            z = float(ligne['Z'])
+            points[nom] = (x, y, z)
+    return points
+
+
+def ecrire_courbes(gm, fichier):
+    with open(fichier + "_courbes.csv", "w", encoding="utf-8", newline='') as f:
+        writer = csv.writer(f, delimiter=';')
+        writer.writerow(["Nom_Courbe", "Points"])
+        
+        for curve_name in gm.getCurves():
+            # Récupération de la liste des points de la courbe
+            pts = gm.getInfos(curve_name, 1).vertices()
+            writer.writerow([curve_name] + pts)
+
+
+def ecrire_surfaces(gm, fichier):
+    """
+    Exporte les surfaces d'un objet gm dans un fichier texte.
+    Chaque ligne contient : Nom_Surface;Point1;Point2;Point3;...
+    """
+
+    ecrire_points(gm, fichier)
+    ecrire_courbes(gm, fichier)
+
+    with open(fichier + "_surfaces.csv", "w", encoding="utf-8", newline='') as f:
+        writer = csv.writer(f, delimiter=';')
+        writer.writerow(["Nom_Surface", "Points"])
+        
+        for surf_name in gm.getSurfaces():
+            # Récupération de la liste des points de la surface
+            pts = gm.getInfos(surf_name, 2).vertices()
+            writer.writerow([surf_name] + pts)
+
+
+def lire_courbes(fichier):
+    courbes = {}
+    with open(fichier + "_courbes.csv", newline='', encoding='utf-8') as f:
+        lecteur = csv.reader(f, delimiter=';')
+        next(lecteur)  # ignorer l'en-tête
+        for ligne in lecteur:
+            nom_courbe = ligne[0]
+            points = ligne[1:]
+            courbes[nom_courbe] = points
+    return courbes
+
+
+def lire_surfaces(fichier):
+    """
+    Lit un fichier de surfaces et retourne un dict :
+    {nom_surface: [liste_points]}
+    """
+    surfaces = {}
+    with open(fichier + "_surfaces.csv", newline='', encoding='utf-8') as f:
+        lecteur = csv.reader(f, delimiter=';')
+        next(lecteur)  # ignorer l'en-tête
+        for ligne in lecteur:
+            nom_surface = ligne[0]
+            points = ligne[1:]
+            surfaces[nom_surface] = points
+    return surfaces
+
+
+def ecrire_correspondances(fichier1, fichier2, epsilon=1e-8):
+    # Trouve les correspondances entre deux fichiers de points.
+    pts1 = lire_points(fichier1)
+    pts2 = lire_points(fichier2)
+    corr_pts = {}
+    for nom1, coord1 in pts1.items():
+        candidats = []
+        for nom2, coord2 in pts2.items():
+            if math.dist(coord1, coord2) <= epsilon:
+                candidats.append(nom2)
+        corr_pts[nom1] = candidats
+    print(corr_pts)
+
+    # Trouve les correspondances entre les courbes.
+    cvr1 = lire_courbes(fichier1)
+    cvr2 = lire_courbes(fichier2)
+
+    corr_crvs = []
+    for nom1, pts1 in cvr1.items():
+        # Liste de candidats pour chaque point
+        candidats_par_point = [corr_pts[p] for p in pts1]
+        # Test de toutes les combinaisons possibles
+        for combinaison in itertools.product(*candidats_par_point):
+            if set(combinaison) in [set(s) for s in cvr2.values()]:
+                # Récupérer le nom de la courbe correspondante dans cvr2
+                for nom2, pts2 in cvr2.items():
+                    if set(combinaison) == set(pts2):
+                        corr_crvs.append((nom1, nom2))
+                        break
+                break
+    print(corr_crvs)
+
+    # Trouve les correspondances entre les surfaces
+    surf1 = lire_surfaces(fichier1)
+    surf2 = lire_surfaces(fichier2)
+
+    corr_surfs = []
+    for nom1, pts1 in surf1.items():
+        # Liste de candidats pour chaque point
+        candidats_par_point = [corr_pts[p] for p in pts1]
+        # Test de toutes les combinaisons possibles
+        for combinaison in itertools.product(*candidats_par_point):
+            if set(combinaison) in [set(s) for s in surf2.values()]:
+                # Récupérer le nom de la surface correspondante dans surf2
+                for nom2, pts2 in surf2.items():
+                    if set(combinaison) == set(pts2):
+                        corr_surfs.append((nom1, nom2))
+                        break
+                break
+    print(corr_surfs)

test_link/test_extrusion.py

@@ -164,3 +164,57 @@ def test_makeRevol_groups():
     assert sorted(grm.getGeomCurves("PLAN", 1)) == ['Crb0004', 'Crb0005']
     assert sorted(grm.getGeomCurves("PZ0_PLAN", 1)) == ['Crb0006', 'Crb0007', 'Crb0008', 'Crb0009', 'Crb0010', 'Crb0011', 'Crb0012']
     assert grm.getGeomVertices("PLAN", 0) == []
+
+def test_trou_oblong(capfd):
+    ctx = Mgx3D.getStdContext()
+    ctx.clearSession() # Clean the session after the previous test
+    gm = ctx.getGeomManager()
+
+    # Changement d'unité de longueur
+    ctx.setLengthUnit(Mgx3D.Unit.micron)
+    # Création du sommet Pt0000
+    gm.newVertex (Mgx3D.Point(0, 0, 0))
+    # Création du sommet Pt0001
+    gm.newVertex (Mgx3D.Point(-30, 0, 0))
+    # Création du sommet Pt0002
+    gm.newVertex (Mgx3D.Point(-30, 415, 0))
+    # Création du sommet Pt0003
+    gm.newVertex (Mgx3D.Point(0, 415, 0))
+    # Création du sommet Pt0004
+    gm.newVertex (Mgx3D.Point(-15, 415, 0))
+    # Création du segment Crb0000
+    gm.newSegment("Pt0003", "Pt0000")
+    # Création du segment Crb0001
+    gm.newSegment("Pt0000", "Pt0001")
+    # Création du segment Crb0002
+    gm.newSegment("Pt0001", "Pt0002")
+    # Création de l'arc de cercle Crb0003
+    gm.newArcCircle("Pt0004", "Pt0003", "Pt0002", False)
+    # Création de la surface Surf0000
+    gm.newPlanarSurface (["Crb0000","Crb0001","Crb0002","Crb0003"], "")
+    # Modifie le groupe TROU
+    gm.addToGroup (["Surf0000"], 2, "TROU")
+    # Révolution de Surf0000
+    gm.makeRevol (["Surf0000"], Mgx3D.RotX(180), False)
+    # Extrusion de Surf0004
+    gm.makeExtrude (["Surf0004"], Mgx3D.Vector(0, 0, -700), False)
+    # Modifie le groupe TROU
+    gm.addToGroup (["Vol0001"], 3, "TROU")
+    # Copie de  Vol0000 Vol0001
+    gm.copy (["Vol0000","Vol0001"], False,"B")
+    # Rotation de Vol0003 Vol0002 suivant  [  [ 0, 0, 0] ,  [ 1, 0, 0] , 180] 
+    gm.rotate (["Vol0003","Vol0002"], Mgx3D.RotX(180))
+    # Translation de Vol0003 Vol0002 suivant  [ 0, 0, -700] 
+    gm.translate (["Vol0003","Vol0002"], Mgx3D.Vector(0, 0, -700))
+    # Collage entre Vol0000 Vol0001 Vol0002 Vol0003
+    gm.glue (["Vol0000","Vol0001","Vol0002","Vol0003"])
+    # Fusion Booléenne de  Vol0000 Vol0001 Vol0002 Vol0003
+    gm.fuse (["Vol0000","Vol0001","Vol0002","Vol0003"])
+
+    assert gm.getNbVertices() == 13
+    assert gm.getNbCurves() == 22
+    assert gm.getNbSurfaces() == 12
+    assert gm.getNbVolumes() == 1
+
+    out, err = capfd.readouterr()
+    assert len(err) == 0