TTF/comparison.py at master · bradcownden/TTF · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
# -*- coding: utf-8 -*-
"""
Created on Thu Dec  3 14:44:31 2015

@author: bradc
"""

"""
Compare values of X, Y, S, R, T between those calculated by recursion
relation vs those calculated by integration. Read in data from files, ensure
sorting convention is the same, then calculate percent difference
"""

import numpy as np
import matplotlib.pyplot as mp
import recursiontools as rt


# Load the data files and calculate the differences
"""
Xme = np.loadtxt("d3X.dat",ndmin=2)
Xme.sort(axis=1)
indx = np.lexsort((Xme[:,4],Xme[:,3],Xme[:,2],Xme[:,1]))
Xmeprime = Xme[indx]

Xnils = np.loadtxt("NilsData/X_30.dat",ndmin=2)
Xnils.sort(axis=1)

compx = np.zeros(Xmeprime.shape[0])

for i in range(Xmeprime.shape[0]):
    compx[i] = (Xmeprime[i][0] - Xnils[i][0])

###########################################################

Yme = np.loadtxt("d3Y.dat",ndmin=2)
Yme.sort(axis=1)
indy = np.lexsort((Yme[:,4],Yme[:,3],Yme[:,2],Yme[:,1]))
Ymeprime = Yme[indy]

Ynils = np.loadtxt("NilsData/Y_30.dat",ndmin=2)
Ynils.sort(axis=1)

compy = np.zeros(Ymeprime.shape[0])

for i in range(Ymeprime.shape[0]):
    compy[i] = (Ymeprime[i][0] - Ynils[i][0])
"""

###########################################################
###########################################################

# Plot the results
"""
mp.figure(1)

mp.subplot(211)
x = range(Xmeprime.shape[0])
mp.plot(x,compx,'bo')
mp.ylabel('Difference in X Coefficients')
mp.xlabel('Index Number')
mp.title('X Coefficients Comparison')

mp.subplot(212)
x = range(Ymeprime.shape[0])
mp.plot(x,compy,'ro')
mp.ylabel('Difference in Y Coefficients')
mp.xlabel('Index Number')
mp.title('Y Coefficients Comparison')
mp.show()
"""

S = np.genfromtxt("d3S.dat",loose=True)
indy = np.lexsort((S[:,3],S[:,2],S[:,1],S[:,0]))
newS = S[indy]
print(newS[1000][4])
print(newS.shape[0])

Snils = np.genfromtxt("NilsData/S_30.dat")
#print(Snils[0:10,:])

comp = np.zeros((newS.shape[0],7))
for i in range(newS.shape[0]):
    ccount=0
    try:
        if newS[i][0] == Snils[i][0] and newS[i][1] == Snils[i][1] \
        and newS[i][2] == Snils[i][2] and newS[i][3] == Snils[i][3] \
        and str(newS[i][4]) != 'nan':
            comp[ccount][0] = newS[i][0]
            comp[ccount][1] = newS[i][1]
            comp[ccount][2] = newS[i][2]
            comp[ccount][3] = newS[i][3]
            comp[ccount][4] = newS[i][4]
            comp[ccount][5] = Snils[i][4]
            comp[ccount][6] = newS[i][4] - Snils[i][4]
            ccount = ccount+1
    except IndexError:
        break

print(comp)