Move definition of TX timestamp and fix bug after RX reset

1 files changed, 14 insertions, 4 deletions

diff --git a/AlignSample.hpp b/AlignSample.hpp
index 78c430f..5757ff9 100644
--- a/AlignSample.hpp
+++ b/AlignSample.hpp
@@ -88,14 +88,24 @@ class AlignSample {
     private:
         bool align();
 
-        double m_rx_sample_time() const {
+        double m_rx_sample_time(size_t offset = 0) const {
             return m_first_rx_sample_time +
-                (double)m_num_rx_samples_dropped / samplerate;
+                (double)(m_num_rx_samples_dropped + offset) / (double)samplerate;
         }
 
-        double m_tx_sample_time() const {
+        double m_tx_sample_time(size_t offset = 0) const {
             return m_first_tx_sample_time +
-                (double)m_num_tx_samples_dropped / samplerate;
+                (double)(m_num_tx_samples_dropped + offset) / (double)samplerate;
+        }
+
+        void m_drop_tx_samples(size_t samples) {
+            m_txsamples.erase(m_txsamples.begin(), m_txsamples.begin() + samples);
+            m_num_tx_samples_dropped += samples;
+        }
+
+        void m_drop_rx_samples(size_t samples) {
+            m_rxsamples.erase(m_rxsamples.begin(), m_rxsamples.begin() + samples);
+            m_num_rx_samples_dropped += samples;
         }
 
         std::mutex m_mutex;