From 9031338597134a8e3811bd36768c3703f906631e Mon Sep 17 00:00:00 2001
From: Alex Forencich <alex@alexforencich.com>
Date: Sun, 6 Dec 2020 00:40:09 -0800
Subject: [PATCH] Accept clock period directly in PtpClock constructor

---
 cocotbext/eth/ptp.py              | 17 +++++++----------
 tests/ptp_clock/test_ptp_clock.py |  6 +-----
 2 files changed, 8 insertions(+), 15 deletions(-)

diff --git a/cocotbext/eth/ptp.py b/cocotbext/eth/ptp.py
index 0711ada..4a71384 100644
--- a/cocotbext/eth/ptp.py
+++ b/cocotbext/eth/ptp.py
@@ -42,11 +42,7 @@ class PtpClock(object):
             pps=None,
             clock=None,
             reset=None,
-            period_ns=0x6,
-            period_fns=0x6666,
-            drift_ns=0x0,
-            drift_fns=0x0002,
-            drift_rate=5,
+            period_ns=6.4,
             *args, **kwargs):
 
         self.log = logging.getLogger(f"cocotb.eth.{type(self).__name__}")
@@ -57,11 +53,12 @@ class PtpClock(object):
         self.clock = clock
         self.reset = reset
 
-        self.period_ns = period_ns
-        self.period_fns = period_fns
-        self.drift_ns = drift_ns
-        self.drift_fns = drift_fns
-        self.drift_rate = drift_rate
+        self.period_ns = 0
+        self.period_fns = 0
+        self.drift_ns = 0
+        self.drift_fns = 0
+        self.drift_rate = 0
+        self.set_period_ns(period_ns)
 
         self.log.info("PTP clock")
         self.log.info("cocotbext-eth version %s", __version__)
diff --git a/tests/ptp_clock/test_ptp_clock.py b/tests/ptp_clock/test_ptp_clock.py
index 57add01..676e148 100644
--- a/tests/ptp_clock/test_ptp_clock.py
+++ b/tests/ptp_clock/test_ptp_clock.py
@@ -52,11 +52,7 @@ class TB(object):
             pps=dut.pps,
             clock=dut.clk,
             reset=dut.rst,
-            period_ns=0x6,
-            period_fns=0x6666,
-            drift_ns=0x0,
-            drift_fns=0x0002,
-            drift_rate=5,
+            period_ns=6.4
         )
 
     async def reset(self):