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 #include "vpx_ports/config.h"
13 #include "vp8/common/idct.h"
14 #include "quantize.h"
15 #include "vp8/common/reconintra.h"
16 #include "vp8/common/reconintra4x4.h"
17 #include "encodemb.h"
18 #include "vp8/common/invtrans.h"
19 #include "vp8/common/recon.h"
20 #include "dct.h"
21 #include "vp8/common/g_common.h"
22 #include "encodeintra.h"
23
24 #define intra4x4ibias_rate 128
25 #define intra4x4pbias_rate 256
26
27
28 #if CONFIG_RUNTIME_CPU_DETECT
29 #define IF_RTCD(x) (x)
30 #else
31 #define IF_RTCD(x) NULL
32 #endif
vp8_encode_intra4x4block(const VP8_ENCODER_RTCD * rtcd,MACROBLOCK * x,BLOCK * be,BLOCKD * b,int best_mode)33 void vp8_encode_intra4x4block(const VP8_ENCODER_RTCD *rtcd, MACROBLOCK *x, BLOCK *be, BLOCKD *b, int best_mode)
34 {
35 vp8_predict_intra4x4(b, best_mode, b->predictor);
36
37 ENCODEMB_INVOKE(&rtcd->encodemb, subb)(be, b, 16);
38
39 x->vp8_short_fdct4x4(be->src_diff, be->coeff, 32);
40
41 x->quantize_b(be, b);
42
43 vp8_inverse_transform_b(IF_RTCD(&rtcd->common->idct), b, 32);
44
45 RECON_INVOKE(&rtcd->common->recon, recon)(b->predictor, b->diff, *(b->base_dst) + b->dst, b->dst_stride);
46 }
47
vp8_encode_intra4x4mby(const VP8_ENCODER_RTCD * rtcd,MACROBLOCK * mb)48 void vp8_encode_intra4x4mby(const VP8_ENCODER_RTCD *rtcd, MACROBLOCK *mb)
49 {
50 int i;
51
52 MACROBLOCKD *x = &mb->e_mbd;
53 vp8_intra_prediction_down_copy(x);
54
55 for (i = 0; i < 16; i++)
56 {
57 BLOCK *be = &mb->block[i];
58 BLOCKD *b = &x->block[i];
59
60 vp8_encode_intra4x4block(rtcd, mb, be, b, b->bmi.mode);
61 }
62
63 return;
64 }
65
vp8_encode_intra16x16mby(const VP8_ENCODER_RTCD * rtcd,MACROBLOCK * x)66 void vp8_encode_intra16x16mby(const VP8_ENCODER_RTCD *rtcd, MACROBLOCK *x)
67 {
68 int b;
69
70 RECON_INVOKE(&rtcd->common->recon, build_intra_predictors_mby)(&x->e_mbd);
71
72 ENCODEMB_INVOKE(&rtcd->encodemb, submby)(x->src_diff, x->src.y_buffer, x->e_mbd.predictor, x->src.y_stride);
73
74 vp8_transform_intra_mby(x);
75
76 vp8_quantize_mby(x);
77
78 #if !(CONFIG_REALTIME_ONLY)
79 #if 1
80 if (x->optimize)
81 vp8_optimize_mby(x, rtcd);
82
83 #endif
84 #endif
85
86 vp8_inverse_transform_mby(IF_RTCD(&rtcd->common->idct), &x->e_mbd);
87
88 RECON_INVOKE(&rtcd->common->recon, recon_mby)
89 (IF_RTCD(&rtcd->common->recon), &x->e_mbd);
90
91 // make sure block modes are set the way we want them for context updates
92 for (b = 0; b < 16; b++)
93 {
94 BLOCKD *d = &x->e_mbd.block[b];
95
96 switch (x->e_mbd.mode_info_context->mbmi.mode)
97 {
98
99 case DC_PRED:
100 d->bmi.mode = B_DC_PRED;
101 break;
102 case V_PRED:
103 d->bmi.mode = B_VE_PRED;
104 break;
105 case H_PRED:
106 d->bmi.mode = B_HE_PRED;
107 break;
108 case TM_PRED:
109 d->bmi.mode = B_TM_PRED;
110 break;
111 default:
112 d->bmi.mode = B_DC_PRED;
113 break;
114
115 }
116 }
117 }
118
vp8_encode_intra16x16mbuv(const VP8_ENCODER_RTCD * rtcd,MACROBLOCK * x)119 void vp8_encode_intra16x16mbuv(const VP8_ENCODER_RTCD *rtcd, MACROBLOCK *x)
120 {
121 vp8_build_intra_predictors_mbuv(&x->e_mbd);
122
123 ENCODEMB_INVOKE(&rtcd->encodemb, submbuv)(x->src_diff, x->src.u_buffer, x->src.v_buffer, x->e_mbd.predictor, x->src.uv_stride);
124
125 vp8_transform_mbuv(x);
126
127 vp8_quantize_mbuv(x);
128
129 #if !(CONFIG_REALTIME_ONLY)
130 #if 1
131
132 if (x->optimize==2 ||(x->optimize && x->rddiv > 1))
133 vp8_optimize_mbuv(x, rtcd);
134
135 #endif
136 #endif
137
138 vp8_inverse_transform_mbuv(IF_RTCD(&rtcd->common->idct), &x->e_mbd);
139
140 vp8_recon_intra_mbuv(IF_RTCD(&rtcd->common->recon), &x->e_mbd);
141 }
142
143