def drawFreeHand():
global CURRENTZONE; global CURRENTPOLYLINE; global ALLZONES
w = WIDGETS['draw']
prev = []; first = []
if CTK.__BUSY__ == False:
CPlot.unselectAllZones()
CTK.saveTree()
CTK.__BUSY__ = True
TTK.sunkButton(w)
CPlot.setState(cursor=1)
buttonState = 0
while CTK.__BUSY__ == True:
if prev == []: # first point
l = []
while l == []:
l = CPlot.getActivePoint()
if l != []: prev = l; first = l
time.sleep(CPlot.__timeStep__)
w.update()
if CTK.__BUSY__ == False: break
else: # next points
diff = -1.
while (diff < 1.e-10):
(buttonState,x,y,z) = CPlot.getMouseState()
l = (x,y,z)
diff = Vector.norm2(Vector.sub(l,prev))
diff1 = Vector.norm2(Vector.sub(l,first))
if (diff1 < 1.e-10): l = first
if (buttonState == 5): break
time.sleep(CPlot.__timeStep__)
w.update()
if (CTK.__BUSY__ == False): break
prev = l
CPlot.unselectAllZones()
if (buttonState == 5): # button released
ALLZONES.append(CURRENTZONE)
CURRENTZONE = None; prev = []; first = []
CURRENTPOLYLINE = []
CTK.TKTREE.updateApp()
if (CTK.__BUSY__ == True and buttonState != 5):
CURRENTPOLYLINE.append((l[0],l[1],l[2]))
if (CURRENTZONE == None):
CTK.t = C.addBase2PyTree(CTK.t, 'CONTOURS', 1)
base = Internal.getNodeFromName1(CTK.t, 'CONTOURS')
nob = C.getNobOfBase(base, CTK.t)
a = D.polyline(CURRENTPOLYLINE)
CURRENTZONE = a
CTK.add(CTK.t, nob, -1, a)
ret = Internal.getParentOfNode(CTK.t, CURRENTZONE)
noz = ret[1]
else:
a = D.polyline(CURRENTPOLYLINE)
CURRENTZONE = a
CTK.replace(CTK.t, nob, noz, a)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CPlot.render()
buttonState = 0
CTK.__BUSY__ = False
TTK.raiseButton(w)
CPlot.setState(cursor=0)
else:
CTK.__BUSY__ = False
surfaces = getSurfaces()
if (surfaces != []):
if (CURRENTZONE != None): ALLZONES += [CURRENTZONE]
for s in ALLZONES:
ret = Internal.getParentOfNode(CTK.t, s)
nob = C.getNobOfBase(ret[0], CTK.t)
a = T.projectOrthoSmooth(s, surfaces)
noz = ret[1]
CTK.replace(CTK.t, nob, noz, a)
#C._fillMissingVariables(CTK.t)
(CTK.Nb, CTK.Nz) = CPlot.updateCPlotNumbering(CTK.t)
CTK.TKTREE.updateApp()
CPlot.render()
CURRENTZONE = None; ALLZONES = []
CURRENTPOLYLINE = []
TTK.raiseButton(w)
CPlot.setState(cursor=0)
def apply3D(density, npts, factor, ntype):
nzs = CPlot.getSelectedZones()
nz = nzs[0]
nob = CTK.Nb[nz]+1
noz = CTK.Nz[nz]
zone = CTK.t[2][nob][2][noz]
ret = getEdges3D(zone, 0.)
if ret is None: return True
(m, r, f, ue, uf, ind) = ret
out = []
# Applique la fonction sur m
i = m[0]
dims = Internal.getZoneDim(i)
np = dims[1]*dims[2]*dims[3]
if ntype == 0: # uniformize
if density > 0: npts = D.getLength(i)*density
if factor > 0: npts = np*factor[0]
npts = int(max(npts, 2))
distrib = G.cart((0,0,0), (1./(npts-1.),1,1), (npts,1,1))
b = G.map(i, distrib)
elif ntype == 1: # refine
if factor < 0: factor = (npts-1.)/(np-1)
else: npts = factor*(np-1)+1
b = G.refine(i, factor, 1)
elif ntype == 2: # stretch (factor=h)
h = factor
l = D.getLength(i)
a = D.getCurvilinearAbscissa(i)
distrib = C.cpVars(a, 's', a, 'CoordinateX')
C._initVars(distrib, 'CoordinateY', 0.)
C._initVars(distrib, 'CoordinateZ', 0.)
distrib = C.rmVars(distrib, 's')
N = dims[1]
val = C.getValue(a, 's', ind)
Xc = CPlot.getActivePoint()
valf = val
Pind = C.getValue(i, 'GridCoordinates', ind)
if ind < N-1: # cherche avec indp1
Pindp1 = C.getValue(i, 'GridCoordinates', ind+1)
v1 = Vector.sub(Pindp1, Pind)
v2 = Vector.sub(Xc, Pind)
if Vector.dot(v1,v2) >= 0:
val2 = C.getValue(a, 's', ind+1)
alpha = Vector.norm(v2)/Vector.norm(v1)
valf = val+alpha*(val2-val)
if ind > 0 and val == valf: # cherche avec indm1
Pindm1 = C.getValue(i, 'GridCoordinates', ind-1)
v1 = Vector.sub(Pindm1, Pind)
v2 = Vector.sub(Xc, Pind)
if Vector.dot(v1,v2) >= 0:
val2 = C.getValue(a, 's', ind-1)
alpha = Vector.norm(v2)/Vector.norm(v1)
valf = val+alpha*(val2-val)
if h < 0: distrib = G.enforcePoint(distrib, valf)
else:
if val == 0: distrib = G.enforcePlusX(distrib, h/l, N/10, 1)
elif val == 1: distrib = G.enforceMoinsX(distrib, h/l, N/10, 1)
else: distrib = G.enforceX(distrib, valf, h/l, N/10, 1)
b = G.map(i, distrib)
elif ntype == 3:
source = factor
b = G.map(i, source, 1)
elif ntype == 4: # smooth (factor=eps, npts=niter)
niter = npts
eps = factor
a = D.getCurvilinearAbscissa(i)
distrib = C.cpVars(a, 's', a, 'CoordinateX')
C._initVars(distrib, 'CoordinateY', 0.)
C._initVars(distrib, 'CoordinateZ', 0.)
distrib = C.rmVars(distrib, 's')
bornes = P.exteriorFaces(distrib)
distrib = T.smooth(distrib, eps=eps, niter=niter,
fixedConstraints=[bornes])
b = G.map(i, distrib, 1)
dimb = Internal.getZoneDim(b)
npts = dimb[1]
out.append(b)
# Raffine les edges si necessaires
if npts != np:
ret = getEdges3D(zone, 2.)
if ret is None: return True
(m, r, f, ue, uf, ind) = ret
for i in r:
dims = Internal.getZoneDim(i)
np = dims[1]*dims[2]*dims[3]
factor = (npts-1.)/(np-1) # npts de m
b = G.refine(i, factor, 1)
out.append(b)
# Garde les autres
out += ue
outf = []
# Rebuild les faces
for i in f:
# trouve les edges de la face
edges = P.exteriorFacesStructured(i)
match = []
for e in edges:
dime = Internal.getZoneDim(e)
np = dime[1]-1
P0 = C.getValue(e, Internal.__GridCoordinates__, 0)
P1 = C.getValue(e, Internal.__GridCoordinates__, np)
for ei in out: # retrouve les edges par leurs extremites
dimei = Internal.getZoneDim(ei)
npi = dimei[1]-1
Q0 = C.getValue(ei, Internal.__GridCoordinates__, 0)
Q1 = C.getValue(ei, Internal.__GridCoordinates__, npi)
t1 = Vector.norm2(Vector.sub(P0,Q0))
t2 = Vector.norm2(Vector.sub(P1,Q1))
if (t1 < 1.e-12 and t2 < 1.e-12): match.append(ei)
if len(match) == 4: # OK
fn = G.TFI(match)
# Projection du patch interieur
#dimsf = Internal.getZoneDim(fn)
#fns = T.subzone(fn, (2,2,1), (dimsf[1]-1,dimsf[2]-1,1))
#fns = T.projectOrtho(fns, [i])
#fn = T.patch(fn, fns, position=(2,2,1))
#fn = T.projectOrtho(fn, [i])
outf.append(fn)
else: return True
outf += uf
try:
b = G.TFI(outf)
CTK.replace(CTK.t, nob, noz, b)
return False
except Exception as e:
Panels.displayErrors([0,str(e)], header='Error: apply3D')
return True