blender/release/scripts/uvpaint.py
Willian Padovani Germano b970eadedf Scripts:
- updating some bundled scripts, thanks to authors Jean-Michel Soler, Campbell Barton and Anthony D'Agostino.

BPython:
- removing wrong fix from BGL.c's glDrawPixels.

note: applied guitargeek's setName patch to Blender.Key, but saw that he updated it with more functionality and assigned to stivs, so I won't commit this old version.
2005-10-11 02:32:58 +00:00

681 lines
18 KiB
Python
Raw Blame History

#!BPY
""" Registration info for Blender menus: <- these words are ignored
Name: 'UVpainter'
Blender: 232
Group: 'UV'
Tip: 'Use vertex paint color value to fill uvmapping'
"""
__author__ = "Jean-Michel Soler (jms)"
__url__ = ("blender", "elysiun",
"Script's homepage, http://jmsoler.free.fr/didacticiel/blender/tutor/cpl_uvpainting.htm",
"Communicate problems and errors, http://www.zoo-logique.org/3D.Blender/newsportal/thread.php?group=3D.Blender")
__version__ = "0.8 08/2005"
__bpydoc__ = """\
This script "paints" uv-mappings with the model's vertex colors.
Usage:
With this script you can export uv-maps filled with vertex colors to TARGA
(.tga) images. To use it the mesh must have proper uv coordinates assigned
in UV Face Select Mode. And to fill the projected faces with color, the mesh
can also have material(s) and vertex colors painted on it.
The script has a GUI with a preview of the results and options like drawing
lines or not, defining size, etc. You can paint vertex colors in the mesh and
see the uv-map updated in the script's window.
Notes:<br>
Material's rgb color is also used to fill the uv-map;<br>
If there are no vertex colors or texture faces in the mesh and you press
the "Make" VColors button in the edit mesh buttons win, the current light setup
is saved as vertex colors for the model;<br>
Check the script's homepage for example images.
Short keys Documentation
KEYS
M : dipslay GUI Menu
D : Set/Unset Documentation
S : Save current window content
in a tga file
Q or ESC : Exit
T : Set/Unset Transparency
L : Set/Unset lines
E : Set/Unset outline
B : Set lines color to Black
W : Set lines color to white
ARROW : displace model on
UP/DOWN/LEFT/RIGHT side
PADPLUS : increase ZOOM
PADMINUS : decrease ZOOM
HOME : cancel display modifs
Mouse button
RIGHTMOUSE : same as arrows
"""
# $Id$
#
#----------------------------------------------
# uvpainter script (c) 04/2004 jean-michel soler
# http://jmsoler.free.fr/util/blenderfile/py/UVpaint05.zip
# this script is released under GPL licence
# for the Blender 2.33 scripts distribution
#----------------------------------------------
# Official page :
# http://jmsoler.free.fr/didacticiel/blender/tutor/cpl_uvpainting.htm
# Communicate problems and errors on:
# http://www.zoo-logique.org/3D.Blender/newsportal/thread.php?group=3D.Blender
#----------------------------------------------
# Page officielle :
# http://jmsoler.free.fr/didacticiel/blender/tutor/cpl_uvpainting.htm
# Communiquer les problemes et erreurs sur:
# http://www.zoo-logique.org/3D.Blender/newsportal/thread.php?group=3D.Blender
#---------------------------------------------
# ce script est propos<6F> sous licence GPL pour etre associe
# a la distribution de Blender 2.33 et suivant
# --------------------------------------------------------------------------
# this script is released under GPL licence
# for the Blender 2.33 scripts package
# --------------------------------------------------------------------------
# ***** BEGIN GPL LICENSE BLOCK *****
#
# Script copyright (C) 2003, 2004: Jean-Michel Soler
#
# 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 *****
# --------------------------------------------------------------------------
import Blender
from Blender.Draw import *
from Blender.BGL import *
from Blender.NMesh import *
try:
import nt
os=nt
except:
import posix
os=posix
def exist(path):
try:
pth=os.stat(Blender.sys.dirname(path))
except:
return 0
return 1
loc0= Blender.Get ("filename").replace('\\','/')
loc0=loc0[:loc0.rfind('/')]
loc2=loc0+'/'+'test00.tga'
mouse_x,mouse_y=0,0
mouse_xr=1
mouse_yr=1
POS=[0,0]
ANC=[0,0]
XY=[0,0]
size=[]
sel=0
X,Y=0,0
TRANSP,EMPTY,DOCU=0,0,0
MENU, SAVE =1,0
glCr=glRasterPos2d
glCl3=glColor3f
glCl4=glColor4f
glRct=glRectf
LC=1.0
xlimit=0
selmatlist=[]
LIM=[0.0,0.0,0.0,0.0]
NOLIM=1
if not NOLIM : LIM=[-1.0,1.0,-1.0,1.0]
TR=0.8
def Doc(size):
S0,S1=40,50
a=[S0,size[3]-S1, .8,.8,.8]
b=[S0*7,size[3]-S1, .8,0.8,0.8]
c=[S0*7,size[3]-S1*7, 0.8,0.8,0.8]
d=[S0,size[3]-S1*7, 0.8,0.8,0.8]
Tcarre(a,b,c,d,0.8)
Lcarre(a,b,c,d,0.0)
DOC=[' ',
'Documentation',
' ',
'KEYS ',
'M : dipslay GUI Menu',
'D : Set/Unset Documentation',
'S : Save current window content',
' in a tga file',
'Q or ESC : Exit',
'T : Set/Unset Transparency',
'L : Set/Unset lines',
'E : Set/Unset outline',
'B : Set lines color to Black ',
'W : Set lines color to white',
'ARROW : displace model on ',
' UP/DOWN/LEFT/RIGHT side' ,
'PADPLUS : increase ZOOM ',
'PADMINUS : decrease ZOOM ',
'HOME : cancel display modifs',
' ',
'Mouse button',
'RIGHTMOUSE : same as arrows',
]
glColor3f(0.0,0.0,0.0)
for D in DOC :
glRasterPos2f(S0+8, size[3]-S1-13*DOC.index(D))
Text(D)
def Ttriangle(a,b,c):
glBegin(GL_TRIANGLES);
glColor4f(a[2],a[3],a[4],TR)
glVertex2f(a[0],a[1]);
glColor4f(b[2],b[3],b[4],TR)
glVertex2f(b[0],b[1]);
glColor4f(c[2],c[3],c[4],TR)
glVertex2f(c[0],c[1]);
glEnd();
def Ftriangle(a,b,c):
glBegin(GL_TRIANGLES);
glColor3f(a[2],a[3],a[4])
glVertex2f(a[0],a[1]);
glColor3f(b[2],b[3],b[4])
glVertex2f(b[0],b[1]);
glColor3f(c[2],c[3],c[4])
glVertex2f(c[0],c[1]);
glEnd();
def Ltriangle(a,b,c,LC=0.5):
TL=[a,b,c,a]
for v in [0,1,2] :
glBegin(GL_LINES);
glColor4f(LC,LC,LC,TR+0.2)
glVertex2f(TL[v][0],TL[v][1]);
glVertex2f(TL[v+1][0],TL[v+1][1]);
glEnd();
def Tcarre(a,b,c,d,LC=1.0):
Ttriangle(a,b,c)
Ttriangle(a,c,d)
def Fcarre(a,b,c,d,LC=1.0):
Ftriangle(a,b,c)
Ftriangle(a,c,d)
def Lcarre(a,b,c,d,LC=0.5):
TL=[a,b,c,d,a]
for v in [0,1,2,3] :
glBegin(GL_LINES);
glColor4f(LC,LC,LC,TR+0.2)
glVertex2f(TL[v][0],TL[v][1]);
glVertex2f(TL[v+1][0],TL[v+1][1]);
glEnd();
def transface(f,x,y,u=0.0, v=0.0):
global xlimit, LIM
global mouse_xr,sel, ANC, X,Y
global mouse_yr, POS, XY,size
global mouse_x, mouse_y
mouse_x=mouse_xr-size[0]
mouse_y=mouse_yr-size[1]
if sel==1:
POS=[mouse_x-ANC[0],mouse_y-ANC[1]]
u,v=POS
a=[0,0,0.0, 0.0,0.0,0.0]
b=[0,0,0.0, 0.0,0.0,0.0]
c=[0,0,0.0, 0.0,0.0,0.0]
d=[0,0,0.0, 0.0,0.0,0.0]
if len(f.v)>=3:
a[0]=int((f.uv[0][0]-LIM[1])*x+u)
a[1]=int((f.uv[0][1]-LIM[3])*y+v)
if a[0]>xlimit:
xlimit=a[0]
a[2]=f.col[0].r/255.0
a[3]=f.col[0].g/255.0
a[4]=f.col[0].b/255.0
c[0]=int((f.uv[2][0]-LIM[1])*x+u)
c[1]=int((f.uv[2][1]-LIM[3])*y+v)
if c[0]>xlimit:
xlimit=c[0]
c[2]=f.col[2].r/255.0
c[3]=f.col[2].g/255.0
c[4]=f.col[2].b/255.0
b[0]=int((f.uv[1][0]-LIM[1])*x+u)
b[1]=int((f.uv[1][1]-LIM[3])*y+v)
if b[0]>xlimit:
xlimit=b[0]
b[2]=f.col[1].r/255.0
b[3]=f.col[1].g/255.0
b[4]=f.col[1].b/255.0
if len(f.v)==4:
d[0]=int((f.uv[3][0]-LIM[1])*x+u)
d[1]=int((f.uv[3][1]-LIM[3])*y+v)
if d[0]>xlimit:
xlimit=d[0]
d[2]=f.col[3].r/255.0
d[3]=f.col[3].g/255.0
d[4]=f.col[3].b/255.0
else:
d=0
#print a,b,c
return a,b,c,d
def extract_faces(me,MENU):
global TMATList, selmatlist
if MENU==2:
listf=[]
for f in me.faces:
if f.mat in selmatlist:
listf.append(f)
return listf
def affiche_mesh(ME,x,y):
global LINE,xlimit,MMENU,XLIMIT,xwin,xlimit,LC
global LIM, EMPTY,TRANSP
if not NOLIM : LIM=[-1.0,1.0,-1.0,1.0]
if ME.getType()=='Mesh':
me=ME.getData()
if MMENU.val==1:
se=me.faces
elif MMENU.val==3:
se=[s for s in me.faces if s in me.getSelectedFaces() or s.sel]
elif MMENU.val==2:
se=extract_faces(me,2)
if not NOLIM :
for s in se:
for u in s.uv:
if u[0] >LIM[0] : LIM[0]=u[0]
if u[0] <LIM[1] : LIM[1]=u[0]
if u[1] >LIM[2] : LIM[2]=u[1]
if u[1] <LIM[3] : LIM[3]=u[1]
xlimit=0
for f in se:
a,b,c,d=transface(f,x,y)
if not EMPTY and not TRANSP:
if len(f.v)==4:
Ftriangle(a,b,c)
Ftriangle(a,c,d)
elif len(f.v)==3:
Ftriangle(a,b,c)
elif not EMPTY :
if len(f.v)==4:
Ttriangle(a,b,c)
Ttriangle(a,c,d)
elif len(f.v)==3:
Ttriangle(a,b,c)
if LINE.val==1 or EMPTY:
for f in se:
a,b,c,d=transface(f,x,y)
if len(f.v)==4:
Lcarre(a,b,c,d,LC)
elif len(f.v)==3:
Ltriangle(a,b,c,LC)
if XLIMIT.val==0:
Lcarre([1,1],[1,y-2],[xlimit+2,y-2],[xlimit+2,1])
else:
Lcarre([1,1],[1,y-2],[xwin-2,y-2],[xwin-2,1])
def write_tgafile(loc2,bitmap,width,height,profondeur):
f=open(loc2,'wb')
Origine_en_haut_a_gauche=32
Origine_en_bas_a_gauche=0
Data_Type_2=2
RVB=profondeur*8
RVBA=32
entete0=[]
for t in range(18):
entete0.append(chr(0))
entete0[2]=chr(Data_Type_2)
entete0[13]=chr(width/256)
entete0[12]=chr(width % 256)
entete0[15]=chr(height/256)
entete0[14]=chr(height % 256)
entete0[16]=chr(RVB)
entete0[17]=chr(Origine_en_bas_a_gauche)
#Origine_en_haut_a_gauche
for t in entete0:
f.write(t)
for t in bitmap:
for c in [2,1,0,3]:
#print t[c]%256
f.write(chr(t[c]*2))
f.close()
def save(x0,y0,dx,dy):
global SAVE
im = Buffer(GL_BYTE,[dx*(dy+1),4])
glReadPixels(x0,y0,dx,dy,GL_RGBA, GL_BYTE,im);
print len(im), dx*dy, dx, dy, len(im)/dy
write_tgafile(loc2,im,dx,dy+1,4)
SAVE=0
Blender.Redraw()
def DOCMat_list(TMATList,ME):
me=Blender.NMesh.GetRaw(ME.getData().name)
if len(me.materials)!=0:
n=0
for mat in me.materials:
TMATList[1][n][0]=mat.R
TMATList[1][n][1]=mat.G
TMATList[1][n][2]=mat.B
n+=1
TMATList[0]=n
else:
TMATList[0]=0
return TMATList
def SELMat_list():
global TMATList,selmatlist
Me=Blender.Object.GetSelected()
if Me!=[]:
if Me[0].getType()=='Mesh':
TMATList=DOCMat_list(TMATList,Me[0])
selmatlist=[]
for TMat in TMATList[2]:
if TMat.val==1.0:
selmatlist.append(TMATList[2].index(TMat))
ERROR=0
else:
ERROR=1
TextERROR='Selected Object is not a mesh.'
else:
ERROR=1
TextERROR='No Selected Object.'
def DOCBONEMENU(TBONEMENU):
pass
# ----------
# uvpaint1
# ----------
NSIZE=Create(1.0)
# ----------
# uvpaint2
# ----------
LINE=Create(0)
# ----------
# uvpaint3
# ----------
TEXT=Create(loc2)
# ----------
# uvpaint4
# ----------
TMENU="MODE MENU %t|All %x1|Material %x2|Selected %x3"
# ----------
# uvpaint4
# ----------
# coming soon : "|vertex group %x4", perhaps in uvpainter v0.5
LCOLOR= Create(64)
SAVE=0
MMENU=Create(3)
TDOCMat = Create(0)
# ----------
TMATList= [0,[],[]]
for t in range(16):
TMATList[1].append([0.0,0.0,0.0])
TMATList[2].append(Create(0))
# ----------
TDOCMat = Create(1)
# ----------
TBONEMENU= Create(1)
# ----------
XLIMIT=Create(0)
y=0
x=0
x0=0
y0=0
xwin=0
n0=32
def draw():
global NSIZE,LINE,x0,y0,y,x,TEXT,MMENU,TDOCMat
global XLIMIT,selmatlist,xwin, LCOLOR, SAVE
global mouse_xr,sel, ANC, X,Y, TRANSP, DOCU
global mouse_yr, POS, XY,size, TRANSP,EMPTY
global MENU, SAVE
size=Buffer(GL_FLOAT, 4)
glGetFloatv(GL_SCISSOR_BOX, size)
size=[int(s) for s in size ]
n0=32
x0=size[0]
y0=size[1]
x=size[2]
y=size[3]
xwin=x
ywin=y
glClear(GL_COLOR_BUFFER_BIT)
glShadeModel(GL_SMOOTH)
#transparence
if TRANSP :
glEnable(GL_BLEND)
glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
SelecMESH=Blender.Object.GetSelected()
if SelecMESH!=[] and SelecMESH[0].getType()=='Mesh':
affiche_mesh(SelecMESH[0],int(y*NSIZE.val),int(y*NSIZE.val-2))
if MENU:
glColor3f(0.0,0.0,0.0)
glRectf(4,size[3],555,size[3]-32 )
glColor3f(1.0,1.0,1.0)
glRasterPos2f(8, size[3]-13)
Text("uvpainter v0.8")
glRasterPos2f(8, size[3]-28)
Text("Jm Soler, 08/2005")
Button("ReDraw" ,16 ,290-118+61 ,size[3]-30 ,60 ,13)
Button("Exit" ,1 ,250-122+63 ,size[3]-30 ,38 ,13)
Button("Save" ,6 ,250-16+61 ,size[3]-30 ,40 ,13)
NSIZE= Slider("Sc:",4 ,290-118+61 ,size[3]-15 , 102, 13, NSIZE.val, 0.1,5.0,0,"SIZE.")
LINE=Toggle("line", 5 ,250-122+63 ,size[3]-15 , 38, 13, LINE.val, "Draw lines")
glRasterPos2f(250-130 ,size[3]-13,)
Text("Mode")
LCOLOR=Number("", 50, 250-77,size[3]-15,18,13,LCOLOR.val,1,64,'Line color, 64-lighter, 1-darker.')
MMENU= Menu(TMENU ,2 ,250-130, size[3]-30, 63, 13, MMENU.val, "MODE menu.")
if MMENU.val==1 or MMENU.val==3:
glRasterPos2f( 250-16+61+42+80,size[3]-13)
if XLIMIT.val:
xl=xwin
else:
xl=xlimit
Text("x :"+"%d"%(xl+2))
glRasterPos2f(250-16+61+42+65*2,size[3]-13)
Text("y :"+"%d"%(y-n0+1))
TEXT=String("to:", 7 , 278+61 ,size[3]-28 , 213, 13, TEXT.val, 256, "Draw lines")
if XLIMIT.val==1:
limit='winlimit'
else:
limit='maxXlimit'
XLIMIT=Toggle(limit, 9 , 250-16+61+42 ,size[3]-15 , 60, 13, XLIMIT.val, "to save picture from x max uv limit, or x window max limit")
if MMENU.val==2:
TDOCMat=Toggle("doc" ,24,250-130+35 ,size[3]-13 , 28, 13, TDOCMat.val)
if TDOCMat.val==1:
SELMat_list()
for t in range(TMATList[0]):
glCl3(TMATList[1][t][0],
TMATList[1][t][1],
TMATList[1][t][2])
glRct((293-16+61)+t*20,
size[3]-13,
(293-16+61)+t*20+20,
size[3]-30,)
TMATList[2][t]=Toggle("%s"%t , 32+t ,(293-16+61)+t*20 ,size[3]-13 ,20 , 13,TMATList[2][t].val)
#else:
# save()
if DOCU:
Doc(size)
def event(evt, val):
global mouse_x, x, mouse_y, y, LC
global LCOLOR, DOCU, MENU, SAVE
global mouse_xr,sel, ANC, X,Y ,NSIZE
global mouse_yr, POS, XY,size, TRANSP,EMPTY
if (evt== QKEY or evt== ESCKEY and not val):
Exit()
elif (evt== TKEY and not val):
TRANSP=abs(TRANSP-1)
elif (evt== EKEY and not val):
EMPTY=abs(EMPTY-1)
elif (evt== MKEY and not val):
MENU=abs(MENU-1)
elif (evt== SKEY and not val):
SAVE=abs(MENU-1)
elif (evt== WKEY and not val):
LC=1.0
LCOLOR.val=64
elif (evt== BKEY and not val):
LC=0.0
LCOLOR.val=1
elif (evt==LEFTARROWKEY and not val) :
POS[0]-=10
elif (evt==RIGHTARROWKEY and not val) :
POS[0]+=10
elif (evt==UPARROWKEY and not val) :
POS[1]+=10
elif (evt==DOWNARROWKEY and not val) :
POS[1]-=10
elif (evt==PADMINUS and not val) :
NSIZE.val-=0.1
elif (evt==PADPLUSKEY and not val) :
NSIZE.val+=0.1
elif (evt==LKEY and not val) :
LINE.val=abs(LINE.val-1)
elif (evt==HOMEKEY and not val) :
NSIZE.val=1.0
POS=[0,0]
elif (evt==DKEY and not val) :
DOCU=abs(DOCU-1)
elif (evt == MOUSEX): mouse_xr = val
elif (evt == MOUSEY): mouse_yr = val
elif (evt == RIGHTMOUSE and val):
if (sel==0 ) :
ANC=[(mouse_xr-size[0])-POS[0],(mouse_yr-size[1])-POS[1]]
sel=1
elif (evt == RIGHTMOUSE and not val):
sel=0
ANC=0,0
Redraw(1)
def bevent(evt):
global LINE,NSIZE,n0,x0,y0,y,TEXT, loc2
global TMATList, selmatlist, TDOCMat,XLIMIT
global xlimit,LCOLOR,LC,SAVE
if (evt== 1):
Exit()
elif (evt== 16):
pass
elif (evt== 4):
ng=NSIZE.val
elif (evt== 6):
if XLIMIT.val==1:
xi=xwin
save(x0,y0,xi+2,int(y-n0))
else:
xi=xlimit
save(x0,y0,xi+2,int(y*NSIZE.val)-n0)
elif (evt== 7):
if exist(TEXT.val):
loc2=TEXT.val
else:
TEXT.val=loc2
elif (evt== 24) or (evt in [32,33,34,35,36,37,38,39,40,41,42,43,44]):
SELMat_list()
elif (evt== 50):
LC=float(LCOLOR.val)/64.0
Blender.Redraw()
Register(draw, event, bevent)