/*
   * Copyright (C) 2015 Google, Inc
   * Written by Simon Glass <sjg@chromium.org>
   *
   * SPDX-License-Identifier:	GPL-2.0+
   */
  
  #include <common.h>
  #include <mmc.h>
  #include <dm.h>
  #include <dm/device-internal.h>
  #include <dm/lists.h>
  #include "mmc_private.h"
  
  DECLARE_GLOBAL_DATA_PTR;
  
  int dm_mmc_send_cmd(struct udevice *dev, struct mmc_cmd *cmd,
  		    struct mmc_data *data)
  {
  	struct mmc *mmc = mmc_get_mmc_dev(dev);
  	struct dm_mmc_ops *ops = mmc_get_ops(dev);
  	int ret;
  
  	mmmc_trace_before_send(mmc, cmd);
  	if (ops->send_cmd)
  		ret = ops->send_cmd(dev, cmd, data);
  	else
  		ret = -ENOSYS;
  	mmmc_trace_after_send(mmc, cmd, ret);
  
  	return ret;
  }
  
  int mmc_send_cmd(struct mmc *mmc, struct mmc_cmd *cmd, struct mmc_data *data)
  {
  	return dm_mmc_send_cmd(mmc->dev, cmd, data);
  }
  
  int dm_mmc_set_ios(struct udevice *dev)
  {
  	struct dm_mmc_ops *ops = mmc_get_ops(dev);
  
  	if (!ops->set_ios)
  		return -ENOSYS;
  	return ops->set_ios(dev);
  }
  
  int mmc_set_ios(struct mmc *mmc)
  {
  	return dm_mmc_set_ios(mmc->dev);
  }
  
  void dm_mmc_send_init_stream(struct udevice *dev)
  {
  	struct dm_mmc_ops *ops = mmc_get_ops(dev);
  
  	if (ops->send_init_stream)
  		ops->send_init_stream(dev);
  }
  
  void mmc_send_init_stream(struct mmc *mmc)
  {
  	dm_mmc_send_init_stream(mmc->dev);
  }
  
  #if CONFIG_IS_ENABLED(MMC_UHS_SUPPORT)
  int dm_mmc_wait_dat0(struct udevice *dev, int state, int timeout)
  {
  	struct dm_mmc_ops *ops = mmc_get_ops(dev);
  
  	if (!ops->wait_dat0)
  		return -ENOSYS;
  	return ops->wait_dat0(dev, state, timeout);
  }
  
  int mmc_wait_dat0(struct mmc *mmc, int state, int timeout)
  {
  	return dm_mmc_wait_dat0(mmc->dev, state, timeout);
  }
  #endif
  
  int dm_mmc_get_wp(struct udevice *dev)
  {
  	struct dm_mmc_ops *ops = mmc_get_ops(dev);
  
  	if (!ops->get_wp)
  		return -ENOSYS;
  	return ops->get_wp(dev);
  }
  
  int mmc_getwp(struct mmc *mmc)
  {
  	return dm_mmc_get_wp(mmc->dev);
  }
  
  int dm_mmc_get_cd(struct udevice *dev)
  {
  	struct dm_mmc_ops *ops = mmc_get_ops(dev);
  
  	if (!ops->get_cd)
  		return -ENOSYS;
  	return ops->get_cd(dev);
  }
  
  int mmc_getcd(struct mmc *mmc)
  {
  	return dm_mmc_get_cd(mmc->dev);
  }
  
  #ifdef MMC_SUPPORTS_TUNING
  int dm_mmc_execute_tuning(struct udevice *dev, uint opcode)
  {
  	struct dm_mmc_ops *ops = mmc_get_ops(dev);
  
  	if (!ops->execute_tuning)
  		return -ENOSYS;
  	return ops->execute_tuning(dev, opcode);
  }
  
  int mmc_execute_tuning(struct mmc *mmc, uint opcode)
  {
  	return dm_mmc_execute_tuning(mmc->dev, opcode);
  }
  #endif
  
  #if CONFIG_IS_ENABLED(MMC_HS400_ES_SUPPORT)
  void dm_mmc_set_enhanced_strobe(struct udevice *dev)
  {
  	struct dm_mmc_ops *ops = mmc_get_ops(dev);
  
  	if (ops->set_enhanced_strobe)
  		ops->set_enhanced_strobe(dev);
  }
  
  void mmc_set_enhanced_strobe(struct mmc *mmc)
  {
  	dm_mmc_set_enhanced_strobe(mmc->dev);
  }
  #endif
  
  int mmc_of_parse(struct udevice *dev, struct mmc_config *cfg)
  {
  	int val;
  
  	val = dev_read_u32_default(dev, "bus-width", 1);
  
  	switch (val) {
  	case 0x8:
  		cfg->host_caps |= MMC_MODE_8BIT;
  		/* fall through */
  	case 0x4:
  		cfg->host_caps |= MMC_MODE_4BIT;
  		/* fall through */
  	case 0x1:
  		cfg->host_caps |= MMC_MODE_1BIT;
  		break;
  	default:
  		dev_err(dev, "Invalid \"bus-width\" value %u!
  ", val);
  		return -EINVAL;
  	}
  
  	/* f_max is obtained from the optional "max-frequency" property */
  	dev_read_u32(dev, "max-frequency", &cfg->f_max);
  
  	if (dev_read_bool(dev, "cap-sd-highspeed"))
  		cfg->host_caps |= MMC_CAP(SD_HS);
  	if (dev_read_bool(dev, "cap-mmc-highspeed"))
  		cfg->host_caps |= MMC_CAP(MMC_HS);
  	if (dev_read_bool(dev, "sd-uhs-sdr12"))
  		cfg->host_caps |= MMC_CAP(UHS_SDR12);
  	if (dev_read_bool(dev, "sd-uhs-sdr25"))
  		cfg->host_caps |= MMC_CAP(UHS_SDR25);
  	if (dev_read_bool(dev, "sd-uhs-sdr50"))
  		cfg->host_caps |= MMC_CAP(UHS_SDR50);
  	if (dev_read_bool(dev, "sd-uhs-sdr104"))
  		cfg->host_caps |= MMC_CAP(UHS_SDR104);
  	if (dev_read_bool(dev, "sd-uhs-ddr50"))
  		cfg->host_caps |= MMC_CAP(UHS_DDR50);
  	if (dev_read_bool(dev, "mmc-ddr-1_8v"))
  		cfg->host_caps |= MMC_CAP(MMC_DDR_52);
  	if (dev_read_bool(dev, "mmc-ddr-1_2v"))
  		cfg->host_caps |= MMC_CAP(MMC_DDR_52);
  	if (dev_read_bool(dev, "mmc-hs200-1_8v"))
  		cfg->host_caps |= MMC_CAP(MMC_HS_200);
  	if (dev_read_bool(dev, "mmc-hs200-1_2v"))
  		cfg->host_caps |= MMC_CAP(MMC_HS_200);
  
  	return 0;
  }
  
  struct mmc *mmc_get_mmc_dev(struct udevice *dev)
  {
  	struct mmc_uclass_priv *upriv;
  
  	if (!device_active(dev))
  		return NULL;
  	upriv = dev_get_uclass_priv(dev);
  	return upriv->mmc;
  }
  
  #if CONFIG_IS_ENABLED(BLK)
  struct mmc *find_mmc_device(int dev_num)
  {
  	struct udevice *dev, *mmc_dev;
  	int ret;
  
  	ret = blk_find_device(IF_TYPE_MMC, dev_num, &dev);
  
  	if (ret) {
  #if !defined(CONFIG_SPL_BUILD) || defined(CONFIG_SPL_LIBCOMMON_SUPPORT)
  		printf("MMC Device %d not found
  ", dev_num);
  #endif
  		return NULL;
  	}
  
  	mmc_dev = dev_get_parent(dev);
  
  	struct mmc *mmc = mmc_get_mmc_dev(mmc_dev);
  
  	return mmc;
  }
  
  int get_mmc_num(void)
  {
  	return max((blk_find_max_devnum(IF_TYPE_MMC) + 1), 0);
  }
  
  int mmc_get_next_devnum(void)
  {
  	return blk_find_max_devnum(IF_TYPE_MMC);
  }
  
  struct blk_desc *mmc_get_blk_desc(struct mmc *mmc)
  {
  	struct blk_desc *desc;
  	struct udevice *dev;
  
  	device_find_first_child(mmc->dev, &dev);
  	if (!dev)
  		return NULL;
  	desc = dev_get_uclass_platdata(dev);
  
  	return desc;
  }
  
  void mmc_do_preinit(void)
  {
  	struct udevice *dev;
  	struct uclass *uc;
  	int ret;
  
  	ret = uclass_get(UCLASS_MMC, &uc);
  	if (ret)
  		return;
  	uclass_foreach_dev(dev, uc) {
  		struct mmc *m = mmc_get_mmc_dev(dev);
  
  		if (!m)
  			continue;
  #ifdef CONFIG_FSL_ESDHC_ADAPTER_IDENT
  		mmc_set_preinit(m, 1);
  #endif
  		if (m->preinit)
  			mmc_start_init(m);
  	}
  }
  
  #if !defined(CONFIG_SPL_BUILD) || defined(CONFIG_SPL_LIBCOMMON_SUPPORT)
  void print_mmc_devices(char separator)
  {
  	struct udevice *dev;
  	char *mmc_type;
  	bool first = true;
  
  	for (uclass_first_device(UCLASS_MMC, &dev);
  	     dev;
  	     uclass_next_device(&dev), first = false) {
  		struct mmc *m = mmc_get_mmc_dev(dev);
  
  		if (!first) {
  			printf("%c", separator);
  			if (separator != '
  ')
  				puts(" ");
  		}
  		if (m->has_init)
  			mmc_type = IS_SD(m) ? "SD" : "eMMC";
  		else
  			mmc_type = NULL;
  
  		printf("%s: %d", m->cfg->name, mmc_get_blk_desc(m)->devnum);
  		if (mmc_type)
  			printf(" (%s)", mmc_type);
  	}
  
  	printf("
  ");
  }
  
  #else
  void print_mmc_devices(char separator) { }
  #endif
  
  int mmc_bind(struct udevice *dev, struct mmc *mmc, const struct mmc_config *cfg)
  {
  	struct blk_desc *bdesc;
  	struct udevice *bdev;
  	int ret, devnum = -1;
  
  	if (!mmc_get_ops(dev))
  		return -ENOSYS;
  #ifndef CONFIG_SPL_BUILD
  	/* Use the fixed index with aliase node's index */
  	ret = dev_read_alias_seq(dev, &devnum);
  	debug("%s: alias ret=%d, devnum=%d
  ", __func__, ret, devnum);
  #endif
  
  	ret = blk_create_devicef(dev, "mmc_blk", "blk", IF_TYPE_MMC,
  			devnum, 512, 0, &bdev);
  	if (ret) {
  		debug("Cannot create block device
  ");
  		return ret;
  	}
  	bdesc = dev_get_uclass_platdata(bdev);
  	mmc->cfg = cfg;
  	mmc->priv = dev;
  
  	/* the following chunk was from mmc_register() */
  
  	/* Setup dsr related values */
  	mmc->dsr_imp = 0;
  	mmc->dsr = 0xffffffff;
  	/* Setup the universal parts of the block interface just once */
  	bdesc->removable = 1;
  
  	/* setup initial part type */
  	bdesc->part_type = cfg->part_type;
  	mmc->dev = dev;
  
  	return 0;
  }
  
  int mmc_unbind(struct udevice *dev)
  {
  	struct udevice *bdev;
  
  	device_find_first_child(dev, &bdev);
  	if (bdev) {
  		device_remove(bdev, DM_REMOVE_NORMAL);
  		device_unbind(bdev);
  	}
  
  	return 0;
  }
  
  static int mmc_select_hwpart(struct udevice *bdev, int hwpart)
  {
  	struct udevice *mmc_dev = dev_get_parent(bdev);
  	struct mmc *mmc = mmc_get_mmc_dev(mmc_dev);
  	struct blk_desc *desc = dev_get_uclass_platdata(bdev);
  
  	if (desc->hwpart == hwpart)
  		return 0;
  
  	if (mmc->part_config == MMCPART_NOAVAILABLE)
  		return -EMEDIUMTYPE;
  
  	return mmc_switch_part(mmc, hwpart);
  }
  
  static int mmc_blk_probe(struct udevice *dev)
  {
  	struct udevice *mmc_dev = dev_get_parent(dev);
  	struct mmc_uclass_priv *upriv = dev_get_uclass_priv(mmc_dev);
  	struct mmc *mmc = upriv->mmc;
  	int ret;
  
  	ret = mmc_init(mmc);
  	if (ret) {
  		debug("%s: mmc_init() failed (err=%d)
  ", __func__, ret);
  		return ret;
  	}
  
  	return 0;
  }
  
  static const struct blk_ops mmc_blk_ops = {
  	.read	= mmc_bread,
  #if CONFIG_IS_ENABLED(MMC_WRITE)
  	.write	= mmc_bwrite,
  	.erase	= mmc_berase,
  #endif
  	.select_hwpart	= mmc_select_hwpart,
  };
  
  U_BOOT_DRIVER(mmc_blk) = {
  	.name		= "mmc_blk",
  	.id		= UCLASS_BLK,
  	.ops		= &mmc_blk_ops,
  	.probe		= mmc_blk_probe,
  };
  #endif /* CONFIG_BLK */
  
  U_BOOT_DRIVER(mmc) = {
  	.name	= "mmc",
  	.id	= UCLASS_MMC,
  };
  
  UCLASS_DRIVER(mmc) = {
  	.id		= UCLASS_MMC,
  	.name		= "mmc",
  	.flags		= DM_UC_FLAG_SEQ_ALIAS,
  	.per_device_auto_alloc_size = sizeof(struct mmc_uclass_priv),
  };