MLIA-549 Integrate TOSA checker into MLIA

- Add new module for TOSA
- Add advisor workflow components
- Use TOSA checker for getting operators compatibility
  information

Change-Id: I769e5e2a84e15779658f0895b4a347384def63bf

1 files changed, 35 insertions, 0 deletions

diff --git a/src/mlia/devices/tosa/handlers.py b/src/mlia/devices/tosa/handlers.py
new file mode 100644
index 0000000..00c18c5
--- /dev/null
+++ b/src/mlia/devices/tosa/handlers.py
@@ -0,0 +1,35 @@
+# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
+# SPDX-License-Identifier: Apache-2.0
+"""TOSA Advisor event handlers."""
+# pylint: disable=R0801
+import logging
+from typing import Optional
+
+from mlia.core._typing import PathOrFileLike
+from mlia.core.events import CollectedDataEvent
+from mlia.core.handlers import WorkflowEventsHandler
+from mlia.devices.tosa.events import TOSAAdvisorEventHandler
+from mlia.devices.tosa.events import TOSAAdvisorStartedEvent
+from mlia.devices.tosa.operators import TOSACompatibilityInfo
+from mlia.devices.tosa.reporters import tosa_formatters
+
+logger = logging.getLogger(__name__)
+
+
+class TOSAEventHandler(WorkflowEventsHandler, TOSAAdvisorEventHandler):
+    """Event handler for TOSA advisor."""
+
+    def __init__(self, output: Optional[PathOrFileLike] = None) -> None:
+        """Init event handler."""
+        super().__init__(tosa_formatters, output)
+
+    def on_tosa_advisor_started(self, event: TOSAAdvisorStartedEvent) -> None:
+        """Handle TOSAAdvisorStartedEvent event."""
+        self.reporter.submit(event.device)
+
+    def on_collected_data(self, event: CollectedDataEvent) -> None:
+        """Handle CollectedDataEvent event."""
+        data_item = event.data_item
+
+        if isinstance(data_item, TOSACompatibilityInfo):
+            self.reporter.submit(data_item.operators, delay_print=True)