gpio: dm: Support manual relocation for gpio

Relocate gpio ops as was done by:
"dm: Add support for all targets which requires MANUAL_RELOC"
(sha1: 484fdf5ba058b07be5ca82763aa2b72063540ef3)

Signed-off-by: Michal Simek <michal.simek@xilinx.com>
Reviewed-by: Simon Glass <sjg@chromium.org>

diff --git a/drivers/gpio/gpio-uclass.c b/drivers/gpio/gpio-uclass.c
index e9d08e2ebc25bec53c4b4a10c041c5849d141f0a..da5e9ba6e524f584ff472e34167e5a7861f1c8c6 100644 (file)
--- a/drivers/gpio/gpio-uclass.c
+++ b/drivers/gpio/gpio-uclass.c
@@ -854,11 +854,46 @@ static int gpio_pre_remove(struct udevice *dev)
        return gpio_renumber(dev);
 }
 
+static int gpio_post_bind(struct udevice *dev)
+{
+#if defined(CONFIG_NEEDS_MANUAL_RELOC)
+       struct dm_gpio_ops *ops = (struct dm_gpio_ops *)device_get_ops(dev);
+       static int reloc_done;
+
+       if (!reloc_done) {
+               if (ops->request)
+                       ops->request += gd->reloc_off;
+               if (ops->free)
+                       ops->free += gd->reloc_off;
+               if (ops->direction_input)
+                       ops->direction_input += gd->reloc_off;
+               if (ops->direction_output)
+                       ops->direction_output += gd->reloc_off;
+               if (ops->get_value)
+                       ops->get_value += gd->reloc_off;
+               if (ops->set_value)
+                       ops->set_value += gd->reloc_off;
+               if (ops->get_open_drain)
+                       ops->get_open_drain += gd->reloc_off;
+               if (ops->set_open_drain)
+                       ops->set_open_drain += gd->reloc_off;
+               if (ops->get_function)
+                       ops->get_function += gd->reloc_off;
+               if (ops->xlate)
+                       ops->xlate += gd->reloc_off;
+
+               reloc_done++;
+       }
+#endif
+       return 0;
+}
+
 UCLASS_DRIVER(gpio) = {
        .id             = UCLASS_GPIO,
        .name           = "gpio",
        .flags          = DM_UC_FLAG_SEQ_ALIAS,
        .post_probe     = gpio_post_probe,
+       .post_bind      = gpio_post_bind,
        .pre_remove     = gpio_pre_remove,
        .per_device_auto_alloc_size = sizeof(struct gpio_dev_priv),
 };