# $Id$
##
##  This file is part of pyFormex 0.8.9  (Fri Nov  9 10:49:51 CET 2012)
##  pyFormex is a tool for generating, manipulating and transforming 3D
##  geometrical models by sequences of mathematical operations.
##  Home page: http://pyformex.org
##  Project page:  http://savannah.nongnu.org/projects/pyformex/
##  Copyright 2004-2012 (C) Benedict Verhegghe (benedict.verhegghe@ugent.be) 
##  Distributed under the GNU General Public License version 3 or later.
##
##
##  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 3 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, see http://www.gnu.org/licenses/.
##

"""ExtrudeMesh

This example illustrates Mesh extrusion using quadratic elements.

First, a Mesh is created consisting of a single point.
The point is extruded in the x-direction, resulting in a line.
The line is further extrude in y-direction to yield a quadratic surface.
A final extrusion in the z-direction delivers a quadratic volume.
"""
from __future__ import print_function
_status = 'checked'
_level = 'beginner'
_topics = ['mesh']
_techniques = ['extrude', 'quadratic']

from gui.draw import *



def run():
    clear()

    nx,ny,nz = 5,3,4
    #nx,ny,nz = 3,2,2
    degree = 2           # create quadratic extrusions, change to 1 for linear
    serendipity = False
    show3Dbyborder = False
    noise = 0.0          # set nonzero to add some noise to the coordinates 
    sleep = 0
    central = nx//2 + ny//2 * nx + nz//2 * ny * nx

    smoothwire()
    view('iso')
    delay(0)

    a = Formex([0.,0.,0.]).toMesh()   # a point at the origin
    print(a.elName())
    draw(a,color='black')

    delay(sleep)

    b = a.extrude(nx,1.,0,degree=degree)  # point extruded to quadratic line 
    print(b.elName())
    draw(b.coords,wait=False)
    draw(b,color='red')

    c = b.extrude(ny,1.,1,degree=degree)  # line extruded to quadratic surface
    if serendipity:
        c = c.convert('quad8')#.compact()
    print(c.elName())
    draw(c.coords,wait=False)
    draw(c,color='blue')

    #c1 = c.trl(2,1.)
    #d = c.connect(c1,degree=2)

    #d = d.convert('hex20')
    d = c.extrude(nz,1.,2,degree=degree)  # surface extruded to quadratic volume
    d = d.compact()
    print(d.elName())
    #d = d.reverse()
    if show3Dbyborder:
        d = d.getBorderMesh()
    print("Shown as %s" % d.elName())
    clear()
    draw(d.coords,wait=False)
    #drawNumbers(d.coords)
    draw(d,color='yellow',bkcolor='black')

    if noise:
        e = d.addNoise(noise)
        draw(e.coords,wait=False,clear=True)
        draw(e,color=cyan)

    drawNumbers(d)
    print("central = %s" % central)
    d.setProp(7)
    d.prop[central] = 0
    adj = [ d.adjacency(i) for i in range(3) ]
    # color the face adjacent elements blue
    a = adj[2][central]
    a = a[a>=0]
    d.prop[a] = 3
    # color the node but not face adjacent elements cyan
    a = adj[0].symdiff(adj[2])[central]
    a = a[a>=0]
    d.prop[a] = 4
    # color the node but not edge adjacent elements red
    a = adj[0].symdiff(adj[1])[central]
    a = a[a>=0]
    d.prop[a] = 1
    clear()
    draw(d)
    export({'ExtrudeMesh':d})
     
if __name__ == 'draw':
    run()
# End
