a05b2090a9
OBJ Importer, Added true FGon support for faces >4 verts. (Does a propper scanfill and optionaly converts it to an FGON.) - Checks for flipping, uses a face fan when scanfill fails. Added the option to split by materials, into seperate mesh objects (Good for large databases) Made smoothGroups optional.
293 lines
9.8 KiB
Python
293 lines
9.8 KiB
Python
# --------------------------------------------------------------------------
|
|
# BPyImage.py version 0.15
|
|
# --------------------------------------------------------------------------
|
|
# helper functions to be used by other scripts
|
|
# --------------------------------------------------------------------------
|
|
# ***** BEGIN GPL LICENSE BLOCK *****
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software Foundation,
|
|
# Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
#
|
|
# ***** END GPL LICENCE BLOCK *****
|
|
# --------------------------------------------------------------------------
|
|
|
|
#===========================================================================#
|
|
# Comprehensive image loader, will search and find the image #
|
|
# Will return a blender image or a new image if the image is missing #
|
|
#===========================================================================#
|
|
import Blender
|
|
from Blender import sys
|
|
try:
|
|
import os
|
|
except:
|
|
os=None
|
|
|
|
#==============================================#
|
|
# Return directory, where the file is #
|
|
#==============================================#
|
|
def stripFile(path):
|
|
lastSlash = max(path.rfind('\\'), path.rfind('/'))
|
|
if lastSlash != -1:
|
|
path = path[:lastSlash]
|
|
newpath= '%s%s' % (path, sys.sep)
|
|
else:
|
|
newpath= path
|
|
# print 'stripping paths:', path, 'to', newpath
|
|
return newpath
|
|
|
|
#==============================================#
|
|
# Strips the slashes from the back of a string #
|
|
#==============================================#
|
|
def stripPath(path):
|
|
return path.split('/')[-1].split('\\')[-1]
|
|
|
|
#====================================================#
|
|
# Strips the prefix off the name before writing #
|
|
#====================================================#
|
|
def stripExt(name): # name is a string
|
|
index = name.rfind('.')
|
|
if index != -1:
|
|
return name[ : index ]
|
|
else:
|
|
return name
|
|
|
|
def getExt(name):
|
|
index = name.rfind('.')
|
|
if index != -1:
|
|
return name[index+1:]
|
|
return name
|
|
|
|
#====================================================#
|
|
# Adds a slash to the end of a path if its not there #
|
|
#====================================================#
|
|
def addSlash(path):
|
|
if path.endswith('\\') or path.endswith('/'):
|
|
return path
|
|
return path + sys.sep
|
|
|
|
|
|
def comprehensiveImageLoad(imagePath, filePath, placeHolder= True, VERBOSE=True):
|
|
if VERBOSE: print 'img:', imagePath, 'file:', filePath
|
|
# When we have the file load it with this. try/except niceness.
|
|
def imageLoad(path):
|
|
#if path.endswith('\\') or path.endswith('/'):
|
|
# raise 'INVALID PATH'
|
|
try:
|
|
|
|
img = Blender.Image.Load(path)
|
|
if VERBOSE: print '\t\tImage loaded "%s"' % path
|
|
return img
|
|
except:
|
|
#raise "Helloo"
|
|
if VERBOSE:
|
|
if sys.exists(path): print '\t\tImage failed loading "%s", mabe its not a format blender can read.' % (path)
|
|
else: print '\t\tImage not found, making a place holder "%s"' % (path)
|
|
if placeHolder:
|
|
img= Blender.Image.New(stripPath(path),1,1,24)
|
|
|
|
img.setName(path.split('/')[-1].split('\\')[-1][0:21]) #path.split('/')[-1].split('\\')[-1][0:21]
|
|
img.filename= path
|
|
return Blender.Image.New(stripPath(path),1,1,24) #blank image
|
|
else:
|
|
return None
|
|
|
|
# Image formats blender can read
|
|
IMAGE_EXT = ['jpg', 'jpeg', 'png', 'tga', 'bmp', 'rgb', 'sgi', 'bw', 'iff', 'lbm', # Blender Internal
|
|
'gif', 'psd', 'tif', 'tiff', 'pct', 'pict', 'pntg', 'qtif'] # Quacktime, worth a try.
|
|
|
|
imageFileName = stripPath(imagePath) # image path only
|
|
imageFileName_lower = imageFileName.lower() # image path only
|
|
|
|
if VERBOSE: print '\tSearchingExisting Images for "%s"' % imagePath
|
|
for i in Blender.Image.Get():
|
|
if stripPath(i.filename.lower()) == imageFileName_lower:
|
|
if VERBOSE: print '\t\tUsing existing image.'
|
|
return i
|
|
|
|
|
|
if VERBOSE: print '\tAttempting to load "%s"' % imagePath
|
|
if sys.exists(imagePath):
|
|
if VERBOSE: print '\t\tFile found where expected "%s".' % imagePath
|
|
return imageLoad(imagePath)
|
|
|
|
|
|
|
|
imageFileName_noext = stripExt(imageFileName) # With no extension.
|
|
imageFileName_noext_lower = stripExt(imageFileName_lower) # With no extension.
|
|
imageFilePath = stripFile(imagePath)
|
|
|
|
# Remove relative path from image path
|
|
if imageFilePath.startswith('./') or imageFilePath.startswith('.\\'):
|
|
imageFilePath = imageFilePath[2:]
|
|
|
|
|
|
# Attempt to load from obj path.
|
|
tmpPath = stripFile(filePath) + stripPath(imageFileName)
|
|
if sys.exists(tmpPath):
|
|
if VERBOSE: print '\t\tFile found in path (1)"%s".' % tmpPath
|
|
return imageLoad(tmpPath)
|
|
|
|
|
|
# os needed if we go any further.
|
|
if os == None:
|
|
if VERBOSE: print '\t\tCreating a placeholder with a face path: "%s".' % imagePath
|
|
return imageLoad(imagePath) # Will jus treturn a placeholder.
|
|
|
|
|
|
# We have os.
|
|
# GATHER PATHS.
|
|
paths = {} # Store possible paths we may use, dict for no doubles.
|
|
tmpPath = addSlash(sys.expandpath('//')) # Blenders path
|
|
if sys.exists(tmpPath):
|
|
if VERBOSE: print '\t\tSearching in %s' % tmpPath
|
|
paths[tmpPath] = [os.listdir(tmpPath)] # Orig name for loading
|
|
paths[tmpPath].append([f.lower() for f in paths[tmpPath][0]]) # Lower case list.
|
|
paths[tmpPath].append([stripExt(f) for f in paths[tmpPath][1]]) # Lower case no ext
|
|
else:
|
|
if VERBOSE: print '\tNo Path: "%s"' % tmpPath
|
|
|
|
tmpPath = imageFilePath
|
|
if sys.exists(tmpPath):
|
|
if VERBOSE: print '\t\tSearching in %s' % tmpPath
|
|
paths[tmpPath] = [os.listdir(tmpPath)] # Orig name for loading
|
|
paths[tmpPath].append([f.lower() for f in paths[tmpPath][0]]) # Lower case list.
|
|
paths[tmpPath].append([stripExt(f) for f in paths[tmpPath][1]]) # Lower case no ext
|
|
else:
|
|
if VERBOSE: print '\tNo Path: "%s"' % tmpPath
|
|
|
|
tmpPath = stripFile(filePath)
|
|
if sys.exists(tmpPath):
|
|
if VERBOSE: print '\t\tSearching in %s' % tmpPath
|
|
paths[tmpPath] = [os.listdir(tmpPath)] # Orig name for loading
|
|
paths[tmpPath].append([f.lower() for f in paths[tmpPath][0]]) # Lower case list.
|
|
paths[tmpPath].append([stripExt(f) for f in paths[tmpPath][1]]) # Lower case no ext
|
|
else:
|
|
if VERBOSE: print '\tNo Path: "%s"' % tmpPath
|
|
|
|
tmpPath = addSlash(Blender.Get('texturesdir'))
|
|
if tmpPath and sys.exists(tmpPath):
|
|
if VERBOSE: print '\t\tSearching in %s' % tmpPath
|
|
paths[tmpPath] = [os.listdir(tmpPath)] # Orig name for loading
|
|
paths[tmpPath].append([f.lower() for f in paths[tmpPath][0]]) # Lower case list.
|
|
paths[tmpPath].append([stripExt(f) for f in paths[tmpPath][1]]) # Lower case no ext
|
|
else:
|
|
if VERBOSE: print '\tNo Path: "%s"' % tmpPath
|
|
|
|
# Add path if relative image patrh was given.
|
|
tmp_paths= paths.keys()
|
|
for k in tmp_paths:
|
|
tmpPath = k + imageFilePath
|
|
if sys.exists(tmpPath):
|
|
paths[tmpPath] = [os.listdir(tmpPath)] # Orig name for loading
|
|
paths[tmpPath].append([f.lower() for f in paths[tmpPath][0]]) # Lower case list.
|
|
paths[tmpPath].append([stripExt(f) for f in paths[tmpPath][1]]) # Lower case no ext
|
|
else:
|
|
if VERBOSE: print '\tNo Path: "%s"' % tmpPath
|
|
# DONE
|
|
|
|
print 'PATHs', paths
|
|
#
|
|
for path, files in paths.iteritems():
|
|
if sys.exists(path + imageFileName):
|
|
if VERBOSE: print '\tFound image at path: "%s" file" "%s"' % (path, imageFileName)
|
|
return imageLoad(path + imageFileName)
|
|
|
|
# If the files not there then well do a case insensitive seek.
|
|
filesOrigCase = files[0]
|
|
filesLower = files[1]
|
|
filesLowerNoExt = files[2]
|
|
|
|
# We are going to try in index the file directly, if its not there just keep on
|
|
|
|
index = None
|
|
try:
|
|
# Is it just a case mismatch?
|
|
index = filesLower.index(imageFileName_lower)
|
|
except:
|
|
try:
|
|
# Have the extensions changed?
|
|
index = filesLowerNoExt.index(imageFileName_noext_lower)
|
|
|
|
ext = getExt( filesLower[index] ) # Get the extension of the file that matches all but ext.
|
|
|
|
# Check that the ext is useable eg- not a 3ds file :)
|
|
if ext.lower() not in IMAGE_EXT:
|
|
index = None
|
|
|
|
except:
|
|
index = None
|
|
|
|
if index != None:
|
|
tmpPath = path + filesOrigCase[index]
|
|
img = imageLoad( tmpPath )
|
|
if img != None:
|
|
if VERBOSE: print '\t\tImage Found "%s"' % tmpPath
|
|
return img
|
|
|
|
|
|
# IMAGE NOT FOUND IN ANY OF THE DIRS!, DO A RECURSIVE SEARCH.
|
|
if VERBOSE: print '\t\tImage Not Found in any of the dirs, doing a recusrive search'
|
|
for path in paths.iterkeys():
|
|
# Were not going to use files
|
|
|
|
|
|
#------------------
|
|
# finds the file starting at the root.
|
|
# def findImage(findRoot, imagePath):
|
|
#W---------------
|
|
|
|
# ROOT, DIRS, FILES
|
|
pathWalk = os.walk(path)
|
|
pathList = [True]
|
|
|
|
matchList = [] # Store a list of (match, size), choose the biggest.
|
|
while True:
|
|
try:
|
|
pathList = pathWalk.next()
|
|
except:
|
|
break
|
|
|
|
for file in pathList[2]:
|
|
file_lower = file.lower()
|
|
# FOUND A MATCH
|
|
if (file_lower == imageFileName_lower) or\
|
|
(stripExt(file_lower) == imageFileName_noext_lower and getExt(file_lower) in IMAGE_EXT):
|
|
name = pathList[0] + sys.sep + file
|
|
size = os.path.getsize(name)
|
|
if VERBOSE: print '\t\t\tfound:', name
|
|
matchList.append( (name, size) )
|
|
|
|
if matchList:
|
|
# Sort by file size
|
|
matchList.sort(lambda A, B: cmp(B[1], A[1]) )
|
|
|
|
if VERBOSE: print '\t\tFound "%s"' % matchList[0][0]
|
|
|
|
# Loop through all we have found
|
|
img = None
|
|
for match in matchList:
|
|
img = imageLoad(match[0]) # 0 - first, 0 - pathname
|
|
if img != None:
|
|
break
|
|
return img
|
|
|
|
# No go.
|
|
if VERBOSE: print '\t\tImage Not Found after looking everywhere! "%s"' % imagePath
|
|
return imageLoad(imagePath) # Will jus treturn a placeholder.
|
|
|
|
|
|
|
|
|
|
|