def process_nif(test_file, bone_list):
print('Processing', test_file)
target_nif = load_nif(test_file, template = template_mesh, settings = current_settings)
if not target_nif.valid:
print(test_file,'Could Not Be Processed')
targets_found = 0
for this_mesh in target_nif.meshes:
targets_found += mainSearch(current_settings, template_mesh, this_mesh, search_resolution = 1, vertFunction = setWeightDictionary, side = side)
save_file(target_nif, test_file, targets_found, current_settings)
def process_nif(this_file, repair_doubles = False):
# if findall(r_nif, test_file) and test_file != current_settings['template']:
print('******NIF******')
print('Processing', this_file)
target_nif = load_nif(this_file, template = template_mesh, settings = current_settings)
if not target_nif.valid:
print(target_nif,'Could Not Be Processed')
return
nif_gender = target_nif.getGender()
if gender_list:
if nif_gender not in gender_list:
print('Incorrect Gender:', this_file )
print('Closing', this_file )
return
mesh_ = template_mesh.get(nif_gender)
verts_ = template_verts.get(nif_gender)
elif nif_gender != current_settings['gender']:
print('Incorrect Gender:', this_file )
print('Closing', this_file )
return
targets_found = 0
for this_mesh in target_nif.meshes:
if gender_list:
targets_found += mainSearch(current_settings, mesh_, this_mesh, search_resolution = 1, vertFunction = setSeams, nmv = target_nmv, act_verts = verts_)
else:
targets_found += mainSearch(current_settings, template_mesh, this_mesh, search_resolution = 1, vertFunction = setSeams, nmv = target_nmv, act_verts = template_verts)
if repair_doubles:
for this_mesh in target_nif.meshes:
if current_settings.get('doubles'):
for tup in this_mesh.nmv_doubles:
average_normal = kg.math_util.vector3([0,0,0])
bad_normal = 0
for v in tup:
if v.normal:
average_normal += v.normal
else:
bad_normal += 1
if bad_normal == len(tup):
continue
for v in tup:
v.setNormal(kg.math_util.normalizeVector(average_normal))
targets_found += 1
save_file(target_nif, this_file, targets_found, current_settings)
def process_tri(this_file):
if template_verts:
print('******TRI******')
mesh_ = template_mesh
verts_ = template_verts
if gender_list or current_settings['gender']:
"""
Determine Gender of Tri File
"""
root_nif_path = re.sub(r_tri_end, '.nif', this_file)
tri_gender = getGenderFlag(root_nif_path)
if tri_gender is 'NONE' and path.exists(root_nif_path):
target_nif = load_nif(root_nif_path, template = False, settings = {})
if not target_nif.valid:
tri_gender = 'NONE'
else:
tri_gender = target_nif.getGender()
if gender_list:
mesh_ = template_mesh.get(tri_gender)
verts_ = template_verts.get(tri_gender)
if not mesh_ or not verts_:
print('Incorrect Gender:', this_file )
print('Closing', this_file )
return
elif tri_gender != current_settings['gender']:
print('Incorrect Gender:', this_file )
print('Closing', this_file )
return
print('Processing', this_file)
target_tri = load_tri(this_file, template = template_mesh, settings = current_settings)
tri_mesh = target_tri.mesh
# for a in zip(sorted([(vert.idx, vert.getLoc(world_loc = True).as_list()) for vert in tri_mesh.getVerts(nmv = True).values()], key = itemgetter(0)),
# sorted([(vert.idx, vert.getLoc(world_loc = False).as_list()) for vert in template_mesh.getVerts(nmv = True).values()], key = itemgetter(0))):
# print (a)
targets_found = mainSearch(current_settings, mesh_, tri_mesh, search_resolution = 1, vertFunction = setTriSeams, nmv = target_nmv, act_verts = verts_, tri = True, world_loc = True)
save_file(target_tri, this_file, targets_found, current_settings)