#include <mach/gpio.h>
 #include <plat/mmc.h>
 
+#include "mux.h"
+
 #if defined(CONFIG_VIDEO_OMAP2) || defined(CONFIG_VIDEO_OMAP2_MODULE)
 
 static struct resource cam_resources[] = {
 
        if (cpu_is_omap34xx()) {
                if (controller_nr == 0) {
-                       omap_cfg_reg(N28_3430_MMC1_CLK);
-                       omap_cfg_reg(M27_3430_MMC1_CMD);
-                       omap_cfg_reg(N27_3430_MMC1_DAT0);
+                       omap_mux_init_signal("sdmmc1_clk",
+                               OMAP_PIN_INPUT_PULLUP);
+                       omap_mux_init_signal("sdmmc1_cmd",
+                               OMAP_PIN_INPUT_PULLUP);
+                       omap_mux_init_signal("sdmmc1_dat0",
+                               OMAP_PIN_INPUT_PULLUP);
                        if (mmc_controller->slots[0].wires == 4 ||
                                mmc_controller->slots[0].wires == 8) {
-                               omap_cfg_reg(N26_3430_MMC1_DAT1);
-                               omap_cfg_reg(N25_3430_MMC1_DAT2);
-                               omap_cfg_reg(P28_3430_MMC1_DAT3);
+                               omap_mux_init_signal("sdmmc1_dat1",
+                                       OMAP_PIN_INPUT_PULLUP);
+                               omap_mux_init_signal("sdmmc1_dat2",
+                                       OMAP_PIN_INPUT_PULLUP);
+                               omap_mux_init_signal("sdmmc1_dat3",
+                                       OMAP_PIN_INPUT_PULLUP);
                        }
                        if (mmc_controller->slots[0].wires == 8) {
-                               omap_cfg_reg(P27_3430_MMC1_DAT4);
-                               omap_cfg_reg(P26_3430_MMC1_DAT5);
-                               omap_cfg_reg(R27_3430_MMC1_DAT6);
-                               omap_cfg_reg(R25_3430_MMC1_DAT7);
+                               omap_mux_init_signal("sdmmc1_dat4",
+                                       OMAP_PIN_INPUT_PULLUP);
+                               omap_mux_init_signal("sdmmc1_dat5",
+                                       OMAP_PIN_INPUT_PULLUP);
+                               omap_mux_init_signal("sdmmc1_dat6",
+                                       OMAP_PIN_INPUT_PULLUP);
+                               omap_mux_init_signal("sdmmc1_dat7",
+                                       OMAP_PIN_INPUT_PULLUP);
                        }
                }
                if (controller_nr == 1) {
                        /* MMC2 */
-                       omap_cfg_reg(AE2_3430_MMC2_CLK);
-                       omap_cfg_reg(AG5_3430_MMC2_CMD);
-                       omap_cfg_reg(AH5_3430_MMC2_DAT0);
+                       omap_mux_init_signal("sdmmc2_clk",
+                               OMAP_PIN_INPUT_PULLUP);
+                       omap_mux_init_signal("sdmmc2_cmd",
+                               OMAP_PIN_INPUT_PULLUP);
+                       omap_mux_init_signal("sdmmc2_dat0",
+                               OMAP_PIN_INPUT_PULLUP);
 
                        /*
                         * For 8 wire configurations, Lines DAT4, 5, 6 and 7 need to be muxed
                         */
                        if (mmc_controller->slots[0].wires == 4 ||
                                mmc_controller->slots[0].wires == 8) {
-                               omap_cfg_reg(AH4_3430_MMC2_DAT1);
-                               omap_cfg_reg(AG4_3430_MMC2_DAT2);
-                               omap_cfg_reg(AF4_3430_MMC2_DAT3);
+                               omap_mux_init_signal("sdmmc2_dat1",
+                                       OMAP_PIN_INPUT_PULLUP);
+                               omap_mux_init_signal("sdmmc2_dat2",
+                                       OMAP_PIN_INPUT_PULLUP);
+                               omap_mux_init_signal("sdmmc2_dat3",
+                                       OMAP_PIN_INPUT_PULLUP);
                        }
                        if (mmc_controller->slots[0].wires == 8) {
-                               omap_cfg_reg(AE4_3430_MMC2_DAT4);
-                               omap_cfg_reg(AH3_3430_MMC2_DAT5);
-                               omap_cfg_reg(AF3_3430_MMC2_DAT6);
-                               omap_cfg_reg(AE3_3430_MMC2_DAT7);
+                               omap_mux_init_signal("sdmmc2_dat4.sdmmc2_dat4",
+                                       OMAP_PIN_INPUT_PULLUP);
+                               omap_mux_init_signal("sdmmc2_dat5.sdmmc2_dat5",
+                                       OMAP_PIN_INPUT_PULLUP);
+                               omap_mux_init_signal("sdmmc2_dat6.sdmmc2_dat6",
+                                       OMAP_PIN_INPUT_PULLUP);
+                               omap_mux_init_signal("sdmmc2_dat7.sdmmc2_dat7",
+                                       OMAP_PIN_INPUT_PULLUP);
                        }
                }
 
 
 #include <mach/irqs.h>
 #include <plat/usb.h>
 
+#include "mux.h"
+
 #if defined(CONFIG_USB_EHCI_HCD) || defined(CONFIG_USB_EHCI_HCD_MODULE)
 
 static struct resource ehci_resources[] = {
 {
        switch (port_mode[0]) {
        case EHCI_HCD_OMAP_MODE_PHY:
-               omap_cfg_reg(Y9_3430_USB1HS_PHY_STP);
-               omap_cfg_reg(Y8_3430_USB1HS_PHY_CLK);
-               omap_cfg_reg(AA14_3430_USB1HS_PHY_DIR);
-               omap_cfg_reg(AA11_3430_USB1HS_PHY_NXT);
-               omap_cfg_reg(W13_3430_USB1HS_PHY_DATA0);
-               omap_cfg_reg(W12_3430_USB1HS_PHY_DATA1);
-               omap_cfg_reg(W11_3430_USB1HS_PHY_DATA2);
-               omap_cfg_reg(Y11_3430_USB1HS_PHY_DATA3);
-               omap_cfg_reg(W9_3430_USB1HS_PHY_DATA4);
-               omap_cfg_reg(Y12_3430_USB1HS_PHY_DATA5);
-               omap_cfg_reg(W8_3430_USB1HS_PHY_DATA6);
-               omap_cfg_reg(Y13_3430_USB1HS_PHY_DATA7);
+               omap_mux_init_signal("hsusb1_stp", OMAP_PIN_OUTPUT);
+               omap_mux_init_signal("hsusb1_clk", OMAP_PIN_OUTPUT);
+               omap_mux_init_signal("hsusb1_dir", OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_nxt", OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_data0", OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_data1", OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_data2", OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_data3", OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_data4", OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_data5", OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_data6", OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_data7", OMAP_PIN_INPUT_PULLDOWN);
                break;
        case EHCI_HCD_OMAP_MODE_TLL:
-               omap_cfg_reg(Y9_3430_USB1HS_TLL_STP);
-               omap_cfg_reg(Y8_3430_USB1HS_TLL_CLK);
-               omap_cfg_reg(AA14_3430_USB1HS_TLL_DIR);
-               omap_cfg_reg(AA11_3430_USB1HS_TLL_NXT);
-               omap_cfg_reg(W13_3430_USB1HS_TLL_DATA0);
-               omap_cfg_reg(W12_3430_USB1HS_TLL_DATA1);
-               omap_cfg_reg(W11_3430_USB1HS_TLL_DATA2);
-               omap_cfg_reg(Y11_3430_USB1HS_TLL_DATA3);
-               omap_cfg_reg(W9_3430_USB1HS_TLL_DATA4);
-               omap_cfg_reg(Y12_3430_USB1HS_TLL_DATA5);
-               omap_cfg_reg(W8_3430_USB1HS_TLL_DATA6);
-               omap_cfg_reg(Y13_3430_USB1HS_TLL_DATA7);
+               omap_mux_init_signal("hsusb1_tll_stp",
+                       OMAP_PIN_INPUT_PULLUP);
+               omap_mux_init_signal("hsusb1_tll_clk",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_tll_dir",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_tll_nxt",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_tll_data0",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_tll_data1",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_tll_data2",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_tll_data3",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_tll_data4",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_tll_data5",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_tll_data6",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb1_tll_data7",
+                       OMAP_PIN_INPUT_PULLDOWN);
                break;
        case EHCI_HCD_OMAP_MODE_UNKNOWN:
                /* FALLTHROUGH */
 
        switch (port_mode[1]) {
        case EHCI_HCD_OMAP_MODE_PHY:
-               omap_cfg_reg(AA10_3430_USB2HS_PHY_STP);
-               omap_cfg_reg(AA8_3430_USB2HS_PHY_CLK);
-               omap_cfg_reg(AA9_3430_USB2HS_PHY_DIR);
-               omap_cfg_reg(AB11_3430_USB2HS_PHY_NXT);
-               omap_cfg_reg(AB10_3430_USB2HS_PHY_DATA0);
-               omap_cfg_reg(AB9_3430_USB2HS_PHY_DATA1);
-               omap_cfg_reg(W3_3430_USB2HS_PHY_DATA2);
-               omap_cfg_reg(T4_3430_USB2HS_PHY_DATA3);
-               omap_cfg_reg(T3_3430_USB2HS_PHY_DATA4);
-               omap_cfg_reg(R3_3430_USB2HS_PHY_DATA5);
-               omap_cfg_reg(R4_3430_USB2HS_PHY_DATA6);
-               omap_cfg_reg(T2_3430_USB2HS_PHY_DATA7);
+               omap_mux_init_signal("hsusb2_stp", OMAP_PIN_OUTPUT);
+               omap_mux_init_signal("hsusb2_clk", OMAP_PIN_OUTPUT);
+               omap_mux_init_signal("hsusb2_dir", OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_nxt", OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_data0",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_data1",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_data2",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_data3",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_data4",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_data5",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_data6",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_data7",
+                       OMAP_PIN_INPUT_PULLDOWN);
                break;
        case EHCI_HCD_OMAP_MODE_TLL:
-               omap_cfg_reg(AA10_3430_USB2HS_TLL_STP);
-               omap_cfg_reg(AA8_3430_USB2HS_TLL_CLK);
-               omap_cfg_reg(AA9_3430_USB2HS_TLL_DIR);
-               omap_cfg_reg(AB11_3430_USB2HS_TLL_NXT);
-               omap_cfg_reg(AB10_3430_USB2HS_TLL_DATA0);
-               omap_cfg_reg(AB9_3430_USB2HS_TLL_DATA1);
-               omap_cfg_reg(W3_3430_USB2HS_TLL_DATA2);
-               omap_cfg_reg(T4_3430_USB2HS_TLL_DATA3);
-               omap_cfg_reg(T3_3430_USB2HS_TLL_DATA4);
-               omap_cfg_reg(R3_3430_USB2HS_TLL_DATA5);
-               omap_cfg_reg(R4_3430_USB2HS_TLL_DATA6);
-               omap_cfg_reg(T2_3430_USB2HS_TLL_DATA7);
+               omap_mux_init_signal("hsusb2_tll_stp",
+                       OMAP_PIN_INPUT_PULLUP);
+               omap_mux_init_signal("hsusb2_tll_clk",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_tll_dir",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_tll_nxt",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_tll_data0",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_tll_data1",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_tll_data2",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_tll_data3",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_tll_data4",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_tll_data5",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_tll_data6",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb2_tll_data7",
+                       OMAP_PIN_INPUT_PULLDOWN);
                break;
        case EHCI_HCD_OMAP_MODE_UNKNOWN:
                /* FALLTHROUGH */
                printk(KERN_WARNING "Port3 can't be used in PHY mode\n");
                break;
        case EHCI_HCD_OMAP_MODE_TLL:
-               omap_cfg_reg(AB3_3430_USB3HS_TLL_STP);
-               omap_cfg_reg(AA6_3430_USB3HS_TLL_CLK);
-               omap_cfg_reg(AA3_3430_USB3HS_TLL_DIR);
-               omap_cfg_reg(Y3_3430_USB3HS_TLL_NXT);
-               omap_cfg_reg(AA5_3430_USB3HS_TLL_DATA0);
-               omap_cfg_reg(Y4_3430_USB3HS_TLL_DATA1);
-               omap_cfg_reg(Y5_3430_USB3HS_TLL_DATA2);
-               omap_cfg_reg(W5_3430_USB3HS_TLL_DATA3);
-               omap_cfg_reg(AB12_3430_USB3HS_TLL_DATA4);
-               omap_cfg_reg(AB13_3430_USB3HS_TLL_DATA5);
-               omap_cfg_reg(AA13_3430_USB3HS_TLL_DATA6);
-               omap_cfg_reg(AA12_3430_USB3HS_TLL_DATA7);
+               omap_mux_init_signal("hsusb3_tll_stp",
+                       OMAP_PIN_INPUT_PULLUP);
+               omap_mux_init_signal("hsusb3_tll_clk",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb3_tll_dir",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb3_tll_nxt",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb3_tll_data0",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb3_tll_data1",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb3_tll_data2",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb3_tll_data3",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb3_tll_data4",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb3_tll_data5",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb3_tll_data6",
+                       OMAP_PIN_INPUT_PULLDOWN);
+               omap_mux_init_signal("hsusb3_tll_data7",
+                       OMAP_PIN_INPUT_PULLDOWN);
                break;
        case EHCI_HCD_OMAP_MODE_UNKNOWN:
                /* FALLTHROUGH */