CRS_Recipe_Functions.ipynb

{"cells":[{"cell_type":"code","execution_count":null,"metadata":{},"outputs":[],"source":["# -*- coding: utf-8 -*-"]},{"cell_type":"markdown","metadata":{},"source":["<br>\n","       Cloud Radar System (CRS) Reflectivity and <br>\n","       Doppler Velocity Quick View Functions<br>\n","<br>\n","       Decription: This script contains the functions<br>\n","       used in the main CRS_Recipe_Code.py script to<br>\n","       plot CRS reflectivity and Doppler velocity data<br>\n","<br>\n","       Authors: Essence Raphael and Yuling Wu <br>\n","       Information and Technology Systems Center (ITSC)<br>\n","       University of Alabama in Huntsville<br>\n","       <br>\n","       Last Edit Date: 30 August 2021<br>\n","<br>"]},{"cell_type":"markdown","metadata":{},"source":["Import Python packages and modules"]},{"cell_type":"code","execution_count":null,"metadata":{},"outputs":[],"source":["import re\n","import sys\n","import numpy as np\n","import os\n","import matplotlib.pyplot as plt\n","import matplotlib.dates as mdates\n","from datetime import datetime, timedelta \n","from matplotlib.colors import ListedColormap\n","from matplotlib import cm \n","from pathlib import Path\n","\n","def select_campaign():\n","    \"\"\"\n","    User selects which CRS campaign dataset they would\n","    like the code to read\n","    Return name of campaign (CRS dataset)\n","    \"\"\"\n","    campaigns={1:'impacts',2:'goesrplt',3:'olympex',4:'iphex'}\n","    print(\"Select the CRS campaign dataset you would like to plot from the list below by number.\")\n","\n","    for i in campaigns:\n","        print(\"Campaign #{}: {}\".format(i,campaigns[i]))\n","\n","    while True:\n","        num = input (\"\\n*Enter Campaign Number or 'Q' to quit: \") \n","        if num.isdigit():\n","            if int(num) in campaigns:\n","                dataset=campaigns[int(num)]\n","                print('The {} campaign is selected'.format(dataset))\n","                break\n","            else: \n","                print(num,\"is not a valid campaign number. Select a flight from the list:\\n\")\n","        elif num=='Q':\n","            return sys.exit(\"User selected 'quit'\")\n","        else:\n","            print('\\n%%Invalid flight No. format%% \\nTry again.\\n')\n","    return dataset\n","\n","def select_flight_impacts(dataDir):\n","    \"\"\"\n","    User selects among the available flights from the files on their computer\n","    Return date in 'yyyymmdd' of the selected flight\n","    \"\"\"\n","        \n","    impacts_files = [path for path in Path(dataDir).rglob('IMPACTS_CRS_L1B_*.h5')]\n","    impacts_files = [os.path.normpath(i) for i in impacts_files]\n","    \n","    #Check whether impacts files were found in the directory. Return 'None' if no\n","    #files were found. Continue through the code if files were found.\n","    if len(impacts_files)==0:\n","        print(\"%%There are no impacts data files in the currect directory. Try again%%\")\n","        return None\n","    else: \n","        pass\n","    \n","    flight_dates = [] \n","    \n","    for i in impacts_files:\n","        fname_split = re.split(r'_', os.path.basename(i))\n","        start_date = fname_split[4]\n","        \n","        flight_dates.append(datetime.strptime(start_date[:8], '%Y%m%d').date().isoformat())\n","   \n","    print('Flight Dates:')\n","    for i in flight_dates:\n","        print('{}'.format(i))\n","\n","    while True:\n","        num = input (\"\\n*Enter date in [yyyy-mm-dd] or Q to quit:\") \n","        if num.isdigit():\n","            if num in flight_dates:\n","                fdate = num\n","                print('Flight date {} is selected.'.format(num))\n","                break\n","            else: \n","                print(num,\"is not a valid flight date. Select a flight from the list.\")\n","        elif num=='Q':\n","            return sys.exit(\"User selected 'quit'\")\n","        else:\n","            try:\n","                Fdate=datetime.strptime(num, '%Y-%m-%d').strftime(\"%Y-%m-%d\")\n","                if Fdate in flight_dates:\n","                    fdate=Fdate\n","                    print('Flight date {} is selected.'.format(fdate))\n","                    break\n","                else: \n","                    print(Fdate,\"is not a valid flight date. Select a flight from the list:\")\n","                    \n","            except ValueError:\n","                print('\\n%%Invalid flight date format%% \\nTry again.\\n')\n","                \n","    selected_files = [path for path in Path(dataDir).rglob('IMPACTS_CRS_L1B_*'+fdate.replace('-','')+'*.h5')]\n","    selected_files = [os.path.normpath(i) for i in selected_files]\n","    return selected_files\n","\n","def select_time_impacts(selected_files):\n","    \"\"\"\n","    User selects among the available flight time periods \n","    Return the selected file name\n","    \"\"\"\n","    \n","    start_times = []\n","    end_times = []\n","    \n","    time_periods = {}\n","    \n","    for i in selected_files:\n","        fname_split = re.split(r'_', os.path.basename(i))\n","        \n","        start_time = fname_split[4]\n","        start_time = start_time[9:11] + ':' + start_time[11:13] + ':' + start_time[13:]\n","        start_times.append(start_time)\n","                \n","        end_time = fname_split[6]\n","        end_time = end_time[9:11] + ':' + end_time[11:13] + ':' + end_time[13:15]\n","        end_times.append(end_time)\n","        \n","        time_periods[selected_files.index(i)+1] = start_time + ' to ' + end_time\n","        \n","    for i in time_periods:\n","        print('The IMPACTS flight periods available for this date are: \\n #{} - {}'.format(i, time_periods[i]))\n","        \n","    while True:\n","        num = input (\"\\n*Enter valid flight period no. or Q to quit: \") \n","        if num.isdigit():\n","            if int(num) in time_periods:\n","                print('Flight period #{}: {} is selected.'.format(int(num),time_periods[int(num)]))\n","                break\n","            else: \n","                print(int(num),\"is not a valid flight period. Select a period no. from the list:\")            \n","        elif num=='Q':\n","            return sys.exit(\"User selected 'quit'\")\n","        else:\n","            print('\\n%%Invalid flight No. format%% \\nTry again.\\n')\n","    return os.path.basename(selected_files[int(num)-1]) #Returns selected file name\n","\n","def select_flight_goesrplt(dataDir):\n","    \"\"\"\n","    User selects among the available flights from the files on their computer \n","    Return date in 'yyyymmdd' of the selected flight\n","    \"\"\"\n","    goesrplt_files = [path for path in Path(dataDir).rglob('GOESR_CRS_L1B_*.nc')]\n","    goesrplt_files = [os.path.normpath(i) for i in goesrplt_files]         \n","    \n","    #Check whether goesrplt files were found in the directory. Return 'None' if no\n","    #files were found. Continue through the code if files were found.\n","    if len(goesrplt_files)==0:\n","        print(\"%%There are no goesrplt data files in the currect directory. Try again%%\")\n","        return None, None\n","    else: \n","        pass\n","    \n","    flight_dates = [] \n","    \n","    for i in goesrplt_files:\n","        fname_split = re.split(r'_', os.path.basename(i))\n","        start_date = fname_split[3]\n","        \n","        flight_dates.append(datetime.strptime(start_date, '%Y%m%d').date().isoformat())\n","   \n","    print('Flight Dates:')\n","    for i in flight_dates:\n","        print('{}'.format(i))\n","\n","    while True:\n","        num = input (\"\\n*Enter date in [yyyy-mm-dd] or Q to quit:\") \n","        if num.isdigit():\n","            if num in flight_dates:\n","                fdate = num\n","                print('Flight date {} is selected.'.format(num))\n","                break\n","            else: \n","                print(num,\"is not a valid flight date. Select a flight from the list.\")\n","        elif num=='Q':\n","            return sys.exit(\"User selected 'quit'\")\n","        else:\n","            try:\n","                Fdate=datetime.strptime(num, '%Y-%m-%d').strftime(\"%Y-%m-%d\")\n","                if Fdate in flight_dates:\n","                    fdate=Fdate\n","                    print('Flight date {} is selected.'.format(fdate))\n","                    break\n","                else: \n","                    print(Fdate,\"is not a valid flight date. Select a flight from the list:\")\n","                    \n","            except ValueError:\n","                print('\\n%%Invalid flight date format%% \\nTry again.\\n')\n","                \n","    selected_files = [path for path in Path(dataDir).rglob('GOESR_CRS_L1B_*'+fdate.replace('-','')+'*.nc')]\n","    selected_files = [os.path.normpath(i) for i in selected_files]\n","    return os.path.basename(selected_files[0]), fdate.replace('-','') \n","\n","\n","def select_flight_olympex(dataDir):\n","    \"\"\"\n","    User selects among the available flights from the files on their computer\n","    Returns the date in 'yyyymmdd' of the selected flight\n","    \"\"\"\n","        \n","    olympex_files = [path for path in Path(dataDir).rglob('olympex_CRS_*.nc')] \n","    olympex_files = [os.path.normpath(i) for i in olympex_files]\n","    \n","    #Check whether olympex files were found in the directory. Return 'None' if no\n","    #files were found. Continue through the code if files were found.\n","    if len(olympex_files)==0:\n","        print(\"%%There are no olympex data files in the currect directory. Try again%%\")\n","        return None, None\n","    else: \n","        pass\n","    \n","    flight_dates = [] \n","    \n","    for i in olympex_files:\n","        fname_split = re.split(r'_|-', os.path.basename(i))\n","        start_date = fname_split[2]\n","        \n","        flight_dates.append(datetime.strptime(start_date, '%Y%m%d').date().isoformat())\n","   \n","    print('Flight Dates:')\n","    for i in flight_dates:\n","        print('{}'.format(i))\n","\n","    while True:\n","        num = input (\"\\n*Enter date in [yyyy-mm-dd] or Q to quit:\") \n","        if num.isdigit():\n","            if num in flight_dates:\n","                fdate = num\n","                print('Flight date {} is selected.'.format(num))\n","                break\n","            else: \n","                print(num,\"is not a valid flight date. Select a flight from the list.\")\n","        elif num=='Q':\n","            return sys.exit(\"User selected 'quit'\")\n","        else:\n","            try:\n","                Fdate=datetime.strptime(num, '%Y-%m-%d').strftime(\"%Y-%m-%d\")\n","                if Fdate in flight_dates:\n","                    fdate=Fdate\n","                    print('Flight date {} is selected.'.format(fdate))\n","                    break\n","                else: \n","                    print(Fdate,\"is not a valid flight date. Select a flight from the list:\")\n","                    \n","            except ValueError:\n","                print('\\n%%Invalid flight date format%% \\nTry again.\\n')\n","                \n","    selected_files = [path for path in Path(dataDir).rglob('olympex_CRS_'+fdate.replace('-','')+'*.nc')]\n","    selected_files = [os.path.normpath(i) for i in selected_files]\n","    return selected_files,fdate.replace('-','')\n","\n","\n","def select_time_olympex(selected_files):\n","    \"\"\"\n","    User selects among the available flight time periods \n","    Return the selected file name\n","    \"\"\"\n","    start_times = []\n","    end_times = []\n","    \n","    time_periods = {}\n","    \n","    for i in selected_files:\n","        fname_split = re.split(r'_|-', os.path.basename(i))\n","        \n","        start_time = fname_split[3]\n","        start_time = start_time[:2] + ':' + start_time[2:4] + ':' + start_time[4:]\n","        start_times.append(start_time)\n","                \n","        end_time = fname_split[5]\n","        end_time = end_time[:2] + ':' + end_time[2:4] + ':' + end_time[4:]\n","        end_times.append(end_time)\n","        \n","        time_periods[selected_files.index(i)+1] = start_time + ' to ' + end_time\n","        \n","    for i in time_periods:\n","        print('The OLYMPEX flight periods available for this date are: \\n #{} - {}'.format(i, time_periods[i]))\n","        \n","    while True:\n","        num = input (\"\\n*Enter valid flight period no. or Q to quit: \") \n","        if num.isdigit():\n","            if int(num) in time_periods:\n","                print('Flight period #{}: {} is selected.'.format(int(num),time_periods[int(num)]))\n","                break\n","            else: \n","                print(int(num),\"is not a valid flight period. Select a period no. from the list:\")            \n","        elif num=='Q':\n","            return sys.exit(\"User selected 'quit'\")\n","        else:\n","            print('\\n%%Invalid flight No. format%% \\nTry again.\\n')\n","    return os.path.basename(selected_files[int(num)-1])\n","\n","def select_flight_iphex(dataDir):\n","    \"\"\"\n","    User select among the available flights by list number \n","    Return date in 'yyyymmdd' of the selected flight\n","    \"\"\"\n","        \n","    iphex_files = [path for path in Path(dataDir).rglob('IPHEX_CRS_L1B_*.nc')]\n","    iphex_files = [os.path.normpath(i) for i in iphex_files]\n","    \n","    #Check whether iphex files were found in the directory. Return 'None' if no\n","    #files were found. Continue through the code if files were found.\n","    if len(iphex_files)==0:\n","        print(\"%%There are no iphex data files in the currect directory. Try again%%\")\n","        return None, None\n","    else: \n","        pass\n","    \n","    flight_dates = [] \n","    \n","    for i in iphex_files:\n","        fname_split = re.split(r'_|-', os.path.basename(i))\n","        start_date = fname_split[3]\n","        \n","        flight_dates.append(datetime.strptime(start_date, '%Y%m%d').date().isoformat())\n","   \n","    print('Flight Dates:')\n","    for i in flight_dates:\n","        print('{}'.format(i))\n","\n","    while True:\n","        num = input (\"\\n*Enter date in [yyyy-mm-dd] or Q to quit:\") \n","        if num.isdigit():\n","            if num in flight_dates:\n","                fdate = num\n","                print('Flight date {} is selected.'.format(num))\n","                break\n","            else: \n","                print(num,\"is not a valid flight date. Select a flight from the list.\")\n","        elif num=='Q':\n","            return sys.exit(\"User selected 'quit'\")\n","        else:\n","            try:\n","                Fdate=datetime.strptime(num, '%Y-%m-%d').strftime(\"%Y-%m-%d\")\n","                if Fdate in flight_dates:\n","                    fdate=Fdate\n","                    print('Flight date {} is selected.'.format(fdate))\n","                    break\n","                else: \n","                    print(Fdate,\"is not a valid flight date. Select a flight from the list:\")\n","                    \n","            except ValueError:\n","                print('\\n%%Invalid flight date format%% \\nTry again.\\n')\n","                \n","    selected_files = [path for path in Path(dataDir).rglob('IPHEX_CRS_L1B_'+fdate.replace('-','')+'*.nc')]\n","    selected_files = [os.path.normpath(i) for i in selected_files]\n","    return selected_files,fdate.replace('-','')\n","\n","    \n","def select_time_iphex(selected_files):\n","    \"\"\"\n","    User selects among the available flight time periods \n","    Return the selected file name\n","    \"\"\"\n","    start_times = []\n","    end_times = []\n","    \n","    time_periods = {}\n","    \n","    for i in selected_files:\n","        fname_split = re.split(r'_|-', os.path.basename(i))\n","        \n","        start_time = fname_split[4]\n","        start_time = start_time[:2] + ':' + start_time[2:4] + ':' + start_time[4:]\n","        start_times.append(start_time)\n","                \n","        end_time = fname_split[6]\n","        end_time = end_time[:2] + ':' + end_time[2:4] + ':' + end_time[4:6]\n","        end_times.append(end_time)\n","        \n","        time_periods[selected_files.index(i)+1] = start_time + ' to ' + end_time\n","        \n","    for i in time_periods:\n","        print('The IPHEX flight periods available for this date are: \\n #{} - {}'.format(i, time_periods[i]))\n","        \n","    while True:\n","        num = input (\"\\n*Enter valid flight period no. or Q to quit: \") \n","        if num.isdigit():\n","            if int(num) in time_periods:\n","                print('Flight period #{}: {} is selected.'.format(int(num),time_periods[int(num)]))\n","                break\n","            else: \n","                print(int(num),\"is not a valid flight period. Select a period no. from the list:\")            \n","        elif num=='Q':\n","            return sys.exit(\"User selected 'quit'\")\n","        else:\n","            print('\\n%%Invalid flight No. format%% \\nTry again.\\n')\n","    return os.path.basename(selected_files[int(num)-1]) #Returns selected file name\n"," \n","def time2hrs(t):\n","    \"\"\" convert time to hours(float) \"\"\"\n","    return t.hour+t.minute/60+t.second/3600\n","\n","def totime(a):\n","    \"\"\"Obtain time (hours) from a time-string\"\"\"\n","    try:\n","        t=datetime.strptime(a,'%H:%M:%S').time()\n","        return time2hrs(t)\n","    except ValueError:\n","        print('%%Invalid time. Try again%%')\n","        return None\n","\n","def totime_impacts(t,d):\n","    \"\"\"Obtain datetime object from a time-string\"\"\"\n","    try:\n","        t=datetime.strptime(t,'%H:%M:%S').time()\n","        d=datetime.strptime(d,'%Y-%m-%d').date()\n","        return datetime.combine(d,t) #returns datatime object containing date/time\n","    except ValueError:\n","        print('%%Invalid time. Try again%%')\n","        return None\n","\n","def CRSsubset(ds,t1=None,t2=None):\n","    \"\"\"\n","    Subset CRS dataset with selected time interval [t1,t2]\n","    User inputs valid options:\n","      1. First enter for interval start t1\n","         Q: quit program\n","         ALL: select entire flight\n","         valid time string in hh:mm:ss \n","      2. Enter for interval end time t2\n","         valid time string in hh:mm:ss \n","    If user enters subset into function directly, t1 and t2 must be strings\n","    Input CRS fullset datase (ds)\n","    Return CRS subset (cs) of selected interval\n","    \"\"\"\n","    if(t1 and t2):\n","        t1=totime(t1)\n","        t2=totime(t2)\n","        return ds.where((ds.timed>=t1) & (ds.timed<=t2),drop=True)\n","\n","    while True:\n","        inp=input(\"\\n*Select subset starting in [hh:mm:ss] UTC\"+\n","                  \"\\n or 'ALL' for entire flight\"+\n","                  \"\\n or Q to quit: \\n\")\n","        if(inp=='Q'): return sys.exit(\"User selected 'quit'\")\n","        elif(inp=='ALL'):\n","            t1,t2=ds.timed[0],ds.timed[-1]\n","            break\n","        else:\n","            t1=totime(inp)\n","            if(not t1): continue\n","            t2=None\n","            while not t2:\n","                inp2=input(\"\\n*Select subset ending in [hh:mm:ss] UTC \"+\n","                           \"\\n or Q to quit: \\n\")\n","                if(inp=='Q'): return sys.exit(\"User selected 'quit'\")\n","                else:\n","                    t2=totime(inp2)\n","            break\n","\n","    #--Make sure selected period within flight timeframe\n","    cs=ds.where((ds.timed>=t1) & (ds.timed<=t2),drop=True)\n","    if(len(cs.timed)==0): print(\"%%No data found in selected period.\"+\n","                               \"\\n  Data missing or selection beyond data range.\")\n","    return cs\n","\n","def CRSsubset_goesrplt(ds,t1=None,t2=None):\n","    \"\"\"\n","    Subset CRS dataset with selected time interval [t1,t2]\n","    User inputs valid options:\n","      1. First enter for interval start t1\n","         Q: quit program\n","         ALL: select entire flight\n","         valid time string in hh:mm:ss \n","      2. Enter for interval end time t2\n","         valid time string in hh:mm:ss\n","    If user enters subset into function directly, t1 and t2 must be strings\n","    Input CRS fullset datase (ds)\n","    Return CRS subset (cs) of selected interval\n","    \"\"\"\n","    if(t1 and t2):\n","        t1=totime(t1)\n","        t2=totime(t2)\n","        return ds.where((ds.time>=t1) & (ds.time<=t2),drop=True)\n","\n","    while True:\n","        inp=input(\"\\n*Select subset starting in [hh:mm:ss] UTC\"+\n","                  \"\\n or 'ALL' for entire flight\"+\n","                  \"\\n or Q to quit: \\n\")\n","        if(inp=='Q'): return sys.exit(\"User selected 'quit'\")\n","        elif(inp=='ALL'):\n","            t1,t2=ds.time[0],ds.time[-1]\n","            break\n","        else:\n","            t1=totime(inp)\n","            if(not t1): continue\n","            t2=None\n","            while not t2:\n","                inp2=input(\"\\n*Select subset ending in [hh:mm:ss] UTC\"+\n","                           \"\\n or Q to quit: \\n\")\n","                if(inp=='Q'): return sys.exit(\"User selected 'quit'\")\n","                else:\n","                    t2=totime(inp2)\n","            break\n","\n","    #--Make sure selected period within flight timeframe\n","    cs=ds.where((ds.time>=t1) & (ds.time<=t2),drop=True)\n","    if(len(cs.time)==0): print(\"%%No data found in selected period.\"+\n","                               \"\\n  Data missing or selection beyond data range.\")\n","    return cs\n","\n","def CRSsubset_impacts(ds,t0,t1=None, d1=None, t2=None, d2=None):\n","    \"\"\"\n","    Subset CRS dataset with selected time and date interval [t1,d1,t2,d2] \n","    User inputs valid options:\n","      1. First enter for interval start t1\n","         Q: quit program\n","         ALL: select entire flight\n","         valid time string in hh:mm:ss \n","      2. Enter for interval start date d1\n","         in YYYY-MM-DD\n","      3. Enter for interval endtime t2\n","         valid time string in hh:mm:ss\n","      4. Enter for interval end date d2\n","         in YYYY-MM-DD\n","    If user enters subset into function directly, t1,d1,t2,and d2 must be strings\n","    Input CRS fullset datase (ds)\n","    Return CRS subset (cs) of selected interval\n","    \"\"\"\n","    time_data = ds['Time']['Data']['TimeUTC']\n","    time_utc = [(t0+timedelta(seconds=float(s))) for s in time_data] #datetime objects\n","    str_times = [x.strftime(\"%H:%M:%S\") for x in time_utc]\n","    time_dates = [x.strftime(\"%Y-%m-%d\") for x in time_utc]\n","    \n","    if(t1 and t2):\n","        start=totime_impacts(t1,d1)\n","        end=totime_impacts(t2,d2)\n","        subset = [num for num in time_utc if num >=start and num <=end]\n","        return subset\n","\n","    while True:\n","        inp=input(\"\\n*Select subset starting time in [hh:mm:ss] UTC\"+\n","                  \"\\n or 'ALL' for entire flight\"+\n","                  \"\\n or Q to quit: \\n\")\n","        if(inp=='Q'): return sys.exit(\"User selected 'quit'\")\n","        elif(inp=='ALL'):\n","            subset = time_utc\n","            return subset\n","        elif inp in str_times:\n","            t1=inp\n","            break\n","        else:\n","            print(inp,\"is not a valid flight time and/or format. Try again.\")\n","    while True:\n","        inp2=input(\"\\n*Select subset starting date in [YYYY-MM-DD] UTC or Q to quit:\")\n","        if(inp2=='Q'): return sys.exit(\"User selected 'quit'\")\n","        elif inp2 in time_dates:\n","            d1=inp2\n","            break\n","        else:\n","            print(inp2,\"is not a valid flight date and/or format. Try again.\")\n","    while True:\n","        inp3=input(\"\\n*Select subset ending time in [hh:mm:ss] UTC or Q to quit: \")\n","        if(inp3=='Q'): return sys.exit(\"User selected 'quit'\")\n","        elif inp3 in str_times:\n","            t2=inp3\n","            break\n","        else:\n","            print(inp3,\"is not a valid flight time and/or format. Try again.\") \n","    while True:\n","        inp4=input(\"\\n*Select subset ending date in [YYYY-MM-DD] UTC or Q to quit:\")\n","        if(inp4=='Q'): return sys.exit(\"User selected 'quit'\")\n","        elif inp4 in time_dates:\n","            d2=inp4\n","            break\n","        else:\n","            print(inp4,\"is not a valid flight date and/or format. Try again.\")\n","        \n","    start=totime_impacts(inp,inp2)\n","    end=totime_impacts(inp3,inp4)\n","    subset = [num for num in time_utc if num >=start and num <=end]\n","    \n","    #--Make sure selected period within flight timeframe\n","    if(len(subset)==0): print(\"%%No data found in selected period.\"+\n","                               \"\\n  Data missing or selection beyond data range.\")\n","    return subset \n","\n","def radarCmaps():\n","    \"\"\"\n","    Make Color maps for radar Ref and DopV\n","    User can use predefined color maps in cmaps\n","    Return color maps\n","    \"\"\"\n","    basecmp = cm.get_cmap('gist_ncar', 256)\n","    newcols = basecmp(np.linspace(0, 1, 200))\n","    topoff  = cm.get_cmap('gray', 128)\n","    combo   = np.vstack((newcols[:180,:],topoff(np.linspace(0.7, 1, 20))))\n","    aerocmp = ListedColormap(combo, name='aerocmp')\n","    cmaps  ={'Ref':aerocmp,'DopV':cm.gist_ncar}\n","    return cmaps\n","\n","def plot_CRS2D(datap,xvar,ZB,plot_start,plot_end,reverseZ=True):\n","    \"\"\"\n","    datap: Variables to be plotted, reflectivity and Doppler velocity\n","    xvar: Horizontal coord., we use [time]\n","    ZB: Vertical coord., we use radar [range]\n","        ZB=0 means at radar/airccraft location/altitude;\n","        ZB largest means near the ground\n","    plot_start: Plot start date/time object for plot title\n","    plot_end: Plot end date/time object for plot title\n","    Note that reverseZ=True would have data away from radar (large ZB) plotted\n","         at bottom, and near radar range plotted at top.\n","    Return image object \"fig\" than can be used to save the plot\n","    \"\"\"\n","    vnames ={'Ref':\"Reflectivity\",'DopV':'Doppler Vel.'}\n","    units  ={'Ref':'[dBZ]',       'DopV':'[m/s]'}\n","    levs   ={'Ref':np.arange(-20,40,2), 'DopV':np.arange(-20,20,2)} #User can adjust colorscale range and intervals\n","    \n","    fig, axs = plt.subplots(nrows=2, ncols=1, figsize=(12, 6))\n","    fig.tight_layout()\n","    fig.subplots_adjust(top=0.9,bottom=0.1,hspace=0.2)\n","\n","    for iv,vnm in enumerate(vnames):\n","        ax,lev,unit,cmp = axs[iv],levs[vnm],units[vnm],radarCmaps()[vnm]\n","        var=np.array([*zip(*datap[vnm])]) #<--move time to col dim (x), and altitude/range to row(y)\n","        xlab='Time (UTC)' if iv==1 else ''\n","        \n","        #Divide the flight period into multiple segments and plot separately \n","        #for more efficient memory usage\n","        div =int(len(xvar)/9)\n","        \n","        cp = ax.contourf(xvar[:div], ZB, var[:,:div],lev,cmap=cmp)\n","        cp = ax.contourf(xvar[:div*2], ZB, var[:,:div*2],lev,cmap=cmp)\n","        cp = ax.contourf(xvar[div*2:div*3], ZB, var[:,div*2:div*3],lev,cmap=cmp)\n","        cp = ax.contourf(xvar[div*3:div*4], ZB, var[:,div*3:div*4],lev,cmap=cmp)\n","        cp = ax.contourf(xvar[div*4:div*5], ZB, var[:,div*4:div*5],lev,cmap=cmp)\n","        cp = ax.contourf(xvar[div*5:div*6], ZB, var[:,div*5:div*6],lev,cmap=cmp)\n","        cp = ax.contourf(xvar[div*6:div*7], ZB, var[:,div*6:div*7],lev,cmap=cmp)\n","        cp = ax.contourf(xvar[div*7:div*8], ZB, var[:,div*7:div*8],lev,cmap=cmp)\n","        cp = ax.contourf(xvar[div*8:], ZB, var[:,div*8:],lev,cmap=cmp)\n","        \n","        ax.set_ylabel('Range from Radar [km]')\n","        ax.set_xlabel(xlab)\n","        if(reverseZ):\n","            ax.set_ylim(ymin=20,ymax=5)\n","            ytpos=4.5\n","        else:\n","            ax.set_ylim(ymin=5,ymax=20)\n","            ytpos=20.6\n","\n","        ax.text(xvar[int(len(xvar)*.5)],ytpos,vnames[vnm],\n","               {'fontsize':13,'ha':'center'})\n","        \n","        #Extract the number of seconds over the entire flight period\n","        period_sec = (xvar[-1] - xvar[0]).seconds \n","        \n","        #Place time ticks on the x-axis of plot based on flight period length\n","        # User can manually change this value to preferred number of ticks by replacing the \"6\" value\n","        ax.xaxis.set_major_locator(mdates.SecondLocator(interval=int(period_sec/6))) \n","        ax.xaxis.set_major_formatter(mdates.DateFormatter('%H:%M:%S')) #<--Format times on x-axis to hh:mm:ss\n","        \n","        #Create plot title based on file dates; a different title is given based on whether the data\n","        #covers a single date or multiple dates\n","        if plot_start.date() != plot_end.date():\n","            plot_title1 = 'CRS Reflectivity and Doppler Velocity ' + plot_start.strftime(\"%B %d, %Y\") + ' - ' + plot_end.strftime(\"%B %d, %Y\")\n","            fig.suptitle(plot_title1,fontsize=14,x=0.415)\n","        else:\n","            plot_title2 = 'CRS Reflectivity and Doppler Velocity ' + plot_start.strftime(\"%B %d, %Y\")\n","            fig.suptitle(plot_title2,fontsize=14,x=0.415)\n","        \n","        #Create and label colorbar\n","        clb=fig.colorbar(cp,ax=ax) \n","        clb.set_label(unit)\n","        print(\"Fig.{} is done for {}\".format(iv, vnm))\n","\n","    plt.show()\n","    return fig\n","\n","def SAVEsubset(cs,fig,fname,dirpath,start,end):\n","    \"\"\"\n","    User selects whether to save the plot image\n","    cs:  selected subset dataset\n","    fig: radar image of cs \n","    fname: original CRS file name\n","    dirpath: where saved image will be located\n","    start: start date/time of plot\n","    end: end date/time of plot \n","    \"\"\"\n","    Save=input(\"\\n*Save plot(y/n)?\")\n","    if(Save.lower()=='y'):\n","        campaign=fname.split('_')[0]\n","        instr=fname.split('_')[1]\n","        fig.savefig(os.path.join(dirpath+campaign+ '_'+instr+ '_'+start+ '_'+end+'.png'),dpi=100,bbox_inches='tight')\n","        plt.close(fig)\n","        print(\"Image saved to \", dirpath+campaign+ '_'+instr+ '_'+start+ '_'+end+'.png\\n') \n","    else: \n","        print(\"No image was saved.\\n\")"]}],"metadata":{"kernelspec":{"display_name":"Python 3.8.10 64-bit","language":"python","name":"python3"},"language_info":{"codemirror_mode":{"name":"ipython","version":3},"file_extension":".py","mimetype":"text/x-python","name":"python","nbconvert_exporter":"python","pygments_lexer":"ipython3","version":"3.8.10"},"vscode":{"interpreter":{"hash":"916dbcbb3f70747c44a77c7bcd40155683ae19c65e1c03b4aa3499c5328201f1"}}},"nbformat":4,"nbformat_minor":2}