spi/imx: add device tree probe support
[pandora-kernel.git] / drivers / spi / spi-imx.c
index 69d6dba..8ac6542 100644 (file)
@@ -34,6 +34,9 @@
 #include <linux/spi/spi.h>
 #include <linux/spi/spi_bitbang.h>
 #include <linux/types.h>
+#include <linux/of.h>
+#include <linux/of_device.h>
+#include <linux/of_gpio.h>
 
 #include <mach/spi.h>
 
@@ -45,9 +48,6 @@
 #define MXC_CSPIINT            0x0c
 #define MXC_RESET              0x1c
 
-#define MX3_CSPISTAT           0x14
-#define MX3_CSPISTAT_RR                (1 << 3)
-
 /* generic defines to abstract from the different register layouts */
 #define MXC_INT_RR     (1 << 0) /* Receive data ready interrupt */
 #define MXC_INT_TE     (1 << 1) /* Transmit FIFO empty interrupt */
@@ -60,12 +60,12 @@ struct spi_imx_config {
 };
 
 enum spi_imx_devtype {
-       SPI_IMX_VER_IMX1,
-       SPI_IMX_VER_0_0,
-       SPI_IMX_VER_0_4,
-       SPI_IMX_VER_0_5,
-       SPI_IMX_VER_0_7,
-       SPI_IMX_VER_2_3,
+       IMX1_CSPI,
+       IMX21_CSPI,
+       IMX27_CSPI,
+       IMX31_CSPI,
+       IMX35_CSPI,     /* CSPI on all i.mx except above */
+       IMX51_ECSPI,    /* ECSPI on i.mx51 and later */
 };
 
 struct spi_imx_data;
@@ -76,7 +76,7 @@ struct spi_imx_devtype_data {
        void (*trigger)(struct spi_imx_data *);
        int (*rx_available)(struct spi_imx_data *);
        void (*reset)(struct spi_imx_data *);
-       unsigned int fifosize;
+       enum spi_imx_devtype devtype;
 };
 
 struct spi_imx_data {
@@ -87,7 +87,6 @@ struct spi_imx_data {
        int irq;
        struct clk *clk;
        unsigned long spi_clk;
-       int *chipselect;
 
        unsigned int count;
        void (*tx)(struct spi_imx_data *);
@@ -96,9 +95,25 @@ struct spi_imx_data {
        const void *tx_buf;
        unsigned int txfifo; /* number of words pushed in tx FIFO */
 
-       struct spi_imx_devtype_data devtype_data;
+       struct spi_imx_devtype_data *devtype_data;
+       int chipselect[0];
 };
 
+static inline int is_imx27_cspi(struct spi_imx_data *d)
+{
+       return d->devtype_data->devtype == IMX27_CSPI;
+}
+
+static inline int is_imx35_cspi(struct spi_imx_data *d)
+{
+       return d->devtype_data->devtype == IMX35_CSPI;
+}
+
+static inline unsigned spi_imx_get_fifosize(struct spi_imx_data *d)
+{
+       return (d->devtype_data->devtype == IMX51_ECSPI) ? 64 : 8;
+}
+
 #define MXC_SPI_BUF_RX(type)                                           \
 static void spi_imx_buf_rx_##type(struct spi_imx_data *spi_imx)                \
 {                                                                      \
@@ -140,14 +155,9 @@ static int mxc_clkdivs[] = {0, 3, 4, 6, 8, 12, 16, 24, 32, 48, 64, 96, 128, 192,
 
 /* MX21, MX27 */
 static unsigned int spi_imx_clkdiv_1(unsigned int fin,
-               unsigned int fspi)
+               unsigned int fspi, unsigned int max)
 {
-       int i, max;
-
-       if (cpu_is_mx21())
-               max = 18;
-       else
-               max = 16;
+       int i;
 
        for (i = 2; i < max; i++)
                if (fspi * mxc_clkdivs[i] >= fin)
@@ -171,30 +181,30 @@ static unsigned int spi_imx_clkdiv_2(unsigned int fin,
        return 7;
 }
 
-#define SPI_IMX2_3_CTRL                0x08
-#define SPI_IMX2_3_CTRL_ENABLE         (1 <<  0)
-#define SPI_IMX2_3_CTRL_XCH            (1 <<  2)
-#define SPI_IMX2_3_CTRL_MODE_MASK      (0xf << 4)
-#define SPI_IMX2_3_CTRL_POSTDIV_OFFSET 8
-#define SPI_IMX2_3_CTRL_PREDIV_OFFSET  12
-#define SPI_IMX2_3_CTRL_CS(cs)         ((cs) << 18)
-#define SPI_IMX2_3_CTRL_BL_OFFSET      20
+#define MX51_ECSPI_CTRL                0x08
+#define MX51_ECSPI_CTRL_ENABLE         (1 <<  0)
+#define MX51_ECSPI_CTRL_XCH            (1 <<  2)
+#define MX51_ECSPI_CTRL_MODE_MASK      (0xf << 4)
+#define MX51_ECSPI_CTRL_POSTDIV_OFFSET 8
+#define MX51_ECSPI_CTRL_PREDIV_OFFSET  12
+#define MX51_ECSPI_CTRL_CS(cs)         ((cs) << 18)
+#define MX51_ECSPI_CTRL_BL_OFFSET      20
 
-#define SPI_IMX2_3_CONFIG      0x0c
-#define SPI_IMX2_3_CONFIG_SCLKPHA(cs)  (1 << ((cs) +  0))
-#define SPI_IMX2_3_CONFIG_SCLKPOL(cs)  (1 << ((cs) +  4))
-#define SPI_IMX2_3_CONFIG_SBBCTRL(cs)  (1 << ((cs) +  8))
-#define SPI_IMX2_3_CONFIG_SSBPOL(cs)   (1 << ((cs) + 12))
+#define MX51_ECSPI_CONFIG      0x0c
+#define MX51_ECSPI_CONFIG_SCLKPHA(cs)  (1 << ((cs) +  0))
+#define MX51_ECSPI_CONFIG_SCLKPOL(cs)  (1 << ((cs) +  4))
+#define MX51_ECSPI_CONFIG_SBBCTRL(cs)  (1 << ((cs) +  8))
+#define MX51_ECSPI_CONFIG_SSBPOL(cs)   (1 << ((cs) + 12))
 
-#define SPI_IMX2_3_INT         0x10
-#define SPI_IMX2_3_INT_TEEN            (1 <<  0)
-#define SPI_IMX2_3_INT_RREN            (1 <<  3)
+#define MX51_ECSPI_INT         0x10
+#define MX51_ECSPI_INT_TEEN            (1 <<  0)
+#define MX51_ECSPI_INT_RREN            (1 <<  3)
 
-#define SPI_IMX2_3_STAT                0x18
-#define SPI_IMX2_3_STAT_RR             (1 <<  3)
+#define MX51_ECSPI_STAT                0x18
+#define MX51_ECSPI_STAT_RR             (1 <<  3)
 
 /* MX51 eCSPI */
-static unsigned int spi_imx2_3_clkdiv(unsigned int fin, unsigned int fspi)
+static unsigned int mx51_ecspi_clkdiv(unsigned int fin, unsigned int fspi)
 {
        /*
         * there are two 4-bit dividers, the pre-divider divides by
@@ -222,36 +232,36 @@ static unsigned int spi_imx2_3_clkdiv(unsigned int fin, unsigned int fspi)
 
        pr_debug("%s: fin: %u, fspi: %u, post: %u, pre: %u\n",
                        __func__, fin, fspi, post, pre);
-       return (pre << SPI_IMX2_3_CTRL_PREDIV_OFFSET) |
-               (post << SPI_IMX2_3_CTRL_POSTDIV_OFFSET);
+       return (pre << MX51_ECSPI_CTRL_PREDIV_OFFSET) |
+               (post << MX51_ECSPI_CTRL_POSTDIV_OFFSET);
 }
 
-static void __maybe_unused spi_imx2_3_intctrl(struct spi_imx_data *spi_imx, int enable)
+static void __maybe_unused mx51_ecspi_intctrl(struct spi_imx_data *spi_imx, int enable)
 {
        unsigned val = 0;
 
        if (enable & MXC_INT_TE)
-               val |= SPI_IMX2_3_INT_TEEN;
+               val |= MX51_ECSPI_INT_TEEN;
 
        if (enable & MXC_INT_RR)
-               val |= SPI_IMX2_3_INT_RREN;
+               val |= MX51_ECSPI_INT_RREN;
 
-       writel(val, spi_imx->base + SPI_IMX2_3_INT);
+       writel(val, spi_imx->base + MX51_ECSPI_INT);
 }
 
-static void __maybe_unused spi_imx2_3_trigger(struct spi_imx_data *spi_imx)
+static void __maybe_unused mx51_ecspi_trigger(struct spi_imx_data *spi_imx)
 {
        u32 reg;
 
-       reg = readl(spi_imx->base + SPI_IMX2_3_CTRL);
-       reg |= SPI_IMX2_3_CTRL_XCH;
-       writel(reg, spi_imx->base + SPI_IMX2_3_CTRL);
+       reg = readl(spi_imx->base + MX51_ECSPI_CTRL);
+       reg |= MX51_ECSPI_CTRL_XCH;
+       writel(reg, spi_imx->base + MX51_ECSPI_CTRL);
 }
 
-static int __maybe_unused spi_imx2_3_config(struct spi_imx_data *spi_imx,
+static int __maybe_unused mx51_ecspi_config(struct spi_imx_data *spi_imx,
                struct spi_imx_config *config)
 {
-       u32 ctrl = SPI_IMX2_3_CTRL_ENABLE, cfg = 0;
+       u32 ctrl = MX51_ECSPI_CTRL_ENABLE, cfg = 0;
 
        /*
         * The hardware seems to have a race condition when changing modes. The
@@ -260,42 +270,42 @@ static int __maybe_unused spi_imx2_3_config(struct spi_imx_data *spi_imx,
         * the same time.
         * So set master mode for all channels as we do not support slave mode.
         */
-       ctrl |= SPI_IMX2_3_CTRL_MODE_MASK;
+       ctrl |= MX51_ECSPI_CTRL_MODE_MASK;
 
        /* set clock speed */
-       ctrl |= spi_imx2_3_clkdiv(spi_imx->spi_clk, config->speed_hz);
+       ctrl |= mx51_ecspi_clkdiv(spi_imx->spi_clk, config->speed_hz);
 
        /* set chip select to use */
-       ctrl |= SPI_IMX2_3_CTRL_CS(config->cs);
+       ctrl |= MX51_ECSPI_CTRL_CS(config->cs);
 
-       ctrl |= (config->bpw - 1) << SPI_IMX2_3_CTRL_BL_OFFSET;
+       ctrl |= (config->bpw - 1) << MX51_ECSPI_CTRL_BL_OFFSET;
 
-       cfg |= SPI_IMX2_3_CONFIG_SBBCTRL(config->cs);
+       cfg |= MX51_ECSPI_CONFIG_SBBCTRL(config->cs);
 
        if (config->mode & SPI_CPHA)
-               cfg |= SPI_IMX2_3_CONFIG_SCLKPHA(config->cs);
+               cfg |= MX51_ECSPI_CONFIG_SCLKPHA(config->cs);
 
        if (config->mode & SPI_CPOL)
-               cfg |= SPI_IMX2_3_CONFIG_SCLKPOL(config->cs);
+               cfg |= MX51_ECSPI_CONFIG_SCLKPOL(config->cs);
 
        if (config->mode & SPI_CS_HIGH)
-               cfg |= SPI_IMX2_3_CONFIG_SSBPOL(config->cs);
+               cfg |= MX51_ECSPI_CONFIG_SSBPOL(config->cs);
 
-       writel(ctrl, spi_imx->base + SPI_IMX2_3_CTRL);
-       writel(cfg, spi_imx->base + SPI_IMX2_3_CONFIG);
+       writel(ctrl, spi_imx->base + MX51_ECSPI_CTRL);
+       writel(cfg, spi_imx->base + MX51_ECSPI_CONFIG);
 
        return 0;
 }
 
-static int __maybe_unused spi_imx2_3_rx_available(struct spi_imx_data *spi_imx)
+static int __maybe_unused mx51_ecspi_rx_available(struct spi_imx_data *spi_imx)
 {
-       return readl(spi_imx->base + SPI_IMX2_3_STAT) & SPI_IMX2_3_STAT_RR;
+       return readl(spi_imx->base + MX51_ECSPI_STAT) & MX51_ECSPI_STAT_RR;
 }
 
-static void __maybe_unused spi_imx2_3_reset(struct spi_imx_data *spi_imx)
+static void __maybe_unused mx51_ecspi_reset(struct spi_imx_data *spi_imx)
 {
        /* drain receive buffer */
-       while (spi_imx2_3_rx_available(spi_imx))
+       while (mx51_ecspi_rx_available(spi_imx))
                readl(spi_imx->base + MXC_CSPIRXDATA);
 }
 
@@ -343,32 +353,7 @@ static void __maybe_unused mx31_trigger(struct spi_imx_data *spi_imx)
        writel(reg, spi_imx->base + MXC_CSPICTRL);
 }
 
-static int __maybe_unused spi_imx0_4_config(struct spi_imx_data *spi_imx,
-               struct spi_imx_config *config)
-{
-       unsigned int reg = MX31_CSPICTRL_ENABLE | MX31_CSPICTRL_MASTER;
-       int cs = spi_imx->chipselect[config->cs];
-
-       reg |= spi_imx_clkdiv_2(spi_imx->spi_clk, config->speed_hz) <<
-               MX31_CSPICTRL_DR_SHIFT;
-
-       reg |= (config->bpw - 1) << MX31_CSPICTRL_BC_SHIFT;
-
-       if (config->mode & SPI_CPHA)
-               reg |= MX31_CSPICTRL_PHA;
-       if (config->mode & SPI_CPOL)
-               reg |= MX31_CSPICTRL_POL;
-       if (config->mode & SPI_CS_HIGH)
-               reg |= MX31_CSPICTRL_SSPOL;
-       if (cs < 0)
-               reg |= (cs + 32) << MX31_CSPICTRL_CS_SHIFT;
-
-       writel(reg, spi_imx->base + MXC_CSPICTRL);
-
-       return 0;
-}
-
-static int __maybe_unused spi_imx0_7_config(struct spi_imx_data *spi_imx,
+static int __maybe_unused mx31_config(struct spi_imx_data *spi_imx,
                struct spi_imx_config *config)
 {
        unsigned int reg = MX31_CSPICTRL_ENABLE | MX31_CSPICTRL_MASTER;
@@ -377,8 +362,12 @@ static int __maybe_unused spi_imx0_7_config(struct spi_imx_data *spi_imx,
        reg |= spi_imx_clkdiv_2(spi_imx->spi_clk, config->speed_hz) <<
                MX31_CSPICTRL_DR_SHIFT;
 
-       reg |= (config->bpw - 1) << MX35_CSPICTRL_BL_SHIFT;
-       reg |= MX31_CSPICTRL_SSCTL;
+       if (is_imx35_cspi(spi_imx)) {
+               reg |= (config->bpw - 1) << MX35_CSPICTRL_BL_SHIFT;
+               reg |= MX31_CSPICTRL_SSCTL;
+       } else {
+               reg |= (config->bpw - 1) << MX31_CSPICTRL_BC_SHIFT;
+       }
 
        if (config->mode & SPI_CPHA)
                reg |= MX31_CSPICTRL_PHA;
@@ -387,7 +376,9 @@ static int __maybe_unused spi_imx0_7_config(struct spi_imx_data *spi_imx,
        if (config->mode & SPI_CS_HIGH)
                reg |= MX31_CSPICTRL_SSPOL;
        if (cs < 0)
-               reg |= (cs + 32) << MX35_CSPICTRL_CS_SHIFT;
+               reg |= (cs + 32) <<
+                       (is_imx35_cspi(spi_imx) ? MX35_CSPICTRL_CS_SHIFT :
+                                                 MX31_CSPICTRL_CS_SHIFT);
 
        writel(reg, spi_imx->base + MXC_CSPICTRL);
 
@@ -399,77 +390,78 @@ static int __maybe_unused mx31_rx_available(struct spi_imx_data *spi_imx)
        return readl(spi_imx->base + MX31_CSPISTATUS) & MX31_STATUS_RR;
 }
 
-static void __maybe_unused spi_imx0_4_reset(struct spi_imx_data *spi_imx)
+static void __maybe_unused mx31_reset(struct spi_imx_data *spi_imx)
 {
        /* drain receive buffer */
-       while (readl(spi_imx->base + MX3_CSPISTAT) & MX3_CSPISTAT_RR)
+       while (readl(spi_imx->base + MX31_CSPISTATUS) & MX31_STATUS_RR)
                readl(spi_imx->base + MXC_CSPIRXDATA);
 }
 
-#define MX27_INTREG_RR         (1 << 4)
-#define MX27_INTREG_TEEN       (1 << 9)
-#define MX27_INTREG_RREN       (1 << 13)
+#define MX21_INTREG_RR         (1 << 4)
+#define MX21_INTREG_TEEN       (1 << 9)
+#define MX21_INTREG_RREN       (1 << 13)
 
-#define MX27_CSPICTRL_POL      (1 << 5)
-#define MX27_CSPICTRL_PHA      (1 << 6)
-#define MX27_CSPICTRL_SSPOL    (1 << 8)
-#define MX27_CSPICTRL_XCH      (1 << 9)
-#define MX27_CSPICTRL_ENABLE   (1 << 10)
-#define MX27_CSPICTRL_MASTER   (1 << 11)
-#define MX27_CSPICTRL_DR_SHIFT 14
-#define MX27_CSPICTRL_CS_SHIFT 19
+#define MX21_CSPICTRL_POL      (1 << 5)
+#define MX21_CSPICTRL_PHA      (1 << 6)
+#define MX21_CSPICTRL_SSPOL    (1 << 8)
+#define MX21_CSPICTRL_XCH      (1 << 9)
+#define MX21_CSPICTRL_ENABLE   (1 << 10)
+#define MX21_CSPICTRL_MASTER   (1 << 11)
+#define MX21_CSPICTRL_DR_SHIFT 14
+#define MX21_CSPICTRL_CS_SHIFT 19
 
-static void __maybe_unused mx27_intctrl(struct spi_imx_data *spi_imx, int enable)
+static void __maybe_unused mx21_intctrl(struct spi_imx_data *spi_imx, int enable)
 {
        unsigned int val = 0;
 
        if (enable & MXC_INT_TE)
-               val |= MX27_INTREG_TEEN;
+               val |= MX21_INTREG_TEEN;
        if (enable & MXC_INT_RR)
-               val |= MX27_INTREG_RREN;
+               val |= MX21_INTREG_RREN;
 
        writel(val, spi_imx->base + MXC_CSPIINT);
 }
 
-static void __maybe_unused mx27_trigger(struct spi_imx_data *spi_imx)
+static void __maybe_unused mx21_trigger(struct spi_imx_data *spi_imx)
 {
        unsigned int reg;
 
        reg = readl(spi_imx->base + MXC_CSPICTRL);
-       reg |= MX27_CSPICTRL_XCH;
+       reg |= MX21_CSPICTRL_XCH;
        writel(reg, spi_imx->base + MXC_CSPICTRL);
 }
 
-static int __maybe_unused mx27_config(struct spi_imx_data *spi_imx,
+static int __maybe_unused mx21_config(struct spi_imx_data *spi_imx,
                struct spi_imx_config *config)
 {
-       unsigned int reg = MX27_CSPICTRL_ENABLE | MX27_CSPICTRL_MASTER;
+       unsigned int reg = MX21_CSPICTRL_ENABLE | MX21_CSPICTRL_MASTER;
        int cs = spi_imx->chipselect[config->cs];
+       unsigned int max = is_imx27_cspi(spi_imx) ? 16 : 18;
 
-       reg |= spi_imx_clkdiv_1(spi_imx->spi_clk, config->speed_hz) <<
-               MX27_CSPICTRL_DR_SHIFT;
+       reg |= spi_imx_clkdiv_1(spi_imx->spi_clk, config->speed_hz, max) <<
+               MX21_CSPICTRL_DR_SHIFT;
        reg |= config->bpw - 1;
 
        if (config->mode & SPI_CPHA)
-               reg |= MX27_CSPICTRL_PHA;
+               reg |= MX21_CSPICTRL_PHA;
        if (config->mode & SPI_CPOL)
-               reg |= MX27_CSPICTRL_POL;
+               reg |= MX21_CSPICTRL_POL;
        if (config->mode & SPI_CS_HIGH)
-               reg |= MX27_CSPICTRL_SSPOL;
+               reg |= MX21_CSPICTRL_SSPOL;
        if (cs < 0)
-               reg |= (cs + 32) << MX27_CSPICTRL_CS_SHIFT;
+               reg |= (cs + 32) << MX21_CSPICTRL_CS_SHIFT;
 
        writel(reg, spi_imx->base + MXC_CSPICTRL);
 
        return 0;
 }
 
-static int __maybe_unused mx27_rx_available(struct spi_imx_data *spi_imx)
+static int __maybe_unused mx21_rx_available(struct spi_imx_data *spi_imx)
 {
-       return readl(spi_imx->base + MXC_CSPIINT) & MX27_INTREG_RR;
+       return readl(spi_imx->base + MXC_CSPIINT) & MX21_INTREG_RR;
 }
 
-static void __maybe_unused spi_imx0_0_reset(struct spi_imx_data *spi_imx)
+static void __maybe_unused mx21_reset(struct spi_imx_data *spi_imx)
 {
        writel(1, spi_imx->base + MXC_RESET);
 }
@@ -535,61 +527,94 @@ static void __maybe_unused mx1_reset(struct spi_imx_data *spi_imx)
        writel(1, spi_imx->base + MXC_RESET);
 }
 
-/*
- * These version numbers are taken from the Freescale driver.  Unfortunately it
- * doesn't support i.MX1, so this entry doesn't match the scheme. :-(
- */
-static struct spi_imx_devtype_data spi_imx_devtype_data[] __devinitdata = {
-#ifdef CONFIG_SPI_IMX_VER_IMX1
-       [SPI_IMX_VER_IMX1] = {
-               .intctrl = mx1_intctrl,
-               .config = mx1_config,
-               .trigger = mx1_trigger,
-               .rx_available = mx1_rx_available,
-               .reset = mx1_reset,
-               .fifosize = 8,
-       },
-#endif
-#ifdef CONFIG_SPI_IMX_VER_0_0
-       [SPI_IMX_VER_0_0] = {
-               .intctrl = mx27_intctrl,
-               .config = mx27_config,
-               .trigger = mx27_trigger,
-               .rx_available = mx27_rx_available,
-               .reset = spi_imx0_0_reset,
-               .fifosize = 8,
-       },
-#endif
-#ifdef CONFIG_SPI_IMX_VER_0_4
-       [SPI_IMX_VER_0_4] = {
-               .intctrl = mx31_intctrl,
-               .config = spi_imx0_4_config,
-               .trigger = mx31_trigger,
-               .rx_available = mx31_rx_available,
-               .reset = spi_imx0_4_reset,
-               .fifosize = 8,
-       },
-#endif
-#ifdef CONFIG_SPI_IMX_VER_0_7
-       [SPI_IMX_VER_0_7] = {
-               .intctrl = mx31_intctrl,
-               .config = spi_imx0_7_config,
-               .trigger = mx31_trigger,
-               .rx_available = mx31_rx_available,
-               .reset = spi_imx0_4_reset,
-               .fifosize = 8,
-       },
-#endif
-#ifdef CONFIG_SPI_IMX_VER_2_3
-       [SPI_IMX_VER_2_3] = {
-               .intctrl = spi_imx2_3_intctrl,
-               .config = spi_imx2_3_config,
-               .trigger = spi_imx2_3_trigger,
-               .rx_available = spi_imx2_3_rx_available,
-               .reset = spi_imx2_3_reset,
-               .fifosize = 64,
-       },
-#endif
+static struct spi_imx_devtype_data imx1_cspi_devtype_data = {
+       .intctrl = mx1_intctrl,
+       .config = mx1_config,
+       .trigger = mx1_trigger,
+       .rx_available = mx1_rx_available,
+       .reset = mx1_reset,
+       .devtype = IMX1_CSPI,
+};
+
+static struct spi_imx_devtype_data imx21_cspi_devtype_data = {
+       .intctrl = mx21_intctrl,
+       .config = mx21_config,
+       .trigger = mx21_trigger,
+       .rx_available = mx21_rx_available,
+       .reset = mx21_reset,
+       .devtype = IMX21_CSPI,
+};
+
+static struct spi_imx_devtype_data imx27_cspi_devtype_data = {
+       /* i.mx27 cspi shares the functions with i.mx21 one */
+       .intctrl = mx21_intctrl,
+       .config = mx21_config,
+       .trigger = mx21_trigger,
+       .rx_available = mx21_rx_available,
+       .reset = mx21_reset,
+       .devtype = IMX27_CSPI,
+};
+
+static struct spi_imx_devtype_data imx31_cspi_devtype_data = {
+       .intctrl = mx31_intctrl,
+       .config = mx31_config,
+       .trigger = mx31_trigger,
+       .rx_available = mx31_rx_available,
+       .reset = mx31_reset,
+       .devtype = IMX31_CSPI,
+};
+
+static struct spi_imx_devtype_data imx35_cspi_devtype_data = {
+       /* i.mx35 and later cspi shares the functions with i.mx31 one */
+       .intctrl = mx31_intctrl,
+       .config = mx31_config,
+       .trigger = mx31_trigger,
+       .rx_available = mx31_rx_available,
+       .reset = mx31_reset,
+       .devtype = IMX35_CSPI,
+};
+
+static struct spi_imx_devtype_data imx51_ecspi_devtype_data = {
+       .intctrl = mx51_ecspi_intctrl,
+       .config = mx51_ecspi_config,
+       .trigger = mx51_ecspi_trigger,
+       .rx_available = mx51_ecspi_rx_available,
+       .reset = mx51_ecspi_reset,
+       .devtype = IMX51_ECSPI,
+};
+
+static struct platform_device_id spi_imx_devtype[] = {
+       {
+               .name = "imx1-cspi",
+               .driver_data = (kernel_ulong_t) &imx1_cspi_devtype_data,
+       }, {
+               .name = "imx21-cspi",
+               .driver_data = (kernel_ulong_t) &imx21_cspi_devtype_data,
+       }, {
+               .name = "imx27-cspi",
+               .driver_data = (kernel_ulong_t) &imx27_cspi_devtype_data,
+       }, {
+               .name = "imx31-cspi",
+               .driver_data = (kernel_ulong_t) &imx31_cspi_devtype_data,
+       }, {
+               .name = "imx35-cspi",
+               .driver_data = (kernel_ulong_t) &imx35_cspi_devtype_data,
+       }, {
+               .name = "imx51-ecspi",
+               .driver_data = (kernel_ulong_t) &imx51_ecspi_devtype_data,
+       }, {
+               /* sentinel */
+       }
+};
+
+static const struct of_device_id spi_imx_dt_ids[] = {
+       { .compatible = "fsl,imx1-cspi", .data = &imx1_cspi_devtype_data, },
+       { .compatible = "fsl,imx21-cspi", .data = &imx21_cspi_devtype_data, },
+       { .compatible = "fsl,imx27-cspi", .data = &imx27_cspi_devtype_data, },
+       { .compatible = "fsl,imx31-cspi", .data = &imx31_cspi_devtype_data, },
+       { .compatible = "fsl,imx35-cspi", .data = &imx35_cspi_devtype_data, },
+       { .compatible = "fsl,imx51-ecspi", .data = &imx51_ecspi_devtype_data, },
+       { /* sentinel */ }
 };
 
 static void spi_imx_chipselect(struct spi_device *spi, int is_active)
@@ -607,21 +632,21 @@ static void spi_imx_chipselect(struct spi_device *spi, int is_active)
 
 static void spi_imx_push(struct spi_imx_data *spi_imx)
 {
-       while (spi_imx->txfifo < spi_imx->devtype_data.fifosize) {
+       while (spi_imx->txfifo < spi_imx_get_fifosize(spi_imx)) {
                if (!spi_imx->count)
                        break;
                spi_imx->tx(spi_imx);
                spi_imx->txfifo++;
        }
 
-       spi_imx->devtype_data.trigger(spi_imx);
+       spi_imx->devtype_data->trigger(spi_imx);
 }
 
 static irqreturn_t spi_imx_isr(int irq, void *dev_id)
 {
        struct spi_imx_data *spi_imx = dev_id;
 
-       while (spi_imx->devtype_data.rx_available(spi_imx)) {
+       while (spi_imx->devtype_data->rx_available(spi_imx)) {
                spi_imx->rx(spi_imx);
                spi_imx->txfifo--;
        }
@@ -635,12 +660,12 @@ static irqreturn_t spi_imx_isr(int irq, void *dev_id)
                /* No data left to push, but still waiting for rx data,
                 * enable receive data available interrupt.
                 */
-               spi_imx->devtype_data.intctrl(
+               spi_imx->devtype_data->intctrl(
                                spi_imx, MXC_INT_RR);
                return IRQ_HANDLED;
        }
 
-       spi_imx->devtype_data.intctrl(spi_imx, 0);
+       spi_imx->devtype_data->intctrl(spi_imx, 0);
        complete(&spi_imx->xfer_done);
 
        return IRQ_HANDLED;
@@ -677,7 +702,7 @@ static int spi_imx_setupxfer(struct spi_device *spi,
        } else
                BUG();
 
-       spi_imx->devtype_data.config(spi_imx, &config);
+       spi_imx->devtype_data->config(spi_imx, &config);
 
        return 0;
 }
@@ -696,7 +721,7 @@ static int spi_imx_transfer(struct spi_device *spi,
 
        spi_imx_push(spi_imx);
 
-       spi_imx->devtype_data.intctrl(spi_imx, MXC_INT_TE);
+       spi_imx->devtype_data->intctrl(spi_imx, MXC_INT_TE);
 
        wait_for_completion(&spi_imx->xfer_done);
 
@@ -723,72 +748,47 @@ static void spi_imx_cleanup(struct spi_device *spi)
 {
 }
 
-static struct platform_device_id spi_imx_devtype[] = {
-       {
-               .name = "imx1-cspi",
-               .driver_data = SPI_IMX_VER_IMX1,
-       }, {
-               .name = "imx21-cspi",
-               .driver_data = SPI_IMX_VER_0_0,
-       }, {
-               .name = "imx25-cspi",
-               .driver_data = SPI_IMX_VER_0_7,
-       }, {
-               .name = "imx27-cspi",
-               .driver_data = SPI_IMX_VER_0_0,
-       }, {
-               .name = "imx31-cspi",
-               .driver_data = SPI_IMX_VER_0_4,
-       }, {
-               .name = "imx35-cspi",
-               .driver_data = SPI_IMX_VER_0_7,
-       }, {
-               .name = "imx51-cspi",
-               .driver_data = SPI_IMX_VER_0_7,
-       }, {
-               .name = "imx51-ecspi",
-               .driver_data = SPI_IMX_VER_2_3,
-       }, {
-               .name = "imx53-cspi",
-               .driver_data = SPI_IMX_VER_0_7,
-       }, {
-               .name = "imx53-ecspi",
-               .driver_data = SPI_IMX_VER_2_3,
-       }, {
-               /* sentinel */
-       }
-};
-
 static int __devinit spi_imx_probe(struct platform_device *pdev)
 {
-       struct spi_imx_master *mxc_platform_info;
+       struct device_node *np = pdev->dev.of_node;
+       const struct of_device_id *of_id =
+                       of_match_device(spi_imx_dt_ids, &pdev->dev);
+       struct spi_imx_master *mxc_platform_info =
+                       dev_get_platdata(&pdev->dev);
        struct spi_master *master;
        struct spi_imx_data *spi_imx;
        struct resource *res;
-       int i, ret;
+       int i, ret, num_cs;
 
-       mxc_platform_info = dev_get_platdata(&pdev->dev);
-       if (!mxc_platform_info) {
+       if (!np && !mxc_platform_info) {
                dev_err(&pdev->dev, "can't get the platform data\n");
                return -EINVAL;
        }
 
-       master = spi_alloc_master(&pdev->dev, sizeof(struct spi_imx_data));
+       ret = of_property_read_u32(np, "fsl,spi-num-chipselects", &num_cs);
+       if (ret < 0)
+               num_cs = mxc_platform_info->num_chipselect;
+
+       master = spi_alloc_master(&pdev->dev,
+                       sizeof(struct spi_imx_data) + sizeof(int) * num_cs);
        if (!master)
                return -ENOMEM;
 
        platform_set_drvdata(pdev, master);
 
        master->bus_num = pdev->id;
-       master->num_chipselect = mxc_platform_info->num_chipselect;
+       master->num_chipselect = num_cs;
 
        spi_imx = spi_master_get_devdata(master);
        spi_imx->bitbang.master = spi_master_get(master);
-       spi_imx->chipselect = mxc_platform_info->chipselect;
 
        for (i = 0; i < master->num_chipselect; i++) {
-               if (spi_imx->chipselect[i] < 0)
+               int cs_gpio = of_get_named_gpio(np, "cs-gpios", i);
+               if (cs_gpio < 0)
+                       cs_gpio = mxc_platform_info->chipselect[i];
+               if (cs_gpio < 0)
                        continue;
+               spi_imx->chipselect[i] = cs_gpio;
                ret = gpio_request(spi_imx->chipselect[i], DRIVER_NAME);
                if (ret) {
                        while (i > 0) {
@@ -810,8 +810,8 @@ static int __devinit spi_imx_probe(struct platform_device *pdev)
 
        init_completion(&spi_imx->xfer_done);
 
-       spi_imx->devtype_data =
-               spi_imx_devtype_data[pdev->id_entry->driver_data];
+       spi_imx->devtype_data = of_id ? of_id->data :
+               (struct spi_imx_devtype_data *) pdev->id_entry->driver_data;
 
        res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
        if (!res) {
@@ -854,10 +854,11 @@ static int __devinit spi_imx_probe(struct platform_device *pdev)
        clk_enable(spi_imx->clk);
        spi_imx->spi_clk = clk_get_rate(spi_imx->clk);
 
-       spi_imx->devtype_data.reset(spi_imx);
+       spi_imx->devtype_data->reset(spi_imx);
 
-       spi_imx->devtype_data.intctrl(spi_imx, 0);
+       spi_imx->devtype_data->intctrl(spi_imx, 0);
 
+       master->dev.of_node = pdev->dev.of_node;
        ret = spi_bitbang_start(&spi_imx->bitbang);
        if (ret) {
                dev_err(&pdev->dev, "bitbang start failed with %d\n", ret);
@@ -920,6 +921,7 @@ static struct platform_driver spi_imx_driver = {
        .driver = {
                   .name = DRIVER_NAME,
                   .owner = THIS_MODULE,
+                  .of_match_table = spi_imx_dt_ids,
                   },
        .id_table = spi_imx_devtype,
        .probe = spi_imx_probe,