rapidtide 2.9.6__py3-none-any.whl → 3.1.3__py3-none-any.whl

rapidtide/workflows/tcfrom2col.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 #
-#   Copyright 2016-2024 Blaise Frederick
+#   Copyright 2016-2025 Blaise Frederick
 #
 #   Licensed under the Apache License, Version 2.0 (the "License");
 #   you may not use this file except in compliance with the License.
@@ -17,14 +17,42 @@
 #
 #
 import argparse
+from argparse import Namespace
+from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
 import numpy as np
+from numpy.typing import NDArray
 
 import rapidtide.io as tide_io
 import rapidtide.util as tide_util
 
 
-def _get_parser():
+def _get_parser() -> Any:
+    """
+    Create and configure argument parser for tcfrom2col command line tool.
+
+    This function sets up an argument parser with required and optional arguments
+    for processing two-column data files into time course output.
+
+    Returns
+    -------
+    argparse.ArgumentParser
+        Configured argument parser object with all required and optional arguments
+        for the tcfrom2col tool.
+
+    Notes
+    -----
+    The parser includes the following required arguments:
+    - infilename: input two column file name
+    - timestep: time step of output time course in seconds
+    - numpoints: number of output time points
+    - outfilename: output time course file name
+
+    Examples
+    --------
+    >>> parser = _get_parser()
+    >>> args = parser.parse_args()
+    """
     # get the command line parameters
     parser = argparse.ArgumentParser(
         prog="tcfrom2col",
@@ -48,7 +76,53 @@ def _get_parser():
     return parser
 
 
-def tcfrom2col(args):
+def tcfrom2col(args: Any) -> None:
+    """
+    Convert two-column data to tidal constituent data.
+
+    This function reads two-column input data, processes it to extract tidal constituents,
+    and writes the results to an output file. The input data is expected to be in a format
+    suitable for tidal analysis, and the output contains the computed tidal constituents.
+
+    Parameters
+    ----------
+    args : Any
+        An object containing the following attributes:
+        - infilename : str
+            Path to the input file containing two-column data
+        - outfilename : str
+            Path to the output file for writing tidal constituent data
+        - numpoints : int
+            Number of data points in the time series
+        - timestep : float
+            Time step between data points
+        - debug : bool
+            Flag to enable debug printing
+
+    Returns
+    -------
+    None
+        This function does not return a value but writes output to a file.
+
+    Notes
+    -----
+    The function uses `tide_io.readvecs` to read input data and `tide_util.maketcfrom2col`
+    to perform the tidal constituent calculation. The time axis is generated based on the
+    number of points and time step provided in the args object.
+
+    Examples
+    --------
+    >>> class Args:
+    ...     def __init__(self):
+    ...         self.infilename = 'input.txt'
+    ...         self.outfilename = 'output.txt'
+    ...         self.numpoints = 1000
+    ...         self.timestep = 300.0
+    ...         self.debug = False
+    ...
+    >>> args = Args()
+    >>> tcfrom2col(args)
+    """
     if args.debug:
         print(args)
 

rapidtide/workflows/tcfrom3col.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 #
-#   Copyright 2016-2024 Blaise Frederick
+#   Copyright 2016-2025 Blaise Frederick
 #
 #   Licensed under the Apache License, Version 2.0 (the "License");
 #   you may not use this file except in compliance with the License.
@@ -17,14 +17,40 @@
 #
 #
 import argparse
+from argparse import Namespace
+from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
 import numpy as np
+from numpy.typing import NDArray
 
 import rapidtide.io as tide_io
 import rapidtide.util as tide_util
 
 
-def _get_parser():
+def _get_parser() -> Any:
+    """
+    Create and configure argument parser for command line interface.
+
+    This function initializes an ArgumentParser object with specific parameters
+    required for processing three-column data files and generating time course output.
+
+    Returns
+    -------
+    argparse.ArgumentParser
+        Configured argument parser object with all required and optional arguments
+
+    Notes
+    -----
+    The parser expects exactly four positional arguments followed by an optional
+    debug flag. The function is designed for use with the tcfrom3col program.
+
+    Examples
+    --------
+    >>> parser = _get_parser()
+    >>> args = parser.parse_args(['input.txt', '0.1', '100', 'output.txt'])
+    >>> print(args.infilename)
+    'input.txt'
+    """
     # get the command line parameters
     parser = argparse.ArgumentParser(
         prog="tcfrom3col",
@@ -48,7 +74,53 @@ def _get_parser():
     return parser
 
 
-def tcfrom3col(args):
+def tcfrom3col(args: Any) -> None:
+    """
+    Convert three-column data to tidal constituent data.
+
+    This function reads three-column input data, processes it to extract tidal constituents,
+    and writes the results to an output file. The input data is expected to contain time,
+    latitude, and longitude columns, and the output contains tidal constituent information.
+
+    Parameters
+    ----------
+    args : Any
+        An object containing the following attributes:
+        - infilename : str
+            Path to the input file containing three-column data
+        - outfilename : str
+            Path to the output file for writing tidal constituent data
+        - numpoints : int
+            Number of data points in the time series
+        - timestep : float
+            Time step between data points
+        - debug : bool
+            Flag to enable debug printing
+
+    Returns
+    -------
+    None
+        This function does not return a value but writes output to a file.
+
+    Notes
+    -----
+    The function uses `tide_io.readvecs` to read input data and `tide_util.maketcfrom3col`
+    to perform the tidal constituent calculation. The time axis is generated using
+    `np.arange` with the specified number of points and time step.
+
+    Examples
+    --------
+    >>> class Args:
+    ...     def __init__(self):
+    ...         self.infilename = 'input.dat'
+    ...         self.outfilename = 'output.dat'
+    ...         self.numpoints = 1000
+    ...         self.timestep = 3600.0
+    ...         self.debug = False
+    ...
+    >>> args = Args()
+    >>> tcfrom3col(args)
+    """
     if args.debug:
         print(args)