mirror of
https://github.com/devZoGok/vb01.git
synced 2026-08-26 19:43:30 +00:00
260 lines
9.9 KiB
Python
260 lines
9.9 KiB
Python
import sys
|
|
|
|
def exportData(ob):
|
|
par = ob.parent
|
|
if(par is not None and par.type == 'ARMATURE'):
|
|
while(par is not None):
|
|
par = par.parent
|
|
|
|
file.write('{\n' + ob.type + ": " + ob.name + '\n')
|
|
file.write('pos: ' + str(ob.location.x) + ' ' + str(ob.location.y) + ' ' + str(ob.location.z) + ' \n')
|
|
file.write('rot: ' + str(ob.rotation_quaternion.w) + ' ' + str(ob.rotation_quaternion.x) + ' ' + str(ob.rotation_quaternion.y) + ' ' + str(ob.rotation_quaternion.z) + ' \n')
|
|
file.write('scale: ' + str(ob.scale.x) + ' ' + str(ob.scale.y) + ' ' + str(ob.scale.z) + ' \n')
|
|
file.write('parent: ' + ('-' if par is None else par.name) + '\n')
|
|
|
|
numAnims = 0
|
|
if(ob.animation_data is not None and len(ob.animation_data.nla_tracks) > 0):
|
|
numAnims = len(ob.animation_data.nla_tracks[0].strips)
|
|
|
|
channels = []
|
|
if(ob.type == 'ARMATURE'):
|
|
file.write('bones: ' + str(len(ob.data.bones)) + ' ' + str(numAnims) + ' \n')
|
|
|
|
for bone in ob.data.bones:
|
|
head = bone.head
|
|
tail = bone.tail
|
|
xAxis = bone.x_axis
|
|
yAxis = bone.y_axis
|
|
ikTarget = '-'
|
|
chainLength = -1
|
|
|
|
ikConstraint = None
|
|
for constr in ob.pose.bones[bone.name].constraints:
|
|
if constr.name == 'IK':
|
|
ikConstraint = constr
|
|
if(ikConstraint is not None and ikConstraint.subtarget):
|
|
ikTarget = ikConstraint.subtarget
|
|
if ikConstraint.chain_count == 0:
|
|
chainLength = 1
|
|
bonePar = bone.parent
|
|
while(bonePar is not None):
|
|
chainLength = chain_count + 1
|
|
bonePar = bonePar.parent
|
|
else:
|
|
chainLength = ikConstraint.chain_count
|
|
|
|
file.write(bone.name + ': ' + ('None' if bone.parent is None else bone.parent.name) + " " + str(bone.length) + " " + str(head.x) + " " + str(head.y) + " " + str(head.z) + " " + str(xAxis.x) + " " + str(xAxis.y) + " " + str(xAxis.z) + " " + str(yAxis.x) + " " + str(yAxis.y) + " " + str(yAxis.z) + " " + ikTarget + ' ' + str(chainLength) + ' \n')
|
|
|
|
elif(ob.type == 'MESH'):
|
|
skeletonName = '-'
|
|
skeleton = None
|
|
if(len(ob.modifiers) > 0 and ob.modifiers['Armature'] and ob.modifiers['Armature'].object):
|
|
skeleton = ob.modifiers['Armature'].object
|
|
skeletonName = skeleton.name
|
|
file.write('skeleton: ' + skeletonName + '\n')
|
|
mesh = ob.data
|
|
|
|
numFaces = len(mesh.polygons)
|
|
numVerts = len(mesh.vertices)
|
|
numGroups = len(ob.vertex_groups)
|
|
numShapeKeys = 0
|
|
if mesh.shape_keys:
|
|
numShapeKeys = len(mesh.shape_keys.key_blocks) - 1
|
|
|
|
file.write('numElements: ' + str(numVerts) + ' ' + str(numFaces) + ' ' + str(numGroups) + ' ' + str(numShapeKeys) + ' \n')
|
|
|
|
file.write('groups:\n')
|
|
if numGroups > 0:
|
|
for group in ob.vertex_groups:
|
|
file.write(group.name + '\n')
|
|
|
|
file.write('vertices:\n')
|
|
print('Exporting vertices...\n')
|
|
for vert in mesh.vertices:
|
|
pos = vert.co
|
|
norm = vert.normal
|
|
weights = []
|
|
for i in range(0, numGroups):
|
|
weights.append(0.0)
|
|
for g in vert.groups:
|
|
weights[g.group] = g.weight
|
|
file.write(str(pos.x) + ' ' + str(pos.y) + ' ' + str(pos.z) + ' ' + str(norm.x) + ' ' + str(norm.y) + ' ' + str(norm.z) + ' ')
|
|
for w in weights:
|
|
file.write(str(w) + ' ')
|
|
file.write('\n')
|
|
|
|
file.write('faces:\n')
|
|
print('Exporting faces...\n')
|
|
mesh.calc_tangents()
|
|
for face in mesh.polygons:
|
|
for vert, loopInd in zip(face.vertices, face.loop_indices):
|
|
uv = mesh.uv_layers[0].data[loopInd].uv
|
|
loop = mesh.loops[loopInd]
|
|
tan = loop.tangent
|
|
bitan = loop.bitangent
|
|
file.write(str(vert) + ' ' + str(uv.x) + ' ' + str(uv.y) + ' ' + str(tan.x) + ' ' + str(tan.y) + ' ' + str(tan.z) + ' ' + str(bitan.x) + ' ' + str(bitan.y) + ' ' + str(bitan.z) + ' \n')
|
|
mesh.free_tangents()
|
|
|
|
file.write('shapeKeys:\n')
|
|
if numShapeKeys > 0:
|
|
print('Exporting shape keys...\n')
|
|
for shape_key in mesh.shape_keys.key_blocks:
|
|
if shape_key.name != 'Basis':
|
|
file.write(shape_key.name + ': ' + str(shape_key.slider_min) + ' ' + str(shape_key.slider_max) + ' \n')
|
|
for vert in shape_key.data:
|
|
pos = vert.co
|
|
file.write(str(pos.x) + ' ' + str(pos.y) + ' ' + str(pos.z) + ' \n')
|
|
channels.extend(mesh.shape_keys.animation_data.drivers)
|
|
elif ob.type == 'LIGHT':
|
|
color = ob.data.color
|
|
outerAngle = 0
|
|
innerAngle = 0
|
|
type = ob.data.type
|
|
if type == 'SPOT':
|
|
outerAngle = ob.data.spot_size
|
|
innerAngle = outerAngle * ob.data.spot_blend
|
|
file.write('type: ' + type + '\n')
|
|
file.write('color: ' + str(color[0]) + ' ' + str(color[1]) + ' ' + str(color[2]) + ' \n')
|
|
file.write('outerAngle: ' + str(outerAngle) + ' \n')
|
|
file.write('innerAngle: ' + str(innerAngle) + ' \n')
|
|
|
|
|
|
file.write('animations: ' + str(numAnims) + '\n')
|
|
if numAnims > 0:
|
|
if ob.type == 'LIGHT':
|
|
readAnimations(ob, len(ob.animation_data.nla_tracks[0].strips))
|
|
if(ob.data.animation_data is not None and len(ob.data.animation_data.nla_tracks) > 0):
|
|
readAnimations(ob.data, len(ob.data.animation_data.nla_tracks[0].strips))
|
|
else:
|
|
readAnimations(ob, numAnims)
|
|
|
|
if ob.animation_data is not None:
|
|
channels.extend(ob.animation_data.drivers)
|
|
readDrivers(channels)
|
|
|
|
def readDrivers(channels):
|
|
file.write('drivers: ' + str(len(channels)) + '\n')
|
|
for channel in channels:
|
|
firstQuoteId = channel.data_path.find('"') + 1
|
|
secondQuoteId = channel.data_path.rfind('"')
|
|
name = (ob.name if firstQuoteId == -1 and secondQuoteId == -1 else channel.data_path[firstQuoteId : secondQuoteId])
|
|
readDriver(channel, name)
|
|
|
|
def getChannelType(channel):
|
|
channelType = ''
|
|
|
|
dotId = channel.data_path.rfind('.') + 1;
|
|
coordType = channel.data_path[dotId :]
|
|
arrInd = channel.array_index
|
|
coords = 'wxyz'
|
|
|
|
if coordType == 'location':
|
|
channelType = 'pos_' + coords[arrInd + 1]
|
|
elif coordType == 'rotation_quaternion':
|
|
channelType = 'rot_' + coords[arrInd]
|
|
elif coordType == 'scale':
|
|
channelType = 'scale_' + coords[arrInd + 1]
|
|
elif coordType == 'color':
|
|
channelType = 'diffuse_color_' + coords[arrInd + 1]
|
|
elif coordType == 'slider_min':
|
|
channelType = 'shape_key_min'
|
|
elif coordType == 'value':
|
|
channelType = 'shape_key_value'
|
|
elif coordType == 'slider_max':
|
|
channelType = 'shape_key_max'
|
|
|
|
return channelType
|
|
|
|
def readChannel(animatableName, channel):
|
|
channelType = getChannelType(channel)
|
|
numKeyframes = len(channel.keyframe_points)
|
|
|
|
file.write(animatableName + ' ' + channelType + ' ' + str(numKeyframes) + ' \n')
|
|
|
|
for i in range(numKeyframes):
|
|
keyframe = channel.keyframe_points[i]
|
|
keyframeId = keyframe.co[0]
|
|
interp = keyframe.interpolation
|
|
interpInt = -1
|
|
|
|
if interp == 'CONSTANT':
|
|
interpInt = 0
|
|
elif interp == 'LINEAR':
|
|
interpInt = 1
|
|
elif interp == 'BEZIER':
|
|
interpInt = 2
|
|
|
|
keyframeValue = keyframe.co[1]
|
|
file.write(str(keyframeValue) + ' ' + str(keyframeId) + ' ' + str(interpInt))
|
|
|
|
leftHandleValue = (keyframe.handle_left[1])
|
|
leftHandleFrame = (keyframe.handle_left[0])
|
|
rightHandleValue = (keyframe.handle_right[1])
|
|
rightHandleFrame = (keyframe.handle_right[0])
|
|
if interp != 'BEZIER':
|
|
rightHandleValue = keyframeValue
|
|
rightHandleFrame = keyframeId
|
|
if i > 0 and channel.keyframe_points[i - 1].interpolation != 'BEZIER':
|
|
leftHandleValue = keyframeValue
|
|
leftHandleFrame = keyframeId
|
|
|
|
file.write(' ' + str(leftHandleValue) + ' ' + str(leftHandleFrame) + ' ')
|
|
file.write(str(rightHandleValue) + ' ' + str(rightHandleFrame) + ' ')
|
|
file.write('\n')
|
|
|
|
def readDriver(channel, driverName):
|
|
#for channel in channels:
|
|
driver = channel.driver
|
|
target = driver.variables[0].targets[0];
|
|
|
|
file.write(target.id.name + ' ' + target.bone_target + ' ' + str(target.transform_type) + ' \n')
|
|
print('Exporting driver ...\n')
|
|
|
|
numKeyframes = len(channel.keyframe_points)
|
|
channelType = getChannelType(channel)
|
|
#file.write(driverName + ' 1 ' + channelType + ' ' + str(numKeyframes) + ' \n')
|
|
|
|
readChannel(driverName, channel)
|
|
|
|
def readAnimations(ob, numAnims):
|
|
for nlaStrip in ob.animation_data.nla_tracks[0].strips:
|
|
animName = nlaStrip.name
|
|
action = bpy.data.actions[animName]
|
|
numAnimBones = len(action.groups)
|
|
file.write('animation: ' + animName + ' ' + str(len(action.fcurves)) + ' \n')
|
|
|
|
print('Exporting animation ' + animName + '...\n')
|
|
|
|
for channel in action.fcurves:
|
|
group = channel.group
|
|
readChannel((ob.name if group is None or group.name == 'Object Transforms' else group.name), channel)
|
|
|
|
fl = bpy.data.filepath
|
|
dotId = 0
|
|
for i in range(len(fl)):
|
|
if(fl[i] == '.'):
|
|
dotId = i
|
|
break
|
|
fl = fl[0 : dotId] + '.vb'
|
|
|
|
objects = []
|
|
selectedObjects = bpy.context.selected_objects
|
|
|
|
for s in selectedObjects:
|
|
sPar = s
|
|
while(sPar):
|
|
sPar = sPar.parent
|
|
inObjects = False
|
|
for so in objects:
|
|
if(so == sPar):
|
|
inObjects = True
|
|
break
|
|
if inObjects == False:
|
|
objects.append(sPar)
|
|
|
|
file = open(fl, 'w')
|
|
for ob in selectedObjects:
|
|
exportData(ob)
|
|
file.write('}\n')
|
|
file.close()
|