]> git.karo-electronics.de Git - karo-tx-linux.git/blobdiff - drivers/input/touchscreen/atmel_mxt_ts.c
Input: atmel_mxt_ts - move input device init into separate function
[karo-tx-linux.git] / drivers / input / touchscreen / atmel_mxt_ts.c
index d3aef870d25519919a5c63d3888ca3cc31205fdd..a248414642d91873e09faed15f2feacdd61a1e26 100644 (file)
 #define MXT_VER_21             21
 #define MXT_VER_22             22
 
-/* Slave addresses */
-#define MXT_APP_LOW            0x4a
-#define MXT_APP_HIGH           0x4b
-#define MXT_BOOT_LOW           0x24
-#define MXT_BOOT_HIGH          0x25
-
 /* Firmware */
 #define MXT_FW_NAME            "maxtouch.fw"
 
 
 /* MXT_TOUCH_MULTI_T9 field */
 #define MXT_TOUCH_CTRL         0
-#define MXT_TOUCH_XORIGIN      1
-#define MXT_TOUCH_YORIGIN      2
-#define MXT_TOUCH_XSIZE                3
-#define MXT_TOUCH_YSIZE                4
-#define MXT_TOUCH_BLEN         6
-#define MXT_TOUCH_TCHTHR       7
-#define MXT_TOUCH_TCHDI                8
-#define MXT_TOUCH_ORIENT       9
-#define MXT_TOUCH_MOVHYSTI     11
-#define MXT_TOUCH_MOVHYSTN     12
-#define MXT_TOUCH_NUMTOUCH     14
-#define MXT_TOUCH_MRGHYST      15
-#define MXT_TOUCH_MRGTHR       16
-#define MXT_TOUCH_AMPHYST      17
-#define MXT_TOUCH_XRANGE_LSB   18
-#define MXT_TOUCH_XRANGE_MSB   19
-#define MXT_TOUCH_YRANGE_LSB   20
-#define MXT_TOUCH_YRANGE_MSB   21
-#define MXT_TOUCH_XLOCLIP      22
-#define MXT_TOUCH_XHICLIP      23
-#define MXT_TOUCH_YLOCLIP      24
-#define MXT_TOUCH_YHICLIP      25
-#define MXT_TOUCH_XEDGECTRL    26
-#define MXT_TOUCH_XEDGEDIST    27
-#define MXT_TOUCH_YEDGECTRL    28
-#define MXT_TOUCH_YEDGEDIST    29
-#define MXT_TOUCH_JUMPLIMIT    30
+#define MXT_T9_ORIENT          9
+#define MXT_T9_RANGE           18
+
+/* MXT_TOUCH_MULTI_T9 status */
+#define MXT_T9_UNGRIP          (1 << 0)
+#define MXT_T9_SUPPRESS                (1 << 1)
+#define MXT_T9_AMP             (1 << 2)
+#define MXT_T9_VECTOR          (1 << 3)
+#define MXT_T9_MOVE            (1 << 4)
+#define MXT_T9_RELEASE         (1 << 5)
+#define MXT_T9_PRESS           (1 << 6)
+#define MXT_T9_DETECT          (1 << 7)
+
+struct t9_range {
+       u16 x;
+       u16 y;
+} __packed;
+
+/* MXT_TOUCH_MULTI_T9 orient */
+#define MXT_T9_ORIENT_SWITCH   (1 << 0)
 
 /* MXT_PROCI_GRIPFACE_T20 field */
 #define MXT_GRIPFACE_CTRL      0
 #define MXT_BACKUP_TIME                50      /* msec */
 #define MXT_RESET_TIME         200     /* msec */
 #define MXT_RESET_TIMEOUT      3000    /* msec */
+#define MXT_CRC_TIMEOUT                1000    /* msec */
 #define MXT_FW_RESET_TIME      3000    /* msec */
 #define MXT_FW_CHG_TIMEOUT     300     /* msec */
 
 #define MXT_FRAME_CRC_PASS     0x04
 #define MXT_APP_CRC_FAIL       0x40    /* valid 7 8 bit only */
 #define MXT_BOOT_STATUS_MASK   0x3f
-
-/* Touch status */
-#define MXT_UNGRIP             (1 << 0)
-#define MXT_SUPPRESS           (1 << 1)
-#define MXT_AMP                        (1 << 2)
-#define MXT_VECTOR             (1 << 3)
-#define MXT_MOVE               (1 << 4)
-#define MXT_RELEASE            (1 << 5)
-#define MXT_PRESS              (1 << 6)
-#define MXT_DETECT             (1 << 7)
-
-/* Touch orient bits */
-#define MXT_XY_SWITCH          (1 << 0)
-#define MXT_X_INVERT           (1 << 1)
-#define MXT_Y_INVERT           (1 << 2)
+#define MXT_BOOT_EXTENDED_ID   (1 << 5)
+#define MXT_BOOT_ID_MASK       0x1f
 
 /* Touchscreen absolute values */
 #define MXT_MAX_AREA           0xff
@@ -259,6 +234,8 @@ struct mxt_data {
        unsigned int max_x;
        unsigned int max_y;
        bool in_bootloader;
+       u32 config_crc;
+       u8 bootloader_addr;
 
        /* Cached parameters from object table */
        u8 T6_reportid;
@@ -272,6 +249,9 @@ struct mxt_data {
 
        /* for reset handling */
        struct completion reset_completion;
+
+       /* for config update handling */
+       struct completion crc_completion;
 };
 
 static size_t mxt_obj_size(const struct mxt_object *obj)
@@ -373,9 +353,103 @@ static int mxt_wait_for_completion(struct mxt_data *data,
        return 0;
 }
 
+static int mxt_bootloader_read(struct mxt_data *data,
+                              u8 *val, unsigned int count)
+{
+       int ret;
+       struct i2c_msg msg;
+
+       msg.addr = data->bootloader_addr;
+       msg.flags = data->client->flags & I2C_M_TEN;
+       msg.flags |= I2C_M_RD;
+       msg.len = count;
+       msg.buf = val;
+
+       ret = i2c_transfer(data->client->adapter, &msg, 1);
+
+       if (ret == 1) {
+               ret = 0;
+       } else {
+               ret = ret < 0 ? ret : -EIO;
+               dev_err(&data->client->dev, "%s: i2c recv failed (%d)\n",
+                       __func__, ret);
+       }
+
+       return ret;
+}
+
+static int mxt_bootloader_write(struct mxt_data *data,
+                               const u8 * const val, unsigned int count)
+{
+       int ret;
+       struct i2c_msg msg;
+
+       msg.addr = data->bootloader_addr;
+       msg.flags = data->client->flags & I2C_M_TEN;
+       msg.len = count;
+       msg.buf = (u8 *)val;
+
+       ret = i2c_transfer(data->client->adapter, &msg, 1);
+       if (ret == 1) {
+               ret = 0;
+       } else {
+               ret = ret < 0 ? ret : -EIO;
+               dev_err(&data->client->dev, "%s: i2c send failed (%d)\n",
+                       __func__, ret);
+       }
+
+       return ret;
+}
+
+static int mxt_lookup_bootloader_address(struct mxt_data *data)
+{
+       u8 appmode = data->client->addr;
+       u8 bootloader;
+
+       switch (appmode) {
+       case 0x4a:
+       case 0x4b:
+       case 0x4c:
+       case 0x4d:
+       case 0x5a:
+       case 0x5b:
+               bootloader = appmode - 0x26;
+               break;
+       default:
+               dev_err(&data->client->dev,
+                       "Appmode i2c address 0x%02x not found\n",
+                       appmode);
+               return -EINVAL;
+       }
+
+       data->bootloader_addr = bootloader;
+       return 0;
+}
+
+static u8 mxt_get_bootloader_version(struct mxt_data *data, u8 val)
+{
+       struct device *dev = &data->client->dev;
+       u8 buf[3];
+
+       if (val & MXT_BOOT_EXTENDED_ID) {
+               if (mxt_bootloader_read(data, &buf[0], 3) != 0) {
+                       dev_err(dev, "%s: i2c failure\n", __func__);
+                       return val;
+               }
+
+               dev_dbg(dev, "Bootloader ID:%d Version:%d\n", buf[1], buf[2]);
+
+               return buf[0];
+       } else {
+               dev_dbg(dev, "Bootloader ID:%d\n", val & MXT_BOOT_ID_MASK);
+
+               return val;
+       }
+}
+
 static int mxt_check_bootloader(struct mxt_data *data, unsigned int state)
 {
-       struct i2c_client *client = data->client;
+       struct device *dev = &data->client->dev;
        u8 val;
        int ret;
 
@@ -396,15 +470,17 @@ recheck:
                         * by writing length 0x000 to device (iff we are in
                         * WAITING_FRAME_DATA state).
                         */
-                       dev_err(&client->dev, "Update wait error %d\n", ret);
+                       dev_err(dev, "Update wait error %d\n", ret);
                        return ret;
                }
        }
 
-       if (i2c_master_recv(client, &val, 1) != 1) {
-               dev_err(&client->dev, "%s: i2c recv failed\n", __func__);
-               return -EIO;
-       }
+       ret = mxt_bootloader_read(data, &val, 1);
+       if (ret)
+               return ret;
+
+       if (state == MXT_WAITING_BOOTLOAD_CMD)
+               val = mxt_get_bootloader_version(data, val);
 
        switch (state) {
        case MXT_WAITING_BOOTLOAD_CMD:
@@ -412,43 +488,37 @@ recheck:
                val &= ~MXT_BOOT_STATUS_MASK;
                break;
        case MXT_FRAME_CRC_PASS:
-               if (val == MXT_FRAME_CRC_CHECK)
+               if (val == MXT_FRAME_CRC_CHECK) {
                        goto recheck;
+               } else if (val == MXT_FRAME_CRC_FAIL) {
+                       dev_err(dev, "Bootloader CRC fail\n");
+                       return -EINVAL;
+               }
                break;
        default:
                return -EINVAL;
        }
 
        if (val != state) {
-               dev_err(&client->dev, "Unvalid bootloader mode state\n");
+               dev_err(dev, "Invalid bootloader state %02X != %02X\n",
+                       val, state);
                return -EINVAL;
        }
 
        return 0;
 }
 
-static int mxt_unlock_bootloader(struct i2c_client *client)
+static int mxt_unlock_bootloader(struct mxt_data *data)
 {
+       int ret;
        u8 buf[2];
 
        buf[0] = MXT_UNLOCK_CMD_LSB;
        buf[1] = MXT_UNLOCK_CMD_MSB;
 
-       if (i2c_master_send(client, buf, 2) != 2) {
-               dev_err(&client->dev, "%s: i2c send failed\n", __func__);
-               return -EIO;
-       }
-
-       return 0;
-}
-
-static int mxt_fw_write(struct i2c_client *client,
-                            const u8 *data, unsigned int frame_size)
-{
-       if (i2c_master_send(client, data, frame_size) != frame_size) {
-               dev_err(&client->dev, "%s: i2c send failed\n", __func__);
-               return -EIO;
-       }
+       ret = mxt_bootloader_write(data, buf, 2);
+       if (ret)
+               return ret;
 
        return 0;
 }
@@ -488,11 +558,6 @@ static int __mxt_read_reg(struct i2c_client *client,
        return ret;
 }
 
-static int mxt_read_reg(struct i2c_client *client, u16 reg, u8 *val)
-{
-       return __mxt_read_reg(client, reg, 1, val);
-}
-
 static int __mxt_write_reg(struct i2c_client *client, u16 reg, u16 len,
                           const void *val)
 {
@@ -540,7 +605,7 @@ mxt_get_object(struct mxt_data *data, u8 type)
                        return object;
        }
 
-       dev_err(&data->client->dev, "Invalid object type\n");
+       dev_err(&data->client->dev, "Invalid object type T%u\n", type);
        return NULL;
 }
 
@@ -589,6 +654,12 @@ static void mxt_input_button(struct mxt_data *data, struct mxt_message *message)
        }
 }
 
+static void mxt_input_sync(struct input_dev *input_dev)
+{
+       input_mt_report_pointer_emulation(input_dev, false);
+       input_sync(input_dev);
+}
+
 static void mxt_input_touchevent(struct mxt_data *data,
                                      struct mxt_message *message, int id)
 {
@@ -598,44 +669,60 @@ static void mxt_input_touchevent(struct mxt_data *data,
        int x;
        int y;
        int area;
-       int pressure;
+       int amplitude;
 
        x = (message->message[1] << 4) | ((message->message[3] >> 4) & 0xf);
        y = (message->message[2] << 4) | ((message->message[3] & 0xf));
+
+       /* Handle 10/12 bit switching */
        if (data->max_x < 1024)
-               x = x >> 2;
+               x >>= 2;
        if (data->max_y < 1024)
-               y = y >> 2;
+               y >>= 2;
 
        area = message->message[4];
-       pressure = message->message[5];
+       amplitude = message->message[5];
 
        dev_dbg(dev,
                "[%u] %c%c%c%c%c%c%c%c x: %5u y: %5u area: %3u amp: %3u\n",
                id,
-               (status & MXT_DETECT) ? 'D' : '.',
-               (status & MXT_PRESS) ? 'P' : '.',
-               (status & MXT_RELEASE) ? 'R' : '.',
-               (status & MXT_MOVE) ? 'M' : '.',
-               (status & MXT_VECTOR) ? 'V' : '.',
-               (status & MXT_AMP) ? 'A' : '.',
-               (status & MXT_SUPPRESS) ? 'S' : '.',
-               (status & MXT_UNGRIP) ? 'U' : '.',
-               x, y, area, pressure);
+               (status & MXT_T9_DETECT) ? 'D' : '.',
+               (status & MXT_T9_PRESS) ? 'P' : '.',
+               (status & MXT_T9_RELEASE) ? 'R' : '.',
+               (status & MXT_T9_MOVE) ? 'M' : '.',
+               (status & MXT_T9_VECTOR) ? 'V' : '.',
+               (status & MXT_T9_AMP) ? 'A' : '.',
+               (status & MXT_T9_SUPPRESS) ? 'S' : '.',
+               (status & MXT_T9_UNGRIP) ? 'U' : '.',
+               x, y, area, amplitude);
 
        input_mt_slot(input_dev, id);
-       input_mt_report_slot_state(input_dev, MT_TOOL_FINGER,
-                                  status & MXT_DETECT);
 
-       if (status & MXT_DETECT) {
+       if (status & MXT_T9_DETECT) {
+               /*
+                * Multiple bits may be set if the host is slow to read
+                * the status messages, indicating all the events that
+                * have happened.
+                */
+               if (status & MXT_T9_RELEASE) {
+                       input_mt_report_slot_state(input_dev,
+                                                  MT_TOOL_FINGER, 0);
+                       mxt_input_sync(input_dev);
+               }
+
+               /* Touch active */
+               input_mt_report_slot_state(input_dev, MT_TOOL_FINGER, 1);
                input_report_abs(input_dev, ABS_MT_POSITION_X, x);
                input_report_abs(input_dev, ABS_MT_POSITION_Y, y);
-               input_report_abs(input_dev, ABS_MT_PRESSURE, pressure);
+               input_report_abs(input_dev, ABS_MT_PRESSURE, amplitude);
                input_report_abs(input_dev, ABS_MT_TOUCH_MAJOR, area);
+       } else {
+               /* Touch no longer active, close out slot */
+               input_mt_report_slot_state(input_dev, MT_TOOL_FINGER, 0);
        }
 }
 
-static unsigned mxt_extract_T6_csum(const u8 *csum)
+static u16 mxt_extract_T6_csum(const u8 *csum)
 {
        return csum[0] | (csum[1] << 8) | (csum[2] << 16);
 }
@@ -653,6 +740,7 @@ static irqreturn_t mxt_process_messages_until_invalid(struct mxt_data *data)
        struct device *dev = &data->client->dev;
        u8 reportid;
        bool update_input = false;
+       u32 crc;
 
        do {
                if (mxt_read_message(data, &message)) {
@@ -664,12 +752,24 @@ static irqreturn_t mxt_process_messages_until_invalid(struct mxt_data *data)
 
                if (reportid == data->T6_reportid) {
                        u8 status = payload[0];
-                       unsigned csum = mxt_extract_T6_csum(&payload[1]);
+
+                       crc = mxt_extract_T6_csum(&payload[1]);
+                       if (crc != data->config_crc) {
+                               data->config_crc = crc;
+                               complete(&data->crc_completion);
+                       }
+
                        dev_dbg(dev, "Status: %02x Config Checksum: %06x\n",
-                               status, csum);
+                               status, data->config_crc);
 
                        if (status & MXT_T6_STATUS_RESET)
                                complete(&data->reset_completion);
+               } else if (!data->input_dev) {
+                       /*
+                        * do not report events if input device
+                        * is not yet registered
+                        */
+                       mxt_dump_message(dev, &message);
                } else if (mxt_is_T9_message(data, &message)) {
                        int id = reportid - data->T9_reportid_min;
                        mxt_input_touchevent(data, &message, id);
@@ -682,10 +782,8 @@ static irqreturn_t mxt_process_messages_until_invalid(struct mxt_data *data)
                }
        } while (reportid != 0xff);
 
-       if (update_input) {
-               input_mt_report_pointer_emulation(data->input_dev, false);
-               input_sync(data->input_dev);
-       }
+       if (update_input)
+               mxt_input_sync(data->input_dev);
 
        return IRQ_HANDLED;
 }
@@ -700,6 +798,9 @@ static irqreturn_t mxt_interrupt(int irq, void *dev_id)
                return IRQ_HANDLED;
        }
 
+       if (!data->object_table)
+               return IRQ_HANDLED;
+
        return mxt_process_messages_until_invalid(data);
 }
 
@@ -756,6 +857,24 @@ static int mxt_soft_reset(struct mxt_data *data)
        return 0;
 }
 
+static void mxt_update_crc(struct mxt_data *data, u8 cmd, u8 value)
+{
+       /*
+        * On failure, CRC is set to 0 and config will always be
+        * downloaded.
+        */
+       data->config_crc = 0;
+       reinit_completion(&data->crc_completion);
+
+       mxt_t6_command(data, cmd, value, true);
+
+       /*
+        * Wait for crc message. On failure, CRC is set to 0 and config will
+        * always be downloaded.
+        */
+       mxt_wait_for_completion(data, &data->crc_completion, MXT_CRC_TIMEOUT);
+}
+
 static int mxt_check_reg_init(struct mxt_data *data)
 {
        const struct mxt_platform_data *pdata = data->pdata;
@@ -770,6 +889,16 @@ static int mxt_check_reg_init(struct mxt_data *data)
                return 0;
        }
 
+       mxt_update_crc(data, MXT_COMMAND_REPORTALL, 1);
+
+       if (data->config_crc == pdata->config_crc) {
+               dev_info(dev, "Config CRC 0x%06X: OK\n", data->config_crc);
+               return 0;
+       }
+
+       dev_info(dev, "Config CRC 0x%06X: does not match 0x%06X\n",
+                data->config_crc, pdata->config_crc);
+
        for (i = 0; i < data->info.object_num; i++) {
                object = data->object_table + i;
 
@@ -789,6 +918,14 @@ static int mxt_check_reg_init(struct mxt_data *data)
                index += size;
        }
 
+       mxt_update_crc(data, MXT_COMMAND_BACKUPNV, MXT_BACKUP_VALUE);
+
+       ret = mxt_soft_reset(data);
+       if (ret)
+               return ret;
+
+       dev_info(dev, "Config successfully updated\n");
+
        return 0;
 }
 
@@ -814,6 +951,19 @@ static int mxt_make_highchg(struct mxt_data *data)
        return 0;
 }
 
+static int mxt_acquire_irq(struct mxt_data *data)
+{
+       int error;
+
+       enable_irq(data->irq);
+
+       error = mxt_make_highchg(data);
+       if (error)
+               return error;
+
+       return 0;
+}
+
 static int mxt_get_info(struct mxt_data *data)
 {
        struct i2c_client *client = data->client;
@@ -832,20 +982,29 @@ static int mxt_get_object_table(struct mxt_data *data)
 {
        struct i2c_client *client = data->client;
        size_t table_size;
+       struct mxt_object *object_table;
        int error;
        int i;
        u8 reportid;
 
        table_size = data->info.object_num * sizeof(struct mxt_object);
+       object_table = kzalloc(table_size, GFP_KERNEL);
+       if (!object_table) {
+               dev_err(&data->client->dev, "Failed to allocate memory\n");
+               return -ENOMEM;
+       }
+
        error = __mxt_read_reg(client, MXT_OBJECT_START, table_size,
-                       data->object_table);
-       if (error)
+                       object_table);
+       if (error) {
+               kfree(object_table);
                return error;
+       }
 
        /* Valid Report IDs start counting from 1 */
        reportid = 1;
        for (i = 0; i < data->info.object_num; i++) {
-               struct mxt_object *object = data->object_table + i;
+               struct mxt_object *object = object_table + i;
                u8 min_id, max_id;
 
                le16_to_cpus(&object->start_address);
@@ -861,7 +1020,7 @@ static int mxt_get_object_table(struct mxt_data *data)
                }
 
                dev_dbg(&data->client->dev,
-                       "Type %2d Start %3d Size %3zd Instances %2zd ReportIDs %3u : %3u\n",
+                       "T%u Start:%u Size:%zu Instances:%zu Report IDs:%u-%u\n",
                        object->type, object->start_address,
                        mxt_obj_size(object), mxt_obj_instances(object),
                        min_id, max_id);
@@ -881,11 +1040,16 @@ static int mxt_get_object_table(struct mxt_data *data)
                }
        }
 
+       data->object_table = object_table;
+
        return 0;
 }
 
 static void mxt_free_object_table(struct mxt_data *data)
 {
+       input_unregister_device(data->input_dev);
+       data->input_dev = NULL;
+
        kfree(data->object_table);
        data->object_table = NULL;
        data->T6_reportid = 0;
@@ -894,60 +1058,187 @@ static void mxt_free_object_table(struct mxt_data *data)
        data->T19_reportid = 0;
 }
 
-static int mxt_initialize(struct mxt_data *data)
+static int mxt_read_t9_resolution(struct mxt_data *data)
 {
        struct i2c_client *client = data->client;
-       struct mxt_info *info = &data->info;
        int error;
-       u8 val;
+       struct t9_range range;
+       unsigned char orient;
+       struct mxt_object *object;
 
-       error = mxt_get_info(data);
+       object = mxt_get_object(data, MXT_TOUCH_MULTI_T9);
+       if (!object)
+               return -EINVAL;
+
+       error = __mxt_read_reg(client,
+                              object->start_address + MXT_T9_RANGE,
+                              sizeof(range), &range);
        if (error)
                return error;
 
-       data->object_table = kcalloc(info->object_num,
-                                    sizeof(struct mxt_object),
-                                    GFP_KERNEL);
-       if (!data->object_table) {
-               dev_err(&client->dev, "Failed to allocate memory\n");
+       le16_to_cpus(&range.x);
+       le16_to_cpus(&range.y);
+
+       error =  __mxt_read_reg(client,
+                               object->start_address + MXT_T9_ORIENT,
+                               1, &orient);
+       if (error)
+               return error;
+
+       /* Handle default values */
+       if (range.x == 0)
+               range.x = 1023;
+
+       if (range.y == 0)
+               range.y = 1023;
+
+       if (orient & MXT_T9_ORIENT_SWITCH) {
+               data->max_x = range.y;
+               data->max_y = range.x;
+       } else {
+               data->max_x = range.x;
+               data->max_y = range.y;
+       }
+
+       dev_dbg(&client->dev,
+               "Touchscreen size X%uY%u\n", data->max_x, data->max_y);
+
+       return 0;
+}
+
+static int mxt_input_open(struct input_dev *dev);
+static void mxt_input_close(struct input_dev *dev);
+
+static int mxt_initialize_t9_input_device(struct mxt_data *data)
+{
+       struct device *dev = &data->client->dev;
+       const struct mxt_platform_data *pdata = data->pdata;
+       struct input_dev *input_dev;
+       int error;
+       unsigned int num_mt_slots;
+       unsigned int mt_flags = 0;
+       int i;
+
+       error = mxt_read_t9_resolution(data);
+       if (error)
+               dev_warn(dev, "Failed to initialize T9 resolution\n");
+
+       input_dev = input_allocate_device();
+       if (!input_dev) {
+               dev_err(dev, "Failed to allocate memory\n");
                return -ENOMEM;
        }
 
+       input_dev->name = "Atmel maXTouch Touchscreen";
+       input_dev->phys = data->phys;
+       input_dev->id.bustype = BUS_I2C;
+       input_dev->dev.parent = dev;
+       input_dev->open = mxt_input_open;
+       input_dev->close = mxt_input_close;
+
+       __set_bit(EV_ABS, input_dev->evbit);
+       __set_bit(EV_KEY, input_dev->evbit);
+       __set_bit(BTN_TOUCH, input_dev->keybit);
+
+       if (pdata->t19_num_keys) {
+               __set_bit(INPUT_PROP_BUTTONPAD, input_dev->propbit);
+
+               for (i = 0; i < pdata->t19_num_keys; i++)
+                       if (pdata->t19_keymap[i] != KEY_RESERVED)
+                               input_set_capability(input_dev, EV_KEY,
+                                                    pdata->t19_keymap[i]);
+
+               mt_flags |= INPUT_MT_POINTER;
+
+               input_abs_set_res(input_dev, ABS_X, MXT_PIXELS_PER_MM);
+               input_abs_set_res(input_dev, ABS_Y, MXT_PIXELS_PER_MM);
+               input_abs_set_res(input_dev, ABS_MT_POSITION_X,
+                                 MXT_PIXELS_PER_MM);
+               input_abs_set_res(input_dev, ABS_MT_POSITION_Y,
+                                 MXT_PIXELS_PER_MM);
+
+               input_dev->name = "Atmel maXTouch Touchpad";
+       }
+
+       /* For single touch */
+       input_set_abs_params(input_dev, ABS_X,
+                            0, data->max_x, 0, 0);
+       input_set_abs_params(input_dev, ABS_Y,
+                            0, data->max_y, 0, 0);
+       input_set_abs_params(input_dev, ABS_PRESSURE,
+                            0, 255, 0, 0);
+
+       /* For multi touch */
+       num_mt_slots = data->T9_reportid_max - data->T9_reportid_min + 1;
+       error = input_mt_init_slots(input_dev, num_mt_slots, mt_flags);
+       if (error) {
+               dev_err(dev, "Error %d initialising slots\n", error);
+               goto err_free_mem;
+       }
+
+       input_set_abs_params(input_dev, ABS_MT_TOUCH_MAJOR,
+                            0, MXT_MAX_AREA, 0, 0);
+       input_set_abs_params(input_dev, ABS_MT_POSITION_X,
+                            0, data->max_x, 0, 0);
+       input_set_abs_params(input_dev, ABS_MT_POSITION_Y,
+                            0, data->max_y, 0, 0);
+       input_set_abs_params(input_dev, ABS_MT_PRESSURE,
+                            0, 255, 0, 0);
+
+       input_set_drvdata(input_dev, data);
+
+       error = input_register_device(input_dev);
+       if (error) {
+               dev_err(dev, "Error %d registering input device\n", error);
+               goto err_free_mem;
+       }
+
+       data->input_dev = input_dev;
+
+       return 0;
+
+err_free_mem:
+       input_free_device(input_dev);
+       return error;
+}
+
+static int mxt_initialize(struct mxt_data *data)
+{
+       struct i2c_client *client = data->client;
+       struct mxt_info *info = &data->info;
+       int error;
+
+       error = mxt_get_info(data);
+       if (error)
+               return error;
+
        /* Get object table information */
        error = mxt_get_object_table(data);
+       if (error) {
+               dev_err(&client->dev, "Error %d reading object table\n", error);
+               return error;
+       }
+
+       mxt_acquire_irq(data);
        if (error)
                goto err_free_object_table;
 
        /* Check register init values */
        error = mxt_check_reg_init(data);
-       if (error)
-               goto err_free_object_table;
-
-       error = mxt_t6_command(data, MXT_COMMAND_BACKUPNV,
-                              MXT_BACKUP_VALUE, false);
-       if (!error)
-               mxt_soft_reset(data);
-
-       /* Update matrix size at info struct */
-       error = mxt_read_reg(client, MXT_MATRIX_X_SIZE, &val);
-       if (error)
+       if (error) {
+               dev_err(&client->dev, "Error %d initializing configuration\n",
+                       error);
                goto err_free_object_table;
-       info->matrix_xsize = val;
+       }
 
-       error = mxt_read_reg(client, MXT_MATRIX_Y_SIZE, &val);
+       error = mxt_initialize_t9_input_device(data);
        if (error)
                goto err_free_object_table;
-       info->matrix_ysize = val;
-
-       dev_info(&client->dev,
-                       "Family ID: %u Variant ID: %u Major.Minor.Build: %u.%u.%02X\n",
-                       info->family_id, info->variant_id, info->version >> 4,
-                       info->version & 0xf, info->build);
 
        dev_info(&client->dev,
-                       "Matrix X Size: %u Matrix Y Size: %u Object Num: %u\n",
-                       info->matrix_xsize, info->matrix_ysize,
-                       info->object_num);
+                "Family: %u Variant: %u Firmware V%u.%u.%02X Objects: %u\n",
+                info->family_id, info->variant_id, info->version >> 4,
+                info->version & 0xf, info->build, info->object_num);
 
        return 0;
 
@@ -956,20 +1247,6 @@ err_free_object_table:
        return error;
 }
 
-static void mxt_calc_resolution(struct mxt_data *data)
-{
-       unsigned int max_x = data->pdata->x_size - 1;
-       unsigned int max_y = data->pdata->y_size - 1;
-
-       if (data->pdata->orient & MXT_XY_SWITCH) {
-               data->max_x = max_y;
-               data->max_y = max_x;
-       } else {
-               data->max_x = max_x;
-               data->max_y = max_y;
-       }
-}
-
 /* Firmware Version is returned as Major.Minor.Build */
 static ssize_t mxt_fw_version_show(struct device *dev,
                                   struct device_attribute *attr, char *buf)
@@ -1050,13 +1327,38 @@ done:
        return error ?: count;
 }
 
+static int mxt_check_firmware_format(struct device *dev,
+                                    const struct firmware *fw)
+{
+       unsigned int pos = 0;
+       char c;
+
+       while (pos < fw->size) {
+               c = *(fw->data + pos);
+
+               if (c < '0' || (c > '9' && c < 'A') || c > 'F')
+                       return 0;
+
+               pos++;
+       }
+
+       /*
+        * To convert file try:
+        * xxd -r -p mXTXXX__APP_VX-X-XX.enc > maxtouch.fw
+        */
+       dev_err(dev, "Aborting: firmware file must be in binary format\n");
+
+       return -EINVAL;
+}
+
 static int mxt_load_fw(struct device *dev, const char *fn)
 {
        struct mxt_data *data = dev_get_drvdata(dev);
-       struct i2c_client *client = data->client;
        const struct firmware *fw = NULL;
        unsigned int frame_size;
        unsigned int pos = 0;
+       unsigned int retry = 0;
+       unsigned int frame = 0;
        int ret;
 
        ret = request_firmware(&fw, fn, dev);
@@ -1065,6 +1367,15 @@ static int mxt_load_fw(struct device *dev, const char *fn)
                return ret;
        }
 
+       /* Check for incorrect enc file */
+       ret = mxt_check_firmware_format(dev, fw);
+       if (ret)
+               goto release_firmware;
+
+       ret = mxt_lookup_bootloader_address(data);
+       if (ret)
+               goto release_firmware;
+
        /* Change to the bootloader mode */
        data->in_bootloader = true;
 
@@ -1074,12 +1385,6 @@ static int mxt_load_fw(struct device *dev, const char *fn)
 
        msleep(MXT_RESET_TIME);
 
-       /* Change to slave address of bootloader */
-       if (client->addr == MXT_APP_LOW)
-               client->addr = MXT_BOOT_LOW;
-       else
-               client->addr = MXT_BOOT_HIGH;
-
        reinit_completion(&data->bl_completion);
 
        ret = mxt_check_bootloader(data, MXT_WAITING_BOOTLOAD_CMD);
@@ -1087,7 +1392,7 @@ static int mxt_load_fw(struct device *dev, const char *fn)
                goto disable_irq;
 
        /* Unlock bootloader */
-       mxt_unlock_bootloader(client);
+       mxt_unlock_bootloader(data);
 
        while (pos < fw->size) {
                ret = mxt_check_bootloader(data, MXT_WAITING_FRAME_DATA);
@@ -1096,21 +1401,34 @@ static int mxt_load_fw(struct device *dev, const char *fn)
 
                frame_size = ((*(fw->data + pos) << 8) | *(fw->data + pos + 1));
 
-               /* We should add 2 at frame size as the the firmware data is not
-                * included the CRC bytes.
-                */
+               /* Take account of CRC bytes */
                frame_size += 2;
 
                /* Write one frame to device */
-               mxt_fw_write(client, fw->data + pos, frame_size);
-
-               ret = mxt_check_bootloader(data, MXT_FRAME_CRC_PASS);
+               ret = mxt_bootloader_write(data, fw->data + pos, frame_size);
                if (ret)
                        goto disable_irq;
 
-               pos += frame_size;
+               ret = mxt_check_bootloader(data, MXT_FRAME_CRC_PASS);
+               if (ret) {
+                       retry++;
+
+                       /* Back off by 20ms per retry */
+                       msleep(retry * 20);
 
-               dev_dbg(dev, "Updated %d bytes / %zd bytes\n", pos, fw->size);
+                       if (retry > 20) {
+                               dev_err(dev, "Retry count exceeded\n");
+                               goto disable_irq;
+                       }
+               } else {
+                       retry = 0;
+                       pos += frame_size;
+                       frame++;
+               }
+
+               if (frame % 50 == 0)
+                       dev_dbg(dev, "Sent %d frames, %d/%zd bytes\n",
+                               frame, pos, fw->size);
        }
 
        /* Wait for flash. */
@@ -1119,6 +1437,8 @@ static int mxt_load_fw(struct device *dev, const char *fn)
        if (ret)
                goto disable_irq;
 
+       dev_dbg(dev, "Sent %d frames, %d bytes\n", frame, pos);
+
        /*
         * Wait for device to reset. Some bootloader versions do not assert
         * the CHG line after bootloading has finished, so ignore potential
@@ -1132,13 +1452,6 @@ disable_irq:
        disable_irq(data->irq);
 release_firmware:
        release_firmware(fw);
-
-       /* Change to slave address of application */
-       if (client->addr == MXT_BOOT_LOW)
-               client->addr = MXT_APP_LOW;
-       else
-               client->addr = MXT_APP_HIGH;
-
        return ret;
 }
 
@@ -1154,14 +1467,11 @@ static ssize_t mxt_update_fw_store(struct device *dev,
                dev_err(dev, "The firmware update failed(%d)\n", error);
                count = error;
        } else {
-               dev_dbg(dev, "The firmware update succeeded\n");
-               mxt_free_object_table(data);
+               dev_info(dev, "The firmware update succeeded\n");
 
-               mxt_initialize(data);
-
-               enable_irq(data->irq);
+               mxt_free_object_table(data);
 
-               error = mxt_make_highchg(data);
+               error = mxt_initialize(data);
                if (error)
                        return error;
        }
@@ -1219,130 +1529,59 @@ static void mxt_input_close(struct input_dev *dev)
 static int mxt_probe(struct i2c_client *client,
                const struct i2c_device_id *id)
 {
-       const struct mxt_platform_data *pdata = dev_get_platdata(&client->dev);
        struct mxt_data *data;
-       struct input_dev *input_dev;
+       const struct mxt_platform_data *pdata = dev_get_platdata(&client->dev);
        int error;
-       unsigned int num_mt_slots;
-       unsigned int mt_flags = 0;
-       int i;
 
        if (!pdata)
                return -EINVAL;
 
        data = kzalloc(sizeof(struct mxt_data), GFP_KERNEL);
-       input_dev = input_allocate_device();
-       if (!data || !input_dev) {
+       if (!data) {
                dev_err(&client->dev, "Failed to allocate memory\n");
-               error = -ENOMEM;
-               goto err_free_mem;
+               return -ENOMEM;
        }
 
-       input_dev->name = "Atmel maXTouch Touchscreen";
        snprintf(data->phys, sizeof(data->phys), "i2c-%u-%04x/input0",
                 client->adapter->nr, client->addr);
 
-       input_dev->phys = data->phys;
-
-       input_dev->id.bustype = BUS_I2C;
-       input_dev->dev.parent = &client->dev;
-       input_dev->open = mxt_input_open;
-       input_dev->close = mxt_input_close;
-
        data->client = client;
-       data->input_dev = input_dev;
        data->pdata = pdata;
        data->irq = client->irq;
+       i2c_set_clientdata(client, data);
 
        init_completion(&data->bl_completion);
        init_completion(&data->reset_completion);
-
-       mxt_calc_resolution(data);
-
-       error = mxt_initialize(data);
-       if (error)
-               goto err_free_mem;
-
-       __set_bit(EV_ABS, input_dev->evbit);
-       __set_bit(EV_KEY, input_dev->evbit);
-       __set_bit(BTN_TOUCH, input_dev->keybit);
-
-       if (pdata->t19_num_keys) {
-               __set_bit(INPUT_PROP_BUTTONPAD, input_dev->propbit);
-
-               for (i = 0; i < pdata->t19_num_keys; i++)
-                       if (pdata->t19_keymap[i] != KEY_RESERVED)
-                               input_set_capability(input_dev, EV_KEY,
-                                                    pdata->t19_keymap[i]);
-
-               mt_flags |= INPUT_MT_POINTER;
-
-               input_abs_set_res(input_dev, ABS_X, MXT_PIXELS_PER_MM);
-               input_abs_set_res(input_dev, ABS_Y, MXT_PIXELS_PER_MM);
-               input_abs_set_res(input_dev, ABS_MT_POSITION_X,
-                                 MXT_PIXELS_PER_MM);
-               input_abs_set_res(input_dev, ABS_MT_POSITION_Y,
-                                 MXT_PIXELS_PER_MM);
-
-               input_dev->name = "Atmel maXTouch Touchpad";
-       }
-
-       /* For single touch */
-       input_set_abs_params(input_dev, ABS_X,
-                            0, data->max_x, 0, 0);
-       input_set_abs_params(input_dev, ABS_Y,
-                            0, data->max_y, 0, 0);
-       input_set_abs_params(input_dev, ABS_PRESSURE,
-                            0, 255, 0, 0);
-
-       /* For multi touch */
-       num_mt_slots = data->T9_reportid_max - data->T9_reportid_min + 1;
-       error = input_mt_init_slots(input_dev, num_mt_slots, mt_flags);
-       if (error)
-               goto err_free_object;
-       input_set_abs_params(input_dev, ABS_MT_TOUCH_MAJOR,
-                            0, MXT_MAX_AREA, 0, 0);
-       input_set_abs_params(input_dev, ABS_MT_POSITION_X,
-                            0, data->max_x, 0, 0);
-       input_set_abs_params(input_dev, ABS_MT_POSITION_Y,
-                            0, data->max_y, 0, 0);
-       input_set_abs_params(input_dev, ABS_MT_PRESSURE,
-                            0, 255, 0, 0);
-
-       input_set_drvdata(input_dev, data);
-       i2c_set_clientdata(client, data);
+       init_completion(&data->crc_completion);
 
        error = request_threaded_irq(client->irq, NULL, mxt_interrupt,
                                     pdata->irqflags | IRQF_ONESHOT,
                                     client->name, data);
        if (error) {
                dev_err(&client->dev, "Failed to register interrupt\n");
-               goto err_free_object;
+               goto err_free_mem;
        }
 
-       error = mxt_make_highchg(data);
-       if (error)
-               goto err_free_irq;
+       disable_irq(client->irq);
 
-       error = input_register_device(input_dev);
+       error = mxt_initialize(data);
        if (error)
                goto err_free_irq;
 
        error = sysfs_create_group(&client->dev.kobj, &mxt_attr_group);
-       if (error)
-               goto err_unregister_device;
+       if (error) {
+               dev_err(&client->dev, "Failure %d creating sysfs group\n",
+                       error);
+               goto err_free_object;
+       }
 
        return 0;
 
-err_unregister_device:
-       input_unregister_device(input_dev);
-       input_dev = NULL;
-err_free_irq:
-       free_irq(client->irq, data);
 err_free_object:
        kfree(data->object_table);
+err_free_irq:
+       free_irq(client->irq, data);
 err_free_mem:
-       input_free_device(input_dev);
        kfree(data);
        return error;
 }