disdrodb 0.0.21__py3-none-any.whl → 0.1.1__py3-none-any.whl

disdrodb/l0/readers/{NCAR/PLOWS_MIPS.py → PARSIVEL2/NCAR/VORTEX_SE_2016_P1.py}

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+
 # -----------------------------------------------------------------------------.
 # Copyright (c) 2021-2023 DISDRODB developers
 #
@@ -15,23 +16,21 @@
 # You should have received a copy of the GNU General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 # -----------------------------------------------------------------------------.
-from disdrodb.l0 import run_l0a
+"""Reader for the VORTEX SE 2016 PARSIVEL2 NCAR data."""
+import pandas as pd
+
 from disdrodb.l0.l0_reader import is_documented_by, reader_generic_docstring
+from disdrodb.l0.l0a_processing import read_raw_text_file
 
 
 @is_documented_by(reader_generic_docstring)
 def reader(
-    raw_dir,
-    processed_dir,
-    station_name,
-    # Processing options
-    force=False,
-    verbose=False,
-    parallel=False,
-    debugging_mode=False,
+    filepath,
+    logger=None,
 ):
+    """Reader."""
     ##------------------------------------------------------------------------.
-    #### - Define column names
+    #### Define column names
     column_names = [
         "day",
         "month",
@@ -50,7 +49,7 @@ def reader(
     ]
 
     ##------------------------------------------------------------------------.
-    #### - Define reader options
+    #### Define reader options
     reader_kwargs = {}
     # - Define delimiter
     reader_kwargs["delimiter"] = ","
@@ -61,6 +60,9 @@ def reader(
     # - Define behaviour when encountering bad lines
     reader_kwargs["on_bad_lines"] = "skip"
 
+    # - Define encoding
+    reader_kwargs["encoding"] = "ISO-8859-1"
+
     # - Define reader engine
     #   - C engine is faster
     #   - Python engine is more feature-complete
@@ -71,67 +73,48 @@ def reader(
     reader_kwargs["compression"] = "infer"
 
     # - Strings to recognize as NA/NaN and replace with standard NA flags
-    #   - Already included: ‘#N/A’, ‘#N/A N/A’, ‘#NA’, ‘-1.#IND’, ‘-1.#QNAN’,
-    #                       ‘-NaN’, ‘-nan’, ‘1.#IND’, ‘1.#QNAN’, ‘<NA>’, ‘N/A’,
-    #                       ‘NA’, ‘NULL’, ‘NaN’, ‘n/a’, ‘nan’, ‘null’
+    #   - Already included: '#N/A', '#N/A N/A', '#NA', '-1.#IND', '-1.#QNAN',
+    #                       '-NaN', '-nan', '1.#IND', '1.#QNAN', '<NA>', 'N/A',
+    #                       'NA', 'NULL', 'NaN', 'n/a', 'nan', 'null'
     reader_kwargs["na_values"] = ["na", "", "error"]
 
     # Skip first row as columns names
     reader_kwargs["header"] = None
 
-    # - Define encoding
-    reader_kwargs["encoding"] = "ISO-8859-1"
-
     ##------------------------------------------------------------------------.
-    #### - Define dataframe sanitizer function for L0 processing
-    def df_sanitizer_fun(df):
-        # - Import pandas
-        import pandas as pd
-
-        # - Define datetime 'time' column
-        df["time"] = (
-            df["day"].astype(str)
-            + "-"
-            + df["month"].astype(str)
-            + "-"
-            + df["year"].astype(str)
-            + " "
-            + df["hour_minute"].astype(str)
-            + ":"
-            + df["second"].astype(str)
-        )
-        df["time"] = pd.to_datetime(df["time"], format="%d-%m-%Y %H%M:%S", errors="coerce")
-
-        # - Drop columns not agreeing with DISDRODB L0 standards
-        columns_to_drop = [
-            "day",
-            "month",
-            "year",
-            "hour_minute",
-            "second",
-        ]
-        df = df.drop(columns=columns_to_drop)
-
-        return df
-
-    ##------------------------------------------------------------------------.
-    #### - Define glob pattern to search data files in <raw_dir>/data/<station_name>
-    glob_patterns = "*.dat"
-
-    ####----------------------------------------------------------------------.
-    #### - Create L0A products
-    run_l0a(
-        raw_dir=raw_dir,
-        processed_dir=processed_dir,
-        station_name=station_name,
-        # Custom arguments of the reader for L0A processing
-        glob_patterns=glob_patterns,
+    #### Read the data
+    df = read_raw_text_file(
+        filepath=filepath,
         column_names=column_names,
         reader_kwargs=reader_kwargs,
-        df_sanitizer_fun=df_sanitizer_fun,
-        # Processing options
-        force=force,
-        verbose=verbose,
-        parallel=parallel,
-        debugging_mode=debugging_mode,
+        logger=logger,
     )
+
+    ##------------------------------------------------------------------------.
+    #### Adapt the dataframe to adhere to DISDRODB L0 standards
+    # Define datetime 'time' column
+    df["time"] = (
+        df["day"].astype(str)
+        + "-"
+        + df["month"].astype(str)
+        + "-"
+        + df["year"].astype(str)
+        + " "
+        + df["hour_minute"].astype(str)
+        + ":"
+        + df["second"].astype(str)
+    )
+    df["time"] = pd.to_datetime(df["time"], format="%d-%m-%Y %H%M:%S")
+
+    # Drop columns not agreeing with DISDRODB L0 standards
+    columns_to_drop = [
+        "day",
+        "month",
+        "year",
+        "hour_minute",
+        "second",
+    ]
+    df = df.drop(columns=columns_to_drop)
+
+    # Return the dataframe adhering to DISDRODB L0 standards
+    return df

disdrodb/l0/readers/PARSIVEL2/NCAR/VORTEX_SE_2016_P2.py

@@ -0,0 +1,118 @@
+#!/usr/bin/env python3
+# -----------------------------------------------------------------------------.
+# Copyright (c) 2021-2023 DISDRODB developers
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+# -----------------------------------------------------------------------------.
+import pandas as pd
+
+from disdrodb.l0.l0_reader import is_documented_by, reader_generic_docstring
+from disdrodb.l0.l0a_processing import read_raw_text_file
+
+
+@is_documented_by(reader_generic_docstring)
+def reader(
+    filepath,
+    logger=None,
+):
+    """Reader."""
+    ##------------------------------------------------------------------------.
+    #### Define column names
+    column_names = [
+        "date",
+        "time",
+        "sensor_battery_voltage",
+        "laser_amplitude",
+        "sensor_heating_current",
+        "sensor_temperature",
+        "number_particles",
+        "mor_visibility",
+        "rainfall_rate_32bit",
+        "rainfall_accumulated_32bit",
+        "weather_code_synop_4680",
+        "reflectivity_32bit",
+        "weather_code_nws",
+        "weather_code_metar_4678",
+        "raw_drop_number",
+    ]
+
+    ##------------------------------------------------------------------------.
+    #### Define reader options
+    reader_kwargs = {}
+    # - Define delimiter
+    reader_kwargs["delimiter"] = ","
+
+    # - Avoid first column to become df index !!!
+    reader_kwargs["index_col"] = False
+
+    # - Define behaviour when encountering bad lines
+    reader_kwargs["on_bad_lines"] = "skip"
+
+    # - Define encoding
+    reader_kwargs["encoding"] = "ISO-8859-1"
+
+    # - Define reader engine
+    #   - C engine is faster
+    #   - Python engine is more feature-complete
+    reader_kwargs["engine"] = "python"
+
+    # - Define on-the-fly decompression of on-disk data
+    #   - Available: gzip, bz2, zip
+    reader_kwargs["compression"] = "infer"
+
+    # - Strings to recognize as NA/NaN and replace with standard NA flags
+    #   - Already included: '#N/A', '#N/A N/A', '#NA', '-1.#IND', '-1.#QNAN',
+    #                       '-NaN', '-nan', '1.#IND', '1.#QNAN', '<NA>', 'N/A',
+    #                       'NA', 'NULL', 'NaN', 'n/a', 'nan', 'null'
+    reader_kwargs["na_values"] = ["na", "", "error"]
+
+    # Skip first row as columns names
+    reader_kwargs["header"] = None
+
+    ##------------------------------------------------------------------------.
+    #### Read the data
+    df = read_raw_text_file(
+        filepath=filepath,
+        column_names=column_names,
+        reader_kwargs=reader_kwargs,
+        logger=logger,
+    )
+
+    ##------------------------------------------------------------------------.
+    #### Adapt the dataframe to adhere to DISDRODB L0 standards
+    # Define datetime 'time' column
+    df["time"] = df["date"] + "-" + df["time"]
+    df["time"] = pd.to_datetime(df["time"], format="%Y%m%d-%H:%M:%S", errors="coerce")
+
+    # Preprocess the raw spectrum
+    # - The '<SPECTRUM>ZERO</SPECTRUM>' indicates no drops detected
+    # --> "" generates an array of zeros in L0B processing
+    df["raw_drop_number"] = df["raw_drop_number"].astype("string")
+    df["raw_drop_number"] = df["raw_drop_number"].str.strip()
+    df["raw_drop_number"] = df["raw_drop_number"].str.replace("<SPECTRUM>ZERO</SPECTRUM>", "")
+
+    # Remove <SPECTRUM> and </SPECTRUM>" acronyms from the raw_drop_number field
+    df["raw_drop_number"] = df["raw_drop_number"].str.replace("<SPECTRUM>", "")
+    df["raw_drop_number"] = df["raw_drop_number"].str.replace("</SPECTRUM>", "")
+
+    # Preprocess the raw spectrum and raw_drop_average_velocity
+    # - Add 0 before every ; if ; not preceded by a digit
+    # - Example: ';;1;;' --> '0;0;1;0;'
+    df["raw_drop_number"] = df["raw_drop_number"].str.replace(r"(?<!\d);", "0;", regex=True)
+
+    # Drop columns not agreeing with DISDRODB L0 standards
+    df = df.drop(columns=["date"])
+
+    # Return the dataframe adhering to DISDRODB L0 standards
+    return df

disdrodb/l0/readers/PARSIVEL2/NCAR/VORTEX_SE_2016_PIPS.py

@@ -0,0 +1,152 @@
+#!/usr/bin/env python3
+# -----------------------------------------------------------------------------.
+# Copyright (c) 2021-2023 DISDRODB developers
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+# -----------------------------------------------------------------------------.
+import pandas as pd
+
+from disdrodb.l0.l0_reader import is_documented_by, reader_generic_docstring
+from disdrodb.l0.l0a_processing import read_raw_text_file
+
+
+@is_documented_by(reader_generic_docstring)
+def reader(
+    filepath,
+    logger=None,
+):
+    """Reader."""
+    ##------------------------------------------------------------------------.
+    #### Define column names
+    column_names = [
+        "logger_time",
+        "record_number",
+        "battery_voltage",
+        "logger_temperature",
+        "wind_direction",
+        "wind_speed",
+        "wind_flag",
+        "fast_temperature",
+        "slow_temperature",
+        "relative_humidity",
+        "pressure",
+        "compass_direction",
+        "gps_time",
+        "gps_status",
+        "gps_latitude",  # DD.DM, DM = decimal minutes/100, DD = degrees
+        "gps_latitude_hemisphere",  # N/S
+        "gps_longitude",
+        "gps_longitude_hemisphere",  # W/E
+        "gps_speed",
+        "gps_direction",
+        "gps_date",
+        "gps_magnetic_version",
+        "gps_altitude",
+        "relative_wind_direction",
+        "dew_point_temperature",
+        "rh",
+        "disdrometer_data",
+    ]
+
+    ##------------------------------------------------------------------------.
+    #### Define reader options
+    reader_kwargs = {}
+    # - Define delimiter
+    reader_kwargs["delimiter"] = ","
+
+    # - Avoid first column to become df index !!!
+    reader_kwargs["index_col"] = False
+
+    # - Define behaviour when encountering bad lines
+    reader_kwargs["on_bad_lines"] = "skip"
+
+    # - Define encoding
+    reader_kwargs["encoding"] = "ISO-8859-1"
+
+    # - Define reader engine
+    #   - C engine is faster
+    #   - Python engine is more feature-complete
+    reader_kwargs["engine"] = "python"
+
+    # - Define on-the-fly decompression of on-disk data
+    #   - Available: gzip, bz2, zip
+    reader_kwargs["compression"] = "infer"
+
+    # - Strings to recognize as NA/NaN and replace with standard NA flags
+    #   - Already included: '#N/A', '#N/A N/A', '#NA', '-1.#IND', '-1.#QNAN',
+    #                       '-NaN', '-nan', '1.#IND', '1.#QNAN', '<NA>', 'N/A',
+    #                       'NA', 'NULL', 'NaN', 'n/a', 'nan', 'null'
+    reader_kwargs["na_values"] = ["na", "", "error"]
+
+    # Skip first row as columns names
+    reader_kwargs["header"] = None
+
+    ##------------------------------------------------------------------------.
+    #### Read the data
+    df = read_raw_text_file(
+        filepath=filepath,
+        column_names=column_names,
+        reader_kwargs=reader_kwargs,
+        logger=logger,
+    )
+
+    ##------------------------------------------------------------------------.
+    #### Adapt the dataframe to adhere to DISDRODB L0 standards
+    # Drop timesteps without disdrometer data
+    df = df[~df["disdrometer_data"].isna()]
+
+    # Retrieve disdrometer data
+    df_data = df["disdrometer_data"].str.split(";", expand=True, n=11)
+
+    # Assign column names
+    column_names = [
+        "serial_number",
+        "rainfall_rate_32bit",
+        "rainfall_accumulated_32bit",
+        "reflectivity_32bit",
+        "sample_interval",
+        "laser_amplitude",
+        "number_particles",
+        "sensor_temperature",
+        "sensor_battery_voltage",
+        "sensor_time",  # Note: logger_time is currently used !
+        "sensor_date",
+        "raw_drop_number",
+    ]
+    df_data.columns = column_names
+
+    # Retrieve time and coordinates information
+    # --> Latitude in degrees_north
+    # --> Longitude in degrees_east
+    df_time = pd.to_datetime(df["logger_time"], errors="coerce")
+    df_lat_sign = df["gps_latitude_hemisphere"].str.replace("N", "1").str.replace("S", "-1")
+    df_lon_sign = df["gps_longitude_hemisphere"].str.replace("E", "1").str.replace("W", "-1")
+    df_lat_sign = df_lat_sign.astype(float)
+    df_lon_sign = df_lon_sign.astype(float)
+    df_lon = df["gps_longitude"].astype(float)
+    df_lat = df["gps_latitude"].astype(float)
+    df_lon = df_lon * df_lon_sign
+    df_lat = df_lat * df_lat_sign
+
+    # Create dataframe
+    df = df_data
+    df["time"] = df_time
+    df["latitude"] = df_lat
+    df["longitude"] = df_lon
+
+    # Drop columns not agreeing with DISDRODB L0 standards
+    df = df.drop(columns=["serial_number", "sensor_time", "sensor_date", "serial_number"])
+
+    # Return the dataframe adhering to DISDRODB L0 standards
+    return df

disdrodb/l0/readers/PARSIVEL2/NETHERLANDS/DELFT.py

@@ -0,0 +1,166 @@
+#!/usr/bin/env python3
+# -----------------------------------------------------------------------------.
+# Copyright (c) 2021-2023 DISDRODB developers
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+# -----------------------------------------------------------------------------.
+import pandas as pd
+
+from disdrodb.l0.l0_reader import is_documented_by, reader_generic_docstring
+from disdrodb.l0.l0a_processing import read_raw_text_file
+
+
+@is_documented_by(reader_generic_docstring)
+def reader(
+    filepath,
+    logger=None,
+):
+    """Reader."""
+    ##------------------------------------------------------------------------.
+    #### Define column names
+    column_names = ["time", "epoch_time", "TO_BE_PARSED"]
+
+    ##------------------------------------------------------------------------.
+    #### Define reader options
+    reader_kwargs = {}
+    # - Define delimiter
+    reader_kwargs["delimiter"] = ";"
+    # - Skip first row as columns names
+    reader_kwargs["header"] = None
+    # - Avoid first column to become df index
+    reader_kwargs["index_col"] = False
+    # - Define behaviour when encountering bad lines
+    reader_kwargs["on_bad_lines"] = "skip"
+    # - Define parser engine
+    #   - C engine is faster
+    #   - Python engine is more feature-complete
+    reader_kwargs["engine"] = "python"
+    # - Define on-the-fly decompression of on-disk data
+    #   - Available: gzip, bz2, zip
+    reader_kwargs["compression"] = "infer"
+    # reader_kwargs['zipped'] = False
+    # reader_kwargs['zipped'] = True
+    # - Strings to recognize as NA/NaN and replace with standard NA flags
+    #   - Already included: '#N/A', '#N/A N/A', '#NA', '-1.#IND', '-1.#QNAN',
+    #                       '-NaN', '-nan', '1.#IND', '1.#QNAN', '<NA>', 'N/A',
+    #                       'NA', 'NULL', 'NaN', 'n/a', 'nan', 'null'
+    reader_kwargs["na_values"] = ["na", "", "error", "-.-", " NA"]
+    ##------------------------------------------------------------------------.
+    #### Read the data
+    df = read_raw_text_file(
+        filepath=filepath,
+        column_names=column_names,
+        reader_kwargs=reader_kwargs,
+        logger=logger,
+    )
+
+    ##------------------------------------------------------------------------.
+    #### Adapt the dataframe to adhere to DISDRODB L0 standards
+    # NOTE: Station 8 has all raw_drop_number corrupted, so it can't be used
+
+    # Remove rows that have a corrupted "TO_BE_PARSED" column
+    # - PAR001, PAR002 have length 3726 (no station_name)
+    # - PAR007 have length 3736 ()
+    df = df.loc[df["TO_BE_PARSED"].astype(str).str.len() >= 3726]
+    df = df.loc[df["TO_BE_PARSED"].astype(str).str.len() <= 3736]
+
+    # Convert 'time' column to datetime
+    df_time = pd.to_datetime(df["time"], format="%Y%m%d-%H%M%S", errors="coerce")
+
+    # Strip values from start and end of the string
+    df["TO_BE_PARSED"] = df["TO_BE_PARSED"].str.lstrip("b'").str.rstrip("'").str.rstrip("\\r\\n'")  # noqa: B005
+
+    # Split the column 'TO_BE_PARSED'
+    df_to_parse = df["TO_BE_PARSED"].str.split(";", expand=True, n=99)
+
+    # Retrieve DISDRODB compliant columns
+    df = df_to_parse.iloc[:, 0:35]
+
+    # Assign column names
+    column_names = [
+        "rainfall_rate_32bit",
+        "rainfall_accumulated_32bit",
+        "weather_code_synop_4680",
+        "weather_code_synop_4677",
+        "weather_code_metar_4678",
+        "weather_code_nws",
+        "reflectivity_32bit",
+        "mor_visibility",
+        "sample_interval",
+        "laser_amplitude",
+        "number_particles",
+        "sensor_temperature",
+        "sensor_serial_number",
+        "firmware_iop",
+        "firmware_dsp",
+        "sensor_heating_current",
+        "sensor_battery_voltage",
+        "sensor_status",
+        "sensor_time_measurement_start",
+        "sensor_time",
+        "sensor_date",
+        "station_name",
+        "station_number",
+        "rainfall_amount_absolute_32bit",
+        "error_code",
+        "sensor_temperature_pcb",
+        "sensor_temperature_receiver",
+        "sensor_temperature_trasmitter",
+        "rainfall_rate_16_bit_30",
+        "rainfall_rate_16_bit_1200",
+        "rainfall_accumulated_16bit",
+        "reflectivity_16bit",
+        "rain_kinetic_energy",
+        "snowfall_rate",
+        "number_particles_all",
+        # "number_particles_all_detected",
+    ]
+    df.columns = column_names
+
+    # Add time column
+    df["time"] = df_time
+
+    # Retrieve raw_drop_concentration
+    df["raw_drop_concentration"] = df_to_parse.iloc[:, 35:67].apply(
+        lambda x: ",".join(x.dropna().astype(str)),
+        axis=1,
+    )
+    # Retrieve raw_drop_average_velocity
+    df["raw_drop_average_velocity"] = df_to_parse.iloc[:, 67:99].apply(
+        lambda x: ",".join(x.dropna().astype(str)),
+        axis=1,
+    )
+
+    # Retrieve raw_drop_number
+    df_raw_drop_number = df_to_parse.iloc[:, 99].squeeze()
+    df_raw_drop_number = df_raw_drop_number.str.replace(r"(\w{3})", r"\1,", regex=True)
+    df["raw_drop_number"] = df_raw_drop_number
+
+    # Drop columns not agreeing with DISDRODB L0 standards
+    columns_to_drop = [
+        "firmware_iop",
+        "firmware_dsp",
+        "sensor_time_measurement_start",
+        "sensor_time",
+        "sensor_date",
+        "station_name",
+        "station_number",
+        "sensor_serial_number",
+        "sensor_serial_number",
+        # "number_particles_all_detected",
+    ]
+    df = df.drop(columns=columns_to_drop)
+
+    # Return the dataframe adhering to DISDRODB L0 standards
+    return df