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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
/*
* Copyright (C) 2018 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef ANDROID_C2_SOFT_HEVC_DEC_H_
#define ANDROID_C2_SOFT_HEVC_DEC_H_
#include <media/stagefright/foundation/ColorUtils.h>
#include <atomic>
#include <SimpleC2Component.h>
#include "ihevc_typedefs.h"
#include "iv.h"
#include "ivd.h"
namespace android {
#define ivdec_api_function ihevcd_cxa_api_function
#define ivdext_create_ip_t ihevcd_cxa_create_ip_t
#define ivdext_create_op_t ihevcd_cxa_create_op_t
#define ivdext_delete_ip_t ihevcd_cxa_delete_ip_t
#define ivdext_delete_op_t ihevcd_cxa_delete_op_t
#define ivdext_ctl_set_num_cores_ip_t ihevcd_cxa_ctl_set_num_cores_ip_t
#define ivdext_ctl_set_num_cores_op_t ihevcd_cxa_ctl_set_num_cores_op_t
#define ivdext_ctl_get_vui_params_ip_t ihevcd_cxa_ctl_get_vui_params_ip_t
#define ivdext_ctl_get_vui_params_op_t ihevcd_cxa_ctl_get_vui_params_op_t
#define ALIGN64(x) ((((x) + 63) >> 6) << 6)
#define MAX_NUM_CORES 4
#define IVDEXT_CMD_CTL_SET_NUM_CORES \
(IVD_CONTROL_API_COMMAND_TYPE_T)IHEVCD_CXA_CMD_CTL_SET_NUM_CORES
#define MIN(a, b) (((a) < (b)) ? (a) : (b))
#define GETTIME(a, b) gettimeofday(a, b);
#define TIME_DIFF(start, end, diff) \
diff = (((end).tv_sec - (start).tv_sec) * 1000000) + \
((end).tv_usec - (start).tv_usec);
struct C2SoftHevcDec : public SimpleC2Component {
class IntfImpl;
C2SoftHevcDec(const char* name, c2_node_id_t id,
const std::shared_ptr<IntfImpl>& intfImpl);
virtual ~C2SoftHevcDec();
// From SimpleC2Component
c2_status_t onInit() override;
c2_status_t onStop() override;
void onReset() override;
void onRelease() override;
c2_status_t onFlush_sm() override;
void process(
const std::unique_ptr<C2Work> &work,
const std::shared_ptr<C2BlockPool> &pool) override;
c2_status_t drain(
uint32_t drainMode,
const std::shared_ptr<C2BlockPool> &pool) override;
private:
status_t createDecoder();
status_t setNumCores();
status_t setParams(size_t stride, IVD_VIDEO_DECODE_MODE_T dec_mode);
status_t getVersion();
status_t initDecoder();
bool setDecodeArgs(ivd_video_decode_ip_t *ps_decode_ip,
ivd_video_decode_op_t *ps_decode_op,
C2ReadView *inBuffer,
C2GraphicView *outBuffer,
size_t inOffset,
size_t inSize,
uint32_t tsMarker);
bool getVuiParams();
// TODO:This is not the right place for colorAspects functions. These should
// be part of c2-vndk so that they can be accessed by all video plugins
// until then, make them feel at home
bool colorAspectsDiffer(const ColorAspects &a, const ColorAspects &b);
void updateFinalColorAspects(
const ColorAspects &otherAspects, const ColorAspects &preferredAspects);
status_t handleColorAspectsChange();
c2_status_t ensureDecoderState(const std::shared_ptr<C2BlockPool> &pool);
void finishWork(uint64_t index, const std::unique_ptr<C2Work> &work);
status_t setFlushMode();
c2_status_t drainInternal(
uint32_t drainMode,
const std::shared_ptr<C2BlockPool> &pool,
const std::unique_ptr<C2Work> &work);
status_t resetDecoder();
void resetPlugin();
status_t deleteDecoder();
// TODO:This is not the right place for this enum. These should
// be part of c2-vndk so that they can be accessed by all video plugins
// until then, make them feel at home
enum {
kNotSupported,
kPreferBitstream,
kPreferContainer,
};
std::shared_ptr<IntfImpl> mIntf;
iv_obj_t *mDecHandle;
std::shared_ptr<C2GraphicBlock> mOutBlock;
uint8_t *mOutBufferFlush;
size_t mNumCores;
IV_COLOR_FORMAT_T mIvColorformat;
uint32_t mOutputDelay;
uint32_t mWidth;
uint32_t mHeight;
uint32_t mStride;
bool mSignalledOutputEos;
bool mSignalledError;
bool mHeaderDecoded;
std::atomic_uint64_t mOutIndex;
// Color aspects. These are ISO values and are meant to detect changes in aspects to avoid
// converting them to C2 values for each frame
struct VuiColorAspects {
uint8_t primaries;
uint8_t transfer;
uint8_t coeffs;
uint8_t fullRange;
// default color aspects
VuiColorAspects()
: primaries(2), transfer(2), coeffs(2), fullRange(0) { }
bool operator==(const VuiColorAspects &o) {
return primaries == o.primaries && transfer == o.transfer && coeffs == o.coeffs
&& fullRange == o.fullRange;
}
} mBitstreamColorAspects;
// profile
struct timeval mTimeStart;
struct timeval mTimeEnd;
C2_DO_NOT_COPY(C2SoftHevcDec);
};
} // namespace android
#endif // ANDROID_C2_SOFT_HEVC_DEC_H_
|
