Add 'projects/rocprofiler-sdk/' from commit 'bf0fad1d5406fbc51403ba1aa9621a9d4a9bce2b'

git-subtree-dir: projects/rocprofiler-sdk
git-subtree-mainline: 50a90550e9
git-subtree-split: bf0fad1d54

projects/rocprofiler-sdk/source/lib/python/rocpd/csv.py

@@ -0,0 +1,109 @@
+#!/usr/bin/env python3
+###############################################################################
+# MIT License
+#
+# Copyright (c) 2023 Advanced Micro Devices, Inc.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+###############################################################################
+
+from .importer import RocpdImportData
+from .time_window import apply_time_window
+from . import output_config
+from . import libpyrocpd
+
+
+def write_csv(importData, config):
+    return libpyrocpd.write_csv(importData, config)
+
+
+def execute(input, config=None, window_args=None, **kwargs):
+
+    importData = RocpdImportData(input)
+
+    apply_time_window(importData, **window_args)
+
+    config = (
+        output_config.output_config(**kwargs)
+        if config is None
+        else config.update(**kwargs)
+    )
+
+    write_csv(importData, config)
+
+
+def add_args(parser):
+    """Add csv arguments."""
+
+    return []
+
+
+def process_args(args, valid_args):
+    ret = {}
+    return ret
+
+
+def main(argv=None):
+    import argparse
+    from .time_window import add_args as add_args_time_window
+    from .time_window import process_args as process_args_time_window
+    from .output_config import add_args as add_args_output_config
+    from .output_config import process_args as process_args_output_config
+    from .output_config import add_generic_args, process_generic_args
+
+    parser = argparse.ArgumentParser(
+        description="Convert rocPD to CSV files",
+        allow_abbrev=False,
+        formatter_class=argparse.RawTextHelpFormatter,
+    )
+
+    required_params = parser.add_argument_group("Required arguments")
+
+    required_params.add_argument(
+        "-i",
+        "--input",
+        required=True,
+        type=output_config.check_file_exists,
+        nargs="+",
+        help="Input path and filename to one or more database(s), separated by spaces",
+    )
+
+    valid_out_config_args = add_args_output_config(parser)
+    valid_generic_args = add_generic_args(parser)
+    valid_time_window_args = add_args_time_window(parser)
+    valid_csv_args = add_args(parser)
+
+    args = parser.parse_args(argv)
+
+    out_cfg_args = process_args_output_config(args, valid_out_config_args)
+    generic_out_cfg_args = process_generic_args(args, valid_generic_args)
+    window_args = process_args_time_window(args, valid_time_window_args)
+    csv_args = process_args(args, valid_csv_args)
+
+    all_args = {
+        **out_cfg_args,
+        **generic_out_cfg_args,
+        **csv_args,
+    }
+
+    execute(args.input, window_args=window_args, **all_args)
+
+
+if __name__ == "__main__":
+    main()