#include <linux/leds.h>
 #include <linux/module.h>
 #include <linux/of.h>
+#include <linux/of_device.h>
 #include <linux/pm.h>
 #include <linux/regulator/consumer.h>
 
 #define TM2_TOUCHKEY_DEV_NAME          "tm2-touchkey"
-#define TM2_TOUCHKEY_KEYCODE_REG       0x03
-#define TM2_TOUCHKEY_BASE_REG          0x00
+
 #define TM2_TOUCHKEY_CMD_LED_ON                0x10
 #define TM2_TOUCHKEY_CMD_LED_OFF       0x20
 #define TM2_TOUCHKEY_BIT_PRESS_EV      BIT(3)
        TM2_TOUCHKEY_KEY_BACK,
 };
 
+struct touchkey_variant {
+       u8 keycode_reg;
+       u8 base_reg;
+};
+
 struct tm2_touchkey_data {
        struct i2c_client *client;
        struct input_dev *input_dev;
        struct led_classdev led_dev;
        struct regulator *vdd;
        struct regulator_bulk_data regulators[2];
+       const struct touchkey_variant *variant;
+};
+
+static const struct touchkey_variant tm2_touchkey_variant = {
+       .keycode_reg = 0x03,
+       .base_reg = 0x00,
+};
+
+static const struct touchkey_variant midas_touchkey_variant = {
+       .keycode_reg = 0x00,
+       .base_reg = 0x00,
 };
 
 static void tm2_touchkey_led_brightness_set(struct led_classdev *led_dev,
 
        regulator_set_voltage(touchkey->vdd, volt, volt);
        i2c_smbus_write_byte_data(touchkey->client,
-                                 TM2_TOUCHKEY_BASE_REG, data);
+                                 touchkey->variant->base_reg, data);
 }
 
 static int tm2_touchkey_power_enable(struct tm2_touchkey_data *touchkey)
        int key;
 
        data = i2c_smbus_read_byte_data(touchkey->client,
-                                       TM2_TOUCHKEY_KEYCODE_REG);
+                                       touchkey->variant->keycode_reg);
        if (data < 0) {
                dev_err(&touchkey->client->dev,
                        "failed to read i2c data: %d\n", data);
        touchkey->client = client;
        i2c_set_clientdata(client, touchkey);
 
+       touchkey->variant = of_device_get_match_data(&client->dev);
+
        touchkey->regulators[0].supply = "vcc";
        touchkey->regulators[1].supply = "vdd";
        error = devm_regulator_bulk_get(&client->dev,
 MODULE_DEVICE_TABLE(i2c, tm2_touchkey_id_table);
 
 static const struct of_device_id tm2_touchkey_of_match[] = {
-       { .compatible = "cypress,tm2-touchkey", },
+       {
+               .compatible = "cypress,tm2-touchkey",
+               .data = &tm2_touchkey_variant,
+       }, {
+               .compatible = "cypress,midas-touchkey",
+               .data = &midas_touchkey_variant,
+       },
        { },
 };
 MODULE_DEVICE_TABLE(of, tm2_touchkey_of_match);