bifandvel.py

   1 # import stuff
   2 import math
   3 import numpy as np
   4 import matplotlib as mpl
   5 import matplotlib.pyplot as plt
   6 import matplotlib.mlab as mlab
   7 import matplotlib.axes as axe
   8 import matplotlib.text as txt
   9
  10 DATAX = '../rk_bifurcate.dat'
  11 DATAV = '../rk_bif_avg_velo.dat'
  12 FOUT = '../test.png'
  13
  14 # some macros
  15 PI = math.pi
  16 L = 2*PI
  17
  18 # options
  19 xpad = 0.002
  20 ypad = 0.075
  21 max = 0.875
  22 tsize = 24
  23 ticksize = 14
  24
  25 # default values
  26 xtks1 = []
  27 ytks1 = [0,L/2,L]
  28 ytks2 = [-1,-0.5,0,0.5,1]
  29 xtkl1 = []
  30 ytkl1 = ["0","L/2","L"]
  31 ylim1 = [0-ypad,L+ypad]
  32 ylim2 = [-1-ypad,1+ypad]
  33 xlbl1 = r''
  34 xlbl2 = r'$l$'
  35 ylbl1 = r'$x(k\tau)$ mod $L$'
  36 ylbl2 = r'$v_x$'
  37 res = (1440/80,900/80) # default dpi is 80
  38
  39 # calculate 3 to 1 ratio of upper to lower plot
  40 left, width = 0.1, 0.8
  41 c = max/4
  42 rect1 = [left, c+0.1, width, c*3]
  43 rect2 = [left, 0.1, width, c]
  44
  45 # title
  46 title = "a = 1.15, w = 0.1, l = 3.0, eta1 = 1.0, eta2 = 1.5, theta = 0, alpha = 83"
  47
  48 # import data
  49 r = mlab.csv2rec(DATAX, delimiter='\t', comments='#')
  50 s = mlab.csv2rec(DATAV, delimiter='\t', comments='#')
  51
  52 xmin = r.r.min()
  53 xmax = r.r.max()
  54
  55 xlim1 = xmin-xpad,xmax+xpad
  56 xlim2 = xmin-xpad,xmax+xpad
  57
  58 # draw figure
  59 fig = plt.figure(figsize=res)
  60 #fig.suptitle(title, size=tsize)
  61
  62 # draw first subplot and setup axis
  63 ax1 = fig.add_axes(rect1)
  64 ax2 = fig.add_axes(rect2)
  65
  66 # put x into the plot
  67 ax1.plot(r.r, r.x, c='r', marker=',', ms=0.1, ls='')
  68 ax1.set_xlim(xlim1)
  69 ax1.set_ylim(ylim1)
  70 ax1.set_yticks(ytks1)
  71 ax1.set_ylabel(ylbl1, size=tsize)
  72 ax1.set_xticklabels(xtkl1, size=ticksize)
  73 ax1.set_yticklabels(ytkl1, size=ticksize)
  74
  75 # put <v> into the plot
  76 ax2.plot(s.r, s.x, c='b', marker=',', ms=0.1, ls='')
  77 ax2.set_xlim(xlim2)
  78 ax2.set_ylim(ylim2)
  79 #ax2.set_xticks(xtks2)
  80 ax2.set_yticks(ytks2)
  81 ax2.set_ylabel(ylbl2, size=tsize)
  82 ax2.set_xlabel(xlbl2, size=tsize)
  83 ax2.set_xticklabels(ax2.get_xticks(), size=ticksize)
  84 ax2.set_yticklabels(ax2.get_yticks(), size=ticksize)
  85
  86 # adjust subplots
  87 plt.subplots_adjust(hspace=0)
  88
  89 # stream the whole mess into a file
  90 plt.savefig(FOUT)
  91
  92 # EOf