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     std::shared_ptr<IntfImpl> mIntf;
67     vpx_codec_ctx_t *mCodecCtx;
68     bool mFrameParallelMode;  // Frame parallel is only supported by VP9 decoder.
69 
70     uint32_t mWidth;
71     uint32_t mHeight;
72     bool mSignalledOutputEos;
73     bool mSignalledError;
74 
75     int mCoreCount;
76     struct ConversionQueue {
77         std::list<std::function<void()>> entries;
78         Condition cond;
79         size_t numPending{0u};
80     };
81     std::shared_ptr<Mutexed<ConversionQueue>> mQueue;
82     std::vector<sp<ConverterThread>> mConverterThreads;
83 
84     status_t initDecoder();
85     status_t destroyDecoder();
86     void finishWork(uint64_t index, const std::unique_ptr<C2Work> &work,
87                     const std::shared_ptr<C2GraphicBlock> &block);
88     status_t outputBuffer(
89             const std::shared_ptr<C2BlockPool> &pool,
90             const std::unique_ptr<C2Work> &work);
91     c2_status_t drainInternal(
92             uint32_t drainMode,
93             const std::shared_ptr<C2BlockPool> &pool,
94             const std::unique_ptr<C2Work> &work);
95 
96     C2_DO_NOT_COPY(C2SoftVpxDec);
97 };
98 
99 }  // namespace android
100 
101 #endif  // ANDROID_C2_SOFT_VPX_DEC_H_
102