Skip to content
Snippets Groups Projects
Commit 67ea44b8 authored by John Phan's avatar John Phan
Browse files

add support for single or multiple uv textures. fixed material id. Deal with... 

add support for single or multiple uv textures. fixed material id. Deal with faces assign to an materials.
parent b7f9bc36
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
io_import_scene_unreal_psk.py
+ 129
− 20
View file @ 67ea44b8
...@@ -146,7 +146,7 @@ def fixRoll(b): ...@@ -146,7 +146,7 @@ def fixRoll(b):
#align Z-axis #align Z-axis
b.roll -= math.degrees(math.atan2(v[0]*v[2]*(1 - v[1]),v[0]*v[0] + v[1]*v[2]*v[2])) b.roll -= math.degrees(math.atan2(v[0]*v[2]*(1 - v[1]),v[0]*v[0] + v[1]*v[2]*v[2]))
def pskimport(infile,importmesh,importbone,bDebugLogPSK): def pskimport(infile,importmesh,importbone,bDebugLogPSK,importmultiuvtextures):
global DEBUGLOG global DEBUGLOG
DEBUGLOG = bDebugLogPSK DEBUGLOG = bDebugLogPSK
print ("--------------------------------------------------") print ("--------------------------------------------------")
...@@ -243,7 +243,8 @@ def pskimport(infile,importmesh,importbone,bDebugLogPSK): ...@@ -243,7 +243,8 @@ def pskimport(infile,importmesh,importbone,bDebugLogPSK):
u2 = UVCoords[indata[2]][1] u2 = UVCoords[indata[2]][1]
v2 = UVCoords[indata[2]][2] v2 = UVCoords[indata[2]][2]
uv.append([u2,v2]) uv.append([u2,v2])
faceuv.append(uv) faceuv.append([uv,indata[3],indata[4],indata[5]])
#print("material:",indata[3])
#print("UV: ",u0,v0) #print("UV: ",u0,v0)
#update the uv var of the last item in the Tmsh.faces list #update the uv var of the last item in the Tmsh.faces list
# which is the face just added above # which is the face just added above
...@@ -270,9 +271,12 @@ def pskimport(infile,importmesh,importbone,bDebugLogPSK): ...@@ -270,9 +271,12 @@ def pskimport(infile,importmesh,importbone,bDebugLogPSK):
printlog("Nbr of MATT0000 records: " + str(recCount) + "\n" ) printlog("Nbr of MATT0000 records: " + str(recCount) + "\n" )
printlog(" - Not importing any material data now. PSKs are texture wrapped! \n") printlog(" - Not importing any material data now. PSKs are texture wrapped! \n")
counter = 0 counter = 0
materialcount = 0
while counter < recCount: while counter < recCount:
counter = counter + 1 counter = counter + 1
indata = unpack('64s6i',pskfile.read(88)) indata = unpack('64s6i',pskfile.read(88))
materialcount += 1
print("Material",counter)
## ##
#================================================================================================== #==================================================================================================
...@@ -530,31 +534,134 @@ def pskimport(infile,importmesh,importbone,bDebugLogPSK): ...@@ -530,31 +534,134 @@ def pskimport(infile,importmesh,importbone,bDebugLogPSK):
me_ob.faces.foreach_set("use_smooth", [False] * len(me_ob.faces)) me_ob.faces.foreach_set("use_smooth", [False] * len(me_ob.faces))
me_ob.update_tag() me_ob.update_tag()
"""
Material setup coding.
First the mesh has to be create first to get the uv texture setup working.
-Create material(s) list in the psk pack data from the list.(to do list)
-Append the material to the from create the mesh object.
-Create Texture(s)
-fae loop for uv assign and assign material index
"""
bpy.ops.object.mode_set(mode='OBJECT')
#===================================================================================================
#Material Setup
#===================================================================================================
print ("-------------------------")
print ("----Creating--Materials--")
print ("-------------------------")
materialname = "pskmat"
materials = []
for matcount in range(materialcount):
#if texturedata != None:
matdata = bpy.data.materials.new(materialname + str(matcount))
#mtex = matdata.texture_slots.new()
#mtex.texture = texture[matcount].data
#print(type(texture[matcount].data))
#print(dir(mtex))
#print(dir(matdata))
#for texno in range(len( bpy.data.textures)):
#print((bpy.data.textures[texno].name))
#print(dir(bpy.data.textures[texno]))
#matdata.active_texture = bpy.data.textures[matcount-1]
#matdata.texture_coords = 'UV'
#matdata.active_texture = texturedata
materials.append(matdata)
for material in materials:
#add material to the mesh list of materials
me_ob.materials.append(material)
#=================================================================================================== #===================================================================================================
#UV Setup #UV Setup
#=================================================================================================== #===================================================================================================
print ("-------------------------")
print ("-- Creating UV Texture --")
print ("-------------------------")
texture = [] texture = []
texturename = "text1" texturename = "text1"
#print(dir(bpy.data)) countm = 0
if (len(faceuv) > 0): #for countm in range(materialcount):
uvtex = me_ob.uv_textures.new() #add one uv texture #psktexname="psk" + str(countm)
for i, face in enumerate(me_ob.faces): #me_ob.uv_textures.new(name=psktexname)
blender_tface= uvtex.data[i] #face if importmultiuvtextures == True:
blender_tface.uv1 = faceuv[i][0] #uv = (0,0) me_ob.uv_textures.new(name="pskuvtexture")
blender_tface.uv2 = faceuv[i][1] #uv = (0,0) #print(dir(bpy.data))
blender_tface.uv3 = faceuv[i][2] #uv = (0,0) if (len(faceuv) > 0):
texture.append(uvtex) for countm in range(len(me_ob.uv_textures)):
me_ob.update()
uvtex = me_ob.uv_textures[countm] #add one uv texture
me_ob.update()
#print("UV TEXTURE NAME:",uvtex.name)
for i, face in enumerate(me_ob.faces):
blender_tface = uvtex.data[i] #face
mfaceuv = faceuv[i]
#print("---------------------------------------")
#print(faceuv[i][1])
#print(dir(face))
face.material_index = faceuv[i][1]
blender_tface.uv1 = mfaceuv[0][0] #uv = (0,0)
blender_tface.uv2 = mfaceuv[0][1] #uv = (0,0)
blender_tface.uv3 = mfaceuv[0][2] #uv = (0,0)
texture.append(uvtex)
else:
for countm in range(materialcount):
psktexname="psk" + str(countm)
me_ob.uv_textures.new(name=psktexname)
#psktexname="psk" + str(countm)
#me_ob.uv_textures.new(name=psktexname)
for countm in range(len(me_ob.uv_textures)):
me_ob.update()
#print(dir(me_ob.uv_textures))
#psktexname="psk" + str(countm)
uvtex = me_ob.uv_textures[countm] #add one uv texture
me_ob.update()
#print("UV TEXTURE NAME:",uvtex.name)
if (len(faceuv) > 0):
counttex = 0
countm = 0
for countm in range(len(me_ob.uv_textures)):
me_ob.update()
#print(dir(me_ob.uv_textures))
psktexname="psk" + str(countm)
uvtex = me_ob.uv_textures[countm] #add one uv texture
me_ob.update()
#print("UV TEXTURE NAME:",uvtex.name)
for i, face in enumerate(me_ob.faces):
blender_tface = uvtex.data[i] #face
mfaceuv = faceuv[i]
#print("---------------------------------------")
#print(faceuv[i][1])
#print(dir(face))
face.material_index = faceuv[i][1]
if countm == faceuv[i][1]:
face.material_index = faceuv[i][1]
blender_tface.uv1 = mfaceuv[0][0] #uv = (0,0)
blender_tface.uv2 = mfaceuv[0][1] #uv = (0,0)
blender_tface.uv3 = mfaceuv[0][2] #uv = (0,0)
else:
#set uv to zero (0,0)
#print("--------------------")
#print(blender_tface.uv1)
#print(blender_tface.uv2)
#print(blender_tface.uv2)
blender_tface.uv1 = [0,0]
#print(blender_tface.uv1)
blender_tface.uv2 = [0,0]
blender_tface.uv3 = [0,0]
texture.append(uvtex)
print("UV TEXTURE LEN:",len(texture))
#for tex in me_ob.uv_textures: #for tex in me_ob.uv_textures:
#print("mesh tex:",dir(tex)) #print("mesh tex:",dir(tex))
#print((tex.name)) #print((tex.name))
#=================================================================================================== #for face in me_ob.faces:
#Material Setup #print(dir(face))
#===================================================================================================
materialname = "mat"
materials = []
'''
matdata = bpy.data.materials.new(materialname) matdata = bpy.data.materials.new(materialname)
#color is 0 - 1 not in 0 - 255 #color is 0 - 1 not in 0 - 255
#matdata.mirror_color=(float(0.04),float(0.08),float(0.44)) #matdata.mirror_color=(float(0.04),float(0.08),float(0.44))
...@@ -574,6 +681,7 @@ def pskimport(infile,importmesh,importbone,bDebugLogPSK): ...@@ -574,6 +681,7 @@ def pskimport(infile,importmesh,importbone,bDebugLogPSK):
for material in materials: for material in materials:
#add material to the mesh list of materials #add material to the mesh list of materials
me_ob.materials.append(material) me_ob.materials.append(material)
'''
#=================================================================================================== #===================================================================================================
# #
#=================================================================================================== #===================================================================================================
...@@ -589,13 +697,13 @@ def pskimport(infile,importmesh,importbone,bDebugLogPSK): ...@@ -589,13 +697,13 @@ def pskimport(infile,importmesh,importbone,bDebugLogPSK):
print ("PSK2Blender completed") print ("PSK2Blender completed")
#End of def pskimport######################### #End of def pskimport#########################
def getInputFilename(filename,importmesh,importbone,bDebugLogPSK): def getInputFilename(filename,importmesh,importbone,bDebugLogPSK,importmultiuvtextures):
checktype = filename.split('\\')[-1].split('.')[1] checktype = filename.split('\\')[-1].split('.')[1]
print ("------------",filename) print ("------------",filename)
if checktype.upper() != 'PSK': if checktype.upper() != 'PSK':
print (" Selected file = ",filename) print (" Selected file = ",filename)
raise (IOError, "The selected input file is not a *.psk file") raise (IOError, "The selected input file is not a *.psk file")
pskimport(filename,importmesh,importbone,bDebugLogPSK) pskimport(filename,importmesh,importbone,bDebugLogPSK,importmultiuvtextures)
from bpy.props import * from bpy.props import *
...@@ -611,10 +719,11 @@ class IMPORT_OT_psk(bpy.types.Operator): ...@@ -611,10 +719,11 @@ class IMPORT_OT_psk(bpy.types.Operator):
filepath = StringProperty(name="File Path", description="Filepath used for importing the OBJ file", maxlen= 1024, default= "") filepath = StringProperty(name="File Path", description="Filepath used for importing the OBJ file", maxlen= 1024, default= "")
importmesh = BoolProperty(name="Mesh", description="Import mesh only. (not yet build.)", default=True) importmesh = BoolProperty(name="Mesh", description="Import mesh only. (not yet build.)", default=True)
importbone = BoolProperty(name="Bones", description="Import bones only. Current not working yet.", default=True) importbone = BoolProperty(name="Bones", description="Import bones only. Current not working yet.", default=True)
importmultiuvtextures = BoolProperty(name="Single UV Texture(s)", description="Single or Multi uv textures.", default=True)
bDebugLogPSK = BoolProperty(name="Debug Log.txt", description="Log the output of raw format. It will save in current file dir. Note this just for testing.", default=False) bDebugLogPSK = BoolProperty(name="Debug Log.txt", description="Log the output of raw format. It will save in current file dir. Note this just for testing.", default=False)
def execute(self, context): def execute(self, context):
getInputFilename(self.filepath,self.importmesh,self.importbone,self.bDebugLogPSK) getInputFilename(self.filepath,self.importmesh,self.importbone,self.bDebugLogPSK,self.importmultiuvtextures)
return {'FINISHED'} return {'FINISHED'}
def invoke(self, context, event): def invoke(self, context, event):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment