Atmospheric Lidar Package: atmospheric_lidar/scripts/licel2scc.py@a1788b77f33b (annotated)

atmospheric_lidar/scripts/licel2scc.py@a1788b77f33b (annotated)

atmospheric_lidar/scripts/licel2scc.py

Fri, 16 Feb 2018 14:39:54 +0200

author: Iannis <i.binietoglou@impworks.gr>
date: Fri, 16 Feb 2018 14:39:54 +0200
changeset 124: a1788b77f33b
parent 121: 4522191fe936
child 126: 4984793c500a
permissions: -rw-r--r--

Attempt to add cloudmasking options.

 """ Command line tool to convert Licel binary files to SCC NetCDF format.
 """
 import argparse
 import glob
 import importlib
 import logging
 import os
 import sys
 from ..licel import LicelLidarMeasurement
 from ..__init__ import __version__
 logger = logging.getLogger(__name__)
 def create_custom_class(custom_netcdf_parameter_path, use_id_as_name=False, temperature=25., pressure=1020.,
                         licel_timezone='UTC'):
     """ This funtion creates a custom LicelLidarMeasurement subclass,
     based on the input provided by the users.
     Parameters
     ----------
     custom_netcdf_parameter_path : str
        The path to the custom channels parameters.
     use_id_as_name : bool
        Defines if channels names are descriptive or transient digitizer IDs.
     temperature : float
        The ground temperature in degrees C (default 25.0).
     pressure : float
        The ground pressure in hPa (default: 1020.0).
     licel_timezone : str
        String describing the timezone according to the tz database.
     Returns
     -------
     CustomLidarMeasurement:
        A custom sub-class of LicelLidarMeasurement
     """
     logger.debug('Reading parameter files: %s' % custom_netcdf_parameter_path)
     custom_netcdf_parameters = read_settings_file(custom_netcdf_parameter_path)
     class CustomLidarMeasurement(LicelLidarMeasurement):
         extra_netcdf_parameters = custom_netcdf_parameters
         def __init__(self, file_list=None):
             super(CustomLidarMeasurement, self).__init__(file_list, use_id_as_name, licel_timezone=licel_timezone)
         def set_PT(self):
             ''' Sets the pressure and temperature at station level. This is used if molecular_calc parameter is
             set to 0 (i.e. use US Standard atmosphere).
             The results are stored in the info dictionary.
             '''
             self.info['Temperature'] = temperature
             self.info['Pressure'] = pressure
     return CustomLidarMeasurement
 def read_settings_file(settings_path):
     """ Read the settings file.
     The file should contain python code."""
     if not os.path.isfile(settings_path):
         logging.error("The provided settings path does not correspond to a file.")
         sys.exit(1)
     dirname, basename = os.path.split(settings_path)
     sys.path.append(dirname)
     module_name, _ = os.path.splitext(basename)
     settings = importlib.import_module(module_name)
     return settings
 def get_cloud_free_files(CustomLidarMeasurement, files, args):
     logger.warning("Starting cloud mask procedure. This is an experimental feature.")
     try:
         from cloudmask import cloudmask  # Import here until we setup a proper installation procedure
     except ImportError:
         logger.error("Cloud mask module could not be loaded. Please install manually.")
         sys.exit(1)
     measurement = CustomLidarMeasurement(files)
     channel = measurement.channels[args.cloudmask_channel]
     cloud_mask = cloudmask.CloudMaskRaw(channel)
     idxs = cloud_mask.cloud_free_periods(args.cloudfree_period, args.cloud_search_height)
     if len(idxs) == 0:  # If no cloud-free period found
         logger.info('No cloud free period found. Nothing converted.')
         sys.exit(1)
     logger.info("{0} cloud free period(s) found.".format(len(idxs)))
     file_list = []
     for idx_min, idx_max in idxs:
         current_files = measurement.files[idx_min:idx_max]
         file_list.append(current_files)
     return file_list
 def get_corrected_measurement_id(args, n):
     """ Correct the provided measurement id, in case of multiple cloud-free periods. """
     if args.measurement_id is not None:
         order = float(args.measurement_id[-2:])
         new_no = order + n
         measurement_id = args.measurement_id[:-2] + str(new_no)
         measurement_no = args.measurement_number  # The same
     else:
         measurement_no =  str(float(args.measurement_number) + n).zfill(2)
         measurement_id = None
     return measurement_id, measurement_no
 def convert_to_scc(CustomLidarMeasurement, files, dark_pattern, measurement_id, measurement_number):
     """ Convert files to SCC. """
     measurement = CustomLidarMeasurement(files)
     # Get a list of files containing dark measurements
     if dark_pattern != "":
         dark_files = glob.glob(dark_pattern)
         if dark_files:
             logger.debug("Using %s as dark measurements files!" % ', '.join(dark_files))
             measurement.dark_measurement = CustomLidarMeasurement(dark_files)
         else:
             logger.warning(
                 'No dark measurement files found when searching for %s. Will not use any dark measurements.' % dark_pattern)
     try:
         measurement = measurement.subset_by_scc_channels()
     except ValueError as err:
         logger.error(err)
         sys.exit(1)
     # Save the netcdf
     logger.info("Saving netcdf")
     measurement.set_measurement_id(measurement_id, measurement_number)
     measurement.save_as_SCC_netcdf()
     logger.info("Created file %s" % measurement.scc_filename)
 def main():
     # Define the command line argument
     parser = argparse.ArgumentParser(description="A program to convert Licel binary files to the SCC NetCDF format.")
     parser.add_argument("parameter_file", help="The path to a parameter file linking licel and SCC channels.")
     parser.add_argument("files",
                         help="Location of licel files. Use relative path and filename wildcards. (default './*.*')",
                         default="./*.*")
     parser.add_argument("-i", '--id_as_name',
                         help="Use transient digitizer ids as channel names, instead of descriptive names",
                         action="store_true")
     parser.add_argument("-m", "--measurement_id", help="The new measurement id", default=None)
     parser.add_argument("-n", "--measurement_number",
                         help="The measurement number for the date from 00 to 99. Used if no id is provided",
                         default="00")
     parser.add_argument("-t", "--temperature", type=float,
                         help="The temperature (in C) at lidar level, required if using US Standard atmosphere",
                         default="25")
     parser.add_argument("-p", "--pressure", type=float,
                         help="The pressure (in hPa) at lidar level, required if using US Standard atmosphere",
                         default="1020")
     parser.add_argument('-D', '--dark_measurements',
                         help="Location of files containing dark measurements. Use relative path and filename wildcars, see 'files' parameter for example.",
                         default="", dest="dark_files"
                         )
     parser.add_argument('--licel_timezone', help="String describing the timezone according to the tz database.",
                         default="UTC", dest="licel_timezone",
                         )
     parser.add_argument('--cloudmask', help="Experimental feature to automatically cloud mask measurements",
                         default=False, action='store_true',
                         )
     parser.add_argument('--cloudmask_channel', help="Name of channel to apply the cloud mask.")
     parser.add_argument('--cloudfree_period', type=float, help="Duration (in min) of cloud-free periods",
                         default="30",
                         )
     parser.add_argument('--cloud_search_height', type=float, help="Maximum altitude (in m) to check for clouds.",
                         default="12000",
                         )
     # Verbosity settings from http://stackoverflow.com/a/20663028
     parser.add_argument('-d', '--debug', help="Print dubuging information.", action="store_const",
                         dest="loglevel", const=logging.DEBUG, default=logging.INFO,
                         )
     parser.add_argument('-s', '--silent', help="Show only warning and error messages.", action="store_const",
                         dest="loglevel", const=logging.WARNING
                         )
     parser.add_argument('--version', help="Show current version.", action='store_true')
     args = parser.parse_args()
     # Get the logger with the appropriate level
     logging.basicConfig(format='%(levelname)s: %(message)s', level=args.loglevel)
     logger = logging.getLogger(__name__)
     # coloredlogs.install(fmt='%(levelname)s: %(message)s', level=args.loglevel)
     # Check for version
     if args.version:
         print("Version: %s" % __version__)
         sys.exit(0)
     # Get a list of files to convert
     files = glob.glob(args.files)
     # If not files found, exit
     if len(files) == 0:
         logger.error("No files found when searching for %s." % args.files)
         sys.exit(1)
     # If everything OK, proceed
     logger.info("Found {0} files matching {1}".format(len(files), os.path.abspath(args.files)))
     CustomLidarMeasurement = create_custom_class(args.parameter_file, args.id_as_name, args.temperature,
                                                  args.pressure, args.licel_timezone)
     if args.cloudmask:
         file_lists = get_cloud_free_files(CustomLidarMeasurement, files, args)
         for n, files in enumerate(file_lists):
             measurement_id, measurement_no = get_corrected_measurement_id(args, n)
             convert_to_scc(CustomLidarMeasurement, files, args.dark_files, measurement_id, measurement_no)
     else:
         convert_to_scc(CustomLidarMeasurement, files, args.dark_files, args.measurement_id, args.measurement_number)

Mercurial > public > atmospheric_lidar / annotate

atmospheric_lidar/scripts/licel2scc.py@a1788b77f33b (annotated)

atmospheric_lidar/scripts/licel2scc.py