setup cli

Author: SBlokhuizen

pyproject.toml

@@ -22,6 +22,7 @@ dependencies = [
     "pygments",
     "plotly",
     "holoviews",
+    "kaleido",
 ]
 dynamic = ["version"]
 

waveform_editor/cli.py

@@ -6,6 +6,8 @@
 from rich.table import Table
 
 import waveform_editor
+from waveform_editor.waveform_exporter import WaveformExporter
+from waveform_editor.yaml_parser import YamlParser
 
 logger = logging.getLogger(__name__)
 
@@ -52,5 +54,52 @@ def print_version():
     console.Console().print(grid)
 
 
+@cli.command("export-csv")
+@click.argument("yaml", type=click.Path(exists=True))
+@click.argument("output", type=str)
+def export_csv(yaml, output):
+    """Export waveform data to a CSV file."""
+    waveform = load_waveform_from_yaml(yaml)
+    exporter = WaveformExporter(waveform)
+    exporter.to_csv(output)
+
+
+@cli.command("export-png")
+@click.argument("yaml", type=click.Path(exists=True))
+@click.argument("output", type=str)
+def export_png(yaml, output):
+    """Export waveform data to a CSV file."""
+    waveform = load_waveform_from_yaml(yaml)
+    exporter = WaveformExporter(waveform)
+    exporter.to_png(output)
+
+
+# # TODO:
+# @cli.command("export-ids")
+# @click.argument("yaml", type=str)
+# @click.argument("uri", type=str)
+# @click.option("--dd-version", type=str)
+# def export_ids(yaml, uri, path):
+#     """Export waveform data to an IDS."""
+#     # waveform = load_waveform_from_yaml(yaml)
+#     # exporter = WaveformExporter(waveform)
+#     # exporter.to_ids(uri, path)
+
+
+def load_waveform_from_yaml(yaml_file):
+    with open(yaml_file) as file:
+        yaml_str = file.read()
+    yaml_parser = YamlParser()
+    yaml_parser.parse_waveforms(yaml_str)
+    annotations = yaml_parser.waveform.annotations
+    if annotations:
+        click.secho(
+            "The following errors and warnings were detected in the YAML file:\n"
+            f"{annotations}",
+            fg="red",
+        )
+    return yaml_parser.waveform
+
+
 if __name__ == "__main__":
     cli()

waveform_editor/waveform.py

@@ -129,6 +129,14 @@ def calc_length(self):
         """Returns the length of the waveform."""
         return self.tendencies[-1].end - self.tendencies[0].start
 
+    def get_start(self):
+        """Returns the start time of the first tendency in the waveform."""
+        return self.tendencies[0].start
+
+    def get_end(self):
+        """Returns the end time of the last tendency in the waveform."""
+        return self.tendencies[-1].end
+
     def _process_waveform(self, waveform):
         """Processes the waveform YAML and populates the tendencies list.