1 /*
2 * Copyright (C) 2021 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 //#define LOG_NDEBUG 0
18 #define LOG_TAG "VideoShaper"
19 #include <utils/Log.h>
20
21 #include <string>
22 #include <inttypes.h>
23
24 #include <media/NdkMediaFormat.h>
25
26 #include "CodecProperties.h"
27 #include "VideoShaper.h"
28 #include "VQops.h"
29
30 namespace android {
31 namespace mediaformatshaper {
32
33 // mediatype-specific operations
34
35 vqOps_t mediaInfo[] = {
36 {
37 .mediaType = "video/avc",
38 .qpMin = 0,
39 .qpMax = 51,
40 .qpDelta = 3,
41 },
42 {
43 .mediaType = "video/hevc",
44 .qpMin = 0,
45 .qpMax = 51,
46 .qpDelta = 3,
47 },
48 {
49 .mediaType = NULL, // matches everything, it must come last
50 .qpMin = INT32_MIN,
51 .qpMax = INT32_MAX,
52 .qpDelta = 3,
53 }
54 };
55 int nMediaInfos = sizeof(mediaInfo) / sizeof(mediaInfo[0]);
56
57 //
58 // Caller retains ownership of and responsibility for inFormat
59 //
60
videoShaper(CodecProperties * codec,AMediaFormat * inFormat,int flags)61 int videoShaper(CodecProperties *codec, AMediaFormat* inFormat, int flags) {
62 if (codec == nullptr) {
63 return -1;
64 }
65 ALOGV("codec %s inFormat %p flags x%x", codec->getName().c_str(), inFormat, flags);
66
67 int ix;
68
69 std::string mediaType = codec->getMediaType();
70 // we should always come out of this with a selection, because the final entry
71 // is deliberaly a NULL -- so that it will act as a default
72 for(ix = 0; mediaInfo[ix].mediaType != NULL; ix++) {
73 if (strcmp(mediaType.c_str(), mediaInfo[ix].mediaType) == 0) {
74 break;
75 }
76 }
77 if (ix >= nMediaInfos) {
78 // shouldn't happen, but if it does .....
79 }
80
81 vqOps_t *info = &mediaInfo[ix];
82
83 // apply any quality transforms in here..
84 (void) VQApply(codec, info, inFormat, flags);
85
86 // We always spread any QP parameters.
87 // Sometimes it's something we inserted here, sometimes it's a value that the user injected.
88 //
89 qpSpreadPerFrameType(inFormat, info->qpDelta, info->qpMin, info->qpMax, /* override */ true);
90
91 //
92 return 0;
93
94 }
95
96 } // namespace mediaformatshaper
97 } // namespace android
98
99