source: anuga_work/development/sudi/sw_1d/periodic_waves/johns/run-plot.py @ 7837

#################Opening analytical solution################
infile = open('analytical_1230_12300.000000.csv', 'r')
table = []
for row in csv.reader(infile):
    table.append(row)
infile.close()
for r in range(len(table)):
    for c in range(len(table[0])):
        table[r][c] = float(table[r][c])

position = zeros(len(table),Float)
analitW12300 = zeros(len(table),Float)
analitP12300 = zeros(len(table),Float)
analitZ12300 = zeros(len(table),Float)
analitH12300 = zeros(len(table),Float)
analitU12300 = zeros(len(table),Float)
for r in range(len(table)):
    for c in range(len(table[0])):
        if c==0:
            position[r] = table[r][c]
        elif c==1:
            analitW12300[r] = table[r][c]
        elif c==2:
            analitP12300[r] = table[r][c]
        elif c==3:
            analitZ12300[r] = table[r][c]
        elif c==4:
            analitH12300[r] = table[r][c]
        elif c==5:
            analitU12300[r] = table[r][c]            

#################Opening numerical_cg solution################
infile = open('numerical_cg_1230_12300.000000.csv', 'r')
table = []
for row in csv.reader(infile):
    table.append(row)
infile.close()
for r in range(len(table)):
    for c in range(len(table[0])):
        table[r][c] = float(table[r][c])

numeric_cgW12300 = zeros(len(table),Float)
numeric_cgP12300 = zeros(len(table),Float)
numeric_cgZ12300 = zeros(len(table),Float)
numeric_cgH12300 = zeros(len(table),Float)
numeric_cgU12300 = zeros(len(table),Float)

for r in range(len(table)):
    for c in range(len(table[0])):
        if c==1:
            numeric_cgW12300[r] = table[r][c]
        elif c==2:
            numeric_cgP12300[r] = table[r][c]
        elif c==3:
            numeric_cgZ12300[r] = table[r][c]
        elif c==4:
            numeric_cgH12300[r] = table[r][c]
        elif c==5:
            numeric_cgU12300[r] = table[r][c]
            


#################Opening numerical_johns solution################
infile = open('numerical_johns_1230_12300.000000.csv', 'r')
table = []
for row in csv.reader(infile):
    table.append(row)
infile.close()
for r in range(len(table)):
    for c in range(len(table[0])):
        table[r][c] = float(table[r][c])

numeric_johnsW12300 = zeros(len(table),Float)
numeric_johnsP12300 = zeros(len(table),Float)
numeric_johnsZ12300 = zeros(len(table),Float)
numeric_johnsH12300 = zeros(len(table),Float)
numeric_johnsU12300 = zeros(len(table),Float)

for r in range(len(table)):
    for c in range(len(table[0])):
        if c==1:
            numeric_johnsW12300[r] = table[r][c]
        elif c==2:
            numeric_johnsP12300[r] = table[r][c]
        elif c==3:
            numeric_johnsZ12300[r] = table[r][c]
        elif c==4:
            numeric_johnsH12300[r] = table[r][c]
        elif c==5:
            numeric_johnsU12300[r] = table[r][c]




















#################Opening analytical solution################
infile = open('analytical_1260_12600.000000.csv', 'r')
table = []
for row in csv.reader(infile):
    table.append(row)
infile.close()
for r in range(len(table)):
    for c in range(len(table[0])):
        table[r][c] = float(table[r][c])

position = zeros(len(table),Float)
analitW12600 = zeros(len(table),Float)
analitP12600 = zeros(len(table),Float)
analitZ12600 = zeros(len(table),Float)
analitH12600 = zeros(len(table),Float)
analitU12600 = zeros(len(table),Float)
for r in range(len(table)):
    for c in range(len(table[0])):
        if c==0:
            position[r] = table[r][c]
        elif c==1:
            analitW12600[r] = table[r][c]
        elif c==2:
            analitP12600[r] = table[r][c]
        elif c==3:
            analitZ12600[r] = table[r][c]
        elif c==4:
            analitH12600[r] = table[r][c]
        elif c==5:
            analitU12600[r] = table[r][c]            

#################Opening numerical_cg solution################
infile = open('numerical_cg_1260_12600.000000.csv', 'r')
table = []
for row in csv.reader(infile):
    table.append(row)
infile.close()
for r in range(len(table)):
    for c in range(len(table[0])):
        table[r][c] = float(table[r][c])

numeric_cgW12600 = zeros(len(table),Float)
numeric_cgP12600 = zeros(len(table),Float)
numeric_cgZ12600 = zeros(len(table),Float)
numeric_cgH12600 = zeros(len(table),Float)
numeric_cgU12600 = zeros(len(table),Float)

for r in range(len(table)):
    for c in range(len(table[0])):
        if c==1:
            numeric_cgW12600[r] = table[r][c]
        elif c==2:
            numeric_cgP12600[r] = table[r][c]
        elif c==3:
            numeric_cgZ12600[r] = table[r][c]
        elif c==4:
            numeric_cgH12600[r] = table[r][c]
        elif c==5:
            numeric_cgU12600[r] = table[r][c]
            


#################Opening numerical_johns solution################
infile = open('numerical_johns_1260_12600.000000.csv', 'r')
table = []
for row in csv.reader(infile):
    table.append(row)
infile.close()
for r in range(len(table)):
    for c in range(len(table[0])):
        table[r][c] = float(table[r][c])

numeric_johnsW12600 = zeros(len(table),Float)
numeric_johnsP12600 = zeros(len(table),Float)
numeric_johnsZ12600 = zeros(len(table),Float)
numeric_johnsH12600 = zeros(len(table),Float)
numeric_johnsU12600 = zeros(len(table),Float)

for r in range(len(table)):
    for c in range(len(table[0])):
        if c==1:
            numeric_johnsW12600[r] = table[r][c]
        elif c==2:
            numeric_johnsP12600[r] = table[r][c]
        elif c==3:
            numeric_johnsZ12600[r] = table[r][c]
        elif c==4:
            numeric_johnsH12600[r] = table[r][c]
        elif c==5:
            numeric_johnsU12600[r] = table[r][c]



#################PLOTTING THE RESULTS##############################################
from pylab import clf,plot,title,xlabel,ylabel,legend,savefig,show,hold,subplot,ion
hold(False)
clf()

plot1 = subplot(311)
plot(#position*1e-4,analitW12300, position*1e-4,numeric_cgW12300, position*1e-4,numeric_johnsW12300,
     position*1e-4,analitW12600, position*1e-4,numeric_cgW12600, position*1e-4,numeric_johnsW12600,
     position*1e-4,analitZ12300,'k-')
#xlabel('Position')
ylabel('Stage')
#plot1.set_xlim([0.0,1.2])
#plot1.set_ylim([-5.0,3.0])#([-9.0e-3,9.0e-3])
#legend(('Analytical Solution', 'C-G Solution', 'Johns Solution'),
#       'lower left', shadow=False)

plot2 = subplot(312)
plot(#position*1e-4,analitP12300, position*1e-4,numeric_cgP12300, position*1e-4,numeric_johnsP12300)
     position*1e-4,analitP12600, position*1e-4,numeric_cgP12600, position*1e-4,numeric_johnsP12600)
#xlabel('Position')
ylabel('Momentum')
#plot1.set_xlim([0.0,1.2])
#plot1.set_ylim([-4.0,3.0])#([-9.0e-3,9.0e-3])
#legend(('Analytical Solution', 'C-G Solution', 'Johns Solution'),
#       'lower left', shadow=False)


plot3 = subplot(313)
plot(#position*1e-4,analitU12300, position*1e-4,numeric_cgU12300, position*1e-4,numeric_johnsU12300)
     position*1e-4,analitU12600, position*1e-4,numeric_cgU12600, position*1e-4,numeric_johnsU12600)
xlabel('Position/10,000')
ylabel('Velocity')
#plot1.set_xlim([0.0,1.2])
#plot3.set_ylim([-4.0,2.0])#([-9.0e-3,9.0e-3])
legend(('Exact Solution', 'C-G Solution', 'Johns Solution'),
       'lower left', shadow=False)

filename = "%s" %("12600.eps")
#savefig(filename)
show()