From e3dad193733b50ea4693a6cecf8f93ba40c6c968 Mon Sep 17 00:00:00 2001
From: Hideki Shimizu <shimiz@cns.s.u-tokyo.ac.jp>
Date: Thu, 7 Nov 2019 17:52:12 +0900
Subject: [PATCH] new attribution, 'Unit'

---
 data.yaml         |  1 +
 timing_diagram.py | 10 +++++++---
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/data.yaml b/data.yaml
index e64d112..5d9298c 100644
--- a/data.yaml
+++ b/data.yaml
@@ -1,6 +1,7 @@
 Title: Timing Diagram
 
 Range : [0, 200]
+Unit : ns
 
 Data:
   Trigger :
diff --git a/timing_diagram.py b/timing_diagram.py
index 587c289..c0b1e35 100644
--- a/timing_diagram.py
+++ b/timing_diagram.py
@@ -33,12 +33,12 @@ draw_start_point = 0.15
 draw_end_point = 0.85
 
 class LogicTiming :
-    def __init__(self, _name, _min=0, _max=1) :
+    def __init__(self, _name, _min=0, _max=1, _unit='a.u.') :
         self.dyanmic_range = [_min, _max]
         self.range = float(_max - _min)
         self.signals = []
         self.name = _name
-        self.unit = 'ns'
+        self.unit = _unit
 
     def set_signal(self, leading, trailing) :
         self.signals.append([leading, trailing])
@@ -46,11 +46,15 @@ class LogicTiming :
 data_i = 0 
 range_canvas = draw_end_point - draw_start_point
 data_range = [ yaml_data["Range"][0], yaml_data["Range"][1] ]
+if 'Unit' in yaml_data :
+    data_unit = yaml_data['Unit']
+else : 
+    data_unit = 'a.u.'
 l = []
 for d in data :
     dname = str(d)
     d = data[d]
-    ch = LogicTiming(dname, _min = data_range[0], _max = data_range[1])
+    ch = LogicTiming(dname, _min = data_range[0], _max = data_range[1], _unit=data_unit)
     for s in d :
         ch.set_signal(float(s[0]), float(s[1]))
     height = pulse_height
-- 
GitLab