Source code for pysatMadrigal.instruments.omni2_imf

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Full license can be found in License.md
# Full author list can be found in .zenodo.json file
# DOI:10.5281/zenodo.3824979
#
# DISTRIBUTION STATEMENT A: Approved for public release. Distribution is
# unlimited.
# ----------------------------------------------------------------------------
"""Supports access to OMNI 2 IMF data archived at Madrigal.

Properties
----------
platform
    'omni2'
name
    'imf'
tag
    None supported
inst_id
    None supported

Note
----
Please provide name (user) and email (password) when downloading data with this
routine.

Glenn Campbell and Bill Rideout completely rebuilt the Madrigal interplanetary
magnetic field data using data from:
ftp://nssdcftp.gsfc.nasa.gov/spacecraft_data/omni/.  The old file had
originally come from Cedar and had gaps even in places where there was data
available.  This new Madrigal file is based on the Omni 2 data set, described
at http://nssdc.gsfc.nasa.gov/omniweb/. (4 May 2004, brideout@haystack.mit.edu)

The OMNI data may be directly downloaded using pysatNASA and is now described
at: https://omniweb.gsfc.nasa.gov/html/ow_data.html

Warnings
--------
The entire data set (27 Nov 1963 through 3 Jun 2019) is provided in a single
file on Madrigal. The download method will break this file up by year for
easier access.

Examples
--------
::


    import datetime as dt
    import pysat
    import pysatMadrigal as py_mad

    # Download IMF data from Madrigal
    imf = pysat.Instrument(inst_module=py_mad.instruments.omni2_imf)
    imf.download(start=py_mad.instruments.omni2_imf.madrigal_start,
                 user='Firstname+Lastname', password='email@address.com')
    imf.load(date=dt.datetime(1981, 1, 1))

"""

import datetime as dt
import functools

import pysat

from pysatMadrigal.instruments.methods import general

# ----------------------------------------------------------------------------
# Instrument attributes

platform = 'omni2'
name = 'imf'
tags = {'': ''}
inst_ids = {'': list(tags.keys())}
pandas_format = True

# Madrigal tags and limits
madrigal_inst_code = 120
madrigal_tag = {'': {'': "30012"}}
madrigal_start = dt.datetime(1963, 11, 27)
madrigal_end = dt.datetime(2019, 6, 4)

# Local attributes
#
# Need a way to get the filename strings for a particular instrument unless
# wildcards start working
supported_tags = {
    inst_id: {tag: general.madrigal_file_format_str(madrigal_inst_code,
                                                    verbose=False)
              for tag in inst_ids[inst_id]} for inst_id in inst_ids.keys()}
remote_tags = {ss: {kk: supported_tags[ss][kk].format(file_type='hdf5')
                    for kk in inst_ids[ss]} for ss in inst_ids.keys()}

# ----------------------------------------------------------------------------
# Instrument test attributes

_test_dates = {inst_id: {tag: madrigal_start for tag in inst_ids[inst_id]}
               for inst_id in inst_ids.keys()}
_test_download = {inst_id: {tag: True for tag in inst_ids[inst_id]}
                  for inst_id in inst_ids.keys()}
_clean_warn = {inst_id: {tag: {clvl: [('logger', 'WARN',
                                       "No cleaning available", clvl)]
                               for clvl in ['clean', 'dusty', 'dirty']}
                         for tag in inst_ids[inst_id]}
               for inst_id in inst_ids.keys()}

# ----------------------------------------------------------------------------
# Instrument methods


[docs] def init(self): """Initialize the Instrument object in support of Madrigal access.""" # Set the standard pysat attributes self.acknowledgements = ''.join([general.cedar_rules(), '\nFor full ', 'acknowledgement info, please see: ', 'https://omniweb.gsfc.nasa.gov/html/', 'citing.html']) self.references = ' '.join(('J.H. King and N.E. Papitashvili, Solar', 'wind spatial scales in and comparisons', 'of hourly Wind and ACE plasma and', 'magnetic field data, J. Geophys. Res.,', 'Vol. 110, No. A2, A02209,', '10.1029/2004JA010649.')) # Remind the user of the Rules of the Road pysat.logger.info(self.acknowledgements) return
[docs] def clean(self): """Raise warning that cleaning is not needed for the OMNI2 data. Note ---- Supports 'clean', 'dusty', 'dirty' in the sense that it prints a message noting there is no cleaning. 'None' is also supported as it signifies no cleaning. Routine is called by pysat, and not by the end user directly. """ pysat.logger.warning("No cleaning available for the collected Omni 2 IMF") return
# ---------------------------------------------------------------------------- # Instrument functions # # Use the default Madrigal and pysat methods file_cadence = madrigal_end - madrigal_start two_digit_year_break = 50 # Set the download routine download = functools.partial(general.download, inst_code=str(madrigal_inst_code), kindat=madrigal_tag['']['']) # Set the list routine list_files = functools.partial(general.list_files, supported_tags=supported_tags, file_cadence=file_cadence, two_digit_year_break=two_digit_year_break) # Set list_remote_files routine list_remote_files = functools.partial(general.list_remote_files, supported_tags=remote_tags, inst_code=madrigal_inst_code, kindats=madrigal_tag, two_digit_year_break=two_digit_year_break)
[docs] def load(fnames, tag='', inst_id=''): """Load the OMNI2 IMF data. Parameters ----------- fnames : list List of filenames tag : str tag name used to identify particular data set to be loaded. This input is nominally provided by pysat itself. (default='') inst_id : str Instrument ID used to identify particular data set to be loaded. This input is nominally provided by pysat itself. (default='') Returns -------- data : pds.DataFrame Object containing IMF data meta : pysat.Meta Object containing metadata such as column names and units Raises ------ ValueError Unexpected time variable names """ # Cycle through all the filenames, getting the desired start and stop times fstart = None fstop = None for fname_date in fnames: # Split the date from the filename fname = fname_date[:-11] fdate = dt.datetime.strptime(fname_date[-10:], '%Y-%m-%d') fstop = fdate if fstart is None: fstart = fdate fstop += dt.timedelta(days=1) # There is only one file for this Instrument data, meta = general.load([fname], tag=tag, inst_id=inst_id) # Test to see if there is data beyond the expected file end if data.index[-1] > madrigal_end: pysat.logger.critical(''.join(['There is data beyond ', '{:}'.format(madrigal_end), ' in the ', 'OMNI2 IMF file, please notify the ', 'pysatMadrigal developers so that they ', 'can update this Instrument.'])) # Select the data for the desired time period data = data[fstart:fstop] return data, meta