1 /*
2 * Copyright 2017 The WebRTC project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include <jni.h>
12 #include <vector>
13
14 #include "third_party/libyuv/include/libyuv/convert.h"
15 #include "third_party/libyuv/include/libyuv/scale.h"
16
17 #include "rtc_base/checks.h"
18 #include "sdk/android/generated_video_jni/NV12Buffer_jni.h"
19
20 namespace webrtc {
21 namespace jni {
22
JNI_NV12Buffer_CropAndScale(JNIEnv * jni,jint crop_x,jint crop_y,jint crop_width,jint crop_height,jint scale_width,jint scale_height,const JavaParamRef<jobject> & j_src,jint src_width,jint src_height,jint src_stride,jint src_slice_height,const JavaParamRef<jobject> & j_dst_y,jint dst_stride_y,const JavaParamRef<jobject> & j_dst_u,jint dst_stride_u,const JavaParamRef<jobject> & j_dst_v,jint dst_stride_v)23 static void JNI_NV12Buffer_CropAndScale(JNIEnv* jni,
24 jint crop_x,
25 jint crop_y,
26 jint crop_width,
27 jint crop_height,
28 jint scale_width,
29 jint scale_height,
30 const JavaParamRef<jobject>& j_src,
31 jint src_width,
32 jint src_height,
33 jint src_stride,
34 jint src_slice_height,
35 const JavaParamRef<jobject>& j_dst_y,
36 jint dst_stride_y,
37 const JavaParamRef<jobject>& j_dst_u,
38 jint dst_stride_u,
39 const JavaParamRef<jobject>& j_dst_v,
40 jint dst_stride_v) {
41 const int src_stride_y = src_stride;
42 const int src_stride_uv = src_stride;
43 const int crop_chroma_x = crop_x / 2;
44 const int crop_chroma_y = crop_y / 2;
45 const int crop_chroma_width = (crop_width + 1) / 2;
46 const int crop_chroma_height = (crop_height + 1) / 2;
47 const int tmp_stride_u = crop_chroma_width;
48 const int tmp_stride_v = crop_chroma_width;
49 const int tmp_size = crop_chroma_height * (tmp_stride_u + tmp_stride_v);
50
51 uint8_t const* src_y =
52 static_cast<uint8_t const*>(jni->GetDirectBufferAddress(j_src.obj()));
53 uint8_t const* src_uv = src_y + src_slice_height * src_stride_y;
54
55 uint8_t* dst_y =
56 static_cast<uint8_t*>(jni->GetDirectBufferAddress(j_dst_y.obj()));
57 uint8_t* dst_u =
58 static_cast<uint8_t*>(jni->GetDirectBufferAddress(j_dst_u.obj()));
59 uint8_t* dst_v =
60 static_cast<uint8_t*>(jni->GetDirectBufferAddress(j_dst_v.obj()));
61
62 // Crop using pointer arithmetic.
63 src_y += crop_x + crop_y * src_stride_y;
64 src_uv += 2 * crop_chroma_x + crop_chroma_y * src_stride_uv;
65
66 std::vector<uint8_t> tmp_buffer(tmp_size);
67 uint8_t* tmp_u = tmp_buffer.data();
68 uint8_t* tmp_v = tmp_u + crop_chroma_height * tmp_stride_u;
69
70 libyuv::SplitUVPlane(src_uv, src_stride_uv, tmp_u, tmp_stride_u, tmp_v,
71 tmp_stride_v, crop_chroma_width, crop_chroma_height);
72
73 libyuv::I420Scale(src_y, src_stride_y, tmp_u, tmp_stride_u, tmp_v,
74 tmp_stride_v, crop_width, crop_height, dst_y, dst_stride_y,
75 dst_u, dst_stride_u, dst_v, dst_stride_v, scale_width,
76 scale_height, libyuv::kFilterBox);
77 }
78
79 } // namespace jni
80 } // namespace webrtc
81