ABelliqueux/blender_io_export_psx_mesh
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use a dict for retrieving object's name from data

Browse Source
This commit is contained in:
ABelliqueux 2021-09-13 15:14:29 +02:00
parent c30a6aedb8
commit a2e9c2ffe1
1 changed files with 102 additions and 81 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

183
io_export_psx_tmesh.py
View File

@ -922,6 +922,69 @@ class ExportMyFormat(bpy.types.Operator, ExportHelper):
"VECTOR " + fileName + "_BKc = { " + BKr + ", " + BKg + ", " + BKb + ", 0 };\n\n" "VECTOR " + fileName + "_BKc = { " + BKr + ", " + BKg + ", " + BKb + ", 0 };\n\n"
) )
level_symbols.append( "VECTOR " + fileName + "_BKc" ) level_symbols.append( "VECTOR " + fileName + "_BKc" )
# Dictionaries
# Sound
# These speaker objects's positions will have to be updated
spkrParents = defaultdict(dict)
spkrOrphans = []
# array of Sound objects
soundFiles = []
# current XA files and channel
freeXAfile = 0
freeXAchannel = 0
# Lights
lmpObjects = {}
for obj in bpy.data.objects:
# Build a dictionary of objects that have child SPEAKER objects
if obj.type == 'SPEAKER':
if obj.data.sound is not None:
# and child of a mesh
if obj.parent is not None:
if obj.parent.type == 'MESH':
parent = obj.parent
# if parent exists in parent list, append to child list
# ~ if obj.parent.name in spkrParents:
# ~ spkrParents[obj.parent.name].append(obj.name)
# ~ else:
# if parent does not exist in list yet, create array
# ~ spkrParents[obj.parent.name] = [obj.name]
# has no parent
else:
# ~ spkrOrphans.append(obj.name)
parent = 0
# get sound informations
objName = obj.name
soundName = obj.data.sound.name
soundPath = bpy.path.abspath(obj.data.sound.filepath)
location = obj.location
volume = int(obj.data.volume)
volume_min = int(obj.data.volume_min)
volume_max = int(obj.data.volume_max)
# convert sound
if obj.data.get('isXA'):
XAsectorsize = 2336 if XAmode else 2352
if freeXAchannel > 7:
freeXAfile += 1
freeXAchannel = 0
convertedSoundPath = sound2XA(soundPath, soundName, bpp=4, XAfile=freeXAfile, XAchannel=freeXAchannel)
XAfile = freeXAfile
XAchannel = freeXAchannel
freeXAchannel += 1
if os.path.exists(convertedSoundPath):
XAsize = os.path.getsize(convertedSoundPath)
XAend = int((( XAsize / XAsectorsize ) - 1))
else:
XAsize = -1
XAend = -1
soundFiles.append( Sound( objName, soundName, soundPath, convertedSoundPath, parent, location, volume, volume_min, volume_max, -1, XAfile, XAchannel, XAsize, XAend ) )
else:
convertedSoundPath = sound2VAG(soundPath, soundName)
soundFiles.append( Sound( objName, soundName, soundPath, convertedSoundPath, parent, location, volume, volume_min, volume_max, -1 ) )
# Build dict of LAMPs objects
# We want to be able to find an object based on it's data name.
if obj.type == 'LAMP':
lmpObjects[obj.data.name] = obj.name
## Camera setup ## Camera setup
# List of points defining the camera path # List of points defining the camera path
camPathPoints = [] camPathPoints = []
@ -989,42 +1052,54 @@ class ExportMyFormat(bpy.types.Operator, ExportHelper):
# Light sources will be similar to Blender's sunlamp # Light sources will be similar to Blender's sunlamp
# A maximum of 3 light sources will be used # A maximum of 3 light sources will be used
# LLM : Local Light Matrix # LLM : Local Light Matrix
if len( bpy.data.lamps ) is not None: if len( lmpObjects ) is not None:
cnt = 0 cnt = 0
pad = 3 - len( bpy.data.lamps ) pad = 3 - len( lmpObjects ) if ( len( lmpObjects ) < 3 ) else 0
f.write( "MATRIX " + fileName + "_lgtmat = {\n") f.write( "MATRIX " + fileName + "_lgtmat = {\n")
for l in range(len(bpy.data.lamps)): for light in sorted(lmpObjects):
# Lightsource energy # Get rid of orphans
energy = int( bpy.data.lamps[ l ].energy * 4096 ) if bpy.data.lamps[light].users == 0:
# Get lightsource's world orientation continue
lightdir = bpy.data.objects[ bpy.data.lamps[ l ].name ].matrix_world * Vector( ( 0, 0, -1, 0 ) ) # PSX can only use 3 light sources
f.write( if cnt < 3 :
"\t" + str( int( lightdir.x * energy ) ) + "," + # Lightsource energy
"\t" + str( int( -lightdir.z * energy ) ) + "," + energy = int( bpy.data.lamps[light].energy * 4096 )
"\t" + str( int( lightdir.y * energy ) ) # ~ energy = int( light.energy * 4096 )
) # Get lightsource's world orientation
if l != len( bpy.data.lamps ) - 1: lightdir = bpy.data.objects[lmpObjects[light]].matrix_world * Vector( ( 0, 0, -1, 0 ) )
f.write(",\n") f.write(
# If less than 3 light sources exist in blender, fill the matrix with 0s. "\t" + str( int( lightdir.x * energy ) ) + "," +
if pad: "\t" + str( int( -lightdir.z * energy ) ) + "," +
f.write(",\n") "\t" + str( int( lightdir.y * energy ) )
while cnt < pad: )
f.write("\t0,0,0") if cnt < 2:
if cnt != pad: f.write(",\n")
f.write(",\n") # If less than 3 light sources exist in blender, fill the matrix with 0s.
if pad:
f.write(",\n")
while cnt < pad:
f.write("\t0,0,0")
if cnt != pad:
f.write(",\n")
cnt += 1
else:
cnt += 1 cnt += 1
f.write("\n\t};\n\n") f.write("\n\t};\n\n")
level_symbols.append( "MATRIX " + fileName + "_lgtmat" ) level_symbols.append( "MATRIX " + fileName + "_lgtmat" )
# LCM : Local Color Matrix # LCM : Local Color Matrix
f.write( "MATRIX " + fileName + "_cmat = {\n") f.write( "MATRIX " + fileName + "_cmat = {\n")
LCM = [] LCM = []
for l in bpy.data.lamps: cnt = 0
# If more than 3 light sources exists, use the 3 first in alphabetic order (same as in Blender's outliner)
for light in sorted(lmpObjects):
# If orphan, get on with it # If orphan, get on with it
if l.users == 0: if bpy.data.lamps[light].users == 0:
continue continue
LCM.append( str( int( l.color.r * 4096 ) if l.color.r else 0 ) ) if cnt < 3 :
LCM.append( str( int( l.color.g * 4096 ) if l.color.g else 0 ) ) LCM.append( str( int( bpy.data.lamps[light].color.r * 4096 ) if bpy.data.lamps[light].color.r else 0 ) )
LCM.append( str( int( l.color.b * 4096 ) if l.color.b else 0 ) ) LCM.append( str( int( bpy.data.lamps[light].color.g * 4096 ) if bpy.data.lamps[light].color.g else 0 ) )
LCM.append( str( int( bpy.data.lamps[light].color.b * 4096 ) if bpy.data.lamps[light].color.b else 0 ) )
cnt += 1
if len(LCM) < 9: if len(LCM) < 9:
while len(LCM) < 9: while len(LCM) < 9:
LCM.append('0') LCM.append('0')
@ -1778,61 +1853,7 @@ class ExportMyFormat(bpy.types.Operator, ExportHelper):
level_symbols.append( "CAMANGLE * " + fileName + "_camPtr" ) level_symbols.append( "CAMANGLE * " + fileName + "_camPtr" )
level_symbols.append( "NODE * " + fileName + "_curNode" ) level_symbols.append( "NODE * " + fileName + "_curNode" )
## Sound ## Sound
# Build a dictionary of objects that have child SPEAKER objects # Use dict generated earlier
# These objects's positions will have to be updated
spkrParents = defaultdict(dict)
spkrOrphans = []
# array of Sound objects
soundFiles = []
# current XA files and channel
freeXAfile = 0
freeXAchannel = 0
for obj in bpy.data.objects:
# if obj is a speaker
if obj.type == 'SPEAKER':
if obj.data.sound is not None:
# and child of a mesh
if obj.parent is not None:
if obj.parent.type == 'MESH':
parent = obj.parent
# if parent exists in parent list, append to child list
# ~ if obj.parent.name in spkrParents:
# ~ spkrParents[obj.parent.name].append(obj.name)
# ~ else:
# if parent does not exist in list yet, create array
# ~ spkrParents[obj.parent.name] = [obj.name]
# has no parent
else:
# ~ spkrOrphans.append(obj.name)
parent = 0
# get sound informations
objName = obj.name
soundName = obj.data.sound.name
soundPath = bpy.path.abspath(obj.data.sound.filepath)
location = obj.location
volume = int(obj.data.volume)
volume_min = int(obj.data.volume_min)
volume_max = int(obj.data.volume_max)
# convert sound
if obj.data.get('isXA'):
XAsectorsize = 2336 if XAmode else 2352
if freeXAchannel > 7:
freeXAfile += 1
freeXAchannel = 0
convertedSoundPath = sound2XA(soundPath, soundName, bpp=4, XAfile=freeXAfile, XAchannel=freeXAchannel)
XAfile = freeXAfile
XAchannel = freeXAchannel
freeXAchannel += 1
if os.path.exists(convertedSoundPath):
XAsize = os.path.getsize(convertedSoundPath)
XAend = int((( XAsize / XAsectorsize ) - 1))
else:
XAsize = -1
XAend = -1
soundFiles.append( Sound( objName, soundName, soundPath, convertedSoundPath, parent, location, volume, volume_min, volume_max, -1, XAfile, XAchannel, XAsize, XAend ) )
else:
convertedSoundPath = sound2VAG(soundPath, soundName)
soundFiles.append( Sound( objName, soundName, soundPath, convertedSoundPath, parent, location, volume, volume_min, volume_max, -1 ) )
# Default values # Default values
XAFiles = "0" XAFiles = "0"
VAGBank = "0" VAGBank = "0"