from matplotlib import rcParams
rcParams['font.family'] = 'sans-serif'
rcParams['font.sans-serif'] = ['Arial']
import numpy as np
import matplotlib.pyplot as pl

def DRAW(data1,data2,data3,fnameF,bb,hh):
    gt='N=0: b'+str(int(bb))+' x h'+str(int(hh))
    xmin=50
    xmax=300
    ymin=0
    ymax=float(int(max(data1[:,1]))+1)
    dx=np.arange(xmin,xmax+50,50)
    dy=np.arange(ymin,ymax+1,0.5)
    fig=pl.figure()
    ax = pl.gca()
    pl.subplot(1,1,1)
    pl.xlabel('c.t.c. of Re-bar (mm)',fontsize=12)
    pl.ylabel('$M/bh^2$ (MPa)',fontsize=12)
    pl.xlim(xmin,xmax)
    pl.xticks(dx)
    pl.ylim(ymin,ymax)
    pl.yticks(dy)
    aspect = 0.7*(ax.get_xlim()[1] - ax.get_xlim()[0]) / (ax.get_ylim()[1] - ax.get_ylim()[0])
    ax.set_aspect(aspect)
    line1=pl.plot(data1[:,0],data1[:,1],marker='o',markersize=8, color='black',linewidth=1.5,label='T32 (double)')
    line2=pl.plot(data2[:,0],data2[:,1],marker='s',markersize=8, color='black',linewidth=1.5,label='T25 (double)')
    line3=pl.plot(data3[:,0],data3[:,1],marker='^',markersize=10,color='black',linewidth=1.5,label='T20 (double)')
    line1[0].set_dashes([6,2,6,2])
    line2[0].set_dashes([8,4,2,4])
    line3[0].set_dashes([8,4,2,4,2,4])
    pl.title('Ultimate Strength of Slab ('+gt+')')
    pl.grid(True,which='both',ls='-', color='0.65')
    pl.legend(shadow=True,prop={'size':12},loc='upper right',handlelength=4)
    pl.savefig(fnameF, dpi=100)
    pl.show()

bb=1000
hh=1000
list_t=[32,25,20]
list_c=[100,125,150,200,250]

for i in list_t:
    TT=int(i)
    fnameW='_'+str(int(TT))+'h'+str(int(hh))+'.txt'
    fout=open(fnameW,'w')
    for j in list_c:
        ctc=int(j)
        fnameR='out_'+str(int(TT))+'ctc'+str(int(ctc))+'h'+str(int(hh))+'.txt'
        fin=open(fnameR,'r')
        for text in fin:
            text=text.strip()
            text=text.split()
            str_i=text[0]
            if str_i=='M0':
                sig_b=float(text[2])
                break
        fin.close()
        print(fnameR,str_i,sig_b)
        print('%15.7e %15.7e'%(ctc,sig_b),file=fout)
    fout.close()

data1=np.loadtxt('_32h1000.txt')
data2=np.loadtxt('_25h1000.txt')
data3=np.loadtxt('_20h1000.txt')
fnameF='fig_rebar_m0.eps'
DRAW(data1,data2,data3,fnameF,bb,hh)