

In [2]:

    
import numpy as np
from ForwardEquation import forwardEquation, forwardEquation_dask
from CalcRays import calcRays,calcRays_dask

def simulateDatapack(templateDatapack,neModelTurbulentTCI,i0, antIdx = -1, timeIdx=[0], dirIdx=-1):
    antennas,antennaLabels = templateDatapack.get_antennas(antIdx = antIdx)
    patches, patchNames = templateDatapack.get_directions(dirIdx = dirIdx)
    times,timestamps = templateDatapack.get_times(timeIdx=timeIdx)
    Na = len(antennas)
    Nt = len(times)
    Nd = len(patches)  
    fixtime = times[Nt>>1]
    phase = templateDatapack.getCenterDirection()
    arrayCenter = templateDatapack.radioArray.getCenter()
    rays = calcRays(antennas,patches,times, arrayCenter, fixtime, phase, neModelTurbulentTCI, templateDatapack.radioArray.frequency, True, 1000, 1000)
    mTCI = neModelTurbulentTCI.copy()
    K_ne = np.mean(mTCI.m)
    mTCI.m /= K_ne
    np.log(mTCI.m,out=mTCI.m)
    g = forwardEquation_dask(rays,K_ne,mTCI,i0)
    datapack = templateDatapack.clone()
    datapack.set_dtec(g,antIdx = antIdx, timeIdx=timeIdx, dirIdx = dirIdx, refAnt=antennaLabels[i0])
    return datapack

def test_simulateDatapack():
    from InitialModel import createTurbulentlModel, createInitialModel
    from RealData import DataPack
    from AntennaFacetSelection import selectAntennaFacets
    from RealData import plotDataPack
    from PlotTools import animateTCISlices
    import os
    datapack = DataPack(filename="output/test/datapackObs.hdf5")
    for i in range(1,5):
        try:
            os.makedirs("output/test/simulate/simulate_{}/fig".format(i))
        except:
            pass
        N = 15
        #datapackSel = selectAntennaFacets(N, datapack, antIdx=-1, dirIdx=-1, timeIdx = [0])
        turbTCI = createTurbulentlModel(datapack,antIdx = -1, timeIdx = [0], dirIdx = -1, zmax = 1000.,spacing=5.)
        nePriorTCI = createInitialModel(datapack,antIdx = -1, timeIdx = [0], dirIdx = -1, zmax = 1000.,spacing=5.)
        turbTCI.save("output/test/simulate/simulate_{}/neModel.hdf5".format(i))
        datapackSim = simulateDatapack(datapack,turbTCI,0, antIdx = -1, timeIdx=[0], dirIdx=-1)
        datapackSim.save("output/test/simulate/simulate_{}/datapackSim.hdf5".format(i))
        plotDataPack(datapackSim,antIdx=-1,timeIdx=[0], dirIdx=-1,figname="output/test/simulate/simulate_{}/dobs".format(i))
        turbTCI.m -= nePriorTCI.m
        animateTCISlices(turbTCI,"output/test/simulate/simulate_{}/fig".format(i))

if __name__ == '__main__':
    test_simulateDatapack()









    



Setting refAnt: CS201HBA0
Using radio array Radio Array: 1.50000e+08 MHz, Longitude 6.84 deg Latitude 52.91 deg Height 30.97 m
Using phase center 217.62933909313725 deg 34.67457328431372 deg
Fixing frame at 2014-08-10T13:00:00.000
Elevation is 46.24627201340199 deg
Sun at zenith angle 40.729142650482565
Creating ionosphere model...
Found domain u in -192.61332681928474..184.8594891581511, v in -202.63304674758123..200.72636450761252, w in -132.6837669150769..1106.7633998610413
Nx=76 Ny=81 Nz=248 number of cells: 1526688
Using radio array Radio Array: 1.50000e+08 MHz, Longitude 6.84 deg Latitude 52.91 deg Height 30.97 m
Using phase center 217.62933909313725 deg 34.67457328431372 deg
Fixing frame at 2014-08-10T13:00:00.000
Elevation is 46.24627201340199 deg
Sun at zenith angle 40.729142650482565
Creating ionosphere model...
Found domain u in -192.61332681928474..184.8594891581511, v in -202.63304674758123..200.72636450761252, w in -132.6837669150769..1106.7633998610413
Nx=76 Ny=81 Nz=248 number of cells: 1526688
Casting rays: 1972
splitting over directions
Loaded 58 antennas, 3595 times, 34 directions
Setting refAnt: CS201HBA0
Setting refAnt: CS201HBA1
plotting levels : [-90261661436.163925, -68145920843.564247, -26096912392.126373, -11385533790.550196, -4308106486.2903709, -1695786094.3626962, -547975203.82301664, -82669503.9079023, -2214631.8468975578, 1637879.105456128, 10358568.449528845, 387433145.1153754, 1664727650.3260827, 5112578638.3855801, 12626859864.152283, 28539880812.205265, 105906421326.01189]






    



C:\Users\josh\Anaconda3\envs\compute\lib\site-packages\matplotlib\contour.py:370: RuntimeWarning: invalid value encountered in true_divide
  dist = np.add.reduce(([(abs(s)[i] / L[i]) for i in range(xsize)]), -1)






    



output/test/simulate/simulate_1/fig/animation.mp4 exists already
Using radio array Radio Array: 1.50000e+08 MHz, Longitude 6.84 deg Latitude 52.91 deg Height 30.97 m
Using phase center 217.62933909313725 deg 34.67457328431372 deg
Fixing frame at 2014-08-10T13:00:00.000
Elevation is 46.24627201340199 deg
Sun at zenith angle 40.729142650482565
Creating ionosphere model...
Found domain u in -192.61332681928474..184.8594891581511, v in -202.63304674758123..200.72636450761252, w in -132.6837669150769..1106.7633998610413
Nx=76 Ny=81 Nz=248 number of cells: 1526688
Using radio array Radio Array: 1.50000e+08 MHz, Longitude 6.84 deg Latitude 52.91 deg Height 30.97 m
Using phase center 217.62933909313725 deg 34.67457328431372 deg
Fixing frame at 2014-08-10T13:00:00.000
Elevation is 46.24627201340199 deg
Sun at zenith angle 40.729142650482565
Creating ionosphere model...
Found domain u in -192.61332681928474..184.8594891581511, v in -202.63304674758123..200.72636450761252, w in -132.6837669150769..1106.7633998610413
Nx=76 Ny=81 Nz=248 number of cells: 1526688
Casting rays: 1972
splitting over directions
Loaded 58 antennas, 3595 times, 34 directions
Setting refAnt: CS201HBA0
Setting refAnt: CS201HBA1
plotting levels : [-107025385762.44676, -83272556551.392136, -38495271751.367401, -17899848625.98056, -6977009047.1549397, -2734084302.6399941, -818776893.12203324, -217801347.83137473, -3743893.8993390477, 8194.7641585593246, 4500187.1341834115, 254312340.96339691, 933347509.91361034, 2138612297.1993058, 6344400847.9517126, 17792746416.098278, 64599449879.079987]
output/test/simulate/simulate_2/fig/animation.mp4 exists already
Using radio array Radio Array: 1.50000e+08 MHz, Longitude 6.84 deg Latitude 52.91 deg Height 30.97 m
Using phase center 217.62933909313725 deg 34.67457328431372 deg
Fixing frame at 2014-08-10T13:00:00.000
Elevation is 46.24627201340199 deg
Sun at zenith angle 40.729142650482565
Creating ionosphere model...
Found domain u in -192.61332681928474..184.8594891581511, v in -202.63304674758123..200.72636450761252, w in -132.6837669150769..1106.7633998610413
Nx=76 Ny=81 Nz=248 number of cells: 1526688
Using radio array Radio Array: 1.50000e+08 MHz, Longitude 6.84 deg Latitude 52.91 deg Height 30.97 m
Using phase center 217.62933909313725 deg 34.67457328431372 deg
Fixing frame at 2014-08-10T13:00:00.000
Elevation is 46.24627201340199 deg
Sun at zenith angle 40.729142650482565
Creating ionosphere model...
Found domain u in -192.61332681928474..184.8594891581511, v in -202.63304674758123..200.72636450761252, w in -132.6837669150769..1106.7633998610413
Nx=76 Ny=81 Nz=248 number of cells: 1526688
Casting rays: 1972
splitting over directions
Loaded 58 antennas, 3595 times, 34 directions
Setting refAnt: CS201HBA0
Setting refAnt: CS201HBA1
plotting levels : [-72355396437.369385, -54514484640.46476, -18627903339.574219, -6835044269.6852913, -2429787193.24962, -887847013.00574565, -255007510.7001133, -5625967.0053434819, -509099.88184136112, 2828641.307807893, 87069794.064546064, 707219920.70301509, 2077240044.9934757, 5528217622.7637205, 12860459005.551594, 27337901652.485275, 70509589017.356918]
output/test/simulate/simulate_3/fig/animation.mp4 exists already
Using radio array Radio Array: 1.50000e+08 MHz, Longitude 6.84 deg Latitude 52.91 deg Height 30.97 m
Using phase center 217.62933909313725 deg 34.67457328431372 deg
Fixing frame at 2014-08-10T13:00:00.000
Elevation is 46.24627201340199 deg
Sun at zenith angle 40.729142650482565
Creating ionosphere model...
Found domain u in -192.61332681928474..184.8594891581511, v in -202.63304674758123..200.72636450761252, w in -132.6837669150769..1106.7633998610413
Nx=76 Ny=81 Nz=248 number of cells: 1526688
Using radio array Radio Array: 1.50000e+08 MHz, Longitude 6.84 deg Latitude 52.91 deg Height 30.97 m
Using phase center 217.62933909313725 deg 34.67457328431372 deg
Fixing frame at 2014-08-10T13:00:00.000
Elevation is 46.24627201340199 deg
Sun at zenith angle 40.729142650482565
Creating ionosphere model...
Found domain u in -192.61332681928474..184.8594891581511, v in -202.63304674758123..200.72636450761252, w in -132.6837669150769..1106.7633998610413
Nx=76 Ny=81 Nz=248 number of cells: 1526688
Casting rays: 1972
splitting over directions
Loaded 58 antennas, 3595 times, 34 directions
Setting refAnt: CS201HBA0
Setting refAnt: CS201HBA1
plotting levels : [-80236428030.771942, -62162267684.531403, -25173525133.160213, -9377412089.8723354, -2835084882.46138, -979217682.74750936, -339474976.7303614, -6402792.8346637636, -351433.52140635409, 3461836.1255883519, 99038465.644033238, 736007769.85921741, 2520181850.6369572, 6364478414.6878557, 14871724461.425919, 33532010988.458546, 104279453230.52753]
output/test/simulate/simulate_4/fig/animation.mp4 exists already