def carve_tetrahedron(curTetId, faceIDsInCurTet):
#remove the known TMP face of this tetrahedron and locally keep the adjacent FIX tetrahedron
for i in faceIDsInCurTet:
if ModelData.dictFaces[i].get_tag() == ClassFace.TMP:
#on the shell
ModelDataFuncs.pre_remove_face_by_faceids([i])
#update the list of shell faces (TMP face must be on the shell)
ModelData.listShellFaceIDs.remove(i)
else:
#deal with adjacent fixed tetrahedron
if i not in ModelData.listShellFaceIDs:
ModelData.listShellFaceIDs.append(i)
tetList = TetraFunc.find_tetids_by_faceid(i)
for tetId in tetList:
if tetId != curTetId:
#keep the tet with fixed or keep shell face
keep_tetrahedron(tetId, TetraFunc.get_faceids_from_tetid(tetId))
else:
print 'weird!!'
if len(faceIDsInCurTet) < 4:
#add the undefined dictFaces as TMP dictFaces
vertexList = ModelData.dictTetrahedrons[curTetId].get_vids()
faceList = []
faceList.append((vertexList[0], vertexList[1], vertexList[2]))
faceList.append((vertexList[0], vertexList[2], vertexList[3]))
faceList.append((vertexList[0], vertexList[3], vertexList[1]))
faceList.append((vertexList[1], vertexList[2], vertexList[3]))
for f in faceList:
isDefined = False
for i in faceIDsInCurTet:
if ModelData.dictFaces[i].is_equal_geometry(ClassFace.Class_face(f)):
isDefined = True
break
if not isDefined :
#add the tmp face and update the list of shell faces
ModelData.listShellFaceIDs.append(ModelDataFuncs.add_face(ClassFace.Class_face(f)))
#remove the tetrahedron
ModelData.dictTetrahedrons.pop(curTetId)
def heuristic_tet_carving():
print("----Heuristic shrinking-wrapping----")
#keep all the tetrahedra that have fix facets on the shell
#in update_stack() this step is not needed because it has already been handled in carving
for fId in ModelData.listShellFaceIDs:
if ModelData.dictFaces[fId].get_tag() == ClassFace.FIX:
tetId = TetraFunc.find_tetids_by_faceid(fId, 1)
keep_tetrahedron(tetId[0], TetraFunc.get_faceids_from_tetid(tetId[0]))
#count
numCount = 0
#initiate Stack and lists
sortedTetIDStack = update_stack()
#heruistic carving
while len(sortedTetIDStack) > 0:
#pop the first not fix element of stack if possible
curTet = sortedTetIDStack.pop(0)
curTetId = curTet[0]
print curTetId
#fetch all the shellFaces (indices) from a given Tet
shellFaceIDsInCurTet = curTet[1]
#fetch all the dictFaces (indices) from a given Tet
faceIDsInCurTet = curTet[2]
#classify the tetrahedron
numTMP = len(shellFaceIDsInCurTet)#equal to the no. of faces on the shell
numFIX = 0
for fid in faceIDsInCurTet:
if ModelData.dictFaces[fid].get_tag() == ClassFace.FIX:
numFIX += 1
numKEEP = len(faceIDsInCurTet) - numTMP - numFIX#keep face is not constrainted as fixed ones
numPRESERV = numFIX + numKEEP
#test by a set of constraints
bStop = False
bPostponed = False
carveTetList = [] #for infinite tetehedra
keepFaceList = [] #for coplanar bounded facets
unboundFaceList = [] #for coplanar unbouned facets
# new code
#if numFIX == 0 and numTMP == 1:
# #constraint3
# if constraintsCtrl['Constraint3'] and not bStop:
# bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[0])
#elif numFIX == 1 and numTMP == 3:
# #constraint5
# if constraintsCtrl['Constraint5'] and not bStop:
# bStop = constraint_5(curTetId, faceIDsInCurTet)
# #check all faces with constraint 3 not needed in theory (flat tetrahedra)
# if constraintsCtrl['Constraint3'] and not bStop:
# bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[0])
# if constraintsCtrl['Constraint3'] and not bStop:
# bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[1])
# if constraintsCtrl['Constraint3'] and not bStop:
# bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[2])
#elif numFIX in (1, 2) and numTMP == 2:
# #c1 5
# if constraintsCtrl['Constraint5'] and not bStop:
# bStop = constraint_5(curTetId, faceIDsInCurTet)
# if constraintsCtrl['Constraint1'] and not bStop:
# #extract the opposite edge
# eList = []
# for i in shellFaceIDsInCurTet:
# for j in range(0, 3):
# if ModelData.dictFaces[i].get_vids()[j] not in eList:
# eList.append(ModelData.dictFaces[i].get_vids()[j])
# else:
# eList.remove(ModelData.dictFaces[i].get_vids()[j])
#
# if not bStop:
# bStop, bPostponed, carveTetList = constraint_1(curTetId, eList)
#
# #check all faces with constraint 3 not needed in theory (flat tetrahedra)
# if constraintsCtrl['Constraint3'] and not bStop:
# bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[0])
# if constraintsCtrl['Constraint3'] and not bStop:
# bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[1])
#elif numFIX in (1, 2, 3) and numTMP == 1:
# #c1 2 3 5
# #constraint5
# if constraintsCtrl['Constraint5'] and not bStop:
# bStop = constraint_5(curTetId, faceIDsInCurTet)
#
# #constraint3
# if constraintsCtrl['Constraint3'] and not bStop:
# bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[0])
# if (constraintsCtrl['Constraint2'] or constraintsCtrl['Constraint1']) and not bStop:
# #extract the candidate face, the opposite edge list (three) the and opposite vert
# tmpFace = shellFaceIDsInCurTet[0]
# #the opposite vertex
# oppVert = -1
# #the opposite edge list
# eList = []
# for v in ModelData.dictTetrahedrons[curTetId].get_vids():
# if v not in ModelData.dictFaces[tmpFace].get_vids():
# oppVert = v
# if numFIX == 1:
# #only two edges
# fixFace = list(set(faceIDsInCurTet) - set(shellFaceIDsInCurTet))
# for v_1 in ModelData.dictFaces[fixFace[0]].get_vids():
# if v_1 != oppVert:
# eList.append((v, v_1))
# else:
# #three edges
# for n in ModelData.dictFaces[tmpFace].get_vids():
# eList.append((v, n))
# break
# #check with constraints 1 2
# #2
# if constraintsCtrl['Constraint2'] and not bStop:
# bStop, bPostponed, carveTetList = constraint_2(curTetId, oppVert)
# #1
# if numFIX == 1:
# #two edges
# if constraintsCtrl['Constraint1'] and not bStop:
# bStop, bPostponed, carveTetList = constraint_1(curTetId, eList[0])
# if constraintsCtrl['Constraint1'] and not bStop:
# bStop, bPostponed, carveTetList = constraint_1(curTetId, eList[1])
# else:
# #three edges
# if constraintsCtrl['Constraint1'] and not bStop:
# bStop, bPostponed, carveTetList = constraint_1(curTetId, eList[0])
# if constraintsCtrl['Constraint1'] and not bStop:
# bStop, bPostponed, carveTetList = constraint_1(curTetId, eList[1])
# if constraintsCtrl['Constraint1'] and not bStop:
# bStop, bPostponed, carveTetList = constraint_1(curTetId, eList[2])
#elif numFIX == 0:
# #check all faces with constraint 3 not needed in theory (flat tetrahedra)
# if constraintsCtrl['Constraint3'] and not bStop:
# for fID in shellFaceIDsInCurTet:
# bStop, keepFaceList, unboundFaceList = constraint_3(fID)
#old code
#every carving step is constrainted, esp. those without fixed faces
if numTMP == 4:
print('Isolated tetrahedron')
pass
#check all faces with constraint 3 not needed in theory
if constraintsCtrl['Constraint3'] and not bStop:
bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[0])
if constraintsCtrl['Constraint3'] and not bStop:
bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[1])
if constraintsCtrl['Constraint3'] and not bStop:
bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[2])
if constraintsCtrl['Constraint3'] and not bStop:
bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[3])
elif numTMP == 3:
#check with constraints 5
if numFIX == 1:
if constraintsCtrl['Constraint5'] and not bStop:
bStop = constraint_5(curTetId, faceIDsInCurTet)
#check all faces with constraint 3 not needed in theory
if constraintsCtrl['Constraint3'] and not bStop:
bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[0])
if constraintsCtrl['Constraint3'] and not bStop:
bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[1])
if constraintsCtrl['Constraint3'] and not bStop:
bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[2])
elif numTMP == 2:
if constraintsCtrl['Constraint1']:
#extract the opposite edge
eList = []
for i in shellFaceIDsInCurTet:
for j in range(0, 3):
if ModelData.dictFaces[i].get_vids()[j] not in eList:
eList.append(ModelData.dictFaces[i].get_vids()[j])
else:
eList.remove(ModelData.dictFaces[i].get_vids()[j])
#1 different with new code
if constraintsCtrl['Constraint1'] and not bStop:
bStop, bPostponed, carveTetList = constraint_1(curTetId, eList)
#5
if numFIX != 0:
if constraintsCtrl['Constraint5'] and not bStop:
bStop = constraint_5(curTetId, faceIDsInCurTet)
#check both faces with constraint 3 not needed in theory
if constraintsCtrl['Constraint3'] and not bStop:
bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[0])
if constraintsCtrl['Constraint3'] and not bStop:
bStop, keepFaceList, unboundFaceList = constraint_3(shellFaceIDsInCurTet[1])
elif numTMP == 1:
if constraintsCtrl['Constraint2'] or constraintsCtrl['Constraint1']:
#extract the candidate face, the opposite edge list (three) the and opposite vert
tmpFace = shellFaceIDsInCurTet[0]
#the opposite vertex
oppVert = -1
#the opposite edge list
eList = []
for v in ModelData.dictTetrahedrons[curTetId].get_vids():
if v not in ModelData.dictFaces[tmpFace].get_vids():
oppVert = v
for n in ModelData.dictFaces[tmpFace].get_vids():
eList.append((v, n))
break
#check with constraints 1 2 3 5
#2 different with new code
if constraintsCtrl['Constraint2'] and not bStop:
bStop, bPostponed, carveTetList = constraint_2(curTetId, oppVert)
#1 different with new code
if constraintsCtrl['Constraint1'] and not bStop:
bStop, bPostponed, carveTetList = constraint_1(curTetId, eList[0])
if constraintsCtrl['Constraint1'] and not bStop:
bStop, bPostponed, carveTetList = constraint_1(curTetId, eList[1])
if constraintsCtrl['Constraint1'] and not bStop:
bStop, bPostponed, carveTetList = constraint_1(curTetId, eList[2])
if numFIX != 0 and constraintsCtrl['Constraint5'] and not bStop:
#5
bStop = constraint_5(curTetId, faceIDsInCurTet)
#3
if numFIX != 3 and constraintsCtrl['Constraint3'] and not bStop:
bStop, keepFaceList, unboundFaceList = constraint_3(tmpFace)
else:
print("Isolated Tetrahedron!", numTMP, numKNOWN)
#raise Exception
#manipulate the tetrahedron
numTet = len(ModelData.dictTetrahedrons)
if bStop:
#keep
if len(keepFaceList) != 0:
#keep a bunch of "coplanar" tetrehedra
for faceId in keepFaceList:
tetId = TetraFunc.find_tetids_by_faceid(faceId, 1)
if tetId != []:
keep_tetrahedron(tetId[0], TetraFunc.get_faceids_from_tetid(tetId[0]))
#carve a bunch of "coplanar" unbouned tetrehedra
if len(unboundFaceList) != 0:
for faceId in unboundFaceList:
tetId = TetraFunc.find_tetids_by_faceid(faceId, 1)
if tetId != []:
carve_tetrahedron(tetId[0], TetraFunc.get_faceids_from_tetid(tetId[0]))
else:
#keep
keep_tetrahedron(curTetId, faceIDsInCurTet)
else:
#carve or postpone
if not bPostponed:
#Carve
bIsCarved = True
carve_tetrahedron(curTetId, faceIDsInCurTet)
else:
#postpone
if ModelData.dictTetrahedrons[curTetId].get_tag() == ClassFace.INF:
#all the tetrehedra are infinite (INF)
for tetId in carveTetList:
#carve a bunch of tetrahedra
carve_tetrahedron(tetId, TetraFunc.get_faceids_from_tetid(tetId))
else:
#postphone the tetrahedron
ModelData.dictTetrahedrons[curTetId].set_tag(ClassFace.INF)
#Update the dictFaces
ModelDataFuncs.remove_faces()
#Update the stack and lists
sortedTetIDStack = update_stack()
#Output intermediate results
if globalOutputCarve and numTet > len(ModelData.dictTetrahedrons) and len(ModelData.dictTetrahedrons) % globalOutputFreq == 0:
ModelDataFuncs.writer_obj(ModelData.strInputFileName + '_' + str(len(ModelData.dictTetrahedrons)) + "_CARVE_.obj")
def update_stack():
#init
sortedTetIDStack = []
#building a list with all the dictTetrahedrons attached with TMP dictFaces and their volume
tmpTetIDs = {}
for fkey in ModelData.listShellFaceIDs:
#all the facets of a fixed tet are already fix
if ModelData.dictFaces[fkey].get_tag() == ClassFace.TMP:
#if (375 in ModelData.dictFaces[fkey].get_vids() or 382 in ModelData.dictFaces[fkey].get_vids()) and 371 in ModelData.dictFaces[fkey].get_vids() and 685 in ModelData.dictFaces[fkey].get_vids():
# aaa = 9
tetList = TetraFunc.find_tetids_by_faceid(fkey, 1)
tetid = tetList[0]
tetShellFaceidList = TetraFunc.get_shell_faceids_from_tetid(tetid)
tetFaceidList = TetraFunc.get_faceids_from_tetid(tetid)#all the known face of this tet
#degree of freedom
DoF = 4 - len(tetFaceidList)
#Carving differential
CDif = 0
for f in tetFaceidList:
if ModelData.dictFaces[f].get_tag() == ClassFace.TMP:
CDif = CDif + 1
CDif = CDif - DoF
#different configurations of heuristics
#no heuristics (speedup)
if not globalHeuristic['validity']:
sortedTetIDStack.append((tetid,tetShellFaceidList, tetFaceidList))
return sortedTetIDStack
#with heuristics (normal)
if not tmpTetIDs.has_key(tetid):
if globalHeuristic['volume']:
tmpTetIDs[tetid] = (tetShellFaceidList, tetFaceidList, ModelData.dictTetrahedrons[tetid].get_tag(), DoF, CDif, ModelData.dictTetrahedrons[tetid].get_volume()) #volume
elif globalHeuristic['flatness']:
tmpTetIDs[tetid] = (tetShellFaceidList, tetFaceidList, ModelData.dictTetrahedrons[tetid].get_tag(), DoF, CDif, ModelData.dictTetrahedrons[tetid].get_depth(ModelData.dictFaces[fkey].get_vids())/ModelData.dictFaces[fkey].get_area())#flatness
elif globalHeuristic['depth']:
tmpTetIDs[tetid] = (tetShellFaceidList, tetFaceidList, ModelData.dictTetrahedrons[tetid].get_tag(), DoF, CDif, ModelData.dictTetrahedrons[tetid].get_depth(ModelData.dictFaces[fkey].get_vids()))#depth
elif globalHeuristic['distanceToCenter']:
#face center to the model center
vids = ModelData.dictFaces[fkey].get_vids()
midPt = SimpleMath.tuple_plus(ModelData.dictVertices[vids[0]], ModelData.dictVertices[vids[1]])
midPt = SimpleMath.tuple_plus(ModelData.dictVertices[vids[2]], midPt)
midPt = SimpleMath.tuple_numproduct(1.0/3.0, midPt)
tmpTetIDs[tetid] = (tetShellFaceidList, tetFaceidList, ModelData.dictTetrahedrons[tetid].get_tag(), DoF, CDif, 1.0/SimpleMath.square_dist_point_to_point(midPt, [0.0, 0.0, 0.0]))#reciprocal distanceToCenter
#tmpTetIDs[tetid] = (tetShellFaceidList, tetFaceidList, ModelData.dictTetrahedrons[tetid].get_tag(), DoF, CDif, ModelData.dictFaces[fkey].get_area())#face area
else:
tmpTetIDs[tetid] = (tetShellFaceidList, tetFaceidList, ModelData.dictTetrahedrons[tetid].get_tag(), DoF, CDif)#simplest
else:
#update the exisiting information
if globalHeuristic['volume']:
volume = ModelData.dictTetrahedrons[tetid].get_volume()
if DoF < tmpTetIDs[tetid][3] or CDif > tmpTetIDs[tetid][4] or volume > tmpTetIDs[tetid][5]:
tmpTetIDs[tetid] = (tetShellFaceidList, tetFaceidList, ModelData.dictTetrahedrons[tetid].get_tag(), DoF, CDif, 1.0/volume) #reciprocal volume
elif globalHeuristic['flatness']:
flatness = ModelData.dictTetrahedrons[tetid].get_depth(ModelData.dictFaces[fkey].get_vids())/ModelData.dictFaces[fkey].get_area()
if DoF < tmpTetIDs[tetid][3] or CDif > tmpTetIDs[tetid][4] or flatness < tmpTetIDs[tetid][5]:
tmpTetIDs[tetid] = (tetShellFaceidList, tetFaceidList, ModelData.dictTetrahedrons[tetid].get_tag(), DoF, CDif, flatness)#flatness
elif globalHeuristic['depth']:
depth = ModelData.dictTetrahedrons[tetid].get_depth(ModelData.dictFaces[fkey].get_vids())
if DoF < tmpTetIDs[tetid][3] or CDif > tmpTetIDs[tetid][4] or depth < tmpTetIDs[tetid][5]:
tmpTetIDs[tetid] = (tetShellFaceidList, tetFaceidList, ModelData.dictTetrahedrons[tetid].get_tag(), DoF, CDif, depth)#depth
elif globalHeuristic['distanceToCenter']:
#face center to the model center
vids = ModelData.dictFaces[fkey].get_vids()
midPt = SimpleMath.tuple_plus(ModelData.dictVertices[vids[0]], ModelData.dictVertices[vids[1]])
midPt = SimpleMath.tuple_plus(ModelData.dictVertices[vids[2]], midPt)
midPt = SimpleMath.tuple_numproduct(1.0/3.0, midPt)
dist = 1.0/SimpleMath.square_dist_point_to_point(midPt, [0.0, 0.0, 0.0])
if DoF < tmpTetIDs[tetid][3] or CDif > tmpTetIDs[tetid][4] or dist > tmpTetIDs[tetid][5]:
tmpTetIDs[tetid] = (tetShellFaceidList, tetFaceidList, ModelData.dictTetrahedrons[tetid].get_tag(), DoF, CDif, dist)#reciprocal distanceToCenter
#sort the dict by the value (volume) and number of neigbour dictFaces into a desascending order
if not globalHeuristic['volume'] and not globalHeuristic['flatness'] and not globalHeuristic['depth'] and not globalHeuristic['distanceToCenter']:
for key in tmpTetIDs:
sortedTetIDStack.append((key, tmpTetIDs[key][0], tmpTetIDs[key][1], tmpTetIDs[key][2], tmpTetIDs[key][3], tmpTetIDs[key][4]))
else:
for key in tmpTetIDs:
sortedTetIDStack.append((key, tmpTetIDs[key][0], tmpTetIDs[key][1], tmpTetIDs[key][2], tmpTetIDs[key][3], tmpTetIDs[key][4], tmpTetIDs[key][5]))
#sort firstly by fix/tmp, secondly by DoF, thirdly by Cd, lastly by volume
sortedTetIDStack = sorted(sortedTetIDStack, key=lambda tet: tet[6], reverse = False)#smaller ratio first
#sortedTetIDStack = sorted(sortedTetIDStack, key=lambda tet: tet[5], reverse = True)#large CDif first
sortedTetIDStack = sorted(sortedTetIDStack, key=lambda tet: tet[4], reverse = False)#smaller DoF first
sortedTetIDStack = sorted(sortedTetIDStack, key=lambda tet: tet[5], reverse = True)#large CDif first
sortedTetIDStack = sorted(sortedTetIDStack, key=lambda tet: tet[3], reverse = True)#TMP tetrahedron first, then INF
str_out = ("number of candidate tetrahedron: {}").format(len(sortedTetIDStack))
os.sys.stdout.write(str_out + "\r")
os.sys.stdout.flush()
return sortedTetIDStack
