257 lines
8.0 KiB
Python
257 lines
8.0 KiB
Python
|
'''
|
||
|
|
UI functionality for the material manager.
|
||
|
|
'''
|
||
|
|
from win32com.client import constants as const
|
||
|
|
import win32com.client
|
||
|
|
xsi = Application
|
||
|
|
addonpath = xsi.InstallationPath(const.siUserAddonPath)
|
||
|
|
PROJECTPATH = xsi.InstallationPath(const.siProjectPath)
|
||
|
|
|
||
|
|
|
||
|
|
def close_mat_OnClicked():
|
||
|
|
PPG.Close()
|
||
|
|
xsi.DeleteObj('MaterialEdit')
|
||
|
|
|
||
|
|
|
||
|
|
def help_mat_OnClicked():
|
||
|
|
ps = xsi.ActiveSceneRoot.AddProperty('CustomProperty', False, 'MatManagerHelp')
|
||
|
|
lay = ps.PPGLayout
|
||
|
|
lay.Language = 'pythonscript'
|
||
|
|
agr = lay.AddGroup
|
||
|
|
egr = lay.EndGroup
|
||
|
|
text = lay.AddStaticText
|
||
|
|
agr('Create', 1)
|
||
|
|
text('''Create a phong material(the only supported material type)
|
||
|
|
with the specified name(line edit to the left of this button.''')
|
||
|
|
egr()
|
||
|
|
agr('Edit', 1)
|
||
|
|
text('''Inspect the first selected material ZE flags and options can be
|
||
|
|
found here, too.''')
|
||
|
|
egr()
|
||
|
|
agr('ZEify / De-ZEify', 1)
|
||
|
|
text('''Adds/Removes Zero Engine flags and material settings to the
|
||
|
|
selected material(s).''')
|
||
|
|
egr()
|
||
|
|
agr('Remove', 1)
|
||
|
|
text('''Removes the selected material(s).''')
|
||
|
|
egr()
|
||
|
|
agr('Assign Tex', 1)
|
||
|
|
text('''A shortcut to assign a texture to the diffuse slot of
|
||
|
|
the selected material. This will launch a file browser.''')
|
||
|
|
egr()
|
||
|
|
agr('Assign / Unassign', 1)
|
||
|
|
text('''Assigns/unassigns the first selected material to the
|
||
|
|
currently selected models.''')
|
||
|
|
egr()
|
||
|
|
xsi.InspectObj(ps, '', 'MatManagerHelp', 4, False)
|
||
|
|
for prop in xsi.ActiveSceneRoot.Properties:
|
||
|
|
if prop.Name == 'MatManagerHelp':
|
||
|
|
xsi.DeleteObj('MatManagerHelp')
|
||
|
|
|
||
|
|
|
||
|
|
def del_mat_OnClicked():
|
||
|
|
materials = get_scene_materials()
|
||
|
|
ppg = PPG.Inspected(0)
|
||
|
|
sel = ppg.Parameters('materials').Value
|
||
|
|
split = sel.split(';')
|
||
|
|
lay = ppg.PPGLayout
|
||
|
|
listbox = lay.Item('materials')
|
||
|
|
items = listbox.UIItems
|
||
|
|
items2 = list(items[:])
|
||
|
|
for matname in split:
|
||
|
|
for mat in materials:
|
||
|
|
if mat.Name == matname:
|
||
|
|
xsi.DeleteObj(mat)
|
||
|
|
break
|
||
|
|
refresh()
|
||
|
|
PPG.Refresh()
|
||
|
|
|
||
|
|
|
||
|
|
def create_mat_OnClicked():
|
||
|
|
ppg = PPG.Inspected(0)
|
||
|
|
name = ppg.Parameters('mat_name').Value
|
||
|
|
args = '$XSI_DSPRESETS\\Shaders\\Material\\Phong', name, '', '', False, ''
|
||
|
|
xsi.SICreateMaterial(*args)
|
||
|
|
refresh()
|
||
|
|
|
||
|
|
|
||
|
|
def edit_mat_OnClicked():
|
||
|
|
materials = get_scene_materials()
|
||
|
|
ppg = PPG.Inspected(0)
|
||
|
|
sel = ppg.Parameters('materials').Value
|
||
|
|
split = sel.split(';')
|
||
|
|
for mat in materials:
|
||
|
|
if mat.Name == split[0]:
|
||
|
|
xsi.InspectObj(mat)
|
||
|
|
|
||
|
|
|
||
|
|
def add_flags_OnClicked():
|
||
|
|
materials = get_scene_materials()
|
||
|
|
ppg = PPG.Inspected(0)
|
||
|
|
sel = ppg.Parameters('materials').Value
|
||
|
|
split = sel.split(';')
|
||
|
|
for matname in split:
|
||
|
|
for mat in materials:
|
||
|
|
if mat.Name == matname:
|
||
|
|
if not get_msh_material_property(mat):
|
||
|
|
add_msh_material_flags(mat)
|
||
|
|
break
|
||
|
|
refresh()
|
||
|
|
|
||
|
|
|
||
|
|
def del_flags_OnClicked():
|
||
|
|
materials = get_scene_materials()
|
||
|
|
ppg = PPG.Inspected(0)
|
||
|
|
sel = ppg.Parameters('materials').Value
|
||
|
|
split = sel.split(';')
|
||
|
|
for matname in split:
|
||
|
|
for mat in materials:
|
||
|
|
if mat.Name == matname:
|
||
|
|
prop = get_msh_material_property(mat)
|
||
|
|
if prop:
|
||
|
|
xsi.DeleteObj(prop)
|
||
|
|
break
|
||
|
|
refresh()
|
||
|
|
|
||
|
|
|
||
|
|
def unassign_mat_OnClicked():
|
||
|
|
if xsi.Selection.Count == 0:
|
||
|
|
msg_box('No objects selected.')
|
||
|
|
return
|
||
|
|
xsi.SIUnAssignMaterial(xsi.Selection)
|
||
|
|
|
||
|
|
|
||
|
|
def assign_mat_OnClicked():
|
||
|
|
materials = get_scene_materials()
|
||
|
|
ppg = PPG.Inspected(0)
|
||
|
|
sel = ppg.Parameters('materials').Value
|
||
|
|
split = sel.split(';')
|
||
|
|
mat2assign = None
|
||
|
|
for mat in materials:
|
||
|
|
if mat.Name == split[0]:
|
||
|
|
mat2assign = mat
|
||
|
|
break
|
||
|
|
if xsi.Selection.Count == 0:
|
||
|
|
msg_box('No objects selected.')
|
||
|
|
return
|
||
|
|
for item in xsi.Selection:
|
||
|
|
if item.Type == 'polySubComponent':
|
||
|
|
cls = item.SubComponent.CreateCluster('matpolycls')
|
||
|
|
xsi.SIAssignMaterial(cls, mat2assign)
|
||
|
|
elif item.Type == 'poly':
|
||
|
|
xsi.SIAssignMaterial(item, mat2assign)
|
||
|
|
elif item.Type == 'polymsh':
|
||
|
|
xsi.SIAssignMaterial(item, mat2assign)
|
||
|
|
refresh()
|
||
|
|
|
||
|
|
|
||
|
|
def assign_tex_OnClicked():
|
||
|
|
materials = get_scene_materials()
|
||
|
|
ppg = PPG.Inspected(0)
|
||
|
|
sel = ppg.Parameters('materials').Value
|
||
|
|
split = sel.split(';')
|
||
|
|
mats = []
|
||
|
|
for matname in split:
|
||
|
|
for mat in materials:
|
||
|
|
if mat.Name == matname:
|
||
|
|
mats.append(mat)
|
||
|
|
break
|
||
|
|
xsiui = win32com.client.Dispatch('XSI.UIToolkit')
|
||
|
|
fb = xsiui.FileBrowser
|
||
|
|
fb.DialogTitle = 'Select a .tga file'
|
||
|
|
fb.InitialDirectory = PROJECTPATH
|
||
|
|
fb.Filter = 'TARGA (*.tga)|*.tga||'
|
||
|
|
fb.ShowOpen()
|
||
|
|
img_filepath = fb.FilePathName
|
||
|
|
if img_filepath:
|
||
|
|
for mat in mats:
|
||
|
|
shader = xsi.CreateShaderFromPreset('$XSI_DSPRESETS\\Shaders\\Texture\\Image.Preset', mat)
|
||
|
|
img_clip = xsi.SICreateImageClip2(img_filepath)
|
||
|
|
xsi.SIConnectShaderToCnxPoint(img_clip, shader.tex, False)
|
||
|
|
xsi.SIConnectShaderToCnxPoint(shader, mat.Shaders(0).Parameters('diffuse'), False)
|
||
|
|
##############################################################################
|
||
|
|
###################### HELPER FUNCTIONS ##################################
|
||
|
|
|
||
|
|
|
||
|
|
def refresh():
|
||
|
|
mats = get_scene_materials('variants')
|
||
|
|
PPG.Inspected(0).PPGLayout.Item('materials').UIItems = mats
|
||
|
|
PPG.Refresh()
|
||
|
|
|
||
|
|
|
||
|
|
def msg_box(message):
|
||
|
|
XSIUIToolkit.MsgBox(message, 0, 'XSIZETools')
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
def get_msh_material_property(material):
|
||
|
|
for prop in material.Properties:
|
||
|
|
if 'ZeroEngine_Flags' in prop.Name:
|
||
|
|
return prop
|
||
|
|
|
||
|
|
|
||
|
|
def add_msh_material_flags(mat):
|
||
|
|
pset = mat.AddProperty('CustomProperty', False, 'ZeroEngine Flags')
|
||
|
|
pset.AddParameter3('tex1', const.siString)
|
||
|
|
pset.AddParameter3('tex2', const.siString)
|
||
|
|
pset.AddParameter3('tex3', const.siString)
|
||
|
|
|
||
|
|
pset.AddParameter3('emissive', const.siBool, 0, '', '', 0)
|
||
|
|
pset.AddParameter3('glow', const.siBool, 0, '', '', 0)
|
||
|
|
pset.AddParameter3('transparency', const.siInt4, 0, 0, 2, 0)
|
||
|
|
pset.AddParameter3('hardedged', const.siBool, 0, '', '', 0)
|
||
|
|
pset.AddParameter3('perpixel', const.siBool, 0, '', '', 0)
|
||
|
|
pset.AddParameter3('additive', const.siBool, 0, '', '', 0)
|
||
|
|
pset.AddParameter3('specular', const.siBool, 0, '', '', 0)
|
||
|
|
pset.AddParameter3('rendertype', const.siInt4, 0, 0, 31, 0)
|
||
|
|
pset.AddParameter3('data0', const.siInt4, 0, 0, 255, 0)
|
||
|
|
pset.AddParameter3('data1', const.siInt4, 0, 0, 255, 0)
|
||
|
|
|
||
|
|
lay = pset.PPGLayout
|
||
|
|
|
||
|
|
lay.AddGroup('ZeroEngine Material Flags', 1)
|
||
|
|
lay.AddGroup('Additional Textures', 1)
|
||
|
|
lay.AddItem('tex1', 'Texture 1')
|
||
|
|
lay.AddItem('tex2', 'Texture 2')
|
||
|
|
lay.AddItem('tex3', 'Texture 3')
|
||
|
|
lay.EndGroup()
|
||
|
|
lay.AddGroup('Flags', 1)
|
||
|
|
lay.AddItem('emissive', 'Emissive')
|
||
|
|
lay.AddItem('glow', 'Glow')
|
||
|
|
lay.AddItem('transparency', 'Transparency')
|
||
|
|
lay.AddItem('hardedged', 'Hardedged Transparency')
|
||
|
|
lay.AddItem('perpixel', 'Per-Pixel Lighting')
|
||
|
|
lay.AddItem('additive', 'Additive Transparency')
|
||
|
|
lay.AddItem('specular', 'Specular')
|
||
|
|
lay.AddItem('rendertype', 'RenderType')
|
||
|
|
lay.AddItem('data0', 'Data0')
|
||
|
|
lay.AddItem('data1', 'Data1')
|
||
|
|
lay.EndGroup()
|
||
|
|
lay.EndGroup()
|
||
|
|
|
||
|
|
|
||
|
|
def get_scene_materials(listtype='list'):
|
||
|
|
mats = []
|
||
|
|
matlibs = xsi.ActiveProject.ActiveScene.MaterialLibraries
|
||
|
|
for lib in matlibs:
|
||
|
|
for mat in lib.Items:
|
||
|
|
mats.append(mat)
|
||
|
|
if listtype == 'variants':
|
||
|
|
newmats = []
|
||
|
|
for index, mat in enumerate(mats):
|
||
|
|
name = [mat.Name]
|
||
|
|
prop = get_msh_material_property(mat)
|
||
|
|
usedby = mat.UsedBy
|
||
|
|
if usedby.Count > 0:
|
||
|
|
if prop:
|
||
|
|
name.append(' - ZE | {0}'.format(usedby.Count))
|
||
|
|
else:
|
||
|
|
name.append(' - {0}'.format(usedby.Count))
|
||
|
|
else:
|
||
|
|
if prop:
|
||
|
|
name.append(' - ZE')
|
||
|
|
newmats.append(''.join(name))
|
||
|
|
newmats.append(mat.Name)
|
||
|
|
return newmats
|
||
|
|
return mats
|