from __future__ import division
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import rcParams

rcParams['font.family'] = 'serif'
rcParams['font.size'] = 12

# Data
masses = np.loadtxt("Allometric_law.csv", skiprows=1, delimiter=",",
                  usecols=(0,))
speeds = np.loadtxt("Allometric_law.csv", skiprows=1, delimiter=",",
                  usecols=(1,))
names = np.loadtxt("Allometric_law.csv", skiprows=1, delimiter=",",
                  usecols=(2,), dtype=str)
groups = np.loadtxt("Allometric_law.csv", skiprows=1, delimiter=",",
                  usecols=(3,), dtype=str)
posx = np.loadtxt("Allometric_law.csv", skiprows=1, delimiter=",",
                  usecols=(4,), dtype=int)
va = np.loadtxt("Allometric_law.csv", skiprows=1, delimiter=",",
                  usecols=(5,), dtype=str)
x_vals = np.logspace(-5, 6)
y_vals = 30*x_vals**(1./6.)

# Big plot
fig = plt.figure(figsize=(15, 10))
ax = plt.subplot(111)
box = ax.get_position()
ax.set_position([box.x0, box.y0, box.width, 0.7*box.height])
plt.plot(x_vals, y_vals, "k", zorder=3, alpha=0.4)
plt.loglog(masses, speeds, "ko", zorder=5)
plt.xlabel("Mass (kg)")
plt.ylabel("Speed (m/s)")
plt.yticks([1, 10, 100, 300], [1, 10, 100, 300])
plt.ylim([1, 300])
for cont in range(len(masses)):
    mass = masses[cont]
    speed = speeds[cont]
    name = names[cont]
    if va[cont] == "top":
        pos = (posx[cont], -10)
    else:
        pos = (posx[cont], 10)
    plt.annotate(name, xy=(mass, speed), xytext=pos,
        textcoords="offset points", ha="center", va=va[cont],
        rotation="vertical", backgroundcolor="white", zorder=4,
        fontsize=10)

# Small plot
ax2 = plt.axes([.2, .6, .4, .32])
plt.loglog(masses[groups=="Insects"], speeds[groups=="Insects"],
           marker="o", mfc="white", lw=0)
plt.loglog(masses[groups=="Birds"], speeds[groups=="Birds"],
           marker="s", mfc="grey", lw=0)
plt.loglog(masses[groups=="Airplanes"], speeds[groups=="Airplanes"],
           marker="^", mfc="black", lw=0)
plt.annotate("Insects", xy=(2e-5, 2))
plt.annotate("Birds", xy=(0.2, 4))
plt.annotate("Airplanes", xy=(4e3, 40))
plt.xticks([1e-5, 1, 1e5], [r"$10^{-5}$", r"$1$", r"$10^{5}$"])
plt.yticks([1, 10, 100], [r"$1$", r"$10$", r"$100$"])
plt.xlabel("Mass (kg)", fontsize=10)
plt.ylabel("Speed (m/s)", fontsize=10)
plt.ylim([1, 300])

plt.savefig("Allometric_Law_of_Body_Mass_vs_Cruising_Speed_in_Constructal_Theory.svg")
plt.show()