From 0e82858eea6e3fdea18080b28a73f0c726cd2a92 Mon Sep 17 00:00:00 2001
From: Sebastien Deronne <sebastien.deronne@gmail.com>
Date: Sat, 7 May 2022 14:00:09 +0200
Subject: [PATCH] wifi: Extend WifiTxVector to indicate whether the Common
 field is present in the HE-SIG-B

---
 src/wifi/model/wifi-tx-vector.cc | 6 ++++++
 src/wifi/model/wifi-tx-vector.h  | 7 +++++++
 2 files changed, 13 insertions(+)

diff --git a/src/wifi/model/wifi-tx-vector.cc b/src/wifi/model/wifi-tx-vector.cc
index a897264e7..71500bd81 100644
--- a/src/wifi/model/wifi-tx-vector.cc
+++ b/src/wifi/model/wifi-tx-vector.cc
@@ -583,6 +583,12 @@ WifiTxVector::GetHeMuUserInfoMap()
     return m_muUserInfos;
 }
 
+bool
+WifiTxVector::IsSigBCompression() const
+{
+    return IsDlMuMimo() && !IsDlOfdma();
+}
+
 void
 WifiTxVector::SetInactiveSubchannels(const std::vector<bool>& inactiveSubchannels)
 {
diff --git a/src/wifi/model/wifi-tx-vector.h b/src/wifi/model/wifi-tx-vector.h
index 85e720c8d..66476f3d4 100644
--- a/src/wifi/model/wifi-tx-vector.h
+++ b/src/wifi/model/wifi-tx-vector.h
@@ -446,6 +446,13 @@ class WifiTxVector
      */
     UserInfoMapOrderedByRus GetUserInfoMapOrderedByRus(uint8_t p20Index) const;
 
+    /**
+     * Indicate whether the Common field is present in the HE-SIG-B field.
+     *
+     * \return true if the Common field is present in the HE-SIG-B, false otherwise
+     */
+    bool IsSigBCompression() const;
+
     /**
      * Set the 20 MHz subchannels that are punctured.
      *