1 /*
2 * Copyright (c) 2019, Intel Corporation
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice shall be included
12 * in all copies or substantial portions of the Software.
13 *
14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
15 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
17 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
18 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
19 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
20 * OTHER DEALINGS IN THE SOFTWARE.
21 */
22
23 //!
24 //! \file encode_av1_vdenc_feature_manager.cpp
25 //! \brief Defines the common interface for av1 vdenc feature manager
26 //! \details The encode feature manager is further sub-divided by codec type
27 //! this file is for the base interface which is shared by all components.
28 //!
29
30 #include "encode_av1_vdenc_feature_manager.h"
31 #include "encode_av1_reference_frames.h"
32 #include "encode_av1_vdenc_const_settings.h"
33
34 namespace encode
35 {
36
CreateConstSettings()37 MOS_STATUS EncodeAv1VdencFeatureManager::CreateConstSettings()
38 {
39 ENCODE_FUNC_CALL();
40 m_featureConstSettings = MOS_New(EncodeAv1VdencConstSettings, m_hwInterface->GetOsInterface());
41 return MOS_STATUS_SUCCESS;
42 }
43
CheckFeatures(void * params)44 MOS_STATUS EncodeAv1VdencFeatureManager::CheckFeatures(void *params)
45 {
46 ENCODE_FUNC_CALL();
47 EncoderParams *encodeParams = (EncoderParams *)params;
48
49 PCODEC_AV1_ENCODE_SEQUENCE_PARAMS av1SeqParams =
50 static_cast<PCODEC_AV1_ENCODE_SEQUENCE_PARAMS>(encodeParams->pSeqParams);
51 ENCODE_CHK_NULL_RETURN(av1SeqParams);
52
53 auto settings = static_cast<EncodeAv1VdencConstSettings*>(m_featureConstSettings);
54 ENCODE_CHK_NULL_RETURN(settings);
55 settings->Update(params);
56
57 if (encodeParams->bNewSeq)
58 {
59 m_ddiTargetUsage = av1SeqParams->TargetUsage;
60 ENCODE_CHK_STATUS_RETURN(MapTargetUsage(av1SeqParams->TargetUsage));
61 m_targetUsage = av1SeqParams->TargetUsage;
62 }
63
64 ENCODE_CHK_STATUS_RETURN(SetPassNum(av1SeqParams));
65
66 return MOS_STATUS_SUCCESS;
67 }
68
CreateFeatures(void * constSettings)69 MOS_STATUS EncodeAv1VdencFeatureManager::CreateFeatures(void *constSettings)
70 {
71 ENCODE_FUNC_CALL();
72
73 EncodeBasicFeature *encBasic = MOS_New(Av1BasicFeature, m_allocator, m_hwInterface, m_trackedBuf, m_recycleResource, constSettings);
74 ENCODE_CHK_STATUS_RETURN(RegisterFeatures(Av1FeatureIDs::basicFeature, encBasic));
75
76 Av1EncodeTile *encTile = MOS_New(Av1EncodeTile, this, m_allocator, m_hwInterface, constSettings);
77 ENCODE_CHK_STATUS_RETURN(RegisterFeatures(Av1FeatureIDs::encodeTile, encTile));
78
79 Av1Segmentation *encSegmentation = MOS_New(Av1Segmentation, this, m_allocator, constSettings);
80 ENCODE_CHK_STATUS_RETURN(RegisterFeatures(Av1FeatureIDs::av1Segmentation, encSegmentation));
81
82 Av1Brc *encBrc = MOS_New(Av1Brc, this, m_allocator, m_hwInterface, constSettings);
83 ENCODE_CHK_STATUS_RETURN(RegisterFeatures(Av1FeatureIDs::av1BrcFeature, encBrc));
84
85 return MOS_STATUS_SUCCESS;
86 }
87
MapTargetUsage(uint8_t & targetUsage)88 MOS_STATUS EncodeAv1VdencFeatureManager::MapTargetUsage(uint8_t &targetUsage)
89 {
90 ENCODE_FUNC_CALL();
91
92 switch (targetUsage)
93 {
94 case 1:
95 case 2:
96 targetUsage = 2;
97 break;
98 case 3:
99 case 4:
100 case 5:
101 targetUsage = 4;
102 break;
103 case 6:
104 case 7:
105 targetUsage = 7;
106 break;
107 default:
108 targetUsage = 4;
109 break;
110 }
111 return MOS_STATUS_SUCCESS;
112 }
113
114
SetPassNum(PCODEC_AV1_ENCODE_SEQUENCE_PARAMS av1SeqParams)115 MOS_STATUS EncodeAv1VdencFeatureManager::SetPassNum(
116 PCODEC_AV1_ENCODE_SEQUENCE_PARAMS av1SeqParams)
117 {
118 ENCODE_FUNC_CALL();
119 ENCODE_CHK_NULL_RETURN(av1SeqParams);
120
121 if (IsRateControlBrc(av1SeqParams->RateControlMethod))
122 {
123 m_passNum = 2;
124 }
125 else
126 {
127 m_passNum = 1;
128 }
129
130 return MOS_STATUS_SUCCESS;
131 }
132 }
133