drivers/gpu/drm/rockchip/cdn-dp-core.c

   1 /*
   2  * Copyright (C) Fuzhou Rockchip Electronics Co.Ltd
   3  * Author: Chris Zhong <zyw@rock-chips.com>
   4  *
   5  * This software is licensed under the terms of the GNU General Public
   6  * License version 2, as published by the Free Software Foundation, and
   7  * may be copied, distributed, and modified under those terms.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  */
  14
  15 #include <drm/drmP.h>
  16 #include <drm/drm_atomic_helper.h>
  17 #include <drm/drm_crtc_helper.h>
  18 #include <drm/drm_dp_helper.h>
  19 #include <drm/drm_edid.h>
  20 #include <drm/drm_of.h>
  21
  22 #include <linux/clk.h>
  23 #include <linux/component.h>
  24 #include <linux/extcon.h>
  25 #include <linux/firmware.h>
  26 #include <linux/regmap.h>
  27 #include <linux/reset.h>
  28 #include <linux/mfd/syscon.h>
  29 #include <linux/phy/phy.h>
  30
  31 #include <sound/hdmi-codec.h>
  32
  33 #include "cdn-dp-core.h"
  34 #include "cdn-dp-reg.h"
  35 #include "rockchip_drm_vop.h"
  36
  37 #define connector_to_dp(c) \
  38                 container_of(c, struct cdn_dp_device, connector)
  39
  40 #define encoder_to_dp(c) \
  41                 container_of(c, struct cdn_dp_device, encoder)
  42
  43 #define GRF_SOC_CON9            0x6224
  44 #define DP_SEL_VOP_LIT          BIT(12)
  45 #define GRF_SOC_CON26           0x6268
  46 #define DPTX_HPD_SEL            (3 << 12)
  47 #define DPTX_HPD_DEL            (2 << 12)
  48 #define DPTX_HPD_SEL_MASK       (3 << 28)
  49
  50 #define CDN_FW_TIMEOUT_MS       (64 * 1000)
  51 #define CDN_DPCD_TIMEOUT_MS     5000
  52 #define CDN_DP_FIRMWARE         "rockchip/dptx.bin"
  53
  54 struct cdn_dp_data {
  55         u8 max_phy;
  56 };
  57
  58 struct cdn_dp_data rk3399_cdn_dp = {
  59         .max_phy = 2,
  60 };
  61
  62 static const struct of_device_id cdn_dp_dt_ids[] = {
  63         { .compatible = "rockchip,rk3399-cdn-dp",
  64                 .data = (void *)&rk3399_cdn_dp },
  65         {}
  66 };
  67
  68 MODULE_DEVICE_TABLE(of, cdn_dp_dt_ids);
  69
  70 static int cdn_dp_grf_write(struct cdn_dp_device *dp,
  71                             unsigned int reg, unsigned int val)
  72 {
  73         int ret;
  74
  75         ret = clk_prepare_enable(dp->grf_clk);
  76         if (ret) {
  77                 DRM_DEV_ERROR(dp->dev, "Failed to prepare_enable grf clock\n");
  78                 return ret;
  79         }
  80
  81         ret = regmap_write(dp->grf, reg, val);
  82         if (ret) {
  83                 DRM_DEV_ERROR(dp->dev, "Could not write to GRF: %d\n", ret);
  84                 return ret;
  85         }
  86
  87         clk_disable_unprepare(dp->grf_clk);
  88
  89         return 0;
  90 }
  91
  92 static int cdn_dp_clk_enable(struct cdn_dp_device *dp)
  93 {
  94         int ret;
  95         u32 rate;
  96
  97         ret = clk_prepare_enable(dp->pclk);
  98         if (ret < 0) {
  99                 DRM_DEV_ERROR(dp->dev, "cannot enable dp pclk %d\n", ret);
 100                 goto err_pclk;
 101         }
 102
 103         ret = clk_prepare_enable(dp->core_clk);
 104         if (ret < 0) {
 105                 DRM_DEV_ERROR(dp->dev, "cannot enable core_clk %d\n", ret);
 106                 goto err_core_clk;
 107         }
 108
 109         ret = pm_runtime_get_sync(dp->dev);
 110         if (ret < 0) {
 111                 DRM_DEV_ERROR(dp->dev, "cannot get pm runtime %d\n", ret);
 112                 goto err_pclk;
 113         }
 114
 115         reset_control_assert(dp->core_rst);
 116         reset_control_assert(dp->dptx_rst);
 117         reset_control_assert(dp->apb_rst);
 118         reset_control_deassert(dp->core_rst);
 119         reset_control_deassert(dp->dptx_rst);
 120         reset_control_deassert(dp->apb_rst);
 121
 122         rate = clk_get_rate(dp->core_clk);
 123         if (!rate) {
 124                 DRM_DEV_ERROR(dp->dev, "get clk rate failed: %d\n", rate);
 125                 goto err_set_rate;
 126         }
 127
 128         cdn_dp_set_fw_clk(dp, rate);
 129         cdn_dp_clock_reset(dp);
 130
 131         return 0;
 132
 133 err_set_rate:
 134         clk_disable_unprepare(dp->core_clk);
 135 err_core_clk:
 136         clk_disable_unprepare(dp->pclk);
 137 err_pclk:
 138         return ret;
 139 }
 140
 141 static void cdn_dp_clk_disable(struct cdn_dp_device *dp)
 142 {
 143         pm_runtime_put_sync(dp->dev);
 144         clk_disable_unprepare(dp->pclk);
 145         clk_disable_unprepare(dp->core_clk);
 146 }
 147
 148 static int cdn_dp_get_port_lanes(struct cdn_dp_port *port)
 149 {
 150         struct extcon_dev *edev = port->extcon;
 151         union extcon_property_value property;
 152         int dptx;
 153         u8 lanes;
 154
 155         dptx = extcon_get_state(edev, EXTCON_DISP_DP);
 156         if (dptx > 0) {
 157                 extcon_get_property(edev, EXTCON_DISP_DP,
 158                                     EXTCON_PROP_USB_SS, &property);
 159                 if (property.intval)
 160                         lanes = 2;
 161                 else
 162                         lanes = 4;
 163         } else {
 164                 lanes = 0;
 165         }
 166
 167         return lanes;
 168 }
 169
 170 static int cdn_dp_get_sink_count(struct cdn_dp_device *dp, u8 *sink_count)
 171 {
 172         int ret;
 173         u8 value;
 174
 175         *sink_count = 0;
 176         ret = cdn_dp_dpcd_read(dp, DP_SINK_COUNT, &value, 1);
 177         if (ret)
 178                 return ret;
 179
 180         *sink_count = DP_GET_SINK_COUNT(value);
 181         return 0;
 182 }
 183
 184 static struct cdn_dp_port *cdn_dp_connected_port(struct cdn_dp_device *dp)
 185 {
 186         struct cdn_dp_port *port;
 187         int i, lanes;
 188
 189         for (i = 0; i < dp->ports; i++) {
 190                 port = dp->port[i];
 191                 lanes = cdn_dp_get_port_lanes(port);
 192                 if (lanes)
 193                         return port;
 194         }
 195         return NULL;
 196 }
 197
 198 static enum drm_connector_status
 199 cdn_dp_connector_detect(struct drm_connector *connector, bool force)
 200 {
 201         struct cdn_dp_device *dp = connector_to_dp(connector);
 202         enum drm_connector_status status = connector_status_disconnected;
 203
 204         mutex_lock(&dp->lock);
 205         if (dp->connected)
 206                 status = connector_status_connected;
 207         mutex_unlock(&dp->lock);
 208
 209         return status;
 210 }
 211
 212 static void cdn_dp_connector_destroy(struct drm_connector *connector)
 213 {
 214         drm_connector_unregister(connector);
 215         drm_connector_cleanup(connector);
 216 }
 217
 218 static const struct drm_connector_funcs cdn_dp_atomic_connector_funcs = {
 219         .dpms = drm_atomic_helper_connector_dpms,
 220         .detect = cdn_dp_connector_detect,
 221         .destroy = cdn_dp_connector_destroy,
 222         .fill_modes = drm_helper_probe_single_connector_modes,
 223         .reset = drm_atomic_helper_connector_reset,
 224         .atomic_duplicate_state = drm_atomic_helper_connector_duplicate_state,
 225         .atomic_destroy_state = drm_atomic_helper_connector_destroy_state,
 226 };
 227
 228 static int cdn_dp_connector_get_modes(struct drm_connector *connector)
 229 {
 230         struct cdn_dp_device *dp = connector_to_dp(connector);
 231         struct edid *edid;
 232         int ret = 0;
 233
 234         mutex_lock(&dp->lock);
 235         edid = dp->edid;
 236         if (edid) {
 237                 DRM_DEV_DEBUG_KMS(dp->dev, "got edid: width[%d] x height[%d]\n",
 238                                   edid->width_cm, edid->height_cm);
 239
 240                 dp->sink_has_audio = drm_detect_monitor_audio(edid);
 241                 ret = drm_add_edid_modes(connector, edid);
 242                 if (ret) {
 243                         drm_mode_connector_update_edid_property(connector,
 244                                                                 edid);
 245                         drm_edid_to_eld(connector, edid);
 246                 }
 247         }
 248         mutex_unlock(&dp->lock);
 249
 250         return ret;
 251 }
 252
 253 static struct drm_encoder *
 254 cdn_dp_connector_best_encoder(struct drm_connector *connector)
 255 {
 256         struct cdn_dp_device *dp = connector_to_dp(connector);
 257
 258         return &dp->encoder;
 259 }
 260
 261 static int cdn_dp_connector_mode_valid(struct drm_connector *connector,
 262                                        struct drm_display_mode *mode)
 263 {
 264         struct cdn_dp_device *dp = connector_to_dp(connector);
 265         struct drm_display_info *display_info = &dp->connector.display_info;
 266         u32 requested, actual, rate, sink_max, source_max = 0;
 267         u8 lanes, bpc;
 268
 269         /* If DP is disconnected, every mode is invalid */
 270         if (!dp->connected)
 271                 return MODE_BAD;
 272
 273         switch (display_info->bpc) {
 274         case 10:
 275                 bpc = 10;
 276                 break;
 277         case 6:
 278                 bpc = 6;
 279                 break;
 280         default:
 281                 bpc = 8;
 282                 break;
 283         }
 284
 285         requested = mode->clock * bpc * 3 / 1000;
 286
 287         source_max = dp->lanes;
 288         sink_max = drm_dp_max_lane_count(dp->dpcd);
 289         lanes = min(source_max, sink_max);
 290
 291         source_max = drm_dp_bw_code_to_link_rate(CDN_DP_MAX_LINK_RATE);
 292         sink_max = drm_dp_max_link_rate(dp->dpcd);
 293         rate = min(source_max, sink_max);
 294
 295         actual = rate * lanes / 100;
 296
 297         /* efficiency is about 0.8 */
 298         actual = actual * 8 / 10;
 299
 300         if (requested > actual) {
 301                 DRM_DEV_DEBUG_KMS(dp->dev,
 302                                   "requested=%d, actual=%d, clock=%d\n",
 303                                   requested, actual, mode->clock);
 304                 return MODE_CLOCK_HIGH;
 305         }
 306
 307         return MODE_OK;
 308 }
 309
 310 static struct drm_connector_helper_funcs cdn_dp_connector_helper_funcs = {
 311         .get_modes = cdn_dp_connector_get_modes,
 312         .best_encoder = cdn_dp_connector_best_encoder,
 313         .mode_valid = cdn_dp_connector_mode_valid,
 314 };
 315
 316 static int cdn_dp_firmware_init(struct cdn_dp_device *dp)
 317 {
 318         int ret;
 319         const u32 *iram_data, *dram_data;
 320         const struct firmware *fw = dp->fw;
 321         const struct cdn_firmware_header *hdr;
 322
 323         hdr = (struct cdn_firmware_header *)fw->data;
 324         if (fw->size != le32_to_cpu(hdr->size_bytes)) {
 325                 DRM_DEV_ERROR(dp->dev, "firmware is invalid\n");
 326                 return -EINVAL;
 327         }
 328
 329         iram_data = (const u32 *)(fw->data + hdr->header_size);
 330         dram_data = (const u32 *)(fw->data + hdr->header_size + hdr->iram_size);
 331
 332         ret = cdn_dp_load_firmware(dp, iram_data, hdr->iram_size,
 333                                    dram_data, hdr->dram_size);
 334         if (ret)
 335                 return ret;
 336
 337         ret = cdn_dp_set_firmware_active(dp, true);
 338         if (ret) {
 339                 DRM_DEV_ERROR(dp->dev, "active ucpu failed: %d\n", ret);
 340                 return ret;
 341         }
 342
 343         return cdn_dp_event_config(dp);
 344 }
 345
 346 static int cdn_dp_get_sink_capability(struct cdn_dp_device *dp,
 347                                       struct cdn_dp_port *port,
 348                                       u8 *sink_count)
 349 {
 350         int ret;
 351         unsigned long timeout = jiffies + msecs_to_jiffies(CDN_DPCD_TIMEOUT_MS);
 352
 353         /*
 354          * Attempt to read sink count & sink capability, retry in case the sink
 355          * may not be ready.
 356          *
 357          * Sinks are *supposed* to come up within 1ms from an off state, but
 358          * some docks need more time to power up.
 359          */
 360         while (time_before(jiffies, timeout)) {
 361                 if (!extcon_get_state(port->extcon, EXTCON_DISP_DP))
 362                         return -ENODEV;
 363
 364                 if (cdn_dp_get_sink_count(dp, sink_count)) {
 365                         usleep_range(5000, 10000);
 366                         continue;
 367                 }
 368
 369                 if (!*sink_count)
 370                         return -ENODEV;
 371
 372                 ret = cdn_dp_dpcd_read(dp, DP_DPCD_REV, dp->dpcd,
 373                                        DP_RECEIVER_CAP_SIZE);
 374                 if (ret) {
 375                         DRM_DEV_ERROR(dp->dev, "Failed to get caps %d\n", ret);
 376                         return ret;
 377                 }
 378
 379                 kfree(dp->edid);
 380                 dp->edid = drm_do_get_edid(&dp->connector,
 381                                            cdn_dp_get_edid_block, dp);
 382                 return 0;
 383         }
 384
 385         DRM_DEV_ERROR(dp->dev, "Get sink capability timed out\n");
 386         return -ETIMEDOUT;
 387 }
 388
 389 static int cdn_dp_enable_phy(struct cdn_dp_device *dp, struct cdn_dp_port *port)
 390 {
 391         union extcon_property_value property;
 392         int ret;
 393
 394         if (!port->phy_enabled) {
 395                 ret = phy_power_on(port->phy);
 396                 if (ret) {
 397                         DRM_DEV_ERROR(dp->dev, "phy power on failed: %d\n",
 398                                       ret);
 399                         goto err_phy;
 400                 }
 401                 port->phy_enabled = true;
 402         }
 403
 404         ret = cdn_dp_grf_write(dp, GRF_SOC_CON26,
 405                                DPTX_HPD_SEL_MASK | DPTX_HPD_SEL);
 406         if (ret) {
 407                 DRM_DEV_ERROR(dp->dev, "Failed to write HPD_SEL %d\n", ret);
 408                 goto err_power_on;
 409         }
 410
 411         ret = cdn_dp_get_hpd_status(dp);
 412         if (ret <= 0) {
 413                 if (!ret)
 414                         DRM_DEV_ERROR(dp->dev, "hpd does not exist\n");
 415                 goto err_power_on;
 416         }
 417
 418         ret = extcon_get_property(port->extcon, EXTCON_DISP_DP,
 419                                   EXTCON_PROP_USB_TYPEC_POLARITY, &property);
 420         if (ret) {
 421                 DRM_DEV_ERROR(dp->dev, "get property failed\n");
 422                 goto err_power_on;
 423         }
 424
 425         port->lanes = cdn_dp_get_port_lanes(port);
 426         ret = cdn_dp_set_host_cap(dp, port->lanes, property.intval);
 427         if (ret) {
 428                 DRM_DEV_ERROR(dp->dev, "set host capabilities failed: %d\n",
 429                               ret);
 430                 goto err_power_on;
 431         }
 432
 433         return 0;
 434
 435 err_power_on:
 436         if (phy_power_off(port->phy))
 437                 DRM_DEV_ERROR(dp->dev, "phy power off failed: %d", ret);
 438         else
 439                 port->phy_enabled = false;
 440
 441 err_phy:
 442         cdn_dp_grf_write(dp, GRF_SOC_CON26,
 443                          DPTX_HPD_SEL_MASK | DPTX_HPD_DEL);
 444         return ret;
 445 }
 446
 447 static int cdn_dp_disable_phy(struct cdn_dp_device *dp,
 448                               struct cdn_dp_port *port)
 449 {
 450         int ret;
 451
 452         if (port->phy_enabled) {
 453                 ret = phy_power_off(port->phy);
 454                 if (ret) {
 455                         DRM_DEV_ERROR(dp->dev, "phy power off failed: %d", ret);
 456                         return ret;
 457                 }
 458         }
 459
 460         port->phy_enabled = false;
 461         port->lanes = 0;
 462         return 0;
 463 }
 464
 465 static int cdn_dp_disable(struct cdn_dp_device *dp)
 466 {
 467         int ret, i;
 468
 469         if (!dp->active)
 470                 return 0;
 471
 472         for (i = 0; i < dp->ports; i++)
 473                 cdn_dp_disable_phy(dp, dp->port[i]);
 474
 475         ret = cdn_dp_grf_write(dp, GRF_SOC_CON26,
 476                                DPTX_HPD_SEL_MASK | DPTX_HPD_DEL);
 477         if (ret) {
 478                 DRM_DEV_ERROR(dp->dev, "Failed to clear hpd sel %d\n",
 479                               ret);
 480                 return ret;
 481         }
 482
 483         cdn_dp_set_firmware_active(dp, false);
 484         cdn_dp_clk_disable(dp);
 485         dp->active = false;
 486         dp->link.rate = 0;
 487         dp->link.num_lanes = 0;
 488         if (!dp->connected) {
 489                 kfree(dp->edid);
 490                 dp->edid = NULL;
 491         }
 492
 493         return 0;
 494 }
 495
 496 static int cdn_dp_enable(struct cdn_dp_device *dp)
 497 {
 498         int ret, i, lanes;
 499         struct cdn_dp_port *port;
 500         u8 sink_count;
 501
 502         port = cdn_dp_connected_port(dp);
 503         if (!port) {
 504                 DRM_DEV_ERROR(dp->dev,
 505                               "Can't enable without connection\n");
 506                 return -ENODEV;
 507         }
 508
 509         if (dp->active)
 510                 return 0;
 511
 512         ret = cdn_dp_clk_enable(dp);
 513         if (ret)
 514                 return ret;
 515
 516         ret = cdn_dp_firmware_init(dp);
 517         if (ret) {
 518                 DRM_DEV_ERROR(dp->dev, "firmware init failed: %d", ret);
 519                 goto err_clk_disable;
 520         }
 521
 522         /* only enable the port that connected with downstream device */
 523         for (i = port->id; i < dp->ports; i++) {
 524                 port = dp->port[i];
 525                 lanes = cdn_dp_get_port_lanes(port);
 526                 if (lanes) {
 527                         ret = cdn_dp_enable_phy(dp, port);
 528                         if (ret)
 529                                 continue;
 530
 531                         ret = cdn_dp_get_sink_capability(dp, port, &sink_count);
 532                         if (ret || (!ret && !sink_count)) {
 533                                 cdn_dp_disable_phy(dp, port);
 534                         } else {
 535                                 dp->active = true;
 536                                 dp->lanes = port->lanes;
 537                                 return 0;
 538                         }
 539                 }
 540         }
 541
 542 err_clk_disable:
 543         cdn_dp_clk_disable(dp);
 544         return ret;
 545 }
 546
 547 static void cdn_dp_encoder_mode_set(struct drm_encoder *encoder,
 548                                     struct drm_display_mode *mode,
 549                                     struct drm_display_mode *adjusted)
 550 {
 551         struct cdn_dp_device *dp = encoder_to_dp(encoder);
 552         struct drm_display_info *display_info = &dp->connector.display_info;
 553         struct rockchip_crtc_state *state;
 554         struct video_info *video = &dp->video_info;
 555         int ret, val;
 556
 557         switch (display_info->bpc) {
 558         case 10:
 559                 video->color_depth = 10;
 560                 break;
 561         case 6:
 562                 video->color_depth = 6;
 563                 break;
 564         default:
 565                 video->color_depth = 8;
 566                 break;
 567         }
 568
 569         video->color_fmt = PXL_RGB;
 570
 571         video->v_sync_polarity = !!(mode->flags & DRM_MODE_FLAG_NVSYNC);
 572         video->h_sync_polarity = !!(mode->flags & DRM_MODE_FLAG_NHSYNC);
 573
 574         ret = drm_of_encoder_active_endpoint_id(dp->dev->of_node, encoder);
 575         if (ret < 0) {
 576                 DRM_DEV_ERROR(dp->dev, "Could not get vop id, %d", ret);
 577                 return;
 578         }
 579
 580         DRM_DEV_DEBUG_KMS(dp->dev, "vop %s output to cdn-dp\n",
 581                           (ret) ? "LIT" : "BIG");
 582         state = to_rockchip_crtc_state(encoder->crtc->state);
 583         if (ret) {
 584                 val = DP_SEL_VOP_LIT | (DP_SEL_VOP_LIT << 16);
 585                 state->output_mode = ROCKCHIP_OUT_MODE_P888;
 586         } else {
 587                 val = DP_SEL_VOP_LIT << 16;
 588                 state->output_mode = ROCKCHIP_OUT_MODE_AAAA;
 589         }
 590
 591         ret = cdn_dp_grf_write(dp, GRF_SOC_CON9, val);
 592         if (ret)
 593                 return;
 594
 595         memcpy(&dp->mode, adjusted, sizeof(*mode));
 596 }
 597
 598 static bool cdn_dp_check_link_status(struct cdn_dp_device *dp)
 599 {
 600         u8 link_status[DP_LINK_STATUS_SIZE];
 601         struct cdn_dp_port *port = cdn_dp_connected_port(dp);
 602         u8 sink_lanes = drm_dp_max_lane_count(dp->dpcd);
 603
 604         if (!port || !dp->link.rate || !dp->link.num_lanes)
 605                 return false;
 606
 607         if (cdn_dp_dpcd_read(dp, DP_LANE0_1_STATUS, link_status,
 608                              DP_LINK_STATUS_SIZE)) {
 609                 DRM_ERROR("Failed to get link status\n");
 610                 return false;
 611         }
 612
 613         /* if link training is requested we should perform it always */
 614         return drm_dp_channel_eq_ok(link_status, min(port->lanes, sink_lanes));
 615 }
 616
 617 static void cdn_dp_encoder_enable(struct drm_encoder *encoder)
 618 {
 619         struct cdn_dp_device *dp = encoder_to_dp(encoder);
 620         int ret;
 621
 622         mutex_lock(&dp->lock);
 623         ret = cdn_dp_enable(dp);
 624         if (ret) {
 625                 DRM_DEV_ERROR(dp->dev, "Failed to enable encoder %d\n",
 626                               ret);
 627                 goto out;
 628         }
 629         if (!cdn_dp_check_link_status(dp)) {
 630                 ret = cdn_dp_train_link(dp);
 631                 if (ret) {
 632                         DRM_DEV_ERROR(dp->dev, "Failed link train %d\n", ret);
 633                         goto out;
 634                 }
 635         }
 636
 637         ret = cdn_dp_set_video_status(dp, CONTROL_VIDEO_IDLE);
 638         if (ret) {
 639                 DRM_DEV_ERROR(dp->dev, "Failed to idle video %d\n", ret);
 640                 goto out;
 641         }
 642
 643         ret = cdn_dp_config_video(dp);
 644         if (ret) {
 645                 DRM_DEV_ERROR(dp->dev, "Failed to config video %d\n", ret);
 646                 goto out;
 647         }
 648
 649         ret = cdn_dp_set_video_status(dp, CONTROL_VIDEO_VALID);
 650         if (ret) {
 651                 DRM_DEV_ERROR(dp->dev, "Failed to valid video %d\n", ret);
 652                 goto out;
 653         }
 654 out:
 655         mutex_unlock(&dp->lock);
 656 }
 657
 658 static void cdn_dp_encoder_disable(struct drm_encoder *encoder)
 659 {
 660         struct cdn_dp_device *dp = encoder_to_dp(encoder);
 661         int ret;
 662
 663         mutex_lock(&dp->lock);
 664         if (dp->active) {
 665                 ret = cdn_dp_disable(dp);
 666                 if (ret) {
 667                         DRM_DEV_ERROR(dp->dev, "Failed to disable encoder %d\n",
 668                                       ret);
 669                 }
 670         }
 671         mutex_unlock(&dp->lock);
 672
 673         /*
 674          * In the following 2 cases, we need to run the event_work to re-enable
 675          * the DP:
 676          * 1. If there is not just one port device is connected, and remove one
 677          *    device from a port, the DP will be disabled here, at this case,
 678          *    run the event_work to re-open DP for the other port.
 679          * 2. If re-training or re-config failed, the DP will be disabled here.
 680          *    run the event_work to re-connect it.
 681          */
 682         if (!dp->connected && cdn_dp_connected_port(dp))
 683                 schedule_work(&dp->event_work);
 684 }
 685
 686 static int cdn_dp_encoder_atomic_check(struct drm_encoder *encoder,
 687                                        struct drm_crtc_state *crtc_state,
 688                                        struct drm_connector_state *conn_state)
 689 {
 690         struct rockchip_crtc_state *s = to_rockchip_crtc_state(crtc_state);
 691
 692         s->output_mode = ROCKCHIP_OUT_MODE_AAAA;
 693         s->output_type = DRM_MODE_CONNECTOR_DisplayPort;
 694
 695         return 0;
 696 }
 697
 698 static const struct drm_encoder_helper_funcs cdn_dp_encoder_helper_funcs = {
 699         .mode_set = cdn_dp_encoder_mode_set,
 700         .enable = cdn_dp_encoder_enable,
 701         .disable = cdn_dp_encoder_disable,
 702         .atomic_check = cdn_dp_encoder_atomic_check,
 703 };
 704
 705 static const struct drm_encoder_funcs cdn_dp_encoder_funcs = {
 706         .destroy = drm_encoder_cleanup,
 707 };
 708
 709 static int cdn_dp_parse_dt(struct cdn_dp_device *dp)
 710 {
 711         struct device *dev = dp->dev;
 712         struct device_node *np = dev->of_node;
 713         struct platform_device *pdev = to_platform_device(dev);
 714         struct resource *res;
 715
 716         dp->grf = syscon_regmap_lookup_by_phandle(np, "rockchip,grf");
 717         if (IS_ERR(dp->grf)) {
 718                 DRM_DEV_ERROR(dev, "cdn-dp needs rockchip,grf property\n");
 719                 return PTR_ERR(dp->grf);
 720         }
 721
 722         res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
 723         dp->regs = devm_ioremap_resource(dev, res);
 724         if (IS_ERR(dp->regs)) {
 725                 DRM_DEV_ERROR(dev, "ioremap reg failed\n");
 726                 return PTR_ERR(dp->regs);
 727         }
 728
 729         dp->core_clk = devm_clk_get(dev, "core-clk");
 730         if (IS_ERR(dp->core_clk)) {
 731                 DRM_DEV_ERROR(dev, "cannot get core_clk_dp\n");
 732                 return PTR_ERR(dp->core_clk);
 733         }
 734
 735         dp->pclk = devm_clk_get(dev, "pclk");
 736         if (IS_ERR(dp->pclk)) {
 737                 DRM_DEV_ERROR(dev, "cannot get pclk\n");
 738                 return PTR_ERR(dp->pclk);
 739         }
 740
 741         dp->spdif_clk = devm_clk_get(dev, "spdif");
 742         if (IS_ERR(dp->spdif_clk)) {
 743                 DRM_DEV_ERROR(dev, "cannot get spdif_clk\n");
 744                 return PTR_ERR(dp->spdif_clk);
 745         }
 746
 747         dp->grf_clk = devm_clk_get(dev, "grf");
 748         if (IS_ERR(dp->grf_clk)) {
 749                 DRM_DEV_ERROR(dev, "cannot get grf clk\n");
 750                 return PTR_ERR(dp->grf_clk);
 751         }
 752
 753         dp->spdif_rst = devm_reset_control_get(dev, "spdif");
 754         if (IS_ERR(dp->spdif_rst)) {
 755                 DRM_DEV_ERROR(dev, "no spdif reset control found\n");
 756                 return PTR_ERR(dp->spdif_rst);
 757         }
 758
 759         dp->dptx_rst = devm_reset_control_get(dev, "dptx");
 760         if (IS_ERR(dp->dptx_rst)) {
 761                 DRM_DEV_ERROR(dev, "no uphy reset control found\n");
 762                 return PTR_ERR(dp->dptx_rst);
 763         }
 764
 765         dp->core_rst = devm_reset_control_get(dev, "core");
 766         if (IS_ERR(dp->core_rst)) {
 767                 DRM_DEV_ERROR(dev, "no core reset control found\n");
 768                 return PTR_ERR(dp->core_rst);
 769         }
 770
 771         dp->apb_rst = devm_reset_control_get(dev, "apb");
 772         if (IS_ERR(dp->apb_rst)) {
 773                 DRM_DEV_ERROR(dev, "no apb reset control found\n");
 774                 return PTR_ERR(dp->apb_rst);
 775         }
 776
 777         return 0;
 778 }
 779
 780 static int cdn_dp_audio_hw_params(struct device *dev,  void *data,
 781                                   struct hdmi_codec_daifmt *daifmt,
 782                                   struct hdmi_codec_params *params)
 783 {
 784         struct cdn_dp_device *dp = dev_get_drvdata(dev);
 785         struct audio_info audio = {
 786                 .sample_width = params->sample_width,
 787                 .sample_rate = params->sample_rate,
 788                 .channels = params->channels,
 789         };
 790         int ret;
 791
 792         mutex_lock(&dp->lock);
 793         if (!dp->active) {
 794                 ret = -ENODEV;
 795                 goto out;
 796         }
 797
 798         switch (daifmt->fmt) {
 799         case HDMI_I2S:
 800                 audio.format = AFMT_I2S;
 801                 break;
 802         case HDMI_SPDIF:
 803                 audio.format = AFMT_SPDIF;
 804                 break;
 805         default:
 806                 DRM_DEV_ERROR(dev, "Invalid format %d\n", daifmt->fmt);
 807                 ret = -EINVAL;
 808                 goto out;
 809         }
 810
 811         ret = cdn_dp_audio_config(dp, &audio);
 812         if (!ret)
 813                 dp->audio_info = audio;
 814
 815 out:
 816         mutex_unlock(&dp->lock);
 817         return ret;
 818 }
 819
 820 static void cdn_dp_audio_shutdown(struct device *dev, void *data)
 821 {
 822         struct cdn_dp_device *dp = dev_get_drvdata(dev);
 823         int ret;
 824
 825         mutex_lock(&dp->lock);
 826         if (!dp->active)
 827                 goto out;
 828
 829         ret = cdn_dp_audio_stop(dp, &dp->audio_info);
 830         if (!ret)
 831                 dp->audio_info.format = AFMT_UNUSED;
 832 out:
 833         mutex_unlock(&dp->lock);
 834 }
 835
 836 static int cdn_dp_audio_digital_mute(struct device *dev, void *data,
 837                                      bool enable)
 838 {
 839         struct cdn_dp_device *dp = dev_get_drvdata(dev);
 840         int ret;
 841
 842         mutex_lock(&dp->lock);
 843         if (!dp->active) {
 844                 ret = -ENODEV;
 845                 goto out;
 846         }
 847
 848         ret = cdn_dp_audio_mute(dp, enable);
 849
 850 out:
 851         mutex_unlock(&dp->lock);
 852         return ret;
 853 }
 854
 855 static int cdn_dp_audio_get_eld(struct device *dev, void *data,
 856                                 u8 *buf, size_t len)
 857 {
 858         struct cdn_dp_device *dp = dev_get_drvdata(dev);
 859
 860         memcpy(buf, dp->connector.eld, min(sizeof(dp->connector.eld), len));
 861
 862         return 0;
 863 }
 864
 865 static const struct hdmi_codec_ops audio_codec_ops = {
 866         .hw_params = cdn_dp_audio_hw_params,
 867         .audio_shutdown = cdn_dp_audio_shutdown,
 868         .digital_mute = cdn_dp_audio_digital_mute,
 869         .get_eld = cdn_dp_audio_get_eld,
 870 };
 871
 872 static int cdn_dp_audio_codec_init(struct cdn_dp_device *dp,
 873                                    struct device *dev)
 874 {
 875         struct hdmi_codec_pdata codec_data = {
 876                 .i2s = 1,
 877                 .spdif = 1,
 878                 .ops = &audio_codec_ops,
 879                 .max_i2s_channels = 8,
 880         };
 881
 882         dp->audio_pdev = platform_device_register_data(
 883                          dev, HDMI_CODEC_DRV_NAME, PLATFORM_DEVID_AUTO,
 884                          &codec_data, sizeof(codec_data));
 885
 886         return PTR_ERR_OR_ZERO(dp->audio_pdev);
 887 }
 888
 889 static int cdn_dp_request_firmware(struct cdn_dp_device *dp)
 890 {
 891         int ret;
 892         unsigned long timeout = jiffies + msecs_to_jiffies(CDN_FW_TIMEOUT_MS);
 893         unsigned long sleep = 1000;
 894
 895         WARN_ON(!mutex_is_locked(&dp->lock));
 896
 897         if (dp->fw_loaded)
 898                 return 0;
 899
 900         /* Drop the lock before getting the firmware to avoid blocking boot */
 901         mutex_unlock(&dp->lock);
 902
 903         while (time_before(jiffies, timeout)) {
 904                 ret = request_firmware(&dp->fw, CDN_DP_FIRMWARE, dp->dev);
 905                 if (ret == -ENOENT) {
 906                         msleep(sleep);
 907                         sleep *= 2;
 908                         continue;
 909                 } else if (ret) {
 910                         DRM_DEV_ERROR(dp->dev,
 911                                       "failed to request firmware: %d\n", ret);
 912                         goto out;
 913                 }
 914
 915                 dp->fw_loaded = true;
 916                 ret = 0;
 917                 goto out;
 918         }
 919
 920         DRM_DEV_ERROR(dp->dev, "Timed out trying to load firmware\n");
 921         ret = -ETIMEDOUT;
 922 out:
 923         mutex_lock(&dp->lock);
 924         return ret;
 925 }
 926
 927 static void cdn_dp_pd_event_work(struct work_struct *work)
 928 {
 929         struct cdn_dp_device *dp = container_of(work, struct cdn_dp_device,
 930                                                 event_work);
 931         int ret;
 932         u8 sink_count;
 933
 934         mutex_lock(&dp->lock);
 935         ret = cdn_dp_request_firmware(dp);
 936         if (ret)
 937                 goto out;
 938
 939         dp->connected = true;
 940
 941         /* Not connected, notify userspace to disable the block */
 942         if (!cdn_dp_connected_port(dp)) {
 943                 DRM_DEV_INFO(dp->dev, "Not connected. Disabling cdn\n");
 944                 dp->connected = false;
 945
 946         /* Connected but not enabled, enable the block */
 947         } else if (!dp->active) {
 948                 DRM_DEV_INFO(dp->dev, "Connected, not enabled. Enabling cdn\n");
 949                 ret = cdn_dp_enable(dp);
 950                 if (ret) {
 951                         DRM_DEV_ERROR(dp->dev, "Enable dp failed %d\n", ret);
 952                         dp->connected = false;
 953                 }
 954
 955         /* Enabled and connected to a dongle without a sink, notify userspace */
 956         } else if (cdn_dp_get_sink_count(dp, &sink_count) || !sink_count) {
 957                 DRM_DEV_INFO(dp->dev, "Connected without sink. Assert hpd\n");
 958                 dp->connected = false;
 959
 960         /* Enabled and connected with a sink, re-train if requested */
 961         } else if (!cdn_dp_check_link_status(dp)) {
 962                 unsigned int rate = dp->link.rate;
 963                 unsigned int lanes = dp->link.num_lanes;
 964                 struct drm_display_mode *mode = &dp->mode;
 965
 966                 DRM_DEV_INFO(dp->dev, "Connected with sink. Re-train link\n");
 967                 ret = cdn_dp_train_link(dp);
 968                 if (ret) {
 969                         dp->connected = false;
 970                         DRM_DEV_ERROR(dp->dev, "Train link failed %d\n", ret);
 971                         goto out;
 972                 }
 973
 974                 /* If training result is changed, update the video config */
 975                 if (mode->clock &&
 976                     (rate != dp->link.rate || lanes != dp->link.num_lanes)) {
 977                         ret = cdn_dp_config_video(dp);
 978                         if (ret) {
 979                                 dp->connected = false;
 980                                 DRM_DEV_ERROR(dp->dev,
 981                                               "Failed to config video %d\n",
 982                                               ret);
 983                         }
 984                 }
 985         }
 986
 987 out:
 988         mutex_unlock(&dp->lock);
 989         drm_helper_hpd_irq_event(dp->drm_dev);
 990 }
 991
 992 static int cdn_dp_pd_event(struct notifier_block *nb,
 993                            unsigned long event, void *priv)
 994 {
 995         struct cdn_dp_port *port = container_of(nb, struct cdn_dp_port,
 996                                                 event_nb);
 997         struct cdn_dp_device *dp = port->dp;
 998
 999         /*
1000          * It would be nice to be able to just do the work inline right here.
1001          * However, we need to make a bunch of calls that might sleep in order
1002          * to turn on the block/phy, so use a worker instead.
1003          */
1004         schedule_work(&dp->event_work);
1005
1006         return NOTIFY_DONE;
1007 }
1008
1009 static int cdn_dp_bind(struct device *dev, struct device *master, void *data)
1010 {
1011         struct cdn_dp_device *dp = dev_get_drvdata(dev);
1012         struct drm_encoder *encoder;
1013         struct drm_connector *connector;
1014         struct cdn_dp_port *port;
1015         struct drm_device *drm_dev = data;
1016         int ret, i;
1017
1018         ret = cdn_dp_parse_dt(dp);
1019         if (ret < 0)
1020                 return ret;
1021
1022         dp->drm_dev = drm_dev;
1023         dp->connected = false;
1024         dp->active = false;
1025
1026         mutex_init(&dp->lock);
1027         INIT_WORK(&dp->event_work, cdn_dp_pd_event_work);
1028
1029         encoder = &dp->encoder;
1030
1031         encoder->possible_crtcs = drm_of_find_possible_crtcs(drm_dev,
1032                                                              dev->of_node);
1033         DRM_DEBUG_KMS("possible_crtcs = 0x%x\n", encoder->possible_crtcs);
1034
1035         ret = drm_encoder_init(drm_dev, encoder, &cdn_dp_encoder_funcs,
1036                                DRM_MODE_ENCODER_TMDS, NULL);
1037         if (ret) {
1038                 DRM_ERROR("failed to initialize encoder with drm\n");
1039                 return ret;
1040         }
1041
1042         drm_encoder_helper_add(encoder, &cdn_dp_encoder_helper_funcs);
1043
1044         connector = &dp->connector;
1045         connector->polled = DRM_CONNECTOR_POLL_HPD;
1046         connector->dpms = DRM_MODE_DPMS_OFF;
1047
1048         ret = drm_connector_init(drm_dev, connector,
1049                                  &cdn_dp_atomic_connector_funcs,
1050                                  DRM_MODE_CONNECTOR_DisplayPort);
1051         if (ret) {
1052                 DRM_ERROR("failed to initialize connector with drm\n");
1053                 goto err_free_encoder;
1054         }
1055
1056         drm_connector_helper_add(connector, &cdn_dp_connector_helper_funcs);
1057
1058         ret = drm_mode_connector_attach_encoder(connector, encoder);
1059         if (ret) {
1060                 DRM_ERROR("failed to attach connector and encoder\n");
1061                 goto err_free_connector;
1062         }
1063
1064         cdn_dp_audio_codec_init(dp, dev);
1065
1066         for (i = 0; i < dp->ports; i++) {
1067                 port = dp->port[i];
1068
1069                 port->event_nb.notifier_call = cdn_dp_pd_event;
1070                 ret = devm_extcon_register_notifier(dp->dev, port->extcon,
1071                                                     EXTCON_DISP_DP,
1072                                                     &port->event_nb);
1073                 if (ret) {
1074                         DRM_DEV_ERROR(dev,
1075                                       "register EXTCON_DISP_DP notifier err\n");
1076                         goto err_free_connector;
1077                 }
1078         }
1079
1080         pm_runtime_enable(dev);
1081
1082         schedule_work(&dp->event_work);
1083
1084         return 0;
1085
1086 err_free_connector:
1087         drm_connector_cleanup(connector);
1088 err_free_encoder:
1089         drm_encoder_cleanup(encoder);
1090         return ret;
1091 }
1092
1093 static void cdn_dp_unbind(struct device *dev, struct device *master, void *data)
1094 {
1095         struct cdn_dp_device *dp = dev_get_drvdata(dev);
1096         struct drm_encoder *encoder = &dp->encoder;
1097         struct drm_connector *connector = &dp->connector;
1098
1099         cancel_work_sync(&dp->event_work);
1100         platform_device_unregister(dp->audio_pdev);
1101         cdn_dp_encoder_disable(encoder);
1102         encoder->funcs->destroy(encoder);
1103         connector->funcs->destroy(connector);
1104
1105         pm_runtime_disable(dev);
1106         release_firmware(dp->fw);
1107         kfree(dp->edid);
1108         dp->edid = NULL;
1109 }
1110
1111 static const struct component_ops cdn_dp_component_ops = {
1112         .bind = cdn_dp_bind,
1113         .unbind = cdn_dp_unbind,
1114 };
1115
1116 int cdn_dp_suspend(struct device *dev)
1117 {
1118         struct cdn_dp_device *dp = dev_get_drvdata(dev);
1119
1120         if (dp->active)
1121                 return cdn_dp_disable(dp);
1122
1123         return 0;
1124 }
1125
1126 int cdn_dp_resume(struct device *dev)
1127 {
1128         struct cdn_dp_device *dp = dev_get_drvdata(dev);
1129
1130         if (dp->fw_loaded)
1131                 schedule_work(&dp->event_work);
1132
1133         return 0;
1134 }
1135
1136 static int cdn_dp_probe(struct platform_device *pdev)
1137 {
1138         struct device *dev = &pdev->dev;
1139         const struct of_device_id *match;
1140         struct cdn_dp_data *dp_data;
1141         struct cdn_dp_port *port;
1142         struct cdn_dp_device *dp;
1143         struct extcon_dev *extcon;
1144         struct phy *phy;
1145         int i;
1146
1147         dp = devm_kzalloc(dev, sizeof(*dp), GFP_KERNEL);
1148         if (!dp)
1149                 return -ENOMEM;
1150         dp->dev = dev;
1151
1152         match = of_match_node(cdn_dp_dt_ids, pdev->dev.of_node);
1153         dp_data = (struct cdn_dp_data *)match->data;
1154
1155         for (i = 0; i < dp_data->max_phy; i++) {
1156                 extcon = extcon_get_edev_by_phandle(dev, i);
1157                 phy = devm_of_phy_get_by_index(dev, dev->of_node, i);
1158
1159                 if (PTR_ERR(extcon) == -EPROBE_DEFER ||
1160                     PTR_ERR(phy) == -EPROBE_DEFER)
1161                         return -EPROBE_DEFER;
1162
1163                 if (IS_ERR(extcon) || IS_ERR(phy))
1164                         continue;
1165
1166                 port = devm_kzalloc(dev, sizeof(*port), GFP_KERNEL);
1167                 if (!dp)
1168                         return -ENOMEM;
1169
1170                 port->extcon = extcon;
1171                 port->phy = phy;
1172                 port->dp = dp;
1173                 port->id = i;
1174                 dp->port[dp->ports++] = port;
1175         }
1176
1177         if (!dp->ports) {
1178                 DRM_DEV_ERROR(dev, "missing extcon or phy\n");
1179                 return -EINVAL;
1180         }
1181
1182         dev_set_drvdata(dev, dp);
1183
1184         return component_add(dev, &cdn_dp_component_ops);
1185 }
1186
1187 static int cdn_dp_remove(struct platform_device *pdev)
1188 {
1189         struct cdn_dp_device *dp = platform_get_drvdata(pdev);
1190
1191         cdn_dp_suspend(dp->dev);
1192         component_del(&pdev->dev, &cdn_dp_component_ops);
1193
1194         return 0;
1195 }
1196
1197 static void cdn_dp_shutdown(struct platform_device *pdev)
1198 {
1199         struct cdn_dp_device *dp = platform_get_drvdata(pdev);
1200
1201         cdn_dp_suspend(dp->dev);
1202 }
1203
1204 static const struct dev_pm_ops cdn_dp_pm_ops = {
1205         SET_SYSTEM_SLEEP_PM_OPS(cdn_dp_suspend,
1206                                 cdn_dp_resume)
1207 };
1208
1209 static struct platform_driver cdn_dp_driver = {
1210         .probe = cdn_dp_probe,
1211         .remove = cdn_dp_remove,
1212         .shutdown = cdn_dp_shutdown,
1213         .driver = {
1214                    .name = "cdn-dp",
1215                    .owner = THIS_MODULE,
1216                    .of_match_table = of_match_ptr(cdn_dp_dt_ids),
1217                    .pm = &cdn_dp_pm_ops,
1218         },
1219 };
1220
1221 module_platform_driver(cdn_dp_driver);
1222
1223 MODULE_AUTHOR("Chris Zhong <zyw@rock-chips.com>");
1224 MODULE_DESCRIPTION("cdn DP Driver");
1225 MODULE_LICENSE("GPL v2");