#ifndef __LINUX_TI_AM335X_TSCADC_MFD_H
 #define __LINUX_TI_AM335X_TSCADC_MFD_H
 
+#include <linux/bitfield.h>
 #include <linux/mfd/core.h>
 #include <linux/units.h>
 
 #define IRQENB_PENUP           BIT(9)
 
 /* Step Configuration */
-#define STEPCONFIG_MODE_MASK   (3 << 0)
-#define STEPCONFIG_MODE(val)   ((val) << 0)
+#define STEPCONFIG_MODE_MASK   GENMASK(1, 0)
+#define STEPCONFIG_MODE(val)   FIELD_PREP(STEPCONFIG_MODE_MASK, (val))
 #define STEPCONFIG_MODE_SWCNT  STEPCONFIG_MODE(1)
 #define STEPCONFIG_MODE_HWSYNC STEPCONFIG_MODE(2)
-#define STEPCONFIG_AVG_MASK    (7 << 2)
-#define STEPCONFIG_AVG(val)    ((val) << 2)
+#define STEPCONFIG_AVG_MASK    GENMASK(4, 2)
+#define STEPCONFIG_AVG(val)    FIELD_PREP(STEPCONFIG_AVG_MASK, (val))
 #define STEPCONFIG_AVG_16      STEPCONFIG_AVG(4)
 #define STEPCONFIG_XPP         BIT(5)
 #define STEPCONFIG_XNN         BIT(6)
 #define STEPCONFIG_YNN         BIT(8)
 #define STEPCONFIG_XNP         BIT(9)
 #define STEPCONFIG_YPN         BIT(10)
-#define STEPCONFIG_RFP(val)    ((val) << 12)
-#define STEPCONFIG_RFP_VREFP   (0x3 << 12)
-#define STEPCONFIG_INM_MASK    (0xF << 15)
-#define STEPCONFIG_INM(val)    ((val) << 15)
+#define STEPCONFIG_RFP_VREFP   GENMASK(13, 12)
+#define STEPCONFIG_RFP(val)    FIELD_PREP(STEPCONFIG_RFP_VREFP, (val))
+#define STEPCONFIG_INM_MASK    GENMASK(18, 15)
+#define STEPCONFIG_INM(val)    FIELD_PREP(STEPCONFIG_INM_MASK, (val))
 #define STEPCONFIG_INM_ADCREFM STEPCONFIG_INM(8)
-#define STEPCONFIG_INP_MASK    (0xF << 19)
-#define STEPCONFIG_INP(val)    ((val) << 19)
+#define STEPCONFIG_INP_MASK    GENMASK(22, 19)
+#define STEPCONFIG_INP(val)    FIELD_PREP(STEPCONFIG_INP_MASK, (val))
 #define STEPCONFIG_INP_AN4     STEPCONFIG_INP(4)
 #define STEPCONFIG_INP_ADCREFM STEPCONFIG_INP(8)
 #define STEPCONFIG_FIFO1       BIT(26)
-#define STEPCONFIG_RFM(val)    ((val) << 23)
-#define STEPCONFIG_RFM_VREFN   (0x3 << 23)
+#define STEPCONFIG_RFM_VREFN   GENMASK(24, 23)
+#define STEPCONFIG_RFM(val)    FIELD_PREP(STEPCONFIG_RFM_VREFN, (val))
 
 /* Delay register */
-#define STEPDELAY_OPEN_MASK    (0x3FFFF << 0)
-#define STEPDELAY_OPEN(val)    ((val) << 0)
+#define STEPDELAY_OPEN_MASK    GENMASK(17, 0)
+#define STEPDELAY_OPEN(val)    FIELD_PREP(STEPDELAY_OPEN_MASK, (val))
 #define STEPCONFIG_OPENDLY     STEPDELAY_OPEN(0x098)
-#define STEPDELAY_SAMPLE_MASK  (0xFF << 24)
-#define STEPDELAY_SAMPLE(val)  ((val) << 24)
+#define STEPDELAY_SAMPLE_MASK  GENMASK(31, 24)
+#define STEPDELAY_SAMPLE(val)  FIELD_PREP(STEPDELAY_SAMPLE_MASK, (val))
 #define STEPCONFIG_SAMPLEDLY   STEPDELAY_SAMPLE(0)
 
 /* Charge Config */
-#define STEPCHARGE_RFP_MASK    (7 << 12)
-#define STEPCHARGE_RFP(val)    ((val) << 12)
+#define STEPCHARGE_RFP_MASK    GENMASK(14, 12)
+#define STEPCHARGE_RFP(val)    FIELD_PREP(STEPCHARGE_RFP_MASK, (val))
 #define STEPCHARGE_RFP_XPUL    STEPCHARGE_RFP(1)
-#define STEPCHARGE_INM_MASK    (0xF << 15)
-#define STEPCHARGE_INM(val)    ((val) << 15)
+#define STEPCHARGE_INM_MASK    GENMASK(18, 15)
+#define STEPCHARGE_INM(val)    FIELD_PREP(STEPCHARGE_INM_MASK, (val))
 #define STEPCHARGE_INM_AN1     STEPCHARGE_INM(1)
-#define STEPCHARGE_INP_MASK    (0xF << 19)
-#define STEPCHARGE_INP(val)    ((val) << 19)
-#define STEPCHARGE_RFM_MASK    (3 << 23)
-#define STEPCHARGE_RFM(val)    ((val) << 23)
+#define STEPCHARGE_INP_MASK    GENMASK(22, 19)
+#define STEPCHARGE_INP(val)    FIELD_PREP(STEPCHARGE_INP_MASK, (val))
+#define STEPCHARGE_RFM_MASK    GENMASK(24, 23)
+#define STEPCHARGE_RFM(val)    FIELD_PREP(STEPCHARGE_RFM_MASK, (val))
 #define STEPCHARGE_RFM_XNUR    STEPCHARGE_RFM(1)
 
 /* Charge delay */
-#define CHARGEDLY_OPEN_MASK    (0x3FFFF << 0)
-#define CHARGEDLY_OPEN(val)    ((val) << 0)
+#define CHARGEDLY_OPEN_MASK    GENMASK(17, 0)
+#define CHARGEDLY_OPEN(val)    FIELD_PREP(CHARGEDLY_OPEN_MASK, (val))
 #define CHARGEDLY_OPENDLY      CHARGEDLY_OPEN(0x400)
 
 /* Control register */
 #define CNTRLREG_STEPID                BIT(1)
 #define CNTRLREG_STEPCONFIGWRT BIT(2)
 #define CNTRLREG_POWERDOWN     BIT(4)
-#define CNTRLREG_AFE_CTRL_MASK (3 << 5)
-#define CNTRLREG_AFE_CTRL(val) ((val) << 5)
+#define CNTRLREG_AFE_CTRL_MASK GENMASK(6, 5)
+#define CNTRLREG_AFE_CTRL(val) FIELD_PREP(CNTRLREG_AFE_CTRL_MASK, (val))
 #define CNTRLREG_4WIRE         CNTRLREG_AFE_CTRL(1)
 #define CNTRLREG_5WIRE         CNTRLREG_AFE_CTRL(2)
 #define CNTRLREG_8WIRE         CNTRLREG_AFE_CTRL(3)
 #define CNTRLREG_TSCENB                BIT(7)
 
 /* FIFO READ Register */
-#define FIFOREAD_DATA_MASK     (0xfff << 0)
-#define FIFOREAD_CHNLID_MASK   (0xf << 16)
+#define FIFOREAD_DATA_MASK     GENMASK(11, 0)
+#define FIFOREAD_CHNLID_MASK   GENMASK(19, 16)
 
 /* DMA ENABLE/CLEAR Register */
 #define DMA_FIFO0              BIT(0)