def keep_tetrahedron(curTetId, faceIDsInCurTet):
#use the tag of the tetrahedron
if ModelData.dictTetrahedrons[curTetId].get_tag == ClassFace.FIX:
return
#set the tag
ModelData.dictTetrahedrons[curTetId].set_tag(ClassFace.FIX)
#deal with all the known faces
for j in faceIDsInCurTet:
if ModelData.dictFaces[j].get_tag() != ClassFace.FIX:
ModelData.dictFaces[j].set_tag(ClassFace.KEEP)
#add the unkown faces and define them as KEEP (in order to seperate KEEP and FIX)
if len(faceIDsInCurTet) < 4:
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 :
tetNormal = []
for v in vertexList:
if v not in f:
#facing outwards
tetNormal = SimpleMath.tuple_minus(ModelData.dictVertices[f[0]], ModelData.dictVertices[v])
break
#do not have to consider semantics at this moment
fId = ModelDataFuncs.add_face(ClassFace.Class_face(ModelDataFuncs.create_orientated_vids(f, tetNormal), ClassFace.KEEP))
def deduce_semantics_of_face(fid):
#detect coplaner neighbours
vertexList = ModelData.dictFaces[fid].get_vids()
curNormal = SimpleMath.get_face_normal([ModelData.dictVertices[vertexList[0]], ModelData.dictVertices[vertexList[1]], ModelData.dictVertices[vertexList[2]]])
if SimpleMath.vector_length_3(curNormal) > SimpleMath.Tol:
#Get the normals of neighbors of this face
tag = [fid]
coplaner_neighbors(fid, curNormal, tag)
#statistic of dominant semantics within tagged faces
counterSemantics = Counter()
for fid in tag:
counterSemantics[(ModelData.dictFaces[fid].get_id(), ModelData.dictFaces[fid].get_type())] += 1
res = counterSemantics.most_common(2)
#get the semantics
strId = ''
strType = ''
if len(res) == 1 and res[0][0] == ('', ''):
#deduce the semantics based on direction
strId = ModelDataFuncs.generate_uuid()
angle = SimpleMath.dot_product_3(curNormal, (0.0, 0.0, 1.0))
if angle <= 1 and angle > 0.087:
#roof 0 to 85
strType = 'BUILDING_ROOF_SURFACE'
elif angle < -0.999:
#ground -0 to -1
strType = 'BUILDING_GROUND_SURFACE'
else:
#wall
strType = 'BUILDING_WALL_SURFACE'
else:
if res[0][0] != ('', ''):
strId = res[0][0][0]
strType = res[0][0][1]
elif res[1][0] != ('', ''):
strId = res[1][0][0]
strType = res[1][0][1]
#assign the semantics to all the raw face in tag
for f in tag:
if ModelData.dictFaces[f].get_id() == '' :
ModelData.dictFaces[f].set_id(strId)
if ModelData.dictFaces[f].get_type() == '' :
ModelData.dictFaces[f].set_type(strType)
else:
ModelData.dictFaces[fid].set_id(ModelDataFuncs.generate_uuid())
ModelData.dictFaces[fid].set_type('BUILDING_WALL_SURFACE')
def sort_facesgroup_by_size(listFaceGroup):
listSize = []
for list in listFaceGroup:
bbox = ModelDataFuncs.get_bbox_faces(list)
listSize.append(SimpleMath.square_dist_point_to_point(bbox[0], bbox[1]), list)
listSize.sort()
#
listFaceGroup = []
for list in listSize:
listFaceGroup.append(list[1])
return listFaceGroup
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 CDT():
print("Start tetrahedralization....")
curDirBefore = os.getcwd()
path = os.path.dirname(os.path.realpath(__file__))
if sizeof(c_voidp) == 4:
#win32
path = os.path.join(path, '..\\tetgendll\\Release')
os.chdir(path)
if not os.path.exists("tetgendll.dll"):
print("DLL missing: " + path + "tetgendll.dll")
Tetrahedralator = CDLL("tetgendll.dll")
elif sizeof(c_voidp) == 8:
#x64
path = os.path.join(path, '..\\tetgendll\\x64\\Release')
os.chdir(path)
if not os.path.exists("tetgendll.dll"):
print("DLL missing: " + path + "tetgendll.dll")
Tetrahedralator = CDLL("tetgendll.dll")
os.chdir(curDirBefore)
#be careful with the indices
#record the indices of inserted dictVertices and map the indices in the dictFaces and tetrahedron to the finally added indices
mapVertTet = {}
iCount = 0
cVertices = (c_double * (len(ModelData.dictVertices) * 3))() #the size of dictVertices *3
for key in ModelData.dictVertices:
cVertices[3*iCount] = c_double(ModelData.dictVertices[key][0])
cVertices[3*iCount+1] = c_double(ModelData.dictVertices[key][1])
cVertices[3*iCount+2] = c_double(ModelData.dictVertices[key][2])
mapVertTet[iCount] = key
iCount += 1
iCount = 0
cFaces = (c_int * (len(ModelData.dictFaces) * 3))() #the size of dictFaces * 3
for key in ModelData.dictFaces:
#be careful with the indices
cFaces[3*iCount] = c_int(ModelDataFuncs.find_key_from_dict_by_exact_value(mapVertTet, ModelData.dictFaces[key].get_vids()[0]))
cFaces[3*iCount+1] = c_int(ModelDataFuncs.find_key_from_dict_by_exact_value(mapVertTet, ModelData.dictFaces[key].get_vids()[1]))
cFaces[3*iCount+2] = c_int(ModelDataFuncs.find_key_from_dict_by_exact_value(mapVertTet, ModelData.dictFaces[key].get_vids()[2]))
iCount += 1
numberOfOutputVerts = c_int(0);
numberOfOutputTriangles = c_int(0);
numberOfOutputTetrahedrons = c_int(0);
try:
Tetrahedralator.simpleTetrahedralize(byref(cVertices), c_int(len(ModelData.dictVertices)), byref(cFaces), c_int(len(ModelData.dictFaces)),
byref(numberOfOutputVerts), byref(numberOfOutputTriangles), byref(numberOfOutputTetrahedrons))
except ValueError:
print("CDT failed")
return False
#check
if numberOfOutputTetrahedrons.value == 0:
print("tetrahedralization failed")
return False
#Get the results
outputVerts = (c_double * (numberOfOutputVerts.value *3))()
outputConvexhullTris = (c_int * (numberOfOutputTriangles.value *3))()
outputTetrahedrons = (c_int * (numberOfOutputTetrahedrons.value *4))()
Tetrahedralator.getResults(pointer(outputVerts), pointer(outputConvexhullTris), pointer(outputTetrahedrons))
#update the ModelData
#dictVertices
if numberOfOutputVerts.value > len(ModelData.dictVertices):
print("{} steiner points inserted (at the back of the original list)").format(numberOfOutputVerts.value - len(ModelData.dictVertices))
for i in range(len(ModelData.dictVertices) * 3, numberOfOutputVerts.value * 3, 3):
#be carefull with the indices (start with 0)
mapVertTet[len(mapVertTet)] = ModelDataFuncs.add_vertex((outputVerts[i], outputVerts[i+1], outputVerts[i+2]))
#dictTetrahedrons
for i in range(0, numberOfOutputTetrahedrons.value * 4, 4):
ModelData.dictTetrahedrons[i/4] = Class_tetrahedron((mapVertTet[outputTetrahedrons[i]], mapVertTet[outputTetrahedrons[i+1]], mapVertTet[outputTetrahedrons[i+2]], mapVertTet[outputTetrahedrons[i+3]]))
#record the the triangles on the shell
isFaceDeleted = False
for i in range(0, numberOfOutputTriangles.value*3, 3):
tris = (mapVertTet[outputConvexhullTris[i]], mapVertTet[outputConvexhullTris[i+1]], mapVertTet[outputConvexhullTris[i+2]])
isExt = False
for f in ModelData.dictFaces:
if ModelData.dictFaces[f].is_equal_geometry(ClassFace.Class_face(tris)):
#found the existing face
isExt = True
#add to the list of shell face
ModelData.listShellFaceIDs.append(f)
break
if isExt == False:
#distance mapping in case this geometry is produced by flipping the coplanar triangle
fId = ModelDataFuncs.find_face_by_mapping(tris)
if fId:
ModelData.listShellFaceIDs.append(ModelDataFuncs.add_face(ClassFace.Class_face(tris, ClassFace.FIX, ModelData.dictFaces[fId].get_id(), ModelData.dictFaces[fId].get_type())))
ModelDataFuncs.pre_remove_face_by_faceids([fId])
isFaceDeleted = True
else:
#add the new face and add the face to the list of shell face
ModelData.listShellFaceIDs.append(ModelDataFuncs.add_face(ClassFace.Class_face(tris)))
#remove faces
if isFaceDeleted:
ModelDataFuncs.remove_faces()
#
print ("After CDT: " + str(len(ModelData.dictTetrahedrons)) + " tetrahedra and " + str(len(ModelData.dictFaces)) + " dictFaces and " + str(len(ModelData.dictVertices)) + " dictVertices")
return True
def extract_mesh_from_tet(bIsSemantics = False):
print("----Extract the output mesh----")
#check
if len(ModelData.dictTetrahedrons) == 0:
print("invalid call extract Mesh")
return
#deletion and deduce semantics
for fkey in ModelData.dictFaces:
if fkey not in ModelData.listShellFaceIDs:
ModelDataFuncs.pre_remove_face_by_faceids([fkey])
#clearup
if ModelDataFuncs.remove_faces() > 0:
print ("Mesh extracted")
#optimisazation
if ModelDataFuncs.optimise_illshaped_shellfaces():
ModelDataFuncs.remove_faces()
ModelDataFuncs.clean_duplicated_vertices()
ModelDataFuncs.clean_unreferenced_vertices()
ModelDataFuncs.restore_normals()
#deduce semantics
deduce_semantics_of_poly(bIsSemantics)
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 do_the_work(inputFilePath, isSemantic, debugControl):
#Init mid files
if os.path.exists(MIDFILEPATH):
os.remove(MIDFILEPATH)
#Init datastructure
ModelDataFuncs.Init()
ModelData.strInputFileName, ext = os.path.splitext(inputFilePath)
#read file
print ("----Processing file----")
print (inputFilePath)
if isSemantic == True:
if os.path.splitext(inputFilePath)[1] == '.obj':
print('WARNING: obj format does not contain semantics')
print('Semantics will be deduced')
print ("Check and tessellate the file...")
MyCov = ConvProvider()
MyCov.convert(inputFilePath, MIDFILEPATH, True)
#read a tesselated objfile
if ModelDataFuncs.reader_obj(MIDFILEPATH) == False:
raise Exception
else:
#poly with semantics
try:
if not ModelDataFuncs.reader_poly_with_semantics(inputFilePath):
return
except:
raise ImportError
else:
#preprocess (convert and tessellation)
print ("Check and tessellate the file...")
MyCov = ConvProvider()
MyCov.convert(inputFilePath, MIDFILEPATH, True)
#read a tesselated objfile
if ModelDataFuncs.reader_obj(MIDFILEPATH) == False:
raise Exception
#invert the normal of the input model
if ModelData.global_INVERT_NORMAL:
ModelDataFuncs.invert_poly_normal()
#only for debug
#tmp = ModelData.centerVertex
#ModelData.centerVertex = (0.0, 0.0, 0.0)
#ModelDataFuncs.writer_obj(ModelData.strInputFileName+"_CLS.obj")
#ModelData.centerVertex = tmp
#
if int(debugControl) == 1:
#Decomposit all the triangles
DecomposeFunc.model_decompositionEx()
#Merge all the coplaner dictFaces
#coplaner_face_merge()
#ModelDataFuncs.writer_obj(ModelData.strInputFileName+"_DEC.obj")
elif int(debugControl) == 2:
#Constrained Tetrahedralization
if False == TetraFunc.CDT():
print("Constrained Delauney Tetrahedralization FAILED!")
return
#ModelDataFuncs.writer_obj(ModelData.strInputFileName+"_CDT.obj")
#Heuristic carving
CarveFunc.heuristic_tet_carving()
#ModelDataFuncs.writer_obj(ModelData.strInputFileName+"_CARVE.obj")
#Reconstruct the mesh from tetrahedron
TetraFunc.extract_mesh_from_tet(isSemantic)
#Output
ModelDataFuncs.writer_obj(ModelData.strInputFileName+"_OUTPUT.obj")
if isSemantic:
ModelDataFuncs.writer_poly_with_semantics(ModelData.strInputFileName + "_OUTPUT.poly")
elif int(debugControl) == 3:
#only deduce the semantics
TetraFunc.deduce_semantics_of_poly(isSemantic)
ModelDataFuncs.writer_poly_with_semantics(ModelData.strInputFileName + "_OUTPUT.poly")
else:
#Decomposit all the triangles
DecomposeFunc.model_decompositionEx()
#Merge all the coplaner dictFaces
#coplaner_face_merge()
ModelDataFuncs.writer_obj(ModelData.strInputFileName +"_DEC.obj")
#Constrained Tetrahedralization
if False == TetraFunc.CDT():
print("Constrained Delauney Tetrahedralization FAILED!")
return
ModelDataFuncs.writer_obj(ModelData.strInputFileName+"_CDT.obj")
#Heuristic carving
CarveFunc.heuristic_tet_carving()
ModelDataFuncs.writer_obj(ModelData.strInputFileName+"_CARVE.obj")
#Reconstruct the mesh from tetrahedron
TetraFunc.extract_mesh_from_tet(isSemantic)
#Output
ModelDataFuncs.writer_obj(ModelData.strInputFileName + "_OUTPUT.obj")
if isSemantic:
ModelDataFuncs.writer_poly_with_semantics(ModelData.strInputFileName + "_OUTPUT.poly")
