// SPDX-License-Identifier: GPL-2.0-only

  /*
   * Maxim MAX77620 MFD Driver
   *
   * Copyright (C) 2016 NVIDIA CORPORATION. All rights reserved.
   *
   * Author:
   *	Laxman Dewangan <ldewangan@nvidia.com>
   *	Chaitanya Bandi <bandik@nvidia.com>
   *	Mallikarjun Kasoju <mkasoju@nvidia.com>

   */
  
  /****************** Teminology used in driver ********************
   * Here are some terminology used from datasheet for quick reference:
   * Flexible Power Sequence (FPS):
   * The Flexible Power Sequencer (FPS) allows each regulator to power up under
   * hardware or software control. Additionally, each regulator can power on
   * independently or among a group of other regulators with an adjustable
   * power-up and power-down delays (sequencing). GPIO1, GPIO2, and GPIO3 can
   * be programmed to be part of a sequence allowing external regulators to be
   * sequenced along with internal regulators. 32KHz clock can be programmed to
   * be part of a sequence.
   * There is 3 FPS confguration registers and all resources are configured to
   * any of these FPS or no FPS.
   */
  
  #include <linux/i2c.h>
  #include <linux/interrupt.h>
  #include <linux/mfd/core.h>
  #include <linux/mfd/max77620.h>

  #include <linux/init.h>

  #include <linux/of.h>
  #include <linux/of_device.h>
  #include <linux/regmap.h>
  #include <linux/slab.h>

  static struct max77620_chip *max77620_scratch;

  static const struct resource gpio_resources[] = {

  	DEFINE_RES_IRQ(MAX77620_IRQ_TOP_GPIO),
  };

  static const struct resource power_resources[] = {

  	DEFINE_RES_IRQ(MAX77620_IRQ_LBT_MBATLOW),
  };

  static const struct resource rtc_resources[] = {

  	DEFINE_RES_IRQ(MAX77620_IRQ_TOP_RTC),
  };

  static const struct resource thermal_resources[] = {

  	DEFINE_RES_IRQ(MAX77620_IRQ_LBT_TJALRM1),
  	DEFINE_RES_IRQ(MAX77620_IRQ_LBT_TJALRM2),
  };
  
  static const struct regmap_irq max77620_top_irqs[] = {
  	REGMAP_IRQ_REG(MAX77620_IRQ_TOP_GLBL, 0, MAX77620_IRQ_TOP_GLBL_MASK),
  	REGMAP_IRQ_REG(MAX77620_IRQ_TOP_SD, 0, MAX77620_IRQ_TOP_SD_MASK),
  	REGMAP_IRQ_REG(MAX77620_IRQ_TOP_LDO, 0, MAX77620_IRQ_TOP_LDO_MASK),
  	REGMAP_IRQ_REG(MAX77620_IRQ_TOP_GPIO, 0, MAX77620_IRQ_TOP_GPIO_MASK),
  	REGMAP_IRQ_REG(MAX77620_IRQ_TOP_RTC, 0, MAX77620_IRQ_TOP_RTC_MASK),
  	REGMAP_IRQ_REG(MAX77620_IRQ_TOP_32K, 0, MAX77620_IRQ_TOP_32K_MASK),
  	REGMAP_IRQ_REG(MAX77620_IRQ_TOP_ONOFF, 0, MAX77620_IRQ_TOP_ONOFF_MASK),
  	REGMAP_IRQ_REG(MAX77620_IRQ_LBT_MBATLOW, 1, MAX77620_IRQ_LBM_MASK),
  	REGMAP_IRQ_REG(MAX77620_IRQ_LBT_TJALRM1, 1, MAX77620_IRQ_TJALRM1_MASK),
  	REGMAP_IRQ_REG(MAX77620_IRQ_LBT_TJALRM2, 1, MAX77620_IRQ_TJALRM2_MASK),
  };
  
  static const struct mfd_cell max77620_children[] = {
  	{ .name = "max77620-pinctrl", },
  	{ .name = "max77620-clock", },
  	{ .name = "max77620-pmic", },
  	{ .name = "max77620-watchdog", },
  	{
  		.name = "max77620-gpio",
  		.resources = gpio_resources,
  		.num_resources = ARRAY_SIZE(gpio_resources),
  	}, {
  		.name = "max77620-rtc",
  		.resources = rtc_resources,
  		.num_resources = ARRAY_SIZE(rtc_resources),
  	}, {
  		.name = "max77620-power",
  		.resources = power_resources,
  		.num_resources = ARRAY_SIZE(power_resources),
  	}, {
  		.name = "max77620-thermal",
  		.resources = thermal_resources,
  		.num_resources = ARRAY_SIZE(thermal_resources),
  	},
  };
  
  static const struct mfd_cell max20024_children[] = {
  	{ .name = "max20024-pinctrl", },
  	{ .name = "max77620-clock", },
  	{ .name = "max20024-pmic", },
  	{ .name = "max77620-watchdog", },
  	{
  		.name = "max77620-gpio",
  		.resources = gpio_resources,
  		.num_resources = ARRAY_SIZE(gpio_resources),
  	}, {
  		.name = "max77620-rtc",
  		.resources = rtc_resources,
  		.num_resources = ARRAY_SIZE(rtc_resources),
  	}, {
  		.name = "max20024-power",
  		.resources = power_resources,
  		.num_resources = ARRAY_SIZE(power_resources),
  	},
  };

  static const struct mfd_cell max77663_children[] = {
  	{ .name = "max77620-pinctrl", },
  	{ .name = "max77620-clock", },
  	{ .name = "max77663-pmic", },
  	{ .name = "max77620-watchdog", },
  	{
  		.name = "max77620-gpio",
  		.resources = gpio_resources,
  		.num_resources = ARRAY_SIZE(gpio_resources),
  	}, {
  		.name = "max77620-rtc",
  		.resources = rtc_resources,
  		.num_resources = ARRAY_SIZE(rtc_resources),
  	}, {
  		.name = "max77663-power",
  		.resources = power_resources,
  		.num_resources = ARRAY_SIZE(power_resources),
  	},
  };

  static const struct regmap_range max77620_readable_ranges[] = {
  	regmap_reg_range(MAX77620_REG_CNFGGLBL1, MAX77620_REG_DVSSD4),
  };
  
  static const struct regmap_access_table max77620_readable_table = {
  	.yes_ranges = max77620_readable_ranges,
  	.n_yes_ranges = ARRAY_SIZE(max77620_readable_ranges),
  };
  
  static const struct regmap_range max20024_readable_ranges[] = {
  	regmap_reg_range(MAX77620_REG_CNFGGLBL1, MAX77620_REG_DVSSD4),
  	regmap_reg_range(MAX20024_REG_MAX_ADD, MAX20024_REG_MAX_ADD),
  };
  
  static const struct regmap_access_table max20024_readable_table = {
  	.yes_ranges = max20024_readable_ranges,
  	.n_yes_ranges = ARRAY_SIZE(max20024_readable_ranges),
  };
  
  static const struct regmap_range max77620_writable_ranges[] = {
  	regmap_reg_range(MAX77620_REG_CNFGGLBL1, MAX77620_REG_DVSSD4),
  };
  
  static const struct regmap_access_table max77620_writable_table = {
  	.yes_ranges = max77620_writable_ranges,
  	.n_yes_ranges = ARRAY_SIZE(max77620_writable_ranges),
  };
  
  static const struct regmap_range max77620_cacheable_ranges[] = {
  	regmap_reg_range(MAX77620_REG_SD0_CFG, MAX77620_REG_LDO_CFG3),
  	regmap_reg_range(MAX77620_REG_FPS_CFG0, MAX77620_REG_FPS_SD3),
  };
  
  static const struct regmap_access_table max77620_volatile_table = {
  	.no_ranges = max77620_cacheable_ranges,
  	.n_no_ranges = ARRAY_SIZE(max77620_cacheable_ranges),
  };
  
  static const struct regmap_config max77620_regmap_config = {
  	.name = "power-slave",
  	.reg_bits = 8,
  	.val_bits = 8,
  	.max_register = MAX77620_REG_DVSSD4 + 1,
  	.cache_type = REGCACHE_RBTREE,
  	.rd_table = &max77620_readable_table,
  	.wr_table = &max77620_writable_table,
  	.volatile_table = &max77620_volatile_table,
  };
  
  static const struct regmap_config max20024_regmap_config = {
  	.name = "power-slave",
  	.reg_bits = 8,
  	.val_bits = 8,
  	.max_register = MAX20024_REG_MAX_ADD + 1,
  	.cache_type = REGCACHE_RBTREE,
  	.rd_table = &max20024_readable_table,
  	.wr_table = &max77620_writable_table,
  	.volatile_table = &max77620_volatile_table,
  };

  static const struct regmap_range max77663_readable_ranges[] = {
  	regmap_reg_range(MAX77620_REG_CNFGGLBL1, MAX77620_REG_CID5),
  };
  
  static const struct regmap_access_table max77663_readable_table = {
  	.yes_ranges = max77663_readable_ranges,
  	.n_yes_ranges = ARRAY_SIZE(max77663_readable_ranges),
  };
  
  static const struct regmap_range max77663_writable_ranges[] = {
  	regmap_reg_range(MAX77620_REG_CNFGGLBL1, MAX77620_REG_CID5),
  };
  
  static const struct regmap_access_table max77663_writable_table = {
  	.yes_ranges = max77663_writable_ranges,
  	.n_yes_ranges = ARRAY_SIZE(max77663_writable_ranges),
  };
  
  static const struct regmap_config max77663_regmap_config = {
  	.name = "power-slave",
  	.reg_bits = 8,
  	.val_bits = 8,
  	.max_register = MAX77620_REG_CID5 + 1,
  	.cache_type = REGCACHE_RBTREE,
  	.rd_table = &max77663_readable_table,
  	.wr_table = &max77663_writable_table,
  	.volatile_table = &max77620_volatile_table,
  };

  /*
   * MAX77620 and MAX20024 has the following steps of the interrupt handling
   * for TOP interrupts:
   * 1. When interrupt occurs from PMIC, mask the PMIC interrupt by setting GLBLM.
   * 2. Read IRQTOP and service the interrupt.
   * 3. Once all interrupts has been checked and serviced, the interrupt service
   *    routine un-masks the hardware interrupt line by clearing GLBLM.
   */
  static int max77620_irq_global_mask(void *irq_drv_data)
  {
  	struct max77620_chip *chip = irq_drv_data;
  	int ret;
  
  	ret = regmap_update_bits(chip->rmap, MAX77620_REG_INTENLBT,
  				 MAX77620_GLBLM_MASK, MAX77620_GLBLM_MASK);
  	if (ret < 0)
  		dev_err(chip->dev, "Failed to set GLBLM: %d
  ", ret);
  
  	return ret;
  }
  
  static int max77620_irq_global_unmask(void *irq_drv_data)
  {
  	struct max77620_chip *chip = irq_drv_data;
  	int ret;
  
  	ret = regmap_update_bits(chip->rmap, MAX77620_REG_INTENLBT,
  				 MAX77620_GLBLM_MASK, 0);
  	if (ret < 0)
  		dev_err(chip->dev, "Failed to reset GLBLM: %d
  ", ret);
  
  	return ret;
  }
  
  static struct regmap_irq_chip max77620_top_irq_chip = {
  	.name = "max77620-top",
  	.irqs = max77620_top_irqs,
  	.num_irqs = ARRAY_SIZE(max77620_top_irqs),
  	.num_regs = 2,
  	.status_base = MAX77620_REG_IRQTOP,
  	.mask_base = MAX77620_REG_IRQTOPM,
  	.handle_pre_irq = max77620_irq_global_mask,
  	.handle_post_irq = max77620_irq_global_unmask,
  };

  /* max77620_get_fps_period_reg_value:  Get FPS bit field value from
   *				       requested periods.
   * MAX77620 supports the FPS period of 40, 80, 160, 320, 540, 1280, 2560
   * and 5120 microseconds. MAX20024 supports the FPS period of 20, 40, 80,
   * 160, 320, 540, 1280 and 2560 microseconds.
   * The FPS register has 3 bits field to set the FPS period as
   * bits		max77620		max20024
   * 000		40			20
   * 001		80			40
   * :::
  */
  static int max77620_get_fps_period_reg_value(struct max77620_chip *chip,
  					     int tperiod)
  {
  	int fps_min_period;
  	int i;
  
  	switch (chip->chip_id) {
  	case MAX20024:
  		fps_min_period = MAX20024_FPS_PERIOD_MIN_US;
  		break;
  	case MAX77620:
  		fps_min_period = MAX77620_FPS_PERIOD_MIN_US;

  		break;

  	case MAX77663:
  		fps_min_period = MAX20024_FPS_PERIOD_MIN_US;
  		break;

  	default:
  		return -EINVAL;
  	}
  
  	for (i = 0; i < 7; i++) {
  		if (fps_min_period >= tperiod)
  			return i;
  		fps_min_period *= 2;
  	}
  
  	return i;
  }
  
  /* max77620_config_fps: Configure FPS configuration registers
   *			based on platform specific information.
   */
  static int max77620_config_fps(struct max77620_chip *chip,
  			       struct device_node *fps_np)
  {
  	struct device *dev = chip->dev;
  	unsigned int mask = 0, config = 0;
  	u32 fps_max_period;
  	u32 param_val;
  	int tperiod, fps_id;
  	int ret;
  	char fps_name[10];
  
  	switch (chip->chip_id) {
  	case MAX20024:
  		fps_max_period = MAX20024_FPS_PERIOD_MAX_US;
  		break;
  	case MAX77620:
  		fps_max_period = MAX77620_FPS_PERIOD_MAX_US;

  		break;

  	case MAX77663:
  		fps_max_period = MAX20024_FPS_PERIOD_MAX_US;
  		break;

  	default:
  		return -EINVAL;
  	}
  
  	for (fps_id = 0; fps_id < MAX77620_FPS_COUNT; fps_id++) {
  		sprintf(fps_name, "fps%d", fps_id);

  		if (of_node_name_eq(fps_np, fps_name))

  			break;
  	}
  
  	if (fps_id == MAX77620_FPS_COUNT) {

  		dev_err(dev, "FPS node name %pOFn is not valid
  ", fps_np);

  		return -EINVAL;
  	}
  
  	ret = of_property_read_u32(fps_np, "maxim,shutdown-fps-time-period-us",
  				   &param_val);
  	if (!ret) {
  		mask |= MAX77620_FPS_TIME_PERIOD_MASK;
  		chip->shutdown_fps_period[fps_id] = min(param_val,
  							fps_max_period);
  		tperiod = max77620_get_fps_period_reg_value(chip,
  				chip->shutdown_fps_period[fps_id]);
  		config |= tperiod << MAX77620_FPS_TIME_PERIOD_SHIFT;
  	}
  
  	ret = of_property_read_u32(fps_np, "maxim,suspend-fps-time-period-us",
  				   &param_val);
  	if (!ret)
  		chip->suspend_fps_period[fps_id] = min(param_val,
  						       fps_max_period);
  
  	ret = of_property_read_u32(fps_np, "maxim,fps-event-source",
  				   &param_val);
  	if (!ret) {
  		if (param_val > 2) {
  			dev_err(dev, "FPS%d event-source invalid
  ", fps_id);
  			return -EINVAL;
  		}
  		mask |= MAX77620_FPS_EN_SRC_MASK;
  		config |= param_val << MAX77620_FPS_EN_SRC_SHIFT;
  		if (param_val == 2) {
  			mask |= MAX77620_FPS_ENFPS_SW_MASK;
  			config |= MAX77620_FPS_ENFPS_SW;
  		}
  	}
  
  	if (!chip->sleep_enable && !chip->enable_global_lpm) {
  		ret = of_property_read_u32(fps_np,
  				"maxim,device-state-on-disabled-event",
  				&param_val);
  		if (!ret) {
  			if (param_val == 0)
  				chip->sleep_enable = true;
  			else if (param_val == 1)
  				chip->enable_global_lpm = true;
  		}
  	}
  
  	ret = regmap_update_bits(chip->rmap, MAX77620_REG_FPS_CFG0 + fps_id,
  				 mask, config);
  	if (ret < 0) {
  		dev_err(dev, "Failed to update FPS CFG: %d
  ", ret);
  		return ret;
  	}
  
  	return 0;
  }
  
  static int max77620_initialise_fps(struct max77620_chip *chip)
  {
  	struct device *dev = chip->dev;
  	struct device_node *fps_np, *fps_child;
  	u8 config;
  	int fps_id;
  	int ret;
  
  	for (fps_id = 0; fps_id < MAX77620_FPS_COUNT; fps_id++) {
  		chip->shutdown_fps_period[fps_id] = -1;
  		chip->suspend_fps_period[fps_id] = -1;
  	}
  
  	fps_np = of_get_child_by_name(dev->of_node, "fps");
  	if (!fps_np)
  		goto skip_fps;
  
  	for_each_child_of_node(fps_np, fps_child) {
  		ret = max77620_config_fps(chip, fps_child);

  		if (ret < 0) {
  			of_node_put(fps_child);

  			return ret;

  		}

  	}
  
  	config = chip->enable_global_lpm ? MAX77620_ONOFFCNFG2_SLP_LPM_MSK : 0;
  	ret = regmap_update_bits(chip->rmap, MAX77620_REG_ONOFFCNFG2,
  				 MAX77620_ONOFFCNFG2_SLP_LPM_MSK, config);
  	if (ret < 0) {
  		dev_err(dev, "Failed to update SLP_LPM: %d
  ", ret);
  		return ret;
  	}
  
  skip_fps:

  	if (chip->chip_id == MAX77663)
  		return 0;

  	/* Enable wake on EN0 pin */
  	ret = regmap_update_bits(chip->rmap, MAX77620_REG_ONOFFCNFG2,
  				 MAX77620_ONOFFCNFG2_WK_EN0,
  				 MAX77620_ONOFFCNFG2_WK_EN0);
  	if (ret < 0) {
  		dev_err(dev, "Failed to update WK_EN0: %d
  ", ret);
  		return ret;
  	}
  
  	/* For MAX20024, SLPEN will be POR reset if CLRSE is b11 */
  	if ((chip->chip_id == MAX20024) && chip->sleep_enable) {
  		config = MAX77620_ONOFFCNFG1_SLPEN | MAX20024_ONOFFCNFG1_CLRSE;
  		ret = regmap_update_bits(chip->rmap, MAX77620_REG_ONOFFCNFG1,
  					 config, config);
  		if (ret < 0) {
  			dev_err(dev, "Failed to update SLPEN: %d
  ", ret);
  			return ret;
  		}
  	}
  
  	return 0;
  }
  
  static int max77620_read_es_version(struct max77620_chip *chip)
  {
  	unsigned int val;
  	u8 cid_val[6];
  	int i;
  	int ret;
  
  	for (i = MAX77620_REG_CID0; i <= MAX77620_REG_CID5; i++) {
  		ret = regmap_read(chip->rmap, i, &val);
  		if (ret < 0) {
  			dev_err(chip->dev, "Failed to read CID: %d
  ", ret);
  			return ret;
  		}
  		dev_dbg(chip->dev, "CID%d: 0x%02x
  ",
  			i - MAX77620_REG_CID0, val);
  		cid_val[i - MAX77620_REG_CID0] = val;
  	}
  
  	/* CID4 is OTP Version  and CID5 is ES version */
  	dev_info(chip->dev, "PMIC Version OTP:0x%02X and ES:0x%X
  ",
  		 cid_val[4], MAX77620_CID5_DIDM(cid_val[5]));
  
  	return ret;
  }

  static void max77620_pm_power_off(void)
  {
  	struct max77620_chip *chip = max77620_scratch;
  
  	regmap_update_bits(chip->rmap, MAX77620_REG_ONOFFCNFG1,
  			   MAX77620_ONOFFCNFG1_SFT_RST,
  			   MAX77620_ONOFFCNFG1_SFT_RST);
  }

  static int max77620_probe(struct i2c_client *client,
  			  const struct i2c_device_id *id)
  {
  	const struct regmap_config *rmap_config;
  	struct max77620_chip *chip;
  	const struct mfd_cell *mfd_cells;
  	int n_mfd_cells;

  	bool pm_off;

  	int ret;
  
  	chip = devm_kzalloc(&client->dev, sizeof(*chip), GFP_KERNEL);
  	if (!chip)
  		return -ENOMEM;
  
  	i2c_set_clientdata(client, chip);
  	chip->dev = &client->dev;
  	chip->irq_base = -1;
  	chip->chip_irq = client->irq;
  	chip->chip_id = (enum max77620_chip_id)id->driver_data;
  
  	switch (chip->chip_id) {
  	case MAX77620:
  		mfd_cells = max77620_children;
  		n_mfd_cells = ARRAY_SIZE(max77620_children);
  		rmap_config = &max77620_regmap_config;
  		break;
  	case MAX20024:
  		mfd_cells = max20024_children;
  		n_mfd_cells = ARRAY_SIZE(max20024_children);
  		rmap_config = &max20024_regmap_config;
  		break;

  	case MAX77663:
  		mfd_cells = max77663_children;
  		n_mfd_cells = ARRAY_SIZE(max77663_children);
  		rmap_config = &max77663_regmap_config;
  		break;

  	default:
  		dev_err(chip->dev, "ChipID is invalid %d
  ", chip->chip_id);
  		return -EINVAL;
  	}
  
  	chip->rmap = devm_regmap_init_i2c(client, rmap_config);
  	if (IS_ERR(chip->rmap)) {
  		ret = PTR_ERR(chip->rmap);

  		dev_err(chip->dev, "Failed to initialise regmap: %d
  ", ret);

  		return ret;
  	}
  
  	ret = max77620_read_es_version(chip);
  	if (ret < 0)
  		return ret;

  	max77620_top_irq_chip.irq_drv_data = chip;

  	ret = devm_regmap_add_irq_chip(chip->dev, chip->rmap, client->irq,
  				       IRQF_ONESHOT | IRQF_SHARED,
  				       chip->irq_base, &max77620_top_irq_chip,
  				       &chip->top_irq_data);
  	if (ret < 0) {
  		dev_err(chip->dev, "Failed to add regmap irq: %d
  ", ret);
  		return ret;
  	}
  
  	ret = max77620_initialise_fps(chip);
  	if (ret < 0)
  		return ret;
  
  	ret =  devm_mfd_add_devices(chip->dev, PLATFORM_DEVID_NONE,
  				    mfd_cells, n_mfd_cells, NULL, 0,
  				    regmap_irq_get_domain(chip->top_irq_data));
  	if (ret < 0) {
  		dev_err(chip->dev, "Failed to add MFD children: %d
  ", ret);
  		return ret;
  	}

  	pm_off = of_device_is_system_power_controller(client->dev.of_node);
  	if (pm_off && !pm_power_off) {
  		max77620_scratch = chip;
  		pm_power_off = max77620_pm_power_off;
  	}

  	return 0;
  }
  
  #ifdef CONFIG_PM_SLEEP
  static int max77620_set_fps_period(struct max77620_chip *chip,
  				   int fps_id, int time_period)
  {
  	int period = max77620_get_fps_period_reg_value(chip, time_period);
  	int ret;
  
  	ret = regmap_update_bits(chip->rmap, MAX77620_REG_FPS_CFG0 + fps_id,
  				 MAX77620_FPS_TIME_PERIOD_MASK,
  				 period << MAX77620_FPS_TIME_PERIOD_SHIFT);
  	if (ret < 0) {
  		dev_err(chip->dev, "Failed to update FPS period: %d
  ", ret);
  		return ret;
  	}
  
  	return 0;
  }
  
  static int max77620_i2c_suspend(struct device *dev)
  {
  	struct max77620_chip *chip = dev_get_drvdata(dev);
  	struct i2c_client *client = to_i2c_client(dev);
  	unsigned int config;
  	int fps;
  	int ret;
  
  	for (fps = 0; fps < MAX77620_FPS_COUNT; fps++) {
  		if (chip->suspend_fps_period[fps] < 0)
  			continue;
  
  		ret = max77620_set_fps_period(chip, fps,
  					      chip->suspend_fps_period[fps]);
  		if (ret < 0)
  			return ret;
  	}
  
  	/*
  	 * For MAX20024: No need to configure SLPEN on suspend as
  	 * it will be configured on Init.
  	 */
  	if (chip->chip_id == MAX20024)
  		goto out;
  
  	config = (chip->sleep_enable) ? MAX77620_ONOFFCNFG1_SLPEN : 0;
  	ret = regmap_update_bits(chip->rmap, MAX77620_REG_ONOFFCNFG1,
  				 MAX77620_ONOFFCNFG1_SLPEN,
  				 config);
  	if (ret < 0) {
  		dev_err(dev, "Failed to configure sleep in suspend: %d
  ", ret);
  		return ret;
  	}

  	if (chip->chip_id == MAX77663)
  		goto out;

  	/* Disable WK_EN0 */
  	ret = regmap_update_bits(chip->rmap, MAX77620_REG_ONOFFCNFG2,
  				 MAX77620_ONOFFCNFG2_WK_EN0, 0);
  	if (ret < 0) {
  		dev_err(dev, "Failed to configure WK_EN in suspend: %d
  ", ret);
  		return ret;
  	}
  
  out:
  	disable_irq(client->irq);
  
  	return 0;
  }
  
  static int max77620_i2c_resume(struct device *dev)
  {
  	struct max77620_chip *chip = dev_get_drvdata(dev);
  	struct i2c_client *client = to_i2c_client(dev);
  	int ret;
  	int fps;
  
  	for (fps = 0; fps < MAX77620_FPS_COUNT; fps++) {
  		if (chip->shutdown_fps_period[fps] < 0)
  			continue;
  
  		ret = max77620_set_fps_period(chip, fps,
  					      chip->shutdown_fps_period[fps]);
  		if (ret < 0)
  			return ret;
  	}
  
  	/*
  	 * For MAX20024: No need to configure WKEN0 on resume as
  	 * it is configured on Init.
  	 */

  	if (chip->chip_id == MAX20024 || chip->chip_id == MAX77663)

  		goto out;
  
  	/* Enable WK_EN0 */
  	ret = regmap_update_bits(chip->rmap, MAX77620_REG_ONOFFCNFG2,
  				 MAX77620_ONOFFCNFG2_WK_EN0,
  				 MAX77620_ONOFFCNFG2_WK_EN0);
  	if (ret < 0) {
  		dev_err(dev, "Failed to configure WK_EN0 n resume: %d
  ", ret);
  		return ret;
  	}
  
  out:
  	enable_irq(client->irq);
  
  	return 0;
  }
  #endif
  
  static const struct i2c_device_id max77620_id[] = {
  	{"max77620", MAX77620},
  	{"max20024", MAX20024},

  	{"max77663", MAX77663},

  	{},
  };

  
  static const struct dev_pm_ops max77620_pm_ops = {
  	SET_SYSTEM_SLEEP_PM_OPS(max77620_i2c_suspend, max77620_i2c_resume)
  };
  
  static struct i2c_driver max77620_driver = {
  	.driver = {
  		.name = "max77620",
  		.pm = &max77620_pm_ops,
  	},
  	.probe = max77620_probe,
  	.id_table = max77620_id,
  };

  builtin_i2c_driver(max77620_driver);