forked from bartvdbraak/blender
65895ba6bd
BPyRender.imageFromObjectsOrtho() returns an image Made all image bake scripts ask before overwriting a file. as well as displaying the newly created image once its rendered.
127 lines
3.1 KiB
Python
127 lines
3.1 KiB
Python
#!BPY
|
|
"""
|
|
Name: 'Bake Normal Map Image from UVs'
|
|
Blender: 242
|
|
Group: 'Image'
|
|
Tooltip: 'Generate a normal map image from selected mesh objects.'
|
|
"""
|
|
__author__= ['Campbell Barton']
|
|
__url__= ('blender', 'elysiun', 'http://www.gametutorials.com')
|
|
__version__= '0.1'
|
|
__bpydoc__= '''\
|
|
Bake Vertex Normals to an image
|
|
|
|
This script makes an image from a meshes vertex normals, using the UV coordinates
|
|
to draw the faces into the image.
|
|
|
|
Make sure your UV Coordinates do not overlap.
|
|
LSCM Unwrapper or archimap unwrapper work well to automaticaly do this.
|
|
'''
|
|
|
|
import Blender
|
|
import BPyRender
|
|
#reload(BPyRender)
|
|
import BPyMessages
|
|
Vector= Blender.Mathutils.Vector
|
|
Create= Blender.Draw.Create
|
|
|
|
|
|
def main():
|
|
# Create the variables.
|
|
# Filename without path or extension.
|
|
# Big LC, gets all unique mesh objects from the selection that have UV coords.
|
|
|
|
def worldnormals(ob):
|
|
nor_mtx= ob.matrixWorld.rotationPart()
|
|
me= ob.getData(mesh=1)
|
|
for v in me.verts:
|
|
v.no= v.no*nor_mtx
|
|
return me
|
|
|
|
ob_s= dict([\
|
|
(ob.getData(name_only=1), ob)\
|
|
for ob in Blender.Object.GetSelected()\
|
|
if ob.getType()=='Mesh' if ob.getData(mesh=1).faceUV]).values()
|
|
|
|
me_s= [worldnormals(ob) for ob in ob_s]
|
|
del ob_s
|
|
|
|
if not me_s:
|
|
BPyMessages.Error_NoMeshUvSelected()
|
|
return
|
|
|
|
newpath= Blender.Get('filename').split('/')[-1].split('\\')[-1].replace('.blend', '_nor.png')
|
|
###PREF_IMAGE_PATH = Create('//%s_nor' % newpath)
|
|
PREF_IMAGE_SIZE = Create(512)
|
|
PREF_IMAGE_BLEED = Create(4)
|
|
PREF_IMAGE_SMOOTH= Create(0)
|
|
|
|
PREF_SEL_FACES_ONLY= Create(0)
|
|
|
|
pup_block = [\
|
|
###('', PREF_IMAGE_PATH, 3, 100, 'Path to new Image. "//" for curent blend dir.'),\
|
|
('Pixel Size:', PREF_IMAGE_SIZE, 64, 4096, 'Image Width and Height.'),\
|
|
('Pixel Bleed:', PREF_IMAGE_BLEED, 0, 64, 'Extend pixels from boundry edges to avoid mipmapping errors on rendering.'),\
|
|
('Smooth lines', PREF_IMAGE_SMOOTH, 'Render smooth lines.'),\
|
|
'',\
|
|
('Selected Faces only', PREF_SEL_FACES_ONLY, 'Only bake from selected faces.'),\
|
|
]
|
|
|
|
if not Blender.Draw.PupBlock('Texface Image Bake', pup_block):
|
|
# Update the normals before we exit
|
|
for me in me_s:
|
|
me.update()
|
|
return
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Defaults for VCol, user cant change
|
|
PREF_IMAGE_WIRE= False
|
|
PREF_IMAGE_WIRE_INVERT= False
|
|
PREF_IMAGE_WIRE_UNDERLAY= False
|
|
|
|
PREF_USE_IMAGE= False
|
|
PREF_USE_VCOL= False
|
|
PREF_USE_MATCOL= False
|
|
PREF_USE_TEXTURE= False
|
|
|
|
PREF_USE_NORMAL= True # of course we need this one
|
|
def file_sel(PREF_IMAGE_PATH):
|
|
|
|
if not (BPyMessages.Warning_SaveOver(PREF_IMAGE_PATH) and \
|
|
BPyMessages.Warning_SaveOver(PREF_IMAGE_PATH+'.png')):
|
|
return
|
|
|
|
image = BPyRender.vcol2image(me_s,\
|
|
PREF_IMAGE_PATH,\
|
|
PREF_IMAGE_SIZE.val,\
|
|
PREF_IMAGE_BLEED.val,\
|
|
PREF_IMAGE_SMOOTH.val,\
|
|
PREF_IMAGE_WIRE,\
|
|
PREF_IMAGE_WIRE_INVERT,\
|
|
PREF_IMAGE_WIRE_UNDERLAY,\
|
|
PREF_USE_IMAGE,\
|
|
PREF_USE_VCOL,\
|
|
PREF_USE_MATCOL,\
|
|
PREF_USE_NORMAL,\
|
|
PREF_USE_TEXTURE,\
|
|
PREF_SEL_FACES_ONLY.val)
|
|
|
|
# Restore normals
|
|
for me in me_s:
|
|
me.update()
|
|
|
|
if image:
|
|
image.makeCurrent()
|
|
|
|
Blender.Window.RedrawAll()
|
|
|
|
Blender.Window.FileSelector(file_sel, 'SAVE PNG', newpath)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|