drivers: edma_rev2: support MUX register in MP space

On some EDMA versions, the channel MUX register may be
placed inside the MP region (instead of the TCD region).
Add support for this. EDMA versions with this quirk will
signal it via the `EDMA_HAS_MP_MUX_FLAG` flag.

Signed-off-by: Laurentiu Mihalcea <laurentiu.mihalcea@nxp.com>

Author: LaurentiuM1234

mcux/mcux-sdk/drivers/edma_rev2/fsl_edma_rev2.c

@@ -53,7 +53,7 @@ static uint32_t EDMA_GetChannelRegBase(edma_config_t *cfg, int channel, uint32_t
 {
     uint32_t index = EDMA_REG_INDEX(reg);
 
-    if (reg == EDMA_MP_CH_GRPRI) {
+    if (reg == EDMA_MP_CH_GRPRI || reg == EDMA_MP_CH_MUX) {
         return cfg->regmap + cfg->registerLayout[index] + channel * 0x4;
     } else {
         return EDMA_GetChannelBase(cfg, channel) + cfg->registerLayout[index];
@@ -118,6 +118,8 @@ uint32_t EDMA_MPRegRead(edma_config_t *cfg, uint32_t reg)
 
 status_t EDMA_SetChannelMux(edma_config_t *cfg, int channel, uint32_t mux)
 {
+    uint32_t muxReg;
+
     if (channel >= cfg->channels) {
         return kStatus_InvalidArgument;
     }
@@ -126,11 +128,17 @@ status_t EDMA_SetChannelMux(edma_config_t *cfg, int channel, uint32_t mux)
         return kStatus_EDMA_InvalidConfiguration;
     }
 
-    if (EDMA_ChannelRegRead(cfg, channel, EDMA_TCD_CH_MUX) != 0 && mux != 0) {
+    if (cfg->flags & EDMA_HAS_MP_MUX_FLAG) {
+        muxReg = EDMA_MP_CH_MUX;
+    } else {
+        muxReg = EDMA_TCD_CH_MUX;
+    }
+
+    if (EDMA_ChannelRegRead(cfg, channel, muxReg) != 0 && mux != 0) {
         return kStatus_Busy;
     }
 
-    EDMA_ChannelRegWrite(cfg, channel, EDMA_TCD_CH_MUX, mux);
+    EDMA_ChannelRegWrite(cfg, channel, muxReg, mux);
 
     return kStatus_Success;
 }

mcux/mcux-sdk/drivers/edma_rev2/fsl_edma_rev2.h

@@ -16,6 +16,7 @@ extern "C" {
 /* flag macros */
 #define EDMA_HAS_CH_MUX_FLAG EDMA_BIT(0)
 #define EDMA_ALLOWS_128B_TRANSFER_FLAG EDMA_BIT(1)
+#define EDMA_HAS_MP_MUX_FLAG EDMA_BIT(2)
 
 /* internal utility macros */
 #define EDMA_BIT(x) (1 << (x))
@@ -58,6 +59,7 @@ extern "C" {
 #define EDMA_MP_HRS_HIGH_INDEX 24
 #define EDMA_TCD_CH_MUX_INDEX 25
 #define EDMA_TCD_CH_MATTR_INDEX 26
+#define EDMA_MP_CH_MUX_INDEX 27
 
 /* utility macros - for external and internal usage */
 #define EDMA_HAS_MUX(cfg) ((cfg)->flags & EDMA_HAS_CH_MUX_FLAG)
@@ -115,6 +117,7 @@ extern "C" {
 #define EDMA_MP_INT_HIGH EDMA_REGISTER_MAKE(EDMA_MP_INT_HIGH_INDEX, kEDMA_RegAccess32)
 #define EDMA_MP_HRS_LOW EDMA_REGISTER_MAKE(EDMA_MP_HRS_LOW_INDEX, kEDMA_RegAccess32)
 #define EDMA_MP_HRS_HIGH EDMA_REGISTER_MAKE(EDMA_MP_HRS_HIGH_INDEX, kEDMA_RegAccess32)
+#define EDMA_MP_CH_MUX EDMA_REGISTER_MAKE(EDMA_MP_CH_MUX_INDEX, kEDMA_RegAccess32)
 /* non-common TCD-related registers */
 #define EDMA_TCD_CH_MUX EDMA_REGISTER_MAKE(EDMA_TCD_CH_MUX_INDEX, kEDMA_RegAccess32)
 #define EDMA_TCD_CH_MATTR EDMA_REGISTER_MAKE(EDMA_TCD_CH_MATTR_INDEX, kEDMA_RegAccess16)