Added ability to specify units along with -t command-line option

1 files changed, 1 insertions, 7 deletions

diff --git a/unit_trace/viz/viewer.py b/unit_trace/viz/viewer.py
index bb1385c..4613c47 100644
--- a/unit_trace/viz/viewer.py
+++ b/unit_trace/viz/viewer.py
@@ -856,14 +856,8 @@ class MainWindow(gtk.Window):
             if ret == gtk.RESPONSE_ACCEPT:
                 err, time = None, None
                 try:
-                    input_tokens = self.text_input_dialog.get_input().split()
-                    if len(input_tokens) > 2:
-                        raise ValueError
+                    time = util.parse_time(self.text_input_dialog.get_input())
 
-                    if len(input_tokens) > 1:
-                        unit = util.parse_unit(input_tokens[1])
-
-                    time = unit.native_to_nsec(float(input_tokens[0]))
                     start, end = self.notebook.get_nth_page(self.notebook.last_page).get_graph_area() \
                                  .get_schedule().get_time_bounds()
                     if time < start or time > end: