def createBubbleMaterial(name, baseColor):
mat = bpy.data.materials.new(name)
mat.use_nodes = True
mat.blend_method = 'BLEND'
principled = PrincipledBSDFWrapper(mat, is_readonly=False)
principled.base_color = baseColor
principled.metallic = 0.5
principled.specular = 0.2
principled.roughness = 0.05
#principled.IOR = 1.1
principled.alpha = 0.3
# principled.specular_texture.image = load_image("/path/to/image.png")
# Export
principled = PrincipledBSDFWrapper(mat, is_readonly=True)
base_color = principled.base_color
specular_texture = principled.specular_texture
if specular_texture and specular_texture.image:
specular_texture_filepath = principled.specular_texture.image.filepath
def handle_materials(context, model, materials, update):
"""
"""
for m in model.Materials:
matname = material_name(m)
if matname not in materials:
blmat = utils.get_iddata(context.blend_data.materials, None,
m.Name, None)
if update:
blmat.use_nodes = True
refl = m.Reflectivity
transp = m.Transparency
ior = m.IndexOfRefraction
roughness = m.ReflectionGlossiness
transrough = m.RefractionGlossiness
spec = m.Shine / 255.0
if m.DiffuseColor == _black and m.Reflectivity > 0.0 and m.Transparency == 0.0:
r, g, b, _ = m.ReflectionColor
elif m.DiffuseColor == _black and m.Reflectivity == 0.0 and m.Transparency > 0.0:
r, g, b, _ = m.TransparentColor
refl = 0.0
elif m.DiffuseColor == _black and m.Reflectivity > 0.0 and m.Transparency > 0.0:
r, g, b, _ = m.TransparentColor
refl = 0.0
else:
r, g, b, a = m.DiffuseColor
if refl > 0.0 and transp > 0.0:
refl = 0.0
principled = PrincipledBSDFWrapper(blmat, is_readonly=False)
principled.base_color = (r / 255.0, g / 255.0, b / 255.0)
principled.metallic = refl
principled.transmission = transp
principled.ior = ior
principled.roughness = roughness
principled.specular = spec
principled.node_principled_bsdf.inputs[
16].default_value = transrough
materials[matname] = blmat
def importWoWOBJ(objectFile, givenParent=None):
baseDir, fileName = os.path.split(objectFile)
print('Parsing OBJ: ' + fileName)
### OBJ wide
material_libs = set()
mtlfile = ""
verts = []
normals = []
uv = []
meshes = []
### Per group
class OBJMesh:
def __init__(self):
self.usemtl = ""
self.name = ""
self.faces = []
curMesh = OBJMesh()
meshIndex = -1
with open(objectFile, 'rb') as f:
for line in f:
line_split = line.split()
if not line_split:
continue
line_start = line_split[0]
if line_start == b'mtllib':
mtlfile = line_split[1]
elif line_start == b'v':
verts.append([float(v) for v in line_split[1:]])
elif line_start == b'vn':
normals.append([float(v) for v in line_split[1:]])
elif line_start == b'vt':
uv.append([float(v) for v in line_split[1:]])
elif line_start == b'f':
line_split = line_split[1:]
meshes[meshIndex].faces.append(
(int(line_split[0].split(b'/')[0]),
int(line_split[1].split(b'/')[0]),
int(line_split[2].split(b'/')[0])))
elif line_start == b'g':
meshIndex += 1
meshes.append(OBJMesh())
meshes[meshIndex].name = line_split[1].decode("utf-8")
elif line_start == b'usemtl':
meshes[meshIndex].usemtl = line_split[1].decode("utf-8")
## Materials file (.mtl)
materials = dict()
matname = ""
matfile = ""
with open(os.path.join(baseDir, mtlfile.decode("utf-8")), 'r') as f:
for line in f:
line_split = line.split()
if not line_split:
continue
line_start = line_split[0]
if line_start == 'newmtl':
matname = line_split[1]
elif line_start == 'map_Kd':
matfile = line_split[1]
materials[matname] = os.path.join(baseDir, matfile)
if bpy.ops.object.select_all.poll():
bpy.ops.object.select_all(action='DESELECT')
# TODO: Better handling for dupes?
objname = os.path.basename(objectFile)
if objname in bpy.data.objects:
objindex = 1
newname = objname
while (newname in bpy.data.objects):
newname = objname + '.' + str(objindex).rjust(3, '0')
objindex += 1
newmesh = bpy.data.meshes.new(objname)
obj = bpy.data.objects.new(objname, newmesh)
## Textures
# TODO: Must be a better way to do this!
materialmapping = dict()
for matname, texturelocation in materials.items():
# Import material only once
if (matname not in bpy.data.materials):
mat = bpy.data.materials.new(name=matname)
mat.use_nodes = True
mat.blend_method = 'CLIP'
principled = PrincipledBSDFWrapper(mat, is_readonly=False)
principled.specular = 0.0
principled.base_color_texture.image = load_image(texturelocation)
mat.node_tree.links.new(
mat.node_tree.nodes['Image Texture'].outputs[1],
mat.node_tree.nodes['Principled BSDF'].inputs[18])
obj.data.materials.append(bpy.data.materials[matname])
# TODO: Must be a better way to do this!
materialmapping[matname] = len(obj.data.materials) - 1
## Meshes
bm = bmesh.new()
i = 0
for v in verts:
vert = bm.verts.new(v)
vert.normal = normals[i]
i = i + 1
bm.verts.ensure_lookup_table()
bm.verts.index_update()
for mesh in meshes:
exampleFaceSet = False
for face in mesh.faces:
try:
## TODO: Must be a better way to do this, this is already much faster than doing material every face, but still.
if exampleFaceSet == False:
bm.faces.new((bm.verts[face[0] - 1], bm.verts[face[1] - 1],
bm.verts[face[2] - 1]))
bm.faces.ensure_lookup_table()
bm.faces[-1].material_index = materialmapping[mesh.usemtl]
bm.faces[-1].smooth = True
exampleFace = bm.faces[-1]
exampleFaceSet = True
else:
## Use example face if set to speed up material copy!
bm.faces.new((bm.verts[face[0] - 1], bm.verts[face[1] - 1],
bm.verts[face[2] - 1]), exampleFace)
except ValueError:
## TODO: Duplicate faces happen for some reason
pass
uv_layer = bm.loops.layers.uv.new()
for face in bm.faces:
for loop in face.loops:
loop[uv_layer].uv = uv[loop.vert.index]
bm.to_mesh(newmesh)
bm.free()
## Rotate object the right way
obj.rotation_euler = [0, 0, 0]
obj.rotation_euler.x = radians(90)
bpy.context.scene.collection.objects.link(obj)
obj.select_set(True)
## WoW coordinate system
max_size = 51200 / 3
map_size = max_size * 2
adt_size = map_size / 64
## Import doodads and/or WMOs
csvPath = objectFile.replace('.obj', '_ModelPlacementInformation.csv')
if os.path.exists(csvPath):
with open(csvPath) as csvFile:
reader = csv.DictReader(csvFile, delimiter=';')
if 'Type' in reader.fieldnames:
importType = 'ADT'
wmoparent = bpy.data.objects.new("WMOs", None)
wmoparent.parent = obj
wmoparent.name = "WMOs"
wmoparent.rotation_euler = [0, 0, 0]
wmoparent.rotation_euler.x = radians(-90)
bpy.context.scene.collection.objects.link(wmoparent)
doodadparent = bpy.data.objects.new("Doodads", None)
doodadparent.parent = obj
doodadparent.name = "Doodads"
doodadparent.rotation_euler = [0, 0, 0]
doodadparent.rotation_euler.x = radians(-90)
bpy.context.scene.collection.objects.link(doodadparent)
else:
importType = 'WMO'
if not givenParent:
print('WMO import without given parent, creating..')
givenParent = bpy.data.objects.new("WMO parent", None)
givenParent.parent = obj
givenParent.name = "Doodads"
givenParent.rotation_euler = [0, 0, 0]
givenParent.rotation_euler.x = radians(-90)
bpy.context.scene.collection.objects.link(givenParent)
for row in reader:
if importType == 'ADT':
if 'importedModelIDs' in bpy.context.scene:
tempModelIDList = bpy.context.scene['importedModelIDs']
else:
tempModelIDList = []
if row['ModelId'] in tempModelIDList:
print('Skipping already imported model ' +
row['ModelId'])
continue
else:
tempModelIDList.append(row['ModelId'])
# ADT CSV
if row['Type'] == 'wmo':
print('ADT WMO import: ' + row['ModelFile'])
# Make WMO parent that holds WMO and doodads
parent = bpy.data.objects.new(
row['ModelFile'] + " parent", None)
parent.parent = wmoparent
parent.location = (17066 - float(row['PositionX']),
(17066 - float(row['PositionZ'])) *
-1, float(row['PositionY']))
parent.rotation_euler = [0, 0, 0]
parent.rotation_euler.x += radians(
float(row['RotationZ']))
parent.rotation_euler.y += radians(
float(row['RotationX']))
parent.rotation_euler.z = radians(
(-90 + float(row['RotationY'])))
if row['ScaleFactor']:
parent.scale = (float(row['ScaleFactor']),
float(row['ScaleFactor']),
float(row['ScaleFactor']))
bpy.context.scene.collection.objects.link(parent)
## Only import OBJ if model is not yet in scene, otherwise copy existing
if row['ModelFile'] not in bpy.data.objects:
importedFile = importWoWOBJ(
os.path.join(baseDir, row['ModelFile']),
parent)
else:
## Don't copy WMOs with doodads!
if os.path.exists(
os.path.join(
baseDir, row['ModelFile'].replace(
'.obj',
'_ModelPlacementInformation.csv'))
):
importedFile = importWoWOBJ(
os.path.join(baseDir, row['ModelFile']),
parent)
else:
originalObject = bpy.data.objects[
row['ModelFile']]
importedFile = originalObject.copy()
importedFile.data = originalObject.data.copy()
bpy.context.scene.collection.objects.link(
importedFile)
importedFile.parent = parent
elif row['Type'] == 'm2':
print('ADT M2 import: ' + row['ModelFile'])
## Only import OBJ if model is not yet in scene, otherwise copy existing
if row['ModelFile'] not in bpy.data.objects:
importedFile = importWoWOBJ(
os.path.join(baseDir, row['ModelFile']))
else:
originalObject = bpy.data.objects[row['ModelFile']]
importedFile = originalObject.copy()
importedFile.rotation_euler = [0, 0, 0]
importedFile.rotation_euler.x = radians(90)
bpy.context.scene.collection.objects.link(
importedFile)
importedFile.parent = doodadparent
importedFile.location.x = (17066 -
float(row['PositionX']))
importedFile.location.y = (
17066 - float(row['PositionZ'])) * -1
importedFile.location.z = float(row['PositionY'])
importedFile.rotation_euler.x += radians(
float(row['RotationZ']))
importedFile.rotation_euler.y += radians(
float(row['RotationX']))
importedFile.rotation_euler.z = radians(
90 + float(row['RotationY']))
if row['ScaleFactor']:
importedFile.scale = (float(row['ScaleFactor']),
float(row['ScaleFactor']),
float(row['ScaleFactor']))
bpy.context.scene['importedModelIDs'] = tempModelIDList
else:
# WMO CSV
print('WMO M2 import: ' + row['ModelFile'])
if row['ModelFile'] not in bpy.data.objects:
importedFile = importWoWOBJ(
os.path.join(baseDir, row['ModelFile']))
else:
originalObject = bpy.data.objects[row['ModelFile']]
importedFile = originalObject.copy()
bpy.context.scene.collection.objects.link(importedFile)
importedFile.location = (float(row['PositionX']) * -1,
float(row['PositionY']) * -1,
float(row['PositionZ']))
importedFile.rotation_euler = [0, 0, 0]
rotQuat = Quaternion(
(float(row['RotationW']), float(row['RotationX']),
float(row['RotationY']), float(row['RotationZ'])))
rotEul = rotQuat.to_euler()
rotEul.x += radians(90)
rotEul.z += radians(180)
importedFile.rotation_euler = rotEul
importedFile.parent = givenParent or obj
if row['ScaleFactor']:
importedFile.scale = (float(row['ScaleFactor']),
float(row['ScaleFactor']),
float(row['ScaleFactor']))
return obj
#objectFile = "D:\\models\\world\\maps\\azeroth\\azeroth_32_32.obj"
#objectFile = "D:\\models\\world\\maps\\kultiras\\kultiras_32_29.obj"
#objectFile = "D:\\models\\world\\wmo\\kultiras\\human\\8hu_kultiras_seabattlement01.obj"
#importWoWOBJ(objectFile)