From: Florent Kermarrec <florent@enjoy-digital.fr>
Date: Fri, 26 Jun 2020 06:36:16 +0000 (+0200)
Subject: interconnect/axi/AXIStreamInterface: add tuser support.
X-Git-Tag: 24jan2021_ls180~134
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=34e9d12ef24ccdb7204445e37614af92ab5e9acf;p=litex.git

interconnect/axi/AXIStreamInterface: add tuser support.
---

diff --git a/litex/soc/interconnect/axi.py b/litex/soc/interconnect/axi.py
index e95e01e0..e094a3d1 100644
--- a/litex/soc/interconnect/axi.py
+++ b/litex/soc/interconnect/axi.py
@@ -141,9 +141,13 @@ class AXILiteInterface:
 # AXI Stream Definition ----------------------------------------------------------------------------
 
 class AXIStreamInterface(stream.Endpoint):
-    def __init__(self, data_width=32):
+    def __init__(self, data_width=32, user_width=0):
         self.data_width = data_width
-        stream.Endpoint.__init__(self, [("data", data_width)])
+        self.user_width = user_width
+        axi_layout = [("data", data_width)]
+        if self.user_width:
+            axi_layout += [("user", user_width)]
+        stream.Endpoint.__init__(self, axi_layout)
 
     def get_ios(self, bus_name="axi"):
         subsignals = [
@@ -152,6 +156,8 @@ class AXIStreamInterface(stream.Endpoint):
             Subsignal("tready", Pins(1)),
             Subsignal("tdata",  Pins(self.data_width)),
         ]
+        if self.user_width:
+            subsignals += [Subsignal("tuser", Pins(self.user_width))]
         ios = [(bus_name , 0) + tuple(subsignals)]
         return ios
 
@@ -163,11 +169,15 @@ class AXIStreamInterface(stream.Endpoint):
             r.append(self.ready.eq(pads.tready))
             r.append(pads.tlast.eq(self.last))
             r.append(pads.tdata.eq(self.data))
+            if self.user_width:
+                r.append(pads.tuser.eq(self.user))
         if mode == "slave":
             r.append(self.valid.eq(pads.tvalid))
             r.append(pads.tready.eq(self.ready))
             r.append(self.last.eq(pads.tlast))
             r.append(self.data.eq(pads.tdata))
+            if self.user_width:
+                r.append(self.user.eq(pads.tuser))
         return r
 
 # AXI Bursts to Beats ------------------------------------------------------------------------------