"""
Thanks to Tony Yu <tsyu80@gmail.com> for the logo design
"""

import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt
import matplotlib.cm as cm
import matplotlib.mlab as mlab
from pylab import rand

mpl.rcParams['xtick.labelsize'] = 6
mpl.rcParams['ytick.labelsize'] = 6
mpl.rcParams['axes.edgecolor'] = 'gray'


axalpha = 0.05
#figcolor = '#EFEFEF'
figcolor = 'white'
dpi = 80
fig = plt.figure(figsize=(7, 1.5),dpi=dpi)
fig.figurePatch.set_edgecolor(figcolor)
fig.figurePatch.set_facecolor(figcolor)


def add_math_background():
    ax = fig.add_axes([0., 0., 1., 1.])

    text = []
    text.append((r"$W^{3\beta}_{\delta_1 \rho_1 \sigma_2} = U^{3\beta}_{\delta_1 \rho_1} + \frac{1}{8 \pi 2} \int^{\alpha_2}_{\alpha_2} d \alpha^\prime_2 \left[\frac{ U^{2\beta}_{\delta_1 \rho_1} - \alpha^\prime_2U^{1\beta}_{\rho_1 \sigma_2} }{U^{0\beta}_{\rho_1 \sigma_2}}\right]$", (0.7, 0.2), 20))
    text.append((r"$\frac{d\rho}{d t} + \rho \vec{v}\cdot\nabla\vec{v} = -\nabla p + \mu\nabla^2 \vec{v} + \rho \vec{g}$",
                (0.35, 0.9), 20))
    text.append((r"$\int_{-\infty}^\infty e^{-x^2}dx=\sqrt{\pi}$",
                (0.15, 0.3), 25))
    #text.append((r"$E = mc^2 = \sqrt{{m_0}^2c^4 + p^2c^2}$",
    #            (0.7, 0.42), 30))
    text.append((r"$F_G = G\frac{m_1m_2}{r^2}$",
                (0.85, 0.7), 30))
    for eq, (x, y), size in text:
        ax.text(x, y, eq, ha='center', va='center', color="blue", alpha=0.25,
                transform=ax.transAxes, fontsize=size)
    ax.set_axis_off()
    return ax

def add_matplotlib_text(ax):
    ax.text(0.3, 0.5, 'matplotlib', color='black', fontsize=70,
               ha='left', va='center', alpha=1.0, transform=ax.transAxes)

def add_polar_bar():
    ax = fig.add_axes([0.05, 0.05, 0.2, 0.9], polar=True)
    ax.axesPatch.set_alpha(axalpha)
    N = 15
    arc = 2. * np.pi
    theta = np.arange(0.0, arc, arc/N)
    radii = 10 * np.random.rand(N)
    width = np.pi / 4 * np.random.rand(N)
    bars = ax.bar(theta, radii, width=width, bottom=0.0)
    for r, bar in zip(radii, bars):
        bar.set_facecolor(cm.jet(r/10.))
        bar.set_alpha(0.6)

def add_histogram():
    ax = fig.add_axes([0.325, 0.125, 0.2, 0.4])
    ax.axesPatch.set_alpha(axalpha)
    mu, sigma = 100, 15
    x = mu + sigma * np.random.randn(500)
    # the histogram of the data
    n, bins, patches = ax.hist(x, 15, normed=1, facecolor='green',
                               edgecolor='green', alpha=0.5)
    y = mlab.normpdf(bins, mu, sigma)
    l = ax.plot(bins, y, 'r', lw=1)
    ax.set_xlim(-4*sigma + mu, 4*sigma + mu)

def add_scatter():
    ax = fig.add_axes([0.6, 0.125, 0.15, 0.4])
    ax.axesPatch.set_alpha(axalpha)
    N = 40
    volume = 100 * rand(N)
    color = 256 * rand(N)
    darkgray = [0.2] * 3
    plt.scatter(rand(N), rand(N), c=color, s=volume, alpha=0.75,
                edgecolor=darkgray)
    plt.axis('tight')
    ax.set_yticks([])
    ax.set_xticks([])
    #ax.set_axis_off()

def add_pcolor():
    ax = fig.add_axes([0.75, 0.125, 0.2, 0.4])
    ax.axesPatch.set_alpha(axalpha)
    x = np.linspace(-3.0, 3.0, 200)
    X,Y = np.meshgrid(x, x)
    Z = (1- X/2 + X**5 + Y**3) * np.exp(-X**2-Y**2)
    limits = (-3,3,-3,3)
    cmap = cm.RdGy_r#(np.linspace(0.2, 1))
    im = plt.imshow(Z, interpolation='bilinear', origin='lower',
                    cmap=cmap, extent=limits)
    levels = np.arange(-1.2,1.6,0.4)
    cset = plt.contour(Z, levels, cmap=cm.hot, origin='lower', extent=limits)
    plt.clabel(cset, inline=1, fmt='%1.1f', fontsize=4)
    plt.colorbar()
    #ax.set_axis_off()

def add_pcolor2():
    ax = fig.add_axes([0.3, 0.125, 0.25, 0.4], aspect='auto')
    ax.axesPatch.set_alpha(axalpha)
    x = np.linspace(-3.0, 3.0, 200)
    X,Y = np.meshgrid(x, x)
    Z = (1- X/2 + X**5 + Y**3) * np.exp(-X**2-Y**2)
    limits = (-3,3,-3,3)
    cmap = cm.RdGy_r#(np.linspace(0.2, 1))
    im = plt.imshow(Z, interpolation='bilinear', origin='lower',
                    cmap=cmap, extent=limits, aspect='auto')
    levels = np.arange(-1.2,1.6,0.4)
    cset = plt.contour(Z, levels, cmap=cm.hot, origin='lower', extent=limits)
    plt.clabel(cset, inline=1, fmt='%1.1f', fontsize=4)
    #plt.colorbar()
    #ax.set_axis_off()

if __name__ == '__main__':
    main_axes = add_math_background()
    add_polar_bar()
#    add_histogram()
#    add_scatter()
# add_pcolor()
    #add_pcolor2()
    add_matplotlib_text(main_axes)
    fig.savefig('logo2.png', facecolor=figcolor, edgecolor=figcolor, dpi=dpi)
    plt.show()