1 /*
2 * Copyright (c) 2010 The WebM 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 #ifndef VP8_ENCODER_RDOPT_H_
12 #define VP8_ENCODER_RDOPT_H_
13
14 #include "./vpx_config.h"
15
16 #ifdef __cplusplus
17 extern "C" {
18 #endif
19
20 #define RDCOST(RM, DM, R, D) (((128 + (R) * (RM)) >> 8) + (DM) * (D))
21
insertsortmv(int arr[],int len)22 static INLINE void insertsortmv(int arr[], int len) {
23 int i, j, k;
24
25 for (i = 1; i <= len - 1; ++i) {
26 for (j = 0; j < i; ++j) {
27 if (arr[j] > arr[i]) {
28 int temp;
29
30 temp = arr[i];
31
32 for (k = i; k > j; k--) arr[k] = arr[k - 1];
33
34 arr[j] = temp;
35 }
36 }
37 }
38 }
39
insertsortsad(int arr[],int idx[],int len)40 static INLINE void insertsortsad(int arr[], int idx[], int len) {
41 int i, j, k;
42
43 for (i = 1; i <= len - 1; ++i) {
44 for (j = 0; j < i; ++j) {
45 if (arr[j] > arr[i]) {
46 int temp, tempi;
47
48 temp = arr[i];
49 tempi = idx[i];
50
51 for (k = i; k > j; k--) {
52 arr[k] = arr[k - 1];
53 idx[k] = idx[k - 1];
54 }
55
56 arr[j] = temp;
57 idx[j] = tempi;
58 }
59 }
60 }
61 }
62
63 extern void vp8_initialize_rd_consts(VP8_COMP *cpi, MACROBLOCK *x, int Qvalue);
64 extern void vp8_rd_pick_inter_mode(VP8_COMP *cpi, MACROBLOCK *x,
65 int recon_yoffset, int recon_uvoffset,
66 int *returnrate, int *returndistortion,
67 int *returnintra, int mb_row, int mb_col);
68 extern void vp8_rd_pick_intra_mode(MACROBLOCK *x, int *rate);
69
get_plane_pointers(const YV12_BUFFER_CONFIG * fb,unsigned char * plane[3],unsigned int recon_yoffset,unsigned int recon_uvoffset)70 static INLINE void get_plane_pointers(const YV12_BUFFER_CONFIG *fb,
71 unsigned char *plane[3],
72 unsigned int recon_yoffset,
73 unsigned int recon_uvoffset) {
74 plane[0] = fb->y_buffer + recon_yoffset;
75 plane[1] = fb->u_buffer + recon_uvoffset;
76 plane[2] = fb->v_buffer + recon_uvoffset;
77 }
78
get_predictor_pointers(const VP8_COMP * cpi,unsigned char * plane[4][3],unsigned int recon_yoffset,unsigned int recon_uvoffset)79 static INLINE void get_predictor_pointers(const VP8_COMP *cpi,
80 unsigned char *plane[4][3],
81 unsigned int recon_yoffset,
82 unsigned int recon_uvoffset) {
83 if (cpi->ref_frame_flags & VP8_LAST_FRAME) {
84 get_plane_pointers(&cpi->common.yv12_fb[cpi->common.lst_fb_idx],
85 plane[LAST_FRAME], recon_yoffset, recon_uvoffset);
86 }
87
88 if (cpi->ref_frame_flags & VP8_GOLD_FRAME) {
89 get_plane_pointers(&cpi->common.yv12_fb[cpi->common.gld_fb_idx],
90 plane[GOLDEN_FRAME], recon_yoffset, recon_uvoffset);
91 }
92
93 if (cpi->ref_frame_flags & VP8_ALTR_FRAME) {
94 get_plane_pointers(&cpi->common.yv12_fb[cpi->common.alt_fb_idx],
95 plane[ALTREF_FRAME], recon_yoffset, recon_uvoffset);
96 }
97 }
98
get_reference_search_order(const VP8_COMP * cpi,int ref_frame_map[4])99 static INLINE void get_reference_search_order(const VP8_COMP *cpi,
100 int ref_frame_map[4]) {
101 int i = 0;
102
103 ref_frame_map[i++] = INTRA_FRAME;
104 if (cpi->ref_frame_flags & VP8_LAST_FRAME) ref_frame_map[i++] = LAST_FRAME;
105 if (cpi->ref_frame_flags & VP8_GOLD_FRAME) ref_frame_map[i++] = GOLDEN_FRAME;
106 if (cpi->ref_frame_flags & VP8_ALTR_FRAME) ref_frame_map[i++] = ALTREF_FRAME;
107 for (; i < 4; ++i) ref_frame_map[i] = -1;
108 }
109
110 extern void vp8_mv_pred(VP8_COMP *cpi, MACROBLOCKD *xd, const MODE_INFO *here,
111 int_mv *mvp, int refframe, int *ref_frame_sign_bias,
112 int *sr, int near_sadidx[]);
113 void vp8_cal_sad(VP8_COMP *cpi, MACROBLOCKD *xd, MACROBLOCK *x,
114 int recon_yoffset, int near_sadidx[]);
115 int VP8_UVSSE(MACROBLOCK *x);
116 int vp8_cost_mv_ref(MB_PREDICTION_MODE m, const int near_mv_ref_ct[4]);
117 void vp8_set_mbmode_and_mvs(MACROBLOCK *x, MB_PREDICTION_MODE mb, int_mv *mv);
118
119 #ifdef __cplusplus
120 } // extern "C"
121 #endif
122
123 #endif // VP8_ENCODER_RDOPT_H_
124