Problemi con ampiezza prima generazione albero
-
24-09-2019 - |
Domanda
ho un problema con ampiezza primo algoritmo, il mio script genera curve in maya, posizionarli, ruotare e scalare in modo mi danno la forma di albero, ho queste variabili
cs = stato attuale,
p = genitore,
nodi = non List nodo visitato
lvl = profondità attuale
maxlvl = profondità massima
il problema è che non posso determinare profondità corrente, e terminare l'albero lì, esiste senza tutti i nodi visitati
qui è il mio script
import random
import math
import maya.cmds as mc
#----define set-----
#--- curve, #children,angle,l--------
#----define rule----
# l=80% ch<=3 angle<=45
l=0.8
ch=6
ang=50
#---list to track----
nodes=[]
#--- begin-----
#---start curve-----
nodes.append(mc.curve(n="a",d=3,p=((0,0,0),(0,1,0),(0,3,0),(0,5,0),(0,6,0)),k=(0,0,0,1,2,2,2)))
cs="" #---current state----
p="a" #---paretn of the current state----
maxlvl=6 #--max depth for the tree
lvl=1 #-- cuurent level----
while (len(nodes)!=0 and lvl<5):
#---generate children number----
chN=random.randint(0,ch)
for j in range(chN):
#----generate node----
mc.select(p,r=1)
cs=mc.duplicate(rr=1)[0]
#---append to node lst----
nodes.append(cs)
#---adjust pos---
pos=mc.pointPosition(p+".cv[4]")
mc.xform(cs,t=(pos[0],pos[1],pos[2]),ws=1)
#---adjust rotation---
mc.rotate(0,0,(random.random()*45+5)*math.pow(-1,j%2),cs,r=1)
#---adjust scale---
mc.scale(0,math.pow(0.8,lvl),0,cs)
j+=1
#--- go to next parent---
nodes.pop(0)
if nodes:
p=nodes[0]
lvl+=1
vi ringrazio in anticipo
Soluzione
È necessario associare la profondità con ogni nodo. O mettere come un membro della classe del nodo, o hanno le voci di coda memorizzano sia la profondità e il nodo in questo modo:
nodes.append((1, firstNode))
while nodes and lvl<5:
lvl, p = nodes.pop(0)
For each child:
#...create child
nodes.append((lvl+1, cs))
Alcune note non collegati:
-
L'uso
collections.deque
invece dilist
:. È più veloce per le code FIFO -
La dichiarazione
j+=1
non è necessaria nel ciclo for.