segmentation_common.c 2.21 KB
Newer Older
John Koleszar's avatar
John Koleszar committed
1
2
3
/*
 *  Copyright (c) 2010 The VP8 project authors. All Rights Reserved.
 *
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
13
14
 */


#include "segmentation_common.h"
#include "vpx_mem/vpx_mem.h"

15
void vp8_update_gf_useage_maps(VP8_COMP *cpi, VP8_COMMON *cm, MACROBLOCK *x)
John Koleszar's avatar
John Koleszar committed
16
17
18
19
20
{
    int mb_row, mb_col;

    MODE_INFO *this_mb_mode_info = cm->mi;

21
    x->gf_active_ptr = (signed char *)cpi->gf_active_flags;
John Koleszar's avatar
John Koleszar committed
22
23
24
25

    if ((cm->frame_type == KEY_FRAME) || (cm->refresh_golden_frame))
    {
        // Reset Gf useage monitors
26
27
        vpx_memset(cpi->gf_active_flags, 1, (cm->mb_rows * cm->mb_cols));
        cpi->gf_active_count = cm->mb_rows * cm->mb_cols;
John Koleszar's avatar
John Koleszar committed
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
    }
    else
    {
        // for each macroblock row in image
        for (mb_row = 0; mb_row < cm->mb_rows; mb_row++)
        {
            // for each macroblock col in image
            for (mb_col = 0; mb_col < cm->mb_cols; mb_col++)
            {

                // If using golden then set GF active flag if not already set.
                // If using last frame 0,0 mode then leave flag as it is
                // else if using non 0,0 motion or intra modes then clear flag if it is currently set
                if ((this_mb_mode_info->mbmi.ref_frame == GOLDEN_FRAME) || (this_mb_mode_info->mbmi.ref_frame == ALTREF_FRAME))
                {
43
                    if (*(x->gf_active_ptr) == 0)
John Koleszar's avatar
John Koleszar committed
44
                    {
45
46
                        *(x->gf_active_ptr) = 1;
                        cpi->gf_active_count ++;
John Koleszar's avatar
John Koleszar committed
47
48
                    }
                }
49
                else if ((this_mb_mode_info->mbmi.mode != ZEROMV) && *(x->gf_active_ptr))
John Koleszar's avatar
John Koleszar committed
50
                {
51
52
                    *(x->gf_active_ptr) = 0;
                    cpi->gf_active_count--;
John Koleszar's avatar
John Koleszar committed
53
54
                }

55
                x->gf_active_ptr++;          // Step onto next entry
John Koleszar's avatar
John Koleszar committed
56
57
58
59
60
61
62
63
64
                this_mb_mode_info++;           // skip to next mb

            }

            // this is to account for the border
            this_mb_mode_info++;
        }
    }
}