From: Rodrigo Alencar <[email protected]>

Add OSK channel with amplitude envelope control capabilities:
- OSK enable/disable via IIO_CHAN_INFO_ENABLE;
- Amplitude ramp rate control via IIO_CHAN_INFO_SAMP_FREQ;
- Amplitude scale factor readback via IIO_CHAN_INFO_RAW (ASF register);
- Automatic OSK step size configurable through the raw_roc extended
  attribute, which allows for selectable step sizes in raw units:
        - 0: no step, means manual mode (NOT pin controlled)
        - 1: I_FS / 2^14 step, automatic mode (pin controlled)
        - 2: 2 I_FS / 2^14 step, automatic mode (pin controlled)
        - 4: 4 I_FS /2^14 step, automatic mode (pin controlled)
        - 8: 8 I_FS /2^14 step, automatic mode (pin controlled)
        - 16383: I_FS step (max), manual mode (pin controlled)

The ASF register is initialized with a default amplitude ramp rate during
device setup to ensure valid readback.

Signed-off-by: Rodrigo Alencar <[email protected]>
---
 drivers/iio/frequency/ad9910.c | 191 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 191 insertions(+)

diff --git a/drivers/iio/frequency/ad9910.c b/drivers/iio/frequency/ad9910.c
index c4e179dda715..890499f67bd5 100644
--- a/drivers/iio/frequency/ad9910.c
+++ b/drivers/iio/frequency/ad9910.c
@@ -242,6 +242,7 @@
  * @AD9910_CHANNEL_DRG_RAMP_UP: DRG ramp up channel
  * @AD9910_CHANNEL_DRG_RAMP_DOWN: DRG ramp down channel
  * @AD9910_CHANNEL_RAM: RAM control output channel
+ * @AD9910_CHANNEL_OSK: Output Shift Keying output channel
  */
 enum ad9910_channel {
        AD9910_CHANNEL_PHY = 100,
@@ -259,6 +260,7 @@ enum ad9910_channel {
        AD9910_CHANNEL_DRG_RAMP_UP = 131,
        AD9910_CHANNEL_DRG_RAMP_DOWN = 132,
        AD9910_CHANNEL_RAM = 140,
+       AD9910_CHANNEL_OSK = 150,
 };
 
 /**
@@ -305,12 +307,14 @@ enum {
        AD9910_CHAN_IDX_DRG_AMP_RAMP_UP,
        AD9910_CHAN_IDX_DRG_AMP_RAMP_DOWN,
        AD9910_CHAN_IDX_RAM,
+       AD9910_CHAN_IDX_OSK,
 };
 
 enum {
        AD9910_POWERDOWN,
        AD9910_DWELL_EN,
        AD9910_ROC,
+       AD9910_ROC_AVAIL,
 };
 
 struct ad9910_data {
@@ -756,6 +760,125 @@ static ssize_t ad9910_drg_roc_write(struct iio_dev 
*indio_dev,
        return len;
 }
 
+static const u32 ad9910_osk_raw_step[] = {
+       0,              /* no step: manual mode (NOT pin controlled) */
+       1,              /* step size 1: automatic mode (pin controlled) */
+       2,              /* step size 2: automatic mode (pin controlled) */
+       4,              /* step size 4: automatic mode (pin controlled) */
+       8,              /* step size 8: automatic mode (pin controlled) */
+       GENMASK(13, 0), /* max step: manual mode (pin controlled) */
+};
+
+static ssize_t ad9910_osk_attrs_read(struct iio_dev *indio_dev,
+                                    uintptr_t private,
+                                    const struct iio_chan_spec *chan,
+                                    char *buf)
+{
+       struct ad9910_state *st = iio_priv(indio_dev);
+       bool auto_en, pinctrl_en;
+       u32 rate, step;
+       u64 roc64;
+
+       guard(mutex)(&st->lock);
+
+       rate = FIELD_GET(AD9910_ASF_RAMP_RATE_MSK, 
st->reg[AD9910_REG_ASF].val32);
+       if (!rate)
+               return -ERANGE;
+
+       switch (private) {
+       case AD9910_ROC:
+               auto_en = FIELD_GET(AD9910_CFR1_SELECT_AUTO_OSK_MSK,
+                                   st->reg[AD9910_REG_CFR1].val32);
+               pinctrl_en = FIELD_GET(AD9910_CFR1_OSK_MANUAL_EXT_CTL_MSK,
+                                      st->reg[AD9910_REG_CFR1].val32);
+               if (auto_en) {
+                       step = FIELD_GET(AD9910_ASF_STEP_SIZE_MSK,
+                                        st->reg[AD9910_REG_ASF].val32);
+                       step = ad9910_osk_raw_step[step + 1];
+               } else if (pinctrl_en) {
+                       step = 
ad9910_osk_raw_step[ARRAY_SIZE(ad9910_osk_raw_step) - 1];
+               } else {
+                       step = ad9910_osk_raw_step[0];
+               }
+
+               roc64 = div_u64((u64)step * st->data.sysclk_freq_hz, 4 * rate);
+               return sysfs_emit(buf, "%llu\n", roc64);
+       case AD9910_ROC_AVAIL: {
+               ssize_t len = 0;
+
+               for (unsigned int i = 0; i < ARRAY_SIZE(ad9910_osk_raw_step); 
i++) {
+                       roc64 = div_u64((u64)ad9910_osk_raw_step[i] * 
st->data.sysclk_freq_hz,
+                                       4 * rate);
+                       len += sysfs_emit_at(buf, len, "%llu ", roc64);
+               }
+
+               buf[len - 1] = '\n'; /* replace last space with a newline */
+               return len;
+       }
+       default:
+               return -EINVAL;
+       }
+}
+
+static ssize_t ad9910_osk_attrs_write(struct iio_dev *indio_dev,
+                                     uintptr_t private,
+                                     const struct iio_chan_spec *chan,
+                                     const char *buf, size_t len)
+{
+       struct ad9910_state *st = iio_priv(indio_dev);
+       u32 idx, reg_val, rate;
+       u64 step;
+       int ret;
+
+       ret = kstrtou64(buf, 10, &step);
+       if (ret)
+               return ret;
+
+       guard(mutex)(&st->lock);
+
+       rate = FIELD_GET(AD9910_ASF_RAMP_RATE_MSK, 
st->reg[AD9910_REG_ASF].val32);
+       if (!rate)
+               return -ERANGE;
+
+       switch (private) {
+       case AD9910_ROC:
+               step = ad9910_rational_scale(step, 4 * rate,
+                                            st->data.sysclk_freq_hz);
+               step = min(step, AD9910_ASF_MAX);
+               idx = find_closest(step, ad9910_osk_raw_step,
+                                  ARRAY_SIZE(ad9910_osk_raw_step));
+               if (idx == ARRAY_SIZE(ad9910_osk_raw_step) - 1) {
+                       /* manual mode: pin-controlled */
+                       reg_val = AD9910_CFR1_OSK_MANUAL_EXT_CTL_MSK;
+               } else if (idx == 0) {
+                       /* manual mode, NOT pin-controlled */
+                       reg_val = 0;
+               } else {
+                       /* auto mode: pin-controlled */
+                       reg_val = FIELD_PREP(AD9910_ASF_STEP_SIZE_MSK, idx - 1);
+                       ret = ad9910_reg32_update(st, AD9910_REG_ASF,
+                                                 AD9910_ASF_STEP_SIZE_MSK,
+                                                 reg_val, false);
+                       if (ret)
+                               return ret;
+
+                       reg_val = AD9910_CFR1_SELECT_AUTO_OSK_MSK;
+               }
+
+               ret = ad9910_reg32_update(st, AD9910_REG_CFR1,
+                                         AD9910_CFR1_SELECT_AUTO_OSK_MSK |
+                                         AD9910_CFR1_OSK_MANUAL_EXT_CTL_MSK,
+                                         reg_val, true);
+               if (ret)
+                       return ret;
+               break;
+       default:
+               return -EINVAL;
+       }
+
+       return len;
+}
+
 static const struct iio_chan_spec_ext_info ad9910_phy_ext_info[] = {
        {
                .name = "powerdown",
@@ -785,6 +908,23 @@ static const struct iio_chan_spec_ext_info 
ad9910_drg_ramp_ext_info[] = {
        { }
 };
 
+static const struct iio_chan_spec_ext_info ad9910_osk_ext_info[] = {
+       {
+               .name = "raw_roc",
+               .read = ad9910_osk_attrs_read,
+               .write = ad9910_osk_attrs_write,
+               .private = AD9910_ROC,
+               .shared = IIO_SEPARATE,
+       },
+       {
+               .name = "raw_roc_available",
+               .read = ad9910_osk_attrs_read,
+               .private = AD9910_ROC_AVAIL,
+               .shared = IIO_SEPARATE,
+       },
+       { }
+};
+
 #define AD9910_PROFILE_CHAN(idx) {                             \
        .type = IIO_ALTCURRENT,                                 \
        .indexed = 1,                                           \
@@ -928,6 +1068,18 @@ static const struct iio_chan_spec ad9910_channels[] = {
                                      BIT(IIO_CHAN_INFO_SAMP_FREQ),
                .parent = &ad9910_channels[AD9910_CHAN_IDX_PHY],
        },
+       [AD9910_CHAN_IDX_OSK] = {
+               .type = IIO_ALTCURRENT,
+               .indexed = 1,
+               .output = 1,
+               .channel = AD9910_CHANNEL_OSK,
+               .address = AD9910_CHAN_IDX_OSK,
+               .info_mask_separate = BIT(IIO_CHAN_INFO_ENABLE) |
+                                     BIT(IIO_CHAN_INFO_RAW) |
+                                     BIT(IIO_CHAN_INFO_SAMP_FREQ),
+               .ext_info = ad9910_osk_ext_info,
+               .parent = &ad9910_channels[AD9910_CHAN_IDX_PHY],
+       },
 };
 
 static int ad9910_read_raw(struct iio_dev *indio_dev,
@@ -964,6 +1116,10 @@ static int ad9910_read_raw(struct iio_dev *indio_dev,
                        *val = FIELD_GET(AD9910_CFR1_RAM_ENABLE_MSK,
                                         st->reg[AD9910_REG_CFR1].val32);
                        break;
+               case AD9910_CHANNEL_OSK:
+                       *val = FIELD_GET(AD9910_CFR1_OSK_ENABLE_MSK,
+                                        st->reg[AD9910_REG_CFR1].val32);
+                       break;
                default:
                        return -EINVAL;
                }
@@ -1019,6 +1175,10 @@ static int ad9910_read_raw(struct iio_dev *indio_dev,
                                          st->reg[AD9910_REG_DRG_LIMIT].val64);
                        iio_val_s64_decompose(tmp64, val, val2);
                        return IIO_VAL_INT_64;
+               case AD9910_CHANNEL_OSK:
+                       *val = FIELD_GET(AD9910_ASF_SCALE_FACTOR_MSK,
+                                        st->reg[AD9910_REG_ASF].val32);
+                       return IIO_VAL_INT;
                default:
                        return -EINVAL;
                }
@@ -1039,6 +1199,10 @@ static int ad9910_read_raw(struct iio_dev *indio_dev,
                        tmp32 = FIELD_GET(AD9910_PROFILE_RAM_STEP_RATE_MSK,
                                          ad9910_ram_profile_val(st));
                        break;
+               case AD9910_CHANNEL_OSK:
+                       tmp32 = FIELD_GET(AD9910_ASF_RAMP_RATE_MSK,
+                                         st->reg[AD9910_REG_ASF].val32);
+                       break;
                default:
                        return -EINVAL;
                }
@@ -1202,6 +1366,11 @@ static int ad9910_write_raw(struct iio_dev *indio_dev,
                        return ad9910_reg32_update(st, AD9910_REG_CFR1,
                                                   AD9910_CFR1_RAM_ENABLE_MSK,
                                                   tmp32, true);
+               case AD9910_CHANNEL_OSK:
+                       tmp32 = FIELD_PREP(AD9910_CFR1_OSK_ENABLE_MSK, !!val);
+                       return ad9910_reg32_update(st, AD9910_REG_CFR1,
+                                                  AD9910_CFR1_OSK_ENABLE_MSK,
+                                                  tmp32, true);
                default:
                        return -EINVAL;
                }
@@ -1292,6 +1461,15 @@ static int ad9910_write_raw(struct iio_dev *indio_dev,
                        return ad9910_reg64_update(st, AD9910_REG_DRG_LIMIT,
                                                   AD9910_DRG_LIMIT_LOWER_MSK,
                                                   tmp64, true);
+               case AD9910_CHANNEL_OSK:
+                       if (val < 0)
+                               return -EINVAL;
+
+                       tmp32 = min_t(u32, val, AD9910_ASF_MAX);
+                       tmp32 = FIELD_PREP(AD9910_ASF_SCALE_FACTOR_MSK, tmp32);
+                       return ad9910_reg32_update(st, AD9910_REG_ASF,
+                                                  AD9910_ASF_SCALE_FACTOR_MSK,
+                                                  tmp32, true);
                default:
                        return -EINVAL;
                }
@@ -1331,6 +1509,11 @@ static int ad9910_write_raw(struct iio_dev *indio_dev,
                        return ad9910_reg64_update(st, 
AD9910_REG_PROFILE(st->profile),
                                                   
AD9910_PROFILE_RAM_STEP_RATE_MSK,
                                                   tmp64, true);
+               case AD9910_CHANNEL_OSK:
+                       return ad9910_reg32_update(st, AD9910_REG_ASF,
+                                                  AD9910_ASF_RAMP_RATE_MSK,
+                                                  
FIELD_PREP(AD9910_ASF_RAMP_RATE_MSK, tmp32),
+                                                  true);
                default:
                        return -EINVAL;
                }
@@ -1410,6 +1593,7 @@ static int ad9910_write_raw_get_fmt(struct iio_dev 
*indio_dev,
        case IIO_CHAN_INFO_RAW:
                switch (chan->channel) {
                case AD9910_CHANNEL_PROFILE_0 ... AD9910_CHANNEL_PROFILE_7:
+               case AD9910_CHANNEL_OSK:
                        return IIO_VAL_INT;
                case AD9910_CHANNEL_DRG_RAMP_UP:
                case AD9910_CHANNEL_DRG_RAMP_DOWN:
@@ -1497,6 +1681,7 @@ static const char * const ad9910_channel_str[] = {
        [AD9910_CHAN_IDX_DRG_AMP_RAMP_UP] = "drg_rising",
        [AD9910_CHAN_IDX_DRG_AMP_RAMP_DOWN] = "drg_falling",
        [AD9910_CHAN_IDX_RAM] = "ram",
+       [AD9910_CHAN_IDX_OSK] = "osk",
 };
 
 static int ad9910_read_label(struct iio_dev *indio_dev,
@@ -1752,6 +1937,12 @@ static int ad9910_setup(struct device *dev, struct 
ad9910_state *st,
                return ret;
 
        /* configure step rate with default values */
+       ret = ad9910_reg32_write(st, AD9910_REG_ASF,
+                                FIELD_PREP(AD9910_ASF_RAMP_RATE_MSK, 1),
+                                false);
+       if (ret)
+               return ret;
+
        ret = ad9910_reg32_write(st, AD9910_REG_DRG_RATE,
                                 FIELD_PREP(AD9910_DRG_RATE_DEC_MSK, 1) |
                                 FIELD_PREP(AD9910_DRG_RATE_INC_MSK, 1),

-- 
2.43.0



Reply via email to