Filtering/Filtering.py at main · MarcusHatton/Filtering · 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
# -*- coding: utf-8 -*-
"""
Created on Wed Nov  2 16:59:20 2022

@author: mjh1n20
"""


# from system.Setup import setup

if __name__ == '__main__':

    from multiprocessing import Process, Pool
    import os
    import numpy as np
#    import matplotlib.pyplot as plt
    import pickle
    from timeit import default_timer as timer
    import h5py
    from scipy.interpolate import interpn
    from scipy.optimize import root, minimize
#    from mpl_toolkits.mplot3d import Axes3D
    from scipy.integrate import solve_ivp, quad
    import cProfile, pstats, io
    # from system.BaseFunctionality import *
    import system.BaseFunctionality
    from system.BaseFunctionality import Base
    from functools import partial
    from itertools import repeat

    system = Base()

    # cProfile.run('fnc()')

#     test_vx_vy = [0.0,0.0] # vx, vy
#     test_vmag_vtheta = [0.0,0.0] # vmag, vtheta
#     coords = [6.0,0.0,0.0]
#     L = 0.01
#     tol = 1e-6

#     start = timer()
#     sol = minimize(residual_ib,x0=test_vx_vy,args=(coords,L),bounds=((-0.7,0.7),(-0.7,0.7)),tol=tol)#,method='CG')
# #     sol = minimize(residual,x0=test_vx_vy,args=(coords,L),bounds=((-0.7,0.7),(-0.7,0.7)),tol=tol)#,method='CG')
# #     sol = minimize(residual_ib,x0=test_vmag_vtheta,args=(coords,L),bounds=((0.0,0.9),(-1.5*np.pi, 1.5*np.pi)),tol=tol)#,method='CG')
#     print("time, mins:", (end - start)/60) # Time in seconds, e.g. 5.38091952400282
#     print("root-found: ",get_U_mu(sol.x))
#     print("blabla")

    t_range = [29.99,30.01]
    x_range = [0.2,0.3]
    y_range = [0.4,0.5]
    initial_guess = [0.0,0.0] # not used... should remove...
    L = 0.01 # need to be able to make this bigger really...
    f_rts = open("runtimes.txt", "a")
    f_obs = open("observers.txt", "a")
    # args = [([tr, tr], [xr, xr], [yr, yr], L, 1, 1, 1, initial_guess) \
    #         for tr in np.linspace(t_range[0], t_range[1], 3)\
    #         for xr in np.linspace(x_range[0], x_range[1], 41)\
    #         for yr in np.linspace(y_range[0], y_range[1], 21)]
    # args = [([tr, tr], [xr, xr], [yr, yr], L, 1, 1, 1, initial_guess) for tr in np.linspace(6.0, 6.0, 1) for xr in np.linspace(0.0, 0.0, 1) for yr in np.linspace(0.0, 0.0, 1)]
    KH_observers = system.find_observers(t_range,x_range,y_range,L,3,26,26,initial_guess)
    # print(KH_observers)
    with open('KH_observers_2998_32626_x0203_y0405.pickle', 'wb') as handle:
        pickle.dump(KH_observers, handle, protocol=pickle.HIGHEST_PROTOCOL)

    np.savetxt('coords2998_32626_x0203_y0405.txt',KH_observers[0])
    np.savetxt('obs2998_32626_x0203_y0405.txt',KH_observers[1])

    # handle = open('KH_observers.pickle', 'wb')
    # start = timer()
    # with Pool(40) as p:
    #     results = p.starmap(system.find_observers, args)
    #     pickle.dump(np.array(results), handle, protocol=pickle.HIGHEST_PROTOCOL)
    #     f_obs.write(str(results))
    #     mid = timer()
        # end = timer()

    # # mid = timer()
    # print("time, mins:", (mid - start)/60,"\n") # Time in seconds, e.g. 5.38091952400282
    # f_rts.write("Parallel time, mins:"+ str((mid - start)/60) + "\n") # Time in seconds, e.g. 5.38091952400282

# #    system.find_observers(t_range1, x_range,y_range,L,1,1,1,initial_guess)
# #    system.find_observers(t_range2, x_range,y_range,L,1,1,1,initial_guess)
# #    print(system.find_observers(t_range3, x_range,y_range,L,1,1,1,initial_guess))
# #    print(system.find_observers(t_range4, x_range,y_range,L,1,1,1,initial_guess))
# #    print(system.find_observers(t_range5, x_range,y_range,L,1,1,1,initial_guess))

    # end = timer()
    # print("time, mins:", (end - mid)/60,"\n") # Time in seconds, e.g. 5.38091952400282
    # f_rts.write("Serial time, mins:"+str((end - mid)/60) + "\n") # Time in seconds, e.g. 5.38091952400282
    # f_rts.close()
    # f_obs.close()