src/mm_player_capture.c

   1 /*
   2  * libmm-player
   3  *
   4  * Copyright(c) 2000 - 2011 Samsung Electronics Co., Ltd. All rights reserved.
   5  *
   6  * Contact: JongHyuk Choi <jhchoi.choi@samsung.com>, YeJin Cho <cho.yejin@samsung.com>,
   7  * Seungbae Shin <seungbae.shin@samsung.com>, YoungHwan An <younghwan_.an@samsung.com>
   8  *
   9  * Licensed under the Apache License, Version 2.0(the "License");
  10  * you may not use this file except in compliance with the License.
  11  * You may obtain a copy of the License at
  12  *
  13  * http://www.apache.org/licenses/LICENSE-2.0
  14  *
  15  * Unless required by applicable law or agreed to in writing, software
  16  * distributed under the License is distributed on an "AS IS" BASIS,
  17  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  18  * See the License for the specific language governing permissions and
  19  * limitations under the License.
  20  *
  21  */
  22
  23 /*===========================================================================================
  24 |                                                                                                                                                                                       |
  25 |  INCLUDE FILES                                                                                                                                                        |
  26 |                                                                                                                                                                                       |
  27 ========================================================================================== */
  28 #include <dlog.h>
  29 #include "mm_player_utils.h"
  30 #include "mm_player_capture.h"
  31 #include "mm_player_priv.h"
  32
  33 #include <mm_util_imgp.h>
  34 #include <gst/video/video-info.h>
  35
  36 //#define       CAPTURE_OUTPUT_DUMP     1
  37 /*---------------------------------------------------------------------------
  38 |    LOCAL VARIABLE DEFINITIONS for internal                                                            |
  39 ---------------------------------------------------------------------------*/
  40
  41 /*---------------------------------------------------------------------------
  42 |    LOCAL FUNCTION PROTOTYPES:                                                                                         |
  43 ---------------------------------------------------------------------------*/
  44 static GstPadProbeReturn __mmplayer_video_capture_probe(GstPad *pad, GstPadProbeInfo *info, gpointer u_data);
  45 static int  __mmplayer_get_video_frame_from_buffer(mm_player_t* player, GstPad *pad, GstBuffer *buffer);
  46 static gpointer __mmplayer_capture_thread(gpointer data);
  47 static void __csc_tiled_to_linear_crop(unsigned char *yuv420_dest, unsigned char *nv12t_src, int yuv420_width, int yuv420_height, int left, int top, int right, int buttom);
  48 static int __tile_4x2_read(int x_size, int y_size, int x_pos, int y_pos);
  49 static int __mm_player_convert_colorspace(mm_player_t* player, unsigned char* src_data, mm_util_img_format src_fmt, unsigned int src_w, unsigned int src_h, mm_util_img_format dst_fmt);
  50 #ifdef CAPTURE_OUTPUT_DUMP
  51 static void capture_output_dump(mm_player_t* player);
  52 #endif
  53
  54 /*===========================================================================================
  55 |                                                                                                                                                                                       |
  56 |  FUNCTION DEFINITIONS                                                                                                                                         |
  57 |                                                                                                                                                                                       |
  58 ========================================================================================== */
  59 int
  60 _mmplayer_initialize_video_capture(mm_player_t* player)
  61 {
  62         int ret = MM_ERROR_NONE;
  63         MMPLAYER_RETURN_VAL_IF_FAIL(player, MM_ERROR_PLAYER_NOT_INITIALIZED);
  64         /* create capture mutex */
  65         g_mutex_init(&player->capture_thread_mutex);
  66
  67         /* create capture cond */
  68         g_cond_init(&player->capture_thread_cond);
  69
  70
  71         player->capture_thread_exit = FALSE;
  72
  73         /* create video capture thread */
  74         player->capture_thread =
  75                         g_thread_try_new("capture_thread", __mmplayer_capture_thread, (gpointer)player, NULL);
  76
  77         if (!player->capture_thread) {
  78                 ret = MM_ERROR_PLAYER_RESOURCE_LIMIT;
  79                 goto ERROR;
  80         }
  81
  82         return ret;
  83
  84 ERROR:
  85         /* capture thread */
  86         g_mutex_clear(&player->capture_thread_mutex);
  87
  88         g_cond_clear(&player->capture_thread_cond);
  89
  90         return ret;
  91 }
  92
  93 int
  94 _mmplayer_release_video_capture(mm_player_t* player)
  95 {
  96         MMPLAYER_RETURN_VAL_IF_FAIL(player, MM_ERROR_PLAYER_NOT_INITIALIZED);
  97         /* release capture thread */
  98         MMPLAYER_CAPTURE_THREAD_LOCK(player);
  99         player->capture_thread_exit = TRUE;
 100         MMPLAYER_CAPTURE_THREAD_SIGNAL(player);
 101         MMPLAYER_CAPTURE_THREAD_UNLOCK(player);
 102
 103         LOGD("waitting for capture thread exit");
 104         g_thread_join(player->capture_thread);
 105         g_mutex_clear(&player->capture_thread_mutex);
 106         g_cond_clear(&player->capture_thread_cond);
 107         LOGD("capture thread released");
 108
 109         return MM_ERROR_NONE;
 110 }
 111
 112 int
 113 _mmplayer_do_video_capture(MMHandleType hplayer)
 114 {
 115         mm_player_t* player = (mm_player_t*) hplayer;
 116         int ret = MM_ERROR_NONE;
 117         GstPad *pad = NULL;
 118
 119         MMPLAYER_FENTER();
 120
 121         MMPLAYER_RETURN_VAL_IF_FAIL(player && player->pipeline, MM_ERROR_PLAYER_NOT_INITIALIZED);
 122
 123         /* capturing or not */
 124         if (player->video_capture_cb_probe_id || player->capture.data
 125                         || player->captured.data[0] || player->captured.data[1]) {
 126                 LOGW("capturing... we can't do any more");
 127                 return MM_ERROR_PLAYER_INVALID_STATE;
 128         }
 129         gint surface_type = 0;
 130         mm_attrs_get_int_by_name(player->attrs, "display_surface_type", &surface_type);
 131
 132         /* check if video pipeline is linked or not */
 133         if (!player->pipeline->videobin) {
 134                 if (surface_type == MM_DISPLAY_SURFACE_REMOTE && player->video_fakesink) {
 135                         LOGD("it is evas surface type.");
 136                 } else {
 137                         LOGW("not ready to capture");
 138                         return MM_ERROR_PLAYER_INVALID_STATE;
 139                 }
 140         }
 141
 142         if (player->pipeline->videobin)
 143                 pad = gst_element_get_static_pad(player->pipeline->videobin[MMPLAYER_V_SINK].gst, "sink");
 144         else // evas surface
 145                 pad = gst_element_get_static_pad(player->video_fakesink, "sink");
 146
 147         if (player->state != MM_PLAYER_STATE_PLAYING) {
 148                 if (player->state == MM_PLAYER_STATE_PAUSED) {
 149                         // get last buffer from video sink
 150                         GstSample *sample = NULL;
 151
 152                         gst_element_get_state(player->pipeline->mainbin[MMPLAYER_M_PIPE].gst, NULL, NULL, 5 * GST_SECOND); //5 seconds
 153
 154                         if (player->pipeline->videobin)
 155                                 g_object_get(player->pipeline->videobin[MMPLAYER_V_SINK].gst, "last-sample", &sample, NULL);
 156                         else // evas surface
 157                                 g_object_get(player->video_fakesink, "last-sample", &sample, NULL);
 158
 159                         if (sample) {
 160                                 GstBuffer *buf = NULL;
 161                                 buf = gst_sample_get_buffer(sample);
 162                                 if (buf) {
 163                                         if (__mmplayer_get_video_frame_from_buffer(player, pad, buf) != MM_ERROR_NONE)
 164                                                 ret = MM_ERROR_PLAYER_INTERNAL;
 165                                 } else {
 166                                         LOGW("failed to get video frame");
 167                                 }
 168                                 gst_sample_unref(sample);
 169                         }
 170                         return ret;
 171                 } else {
 172                         LOGW("invalid state(%d) to capture", player->state);
 173                         gst_object_unref(GST_OBJECT(pad));
 174                         pad = NULL;
 175                         return MM_ERROR_PLAYER_INVALID_STATE;
 176                 }
 177         }
 178
 179         /* register probe */
 180         player->video_capture_cb_probe_id = gst_pad_add_probe(pad, GST_PAD_PROBE_TYPE_BUFFER,
 181                 __mmplayer_video_capture_probe, player, NULL);
 182
 183         gst_object_unref(GST_OBJECT(pad));
 184         pad = NULL;
 185
 186         MMPLAYER_FLEAVE();
 187
 188         return ret;
 189 }
 190
 191 int
 192 __mmplayer_handle_orientation(mm_player_t* player, int orientation, int format)
 193 {
 194         unsigned char *src_buffer = NULL;
 195         int ret = MM_ERROR_NONE;
 196         unsigned char *dst_frame = NULL;
 197         unsigned int dst_width = 0;
 198         unsigned int dst_height = 0;
 199         unsigned int dst_size = 0;
 200         mm_util_img_rotate_type rot_enum = MM_UTIL_ROTATE_NUM;
 201
 202         player->capture.orientation = orientation;
 203
 204         if (orientation == 90 || orientation == 270) {
 205                 dst_width = player->captured.height[0];
 206                 dst_height = player->captured.width[0];
 207                 LOGD("interchange width & height");
 208         } else if (orientation == 180) {
 209                 dst_width = player->captured.width[0];
 210                 dst_height = player->captured.height[0];
 211         } else if (orientation == 0) {
 212                 LOGE("no need handle orientation : %d", orientation);
 213                 player->capture.width = player->captured.width[0];
 214                 player->capture.height = player->captured.height[0];
 215                 return MM_ERROR_NONE;
 216         } else
 217                 LOGE("wrong orientation value...");
 218
 219     /* height & width will be interchanged for 90 and 270 orientation */
 220         ret = mm_util_get_image_size(format, dst_width, dst_height, &dst_size);
 221         if (ret != MM_ERROR_NONE) {
 222                 LOGE("failed to get destination frame size");
 223                 return ret;
 224         }
 225
 226         LOGD("before rotation : dst_width = %d and dst_height = %d", dst_width, dst_height);
 227
 228         dst_frame = (unsigned char*) malloc(dst_size);
 229         if (!dst_frame) {
 230                 LOGE("failed to allocate memory");
 231                 return MM_ERROR_PLAYER_NO_FREE_SPACE;
 232         }
 233
 234         src_buffer = (unsigned char*)player->capture.data;
 235
 236         /* convert orientation degree into enum here */
 237         switch (orientation) {
 238         case 90:
 239                 rot_enum = MM_UTIL_ROTATE_90;
 240                 break;
 241         case 180:
 242                 rot_enum = MM_UTIL_ROTATE_180;
 243                 break;
 244         case 270:
 245                 rot_enum = MM_UTIL_ROTATE_270;
 246                 break;
 247         default:
 248                 LOGE("wrong rotate value");
 249                 break;
 250         }
 251
 252         LOGD("source buffer for rotation = %p and rotation = %d", src_buffer, rot_enum);
 253
 254         ret = mm_util_rotate_image(src_buffer,
 255                         player->captured.width[0], player->captured.height[0], format,
 256                         dst_frame, &dst_width, &dst_height, rot_enum);
 257         if (ret != MM_ERROR_NONE) {
 258                 LOGE("failed to do rotate image");
 259                 free(dst_frame);
 260                 return ret;
 261         }
 262
 263         LOGD("after rotation same stride: dst_width = %d and dst_height = %d", dst_width, dst_height);
 264
 265         g_free(src_buffer);
 266
 267         player->capture.data = dst_frame;
 268         player->capture.size = dst_size;
 269         player->capture.orientation = orientation;
 270         player->capture.width = dst_width;
 271         player->capture.height = dst_height;
 272
 273         player->captured.width[0] = player->captured.stride_width[0] = dst_width;
 274         player->captured.height[0] = player->captured.stride_height[0] = dst_height;
 275
 276         return ret;
 277 }
 278
 279
 280 static gpointer
 281 __mmplayer_capture_thread(gpointer data)
 282 {
 283         mm_player_t* player = (mm_player_t*) data;
 284         MMMessageParamType msg = {0, };
 285         unsigned char * src_buffer = NULL;
 286         unsigned char * linear_y_plane = NULL;
 287         unsigned char * linear_uv_plane = NULL;
 288         int orientation = 0;
 289         int ret = 0;
 290         int planes[MAX_BUFFER_PLANE] = {0, };
 291         unsigned char * p_buf = NULL;
 292         unsigned char * temp = NULL;
 293         int i, j;
 294
 295         MMPLAYER_RETURN_VAL_IF_FAIL(player, NULL);
 296
 297         MMPLAYER_CAPTURE_THREAD_LOCK(player);
 298         while (!player->capture_thread_exit) {
 299                 LOGD("capture thread started. waiting for signal");
 300                 MMPLAYER_CAPTURE_THREAD_WAIT(player);
 301
 302                 if (player->capture_thread_exit) {
 303                         LOGD("exiting capture thread");
 304                         goto EXIT;
 305                 }
 306                 LOGD("capture thread is received signal");
 307
 308                 /* NOTE: Don't use MMPLAYER_CMD_LOCK() here.
 309                  * Because deadlock can be happened if other player api is used in message callback.
 310                  */
 311                 if (player->video_cs == MM_PLAYER_COLORSPACE_NV12_TILED) {
 312                         /* Colorspace conversion : NV12T-> NV12-> RGB888 */
 313                         int ret = 0;
 314                         int linear_y_plane_size;
 315                         int linear_uv_plane_size;
 316                         int width = player->captured.width[0];
 317                         int height = player->captured.height[0];
 318
 319                         linear_y_plane_size = (width * height);
 320                         linear_uv_plane_size = (width * height / 2);
 321
 322                         linear_y_plane = (unsigned char*) g_try_malloc(linear_y_plane_size);
 323                         if (linear_y_plane == NULL) {
 324                                 msg.code = MM_ERROR_PLAYER_NO_FREE_SPACE;
 325                                 goto ERROR;
 326                         }
 327
 328                         linear_uv_plane = (unsigned char*) g_try_malloc(linear_uv_plane_size);
 329                         if (linear_uv_plane == NULL) {
 330                                 msg.code = MM_ERROR_PLAYER_NO_FREE_SPACE;
 331                                 goto ERROR;
 332                         }
 333                         /* NV12 tiled to linear */
 334                         __csc_tiled_to_linear_crop(linear_y_plane,
 335                                         player->captured.data[0], width, height, 0, 0, 0, 0);
 336                         __csc_tiled_to_linear_crop(linear_uv_plane,
 337                                         player->captured.data[1], width, height / 2, 0, 0, 0, 0);
 338
 339                         for (i = 0; i < player->captured.handle_num; i++) {
 340                                 MMPLAYER_FREEIF(player->captured.data[i]);
 341                         }
 342
 343                         src_buffer = (unsigned char*) g_try_malloc(linear_y_plane_size+linear_uv_plane_size);
 344
 345                         if (src_buffer == NULL) {
 346                                 msg.code = MM_ERROR_PLAYER_NO_FREE_SPACE;
 347                                 goto ERROR;
 348                         }
 349                         memset(src_buffer, 0x00, linear_y_plane_size+linear_uv_plane_size);
 350                         memcpy(src_buffer, linear_y_plane, linear_y_plane_size);
 351                         memcpy(src_buffer+linear_y_plane_size, linear_uv_plane, linear_uv_plane_size);
 352
 353                         /* NV12 linear to RGB888 */
 354                         ret = __mm_player_convert_colorspace(player, src_buffer, MM_UTIL_IMG_FMT_NV12,
 355                                 width, height, MM_UTIL_IMG_FMT_RGB888);
 356
 357                         if (ret != MM_ERROR_NONE) {
 358                                 LOGE("failed to convert nv12 linear");
 359                                 goto ERROR;
 360                         }
 361                         /* clean */
 362                         MMPLAYER_FREEIF(src_buffer);
 363                         MMPLAYER_FREEIF(linear_y_plane);
 364                         MMPLAYER_FREEIF(linear_uv_plane);
 365                 } else if (MM_PLAYER_COLORSPACE_NV12 == player->video_cs) {
 366                         #define MM_ALIGN(x, a)      (((x) +(a) - 1) & ~((a) - 1))
 367                         int ret = 0;
 368                         /* using original width otherwises, app can't know aligned to resize */
 369                         planes[0] =  player->captured.stride_width[0] * player->captured.stride_height[0];
 370                         planes[1] = player->captured.stride_width[1] * player->captured.stride_height[1];
 371                         int src_buffer_size = planes[0] + planes[1];
 372                         src_buffer = (unsigned char*) g_try_malloc(src_buffer_size);
 373                         p_buf = src_buffer;
 374
 375                         if (!src_buffer_size) {
 376                                 LOGE("invalid data size");
 377                                 goto ERROR;
 378                         }
 379
 380                         if (!src_buffer) {
 381                                 msg.code = MM_ERROR_PLAYER_NO_FREE_SPACE;
 382                                 goto ERROR;
 383                         }
 384
 385                         memset(src_buffer, 0x00, src_buffer_size);
 386
 387                         temp = player->captured.data[0];
 388
 389                         /* set Y plane */
 390                         for (i = 0; i < player->captured.height[0]; i++) {
 391                                 memcpy(p_buf, temp, player->captured.width[0]);
 392                                 p_buf += player->captured.width[0];
 393                                 temp += player->captured.stride_width[0];
 394                         }
 395
 396                         temp = player->captured.data[1];
 397
 398                         /* set UV plane*/
 399                         for (j = 0; j < player->captured.height[1]; j++) {
 400                                 memcpy(p_buf, temp, player->captured.width[1]);
 401                                 p_buf += player->captured.width[1];
 402                                 temp += player->captured.stride_width[1];
 403                         }
 404
 405                         /* NV12 -> RGB888 */
 406                         ret = __mm_player_convert_colorspace(player, (unsigned char*)src_buffer, MM_UTIL_IMG_FMT_NV12,
 407                                 player->captured.width[0], player->captured.height[0], MM_UTIL_IMG_FMT_RGB888);
 408                         if (ret != MM_ERROR_NONE) {
 409                                 LOGE("failed to convert nv12 linear");
 410                                 goto ERROR;
 411                         }
 412 #ifdef CAPTURE_OUTPUT_DUMP
 413                         capture_output_dump(player);
 414 #endif
 415                         /* clean */
 416                         MMPLAYER_FREEIF(src_buffer);
 417                         /* free captured buf */
 418                         for (i = 0; i < player->captured.handle_num; i++) {
 419                                 MMPLAYER_FREEIF(player->captured.data[i]);
 420                         }
 421                 } else if (MM_PLAYER_COLORSPACE_I420 == player->video_cs) {
 422                         planes[0] = player->captured.stride_width[0] * player->captured.stride_height[0];
 423                         planes[1] = planes[2] = (player->captured.stride_width[0]>>1) * (player->captured.stride_height[0]>>1);
 424
 425                         src_buffer = (unsigned char*) g_try_malloc(player->captured.stride_width[0] * player->captured.stride_height[0]*3/2);
 426                         p_buf = src_buffer;
 427                         /* set Y plane */
 428                         memset(p_buf, 0x00, planes[0]);
 429                         temp = player->captured.data[0];
 430
 431                         for (i = 0; i < player->captured.height[0]; i++) {
 432                                 memcpy(p_buf, temp, player->captured.width[0]);
 433                                 temp += player->captured.stride_width[0];
 434                                 p_buf += player->captured.width[0];
 435                         }
 436
 437                         /* set U plane */
 438                         memset(p_buf, 0x00, planes[1]);
 439                         temp = player->captured.data[1];
 440
 441                         for (i = 0; i < player->captured.height[1]; i++) {
 442                                 memcpy(p_buf, temp, player->captured.width[1]);
 443                                 temp += player->captured.stride_width[1];
 444                                 p_buf += player->captured.width[1];
 445                         }
 446
 447                         /* set V plane */
 448                         memset(p_buf, 0x00, planes[2]);
 449                         temp = player->captured.data[2];
 450
 451                         for (i = 0; i < player->captured.height[2]; i++) {
 452                                 memcpy(p_buf, temp, player->captured.width[2]);
 453                                 temp += player->captured.stride_width[2];
 454                                 p_buf += player->captured.width[2];
 455                         }
 456
 457                         /* I420 -> RGB888 */
 458                         ret = __mm_player_convert_colorspace(player, (unsigned char*)src_buffer, MM_UTIL_IMG_FMT_I420,
 459                                 player->captured.width[0], player->captured.height[0], MM_UTIL_IMG_FMT_RGB888);
 460                         if (ret != MM_ERROR_NONE) {
 461                                 LOGE("failed to convert I420 linear");
 462                                 goto ERROR;
 463                         }
 464 #ifdef CAPTURE_OUTPUT_DUMP
 465                         capture_output_dump(player);
 466 #endif
 467                         /* clean */
 468                         MMPLAYER_FREEIF(src_buffer);
 469                         /* free captured buf */
 470                         for (i = 0; i < player->captured.handle_num; i++) {
 471                                 MMPLAYER_FREEIF(player->captured.data[i]);
 472                         }
 473                 }
 474
 475                 ret = _mmplayer_get_video_rotate_angle((MMHandleType)player, &orientation);
 476                 if (ret != MM_ERROR_NONE) {
 477                         LOGE("failed to get rotation angle");
 478                         goto ERROR;
 479                 }
 480
 481                 LOGD("orientation value = %d", orientation);
 482
 483                 ret = __mmplayer_handle_orientation(player, orientation, MM_UTIL_IMG_FMT_RGB888);
 484                 if (ret != MM_ERROR_NONE) {
 485                         LOGE("failed to convert nv12 linear");
 486                         goto ERROR;
 487                 }
 488
 489                 player->capture.fmt = MM_PLAYER_COLORSPACE_RGB888;
 490                 msg.data = &player->capture;
 491                 msg.size = player->capture.size;
 492 //              msg.captured_frame.width = player->capture.width;
 493 //              msg.captured_frame.height = player->capture.height;
 494 //              msg.captured_frame.orientation = player->capture.orientation;
 495
 496                 if (player->cmd >= MMPLAYER_COMMAND_START) {
 497                         MMPLAYER_POST_MSG(player, MM_MESSAGE_VIDEO_CAPTURED, &msg);
 498                         LOGD("returned from capture message callback");
 499                 }
 500
 501                 //MMPLAYER_FREEIF(player->capture.data);
 502                 continue;
 503 ERROR:
 504                 MMPLAYER_FREEIF(src_buffer);
 505
 506                 for (i = 0; i < player->captured.handle_num; i++) {
 507                         MMPLAYER_FREEIF(player->captured.data[i]);
 508                 }
 509
 510                 if (player->video_cs == MM_PLAYER_COLORSPACE_NV12_TILED) {
 511                         /* clean */
 512                         MMPLAYER_FREEIF(linear_y_plane);
 513                         MMPLAYER_FREEIF(linear_uv_plane);
 514                 }
 515
 516                 msg.union_type = MM_MSG_UNION_CODE;
 517
 518                 MMPLAYER_POST_MSG(player, MM_MESSAGE_VIDEO_NOT_CAPTURED, &msg);
 519         }
 520
 521 EXIT:
 522         MMPLAYER_CAPTURE_THREAD_UNLOCK(player);
 523         return NULL;
 524 }
 525
 526 /**
 527   * The output is fixed as RGB888
 528   */
 529 static int
 530 __mmplayer_get_video_frame_from_buffer(mm_player_t* player, GstPad *pad, GstBuffer *buffer)
 531 {
 532         int ret = MM_ERROR_NONE;
 533         gint planes[MAX_BUFFER_PLANE] = {0, };
 534         gint i = 0;
 535         gint src_width = 0;
 536         gint src_height = 0;
 537         GstCaps *caps = NULL;
 538         GstStructure *structure = NULL;
 539         GstMapInfo mapinfo = GST_MAP_INFO_INIT;
 540         GstMemory *memory = NULL;
 541         mm_util_img_format src_fmt = MM_UTIL_IMG_FMT_YUV420;
 542         mm_util_img_format dst_fmt = MM_UTIL_IMG_FMT_RGB888; // fixed
 543
 544         MMPLAYER_FENTER();
 545
 546         MMPLAYER_RETURN_VAL_IF_FAIL(player, MM_ERROR_PLAYER_NOT_INITIALIZED);
 547         MMPLAYER_RETURN_VAL_IF_FAIL(buffer, MM_ERROR_INVALID_ARGUMENT);
 548
 549         /* get fourcc */
 550         caps = gst_pad_get_current_caps(pad);
 551
 552         MMPLAYER_RETURN_VAL_IF_FAIL(caps, MM_ERROR_INVALID_ARGUMENT);
 553         MMPLAYER_LOG_GST_CAPS_TYPE(caps);
 554
 555         structure = gst_caps_get_structure(caps, 0);
 556         if (!structure) {
 557                 LOGE("cannot get structure from caps.\n");
 558                 ret = MM_ERROR_PLAYER_INTERNAL;
 559                 goto ERROR;
 560         }
 561
 562         /* init capture image buffer */
 563         memset(&player->capture, 0x00, sizeof(MMPlayerVideoCapture));
 564
 565         gst_structure_get_int(structure, "width", &src_width);
 566         gst_structure_get_int(structure, "height", &src_height);
 567
 568         /* check rgb or yuv */
 569         if (gst_structure_has_name(structure, "video/x-raw")) {
 570                 /* NV12T */
 571                 const gchar *gst_format = gst_structure_get_string(structure, "format");
 572
 573                 if (!g_strcmp0(gst_format, "ST12"))
 574                         player->video_cs = MM_PLAYER_COLORSPACE_NV12_TILED;
 575                 else if (!g_strcmp0(gst_format, "S420"))
 576                         player->video_cs = MM_PLAYER_COLORSPACE_I420;
 577                 else if (!g_strcmp0(gst_format, "SN12"))
 578                         player->video_cs = MM_PLAYER_COLORSPACE_NV12;
 579                 else if (!g_strcmp0(gst_format, "BGRx"))
 580                         player->video_cs = MM_PLAYER_COLORSPACE_BGRx;
 581                 else
 582                         player->video_cs = MM_PLAYER_COLORSPACE_MAX;
 583
 584                 LOGI("captured format is %s\n", gst_format);
 585
 586                 if (!g_strcmp0(gst_format, "ST12") || !g_strcmp0(gst_format, "SN12")
 587                         || !g_strcmp0(gst_format, "S420")) {
 588                         guint n;
 589                         MMVideoBuffer *proved = NULL;
 590
 591                         /* get video frame info from proved buffer */
 592                         n = gst_buffer_n_memory(buffer);
 593                         memory = gst_buffer_peek_memory(buffer, n-1);
 594                         gst_memory_map(memory, &mapinfo, GST_MAP_READ);
 595                         proved = (MMVideoBuffer *)mapinfo.data;
 596
 597                         if (!proved || !proved->data[0] || !proved->data[1]) {
 598                                 LOGE("fail to gst_memory_map");
 599                                 ret = MM_ERROR_PLAYER_INTERNAL;
 600                                 goto ERROR;
 601                         }
 602
 603                         memcpy(&player->captured, proved, sizeof(MMVideoBuffer));
 604
 605                         planes[0] = proved->size[0];
 606                         planes[1] = proved->size[1];
 607                         planes[2] = proved->size[2];
 608
 609                         /* Y */
 610                         player->captured.data[0] = g_try_malloc(planes[0]);
 611                         if (!player->captured.data[0]) {
 612                                 gst_memory_unmap(memory, &mapinfo);
 613                                 LOGE("no free space\n");
 614                                 ret = MM_ERROR_PLAYER_NO_FREE_SPACE;
 615                                 goto ERROR;
 616                         }
 617
 618                         /* U */
 619                         if (proved->size[1]) {
 620                                 player->captured.data[1] = g_try_malloc(planes[1]);
 621
 622                                 if (!player->captured.data[1]) {
 623                                         LOGE("no free space\n");
 624                                         gst_memory_unmap(memory, &mapinfo);
 625                                         ret = MM_ERROR_PLAYER_NO_FREE_SPACE;
 626                                         goto ERROR;
 627                                 }
 628                         }
 629
 630                         /* V */
 631                         if (proved->size[2]) {
 632                                 player->captured.data[2] = g_try_malloc(planes[2]);
 633
 634                                 if (!player->captured.data[2]) {
 635                                         LOGE("no free space\n");
 636                                         gst_memory_unmap(memory, &mapinfo);
 637                                         ret = MM_ERROR_PLAYER_NO_FREE_SPACE;
 638                                         goto ERROR;
 639                                 }
 640                         }
 641
 642                         LOGD("Buffer type %d", proved->type);
 643                         if (proved->type == MM_VIDEO_BUFFER_TYPE_TBM_BO) {
 644
 645                                 for (i = 0; i < player->captured.handle_num; i++) {
 646                                         tbm_bo_ref(proved->handle.bo[i]);
 647                                 }
 648                                 LOGD("plane[0] : %p, size %d", proved->data[0], planes[0]);
 649
 650                                 if (proved->data[0])
 651                                         memcpy(player->captured.data[0], proved->data[0], planes[0]);
 652
 653                                 LOGD("plane[1] : %p, size %d", proved->data[1], planes[1]);
 654
 655                                 if (proved->data[1])
 656                                         memcpy(player->captured.data[1], proved->data[1], planes[1]);
 657
 658                                 if (player->video_cs == MM_PLAYER_COLORSPACE_I420) {
 659                                         LOGD("plane[2] : %p, size %d", proved->data[2], planes[2]);
 660
 661                                         if (proved->data[2])
 662                                                 memcpy(player->captured.data[2], proved->data[2], planes[2]);
 663                                 }
 664
 665                                 for (i = 0; i < player->captured.handle_num; i++) {
 666                                         tbm_bo_unref(proved->handle.bo[i]);
 667                                 }
 668                         } else {
 669                                 LOGE("Not support video buffer type %d", proved->type);
 670                                 gst_memory_unmap(memory, &mapinfo);
 671                                 ret = MM_ERROR_PLAYER_INTERNAL;
 672                                 goto ERROR;
 673                         }
 674
 675                         gst_memory_unmap(memory, &mapinfo);
 676                         goto DONE;
 677                 } else {
 678                         GstVideoInfo format_info;
 679                         gst_video_info_from_caps(&format_info, caps);
 680
 681                         switch (GST_VIDEO_INFO_FORMAT(&format_info)) {
 682                         case GST_VIDEO_FORMAT_I420:
 683                                 src_fmt = MM_UTIL_IMG_FMT_I420;
 684                                 break;
 685                         case GST_VIDEO_FORMAT_BGRA:
 686                                 src_fmt = MM_UTIL_IMG_FMT_BGRA8888;
 687                                 break;
 688                         case GST_VIDEO_FORMAT_BGRx:
 689                                 src_fmt = MM_UTIL_IMG_FMT_BGRX8888;
 690                                 break;
 691                         default:
 692                                 LOGE("unknown format to capture\n");
 693                                 ret = MM_ERROR_PLAYER_INTERNAL;
 694                                 goto ERROR;
 695                                 break;
 696                         }
 697                 }
 698         } else {
 699                 LOGE("unknown format to capture\n");
 700                 ret = MM_ERROR_PLAYER_INTERNAL;
 701                 goto ERROR;
 702         }
 703
 704         gst_buffer_map(buffer, &mapinfo, GST_MAP_READ);
 705         __mm_player_convert_colorspace(player, mapinfo.data, src_fmt, src_width, src_height, dst_fmt);
 706         gst_buffer_unmap(buffer, &mapinfo);
 707
 708 DONE:
 709         /* do convert colorspace */
 710         MMPLAYER_CAPTURE_THREAD_SIGNAL(player);
 711
 712         if (caps) {
 713                 gst_caps_unref(caps);
 714                 caps = NULL;
 715         }
 716
 717         return ret;
 718
 719 ERROR:
 720
 721         if (caps) {
 722                 gst_caps_unref(caps);
 723                 caps = NULL;
 724         }
 725
 726         for (i = 0; i < player->captured.handle_num; i++) {
 727                 MMPLAYER_FREEIF(player->captured.data[i]);
 728         }
 729
 730         MMPLAYER_FLEAVE();
 731         return ret;
 732 }
 733
 734 static GstPadProbeReturn
 735 __mmplayer_video_capture_probe(GstPad *pad, GstPadProbeInfo *info, gpointer u_data)
 736 {
 737         mm_player_t* player = (mm_player_t*) u_data;
 738         GstBuffer *buffer = NULL;
 739         int ret = MM_ERROR_NONE;
 740
 741         MMPLAYER_RETURN_VAL_IF_FAIL(info->data, GST_PAD_PROBE_REMOVE);
 742         MMPLAYER_FENTER();
 743
 744         buffer = gst_pad_probe_info_get_buffer(info);
 745         ret = __mmplayer_get_video_frame_from_buffer(player, pad, buffer);
 746
 747         if (ret != MM_ERROR_NONE) {
 748                 LOGE("failed to get video frame");
 749                 return GST_PAD_PROBE_REMOVE;
 750         }
 751
 752         /* remove probe to be called at one time */
 753         if (player->video_capture_cb_probe_id) {
 754                 gst_pad_remove_probe(pad, player->video_capture_cb_probe_id);
 755                 player->video_capture_cb_probe_id = 0;
 756         }
 757
 758         MMPLAYER_FLEAVE();
 759
 760         return GST_PAD_PROBE_OK;
 761 }
 762
 763 static int
 764 __mm_player_convert_colorspace(mm_player_t* player, unsigned char* src_data, mm_util_img_format src_fmt, unsigned int src_w, unsigned int src_h, mm_util_img_format dst_fmt)
 765 {
 766         unsigned char *dst_data = NULL;
 767         unsigned int dst_size;
 768         int ret = MM_ERROR_NONE;
 769
 770         MMPLAYER_RETURN_VAL_IF_FAIL(player, MM_ERROR_PLAYER_INTERNAL);
 771         ret = mm_util_get_image_size(dst_fmt, src_w, src_h, &dst_size);
 772
 773         if (ret != MM_ERROR_NONE) {
 774                 LOGE("failed to get image size for capture, %d\n", ret);
 775                 return MM_ERROR_PLAYER_INTERNAL;
 776         }
 777
 778         SECURE_LOGD("width: %d, height: %d to capture, dest size: %d\n", src_w, src_h, dst_size);
 779
 780         dst_data = (unsigned char*)g_malloc0(dst_size);
 781
 782         if (!dst_data) {
 783                 LOGE("no free space to capture\n");
 784                 g_free(dst_data);
 785                 return MM_ERROR_PLAYER_NO_FREE_SPACE;
 786         }
 787
 788         ret = mm_util_convert_colorspace(src_data, src_w, src_h, src_fmt, dst_data, dst_fmt);
 789
 790         if (ret != MM_ERROR_NONE) {
 791                 LOGE("failed to convert for capture, %d\n", ret);
 792                 g_free(dst_data);
 793                 return MM_ERROR_PLAYER_INTERNAL;
 794         }
 795
 796         player->capture.size = dst_size;
 797         player->capture.data = dst_data;
 798
 799         return MM_ERROR_NONE;
 800 }
 801
 802 /*
 803  * Get tiled address of position(x,y)
 804  *
 805  * @param x_size
 806  *   width of tiled[in]
 807  *
 808  * @param y_size
 809  *   height of tiled[in]
 810  *
 811  * @param x_pos
 812  *   x position of tield[in]
 813  *
 814  * @param src_size
 815  *   y position of tield[in]
 816  *
 817  * @return
 818  *   address of tiled data
 819  */
 820 static int
 821 __tile_4x2_read(int x_size, int y_size, int x_pos, int y_pos)
 822 {
 823         int pixel_x_m1, pixel_y_m1;
 824         int roundup_x;
 825         int linear_addr0, linear_addr1, bank_addr ;
 826         int x_addr;
 827         int trans_addr;
 828
 829         pixel_x_m1 = x_size -1;
 830         pixel_y_m1 = y_size -1;
 831
 832         roundup_x = ((pixel_x_m1 >> 7) + 1);
 833
 834         x_addr = x_pos >> 2;
 835
 836         if ((y_size <= y_pos+32) && (y_pos < y_size) &&
 837                 (((pixel_y_m1 >> 5) & 0x1) == 0) && (((y_pos >> 5) & 0x1) == 0)) {
 838                 linear_addr0 = (((y_pos & 0x1f) <<4) | (x_addr & 0xf));
 839                 linear_addr1 = (((y_pos >> 6) & 0xff) * roundup_x +((x_addr >> 6) & 0x3f));
 840
 841                 if (((x_addr >> 5) & 0x1) == ((y_pos >> 5) & 0x1))
 842                         bank_addr = ((x_addr >> 4) & 0x1);
 843                 else
 844                         bank_addr = 0x2 | ((x_addr >> 4) & 0x1);
 845         } else {
 846                 linear_addr0 = (((y_pos & 0x1f) << 4) | (x_addr & 0xf));
 847                 linear_addr1 = (((y_pos >> 6) & 0xff) * roundup_x +((x_addr >> 5) & 0x7f));
 848
 849                 if (((x_addr >> 5) & 0x1) == ((y_pos >> 5) & 0x1))
 850                         bank_addr = ((x_addr >> 4) & 0x1);
 851                 else
 852                         bank_addr = 0x2 | ((x_addr >> 4) & 0x1);
 853         }
 854
 855         linear_addr0 = linear_addr0 << 2;
 856         trans_addr = (linear_addr1 <<13) | (bank_addr << 11) | linear_addr0;
 857
 858         return trans_addr;
 859 }
 860
 861 /*
 862  * Converts tiled data to linear
 863  * Crops left, top, right, buttom
 864  * 1. Y of NV12T to Y of YUV420P
 865  * 2. Y of NV12T to Y of YUV420S
 866  * 3. UV of NV12T to UV of YUV420S
 867  *
 868  * @param yuv420_dest
 869  *   Y or UV plane address of YUV420[out]
 870  *
 871  * @param nv12t_src
 872  *   Y or UV plane address of NV12T[in]
 873  *
 874  * @param yuv420_width
 875  *   Width of YUV420[in]
 876  *
 877  * @param yuv420_height
 878  *   Y: Height of YUV420, UV: Height/2 of YUV420[in]
 879  *
 880  * @param left
 881  *   Crop size of left
 882  *
 883  * @param top
 884  *   Crop size of top
 885  *
 886  * @param right
 887  *   Crop size of right
 888  *
 889  * @param buttom
 890  *   Crop size of buttom
 891  */
 892 static void
 893 __csc_tiled_to_linear_crop(unsigned char *yuv420_dest, unsigned char *nv12t_src, int yuv420_width, int yuv420_height,
 894                                                         int left, int top, int right, int buttom)
 895 {
 896         int i, j;
 897         int tiled_offset = 0, tiled_offset1 = 0;
 898         int linear_offset = 0;
 899         int temp1 = 0, temp2 = 0, temp3 = 0, temp4 = 0;
 900
 901         temp3 = yuv420_width-right;
 902         temp1 = temp3-left;
 903         /* real width is greater than or equal 256 */
 904         if (temp1 >= 256) {
 905                 for (i = top; i < yuv420_height-buttom; i = i+1) {
 906                         j = left;
 907                         temp3 = (j>>8)<<8;
 908                         temp3 = temp3>>6;
 909                         temp4 = i>>5;
 910                         if (temp4 & 0x1) {
 911                                 /* odd fomula: 2+x+(x>>2)<<2+x_block_num*(y-1) */
 912                                 tiled_offset = temp4-1;
 913                                 temp1 = ((yuv420_width+127)>>7)<<7;
 914                                 tiled_offset = tiled_offset*(temp1>>6);
 915                                 tiled_offset = tiled_offset+temp3;
 916                                 tiled_offset = tiled_offset+2;
 917                                 temp1 = (temp3>>2)<<2;
 918                                 tiled_offset = tiled_offset+temp1;
 919                                 tiled_offset = tiled_offset<<11;
 920                                 tiled_offset1 = tiled_offset+2048*2;
 921                                 temp4 = 8;
 922                         } else {
 923                                 temp2 = ((yuv420_height+31)>>5)<<5;
 924                                 if ((i+32) < temp2) {
 925                                         /* even1 fomula: x+((x+2)>>2)<<2+x_block_num*y */
 926                                         temp1 = temp3+2;
 927                                         temp1 = (temp1>>2)<<2;
 928                                         tiled_offset = temp3+temp1;
 929                                         temp1 = ((yuv420_width+127)>>7)<<7;
 930                                         tiled_offset = tiled_offset+temp4*(temp1>>6);
 931                                         tiled_offset = tiled_offset<<11;
 932                                         tiled_offset1 = tiled_offset+2048*6;
 933                                         temp4 = 8;
 934                                 } else {
 935                                         /* even2 fomula: x+x_block_num*y */
 936                                         temp1 = ((yuv420_width+127)>>7)<<7;
 937                                         tiled_offset = temp4*(temp1>>6);
 938                                         tiled_offset = tiled_offset+temp3;
 939                                         tiled_offset = tiled_offset<<11;
 940                                         tiled_offset1 = tiled_offset+2048*2;
 941                                         temp4 = 4;
 942                                 }
 943                         }
 944
 945                         temp1 = i&0x1F;
 946                         tiled_offset = tiled_offset+64*(temp1);
 947                         tiled_offset1 = tiled_offset1+64*(temp1);
 948                         temp2 = yuv420_width-left-right;
 949                         linear_offset = temp2*(i-top);
 950                         temp3 = ((j+256)>>8)<<8;
 951                         temp3 = temp3-j;
 952                         temp1 = left&0x3F;
 953                         if (temp3 > 192) {
 954                                 memcpy(yuv420_dest+linear_offset, nv12t_src+tiled_offset+temp1, 64-temp1);
 955                                 memcpy(yuv420_dest+linear_offset+64-temp1, nv12t_src+tiled_offset+2048, 64);
 956                                 memcpy(yuv420_dest+linear_offset+128-temp1, nv12t_src+tiled_offset1, 64);
 957                                 memcpy(yuv420_dest+linear_offset+192-temp1, nv12t_src+tiled_offset1+2048, 64);
 958                                 linear_offset = linear_offset+256-temp1;
 959                         } else if (temp3 > 128) {
 960                                 memcpy(yuv420_dest+linear_offset, nv12t_src+tiled_offset+2048+temp1, 64-temp1);
 961                                 memcpy(yuv420_dest+linear_offset+64-temp1, nv12t_src+tiled_offset1, 64);
 962                                 memcpy(yuv420_dest+linear_offset+128-temp1, nv12t_src+tiled_offset1+2048, 64);
 963                                 linear_offset = linear_offset+192-temp1;
 964                         } else if (temp3 > 64) {
 965                                 memcpy(yuv420_dest+linear_offset, nv12t_src+tiled_offset1+temp1, 64-temp1);
 966                                 memcpy(yuv420_dest+linear_offset+64-temp1, nv12t_src+tiled_offset1+2048, 64);
 967                                 linear_offset = linear_offset+128-temp1;
 968                         } else if (temp3 > 0) {
 969                                 memcpy(yuv420_dest+linear_offset, nv12t_src+tiled_offset1+2048+temp1, 64-temp1);
 970                                 linear_offset = linear_offset+64-temp1;
 971                         }
 972
 973                         tiled_offset = tiled_offset+temp4*2048;
 974                         j = (left>>8)<<8;
 975                         j = j + 256;
 976                         temp2 = yuv420_width-right-256;
 977                         for (; j <= temp2; j = j+256) {
 978                                 memcpy(yuv420_dest+linear_offset, nv12t_src+tiled_offset, 64);
 979                                 tiled_offset1 = tiled_offset1+temp4*2048;
 980                                 memcpy(yuv420_dest+linear_offset+64, nv12t_src+tiled_offset+2048, 64);
 981                                 memcpy(yuv420_dest+linear_offset+128, nv12t_src+tiled_offset1, 64);
 982                                 tiled_offset = tiled_offset+temp4*2048;
 983                                 memcpy(yuv420_dest+linear_offset+192, nv12t_src+tiled_offset1+2048, 64);
 984                                 linear_offset = linear_offset+256;
 985                         }
 986
 987                         tiled_offset1 = tiled_offset1+temp4*2048;
 988                         temp2 = yuv420_width-right-j;
 989                         if (temp2 > 192) {
 990                                 memcpy(yuv420_dest+linear_offset, nv12t_src+tiled_offset, 64);
 991                                 memcpy(yuv420_dest+linear_offset+64, nv12t_src+tiled_offset+2048, 64);
 992                                 memcpy(yuv420_dest+linear_offset+128, nv12t_src+tiled_offset1, 64);
 993                                 memcpy(yuv420_dest+linear_offset+192, nv12t_src+tiled_offset1+2048, temp2-192);
 994                         } else if (temp2 > 128) {
 995                                 memcpy(yuv420_dest+linear_offset, nv12t_src+tiled_offset, 64);
 996                                 memcpy(yuv420_dest+linear_offset+64, nv12t_src+tiled_offset+2048, 64);
 997                                 memcpy(yuv420_dest+linear_offset+128, nv12t_src+tiled_offset1, temp2-128);
 998                         } else if (temp2 > 64) {
 999                                 memcpy(yuv420_dest+linear_offset, nv12t_src+tiled_offset, 64);
1000                                 memcpy(yuv420_dest+linear_offset+64, nv12t_src+tiled_offset+2048, temp2-64);
1001                         } else
1002                                 memcpy(yuv420_dest+linear_offset, nv12t_src+tiled_offset, temp2);
1003                 }
1004         } else if (temp1 >= 64) {
1005                 for (i = top; i < (yuv420_height-buttom); i = i+1) {
1006                         j = left;
1007                         tiled_offset = __tile_4x2_read(yuv420_width, yuv420_height, j, i);
1008                         temp2 = ((j+64)>>6)<<6;
1009                         temp2 = temp2-j;
1010                         linear_offset = temp1*(i-top);
1011                         temp4 = j&0x3;
1012                         tiled_offset = tiled_offset+temp4;
1013                         memcpy(yuv420_dest+linear_offset, nv12t_src+tiled_offset, temp2);
1014                         linear_offset = linear_offset+temp2;
1015                         j = j+temp2;
1016                         if ((j+64) <= temp3) {
1017                                 tiled_offset = __tile_4x2_read(yuv420_width, yuv420_height, j, i);
1018                                 memcpy(yuv420_dest+linear_offset, nv12t_src+tiled_offset, 64);
1019                                 linear_offset = linear_offset+64;
1020                                 j = j+64;
1021                         }
1022                         if ((j+64) <= temp3) {
1023                                 tiled_offset = __tile_4x2_read(yuv420_width, yuv420_height, j, i);
1024                                 memcpy(yuv420_dest+linear_offset, nv12t_src+tiled_offset, 64);
1025                                 linear_offset = linear_offset+64;
1026                                 j = j+64;
1027                         }
1028                         if (j < temp3) {
1029                                 tiled_offset = __tile_4x2_read(yuv420_width, yuv420_height, j, i);
1030                                 temp2 = temp3-j;
1031                                 memcpy(yuv420_dest+linear_offset, nv12t_src+tiled_offset, temp2);
1032                         }
1033                 }
1034         } else {
1035                 for (i = top; i < (yuv420_height-buttom); i = i+1) {
1036                         linear_offset = temp1*(i-top);
1037                         for (j = left; j < (yuv420_width-right); j = j+2) {
1038                                 tiled_offset = __tile_4x2_read(yuv420_width, yuv420_height, j, i);
1039                                 temp4 = j&0x3;
1040                                 tiled_offset = tiled_offset+temp4;
1041                                 memcpy(yuv420_dest+linear_offset, nv12t_src+tiled_offset, 2);
1042                                 linear_offset = linear_offset+2;
1043                         }
1044                 }
1045         }
1046 }
1047
1048 #ifdef CAPTURE_OUTPUT_DUMP /* for capture output dump */
1049 static void capture_output_dump(mm_player_t* player)
1050 {
1051         unsigned char *temp = NULL;
1052         char file[100] = { 0, };
1053         FILE *fp = NULL;
1054         int ret = 0;
1055         int i = 0, j = 0;
1056
1057         LOGE("capture output dump start. size = %d", player->capture.size);
1058         sprintf(file, "/tmp/dec_output_dump_%dx%d.yuv", player->captured.width[0], player->captured.height[0]);
1059         fp = fopen(file, "ab");
1060
1061         for (i = 0; i < player->captured.plane_num; i++) {
1062                 temp = player->captured.data[i];
1063
1064                 for (j = 0; j < player->captured.height[i]; j++) {
1065                         ret = fwrite(temp, player->captured.width[i], 1, fp);
1066                         temp += player->captured.width[i];
1067                 }
1068         }
1069         LOGE("capture yuv dumped!! ret = %d", ret);
1070         fclose(fp);
1071
1072         temp = (unsigned char *)player->capture.data;
1073         sprintf(file, "/tmp/dec_output_dump_%dx%d.rgb", player->captured.width[0], player->captured.height[0]);
1074         fp = fopen(file, "ab");
1075         ret = fwrite(temp, player->capture.size, 1, fp);
1076         fclose(fp);
1077         LOGE("capture rgb dumped!! ret = %d", ret);
1078 }
1079 #endif