pmic: Add support for setting transmission length in uclass private data

[u-boot] / drivers / power / pmic / pmic-uclass.c

diff --git a/drivers/power/pmic/pmic-uclass.c b/drivers/power/pmic/pmic-uclass.c
index 5f0f6ff93e14449ce9c8064819c35c5c7b064728..d7e559a61560dc7d0a422661c67dac2fa0e76ccb 100644 (file)
--- a/drivers/power/pmic/pmic-uclass.c
+++ b/drivers/power/pmic/pmic-uclass.c
@@ -165,7 +165,17 @@ int pmic_clrsetbits(struct udevice *dev, uint reg, uint clr, uint set)
        return pmic_reg_write(dev, reg, byte);
 }
 
+static int pmic_pre_probe(struct udevice *dev)
+{
+       struct uc_pmic_priv *pmic_priv = dev_get_uclass_priv(dev);
+
+       pmic_priv->trans_len = 1;
+       return 0;
+}
+
 UCLASS_DRIVER(pmic) = {
        .id             = UCLASS_PMIC,
        .name           = "pmic",
+       .pre_probe      = pmic_pre_probe,
+       .per_device_auto_alloc_size = sizeof(struct uc_pmic_priv),
 };