1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
// Copyright 2020 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef ANDROID_V4L2_CODEC2_COMMON_HELPERS_H
#define ANDROID_V4L2_CODEC2_COMMON_HELPERS_H
#include <C2Config.h>
#include <system/graphics.h>
#include <ui/Size.h>
#include <v4l2_codec2/common/VideoPixelFormat.h>
namespace android {
// The encoder parameter set.
// |mInputFormat| is the pixel format of the input frames.
// |mInputVisibleSize| is the resolution of the input frames.
// |mOutputProfile| is the codec profile of the encoded output stream.
// |mInitialBitrate| is the initial bitrate of the encoded output stream, in bits per second.
// |mInitialFramerate| is the initial requested framerate.
// |mH264OutputLevel| is H264 level of encoded output stream.
// |mStorageType| is the storage type of video frame provided on encode().
struct VideoEncoderAcceleratorConfig {
enum VideoFrameStorageType {
SHMEM = 0,
DMABUF = 1,
};
VideoPixelFormat mInputFormat;
ui::Size mInputVisibleSize;
C2Config::profile_t mOutputProfile;
uint32_t mInitialBitrate;
uint32_t mInitialFramerate;
uint8_t mH264OutputLevel;
VideoFrameStorageType mStorageType;
};
// Convert the specified C2Config level to a V4L2 level.
uint8_t c2LevelToV4L2Level(C2Config::level_t level);
// Get the specified graphics block in YCbCr format.
android_ycbcr getGraphicBlockInfo(const C2ConstGraphicBlock& block);
// Try to extract SPS and PPS NAL units from the specified H.264 |data| stream. If found the data
// will be copied (after resizing) into the provided |sps| and |pps| buffers. Returns whether
// extraction was successful.
bool extractSPSPPS(const uint8_t* data, size_t length, std::vector<uint8_t>* sps,
std::vector<uint8_t>* pps);
// When encoding a video the codec-specific data (CSD; e.g. SPS and PPS for H264 encoding) will be
// concatenated to the first encoded slice. This function extracts the CSD out of the bitstream and
// stores it into |csd|. Returns whether extracting CSD info was successful.
bool extractCSDInfo(std::unique_ptr<C2StreamInitDataInfo::output>* const csd, const uint8_t* data,
size_t length);
// Prepend the specified |sps| and |pps| NAL units (without start codes) to the H.264 |data| stream.
// The result is copied into |dst|. The provided |sps| and |pps| data will be updated if an SPS or
// PPS NAL unit is encountered. Returns the size of the new data, will be 0 if an error occurred.
size_t prependSPSPPSToIDR(const uint8_t* src, size_t srcSize, uint8_t* dst, size_t dstSize,
std::vector<uint8_t>* sps, std::vector<uint8_t>* pps);
} // namespace android
#endif // ANDROID_V4L2_CODEC2_COMMON_HELPERS_H
|
