From 7428fec2000cb223850b853f2ef2139932064481 Mon Sep 17 00:00:00 2001
From: Astro <astro@spaceboyz.net>
Date: Thu, 23 May 2019 15:36:34 +0200
Subject: [PATCH] uart: add more channel_sts flags, wait for tx_fifo_empty()
 before sending

---
 src/uart/mod.rs  |  6 ++++++
 src/uart/regs.rs | 19 +++++++++++++++++++
 2 files changed, 25 insertions(+)

diff --git a/src/uart/mod.rs b/src/uart/mod.rs
index a146e41..cfc0a98 100644
--- a/src/uart/mod.rs
+++ b/src/uart/mod.rs
@@ -146,10 +146,16 @@ impl Uart {
     pub fn tx_fifo_full(&self) -> bool {
         self.regs.channel_sts.read().txfull()
     }
+
+    pub fn tx_fifo_empty(&self) -> bool {
+        self.regs.channel_sts.read().txempty()
+    }
 }
 
 impl fmt::Write for Uart {
     fn write_str(&mut self, s: &str) -> Result<(), fmt::Error> {
+        while !self.tx_fifo_empty() {}
+
         for b in s.bytes() {
             self.write_byte(b);
         }
diff --git a/src/uart/regs.rs b/src/uart/regs.rs
index 2c01390..1a5195d 100644
--- a/src/uart/regs.rs
+++ b/src/uart/regs.rs
@@ -74,7 +74,26 @@ register!(baud_rate_gen, BaudRateGen, RW, u32);
 register_bits!(baud_rate_gen, cd, u16, 0, 15);
 
 register!(channel_sts, ChannelSts, RO, u32);
+/// Transmitter FIFO Nearly Full
+register_bit!(channel_sts, tnful, 14);
+/// Tx FIFO fill level is greater than or equal to TTRIG?
+register_bit!(channel_sts, ttrig, 13);
+/// Rx FIFO fill level is greater than or equal to FDEL?
+register_bit!(channel_sts, flowdel, 12);
+/// Transmitter state machine active?
+register_bit!(channel_sts, tactive, 11);
+/// Receiver state machine active?
+register_bit!(channel_sts, ractive, 10);
+/// Tx FIFO is full?
 register_bit!(channel_sts, txfull, 4);
+/// Tx FIFO is empty?
+register_bit!(channel_sts, txempty, 3);
+/// Rx FIFO is full?
+register_bit!(channel_sts, rxfull, 2);
+/// Rx FIFO is empty?
+register_bit!(channel_sts, rxempty, 1);
+/// Rx FIFO fill level is greater than or equal to RTRIG?
+register_bit!(channel_sts, rxovr, 0);
 
 register!(tx_rx_fifo, TxRxFifo, RW, u32);
 register_bits!(tx_rx_fifo, data, u32, 0, 31);