blender/release/scripts/disp_paint.py
Willian Padovani Germano 6b1eb45dda Scripts:
- Campbell Barton updated his sel_same.py script;
- Added to Mesh scripts a call to Window.EditMode(0) to leave editmode before changing meshes.

BPython:
- small doc fixes / updates;
- added a call to undo_push_mesh inside Window.EditMode(0).

Mesh scripts could change the mesh but not the editmesh -- that would then overwrite the changed mesh.  Made all mesh scripts leave edit mode before changing a mesh.
2004-07-28 17:46:29 +00:00

435 lines
14 KiB
Python

#!BPY
""" Registration info for Blender menus: <- these words are ignored
Name: 'Dispaint'
Blender: 233
Group: 'Mesh'
Tip: 'Use vertex paint color value to modify shape displacing vertices along normal.'
"""
# $Id$
#
#----------------------------------------------
# jm soler, displacement paint 03/2002 - > 05/2004: disp_paintf
# Terrain Noise added suugered by Jimmy Haze
#----------------------------------------------
# Page officielle :
# http://jmsoler.free.fr/didacticiel/blender/tutor/cpl_displacementpainting.htm
# Communiquer les problemes et erreurs sur:
# http://www.zoo-logique.org/3D.Blender/newsportal/thread.php?group=3D.Blender
# --------------------------------------------------------------------------
# ce script est proposé sous licence GPL pour etre associe
# a la distribution de Blender 2.33
# --------------------------------------------------------------------------
# this script is released under GPL licence
# for the Blender 2.33 scripts package
# --------------------------------------------------------------------------
# ***** BEGIN GPL LICENSE BLOCK *****
#
# 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 *****
# --------------------------------------------------------------------------
# 09/07/04 : Noise functions based on a piece of script by Jimmy Haze.
# --------------------------------------------------------------------------
import Blender
from Blender import *
from Blender.Draw import *
from Blender.BGL import *
from Blender.Noise import *
from Blender.Scene import *
sc=Scene.getCurrent()
# niveau du deplacement
ng=0.5
# noise default
NOISE=1
# profondeur des couleurs primaires rgb
maxcol=255.0*3
# limitation de la zone de travail sur le
# le materiau numer mat du tableau d'indices
# des materiaux. Par defaut mat =-1 ce qui signifie
# que toute la surface est traitee
mat=[]
vindex=[]
ORIName=''
NEWName=''
ERROR=0
TextERROR=''
def copy_transform(ozero,Obis):
Obis.setSize(ozero.getSize());
Obis.setEuler(ozero.getEuler());
Obis.setLocation(ozero.getLocation())
return Obis
def traite_face(f):
global vindexm, ng, NOISE
global H,lacunarity,octaves,offset,basis
if ORIENTMenu.val==1:
for z in range(len(f.v)):
c=0.0
if vindex[f.v[z].index]!=0:
c=float(f.col[z].r+f.col[z].b+f.col[z].g)/maxcol*ng/vindex[f.v[z].index]
else:
c=0
f.v[z].co[0]=f.v[z].co[0]+f.v[z].no[0]*c
f.v[z].co[1]=f.v[z].co[1]+f.v[z].no[1]*c
f.v[z].co[2]=f.v[z].co[2]+f.v[z].no[2]*c
elif ORIENTMenu.val==2:
for z in range(len(f.v)):
c=0.0
if vindex[f.v[z].index]!=0:
c=float(f.col[z].r+f.col[z].b+f.col[z].g)/maxcol*ng/vindex[f.v[z].index]
else:
c=0
for t in range(3):
if TAXEList[1][t].val==1:
f.v[z].co[t]=f.v[z].co[t]+c
elif ORIENTMenu.val==3 and NOISE<9:
for z in range(len(f.v)):
c=0.0
if vindex[f.v[z].index]!=0:
nx=f.v[z].co[0]/4
ny=f.v[z].co[1]/4
nz=f.v[z].co[2]/4
nn = ng + noise((nx,ny,nz),NOISE)
c=float(f.col[z].r+f.col[z].b+f.col[z].g)/maxcol*nn/vindex[f.v[z].index]
else:
c=0
f.v[z].co[0]=f.v[z].co[0]+f.v[z].no[0]*c
f.v[z].co[1]=f.v[z].co[1]+f.v[z].no[1]*c
f.v[z].co[2]=f.v[z].co[2]+f.v[z].no[2]*c
elif ORIENTMenu.val==3 and NOISE==9:
for z in range(len(f.v)):
c=0.0
if vindex[f.v[z].index]!=0:
nx=f.v[z].co[0]/4
ny=f.v[z].co[1]/4
nz=f.v[z].co[2]/4
nn = ng + cellNoise((nx,ny,nz))
c=float(f.col[z].r+f.col[z].b+f.col[z].g)/maxcol*nn/vindex[f.v[z].index]
else:
c=0
f.v[z].co[0]=f.v[z].co[0]+f.v[z].no[0]*c
f.v[z].co[1]=f.v[z].co[1]+f.v[z].no[1]*c
f.v[z].co[2]=f.v[z].co[2]+f.v[z].no[2]*c
elif ORIENTMenu.val==3 and NOISE==10:
for z in range(len(f.v)):
c=0.0
if vindex[f.v[z].index]!=0:
nx=f.v[z].co[0]/4
ny=f.v[z].co[1]/4
nz=f.v[z].co[2]/4
nn = ng + heteroTerrain((nx,ny,nz),H,lacunarity,octaves,offset,basis)
c=float(f.col[z].r+f.col[z].b+f.col[z].g)/maxcol*nn/vindex[f.v[z].index]
else:
c=0
f.v[z].co[0]=f.v[z].co[0]+f.v[z].no[0]*c
f.v[z].co[1]=f.v[z].co[1]+f.v[z].no[1]*c
f.v[z].co[2]=f.v[z].co[2]+f.v[z].no[2]*c
def paint():
global MODEMenu, vindex,ng, mat, ORIName, NEWName
Me=Object.GetSelected()
if Me!=[]:
if Me[0].getType()=='Mesh':
vindex=[]
ORIName=Me[0].getData().name
me1=NMesh.GetRaw(Me[0].getData().name)
try:
o=Object.Get('newMESH')
me=o.getData()
me=me1
except:
o=Object.New('Mesh','newMESH')
sc.link(o)
me=me1
name='new.002'
for m in me.verts:
vindex.append(0)
for f in me.faces:
for v in f.v:
if MODEMenu.val!=2:
if MODEMenu.val==1:
vindex[v.index]+=1
else:
if v.sel==1:
vindex[v.index]+=1
else:
#print mat
if f.mat in mat:
vindex[v.index]+=1
for f in me.faces:
if MODEMenu.val==2:
if f.mat in mat:
traite_face(f)
else:
traite_face(f)
Me[0].link(me)
#o=copy_transform(Me[0],o)
def NEWMEcreation(name):
nomdelobjet=""; objnumber=-1; namelist=[]
obj=Object.Get()
for ozero in obj:
if ozero.getType()=='Mesh':
namelist.append(ozero.getData().name)
if ozero.getData().name==name:
objnumber=obj.index(ozero)
if objnumber!=-1:
ozero=obj[objnumber]
nomdelobjet=ozero.getName()
Mesh=Blender.NMesh.GetRawFromObject(nomdelobjet)
n=0; name2=name[:];ok=0
while ok==0:
for n0 in namelist:
if n0.find(name2)==0:
ok=0;name2=name[0:name.find('.')+1]+'%s'%(n+1)
else: ok=1
n+=1
Mesh.name=name2
Obis = Blender.NMesh.PutRaw(Mesh,name2)
copy_transform(ozero,Obis)
Obis.makeDisplayList()
def DOCMat_list(TMATList):
global mat
Me=Object.GetSelected()
if Me!=[]:
if Me[0].getType()=='Mesh':
me=NMesh.GetRaw(Me[0].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
MOname = "MODE MENU %t|Normal %x1|Material %x2|Selected %x3"
ORname = "ORIENT MENU %t|From Normal %x1|Local Axes %x2| Noise %x3"
NOname = "NOISE MENU %t|BLENDER %x1|STDPERLIN %x2|NEWPERLIN %x3|VORONOI_F1%x4|VORONOI_F2%x5|VORONOI_F3%x6|VORONOI_F4%x7|VORONOI_F2F1%x8|VORONOI_CRACKLE%x9|CELLNOISE%x10|HETEROTENOISE%x11"
MODEMenu = Create(1)
ORIENTMenu = Create(1)
NOISEMenu = Create(1)
NSIZE = Create(1.0)
TDOCMat = Create(0)
NRepeat = Create(1)
H=1.0
lacunarity=2.0
octaves=5.0
offset=1.0
basis=3
HBout=Create(H)
lacunarityBout=Create(lacunarity)
octavesBout=Create(octaves)
offsetBout=Create(offset)
basisBout=Create(basis)
TMATList= [0,[],[]]
for t in range(16):
TMATList[1].append([0.0,0.0,0.0])
TMATList[2].append(Create(0))
TAXEList=[['X','Y','Z'],[]]
for t in range(3):
TAXEList[1].append(Create(0))
glCr=glRasterPos2d
glCl3=glColor3f
glCl4=glColor4f
glRct=glRectf
def draw():
global MODEMenu, NSIZE, TDOCMat,TMATList, TAXEList
global mat, ORIName, NEWName, ORIENTMenu
global NRepeat, ERROR, TextERROR , NOISE, NOISEMenu
global HBout,lacunarityBout,octavesBout,offsetBout,basisBout
size=Buffer(GL_FLOAT, 4)
glGetFloatv(GL_SCISSOR_BOX, size)
size= size.list
for s in [0,1,2,3]: size[s]=int(size[s])
glClear(GL_COLOR_BUFFER_BIT)
glColor3f(0.0,0.0,0.0)
glRectf(4,size[3],534,size[3]-32 )
glColor3f(1.0,1.0,1.0)
glRasterPos2f(20, size[3]-15)
Text("Script Python de displacement painting")
glRasterPos2f(20, size[3]-28)
Text("Jean-michel Soler, juillet 2004")
n0=70
n1=55
Button("Create" ,17 ,5 ,size[3]-n0+16 ,60 ,20)
Button("Action" ,16 ,5 ,size[3]-n0-4 ,60 ,20)
Button("Exit" ,1 ,5 ,size[3]-n0-24 ,60 ,20)
NRepeat=Number("repeat" ,5 ,5 ,size[3]-n0-50 ,75 ,20, NRepeat.val,1,10)
glColor3f(0.0,0.0,0.0)
glRasterPos2f(80 ,size[3]-n0+24)
Text("MODE")
MODEMenu= Menu(MOname, 2 ,80 ,size[3]-n0 ,100,20, MODEMenu.val, "MODE menu.")
if MODEMenu.val==2:
TDOCMat=Toggle("Doc Mat" ,24 ,180 ,size[3]-n0 ,60 ,20,TDOCMat.val)
if TDOCMat.val==1:
#print TMATList
for t in range(TMATList[0]):
glCl3(TMATList[1][t][0],
TMATList[1][t][1],
TMATList[1][t][2])
glRct(80+t*40,
size[3]-n0-60,
80+t*40+40,
size[3]-n0-60+40)
TMATList[2][t]=Toggle("%s"%t , 32+t ,80+t*40+5 ,size[3]-n0-50 ,30 , 20,TMATList[2][t].val)
glColor3f(1.0,0.3,0.0)
glRasterPos2f(80+40+5 ,size[3]-n0-80)
if ERROR==1:
Text('Last error : '+TextERROR)
else:
Text('Last error : ')
glColor3f(0.0,0.0,0.0)
glRasterPos2f(240 ,size[3]-n0+24)
Text("ORIENTATION")
ORIENTMenu= Menu(ORname, 3 ,240 ,size[3]-n0 ,100,20, ORIENTMenu.val, "ORIENT menu.")
if ORIENTMenu.val==2 :
for t in range(3):
TAXEList[1][t]=Toggle("%s"%TAXEList[0][t],
40+t,
240+100+t*30 , size[3]-n0 ,30 , 20,
TAXEList[1][t].val)
if ORIENTMenu.val==3 :
glRasterPos2f(20 ,size[3]-n0-90+24)
Text("NOISE")
NOISEMenu= Menu(NOname, 45 , 240 ,size[3]-n0-110 ,110,20, NOISEMenu.val, "NOISE menu.")
if NOISEMenu.val==11:
HBout= Slider("H", 46 ,110 ,size[3]-n0-190 ,125,20, HBout.val, -2.0,+2.0,0,)
lacunarityBout=Slider("lacunarity",47 ,110 ,size[3]-n0-110 ,125,20, lacunarityBout.val, -4.0,+4.0,0,)
octavesBout=Slider("octave", 48 ,110 ,size[3]-n0-130 ,125,20, octavesBout.val, -10.0,+10.0,0,)
offsetBout=Slider("offset", 49 ,110 ,size[3]-n0-150 ,125,20, offsetBout.val, -5.0,+5.0,0,)
basisBout=Slider("noise", 50 ,110 ,size[3]-n0-170 ,125,20, basisBout.val, 0,9,0,)
NSIZE= Slider("Disp Size", 4 ,80 ,size[3]-n0-20 ,260,20, NSIZE.val, -4.0,+4.0,0,"SIZE.")
def event(evt, val):
if ((evt== QKEY or evt== ESCKEY) and not val): Exit()
def bevent(evt):
global MODEMenu, NSIZE, ng, TMATList
global mat, ORIENTMenu, NRepeat, TAXEList
global ERROR,TextERROR, NOISE, NOISEMenu
global HBout,lacunarityBout,octavesBout,offsetBout,basisBout
global H,lacunarity,octaves,offset,basis
if (evt== 1):
Exit()
elif (evt== 16):
for n in range(NRepeat.val):
paint()
elif (evt== 4):
ng=NSIZE.val
elif (evt== 24) or (evt in [32,33,34,35,36,37,38,39,40,41,42,43,44]):
Me=Object.GetSelected()
if Me!=[]:
if Me[0].getType()=='Mesh':
TMATList=DOCMat_list(TMATList)
mat=[]
for TMat in TMATList[2]:
if TMat.val==1.0:
mat.append(TMATList[2].index(TMat))
ERROR=0
else:
ERROR=1
TextERROR='Selected Object is not a mesh.'
else:
ERROR=1
TextERROR='No Selected Object.'
elif (evt== 17):
NEWMEcreation('new.002')
elif (evt== 45):
NOISE=NOISEMenu.val-1
elif (evt in [46,47,48,49, 50]):
H=HBout.val
lacunarity=lacunarityBout.val
octaves=octavesBout.val
offset=offsetBout.val
basis=basisBout.val
Blender.Redraw()
Window.EditMode(0)
Register(draw, event, bevent)