*
  */
 
+#include <linux/clk.h>
 #include <linux/module.h>
 #include <linux/interrupt.h>
 #include <linux/uaccess.h>
 #include <linux/sched.h>
 #include <linux/wait.h>
 
-#include <plat/dmtimer.h>
-#include <plat/clock.h>
-
 #include <media/lirc.h>
 #include <media/lirc_dev.h>
+#include <linux/platform_data/pwm_omap_dmtimer.h>
 #include <linux/platform_data/media/ir-rx51.h>
 
 #define LIRC_RX51_DRIVER_FEATURES (LIRC_CAN_SET_SEND_DUTY_CYCLE |      \
 #define TIMER_MAX_VALUE 0xffffffff
 
 struct lirc_rx51 {
-       struct omap_dm_timer *pwm_timer;
-       struct omap_dm_timer *pulse_timer;
+       pwm_omap_dmtimer *pwm_timer;
+       pwm_omap_dmtimer *pulse_timer;
+       struct pwm_omap_dmtimer_pdata *dmtimer;
        struct device        *dev;
        struct lirc_rx51_platform_data *pdata;
        wait_queue_head_t     wqueue;
 
 static void lirc_rx51_on(struct lirc_rx51 *lirc_rx51)
 {
-       omap_dm_timer_set_pwm(lirc_rx51->pwm_timer, 0, 1,
-                             OMAP_TIMER_TRIGGER_OVERFLOW_AND_COMPARE);
+       lirc_rx51->dmtimer->set_pwm(lirc_rx51->pwm_timer, 0, 1,
+                               PWM_OMAP_DMTIMER_TRIGGER_OVERFLOW_AND_COMPARE);
 }
 
 static void lirc_rx51_off(struct lirc_rx51 *lirc_rx51)
 {
-       omap_dm_timer_set_pwm(lirc_rx51->pwm_timer, 0, 1,
-                             OMAP_TIMER_TRIGGER_NONE);
+       lirc_rx51->dmtimer->set_pwm(lirc_rx51->pwm_timer, 0, 1,
+                                   PWM_OMAP_DMTIMER_TRIGGER_NONE);
 }
 
 static int init_timing_params(struct lirc_rx51 *lirc_rx51)
 
        load = -(lirc_rx51->fclk_khz * 1000 / lirc_rx51->freq);
        match = -(lirc_rx51->duty_cycle * -load / 100);
-       omap_dm_timer_set_load(lirc_rx51->pwm_timer, 1, load);
-       omap_dm_timer_set_match(lirc_rx51->pwm_timer, 1, match);
-       omap_dm_timer_write_counter(lirc_rx51->pwm_timer, TIMER_MAX_VALUE - 2);
-       omap_dm_timer_start(lirc_rx51->pwm_timer);
-       omap_dm_timer_set_int_enable(lirc_rx51->pulse_timer, 0);
-       omap_dm_timer_start(lirc_rx51->pulse_timer);
+       lirc_rx51->dmtimer->set_load(lirc_rx51->pwm_timer, 1, load);
+       lirc_rx51->dmtimer->set_match(lirc_rx51->pwm_timer, 1, match);
+       lirc_rx51->dmtimer->write_counter(lirc_rx51->pwm_timer, TIMER_MAX_VALUE - 2);
+       lirc_rx51->dmtimer->start(lirc_rx51->pwm_timer);
+       lirc_rx51->dmtimer->set_int_enable(lirc_rx51->pulse_timer, 0);
+       lirc_rx51->dmtimer->start(lirc_rx51->pulse_timer);
 
        lirc_rx51->match = 0;
 
        BUG_ON(usec < 0);
 
        if (lirc_rx51->match == 0)
-               counter = omap_dm_timer_read_counter(lirc_rx51->pulse_timer);
+               counter = lirc_rx51->dmtimer->read_counter(lirc_rx51->pulse_timer);
        else
                counter = lirc_rx51->match;
 
        counter += (u32)(lirc_rx51->fclk_khz * usec / (1000));
-       omap_dm_timer_set_match(lirc_rx51->pulse_timer, 1, counter);
-       omap_dm_timer_set_int_enable(lirc_rx51->pulse_timer,
-                                    OMAP_TIMER_INT_MATCH);
-       if (tics_after(omap_dm_timer_read_counter(lirc_rx51->pulse_timer),
+       lirc_rx51->dmtimer->set_match(lirc_rx51->pulse_timer, 1, counter);
+       lirc_rx51->dmtimer->set_int_enable(lirc_rx51->pulse_timer,
+                                          PWM_OMAP_DMTIMER_INT_MATCH);
+       if (tics_after(lirc_rx51->dmtimer->read_counter(lirc_rx51->pulse_timer),
                       counter)) {
                return 1;
        }
        unsigned int retval;
        struct lirc_rx51 *lirc_rx51 = ptr;
 
-       retval = omap_dm_timer_read_status(lirc_rx51->pulse_timer);
+       retval = lirc_rx51->dmtimer->read_status(lirc_rx51->pulse_timer);
        if (!retval)
                return IRQ_NONE;
 
-       if (retval & ~OMAP_TIMER_INT_MATCH)
+       if (retval & ~PWM_OMAP_DMTIMER_INT_MATCH)
                dev_err_ratelimited(lirc_rx51->dev,
                                ": Unexpected interrupt source: %x\n", retval);
 
-       omap_dm_timer_write_status(lirc_rx51->pulse_timer,
-                               OMAP_TIMER_INT_MATCH    |
-                               OMAP_TIMER_INT_OVERFLOW |
-                               OMAP_TIMER_INT_CAPTURE);
+       lirc_rx51->dmtimer->write_status(lirc_rx51->pulse_timer,
+                                        PWM_OMAP_DMTIMER_INT_MATCH |
+                                        PWM_OMAP_DMTIMER_INT_OVERFLOW |
+                                        PWM_OMAP_DMTIMER_INT_CAPTURE);
        if (lirc_rx51->wbuf_index < 0) {
                dev_err_ratelimited(lirc_rx51->dev,
                                ": BUG wbuf_index has value of %i\n",
        /* Stop TX here */
        lirc_rx51_off(lirc_rx51);
        lirc_rx51->wbuf_index = -1;
-       omap_dm_timer_stop(lirc_rx51->pwm_timer);
-       omap_dm_timer_stop(lirc_rx51->pulse_timer);
-       omap_dm_timer_set_int_enable(lirc_rx51->pulse_timer, 0);
+       lirc_rx51->dmtimer->stop(lirc_rx51->pwm_timer);
+       lirc_rx51->dmtimer->stop(lirc_rx51->pulse_timer);
+       lirc_rx51->dmtimer->set_int_enable(lirc_rx51->pulse_timer, 0);
        wake_up_interruptible(&lirc_rx51->wqueue);
 
        return IRQ_HANDLED;
        struct clk *clk_fclk;
        int retval, pwm_timer = lirc_rx51->pwm_timer_num;
 
-       lirc_rx51->pwm_timer = omap_dm_timer_request_specific(pwm_timer);
+       lirc_rx51->pwm_timer = lirc_rx51->dmtimer->request_specific(pwm_timer);
        if (lirc_rx51->pwm_timer == NULL) {
                dev_err(lirc_rx51->dev, ": Error requesting GPT%d timer\n",
                        pwm_timer);
                return -EBUSY;
        }
 
-       lirc_rx51->pulse_timer = omap_dm_timer_request();
+       lirc_rx51->pulse_timer = lirc_rx51->dmtimer->request();
        if (lirc_rx51->pulse_timer == NULL) {
                dev_err(lirc_rx51->dev, ": Error requesting pulse timer\n");
                retval = -EBUSY;
                goto err1;
        }
 
-       omap_dm_timer_set_source(lirc_rx51->pwm_timer, OMAP_TIMER_SRC_SYS_CLK);
-       omap_dm_timer_set_source(lirc_rx51->pulse_timer,
-                               OMAP_TIMER_SRC_SYS_CLK);
+       lirc_rx51->dmtimer->set_source(lirc_rx51->pwm_timer,
+                                      PWM_OMAP_DMTIMER_SRC_SYS_CLK);
+       lirc_rx51->dmtimer->set_source(lirc_rx51->pulse_timer,
+                                      PWM_OMAP_DMTIMER_SRC_SYS_CLK);
 
-       omap_dm_timer_enable(lirc_rx51->pwm_timer);
-       omap_dm_timer_enable(lirc_rx51->pulse_timer);
+       lirc_rx51->dmtimer->enable(lirc_rx51->pwm_timer);
+       lirc_rx51->dmtimer->enable(lirc_rx51->pulse_timer);
 
-       lirc_rx51->irq_num = omap_dm_timer_get_irq(lirc_rx51->pulse_timer);
+       lirc_rx51->irq_num = lirc_rx51->dmtimer->get_irq(lirc_rx51->pulse_timer);
        retval = request_irq(lirc_rx51->irq_num, lirc_rx51_interrupt_handler,
                             IRQF_SHARED, "lirc_pulse_timer", lirc_rx51);
        if (retval) {
                goto err2;
        }
 
-       clk_fclk = omap_dm_timer_get_fclk(lirc_rx51->pwm_timer);
-       lirc_rx51->fclk_khz = clk_fclk->rate / 1000;
+       clk_fclk = lirc_rx51->dmtimer->get_fclk(lirc_rx51->pwm_timer);
+       lirc_rx51->fclk_khz = clk_get_rate(clk_fclk) / 1000;
 
        return 0;
 
 err2:
-       omap_dm_timer_free(lirc_rx51->pulse_timer);
+       lirc_rx51->dmtimer->free(lirc_rx51->pulse_timer);
 err1:
-       omap_dm_timer_free(lirc_rx51->pwm_timer);
+       lirc_rx51->dmtimer->free(lirc_rx51->pwm_timer);
 
        return retval;
 }
 
 static int lirc_rx51_free_port(struct lirc_rx51 *lirc_rx51)
 {
-       omap_dm_timer_set_int_enable(lirc_rx51->pulse_timer, 0);
+       lirc_rx51->dmtimer->set_int_enable(lirc_rx51->pulse_timer, 0);
        free_irq(lirc_rx51->irq_num, lirc_rx51);
        lirc_rx51_off(lirc_rx51);
-       omap_dm_timer_disable(lirc_rx51->pwm_timer);
-       omap_dm_timer_disable(lirc_rx51->pulse_timer);
-       omap_dm_timer_free(lirc_rx51->pwm_timer);
-       omap_dm_timer_free(lirc_rx51->pulse_timer);
+       lirc_rx51->dmtimer->disable(lirc_rx51->pwm_timer);
+       lirc_rx51->dmtimer->disable(lirc_rx51->pulse_timer);
+       lirc_rx51->dmtimer->free(lirc_rx51->pwm_timer);
+       lirc_rx51->dmtimer->free(lirc_rx51->pulse_timer);
        lirc_rx51->wbuf_index = -1;
 
        return 0;
 {
        lirc_rx51_driver.features = LIRC_RX51_DRIVER_FEATURES;
        lirc_rx51.pdata = dev->dev.platform_data;
+       if (!lirc_rx51.pdata->dmtimer) {
+               dev_err(&dev->dev, "no dmtimer?\n");
+               return -ENODEV;
+       }
+
        lirc_rx51.pwm_timer_num = lirc_rx51.pdata->pwm_timer;
+       lirc_rx51.dmtimer = lirc_rx51.pdata->dmtimer;
        lirc_rx51.dev = &dev->dev;
        lirc_rx51_driver.dev = &dev->dev;
        lirc_rx51_driver.minor = lirc_register_driver(&lirc_rx51_driver);