Move the coupled regulators voltage balancing code to the separate
function and allow to call it from the custom regulator couplers.

Signed-off-by: Marek Szyprowski <m.szyprow...@samsung.com>
---
 drivers/regulator/core.c          | 49 ++++++++++++++++++-------------
 include/linux/regulator/coupler.h |  8 +++++
 2 files changed, 37 insertions(+), 20 deletions(-)

diff --git a/drivers/regulator/core.c b/drivers/regulator/core.c
index 941783a14b45..370c655ad8f6 100644
--- a/drivers/regulator/core.c
+++ b/drivers/regulator/core.c
@@ -3642,36 +3642,19 @@ static int regulator_get_optimal_voltage(struct 
regulator_dev *rdev,
        return done;
 }
 
-static int regulator_balance_voltage(struct regulator_dev *rdev,
-                                    suspend_state_t state)
+int regulator_do_balance_voltage(struct regulator_dev *rdev,
+                                suspend_state_t state, bool skip_coupled)
 {
        struct regulator_dev **c_rdevs;
        struct regulator_dev *best_rdev;
        struct coupling_desc *c_desc = &rdev->coupling_desc;
-       struct regulator_coupler *coupler = c_desc->coupler;
        int i, ret, n_coupled, best_min_uV, best_max_uV, best_c_rdev;
        unsigned int delta, best_delta;
        unsigned long c_rdev_done = 0;
        bool best_c_rdev_done;
 
        c_rdevs = c_desc->coupled_rdevs;
-       n_coupled = c_desc->n_coupled;
-
-       /*
-        * If system is in a state other than PM_SUSPEND_ON, don't check
-        * other coupled regulators.
-        */
-       if (state != PM_SUSPEND_ON)
-               n_coupled = 1;
-
-       if (c_desc->n_resolved < n_coupled) {
-               rdev_err(rdev, "Not all coupled regulators registered\n");
-               return -EPERM;
-       }
-
-       /* Invoke custom balancer for customized couplers */
-       if (coupler && coupler->balance_voltage)
-               return coupler->balance_voltage(coupler, rdev, state);
+       n_coupled = skip_coupled ? 1 : c_desc->n_coupled;
 
        /*
         * Find the best possible voltage change on each loop. Leave the loop
@@ -3742,6 +3725,32 @@ static int regulator_balance_voltage(struct 
regulator_dev *rdev,
        return ret;
 }
 
+static int regulator_balance_voltage(struct regulator_dev *rdev,
+                                    suspend_state_t state)
+{
+       struct coupling_desc *c_desc = &rdev->coupling_desc;
+       struct regulator_coupler *coupler = c_desc->coupler;
+       bool skip_coupled = false;
+
+       /*
+        * If system is in a state other than PM_SUSPEND_ON, don't check
+        * other coupled regulators.
+        */
+       if (state != PM_SUSPEND_ON)
+               skip_coupled = true;
+
+       if (c_desc->n_resolved < c_desc->n_coupled) {
+               rdev_err(rdev, "Not all coupled regulators registered\n");
+               return -EPERM;
+       }
+
+       /* Invoke custom balancer for customized couplers */
+       if (coupler && coupler->balance_voltage)
+               return coupler->balance_voltage(coupler, rdev, state);
+
+       return regulator_do_balance_voltage(rdev, state, skip_coupled);
+}
+
 /**
  * regulator_set_voltage - set regulator output voltage
  * @regulator: regulator source
diff --git a/include/linux/regulator/coupler.h 
b/include/linux/regulator/coupler.h
index 0212d6255e4e..5f86824bd117 100644
--- a/include/linux/regulator/coupler.h
+++ b/include/linux/regulator/coupler.h
@@ -62,6 +62,8 @@ int regulator_get_voltage_rdev(struct regulator_dev *rdev);
 int regulator_set_voltage_rdev(struct regulator_dev *rdev,
                               int min_uV, int max_uV,
                               suspend_state_t state);
+int regulator_do_balance_voltage(struct regulator_dev *rdev,
+                                suspend_state_t state, bool skip_coupled);
 #else
 static inline int regulator_coupler_register(struct regulator_coupler *coupler)
 {
@@ -92,6 +94,12 @@ static inline int regulator_set_voltage_rdev(struct 
regulator_dev *rdev,
 {
        return -EINVAL;
 }
+static inline int regulator_do_balance_voltage(struct regulator_dev *rdev,
+                                              suspend_state_t state,
+                                              bool skip_coupled)
+{
+       return -EINVAL;
+}
 #endif
 
 #endif
-- 
2.17.1

Reply via email to