encodeintra.c 3.37 KB
Newer Older
John Koleszar's avatar
John Koleszar committed
1
/*
2
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
John Koleszar's avatar
John Koleszar committed
3
 *
4
 *  Use of this source code is governed by a BSD-style license
5
6
 *  that can be found in the LICENSE file in the root of the source
 *  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.
John Koleszar's avatar
John Koleszar committed
9
10
11
12
 */


#include "vpx_ports/config.h"
John Koleszar's avatar
John Koleszar committed
13
#include "vp8/common/idct.h"
John Koleszar's avatar
John Koleszar committed
14
#include "quantize.h"
John Koleszar's avatar
John Koleszar committed
15
16
#include "vp8/common/reconintra.h"
#include "vp8/common/reconintra4x4.h"
John Koleszar's avatar
John Koleszar committed
17
#include "encodemb.h"
John Koleszar's avatar
John Koleszar committed
18
19
#include "vp8/common/invtrans.h"
#include "vp8/common/recon.h"
John Koleszar's avatar
John Koleszar committed
20
#include "dct.h"
John Koleszar's avatar
John Koleszar committed
21
#include "vp8/common/g_common.h"
John Koleszar's avatar
John Koleszar committed
22
23
24
25
26
27
28
29
#include "encodeintra.h"


#if CONFIG_RUNTIME_CPU_DETECT
#define IF_RTCD(x) (x)
#else
#define IF_RTCD(x) NULL
#endif
30
31
32
33


void vp8_encode_intra4x4block(const VP8_ENCODER_RTCD *rtcd,
                              MACROBLOCK *x, int ib)
John Koleszar's avatar
John Koleszar committed
34
{
35
36
37
    BLOCKD *b = &x->e_mbd.block[ib];
    BLOCK *be = &x->block[ib];

38
    RECON_INVOKE(&rtcd->common->recon, intra4x4_predict)
Scott LaVarnway's avatar
Scott LaVarnway committed
39
                (b, b->bmi.as_mode, b->predictor);
John Koleszar's avatar
John Koleszar committed
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

    ENCODEMB_INVOKE(&rtcd->encodemb, subb)(be, b, 16);

    x->vp8_short_fdct4x4(be->src_diff, be->coeff, 32);

    x->quantize_b(be, b);

    vp8_inverse_transform_b(IF_RTCD(&rtcd->common->idct), b, 32);

    RECON_INVOKE(&rtcd->common->recon, recon)(b->predictor, b->diff, *(b->base_dst) + b->dst, b->dst_stride);
}

void vp8_encode_intra4x4mby(const VP8_ENCODER_RTCD *rtcd, MACROBLOCK *mb)
{
    int i;

    MACROBLOCKD *x = &mb->e_mbd;
    vp8_intra_prediction_down_copy(x);

    for (i = 0; i < 16; i++)
60
        vp8_encode_intra4x4block(rtcd, mb, i);
John Koleszar's avatar
John Koleszar committed
61
62
63
64
65
66
67
    return;
}

void vp8_encode_intra16x16mby(const VP8_ENCODER_RTCD *rtcd, MACROBLOCK *x)
{
    int b;

68
    RECON_INVOKE(&rtcd->common->recon, build_intra_predictors_mby)(&x->e_mbd);
John Koleszar's avatar
John Koleszar committed
69
70
71
72
73
74
75

    ENCODEMB_INVOKE(&rtcd->encodemb, submby)(x->src_diff, x->src.y_buffer, x->e_mbd.predictor, x->src.y_stride);

    vp8_transform_intra_mby(x);

    vp8_quantize_mby(x);

76
    if (x->optimize)
John Koleszar's avatar
John Koleszar committed
77
78
79
80
        vp8_optimize_mby(x, rtcd);

    vp8_inverse_transform_mby(IF_RTCD(&rtcd->common->idct), &x->e_mbd);

81
82
    RECON_INVOKE(&rtcd->common->recon, recon_mby)
        (IF_RTCD(&rtcd->common->recon), &x->e_mbd);
John Koleszar's avatar
John Koleszar committed
83
84
85
86
87
88

    // make sure block modes are set the way we want them for context updates
    for (b = 0; b < 16; b++)
    {
        BLOCKD *d = &x->e_mbd.block[b];

89
        switch (x->e_mbd.mode_info_context->mbmi.mode)
John Koleszar's avatar
John Koleszar committed
90
91
        {
        case DC_PRED:
Scott LaVarnway's avatar
Scott LaVarnway committed
92
            d->bmi.as_mode = B_DC_PRED;
John Koleszar's avatar
John Koleszar committed
93
94
            break;
        case V_PRED:
Scott LaVarnway's avatar
Scott LaVarnway committed
95
            d->bmi.as_mode = B_VE_PRED;
John Koleszar's avatar
John Koleszar committed
96
97
            break;
        case H_PRED:
Scott LaVarnway's avatar
Scott LaVarnway committed
98
            d->bmi.as_mode = B_HE_PRED;
John Koleszar's avatar
John Koleszar committed
99
100
            break;
        case TM_PRED:
Scott LaVarnway's avatar
Scott LaVarnway committed
101
            d->bmi.as_mode = B_TM_PRED;
John Koleszar's avatar
John Koleszar committed
102
103
            break;
        default:
Scott LaVarnway's avatar
Scott LaVarnway committed
104
            d->bmi.as_mode = B_DC_PRED;
John Koleszar's avatar
John Koleszar committed
105
106
107
108
109
110
111
            break;
        }
    }
}

void vp8_encode_intra16x16mbuv(const VP8_ENCODER_RTCD *rtcd, MACROBLOCK *x)
{
112
    RECON_INVOKE(&rtcd->common->recon, build_intra_predictors_mbuv)(&x->e_mbd);
John Koleszar's avatar
John Koleszar committed
113
114
115
116
117
118
119

    ENCODEMB_INVOKE(&rtcd->encodemb, submbuv)(x->src_diff, x->src.u_buffer, x->src.v_buffer, x->e_mbd.predictor, x->src.uv_stride);

    vp8_transform_mbuv(x);

    vp8_quantize_mbuv(x);

Yaowu Xu's avatar
Yaowu Xu committed
120
    if (x->optimize)
John Koleszar's avatar
John Koleszar committed
121
122
123
124
125
126
127
        vp8_optimize_mbuv(x, rtcd);

    vp8_inverse_transform_mbuv(IF_RTCD(&rtcd->common->idct), &x->e_mbd);

    vp8_recon_intra_mbuv(IF_RTCD(&rtcd->common->recon), &x->e_mbd);
}