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