1 /*
2  * Copyright (C) 2018 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 #ifndef ANDROID_C2_SOFT_VPX_DEC_H_
18 #define ANDROID_C2_SOFT_VPX_DEC_H_
19 
20 #include <SimpleC2Component.h>
21 
22 
23 #include "vpx/vpx_decoder.h"
24 #include "vpx/vp8dx.h"
25 
26 namespace android {
27 
28 struct C2SoftVpxDec : public SimpleC2Component {
29     class IntfImpl;
30 
31     C2SoftVpxDec(const char* name, c2_node_id_t id,
32               const std::shared_ptr<IntfImpl>& intfImpl);
33     virtual ~C2SoftVpxDec();
34 
35     // From SimpleC2Component
36     c2_status_t onInit() override;
37     c2_status_t onStop() override;
38     void onReset() override;
39     void onRelease() override;
40     c2_status_t onFlush_sm() override;
41     void process(
42             const std::unique_ptr<C2Work> &work,
43             const std::shared_ptr<C2BlockPool> &pool) override;
44     c2_status_t drain(
45             uint32_t drainMode,
46             const std::shared_ptr<C2BlockPool> &pool) override;
47  private:
48     enum {
49         MODE_VP8,
50         MODE_VP9,
51     } mMode;
52 
53     struct ConversionQueue;
54 
55     class ConverterThread : public Thread {
56     public:
57         explicit ConverterThread(
58                 const std::shared_ptr<Mutexed<ConversionQueue>> &queue);
59         ~ConverterThread() override = default;
60         bool threadLoop() override;
61 
62     private:
63         std::shared_ptr<Mutexed<ConversionQueue>> mQueue;
64     };
65 
66     // configurations used by component in process
67     // (TODO: keep this in intf but make them internal only)
68     std::shared_ptr<C2StreamPixelFormatInfo::output> mPixelFormatInfo;
69     std::shared_ptr<C2StreamColorAspectsTuning::output> mColorAspects;
70 
71     std::shared_ptr<IntfImpl> mIntf;
72     vpx_codec_ctx_t *mCodecCtx;
73     bool mFrameParallelMode;  // Frame parallel is only supported by VP9 decoder.
74 
75     uint32_t mHalPixelFormat;
76     uint32_t mWidth;
77     uint32_t mHeight;
78     bool mSignalledOutputEos;
79     bool mSignalledError;
80 
81     int mCoreCount;
82     struct ConversionQueue {
83         std::list<std::function<void()>> entries;
84         Condition cond;
85         size_t numPending{0u};
86     };
87     std::shared_ptr<Mutexed<ConversionQueue>> mQueue;
88     std::vector<sp<ConverterThread>> mConverterThreads;
89 
90     status_t initDecoder();
91     status_t destroyDecoder();
92     void finishWork(uint64_t index, const std::unique_ptr<C2Work> &work,
93                     const std::shared_ptr<C2GraphicBlock> &block);
94     status_t outputBuffer(
95             const std::shared_ptr<C2BlockPool> &pool,
96             const std::unique_ptr<C2Work> &work);
97     c2_status_t drainInternal(
98             uint32_t drainMode,
99             const std::shared_ptr<C2BlockPool> &pool,
100             const std::unique_ptr<C2Work> &work);
101 
102     C2_DO_NOT_COPY(C2SoftVpxDec);
103 };
104 
105 }  // namespace android
106 
107 #endif  // ANDROID_C2_SOFT_VPX_DEC_H_
108