From 233d63077e5d56a467cffcae4bf9f163fd1973a0 Mon Sep 17 00:00:00 2001
From: Christophe Bedard <bedard.christophe@gmail.com>
Date: Wed, 6 May 2020 10:14:32 -0400
Subject: [PATCH] Add 'quiet' option for Processor to not print any output

Signed-off-by: Christophe Bedard <bedard.christophe@gmail.com>
---
 .../tracetools_analysis/processor/__init__.py | 20 +++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/tracetools_analysis/tracetools_analysis/processor/__init__.py b/tracetools_analysis/tracetools_analysis/processor/__init__.py
index 9b5a585..f6007ec 100644
--- a/tracetools_analysis/tracetools_analysis/processor/__init__.py
+++ b/tracetools_analysis/tracetools_analysis/processor/__init__.py
@@ -289,12 +289,14 @@ class Processor():
     def __init__(
         self,
         *handlers: EventHandler,
+        quiet: bool = False,
         **kwargs,
     ) -> None:
         """
         Create a Processor.
 
         :param handlers: the `EventHandler`s to use for processing
+        :param quiet: whether to not print any output, like progress information
         :param kwargs: the parameters to pass on to new handlers
         """
         self._initial_handlers = list(handlers)
@@ -303,9 +305,10 @@ class Processor():
         self._expanded_handlers = self._expand_dependencies(*handlers, **kwargs)
         self._handler_multimap = self._get_handler_maps(self._expanded_handlers)
         self._register_with_handlers(self._expanded_handlers)
+        self._quiet = quiet
         self._progress_display = ProcessingProgressDisplay(
             [type(handler).__name__ for handler in self._expanded_handlers],
-        )
+        ) if not self._quiet else None
         self._processing_done = False
 
     @staticmethod
@@ -401,12 +404,17 @@ class Processor():
             self._check_required_events(events)
 
         if not self._processing_done:
-            self._progress_display.set_work_total(len(events))
-            for event in events:
-                self._process_event(event)
-                self._progress_display.did_work()
+            # Split into two versions so that performance is optimal
+            if self._progress_display is None:
+                for event in events:
+                    self._process_event(event)
+            else:
+                self._progress_display.set_work_total(len(events))
+                for event in events:
+                    self._process_event(event)
+                    self._progress_display.did_work()
+                self._progress_display.done(erase=erase_progress)
             self._processing_done = True
-            self._progress_display.done(erase=erase_progress)
 
     def _process_event(self, event: DictEvent) -> None:
         """Process a single event."""