src/Tools/blocFissure/gmu/creeZoneDefautGeom.py

   1 # -*- coding: utf-8 -*-
   2
   3 import logging
   4 from .geomsmesh import geompy
   5 from .geomsmesh import geomPublish
   6 from .geomsmesh import geomPublishInFather
   7 from . import initLog
   8 from .prolongeVertices import prolongeVertices
   9
  10 # -----------------------------------------------------------------------------
  11 # --- zone de defaut, constructions geometrique avec CAO d'origine
  12
  13 def creeZoneDefautGeom(objetSain, shapeDefaut, origShapes, verticesShapes, dmoyen, lgExtrusion=50):
  14   """
  15   Construction CAO de la zone à remailler, quand on utilise la CAO d'origine,
  16   apres appel creeZoneDefautMaillage
  17   @param objetSain : la géometrie de l'objet initial
  18   @param shapeDefaut : objet géometrique représentant la fissure
  19   (selon les cas, un point central, ou une shape plus complexe,
  20   dont on ne garde que les vertices)
  21   @param origShapes : liste id subShapes
  22   @param verticesShapes : listes noeuds de bord
  23   @param dmoyen : longueur arete moyenne bord
  24   @lgExtrusion : distance d'extrusion de la face du defaut
  25   (ne vaut que pour des fissures courtes)
  26   @return (facesDefaut, centreDefaut, normalDefaut, extrusionDefaut)
  27   """
  28   logging.info("start")
  29
  30   trace = True
  31   faces = []
  32   curves = []
  33   cdgs = []
  34   projs = []
  35   normals = []
  36   extrusions = []
  37   partitions = []
  38   decoupes = []
  39
  40   for ishape, vertices in enumerate(verticesShapes):
  41     aShape = origShapes[ishape]
  42     [face] = geompy.SubShapes(objetSain, [aShape])
  43     faces.append(face)
  44     curve = geompy.MakePolyline(vertices, False)
  45     curves.append(curve)
  46     if trace:
  47       name="poly_%d"%aShape
  48       geomPublish(initLog.debug, curve, name)
  49     #
  50     cdg = geompy.MakeCDG(curve)
  51     cdgs.append(cdg)
  52     if trace:
  53       name="cdgpoly_%d"%aShape
  54       geomPublish(initLog.debug, cdg, name)
  55     #
  56     projCdg = geompy.MakeProjection(cdg, face)
  57     projs.append(projCdg)
  58     if trace:
  59       name="projCdg_%d"%aShape
  60       geomPublish(initLog.debug, projCdg, name)
  61     #
  62     normal = geompy.GetNormal(face, projCdg)
  63     normals.append(normal)
  64     if trace:
  65       name="normal_%d"%aShape
  66       geomPublish(initLog.debug, normal, name)
  67     #
  68     extrusion = geompy.MakePrismVecH2Ways(curve, normal, 10)
  69     extrusions.append(extrusion)
  70     if trace:
  71       name="extrusion_%d"%aShape
  72       geomPublish(initLog.debug, extrusion, name)
  73     #
  74     verticesProlongees = prolongeVertices(vertices)
  75     #
  76     curveprol = geompy.MakePolyline(verticesProlongees, False)
  77     if trace:
  78       name="polyProl_%d"%aShape
  79       geomPublish(initLog.debug, curveprol, name)
  80     #
  81     extruprol = geompy.MakePrismVecH2Ways(curveprol, normal, 10)
  82     if trace:
  83       name="extruProl_%d"%aShape
  84       geomPublish(initLog.debug, extruprol, name)
  85     #
  86     partition = geompy.MakePartition([face], [extruprol], [], [], geompy.ShapeType["FACE"], 0, [], 0)
  87     partitions.append(partition)
  88     if trace:
  89       name="partition_%d"%aShape
  90       geomPublish(initLog.debug, partition, name)
  91     pass
  92   #
  93
  94   centreSphere = geompy.MakeCDG(shapeDefaut)
  95   geomPublish(initLog.debug, centreSphere, "cdg_defaut")
  96   ccurves = geompy.MakeCompound(curves)
  97   gravCenter = geompy.MakeCDG(ccurves)
  98   geomPublish(initLog.debug, gravCenter, "cdg_curves")
  99   for i in range(len(partitions)):
 100     if trace:
 101       logging.debug(" --- original shape %s", origShapes[i])
 102     dists = []
 103     facesToSort = []
 104     subFaces = geompy.ExtractShapes(partitions[i], geompy.ShapeType["FACE"], True)
 105     for aFace in subFaces:
 106       cdg = geompy.MakeCDG(aFace)
 107       distance = geompy.MinDistance(cdg, centreSphere)
 108       dists.append(distance)
 109       facesToSort.append(aFace)
 110       if trace:
 111         logging.debug("distance = %s", distance)
 112         pass
 113       pass
 114     if len(dists) > 0:
 115       minDist = min(dists)
 116       for j,d in enumerate(dists):
 117         if d == minDist:
 118           aFace = facesToSort[j]
 119           name="decoupe_%d"%origShapes[i]
 120           geomPublish(initLog.debug, aFace, name)
 121           decoupes.append(aFace)
 122           break
 123         pass
 124       pass
 125
 126   facesDefaut = decoupes[0]
 127   if len(decoupes) > 1:
 128     facesDefaut = geompy.MakePartition(decoupes, [], [], [], geompy.ShapeType["FACE"], 0, [], 0)
 129   geomPublish(initLog.debug, facesDefaut, "facesDefaut")
 130
 131   shells=[]
 132   if len(decoupes) > 1: # plusieurs faces de defaut
 133     subFaces = geompy.ExtractShapes(facesDefaut, geompy.ShapeType["FACE"], True)
 134     # --- regroupe les subFaces en shells connectes
 135     theFaces = list(subFaces) # copy
 136     while len(theFaces) > 0:
 137       logging.debug("------- len(theFaces) %s" , len(theFaces))
 138       theFace = theFaces[0]
 139       logging.debug("  start with face %s",theFaces[0])
 140       theFaces[0:1] = []
 141       aShell = [theFace]
 142       toAdd =[theFace]
 143       while len(toAdd) > 0:
 144         toAdd = []
 145         toRemove = []
 146         for i in range(len(theFaces)):
 147           logging.debug("  try %s", theFaces[i])
 148           for aFace in aShell:
 149             logging.debug("    with %s", aFace)
 150             try:
 151               edge = geompy.GetSharedShapesMulti([aFace, theFaces[i]], geompy.ShapeType["EDGE"])
 152               edgeShared = True
 153             except:
 154               edgeShared = False
 155             if edgeShared:
 156               if theFaces[i] not in toAdd:
 157                 toAdd.append(theFaces[i])
 158                 toRemove.append(i)
 159                 logging.debug("    --- add %s", theFaces[i])
 160         aShell += toAdd
 161         for k in sorted(toRemove, reverse=True):
 162           theFaces[k:k+1] = []
 163       theShell = geompy.MakeShell(aShell)
 164       name = "theShell%d"%len(shells)
 165       geomPublish(initLog.debug, theShell,name)
 166       shells.append(theShell)
 167     #
 168     distances = []
 169     for aShell in shells: # --- trouver le shell en contact avec la fissure
 170       distances.append(geompy.MinDistance(aShell, shapeDefaut))
 171     minDist = min(distances)
 172     for index in range(len(distances)):
 173       if distances[index] == minDist:
 174         break
 175     theShellDefaut = shells[index]
 176     #
 177   else: # --- une seule face de defaut
 178     subFaces = [facesDefaut]
 179     theShellDefaut = geompy.MakeShell(subFaces)
 180   if trace:
 181     geomPublish(initLog.debug, theShellDefaut,"theShellDefaut")
 182
 183   theFaces = geompy.ExtractShapes(theShellDefaut, geompy.ShapeType["FACE"], True)
 184   distances = []
 185   for aFace in theFaces:
 186     distances.append(geompy.MinDistance(aFace, centreSphere))
 187   minDist = min(distances)
 188   for index in range(len(distances)):
 189     if distances[index] == minDist:
 190       break
 191
 192   centreDefaut = geompy.MakeProjection(centreSphere, theFaces[index])
 193   if trace:
 194     geomPublish(initLog.debug, centreDefaut, "centreDefaut")
 195   normalDefaut = geompy.GetNormal(subFaces[index], centreDefaut)
 196   if trace:
 197     geomPublish(initLog.debug, normalDefaut, "normalDefaut")
 198   extrusionDefaut = geompy.MakePrismVecH(theShellDefaut, normalDefaut, -lgExtrusion)
 199   info = geompy.ShapeInfo(extrusionDefaut)
 200   logging.debug("shape info %s", info)
 201   if (info['SOLID'] > 1) or (info['COMPOUND'] > 0) :
 202     solids = geompy.ExtractShapes(extrusionDefaut, geompy.ShapeType["SOLID"], True)
 203     solid0 = solids[0]
 204     for i in range(1,len(solids)):
 205       solid0 = geompy.MakeFuse(solid0, solids[i])
 206     extrusionDefaut = solid0
 207   if trace:
 208     geomPublish(initLog.debug, extrusionDefaut, "extrusionDefaut")
 209
 210   return facesDefaut, centreDefaut, normalDefaut, extrusionDefaut