vp8/encoder/segmentation.c

   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 "segmentation.h"
  13 #include "vpx_mem/vpx_mem.h"
  14
  15 void vp8_update_gf_useage_maps(VP8_COMP *cpi, VP8_COMMON *cm, MACROBLOCK *x)
  16 {
  17     int mb_row, mb_col;
  18
  19     MODE_INFO *this_mb_mode_info = cm->mi;
  20
  21     x->gf_active_ptr = (signed char *)cpi->gf_active_flags;
  22
  23     if ((cm->frame_type == KEY_FRAME) || (cm->refresh_golden_frame))
  24     {
  25         // Reset Gf useage monitors
  26         vpx_memset(cpi->gf_active_flags, 1, (cm->mb_rows * cm->mb_cols));
  27         cpi->gf_active_count = cm->mb_rows * cm->mb_cols;
  28     }
  29     else
  30     {
  31         // for each macroblock row in image
  32         for (mb_row = 0; mb_row < cm->mb_rows; mb_row++)
  33         {
  34             // for each macroblock col in image
  35             for (mb_col = 0; mb_col < cm->mb_cols; mb_col++)
  36             {
  37
  38                 // If using golden then set GF active flag if not already set.
  39                 // If using last frame 0,0 mode then leave flag as it is
  40                 // else if using non 0,0 motion or intra modes then clear flag if it is currently set
  41                 if ((this_mb_mode_info->mbmi.ref_frame == GOLDEN_FRAME) || (this_mb_mode_info->mbmi.ref_frame == ALTREF_FRAME))
  42                 {
  43                     if (*(x->gf_active_ptr) == 0)
  44                     {
  45                         *(x->gf_active_ptr) = 1;
  46                         cpi->gf_active_count ++;
  47                     }
  48                 }
  49                 else if ((this_mb_mode_info->mbmi.mode != ZEROMV) && *(x->gf_active_ptr))
  50                 {
  51                     *(x->gf_active_ptr) = 0;
  52                     cpi->gf_active_count--;
  53                 }
  54
  55                 x->gf_active_ptr++;          // Step onto next entry
  56                 this_mb_mode_info++;           // skip to next mb
  57
  58             }
  59
  60             // this is to account for the border
  61             this_mb_mode_info++;
  62         }
  63     }
  64 }