h264_dpb.cc revision 5821806d5e7f356e8fa4b058a389a808ea183019
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <algorithm> 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/stl_util.h" 9a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles)#include "content/common/gpu/media/h264_dpb.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace content { 12868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)H264DPB::H264DPB() {} 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)H264DPB::~H264DPB() {} 15a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void H264DPB::Clear() { 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pics_.clear(); 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void H264DPB::RemoveByPOC(int poc) { 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (Pictures::iterator it = pics_.begin(); it != pics_.end(); ++it) { 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if ((*it)->pic_order_cnt == poc) { 233551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) pics_.erase(it); 243551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) return; 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED() << "Missing POC: " << poc; 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void H264DPB::RemoveUnused() { 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (Pictures::iterator it = pics_.begin(); it != pics_.end(); ) { 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if ((*it)->outputted && !(*it)->ref) 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) it = pics_.erase(it); 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) else 3590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ++it; 3690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 37c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void H264DPB::StorePic(H264Picture* pic) { 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_LT(pics_.size(), kDPBMaxSize); 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DVLOG(3) << "Adding PicNum: " << pic->pic_num << " ref: " << (int)pic->ref 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << " longterm: " << (int)pic->long_term << " to DPB"; 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pics_.push_back(pic); 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int H264DPB::CountRefPics() { 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int ret = 0; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < pics_.size(); ++i) { 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (pics_[i]->ref) 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ++ret; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 52a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) return ret; 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void H264DPB::MarkAllUnusedForRef() { 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < pics_.size(); ++i) 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pics_[i]->ref = false; 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)H264Picture* H264DPB::GetShortRefPicByPicNum(int pic_num) { 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < pics_.size(); ++i) { 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) H264Picture* pic = pics_[i]; 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (pic->ref && !pic->long_term && pic->pic_num == pic_num) 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return pic; 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DVLOG(1) << "Missing short ref pic num: " << pic_num; 6890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) return NULL; 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 70ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch 71ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben MurdochH264Picture* H264DPB::GetLongRefPicByLongTermPicNum(int pic_num) { 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < pics_.size(); ++i) { 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) H264Picture* pic = pics_[i]; 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (pic->ref && pic->long_term && pic->long_term_pic_num == pic_num) 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return pic; 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DVLOG(1) << "Missing long term pic num: " << pic_num; 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NULL; 80868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)} 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)H264Picture* H264DPB::GetLowestFrameNumWrapShortRefPic() { 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) H264Picture* ret = NULL; 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < pics_.size(); ++i) { 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) H264Picture* pic = pics_[i]; 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (pic->ref && !pic->long_term && 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) (!ret || pic->frame_num_wrap < ret->frame_num_wrap)) 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ret = pic; 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 90a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) return ret; 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void H264DPB::GetNotOutputtedPicsAppending(H264Picture::PtrVector& out) { 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < pics_.size(); ++i) { 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) H264Picture* pic = pics_[i]; 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!pic->outputted) 972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) out.push_back(pic); 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 99f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)} 100f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 10158537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)void H264DPB::GetShortTermRefPicsAppending(H264Picture::PtrVector& out) { 102a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) for (size_t i = 0; i < pics_.size(); ++i) { 103a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) H264Picture* pic = pics_[i]; 104a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) if (pic->ref && !pic->long_term) 10558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) out.push_back(pic); 106a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) } 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void H264DPB::GetLongTermRefPicsAppending(H264Picture::PtrVector& out) { 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < pics_.size(); ++i) { 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) H264Picture* pic = pics_[i]; 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (pic->ref && pic->long_term) 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) out.push_back(pic); 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1167d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace content 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)