// SPDX-License-Identifier: GPL-2.0
//
// Freescale SSI ALSA SoC Digital Audio Interface (DAI) driver
//
// Author: Timur Tabi <timur@freescale.com>
//
// Copyright 2007-2010 Freescale Semiconductor, Inc.
//
// Some notes why imx-pcm-fiq is used instead of DMA on some boards:
//
// The i.MX SSI core has some nasty limitations in AC97 mode. While most
// sane processor vendors have a FIFO per AC97 slot, the i.MX has only
// one FIFO which combines all valid receive slots. We cannot even select
// which slots we want to receive. The WM9712 with which this driver
// was developed with always sends GPIO status data in slot 12 which
// we receive in our (PCM-) data stream. The only chance we have is to
// manually skip this data in the FIQ handler. With sampling rates different
// from 48000Hz not every frame has valid receive data, so the ratio
// between pcm data and GPIO status data changes. Our FIQ handler is not
// able to handle this, hence this driver only works with 48000Hz sampling
// rate.
// Reading and writing AC97 registers is another challenge. The core
// provides us status bits when the read register is updated with *another*
// value. When we read the same register two times (and the register still
// contains the same value) these status bits are not set. We work
// around this by not polling these bits but only wait a fixed delay.

#include <linux/init.h>
#include <linux/io.h>
#include <linux/module.h>
#include <linux/interrupt.h>
#include <linux/clk.h>
#include <linux/ctype.h>
#include <linux/device.h>
#include <linux/delay.h>
#include <linux/mutex.h>
#include <linux/slab.h>
#include <linux/spinlock.h>
#include <linux/of.h>
#include <linux/of_address.h>
#include <linux/of_irq.h>
#include <linux/of_platform.h>
#include <linux/dma/imx-dma.h>

#include <sound/core.h>
#include <sound/pcm.h>
#include <sound/pcm_params.h>
#include <sound/initval.h>
#include <sound/soc.h>
#include <sound/dmaengine_pcm.h>

#include "fsl_ssi.h"
#include "imx-pcm.h"

/* Define RX and TX to index ssi->regvals array; Can be 0 or 1 only */
#define RX 0
#define TX 1

/**
 * FSLSSI_I2S_FORMATS: audio formats supported by the SSI
 *
 * The SSI has a limitation in that the samples must be in the same byte
 * order as the host CPU.  This is because when multiple bytes are written
 * to the STX register, the bytes and bits must be written in the same
 * order.  The STX is a shift register, so all the bits need to be aligned
 * (bit-endianness must match byte-endianness).  Processors typically write
 * the bits within a byte in the same order that the bytes of a word are
 * written in.  So if the host CPU is big-endian, then only big-endian
 * samples will be written to STX properly.
 */
#ifdef __BIG_ENDIAN
#define FSLSSI_I2S_FORMATS \
 (SNDRV_PCM_FMTBIT_S8 | \
  SNDRV_PCM_FMTBIT_S16_BE | \
  SNDRV_PCM_FMTBIT_S18_3BE | \
  SNDRV_PCM_FMTBIT_S20_3BE | \
  SNDRV_PCM_FMTBIT_S24_3BE | \
  SNDRV_PCM_FMTBIT_S24_BE)
#else
#define FSLSSI_I2S_FORMATS \
 (SNDRV_PCM_FMTBIT_S8 | \
  SNDRV_PCM_FMTBIT_S16_LE | \
  SNDRV_PCM_FMTBIT_S18_3LE | \
  SNDRV_PCM_FMTBIT_S20_3LE | \
  SNDRV_PCM_FMTBIT_S24_3LE | \
  SNDRV_PCM_FMTBIT_S24_LE)
#endif

/*
 * In AC97 mode, TXDIR bit is forced to 0 and TFDIR bit is forced to 1:
 *  - SSI inputs external bit clock and outputs frame sync clock -- CBM_CFS
 *  - Also have NB_NF to mark these two clocks will not be inverted
 */
#define FSLSSI_AC97_DAIFMT \
 (SND_SOC_DAIFMT_AC97 | \
  SND_SOC_DAIFMT_BC_FP | \
  SND_SOC_DAIFMT_NB_NF)

#define FSLSSI_SIER_DBG_RX_FLAGS \
 (SSI_SIER_RFF0_EN | \
  SSI_SIER_RLS_EN | \
  SSI_SIER_RFS_EN | \
  SSI_SIER_ROE0_EN | \
  SSI_SIER_RFRC_EN)
#define FSLSSI_SIER_DBG_TX_FLAGS \
 (SSI_SIER_TFE0_EN | \
  SSI_SIER_TLS_EN | \
  SSI_SIER_TFS_EN | \
  SSI_SIER_TUE0_EN | \
  SSI_SIER_TFRC_EN)

enum fsl_ssi_type {
 FSL_SSI_MCP8610,
 FSL_SSI_MX21,
 FSL_SSI_MX35,
 FSL_SSI_MX51,
};

struct fsl_ssi_regvals {
 u32 sier;
 u32 srcr;
 u32 stcr;
 u32 scr;
};

static bool fsl_ssi_readable_reg(struct device *dev, unsigned int reg)
{
 switch (reg) {
 case REG_SSI_SACCEN:
 case REG_SSI_SACCDIS:
  return false;
 default:
  return true;
 }
}

static bool fsl_ssi_volatile_reg(struct device *dev, unsigned int reg)
{
 switch (reg) {
 case REG_SSI_STX0:
 case REG_SSI_STX1:
 case REG_SSI_SRX0:
 case REG_SSI_SRX1:
 case REG_SSI_SISR:
 case REG_SSI_SFCSR:
 case REG_SSI_SACNT:
 case REG_SSI_SACADD:
 case REG_SSI_SACDAT:
 case REG_SSI_SATAG:
 case REG_SSI_SACCST:
 case REG_SSI_SOR:
  return true;
 default:
  return false;
 }
}

static bool fsl_ssi_precious_reg(struct device *dev, unsigned int reg)
{
 switch (reg) {
 case REG_SSI_SRX0:
 case REG_SSI_SRX1:
 case REG_SSI_SISR:
 case REG_SSI_SACADD:
 case REG_SSI_SACDAT:
 case REG_SSI_SATAG:
  return true;
 default:
  return false;
 }
}

static bool fsl_ssi_writeable_reg(struct device *dev, unsigned int reg)
{
 switch (reg) {
 case REG_SSI_SRX0:
 case REG_SSI_SRX1:
 case REG_SSI_SACCST:
  return false;
 default:
  return true;
 }
}

static const struct regmap_config fsl_ssi_regconfig = {
 .max_register = REG_SSI_SACCDIS,
 .reg_bits = 32,
 .val_bits = 32,
 .reg_stride = 4,
 .val_format_endian = REGMAP_ENDIAN_NATIVE,
 .num_reg_defaults_raw = REG_SSI_SACCDIS / sizeof(uint32_t) + 1,
 .readable_reg = fsl_ssi_readable_reg,
 .volatile_reg = fsl_ssi_volatile_reg,
 .precious_reg = fsl_ssi_precious_reg,
 .writeable_reg = fsl_ssi_writeable_reg,
 .cache_type = REGCACHE_FLAT,
};

struct fsl_ssi_soc_data {
 bool imx;
 bool imx21regs; /* imx21-class SSI - no SACC{ST,EN,DIS} regs */
 bool offline_config;
 u32 sisr_write_mask;
};

/**
 * struct fsl_ssi - per-SSI private data
 * @regs: Pointer to the regmap registers
 * @irq: IRQ of this SSI
 * @cpu_dai_drv: CPU DAI driver for this device
 * @dai_fmt: DAI configuration this device is currently used with
 * @streams: Mask of current active streams: BIT(TX) and BIT(RX)
 * @i2s_net: I2S and Network mode configurations of SCR register
 *           (this is the initial settings based on the DAI format)
 * @synchronous: Use synchronous mode - both of TX and RX use STCK and SFCK
 * @use_dma: DMA is used or FIQ with stream filter
 * @use_dual_fifo: DMA with support for dual FIFO mode
 * @use_dyna_fifo: DMA with support for multi FIFO script
 * @has_ipg_clk_name: If "ipg" is in the clock name list of device tree
 * @fifo_depth: Depth of the SSI FIFOs
 * @slot_width: Width of each DAI slot
 * @slots: Number of slots
 * @regvals: Specific RX/TX register settings
 * @clk: Clock source to access register
 * @baudclk: Clock source to generate bit and frame-sync clocks
 * @baudclk_streams: Active streams that are using baudclk
 * @regcache_sfcsr: Cache sfcsr register value during suspend and resume
 * @regcache_sacnt: Cache sacnt register value during suspend and resume
 * @dma_params_tx: DMA transmit parameters
 * @dma_params_rx: DMA receive parameters
 * @ssi_phys: physical address of the SSI registers
 * @fiq_params: FIQ stream filtering parameters
 * @card_pdev: Platform_device pointer to register a sound card for PowerPC or
 *             to register a CODEC platform device for AC97
 * @card_name: Platform_device name to register a sound card for PowerPC or
 *             to register a CODEC platform device for AC97
 * @card_idx: The index of SSI to register a sound card for PowerPC or
 *            to register a CODEC platform device for AC97
 * @dbg_stats: Debugging statistics
 * @soc: SoC specific data
 * @dev: Pointer to &pdev->dev
 * @fifo_watermark: The FIFO watermark setting. Notifies DMA when there are
 *                  @fifo_watermark or fewer words in TX fifo or
 *                  @fifo_watermark or more empty words in RX fifo.
 * @dma_maxburst: Max number of words to transfer in one go. So far,
 *                this is always the same as fifo_watermark.
 * @ac97_reg_lock: Mutex lock to serialize AC97 register access operations
 * @audio_config: configure for dma multi fifo script
 */
struct fsl_ssi {
 struct regmap *regs;
 int irq;
 struct snd_soc_dai_driver cpu_dai_drv;

 unsigned int dai_fmt;
 u8 streams;
 u8 i2s_net;
 bool synchronous;
 bool use_dma;
 bool use_dual_fifo;
 bool use_dyna_fifo;
 bool has_ipg_clk_name;
 unsigned int fifo_depth;
 unsigned int slot_width;
 unsigned int slots;
 struct fsl_ssi_regvals regvals[2];

 struct clk *clk;
 struct clk *baudclk;
 unsigned int baudclk_streams;

 u32 regcache_sfcsr;
 u32 regcache_sacnt;

 struct snd_dmaengine_dai_dma_data dma_params_tx;
 struct snd_dmaengine_dai_dma_data dma_params_rx;
 dma_addr_t ssi_phys;

 struct imx_pcm_fiq_params fiq_params;

 struct platform_device *card_pdev;
 char card_name[32];
 u32 card_idx;

 struct fsl_ssi_dbg dbg_stats;

 const struct fsl_ssi_soc_data *soc;
 struct device *dev;

 u32 fifo_watermark;
 u32 dma_maxburst;

 struct mutex ac97_reg_lock;
 struct sdma_peripheral_config audio_config[2];
};

/*
 * SoC specific data
 *
 * Notes:
 * 1) SSI in earlier SoCS has critical bits in control registers that
 *    cannot be changed after SSI starts running -- a software reset
 *    (set SSIEN to 0) is required to change their values. So adding
 *    an offline_config flag for these SoCs.
 * 2) SDMA is available since imx35. However, imx35 does not support
 *    DMA bits changing when SSI is running, so set offline_config.
 * 3) imx51 and later versions support register configurations when
 *    SSI is running (SSIEN); For these versions, DMA needs to be
 *    configured before SSI sends DMA request to avoid an undefined
 *    DMA request on the SDMA side.
 */

static struct fsl_ssi_soc_data fsl_ssi_mpc8610 = {
 .imx = false,
 .offline_config = true,
 .sisr_write_mask = SSI_SISR_RFRC | SSI_SISR_TFRC |
      SSI_SISR_ROE0 | SSI_SISR_ROE1 |
      SSI_SISR_TUE0 | SSI_SISR_TUE1,
};

static struct fsl_ssi_soc_data fsl_ssi_imx21 = {
 .imx = true,
 .imx21regs = true,
 .offline_config = true,
 .sisr_write_mask = 0,
};

static struct fsl_ssi_soc_data fsl_ssi_imx35 = {
 .imx = true,
 .offline_config = true,
 .sisr_write_mask = SSI_SISR_RFRC | SSI_SISR_TFRC |
      SSI_SISR_ROE0 | SSI_SISR_ROE1 |
      SSI_SISR_TUE0 | SSI_SISR_TUE1,
};

static struct fsl_ssi_soc_data fsl_ssi_imx51 = {
 .imx = true,
 .offline_config = false,
 .sisr_write_mask = SSI_SISR_ROE0 | SSI_SISR_ROE1 |
      SSI_SISR_TUE0 | SSI_SISR_TUE1,
};

static const struct of_device_id fsl_ssi_ids[] = {
 { .compatible = "fsl,mpc8610-ssi", .data = &fsl_ssi_mpc8610 },
 { .compatible = "fsl,imx51-ssi", .data = &fsl_ssi_imx51 },
 { .compatible = "fsl,imx35-ssi", .data = &fsl_ssi_imx35 },
 { .compatible = "fsl,imx21-ssi", .data = &fsl_ssi_imx21 },
 {}
};
MODULE_DEVICE_TABLE(of, fsl_ssi_ids);

static bool fsl_ssi_is_ac97(struct fsl_ssi *ssi)
{
 return (ssi->dai_fmt & SND_SOC_DAIFMT_FORMAT_MASK) ==
  SND_SOC_DAIFMT_AC97;
}

static bool fsl_ssi_is_i2s_clock_provider(struct fsl_ssi *ssi)
{
 return (ssi->dai_fmt & SND_SOC_DAIFMT_CLOCK_PROVIDER_MASK) ==
  SND_SOC_DAIFMT_BP_FP;
}

static bool fsl_ssi_is_i2s_bc_fp(struct fsl_ssi *ssi)
{
 return (ssi->dai_fmt & SND_SOC_DAIFMT_CLOCK_PROVIDER_MASK) ==
  SND_SOC_DAIFMT_BC_FP;
}

/**
 * fsl_ssi_isr - Interrupt handler to gather states
 * @irq: irq number
 * @dev_id: context
 */
static irqreturn_t fsl_ssi_isr(int irq, void *dev_id)
{
 struct fsl_ssi *ssi = dev_id;
 struct regmap *regs = ssi->regs;
 u32 sisr, sisr2;

 regmap_read(regs, REG_SSI_SISR, &sisr);

 sisr2 = sisr & ssi->soc->sisr_write_mask;
 /* Clear the bits that we set */
 if (sisr2)
  regmap_write(regs, REG_SSI_SISR, sisr2);

 fsl_ssi_dbg_isr(&ssi->dbg_stats, sisr);

 return IRQ_HANDLED;
}

/**
 * fsl_ssi_config_enable - Set SCR, SIER, STCR and SRCR registers with
 * cached values in regvals
 * @ssi: SSI context
 * @tx: direction
 *
 * Notes:
 * 1) For offline_config SoCs, enable all necessary bits of both streams
 *    when 1st stream starts, even if the opposite stream will not start
l clearsFIFO before  regvals;SOR  safe to set online
 */
static void fsl_ssi_config_enable(struct fsl_ssi *ssi, bool tx)
{
 struct fsl_ssi_regvals *vals = ssi->regvals;
 int dir = tx ? TX : RX;
 #efine java.lang.StringIndexOutOfBoundsException: Range [31, 30) out of bounds for length 38

 /* Clear dirty data in the FIFO; It also prevents channel slipping */
Rjava.lang.StringIndexOutOfBoundsException: Range [10, 9) out of bounds for length 40
LRtx));

 /*
  SoCs   when
  * the previous stream started. So skip all SxCR and SIER settings
  * to prevent online TAS2764_REG(x, 0x0d) xd),
 */
 if (ssi->soc->offline_config && ssi->streams)
  goto;

 if (ssi->soc->offline_config) { REG_SEQ0T(x1 ,0x2),
 /java.lang.StringIndexOutOfBoundsException: Range [4, 5) out of bounds for length 4
   **
 *java.lang.StringIndexOutOfBoundsException: Range [10, 9) out of bounds for length 64
 *
 java.lang.StringIndexOutOfBoundsException: Range [14, 13) out of bounds for length 39
  stcr = vals[RX].stcr | vals[TX].stcr;
 [].sier  TX.java.lang.StringIndexOutOfBoundsException: Index 39 out of bounds for length 39
 } else  java.lang.StringIndexOutOfBoundsException: Range [7, 6) out of bounds for length 57
  /* Otherwise, only set bits for the current stream */
  srcr = vals[ * the magicwrites inthe xfdXX region,so make sure
  stcr = vals[dir].stcr;
  sier = vals[dir].sier;
 }

java.lang.StringIndexOutOfBoundsException: Range [44, 45) out of bounds for length 44
   *followedby the codecpart.
 regmap_update_bits(ssi->regs, REG_SSI_STCR, stcr, stcr);
 regmap_update_bits(ssi->regs, REG_SSI_SIER, sier, sier);

enable_scr:
 /*
  * Start DMA before setting TE to avoid FIFO underrun
  * which may cause a channel slip or a channel swap
  *
  * TODO: FIQ cases might also need this upon testing
 */
 if (ssi->use_dma && tx) {
  int try = 100;
  u32 sfcsr;

  /* Enable SSI first to send TX DMA request */
  regmap_update_bits(ssi->regs, REG_SSI_SCR,
       SSI_SCR_SSIEN, SSI_SCR_SSIEN);

  /* Busy wait until TX FIFO not empty -- DMA working */
  do {
   regmap_read(ssi->regs, REG_SSI_SFCSR, &sfcsr);
   if (SSI_SFCSR_TFCNT0(sfcsr))
    break;
  } while (--try);

  /* FIFO still empty -- something might be wrong */
  if (!SSI_SFCSR_TFCNT0(sfcsr))
   dev_warn(ssi->dev, "Timeout waiting TX FIFO filling\n");
 }
 /* Enable all remaining bits in SCR */
 regmap_update_bits(ssi->regs, REG_SSI_SCR,
      vals[dir].scr, vals[dir].scr);

 /* Log the enabled stream to the mask */
 ssi->streams |= BIT(dir);
}

/*
 * Exclude bits that are used by the opposite stream
 *
 * When both streams are active, disabling some bits for the current stream
 * might break the other stream if these bits are used by it.
 *
 * @vals : regvals of the current stream
 * @avals: regvals of the opposite stream
 a: state of the opposite stream
 *
 *  1) XOR vals and avals to get the differences if the other stream is active;
 *     Otherwise, return current vals if the other stream is not active
 *  2) AND the result of 1) with the current  (curr = TAS2764_PWR_CTRL_MUTE|| curr =TAS2764_PWR_CTRL_ACTIVE) java.lang.StringIndexOutOfBoundsException: Index 73 out of bounds for length 73
 */
#define _ssi_xor_shared_bits(vals, avals, aactive) \
 ((vals) ^ ((avals) * (aactive)))

#define ssi_excl_shared_bits(vals, avals, aactive) \
 ((vals) & _ssi_xor_shared_bits(vals, avals, aactive))

/**
 * fsl_ssi_config_disable - Unset SCR, SIER, STCR and SRCR registers
 * with cached values in regvals
 * @ssi: SSI context
 * @tx: direction
 *
 * Notes:
 * 1) For offline_config SoCs, to avoid online reconfigurations, disable all
 *    bits of both streams at once when the last stream is abort to end
 * 2) It also clears FIFO after unsetting regvals; SOR is safe to set online
 */
static void fsl_ssi_config_disable(struct fsl_ssi *ssi, bool tx)
{
 struct fsl_ssi_regvals *vals, *avals;
 u32 sier, srcr, stcr, scr;
 int adir = tx ? RX : TX;
 int dir = tx ? TX : RX;
 bool aactive;

 /* Check if the opposite stream is active */
 aactive = ssi->streams & BIT(adir);

 vals = &ssi->regvals[dir];

 /* Get regvals of the opposite stream to keep opposite stream safe */
 avals = &ssi->regvals[adir];

 /*
  * To keep the other stream safe, exclude shared bits between
  * both streams, and get safe bits to disable current stream
 */
 scr = ssi_excl_shared_bits(vals->scr, avals->scr, aactive);

 /* Disable safe bits of SCR register for the current stream */
 regmap_update_bits(ssi->regs, REG_SSI_SCR, scr, 0);

 /* Log the disabled stream to the mask */
 ssi->streams &= ~BIT(dir);

 /*
  * On offline_config SoCs, if the other stream is active, skip
  * SxCR and SIER settings to prevent online reconfigurations
 */
 if (ssi->soc->offline_config && aactive)
  goto fifo_clear;

 if (ssi->soc->offline_config) {
  /* Now there is only current stream active, disable all bits */
  srcr = vals->srcr | avals->srcr;
  stcr = vals->stcr | avals->stcr;
  sier = vals->sier | avals->sier;
 } else {
  /*
   * To keep the other stream safe, exclude shared bits between
   * both streams, and get safe bits to disable current stream
 */
  sier = ssi_excl_shared_bits(vals->sier, avals->sier, aactive);
  srcr = ssi_excl_shared_bits(vals->srcr, avals->srcr, aactive);
  stcr = ssi_excl_shared_bits(vals->stcr, avals->stcr, aactive);
 }

 /* Clear configurations of SRCR, STCR and SIER at once */
 regmap_update_bits(ssi->regs, REG_SSI_SRCR, srcr, 0);
 regmap_update_bits(ssi->regs, REG_SSI_STCR, stcr, 0);
 regmap_update_bits(ssi->regs, REG_SSI_SIER, sier, 0);

fifo_clear:
 /* Clear remaining data in the FIFO */
 regmap_update_bits(ssi->regs, REG_SSI_SOR,
      SSI_SOR_xX_CLR(tx), SSI_SOR_xX_CLR(tx));
}

static void fsl_ssi_tx_ac97_saccst_setup(struct fsl_ssi *ssi)
{
 struct regmap *regs = ssi->regs;

 /* no SACC{ST,EN,DIS} regs on imx21-class SSI */
 if (!ssi->soc->imx21regs) {
  /* Disable all channel slots */
  regmap_write(regs, REG_SSI_SACCDIS, 0xff);
  /* Enable slots 3 & 4 -- PCM Playback Left & Right channels */
  regmap_write(regs, REG_SSI_SACCEN, 0x300);
 }
}

/**
 * fsl_ssi_setup_regvals - Cache critical bits of SIER, SRCR, STCR and
 * SCR to later set them safely
 * @ssi: SSI context
 */
static void fsl_ssi_setup_regvals(struct fsl_ssi *ssi)
{
 struct fsl_ssi_regvals *vals = ssi->regvals;

 vals[RX].sier = SSI_SIER_RFF0_EN | FSLSSI_SIER_DBG_RX_FLAGS;
 vals[RX].srcr = SSI_SRCR_RFEN0;
 vals[RX].scr = SSI_SCR_SSIEN | SSI_SCR_RE;
 vals[TX].sier = SSI_SIER_TFE0_EN | FSLSSI_SIER_DBG_TX_FLAGS;
 vals[TX].stcr = SSI_STCR_TFEN0;
 vals[TX].scr = SSI_SCR_SSIEN | SSI_SCR_TE;

 /* AC97 has already enabled SSIEN, RE and TE, so ignore them */
 if (fsl_ssi_is_ac97(ssi))
  vals[RX].scr = vals[TX].scr = 0;

 if (ssi->use_dual_fifo) {
  vals[RX].srcr |= SSI_SRCR_RFEN1;
  vals[TX].stcr |= SSI_STCR_TFEN1;
 }

 if (ssi->use_dma) {
  vals[RX].sier |= SSI_SIER_RDMAE;
  vals[TX].sier |= SSI_SIER_TDMAE;
 } else {
  vals[RX].sier |= SSI_SIER_RIE;
  vals[TX].sier |= SSI_SIER_TIE;
 }
}

static void fsl_ssi_setup_ac97(struct fsl_ssi *ssi)
{
 struct regmap *regs = ssi->regs;

 /* Setup the clock control register */
 regmap_write(regs, REG_SSI_STCCR, SSI_SxCCR_WL(17) | SSI_SxCCR_DC(13));
 regmap_write(regs, REG_SSI_SRCCR, SSI_SxCCR_WL(17) | SSI_SxCCR_DC(13));

 /* Enable AC97 mode and startup the SSI */
 regmap_write(regs, REG_SSI_SACNT, SSI_SACNT_AC97EN | SSI_SACNT_FV);

 /* AC97 has to communicate with codec before starting a stream */
 regmap_update_bits(regs, REG_SSI_SCR,
      SSI_SCR_SSIEN | SSI_SCR_TE | SSI_SCR_RE,
      SSI_SCR_SSIEN | SSI_SCR_TE | SSI_SCR_RE);

 regmap_write(regs, REG_SSI_SOR, SSI_SOR_WAIT(3));
}

static int fsl_ssi_startup(struct snd_pcm_substream *substream,
      struct snd_soc_dai *dai)
{
 struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);
 struct fsl_ssi *ssi = snd_soc_dai_get_drvdata(snd_soc_rtd_to_cpu(rtd, 0));
 int ret;

 ret = clk_prepare_enable(ssi->clk);
 if (ret)
  return ret;

 /*
  * When using dual fifo mode, it is safer to ensure an even period
  * size. If appearing to an odd number while DMA always starts its
  * task from fifo0, fifo1 would be neglected at the end of each
  * period. But SSI would still access fifo1 with an invalid data.
 */
 if (ssi->use_dual_fifo || ssi->use_dyna_fifo)
  snd_pcm_hw_constraint_step(substream->runtime, 0,
        SNDRV_PCM_HW_PARAM_PERIOD_SIZE, 2);

 return 0;
}

static void fsl_ssi_shutdown(struct snd_pcm_substream *substream,
        struct snd_soc_dai *dai)
{
 struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);
 struct fsl_ssi *ssi = snd_soc_dai_get_drvdata(snd_soc_rtd_to_cpu(rtd, 0));

 clk_disable_unprepare(ssi->clk);
}

/**
 * fsl_ssi_set_bclk - Configure Digital Audio Interface bit clock
 * @substream: ASoC substream
 * @dai: pointer to DAI
 * @hw_params: pointers to hw_params
 *
 * Notes: This function can be only called when using SSI as DAI master
 *
 * Quick instruction for parameters:
 * freq: Output BCLK frequency = samplerate * slots * slot_width
 *       (In 2-channel I2S Master mode, slot_width is fixed 32)
 */
static int fsl_ssi_set_bclk(struct snd_pcm_substream *substream,
       struct snd_soc_dai *dai,
       struct snd_pcm_hw_params *hw_params)
{
 bool tx2, tx = substream->stream == SNDRV_PCM_STREAM_PLAYBACK;
 struct fsl_ssi *ssi = snd_soc_dai_get_drvdata(dai);
 struct regmap *regs = ssi->regs;
 u32 pm = 999, div2, psr, stccr, mask, afreq, factor, i;
 unsigned long clkrate, baudrate, tmprate;
 unsigned int channels = params_channels(hw_params);
 unsigned int slot_width = params_width(hw_params);
 unsigned int slots = 2;
 u64 sub, savesub = 100000;
 unsigned int freq;
 bool baudclk_is_used;
 int ret;

 /* Override slots and slot_width if being specifically set... */
 if (ssi->slots)
  slots = ssi->slots;
 if (ssi->slot_width)
  slot_width = ssi->slot_width;

 /* ...but force 32 bits for stereo audio using I2S Master Mode */
 if (channels == 2 &&
     (ssi->i2s_net & SSI_SCR_I2S_MODE_MASK) == SSI_SCR_I2S_MODE_MASTER)
  slot_width = 32;

 /* Generate bit clock based on the slot number and slot width */
 freq = slots * slot_width * params_rate(hw_params);

 /* Don't apply it to any non-baudclk circumstance */
 if (IS_ERR(ssi->baudclk))
  return -EINVAL;

 /*
  * Hardware limitation: The bclk rate must be
  * never greater than 1/5 IPG clock rate
 */
 if (freq * 5 > clk_get_rate(ssi->clk)) {
  dev_err(dai->dev, "bitclk > ipgclk / 5\n");
  return -EINVAL;
 }

 baudclk_is_used = ssi->baudclk_streams & ~(BIT(substream->stream));

 /* It should be already enough to divide clock by setting pm alone */
 psr = 0;
 div2 = 0;

 factor = (div2 + 1) * (7 * psr + 1) * 2;

 for (i = 0; i < 255; i++) {
  tmprate = freq * factor * (i + 1);

  if (baudclk_is_used)
   clkrate = clk_get_rate(ssi->baudclk);
  else
   clkrate = clk_round_rate(ssi->baudclk, tmprate);

  clkrate /= factor;
  afreq = clkrate / (i + 1);

  if (freq == afreq)
   sub = 0;
  else if (freq / afreq == 1)
   sub = freq - afreq;
  else if (afreq / freq == 1)
   sub = afreq - freq;
  else
   continue;

  /* Calculate the fraction */
  sub *= 100000;
  do_div(sub, freq);

  if (sub < savesub && !(i == 0)) {
   baudrate = tmprate;
   savesub = sub;
   pm = i;
  }

  /* We are lucky */
  if (savesub == 0)
   break;
 }

 /* No proper pm found if it is still remaining the initial value */
 if (pm == 999) {
  dev_err(dai->dev, "failed to handle the required sysclk\n");
  return -EINVAL;
 }

 stccr = SSI_SxCCR_PM(pm + 1);
 mask = SSI_SxCCR_PM_MASK | SSI_SxCCR_DIV2 | SSI_SxCCR_PSR;

 /* STCCR is used for RX in synchronous mode */
 tx2 = tx || ssi->synchronous;
 regmap_update_bits(regs, REG_SSI_SxCCR(tx2), mask, stccr);

 if (!baudclk_is_used) {
  ret = clk_set_rate(ssi->baudclk, baudrate);
  if (ret) {
   dev_err(dai->dev, "failed to set baudclk rate\n");
   return -EINVAL;
  }
 }

 return 0;
}

/**
 * fsl_ssi_hw_params - Configure SSI based on PCM hardware parameters
 * @substream: ASoC substream
 * @hw_params: pointers to hw_params
 * @dai: pointer to DAI
 *
 * Notes:
 * 1) SxCCR.WL bits are critical bits that require SSI to be temporarily
 *    disabled on offline_config SoCs. Even for online configurable SoCs
 *    running in synchronous mode (both TX and RX use STCCR), it is not
 *    safe to re-configure them when both two streams start running.
 * 2) SxCCR.PM, SxCCR.DIV2 and SxCCR.PSR bits will be configured in the
 *    fsl_ssi_set_bclk() if SSI is the DAI clock master.
 */
static int fsl_ssi_hw_params(struct snd_pcm_substream *substream,
        struct snd_pcm_hw_params *hw_params,
        struct snd_soc_dai *dai)
{
 bool tx2, tx = substream->stream == SNDRV_PCM_STREAM_PLAYBACK;
 struct fsl_ssi *ssi = snd_soc_dai_get_drvdata(dai);
 struct fsl_ssi_regvals *vals = ssi->regvals;
 struct regmap *regs = ssi->regs;
 unsigned int channels = params_channels(hw_params);
 unsigned int sample_size = params_width(hw_params);
 u32 wl = SSI_SxCCR_WL(sample_size);
 int ret;

 if (fsl_ssi_is_i2s_clock_provider(ssi)) {
  ret = fsl_ssi_set_bclk(substream, dai, hw_params);
  if (ret)
   return ret;

  /* Do not enable the clock if it is already enabled */
  if (!(ssi->baudclk_streams & BIT(substream->stream))) {
   ret = clk_prepare_enable(ssi->baudclk);
   if (ret)
    return ret;

   ssi->baudclk_streams |= BIT(substream->stream);
  }
 }

 /*
  * SSI is properly configured if it is enabled and running in
  * the synchronous mode; Note that AC97 mode is an exception
  * that should set separate configurations for STCCR and SRCCR
  * despite running in the synchronous mode.
 */
 if (ssi->streams && ssi->synchronous)
  return 0;

 if (!fsl_ssi_is_ac97(ssi)) {
  /*
   * Keep the ssi->i2s_net intact while having a local variable
   * to override settings for special use cases. Otherwise, the
   * ssi->i2s_net will lose the settings for regular use cases.
 */
  u8 i2s_net = ssi->i2s_net;

  /* Normal + Network mode to send 16-bit data in 32-bit frames */
  if (fsl_ssi_is_i2s_bc_fp(ssi) && sample_size == 16)
   i2s_net = SSI_SCR_I2S_MODE_NORMAL | SSI_SCR_NET;

  /* Use Normal mode to send mono data at 1st slot of 2 slots */
  if (channels == 1)
   i2s_net = SSI_SCR_I2S_MODE_NORMAL;

  regmap_update_bits(regs, REG_SSI_SCR,
       SSI_SCR_I2S_NET_MASK, i2s_net);
 }

 /* In synchronous mode, the SSI uses STCCR for capture */
 tx2 = tx || ssi->synchronous;
 regmap_update_bits(regs, REG_SSI_SxCCR(tx2), SSI_SxCCR_WL_MASK, wl);

 if (ssi->use_dyna_fifo) {
  if (channels == 1) {
   ssi->audio_config[0].n_fifos_dst = 1;
   ssi->audio_config[1].n_fifos_src = 1;
   vals[RX].srcr &= ~SSI_SRCR_RFEN1;
   vals[TX].stcr &= ~SSI_STCR_TFEN1;
   vals[RX].scr  &= ~SSI_SCR_TCH_EN;
   vals[TX].scr  &= ~SSI_SCR_TCH_EN;
  } else {
   ssi->audio_config[0].n_fifos_dst = 2;
   ssi->audio_config[1].n_fifos_src = 2;
   vals[RX].srcr |= SSI_SRCR_RFEN1;
   vals[TX].stcr |= SSI_STCR_TFEN1;
   vals[RX].scr  |= SSI_SCR_TCH_EN;
   vals[TX].scr  |= SSI_SCR_TCH_EN;
  }
  ssi->dma_params_tx.peripheral_config = &ssi->audio_config[0];
  ssi->dma_params_tx.peripheral_size = sizeof(ssi->audio_config[0]);
  ssi->dma_params_rx.peripheral_config = &ssi->audio_config[1];
  ssi->dma_params_rx.peripheral_size = sizeof(ssi->audio_config[1]);
 }

 return 0;
}

static int fsl_ssi_hw_free(struct snd_pcm_substream *substream,
      struct snd_soc_dai *dai)
{
 struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);
 struct fsl_ssi *ssi = snd_soc_dai_get_drvdata(snd_soc_rtd_to_cpu(rtd, 0));

 if (fsl_ssi_is_i2s_clock_provider(ssi) &&
     ssi->baudclk_streams & BIT(substream->stream)) {
  clk_disable_unprepare(ssi->baudclk);
  ssi->baudclk_streams &= ~BIT(substream->stream);
 }

 return 0;
}

static int _fsl_ssi_set_dai_fmt(struct fsl_ssi *ssi, unsigned int fmt)
{
 u32 strcr = 0, scr = 0, stcr, srcr, mask;
 unsigned int slots;

 ssi->dai_fmt = fmt;

 /* Synchronize frame sync clock for TE to avoid data slipping */
 scr |= SSI_SCR_SYNC_TX_FS;

 /* Set to default shifting settings: LSB_ALIGNED */
 strcr |= SSI_STCR_TXBIT0;

 /* Use Network mode as default */
 ssi->i2s_net = SSI_SCR_NET;
 switch (fmt & SND_SOC_DAIFMT_FORMAT_MASK) {
 case SND_SOC_DAIFMT_I2S:
  switch (fmt & SND_SOC_DAIFMT_CLOCK_PROVIDER_MASK) {
  case SND_SOC_DAIFMT_BP_FP:
   if (IS_ERR(ssi->baudclk)) {
    dev_err(ssi->dev,
     "missing baudclk for master mode\n");
    return -EINVAL;
   }
   fallthrough;
  case SND_SOC_DAIFMT_BC_FP:
   ssi->i2s_net |= SSI_SCR_I2S_MODE_MASTER;
   break;
  case SND_SOC_DAIFMT_BC_FC:
   ssi->i2s_net |= SSI_SCR_I2S_MODE_SLAVE;
   break;
  default:
   return -EINVAL;
  }

  slots = ssi->slots ? : 2;
  regmap_update_bits(ssi->regs, REG_SSI_STCCR,
       SSI_SxCCR_DC_MASK, SSI_SxCCR_DC(slots));
  regmap_update_bits(ssi->regs, REG_SSI_SRCCR,
       SSI_SxCCR_DC_MASK, SSI_SxCCR_DC(slots));

  /* Data on rising edge of bclk, frame low, 1clk before data */
  strcr |= SSI_STCR_TFSI | SSI_STCR_TSCKP | SSI_STCR_TEFS;
  break;
 case SND_SOC_DAIFMT_LEFT_J:
  /* Data on rising edge of bclk, frame high */
  strcr |= SSI_STCR_TSCKP;
  break;
 case SND_SOC_DAIFMT_DSP_A:
  /* Data on rising edge of bclk, frame high, 1clk before data */
  strcr |= SSI_STCR_TFSL | SSI_STCR_TSCKP | SSI_STCR_TEFS;
  break;
 case SND_SOC_DAIFMT_DSP_B:
  /* Data on rising edge of bclk, frame high */
  strcr |= SSI_STCR_TFSL | SSI_STCR_TSCKP;
  break;
 case SND_SOC_DAIFMT_AC97:
  /* Data on falling edge of bclk, frame high, 1clk before data */
  strcr |= SSI_STCR_TEFS;
  break;
 default:
  return -EINVAL;
 }

 scr |= ssi->i2s_net;

 /* DAI clock inversion */
 switch (fmt & SND_SOC_DAIFMT_INV_MASK) {
 case SND_SOC_DAIFMT_NB_NF:
  /* Nothing to do for both normal cases */
  break;
 case SND_SOC_DAIFMT_IB_NF:
  /* Invert bit clock */
  strcr ^= SSI_STCR_TSCKP;
  break;
 case SND_SOC_DAIFMT_NB_IF:
  /* Invert frame clock */
  strcr ^= SSI_STCR_TFSI;
  break;
 case SND_SOC_DAIFMT_IB_IF:
  /* Invert both clocks */
  strcr ^= SSI_STCR_TSCKP;
  strcr ^= SSI_STCR_TFSI;
  break;
 default:
  return -EINVAL;
 }

 /* DAI clock provider masks */
 switch (fmt & SND_SOC_DAIFMT_CLOCK_PROVIDER_MASK) {
 case SND_SOC_DAIFMT_BP_FP:
  /* Output bit and frame sync clocks */
  strcr |= SSI_STCR_TFDIR | SSI_STCR_TXDIR;
  scr |= SSI_SCR_SYS_CLK_EN;
  break;
 case SND_SOC_DAIFMT_BC_FC:
  /* Input bit or frame sync clocks */
  break;
 case SND_SOC_DAIFMT_BC_FP:
  /* Input bit clock but output frame sync clock */
  strcr |= SSI_STCR_TFDIR;
  break;
 default:
  return -EINVAL;
 }

 stcr = strcr;
 srcr = strcr;

 /* Set SYN mode and clear RXDIR bit when using SYN or AC97 mode */
 if (ssi->synchronous || fsl_ssi_is_ac97(ssi)) {
  srcr &= ~SSI_SRCR_RXDIR;
  scr |= SSI_SCR_SYN;
 }

 mask = SSI_STCR_TFDIR | SSI_STCR_TXDIR | SSI_STCR_TSCKP |
        SSI_STCR_TFSL | SSI_STCR_TFSI | SSI_STCR_TEFS | SSI_STCR_TXBIT0;

 regmap_update_bits(ssi->regs, REG_SSI_STCR, mask, stcr);
 regmap_update_bits(ssi->regs, REG_SSI_SRCR, mask, srcr);

 mask = SSI_SCR_SYNC_TX_FS | SSI_SCR_I2S_MODE_MASK |
        SSI_SCR_SYS_CLK_EN | SSI_SCR_SYN;
 regmap_update_bits(ssi->regs, REG_SSI_SCR, mask, scr);

 return 0;
}

/**
 * fsl_ssi_set_dai_fmt - Configure Digital Audio Interface (DAI) Format
 * @dai: pointer to DAI
 * @fmt: format mask
 */
static int fsl_ssi_set_dai_fmt(struct snd_soc_dai *dai, unsigned int fmt)
{
 struct fsl_ssi *ssi = snd_soc_dai_get_drvdata(dai);

 /* AC97 configured DAIFMT earlier in the probe() */
 if (fsl_ssi_is_ac97(ssi))
  return 0;

 return _fsl_ssi_set_dai_fmt(ssi, fmt);
}

/**
 * fsl_ssi_set_dai_tdm_slot - Set TDM slot number and slot width
 * @dai: pointer to DAI
 * @tx_mask: mask for TX
 * @rx_mask: mask for RX
 * @slots: number of slots
 * @slot_width: number of bits per slot
 */
static int fsl_ssi_set_dai_tdm_slot(struct snd_soc_dai *dai, u32 tx_mask,
        u32 rx_mask, int slots, int slot_width)
{
 struct fsl_ssi *ssi = snd_soc_dai_get_drvdata(dai);
 struct regmap *regs = ssi->regs;
 u32 val;

 /* The word length should be 8, 10, 12, 16, 18, 20, 22 or 24 */
 if (slot_width & 1 || slot_width < 8 || slot_width > 24) {
  dev_err(dai->dev, "invalid slot width: %d\n", slot_width);
  return -EINVAL;
 }

 /* The slot number should be >= 2 if using Network mode or I2S mode */
 if (ssi->i2s_net && slots < 2) {
  dev_err(dai->dev, "slot number should be >= 2 in I2S or NET\n");
  return -EINVAL;
 }

 regmap_update_bits(regs, REG_SSI_STCCR,
      SSI_SxCCR_DC_MASK, SSI_SxCCR_DC(slots));
 regmap_update_bits(regs, REG_SSI_SRCCR,
      SSI_SxCCR_DC_MASK, SSI_SxCCR_DC(slots));

 /* Save the SCR register value */
 regmap_read(regs, REG_SSI_SCR, &val);
 /* Temporarily enable SSI to allow SxMSKs to be configurable */
 regmap_update_bits(regs, REG_SSI_SCR, SSI_SCR_SSIEN, SSI_SCR_SSIEN);

 regmap_write(regs, REG_SSI_STMSK, ~tx_mask);
 regmap_write(regs, REG_SSI_SRMSK, ~rx_mask);

 /* Restore the value of SSIEN bit */
 regmap_update_bits(regs, REG_SSI_SCR, SSI_SCR_SSIEN, val);

 ssi->slot_width = slot_width;
 ssi->slots = slots;

 return 0;
}

/**
 * fsl_ssi_trigger - Start or stop SSI and corresponding DMA transaction.
 * @substream: ASoC substream
 * @cmd: trigger command
 * @dai: pointer to DAI
 *
 * The DMA channel is in external master start and pause mode, which
 * means the SSI completely controls the flow of data.
 */
static int fsl_ssi_trigger(struct snd_pcm_substream *substream, int cmd,
      struct snd_soc_dai *dai)
{
 struct snd_soc_pcm_runtime *rtd = snd_soc_substream_to_rtd(substream);
 struct fsl_ssi *ssi = snd_soc_dai_get_drvdata(snd_soc_rtd_to_cpu(rtd, 0));
 bool tx = substream->stream == SNDRV_PCM_STREAM_PLAYBACK;

 switch (cmd) {
 case SNDRV_PCM_TRIGGER_START:
 case SNDRV_PCM_TRIGGER_RESUME:
 case SNDRV_PCM_TRIGGER_PAUSE_RELEASE:
  /*
   * SACCST might be modified via AC Link by a CODEC if it sends
   * extra bits in their SLOTREQ requests, which'll accidentally
   * send valid data to slots other than normal playback slots.
   *
   * To be safe, configure SACCST right before TX starts.
 */
  if (tx && fsl_ssi_is_ac97(ssi))
   fsl_ssi_tx_ac97_saccst_setup(ssi);
  fsl_ssi_config_enable(ssi, tx);
  break;

 case SNDRV_PCM_TRIGGER_STOP:
 case SNDRV_PCM_TRIGGER_SUSPEND:
 case SNDRV_PCM_TRIGGER_PAUSE_PUSH:
  fsl_ssi_config_disable(ssi, tx);
  break;

 default:
  return -EINVAL;
 }

 return 0;
}

static int fsl_ssi_dai_probe(struct snd_soc_dai *dai)
{
 struct fsl_ssi *ssi = snd_soc_dai_get_drvdata(dai);

 if (ssi->soc->imx && ssi->use_dma)
  snd_soc_dai_init_dma_data(dai, &ssi->dma_params_tx,
       &ssi->dma_params_rx);

 return 0;
}

static const struct snd_soc_dai_ops fsl_ssi_dai_ops = {
 .probe = fsl_ssi_dai_probe,
 .startup = fsl_ssi_startup,
 .shutdown = fsl_ssi_shutdown,
 .hw_params = fsl_ssi_hw_params,
 .hw_free = fsl_ssi_hw_free,
 .set_fmt = fsl_ssi_set_dai_fmt,
 .set_tdm_slot = fsl_ssi_set_dai_tdm_slot,
 .trigger = fsl_ssi_trigger,
};

static struct snd_soc_dai_driver fsl_ssi_dai_template = {
 .playback = {
  .stream_name = "CPU-Playback",
  .channels_min = 1,
  .channels_max = 32,
  .rates = SNDRV_PCM_RATE_CONTINUOUS,
  .formats = FSLSSI_I2S_FORMATS,
 },
 .capture = {
  .stream_name = "CPU-Capture",
  .channels_min = 1,
  .channels_max = 32,
  .rates = SNDRV_PCM_RATE_CONTINUOUS,
  .formats = FSLSSI_I2S_FORMATS,
 },
 .ops = &fsl_ssi_dai_ops,
};

static const struct snd_soc_component_driver fsl_ssi_component = {
 .name = "fsl-ssi",
 .legacy_dai_naming = 1,
};

static struct snd_soc_dai_driver fsl_ssi_ac97_dai = {
 .symmetric_channels = 1,
 .playback = {
  .stream_name = "CPU AC97 Playback",
  .channels_min = 2,
  .channels_max = 2,
  .rates = SNDRV_PCM_RATE_8000_48000,
  .formats = SNDRV_PCM_FMTBIT_S16 | SNDRV_PCM_FMTBIT_S20,
 },
 .capture = {
  .stream_name = "CPU AC97 Capture",
  .channels_min = 2,
  .channels_max = 2,
  .rates = SNDRV_PCM_RATE_48000,
  /* 16-bit capture is broken (errata ERR003778) */
  .formats = SNDRV_PCM_FMTBIT_S20,
 },
 .ops = &fsl_ssi_dai_ops,
};

static struct fsl_ssi *fsl_ac97_data;

static void fsl_ssi_ac97_write(struct snd_ac97 *ac97, unsigned short reg,
          unsigned short val)
{
 struct regmap *regs = fsl_ac97_data->regs;
 unsigned int lreg;
 unsigned int lval;
 int ret;

 if (reg > 0x7f)
  return;

 mutex_lock(&fsl_ac97_data->ac97_reg_lock);

 ret = clk_prepare_enable(fsl_ac97_data->clk);
 if (ret) {
  pr_err("ac97 write clk_prepare_enable failed: %d\n",
   ret);
  goto ret_unlock;
 }

 lreg = reg <<  12;
 regmap_write(regs, REG_SSI_SACADD, lreg);

 lval = val << 4;
 regmap_write(regs, REG_SSI_SACDAT, lval);

 regmap_update_bits(regs, REG_SSI_SACNT,
      SSI_SACNT_RDWR_MASK, SSI_SACNT_WR);
 udelay(100);

 clk_disable_unprepare(fsl_ac97_data->clk);

ret_unlock:
 mutex_unlock(&fsl_ac97_data->ac97_reg_lock);
}

static unsigned short fsl_ssi_ac97_read(struct snd_ac97 *ac97,
     unsigned short reg)
{
 struct regmap *regs = fsl_ac97_data->regs;
 unsigned short val = 0;
 u32 reg_val;
 unsigned int lreg;
 int ret;

 mutex_lock(&fsl_ac97_data->ac97_reg_lock);

 ret = clk_prepare_enable(fsl_ac97_data->clk);
 if (ret) {
  pr_err("ac97 read clk_prepare_enable failed: %d\n", ret);
  goto ret_unlock;
 }

 lreg = (reg & 0x7f) <<  12;
 regmap_write(regs, REG_SSI_SACADD, lreg);
 regmap_update_bits(regs, REG_SSI_SACNT,
      SSI_SACNT_RDWR_MASK, SSI_SACNT_RD);

 udelay(100);

 regmap_read(regs, REG_SSI_SACDAT, ®_val);
 val = (reg_val >> 4) & 0xffff;

 clk_disable_unprepare(fsl_ac97_data->clk);

ret_unlock:
 mutex_unlock(&fsl_ac97_data->ac97_reg_lock);
 return val;
}

static struct snd_ac97_bus_ops fsl_ssi_ac97_ops = {
 .read = fsl_ssi_ac97_read,
 .write = fsl_ssi_ac97_write,
};

/**
 * fsl_ssi_hw_init - Initialize SSI registers
 * @ssi: SSI context
 */
static int fsl_ssi_hw_init(struct fsl_ssi *ssi)
{
 u32 wm = ssi->fifo_watermark;

 /* Initialize regvals */
 fsl_ssi_setup_regvals(ssi);

 /* Set watermarks */
 regmap_write(ssi->regs, REG_SSI_SFCSR,
       SSI_SFCSR_TFWM0(wm) | SSI_SFCSR_RFWM0(wm) |
       SSI_SFCSR_TFWM1(wm) | SSI_SFCSR_RFWM1(wm));

 /* Enable Dual FIFO mode */
 if (ssi->use_dual_fifo)
  regmap_update_bits(ssi->regs, REG_SSI_SCR,
       SSI_SCR_TCH_EN, SSI_SCR_TCH_EN);

 /* AC97 should start earlier to communicate with CODECs */
 if (fsl_ssi_is_ac97(ssi)) {
  _fsl_ssi_set_dai_fmt(ssi, ssi->dai_fmt);
  fsl_ssi_setup_ac97(ssi);
 }

 return 0;
}

/**
 * fsl_ssi_hw_clean - Clear SSI registers
 * @ssi: SSI context
 */
static void fsl_ssi_hw_clean(struct fsl_ssi *ssi)
{
 /* Disable registers for AC97 */
 if (fsl_ssi_is_ac97(ssi)) {
  /* Disable TE and RE bits first */
  regmap_update_bits(ssi->regs, REG_SSI_SCR,
       SSI_SCR_TE | SSI_SCR_RE, 0);
  /* Disable AC97 mode */
  regmap_write(ssi->regs, REG_SSI_SACNT, 0);
  /* Unset WAIT bits */
  regmap_write(ssi->regs, REG_SSI_SOR, 0);
  /* Disable SSI -- software reset */
  regmap_update_bits(ssi->regs, REG_SSI_SCR, SSI_SCR_SSIEN, 0);
 }
}

/*
 * Make every character in a string lower-case
 */
static void make_lowercase(char *s)
{
 if (!s)
  return;
 for (; *s; s++)
  *s = tolower(*s);
}

static int fsl_ssi_imx_probe(struct platform_device *pdev,
        struct fsl_ssi *ssi, void __iomem *iomem)
{
 struct device *dev = &pdev->dev;
 int ret;

 /* Backward compatible for a DT without ipg clock name assigned */
 if (ssi->has_ipg_clk_name)
  ssi->clk = devm_clk_get(dev, "ipg");
 else
  ssi->clk = devm_clk_get(dev, NULL);
 if (IS_ERR(ssi->clk)) {
  ret = PTR_ERR(ssi->clk);
  dev_err(dev, "failed to get clock: %d\n", ret);
  return ret;
 }

 /* Enable the clock since regmap will not handle it in this case */
 if (!ssi->has_ipg_clk_name) {
  ret = clk_prepare_enable(ssi->clk);
  if (ret) {
   dev_err(dev, "clk_prepare_enable failed: %d\n", ret);
   return ret;
  }
 }

 /* Do not error out for consumer cases that live without a baud clock */
 ssi->baudclk = devm_clk_get(dev, "baud");
 if (IS_ERR(ssi->baudclk))
  dev_dbg(dev, "failed to get baud clock: %ld\n",
    PTR_ERR(ssi->baudclk));

 ssi->dma_params_tx.maxburst = ssi->dma_maxburst;
 ssi->dma_params_rx.maxburst = ssi->dma_maxburst;
 ssi->dma_params_tx.addr = ssi->ssi_phys + REG_SSI_STX0;
 ssi->dma_params_rx.addr = ssi->ssi_phys + REG_SSI_SRX0;

 /* Use even numbers to avoid channel swap due to SDMA script design */
 if (ssi->use_dual_fifo || ssi->use_dyna_fifo) {
  ssi->dma_params_tx.maxburst &= ~0x1;
  ssi->dma_params_rx.maxburst &= ~0x1;
 }

 if (!ssi->use_dma) {
  /*
   * Some boards use an incompatible codec. Use imx-fiq-pcm-audio
   * to get it working, as DMA is not possible in this situation.
 */
  ssi->fiq_params.irq = ssi->irq;
  ssi->fiq_params.base = iomem;
  ssi->fiq_params.dma_params_rx = &ssi->dma_params_rx;
  ssi->fiq_params.dma_params_tx = &ssi->dma_params_tx;

  ret = imx_pcm_fiq_init(pdev, &ssi->fiq_params);
  if (ret)
   goto error_pcm;
 } else {
  ret = imx_pcm_dma_init(pdev);
  if (ret) {
   dev_err_probe(dev, ret, "Failed to init PCM DMA\n");
   goto error_pcm;
  }
 }

 return 0;

error_pcm:
 if (!ssi->has_ipg_clk_name)
  clk_disable_unprepare(ssi->clk);

 return ret;
}

static void fsl_ssi_imx_clean(struct platform_device *pdev, struct fsl_ssi *ssi)
{
 if (!ssi->use_dma)
  imx_pcm_fiq_exit(pdev);
 if (!ssi->has_ipg_clk_name)
  clk_disable_unprepare(ssi->clk);
}

static int fsl_ssi_probe_from_dt(struct fsl_ssi *ssi)
{
 struct device *dev = ssi->dev;
 struct device_node *np = dev->of_node;
 const char *p, *sprop;
 const __be32 *iprop;
 u32 dmas[4];
 int ret;

 ret = of_property_match_string(np, "clock-names", "ipg");
 /* Get error code if not found */
 ssi->has_ipg_clk_name = ret >= 0;

 /* Check if being used in AC97 mode */
 sprop = of_get_property(np, "fsl,mode", NULL);
 if (sprop && !strcmp(sprop, "ac97-slave")) {
  ssi->dai_fmt = FSLSSI_AC97_DAIFMT;

  ret = of_property_read_u32(np, "cell-index", &ssi->card_idx);
  if (ret) {
   dev_err(dev, "failed to get SSI index property\n");
   return -EINVAL;
  }
  strcpy(ssi->card_name, "ac97-codec");
 } else if (!of_property_read_bool(np, "fsl,ssi-asynchronous")) {
  /*
   * In synchronous mode, STCK and STFS ports are used by RX
   * as well. So the software should limit the sample rates,
   * sample bits and channels to be symmetric.
   *
   * This is exclusive with FSLSSI_AC97_FORMATS as AC97 runs
   * in the SSI synchronous mode however it does not have to
   * limit symmetric sample rates and sample bits.
 */
  ssi->synchronous = true;
 }

 /* Select DMA or FIQ */
 ssi->use_dma = !of_property_read_bool(np, "fsl,fiq-stream-filter");

 /* Fetch FIFO depth; Set to 8 for older DT without this property */
 iprop = of_get_property(np, "fsl,fifo-depth", NULL);
 if (iprop)
  ssi->fifo_depth = be32_to_cpup(iprop);
 else
  ssi->fifo_depth = 8;

 /* Use dual FIFO mode depending on the support from SDMA script */
 ret = of_property_read_u32_array(np, "dmas", dmas, 4);
 if (ssi->use_dma && !ret && dmas[2] == IMX_DMATYPE_SSI_DUAL)
  ssi->use_dual_fifo = true;

 if (ssi->use_dma && !ret && dmas[2] == IMX_DMATYPE_MULTI_SAI)
  ssi->use_dyna_fifo = true;
 /*
  * Backward compatible for older bindings by manually triggering the
  * machine driver's probe(). Use /compatible property, including the
  * address of CPU DAI driver structure, as the name of machine driver
  *
  * If card_name is set by AC97 earlier, bypass here since it uses a
  * different name to register the device.
 */
 if (!ssi->card_name[0] && of_get_property(np, "codec-handle", NULL)) {
  struct device_node *root = of_find_node_by_path("/");

  sprop = of_get_property(root, "compatible", NULL);
  of_node_put(root);
  /* Strip "fsl," in the compatible name if applicable */
  p = strrchr(sprop, ',');
  if (p)
   sprop = p + 1;
  snprintf(ssi->card_name, sizeof(ssi->card_name),
    "snd-soc-%s", sprop);
  make_lowercase(ssi->card_name);
  ssi->card_idx = 0;
 }

 return 0;
}

static int fsl_ssi_probe(struct platform_device *pdev)
{
 struct regmap_config regconfig = fsl_ssi_regconfig;
 struct device *dev = &pdev->dev;
 struct fsl_ssi *ssi;
 struct resource *res;
 void __iomem *iomem;
 int ret = 0;

 ssi = devm_kzalloc(dev, sizeof(*ssi), GFP_KERNEL);
 if (!ssi)
  return -ENOMEM;

 ssi->dev = dev;
 ssi->soc = of_device_get_match_data(&pdev->dev);

 /* Probe from DT */
 ret = fsl_ssi_probe_from_dt(ssi);
 if (ret)
  return ret;

 if (fsl_ssi_is_ac97(ssi)) {
  memcpy(&ssi->cpu_dai_drv, &fsl_ssi_ac97_dai,
         sizeof(fsl_ssi_ac97_dai));
  fsl_ac97_data = ssi;
 } else {
  memcpy(&ssi->cpu_dai_drv, &fsl_ssi_dai_template,
         sizeof(fsl_ssi_dai_template));
 }
 ssi->cpu_dai_drv.name = dev_name(dev);

 iomem = devm_platform_get_and_ioremap_resource(pdev, 0, &res);
 if (IS_ERR(iomem))
  return PTR_ERR(iomem);
 ssi->ssi_phys = res->start;

 if (ssi->soc->imx21regs) {
  /* No SACC{ST,EN,DIS} regs in imx21-class SSI */
  regconfig.max_register = REG_SSI_SRMSK;
  regconfig.num_reg_defaults_raw =
   REG_SSI_SRMSK / sizeof(uint32_t) + 1;
 }

 if (ssi->has_ipg_clk_name)
  ssi->regs = devm_regmap_init_mmio_clk(dev, "ipg", iomem,
            ®config);
 else
  ssi->regs = devm_regmap_init_mmio(dev, iomem, ®config);
 if (IS_ERR(ssi->regs)) {
  dev_err(dev, "failed to init register map\n");
  return PTR_ERR(ssi->regs);
 }

 ssi->irq = platform_get_irq(pdev, 0);
 if (ssi->irq < 0)
  return ssi->irq;

 /* Set software limitations for synchronous mode except AC97 */
 if (ssi->synchronous && !fsl_ssi_is_ac97(ssi)) {
  ssi->cpu_dai_drv.symmetric_rate = 1;
  ssi->cpu_dai_drv.symmetric_channels = 1;
  ssi->cpu_dai_drv.symmetric_sample_bits = 1;
 }

 /*
  * Configure TX and RX DMA watermarks -- when to send a DMA request
  *
  * Values should be tested to avoid FIFO under/over run. Set maxburst
  * to fifo_watermark to maxiumize DMA transaction to reduce overhead.
 */
 switch (ssi->fifo_depth) {
 case 15:
  /*
   * Set to 8 as a balanced configuration -- When TX FIFO has 8
   * empty slots, send a DMA request to fill these 8 slots. The
   * remaining 7 slots should be able to allow DMA to finish the
   * transaction before TX FIFO underruns; Same applies to RX.
   *
   * Tested with cases running at 48kHz @ 16 bits x 16 channels
 */
  ssi->fifo_watermark = 8;
  ssi->dma_maxburst = 8;
  break;
 case 8:
 default:
  /* Safely use old watermark configurations for older chips */
  ssi->fifo_watermark = ssi->fifo_depth - 2;
  ssi->dma_maxburst = ssi->fifo_depth - 2;
  break;
 }

 dev_set_drvdata(dev, ssi);

 if (ssi->soc->imx) {
  ret = fsl_ssi_imx_probe(pdev, ssi, iomem);
  if (ret)
   return ret;
 }

 if (fsl_ssi_is_ac97(ssi)) {
  mutex_init(&ssi->ac97_reg_lock);
  ret = snd_soc_set_ac97_ops_of_reset(&fsl_ssi_ac97_ops, pdev);
  if (ret) {
   dev_err(dev, "failed to set AC'97 ops\n");
   goto error_ac97_ops;
  }
 }

 ret = devm_snd_soc_register_component(dev, &fsl_ssi_component,
           &ssi->cpu_dai_drv, 1);
 if (ret) {
  dev_err(dev, "failed to register DAI: %d\n", ret);
  goto error_asoc_register;
 }

 if (ssi->use_dma) {
  ret = devm_request_irq(dev, ssi->irq, fsl_ssi_isr, 0,
           dev_name(dev), ssi);
  if (ret < 0) {
   dev_err(dev, "failed to claim irq %u\n", ssi->irq);
   goto error_asoc_register;
  }
 }

 fsl_ssi_debugfs_create(&ssi->dbg_stats, dev);

 /* Initially configures SSI registers */
 fsl_ssi_hw_init(ssi);

 /* Register a platform device for older bindings or AC97 */
 if (ssi->card_name[0]) {
  struct device *parent = dev;
  /*
   * Do not set SSI dev as the parent of AC97 CODEC device since
   * it does not have a DT node. Otherwise ASoC core will assume
   * CODEC has the same DT node as the SSI, so it may bypass the
   * dai_probe() of SSI and then cause NULL DMA data pointers.
 */
  if (fsl_ssi_is_ac97(ssi))
   parent = NULL;

  ssi->card_pdev = platform_device_register_data(parent,
    ssi->card_name, ssi->card_idx, NULL, 0);
  if (IS_ERR(ssi->card_pdev)) {
   ret = PTR_ERR(ssi->card_pdev);
   dev_err(dev, "failed to register %s: %d\n",
    ssi->card_name, ret);
   goto error_sound_card;
  }
 }

 return 0;

error_sound_card:
 fsl_ssi_debugfs_remove(&ssi->dbg_stats);
error_asoc_register:
 if (fsl_ssi_is_ac97(ssi))
  snd_soc_set_ac97_ops(NULL);
error_ac97_ops:
 if (fsl_ssi_is_ac97(ssi))
  mutex_destroy(&ssi->ac97_reg_lock);

 if (ssi->soc->imx)
  fsl_ssi_imx_clean(pdev, ssi);

 return ret;
}

static void fsl_ssi_remove(struct platform_device *pdev)
{
 struct fsl_ssi *ssi = dev_get_drvdata(&pdev->dev);

 fsl_ssi_debugfs_remove(&ssi->dbg_stats);

 if (ssi->card_pdev)
  platform_device_unregister(ssi->card_pdev);

 /* Clean up SSI registers */
 fsl_ssi_hw_clean(ssi);

 if (ssi->soc->imx)
  fsl_ssi_imx_clean(pdev, ssi);

 if (fsl_ssi_is_ac97(ssi)) {
  snd_soc_set_ac97_ops(NULL);
  mutex_destroy(&ssi->ac97_reg_lock);
 }
}

static int fsl_ssi_suspend(struct device *dev)
{
 struct fsl_ssi *ssi = dev_get_drvdata(dev);
 struct regmap *regs = ssi->regs;

 regmap_read(regs, REG_SSI_SFCSR, &ssi->regcache_sfcsr);
 regmap_read(regs, REG_SSI_SACNT, &ssi->regcache_sacnt);

 regcache_cache_only(regs, true);
 regcache_mark_dirty(regs);

 return 0;
}

static int fsl_ssi_resume(struct device *dev)
{
 struct fsl_ssi *ssi = dev_get_drvdata(dev);
 struct regmap *regs = ssi->regs;

 regcache_cache_only(regs, false);

 regmap_update_bits(regs, REG_SSI_SFCSR,
      SSI_SFCSR_RFWM1_MASK | SSI_SFCSR_TFWM1_MASK |
      SSI_SFCSR_RFWM0_MASK | SSI_SFCSR_TFWM0_MASK,
      ssi->regcache_sfcsr);
 regmap_write(regs, REG_SSI_SACNT, ssi->regcache_sacnt);

 return regcache_sync(regs);
}

static const struct dev_pm_ops fsl_ssi_pm = {
 SYSTEM_SLEEP_PM_OPS(fsl_ssi_suspend, fsl_ssi_resume)
};

static struct platform_driver fsl_ssi_driver = {
 .driver = {
  .name = "fsl-ssi-dai",
  .of_match_table = fsl_ssi_ids,
  .pm = pm_sleep_ptr(&fsl_ssi_pm),
 },
 .probe = fsl_ssi_probe,
 .remove = fsl_ssi_remove,
};

module_platform_driver(fsl_ssi_driver);

MODULE_ALIAS("platform:fsl-ssi-dai");
MODULE_AUTHOR("Timur Tabi <timur@freescale.com>");
MODULE_DESCRIPTION("Freescale Synchronous Serial Interface (SSI) ASoC Driver");
MODULE_LICENSE("GPL v2");