meteortools.rmsutils.pickleToKml

  1# Copyright (C) 2018-2023 Mark McIntyre
  2import numpy as np
  3import json
  4import os 
  5import pandas as pd
  6import matplotlib.pyplot as plt
  7
  8from ..fileformats import trackCsvtoKML
  9try:
 10    from wmpl.Utils.Pickling import loadPickle
 11except:
 12    print('WMPL not available')
 13
 14
 15def pickleToKml(picklename, outdir=None):
 16    """Create a KML file of the trajectory in a solution pickle
 17     
 18    Arguments:  
 19        picklename: [string] full path to the solution pickle file  
 20
 21    Keyword Args:  
 22        outdir: [string] where to save the KML. Default same place as pickle. 
 23    
 24    Returns:   
 25        the KML file as a tuple  
 26    """
 27    pickpath, trajname = os.path.split(picklename)
 28    data = loadPickle(pickpath, trajname).toJson()
 29    jsd = json.loads(data)
 30    lats = []
 31    lons = []
 32    hts = [] 
 33    times = []
 34    for js in jsd['observations']:
 35        k = list(js.keys())[0]
 36        thisstat = js[k]
 37        lats += thisstat['model_lat']
 38        lons += thisstat['model_lon']
 39        hts += thisstat['model_ht']
 40        times += thisstat['time_data']
 41    df = pd.DataFrame({"lats": np.degrees(lats), "lons": np.degrees(lons), "alts": hts, "times": times})
 42    df = df.sort_values(by=['times', 'lats'])
 43    os.makedirs(outdir, exist_ok=True)
 44    return trackCsvtoKML(trajname.replace('.pickle', 'csv'), df, saveOutput=True, outdir=outdir)
 45
 46
 47def pickleTo2dTrack(picklename, outdir=None):
 48    """Create a 2d graph of the trajectory in a solution pickle
 49     
 50    Arguments:  
 51        picklename: [string] full path to the solution pickle file  
 52
 53    Keyword Args:  
 54        outdir: [string] where to save the plot. Default same place as pickle. 
 55    
 56    Returns:   
 57        none
 58    """
 59    pickpath, trajname = os.path.split(picklename)
 60    data = loadPickle(pickpath, trajname).toJson()
 61    jsd = json.loads(data)
 62    alts = [] 
 63    times = []
 64    rngs = []
 65    for js in jsd['observations']:
 66        k = list(js.keys())[0]
 67        thisstat = js[k]
 68        alts += thisstat['model_ht']
 69        times += thisstat['time_data']
 70        rngs += thisstat['model_range']
 71
 72    plt.clf()
 73    fig = plt.gcf()
 74    fig.set_size_inches(11.6, 8.26)
 75
 76    plt.plot(times, alts, linewidth=2)
 77    ax = plt.gca()
 78    ax.set(xlabel="Time(s)", ylabel="Altitude (m)")
 79    #print(min(alts), max(alts))
 80    ax.set_ylim(bottom = min(min(alts), 25000), top=max(max(alts), 120000))
 81    orig, fname = os.path.split(picklename)
 82    if outdir is None:
 83        outdir = orig
 84    fname = fname.replace('.pickle', '_2dtrack.png')
 85    f3dname = os.path.join(outdir, fname)
 86
 87    plt.title('Trajectory of {}'.format(fname[:15]))
 88    plt.tight_layout()
 89    plt.savefig(f3dname, dpi=200)
 90    plt.close()
 91
 92    plt.clf()
 93    fig = plt.gcf()
 94    fig.set_size_inches(11.6, 8.26)
 95
 96    plt.plot(rngs, alts, linewidth=2)
 97    ax = plt.gca()
 98    ax.set(xlabel="Range (m)", ylabel="Altitude (m)")
 99    #print(min(alts), max(alts))
100    ax.set_ylim(bottom = min(min(alts), 25000), top=max(max(alts), 120000))
101    orig, fname = os.path.split(picklename)
102    if outdir is None:
103        outdir = orig
104    fname = fname.replace('.pickle', '_2drange.png')
105    f3dname = os.path.join(outdir, fname)
106
107    plt.title('Trajectory of {}'.format(fname[:15]))
108    plt.tight_layout()
109    plt.savefig(f3dname, dpi=200)
110    plt.close()
111    return