Added the code written in the field

Author: nholschuh

cresis_season.py

@@ -0,0 +1,98 @@
+################ This is the import statement required to reference scripts within the package
+import os,sys,glob
+ndh_tools_path_opts = [
+    '/mnt/data01/Code/',
+    '/home/common/HolschuhLab/Code/'
+]
+for i in ndh_tools_path_opts:
+    if os.path.isfile(i): sys.path.append(i)
+################################################################################################
+
+
+def cresis_season(y,m=0,d=0,ant1_gre2=1):
+    """
+    % (C) Nick Holschuh - Amherst College -- 2022 ([email protected])
+    %
+    %     This function finds the season name associated with a flight day
+    % There are a small number of flight days in which both campaigns
+    % were running apparently. By default, this code will assume you want
+    % the Antarctic Season. If you would prefer it prompt you to ask for
+    % a selection, chanage the value to zero.  
+    %
+    %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+    % The inputs are:
+    %
+    %     y - either the year, or a string for the filename you want the season for
+    %     m - the month
+    %     d - the day
+    %     ant1_gre2 - force a particular continent for days where there are surveys in both places
+    %
+    %%%%%%%%%%%%%%%
+    % The outputs are:
+    %     season_out - A dictionary with information about the matching season
+    %
+    %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+    """     
+    from datetime import date
+    import numpy as np
+    import os
+    import sys
+    import glob
+    sys.path.append('/mnt/data01/Code/')
+    import NDH_Tools as ndh
+
+    ########################### Here we find the season metadata. This file was produced
+    ########################### from an external matlab script
+    season_metadata_dirs = [
+        '/mnt/data01/Data/RadarData/CReSIS_Filestructure/'
+    ]
+    for i in season_metadata_dirs:
+        if os.path.isdir(i): season_metadata_path=i
+
+
+    season_opts = ndh.loadmat(season_metadata_path+'season_metadata.mat')
+    
+    if isinstance(y,str) == 1:
+        if y[0] == 'D':
+            m = int(y[9:11])
+            d = int(y[11:13])
+            y = int(y[5:9])
+        else:
+            m = int(y[4:6])
+            d = int(y[6:8])    
+            y = int(y[0:4])
+    
+    target_date = date.toordinal(date(y,m,d))
+    
+    full_dates = np.concatenate([season_opts['a_dates'],season_opts['g_dates']])
+    ############### Matlab starts at year 0, python at year 1. So there is a 1 year offset applied here
+    full_dates[:,0] = full_dates[:,0]-366
+    full_dates = np.concatenate([full_dates,np.expand_dims(np.concatenate([np.ones(len(season_opts['a_dates'][:,0])),
+                                           np.ones(len(season_opts['g_dates'][:,0]))*2]),0).T],1)
+    
+    match_ind = np.where(full_dates[:,0] == target_date)[0]
+    exact_flag = 1;
+    
+    if ant1_gre2 == 1:
+        if len(match_ind) > 1:
+            match_ind = match_ind[0]
+    else:
+         if len(match_ind) > 1:
+            match_ind = match_ind[1]   
+        
+    if len(match_ind) == 0:
+        match_ind = ndh.find_nearest(full_dates[:,0],target_date);
+        match_ind = match_ind['index'][0]
+        exact_flag = 0;
+    
+    ############### You have to subtract one from the match ind to deal with matlabs indexing
+    if full_dates[match_ind,2] == 1:
+        season_out = season_opts['a_names'][int(full_dates[match_ind,1]-1)]
+    else:
+        season_out = season_opts['g_names'][int(full_dates[match_ind,1]-1)]
+        
+    if exact_flag == 0:
+        print('No exact match was found -- closest suggested season is the following: '+season_out[0])
+    
+    season_out = {'exact_match':exact_flag,'season':season_out[0],'data_dir':season_out[1],'Date':season_out[2]}
+    return season_out

find_cresisfiles.py

@@ -0,0 +1,81 @@
+################ This is the import statement required to reference scripts within the package
+import os,sys,glob
+ndh_tools_path_opts = [
+    '/mnt/data01/Code/',
+    '/home/common/HolschuhLab/Code/'
+]
+for i in ndh_tools_path_opts:
+    if os.path.isfile(i): sys.path.append(i)
+################################################################################################
+
+
+def find_cresisfiles(y,m=0,d=0,seg=0,frame=0):
+    """
+    % (C) Nick Holschuh - Amherst College -- 2022 ([email protected])
+    %
+    %     This function finds the season name associated with a flight day
+    %
+    %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+    % The inputs are:
+    %
+    %     y - either the year, or a string for the filename you want the season for
+    %     m - the month
+    %     d - the day
+    %     ant1_gre2 - force a particular continent for days where there are surveys in both places
+    %
+    %%%%%%%%%%%%%%%
+    % The outputs are:
+    %     season_out - A dictionary with information about the matching season
+    %
+    %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+    """     
+    from datetime import date
+    import os
+    import sys
+    import glob
+    sys.path.append('/mnt/data01/Code/')
+    import NDH_Tools as ndh
+
+    root_dir = '/mnt/data01/Data/RadarData/CReSIS_Filestructure/ct_data/rds/'
+    
+    if isinstance(y,str) == 1:
+        if y[0] == 'D':
+            seg = int(y[14:16])
+            frm = int(y[17:20])
+            m = int(y[9:11])
+            d = int(y[11:13])
+            y = int(y[5:9])
+        else:
+            seg = int(y[9:11])
+            frm = int(y[13:16])
+            m = int(y[4:6])
+            d = int(y[6:8])    
+            y = int(y[0:4])
+                
+    season = ndh.cresis_season(y,m,d)
+    dayseg_str = '%0.4d%0.2d%0.2d_%0.2d' % (y,m,d,seg)
+    filestr = 'Data_%s_%0.3d' % (dayseg_str,frm)
+    
+    processing_types = sorted(glob.glob(root_dir+season['season']+'/*/'))
+    search_types = ['standard','music','surf','DEM']
+    dir_names = [[],[],[],[]]
+    found_files = [[],[],[],[]]
+    
+    for ind0,ptype in enumerate(search_types):
+        type_fdrs,type_fdrs_ind = ndh.str_compare(processing_types,ptype)
+    
+        for ind1,type_fdr in enumerate(type_fdrs):
+            file_opts = sorted(glob.glob(type_fdr+dayseg_str+'/'+filestr+'.mat'))
+            for ind2, file_select in enumerate(file_opts):
+                found_files[ind0].append(file_select)
+                temp_dir_name = file_select.split('/')
+                dir_names[ind0].append(temp_dir_name[-3])
+    
+            #found_files[ind0] = ndh.flatten_list(found_files[ind0]);
+    
+    found_files = {'standard':found_files[0],'standard_dirs':dir_names[0],
+                   'music':found_files[1],'music_dirs':dir_names[1],
+                   'surf':found_files[2],'surf_dirs':dir_names[2],
+                   'DEM':found_files[3],'DEM_dirs':dir_names[3]}
+    
+    return found_files