regmap: Supply ranges to the sync operations

In order to allow us to support partial sync operations add minimum and
maximum register arguments to the sync operation and update the rbtree
and lzo caches to use this new information. The LZO implementation is
obviously not good, we could exit the iteration earlier, but there may
be room for more wide reaching optimisation there.

Signed-off-by: Mark Brown <broonie@opensource.wolfsonmicro.com>

4 files changed, 32 insertions, 7 deletions

diff --git a/drivers/base/regmap/internal.h b/drivers/base/regmap/internal.h
index d141b80479b5..9c1d62e3e15c 100644
--- a/drivers/base/regmap/internal.h
+++ b/drivers/base/regmap/internal.h
@@ -87,7 +87,7 @@ struct regcache_ops {
         int (*exit)(struct regmap *map);
         int (*read)(struct regmap *map, unsigned int reg, unsigned int *value);
         int (*write)(struct regmap *map, unsigned int reg, unsigned int value);
-        int (*sync)(struct regmap *map);
+        int (*sync)(struct regmap *map, unsigned int min, unsigned int max);
 };
 
 bool regmap_writeable(struct regmap *map, unsigned int reg);
diff --git a/drivers/base/regmap/regcache-lzo.c b/drivers/base/regmap/regcache-lzo.c
index b7d16143edeb..5e964e9b2bab 100644
--- a/drivers/base/regmap/regcache-lzo.c
+++ b/drivers/base/regmap/regcache-lzo.c
@@ -331,7 +331,8 @@ out:
         return ret;
 }
 
-static int regcache_lzo_sync(struct regmap *map)
+static int regcache_lzo_sync(struct regmap *map, unsigned int min,
+                             unsigned int max)
 {
         struct regcache_lzo_ctx **lzo_blocks;
         unsigned int val;
@@ -339,7 +340,12 @@ static int regcache_lzo_sync(struct regmap *map)
         int ret;
 
         lzo_blocks = map->cache;
-        for_each_set_bit(i, lzo_blocks[0]->sync_bmp, lzo_blocks[0]->sync_bmp_nbits) {
+        i = min;
+        for_each_set_bit_from(i, lzo_blocks[0]->sync_bmp,
+                              lzo_blocks[0]->sync_bmp_nbits) {
+                if (i > max)
+                        continue;
+
                 ret = regcache_read(map, i, &val);
                 if (ret)
                         return ret;
diff --git a/drivers/base/regmap/regcache-rbtree.c b/drivers/base/regmap/regcache-rbtree.c
index 32620c4f1683..bae183c6bcb1 100644
--- a/drivers/base/regmap/regcache-rbtree.c
+++ b/drivers/base/regmap/regcache-rbtree.c
@@ -357,7 +357,8 @@ static int regcache_rbtree_write(struct regmap *map, unsigned int reg,
         return 0;
 }
 
-static int regcache_rbtree_sync(struct regmap *map)
+static int regcache_rbtree_sync(struct regmap *map, unsigned int min,
+                                unsigned int max)
 {
         struct regcache_rbtree_ctx *rbtree_ctx;
         struct rb_node *node;
@@ -365,12 +366,30 @@ static int regcache_rbtree_sync(struct regmap *map)
         unsigned int regtmp;
         unsigned int val;
         int ret;
-        int i;
+        int i, base, end;
 
         rbtree_ctx = map->cache;
         for (node = rb_first(&rbtree_ctx->root); node; node = rb_next(node)) {
                 rbnode = rb_entry(node, struct regcache_rbtree_node, node);
-                for (i = 0; i < rbnode->blklen; i++) {
+
+                if (rbnode->base_reg < min)
+                        continue;
+                if (rbnode->base_reg > max)
+                        break;
+                if (rbnode->base_reg + rbnode->blklen < min)
+                        continue;
+
+                if (min < rbnode->base_reg + rbnode->blklen)
+                        base = min - rbnode->base_reg;
+                else
+                        base = 0;
+
+                if (max < rbnode->base_reg + rbnode->blklen)
+                        end = rbnode->base_reg + rbnode->blklen - max;
+                else
+                        end = rbnode->blklen;
+
+                for (i = base; i < end; i++) {
                         regtmp = rbnode->base_reg + i;
                         val = regcache_rbtree_get_register(rbnode, i,
                                                            map->cache_word_size);
diff --git a/drivers/base/regmap/regcache.c b/drivers/base/regmap/regcache.c
index 46c42d3a4655..aec5a7486a29 100644
--- a/drivers/base/regmap/regcache.c
+++ b/drivers/base/regmap/regcache.c
@@ -283,7 +283,7 @@ int regcache_sync(struct regmap *map)
         }
         map->cache_bypass = 0;
 
-        ret = map->cache_ops->sync(map);
+        ret = map->cache_ops->sync(map, 0, map->max_register);
 
         if (ret == 0)
                 map->cache_dirty = false;