feat: add support for downloading Cluster CODIF psd data

Author: LenaMacke

pyspedas/projects/cluster/cis.py

@@ -7,6 +7,7 @@
 
 def cis(trange:List[str]=['2018-11-5', '2018-11-6'],
         probe:Union[str,List[str]]='1',
+        option:str='mom',
         datatype:str='pp',
         prefix:str='',
         suffix:str='',
@@ -33,6 +34,10 @@ def cis(trange:List[str]=['2018-11-5', '2018-11-6'],
             List of probes to load.  Valid options: '1','2','3','4'
             Default: '1'
 
+        option: str
+            The data option to load. Valid options: 'mom' (moments), 'psd_h1' (H+ PSD), 'psd_he1' (He+ PSD), 'psd_o1' (O+ PSD)
+            Default: 'mom'
+
         datatype: str
             Data type; Valid options:
             Default: 'pp'
@@ -94,6 +99,14 @@ def cis(trange:List[str]=['2018-11-5', '2018-11-6'],
     >>> tplot(['N_p__C1_PP_CIS','N_O1__C1_PP_CIS','N_He1__C1_PP_CIS','N_He2__C1_PP_CIS','N_HIA__C1_PP_CIS'])
 
     """
-    return load(instrument='cis', trange=trange, probe=probe, datatype=datatype, prefix=prefix, suffix=suffix, get_support_data=get_support_data, varformat=varformat, varnames=varnames, downloadonly=downloadonly, notplot=notplot, no_update=no_update, time_clip=time_clip, force_download=force_download)
+    
+    if option == 'mom':
+        species = None
+    elif option.startswith('psd'):
+        species = option.split('_')[1]
+    else:
+        raise ValueError("Invalid option: " + option + ". Valid options are 'mom', 'psd_h1' 'psd_he1', 'psd_o1'.")
+
+    return load(instrument='cis', trange=trange, probe=probe, datatype=datatype, prefix=prefix, suffix=suffix, get_support_data=get_support_data, varformat=varformat, varnames=varnames, downloadonly=downloadonly, notplot=notplot, no_update=no_update, time_clip=time_clip, force_download=force_download, species=species)
 
 

pyspedas/projects/cluster/load.py

@@ -18,7 +18,9 @@ def load(trange:List[str]=['2018-11-5', '2018-11-6'],
          notplot:bool=False,
          no_update:bool=False,
          time_clip:bool=False,
-         force_download=False) -> List[str]:
+         force_download=False,
+         species=None
+         ) -> List[str]:
     """
     Load data from Cluster instruments
 
@@ -85,6 +87,9 @@ def load(trange:List[str]=['2018-11-5', '2018-11-6'],
             Download file even if local version is more recent than server version
             Default: False
 
+        species: str
+            Species to load psd data for. Valid options are: 'h1', 'he1', 'o1'.
+            Default: None
 
     Returns
     -------
@@ -137,7 +142,16 @@ def load(trange:List[str]=['2018-11-5', '2018-11-6'],
         elif instrument == 'aspoc':
             pathformat = 'c'+prb+'/'+datatype+'/asp/%Y/c'+prb+'_'+datatype+'_asp_%Y%m%d_v??.cdf'
         elif instrument == 'cis':
-            pathformat = 'c'+prb+'/'+datatype+'/'+instrument+'/%Y/c'+prb+'_'+datatype+'_'+instrument+'_%Y%m%d_v??.cdf'
+            if species == None:
+                pathformat = 'c'+prb+'/'+datatype+'/'+instrument+'/%Y/c'+prb+'_'+datatype+'_'+instrument+'_%Y%m%d_v??.cdf'
+            else: 
+                if species == 'h1':
+                    scd = 'proton' # second name for H+
+                if species == 'he1':
+                    scd = 'heplus' # second name for He+
+                if species == 'o1':
+                    scd = 'oplus' # second name for O+
+                pathformat = 'c'+prb+'/'+instrument+'-codif/'+scd+'_3ddist_highsens_phasespacedens'+'/%Y/c'+prb+'_cp_'+instrument+'-codif_hs_'+species+'_psd'+'_%Y%m%d_v????????.cdf'
         elif instrument == 'dwp':
             pathformat = 'c'+prb+'/'+datatype+'/'+instrument+'/%Y/c'+prb+'_'+datatype+'_'+instrument+'_%Y%m%d_v??.cdf'
         elif instrument == 'edi':