/sunpy/lightcurve/sources/sdo.py

# -*- coding: utf-8 -*-
"""Provides programs to process and analyze EVE data."""
from __future__ import absolute_import

from sunpy.lightcurve import LightCurve

import os
from pandas.io.parsers import read_csv
from datetime import datetime  

class EVELightCurve(LightCurve):
    """SDO EVE light curve definition
    
    Examples
    --------
    >>> import sunpy
    >>> eve = sunpy.lightcurve.EVELightCurve()
    >>> eve = sunpy.lightcurve.EVELightCurve('~/Downloads/EVE_Fe_IX_171_averages.csv')
    >>> eve = sunpy.lightcurve.EVELightCurve('2012/06/20')
    >>> eve = sunpy.lightcurve.EVELightCurve("http://lasp.colorado.edu/eve/data_access/quicklook/quicklook_data/L0CS/LATEST_EVE_L0CS_DIODES_1m.txt")
    >>> 
    >>> eve.show()
    
    References
    ----------
    | http://lasp.colorado.edu/home/eve/data/data-access/
    """
    def __init__(self, *args, **kwargs):
        LightCurve.__init__(self, *args, **kwargs)

    def show(self, **kwargs):
        # Choose title if none was specified
        if not kwargs.has_key("title"):
            if len(self.data.columns) > 1:
                kwargs['title'] = 'EVE GOES Proxy Xray Flux (1 minute data)'
            else:
                if self._filename is not None:
                    base = self._filename.replace('_', ' ')
                    kwargs['title'] = os.path.splitext(base)[0]
                else:
                    kwargs['title'] = 'EVE Averages'

        #LightCurve.show(kwargs)
        from matplotlib import pyplot as plt
        self.data.plot(**kwargs)
        plt.show()
        
    def _get_default_uri(self):
        """Load latest level 0CS if no other data is specified"""
        return "http://lasp.colorado.edu/eve/data_access/quicklook/quicklook_data/L0CS/LATEST_EVE_L0CS_DIODES_1m.txt"
    
    def _get_url_for_date(self, date):
        """Returns a URL to the EVE data for the specified date
        
            @NOTE: currently only supports downloading level 0 data
        """
        base_url = 'http://lasp.colorado.edu/eve/data/quicklook/L0CS/SpWx/'
        return base_url + date.strftime('%Y/%Y%m%d') + '_EVE_L0CS_DIODES_1m.txt'
    
    def _parse_csv(self, filepath):
        """Parses an EVE CSV file"""
        fp = open(filepath, 'rb')
        
        # Determine type of EVE CSV file and parse
        line1 = fp.readline()
        fp.seek(0)

        if line1.startswith("Date"):
            return self._parse_average_csv(fp)
        elif line1.startswith(";"):
            return self._parse_level_0cs(fp)
    
    def _parse_average_csv(self, fp):
        """Parses an EVE Averages file"""
        return "", read_csv(fp, sep=",", index_col=0, parse_dates=True)
    
    def _parse_level_0cs(self, fp):
        """Parses and EVE Level 0CS file"""
        header = ""
        
        fields = ('xrs-b', 'xrs-a', 'sem', 'ESPquad', 'esp171', 
                  'esp257', 'esp304', 'esp366', 'espdark', 'megsp', 'megsdark', 
                  'q0esp', 'q1esp', 'q2esp', 'q3esp', 'cmlat', 'cmlon')
        
        line = fp.readline()
        
        # Read comment at top of file
        while line.startswith(";"):
            header += line
            line = fp.readline()
            
        # Next line is YYYY DOY MM DD
        parts = line.split(" ")
                
        year = int(parts[0])
        month = int(parts[2])
        day = int(parts[3])
        
        # function to parse date column (HHMM)
        parser = lambda x: datetime(year, month, day, int(x[0:2]), int(x[2:4]))

        data = read_csv(fp, sep="\s*", names=fields, index_col=0, date_parser=parser)
        
        return header, data