* to the previous generations.
  *
  * Currently the driver only handles Multitouch events with already
- * programmed firmware and "config" for "Revision D" Berlin IC.
+ * programmed firmware and "config" for "Revision A/D" Berlin IC.
  *
  * Support is missing for:
  * - ESD Management
  * - "Config" update/flashing
  * - Stylus Events
  * - Gesture Events
- * - Support for older revisions (A & B)
+ * - Support for revision B
  */
 
 #include <linux/bitfield.h>
 #include <linux/input.h>
 #include <linux/input/mt.h>
 #include <linux/input/touchscreen.h>
+#include <linux/property.h>
 #include <linux/regmap.h>
 #include <linux/regulator/consumer.h>
 #include <linux/sizes.h>
 
 #define GOODIX_BERLIN_DEV_CONFIRM_VAL          0xAA
 #define GOODIX_BERLIN_BOOTOPTION_ADDR          0x10000
-#define GOODIX_BERLIN_FW_VERSION_INFO_ADDR     0x10014
 
 #define GOODIX_BERLIN_IC_INFO_MAX_LEN          SZ_1K
-#define GOODIX_BERLIN_IC_INFO_ADDR             0x10070
 
 #define GOODIX_BERLIN_CHECKSUM_SIZE            sizeof(u16)
 
        /* Runtime parameters extracted from IC_INFO buffer  */
        u32 touch_data_addr;
 
+       const struct goodix_berlin_ic_data *ic_data;
+
        struct goodix_berlin_event event;
 };
 
 {
        int error;
 
-       error = regmap_raw_read(cd->regmap, GOODIX_BERLIN_FW_VERSION_INFO_ADDR,
+       error = regmap_raw_read(cd->regmap, cd->ic_data->fw_version_info_addr,
                                &cd->fw_version, sizeof(cd->fw_version));
        if (error) {
                dev_err(cd->dev, "error reading fw version, %d\n", error);
        if (!afe_data)
                return -ENOMEM;
 
-       error = regmap_raw_read(cd->regmap, GOODIX_BERLIN_IC_INFO_ADDR,
+       error = regmap_raw_read(cd->regmap, cd->ic_data->ic_info_addr,
                                &length_raw, sizeof(length_raw));
        if (error) {
                dev_err(cd->dev, "failed get ic info length, %d\n", error);
                return -EINVAL;
        }
 
-       error = regmap_raw_read(cd->regmap, GOODIX_BERLIN_IC_INFO_ADDR,
-                               afe_data, length);
+       error = regmap_raw_read(cd->regmap, cd->ic_data->ic_info_addr, afe_data,
+                               length);
        if (error) {
                dev_err(cd->dev, "failed get ic info data, %d\n", error);
                return error;
 EXPORT_SYMBOL_GPL(goodix_berlin_groups);
 
 int goodix_berlin_probe(struct device *dev, int irq, const struct input_id *id,
-                       struct regmap *regmap)
+                       struct regmap *regmap,
+                       const struct goodix_berlin_ic_data *ic_data)
 {
        struct goodix_berlin_core *cd;
        int error;
        cd->dev = dev;
        cd->regmap = regmap;
        cd->irq = irq;
+       cd->ic_data = ic_data;
 
        cd->reset_gpio = devm_gpiod_get_optional(dev, "reset", GPIOD_OUT_HIGH);
        if (IS_ERR(cd->reset_gpio))
 
 
 static int goodix_berlin_i2c_probe(struct i2c_client *client)
 {
+       const struct goodix_berlin_ic_data *ic_data =
+               i2c_get_match_data(client);
        struct regmap *regmap;
        int error;
 
                return PTR_ERR(regmap);
 
        error = goodix_berlin_probe(&client->dev, client->irq,
-                                   &goodix_berlin_i2c_input_id, regmap);
+                                   &goodix_berlin_i2c_input_id, regmap,
+                                   ic_data);
        if (error)
                return error;
 
        return 0;
 }
 
+static const struct goodix_berlin_ic_data gt9916_data = {
+       .fw_version_info_addr = GOODIX_BERLIN_FW_VERSION_INFO_ADDR_D,
+       .ic_info_addr = GOODIX_BERLIN_IC_INFO_ADDR_D,
+};
+
 static const struct i2c_device_id goodix_berlin_i2c_id[] = {
-       { "gt9916" },
+       { .name = "gt9916", .driver_data = (long)>9916_data },
        { }
 };
 
 MODULE_DEVICE_TABLE(i2c, goodix_berlin_i2c_id);
 
 static const struct of_device_id goodix_berlin_i2c_of_match[] = {
-       { .compatible = "goodix,gt9916", },
+       { .compatible = "goodix,gt9916", .data = >9916_data },
        { }
 };
 MODULE_DEVICE_TABLE(of, goodix_berlin_i2c_of_match);
 
 
 #define GOODIX_BERLIN_SPI_TRANS_PREFIX_LEN     1
 #define GOODIX_BERLIN_REGISTER_WIDTH           4
-#define GOODIX_BERLIN_SPI_READ_DUMMY_LEN       3
-#define GOODIX_BERLIN_SPI_READ_PREFIX_LEN      (GOODIX_BERLIN_SPI_TRANS_PREFIX_LEN + \
+#define GOODIX_BERLIN_SPI_READ_DUMMY_LEN_A     4
+#define GOODIX_BERLIN_SPI_READ_DUMMY_LEN_D     3
+#define GOODIX_BERLIN_SPI_READ_PREFIX_LEN_A    (GOODIX_BERLIN_SPI_TRANS_PREFIX_LEN + \
                                                 GOODIX_BERLIN_REGISTER_WIDTH + \
-                                                GOODIX_BERLIN_SPI_READ_DUMMY_LEN)
+                                                GOODIX_BERLIN_SPI_READ_DUMMY_LEN_A)
+#define GOODIX_BERLIN_SPI_READ_PREFIX_LEN_D    (GOODIX_BERLIN_SPI_TRANS_PREFIX_LEN + \
+                                                GOODIX_BERLIN_REGISTER_WIDTH + \
+                                                GOODIX_BERLIN_SPI_READ_DUMMY_LEN_D)
 #define GOODIX_BERLIN_SPI_WRITE_PREFIX_LEN     (GOODIX_BERLIN_SPI_TRANS_PREFIX_LEN + \
                                                 GOODIX_BERLIN_REGISTER_WIDTH)
 
                                  size_t val_size)
 {
        struct spi_device *spi = context;
+       const struct goodix_berlin_ic_data *ic_data = spi_get_device_match_data(spi);
        struct spi_transfer xfers;
        struct spi_message spi_msg;
        const u32 *reg = reg_buf; /* reg is stored as native u32 at start of buffer */
                return -EINVAL;
 
        u8 *buf __free(kfree) =
-               kzalloc(GOODIX_BERLIN_SPI_READ_PREFIX_LEN + val_size,
-                       GFP_KERNEL);
+               kzalloc(ic_data->read_prefix_len + val_size, GFP_KERNEL);
        if (!buf)
                return -ENOMEM;
 
        spi_message_init(&spi_msg);
        memset(&xfers, 0, sizeof(xfers));
 
-       /* buffer format: 0xF1 + addr(4bytes) + dummy(3bytes) + data */
+       /* buffer format: 0xF1 + addr(4bytes) + dummy(3/4bytes) + data */
        buf[0] = GOODIX_BERLIN_SPI_READ_FLAG;
        put_unaligned_be32(*reg, buf + GOODIX_BERLIN_SPI_TRANS_PREFIX_LEN);
        memset(buf + GOODIX_BERLIN_SPI_TRANS_PREFIX_LEN + GOODIX_BERLIN_REGISTER_WIDTH,
-              0xff, GOODIX_BERLIN_SPI_READ_DUMMY_LEN);
+              0xff, ic_data->read_dummy_len);
 
        xfers.tx_buf = buf;
        xfers.rx_buf = buf;
-       xfers.len = GOODIX_BERLIN_SPI_READ_PREFIX_LEN + val_size;
+       xfers.len = ic_data->read_prefix_len + val_size;
        xfers.cs_change = 0;
        spi_message_add_tail(&xfers, &spi_msg);
 
                return error;
        }
 
-       memcpy(val_buf, buf + GOODIX_BERLIN_SPI_READ_PREFIX_LEN, val_size);
+       memcpy(val_buf, buf + ic_data->read_prefix_len, val_size);
        return error;
 }
 
 
 static int goodix_berlin_spi_probe(struct spi_device *spi)
 {
+       const struct goodix_berlin_ic_data *ic_data = spi_get_device_match_data(spi);
        struct regmap_config regmap_config;
        struct regmap *regmap;
        size_t max_size;
        max_size = spi_max_transfer_size(spi);
 
        regmap_config = goodix_berlin_spi_regmap_conf;
-       regmap_config.max_raw_read = max_size - GOODIX_BERLIN_SPI_READ_PREFIX_LEN;
+       regmap_config.max_raw_read = max_size - ic_data->read_prefix_len;
        regmap_config.max_raw_write = max_size - GOODIX_BERLIN_SPI_WRITE_PREFIX_LEN;
 
        regmap = devm_regmap_init(&spi->dev, NULL, spi, ®map_config);
                return PTR_ERR(regmap);
 
        error = goodix_berlin_probe(&spi->dev, spi->irq,
-                                   &goodix_berlin_spi_input_id, regmap);
+                                   &goodix_berlin_spi_input_id, regmap,
+                                   ic_data);
        if (error)
                return error;
 
        return 0;
 }
 
+static const struct goodix_berlin_ic_data gt9897_data = {
+       .fw_version_info_addr = GOODIX_BERLIN_FW_VERSION_INFO_ADDR_A,
+       .ic_info_addr = GOODIX_BERLIN_IC_INFO_ADDR_A,
+       .read_dummy_len = GOODIX_BERLIN_SPI_READ_DUMMY_LEN_A,
+       .read_prefix_len = GOODIX_BERLIN_SPI_READ_PREFIX_LEN_A,
+};
+
+static const struct goodix_berlin_ic_data gt9916_data = {
+       .fw_version_info_addr = GOODIX_BERLIN_FW_VERSION_INFO_ADDR_D,
+       .ic_info_addr = GOODIX_BERLIN_IC_INFO_ADDR_D,
+       .read_dummy_len = GOODIX_BERLIN_SPI_READ_DUMMY_LEN_D,
+       .read_prefix_len = GOODIX_BERLIN_SPI_READ_PREFIX_LEN_D,
+};
+
 static const struct spi_device_id goodix_berlin_spi_ids[] = {
-       { "gt9916" },
+       { .name = "gt9897", .driver_data = (long)>9897_data },
+       { .name = "gt9916", .driver_data = (long)>9916_data },
        { },
 };
 MODULE_DEVICE_TABLE(spi, goodix_berlin_spi_ids);
 
 static const struct of_device_id goodix_berlin_spi_of_match[] = {
-       { .compatible = "goodix,gt9916", },
+       { .compatible = "goodix,gt9897", .data = >9897_data },
+       { .compatible = "goodix,gt9916", .data = >9916_data },
        { }
 };
 MODULE_DEVICE_TABLE(of, goodix_berlin_spi_of_match);