#  Copyright (C) 2007-2016 Alan W. Irwin

# plmtex3, plptex3 demo.
#
#  This file is part of PLplot.
#
#  PLplot is free software; you can redistribute it and/or modify
#  it under the terms of the GNU Library General Public License as published
#  by the Free Software Foundation; either version 2 of the License, or
#  (at your option) any later version.
#
#  PLplot 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 Library General Public License for more details.
#
#  You should have received a copy of the GNU Library General Public License
#  along with PLplot; if not, write to the Free Software
#  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
#
from numpy import *

# Choose these values to correspond to tick marks.
XPTS = 2
YPTS = 2

NREVOLUTION = 16
NROTATION = 8
NSHEAR = 8

xmin = 0.
xmax = 1.0
xmid = 0.5*(xmax + xmin)
xrange = xmax - xmin

ymin = 0.
ymax = 1.0
ymid = 0.5*(ymax + ymin)
yrange = ymax - ymin

zmin = 0.
zmax = 1.0
zmid = 0.5*(zmax + zmin)
zrange = zmax - zmin
ysmin    = ymin + 0.1 * yrange
ysmax    = ymax - 0.1 * yrange
ysrange  = ysmax - ysmin
dysrot   = ysrange / float( NROTATION - 1 )
dysshear = ysrange / float( NSHEAR - 1 )
zsmin    = zmin + 0.1 * zrange
zsmax    = zmax - 0.1 * zrange
zsrange  = zsmax - zsmin
dzsrot   = zsrange / float( NROTATION - 1 )
dzsshear = zsrange / float( NSHEAR - 1 )

pstring = "The future of our civilization depends on software freedom."

def main(w):

 if 1:
    x = xmin + (xmax-xmin)*arange(XPTS)/float(XPTS-1)
    y = ymin + (ymax-ymin)*arange(YPTS)/float(YPTS-1)
    z = reshape(0.+zeros(XPTS*YPTS),[XPTS,YPTS])

 if 1:
    # Page 1: Demonstrate inclination and shear capability pattern.

    w.pladv(0)
    w.plvpor(-0.15, 1.15, -0.05, 1.05)
    w.plwind(-1.2, 1.2, -0.8, 1.5)
    w.plw3d(1.0, 1.0, 1.0, xmin, xmax, ymin, ymax, zmin, zmax, 20., 45.)

    w.plcol0(2)
    w.plbox3(
    "b", "", xmax-xmin, 0,
    "b", "", ymax-ymin, 0,
    "bcd", "", zmax-zmin, 0)

    # z = zmin.
    w.plschr(0., 1.0)
    for i in range(NREVOLUTION):
        omega = 2.*pi*(float(i)/float(NREVOLUTION))
        sin_omega = sin(omega)
        cos_omega = cos(omega)
        x_inclination = 0.5*xrange*cos_omega
        y_inclination = 0.5*yrange*sin_omega
        z_inclination = 0.
        x_shear = -0.5*xrange*sin_omega
        y_shear = 0.5*yrange*cos_omega
        z_shear = 0.
        w.plptex3(
        xmid, ymid, zmin,
        x_inclination, y_inclination, z_inclination,
        x_shear, y_shear, z_shear,
        0.0, "  revolution")

    # x = xmax.
    w.plschr(0., 1.0)
    for i in range(NREVOLUTION):
        omega = 2.*pi*(float(i)/float(NREVOLUTION))
        sin_omega = sin(omega)
        cos_omega = cos(omega)
        x_inclination = 0.
        y_inclination = -0.5*yrange*cos_omega
        z_inclination = 0.5*zrange*sin_omega
        x_shear = 0.
        y_shear = 0.5*yrange*sin_omega
        z_shear = 0.5*zrange*cos_omega
        w.plptex3(
        xmax, ymid, zmid,
        x_inclination, y_inclination, z_inclination,
        x_shear, y_shear, z_shear,
        0.0, "  revolution")

    # y = ymax.
    w.plschr(0., 1.0)
    for i in range(NREVOLUTION):
        omega = 2.*pi*(float(i)/float(NREVOLUTION))
        sin_omega = sin(omega)
        cos_omega = cos(omega)
        x_inclination = 0.5*xrange*cos_omega
        y_inclination = 0.
        z_inclination = 0.5*zrange*sin_omega
        x_shear = -0.5*xrange*sin_omega
        y_shear = 0.
        z_shear = 0.5*zrange*cos_omega
        w.plptex3(
        xmid, ymax, zmid,
        x_inclination, y_inclination, z_inclination,
        x_shear, y_shear, z_shear,
        0.0, "  revolution")
    # Draw minimal 3D grid to finish defining the 3D box.
    w.plmesh(x, y, z, w.DRAW_LINEXY)

    # Page 2: Demonstrate rotation of string around its axis.

    w.pladv(0)
    w.plvpor(-0.15, 1.15, -0.05, 1.05)
    w.plwind(-1.2, 1.2, -0.8, 1.5)
    w.plw3d(1.0, 1.0, 1.0, xmin, xmax, ymin, ymax, zmin, zmax, 20., 45.)

    w.plcol0(2)
    w.plbox3(
    "b", "", xmax-xmin, 0,
    "b", "", ymax-ymin, 0,
    "bcd", "", zmax-zmin, 0)

    # y = ymax.
    w.plschr(0., 1.0)
    x_inclination = 1.
    y_inclination = 0.
    z_inclination = 0.
    x_shear = 0.
    for i in range(NROTATION):
        omega = 2.*pi*(float(i)/float(NROTATION))
        sin_omega = sin(omega)
        cos_omega = cos(omega)
        y_shear = 0.5*yrange*sin_omega
        z_shear = 0.5*zrange*cos_omega
        zs = zsmax - dzsrot * float(i)
        w.plptex3(
        xmid, ymax, zs,
        x_inclination, y_inclination, z_inclination,
        x_shear, y_shear, z_shear,
        0.5, "rotation for y = y#dmax#u")

    # x = xmax.
    w.plschr(0., 1.0)
    x_inclination = 0.
    y_inclination = -1.
    z_inclination = 0.
    y_shear = 0.
    for i in range(NROTATION):
        omega = 2.*pi*(float(i)/float(NROTATION))
        sin_omega = sin(omega)
        cos_omega = cos(omega)
        x_shear = 0.5*xrange*sin_omega
        z_shear = 0.5*zrange*cos_omega
        zs = zsmax - dzsrot * float(i)
        w.plptex3(
        xmax, ymid, zs,
        x_inclination, y_inclination, z_inclination,
        x_shear, y_shear, z_shear,
        0.5, "rotation for x = x#dmax#u")

    # z = zmin.
    w.plschr(0., 1.0)
    x_inclination = 1.
    y_inclination = 0.
    z_inclination = 0.
    x_shear = 0.
    for i in range(NROTATION):
        omega = 2.*pi*(float(i)/float(NROTATION))
        sin_omega = sin(omega)
        cos_omega = cos(omega)
        y_shear = 0.5*yrange*cos_omega
        z_shear = 0.5*zrange*sin_omega
        ys = ysmax - dysrot * float(i)
        w.plptex3(
        xmid, ys, zmin,
        x_inclination, y_inclination, z_inclination,
        x_shear, y_shear, z_shear,
        0.5, "rotation for z = z#dmin#u")

    # Draw minimal 3D grid to finish defining the 3D box.
    w.plmesh(x, y, z, w.DRAW_LINEXY)

    # Page 3: Demonstrate shear of string along its axis.

    # Work around xcairo and pngcairo (but not pscairo) problems for
    # shear vector too close to axis of string. (N.B. no workaround
    # would be domega = 0.)
    domega = 0.05
    w.pladv(0)
    w.plvpor(-0.15, 1.15, -0.05, 1.05)
    w.plwind(-1.2, 1.2, -0.8, 1.5)
    w.plw3d(1.0, 1.0, 1.0, xmin, xmax, ymin, ymax, zmin, zmax, 20., 45.)

    w.plcol0(2)
    w.plbox3(
    "b", "", xmax-xmin, 0,
    "b", "", ymax-ymin, 0,
    "bcd", "", zmax-zmin, 0)

    # y = ymax.
    w.plschr(0., 1.0)
    x_inclination = 1.
    y_inclination = 0.
    z_inclination = 0.
    y_shear = 0.
    for i in range(NSHEAR):
        omega = domega + 2.*pi*(float(i)/float(NSHEAR))
        sin_omega = sin(omega)
        cos_omega = cos(omega)
        x_shear = 0.5*xrange*sin_omega
        z_shear = 0.5*zrange*cos_omega
        zs = zsmax - dzsshear * float(i)
        w.plptex3(
        xmid, ymax, zs,
        x_inclination, y_inclination, z_inclination,
        x_shear, y_shear, z_shear,
        0.5, "shear for y = y#dmax#u")

    # x = xmax.
    w.plschr(0., 1.0)
    x_inclination = 0.
    y_inclination = -1.
    z_inclination = 0.
    x_shear = 0.
    for i in range(NSHEAR):
        omega = domega + 2.*pi*(float(i)/float(NSHEAR))
        sin_omega = sin(omega)
        cos_omega = cos(omega)
        y_shear = -0.5*yrange*sin_omega
        z_shear = 0.5*zrange*cos_omega
        zs = zsmax - dzsshear * float(i)
        w.plptex3(
        xmax, ymid, zs,
        x_inclination, y_inclination, z_inclination,
        x_shear, y_shear, z_shear,
        0.5, "shear for x = x#dmax#u")

    # z = zmin.
    w.plschr(0., 1.0)
    x_inclination = 1.
    y_inclination = 0.
    z_inclination = 0.
    z_shear = 0.
    for i in range(NSHEAR):
        omega = domega + 2.*pi*(float(i)/float(NSHEAR))
        sin_omega = sin(omega)
        cos_omega = cos(omega)
        y_shear = 0.5*yrange*cos_omega
        x_shear = 0.5*xrange*sin_omega
        ys = ysmax - dysshear * float(i)
        w.plptex3(
        xmid, ys, zmin,
        x_inclination, y_inclination, z_inclination,
        x_shear, y_shear, z_shear,
        0.5, "shear for z = z#dmin#u")

    # Draw minimal 3D grid to finish defining the 3D box.
    w.plmesh(x, y, z, w.DRAW_LINEXY)

    # Page 4: Demonstrate drawing a string on a 3D path.
    w.pladv(0)
    w.plvpor(-0.15, 1.15, -0.05, 1.05)
    w.plwind(-1.2, 1.2, -0.8, 1.5)
    w.plw3d(1.0, 1.0, 1.0, xmin, xmax, ymin, ymax, zmin, zmax, 40., -30.)

    w.plcol0(2)
    w.plbox3(
    "b", "", xmax-xmin, 0,
    "b", "", ymax-ymin, 0,
    "bcd", "", zmax-zmin, 0)

    w.plschr(0., 1.2)
    # domega controls the spacing between the various characters of the
    # string and also the maximum value of omega for the given number
    # of characters in *pstring.
    domega = 2.*pi/len(pstring)
    omega = 0.
    # 3D function is a helix of the given radius and pitch
    radius = 0.5
    pitch = 1./(2.*pi)
    for character in pstring:
        sin_omega = sin(omega)
        cos_omega = cos(omega)
        xpos = xmid + radius*sin_omega
        ypos = ymid - radius*cos_omega
        zpos = zmin + pitch*omega
        # In general, the inclination is proportional to the derivative of
        # the position wrt theta.
        x_inclination = radius*cos_omega
        y_inclination = radius*sin_omega
        z_inclination = pitch
        # The shear vector should be perpendicular to the 3D line with Z
        # component maximized, but for low pitch a good approximation is
        # a constant vector that is parallel to the Z axis.
        x_shear = 0.
        y_shear = 0.
        z_shear = 1.
        w.plptex3(
        xpos, ypos, zpos,
        x_inclination, y_inclination, z_inclination,
        x_shear, y_shear, z_shear,
        0.5, character)
        omega += domega

    # Draw minimal 3D grid to finish defining the 3D box.
    w.plmesh(x, y, z, w.DRAW_LINEXY)

    # Page 5: Demonstrate w.plmtex3 axis labelling capability.
    w.pladv(0)
    w.plvpor(-0.15, 1.15, -0.05, 1.05)
    w.plwind(-1.2, 1.2, -0.8, 1.5)
    w.plw3d(1.0, 1.0, 1.0, xmin, xmax, ymin, ymax, zmin, zmax, 20., 45.)

    w.plcol0(2)
    w.plbox3(
    "b", "", xmax-xmin, 0,
    "b", "", ymax-ymin, 0,
    "bcd", "", zmax-zmin, 0)

    w.plschr(0., 1.0)
    w.plmtex3("xp", 3.0, 0.5, 0.5, "Arbitrarily displaced")
    w.plmtex3("xp", 4.5, 0.5, 0.5, "primary X-axis label")
    w.plmtex3("xs", -2.5, 0.5, 0.5, "Arbitrarily displaced")
    w.plmtex3("xs", -1.0, 0.5, 0.5, "secondary X-axis label")
    w.plmtex3("yp", 3.0, 0.5, 0.5, "Arbitrarily displaced")
    w.plmtex3("yp", 4.5, 0.5, 0.5, "primary Y-axis label")
    w.plmtex3("ys", -2.5, 0.5, 0.5, "Arbitrarily displaced")
    w.plmtex3("ys", -1.0, 0.5, 0.5, "secondary Y-axis label")
    w.plmtex3("zp", 4.5, 0.5, 0.5, "Arbitrarily displaced")
    w.plmtex3("zp", 3.0, 0.5, 0.5, "primary Z-axis label")
    w.plmtex3("zs", -2.5, 0.5, 0.5, "Arbitrarily displaced")
    w.plmtex3("zs", -1.0, 0.5, 0.5, "secondary Z-axis label")
    # Draw minimal 3D grid to finish defining the 3D box.
    w.plmesh(x, y, z, w.DRAW_LINEXY)

    # Restore defaults
    w.plschr( 0.0, 1.0 )

    # Must be done independently because otherwise this changes output files
    # and destroys agreement with C examples.
    #w.plcol0(1)