%%html
<style>
.cell-output-ipywidget-background {
background-color: transparent !important;
}
:root {
--jp-widgets-color: var(--vscode-editor-foreground);
--jp-widgets-font-size: var(--vscode-editor-font-size);
}
</style>
import datetime
import pyaurorax
aurorax = pyaurorax.PyAuroraX()
# download an hour of TREx NIR data
dataset_name = "TREX_NIR_RAW"
start_dt = datetime.datetime(2021, 11, 4, 3, 0)
end_dt = datetime.datetime(2021, 11, 4, 3, 59)
site_uid = "gill"
r = aurorax.data.ucalgary.download(dataset_name, start_dt, end_dt, site_uid=site_uid)
Downloading TREX_NIR_RAW files: 0%| | 0.00/51.2M [00:00<?, ?B/s]
# read in the hour of data
data = aurorax.data.ucalgary.read(r.dataset, r.filenames, n_parallel=5)
# set up working with the tools by just making a shorter name for our future calls
at = aurorax.tools
# scale all the images
#
# NOTE: you can scale all images or just one image
images_scaled = at.scale_intensity(data.data, min=400, max=2000)
at.display(images_scaled[:, :, 0], cmap="gray")
keogram = at.keogram.create(images_scaled, data.timestamp)
at.display(keogram.data, cmap="gray", figsize=(12, 4), aspect="auto")
at.set_theme("dark")
plot_dt = start_dt # need this timestamp for later plots, so we reassign it
keogram.plot(
title="TREx NIR Gillam %s" % (plot_dt.strftime("%Y-%m-%d UT%d")),
figsize=(12, 4),
cmap="gray",
aspect="auto",
)
# download skymap files
start_dt = datetime.datetime(2020, 1, 1)
end_dt = datetime.datetime(2022, 1, 1)
r = aurorax.data.ucalgary.download("TREX_NIR_SKYMAP_IDLSAV", start_dt, end_dt, site_uid=site_uid, overwrite=True)
r.filenames
Downloading TREX_NIR_SKYMAP_IDLSAV files: 0%| | 0.00/21.2M [00:00<?, ?B/s]
[PosixPath('/home/darrenc/pyaurorax_data/TREX_NIR_SKYMAP_IDLSAV/gill/gill_20200120/nir_skymap_gill_20200120-+_v01.sav'), PosixPath('/home/darrenc/pyaurorax_data/TREX_NIR_SKYMAP_IDLSAV/gill/gill_20200315/nir_skymap_gill_20200315-+_v01.sav'), PosixPath('/home/darrenc/pyaurorax_data/TREX_NIR_SKYMAP_IDLSAV/gill/gill_20201221/nir_skymap_gill_20201221-+_v01.sav'), PosixPath('/home/darrenc/pyaurorax_data/TREX_NIR_SKYMAP_IDLSAV/gill/gill_20210312/nir_skymap_gill_20210312-+_v01.sav'), PosixPath('/home/darrenc/pyaurorax_data/TREX_NIR_SKYMAP_IDLSAV/gill/gill_20210819/nir_skymap_gill_20210819-+_v01.sav'), PosixPath('/home/darrenc/pyaurorax_data/TREX_NIR_SKYMAP_IDLSAV/gill/gill_20211209/nir_skymap_gill_20211209-+_v01.sav')]
# we want to use the second last one, since that's for the time period of the data we're looking at
skymaps = aurorax.data.ucalgary.read(r.dataset, r.filenames[-2])
skymaps.data[0].pretty_print()
Skymap: filename : /home/darrenc/pyaurorax_data/TREX_NIR_SKYMAP_IDLSAV/gill/gill_20210819/nir_skymap_gill_20210819-+_v01.sav full_azimuth : array(dims=(256, 256), dtype=>f4) full_elevation : array(dims=(256, 256), dtype=>f4) full_map_altitude : array(dims=(3,), dtype=>f4) full_map_latitude : array(dims=(3, 257, 257), dtype=>f4) full_map_longitude : array(dims=(3, 257, 257), dtype=>f4) generation_info : SkymapGenerationInfo(...) get_precalculated_altitudes: <bound method Skymap.get_precalculated_altitudes of Skymap(project_uid=nir, site_uid=gill, imager_uid=nir-216, site_map_latitude=56.376724, site_map_longitude=265.356323, ...)> imager_uid : nir-216 project_uid : nir site_map_altitude : 0.0 site_map_latitude : 56.376724 site_map_longitude : 265.35632 site_uid : gill version : v01
keogram.set_geographic_latitudes(skymaps.data[0])
keogram.plot(
figsize=(12, 4),
cmap="gray",
y_type="geo",
aspect="auto",
title="TREx NIR Gillam %s" % (plot_dt.strftime("%Y-%m-%d UT%d")),
)
keogram.set_magnetic_latitudes(skymaps.data[0], start_dt)
keogram.plot(
figsize=(12, 4),
cmap="gray",
y_type="mag",
aspect="auto",
title="TREx NIR Gillam %s" % (plot_dt.strftime("%Y-%m-%d UT%d")),
)